diff --git a/server/knowledge_base/kb_doc_api.py b/server/knowledge_base/kb_doc_api.py index a38f93a..0341edd 100644 --- a/server/knowledge_base/kb_doc_api.py +++ b/server/knowledge_base/kb_doc_api.py @@ -31,7 +31,7 @@ def search_docs( description="知识库匹配相关度阈值,取值范围在0-1之间," "SCORE越小,相关度越高," "取到1相当于不筛选,建议设置在0.5左右", - ge=0, le=1), + ge=0, le=2), file_name: str = Body("", description="文件名称,支持 sql 通配符"), metadata: dict = Body({}, description="根据 metadata 进行过滤,仅支持一级键"), ) -> List[DocumentWithVSId]: diff --git a/webui_pages/knowledge_base/knowledge_base.py b/webui_pages/knowledge_base/knowledge_base.py index 0dc02b2..61748d6 100644 --- a/webui_pages/knowledge_base/knowledge_base.py +++ b/webui_pages/knowledge_base/knowledge_base.py @@ -330,7 +330,7 @@ def knowledge_base_page(api: ApiRequest, is_lite: bool = None): file_name = selected_rows[0]["file_name"] print(f"选中的file_name:{file_name},kb:{selected_kb}") docs = api.search_kb_docs(knowledge_base_name=selected_kb, file_name=file_name) - print(f"选中的file_name:api.search_kb_docs:{docs}") + #print(f"选中的file_name:api.search_kb_docs:{docs}") if isinstance(docs, list): data = [ {"seq": i + 1, "id": x["id"], "page_content": x["page_content"], "source": x["metadata"].get("source"),