From 4cdd2a5e7972de7fe877614edc6055e9859fc923 Mon Sep 17 00:00:00 2001 From: wvivi2023 Date: Tue, 7 Nov 2023 14:30:01 +0800 Subject: [PATCH] enhance search similarity enhance search similarity --- server/knowledge_base/kb_doc_api.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/server/knowledge_base/kb_doc_api.py b/server/knowledge_base/kb_doc_api.py index 831b139..c6b68b3 100644 --- a/server/knowledge_base/kb_doc_api.py +++ b/server/knowledge_base/kb_doc_api.py @@ -30,7 +30,7 @@ def search_docs(query: str = Body(..., description="用户输入", examples=[" if kb is None: return [] # query = "根据国网安徽信通公司安全准入实施要求，" + query - pre_doc = kb.search_docs(query, 1) + pre_doc = kb.search_docs(query, 1, None) print(f"len(pre_doc):{len(pre_doc)}") if len(pre_doc) > 0: print(f"search_docs, len(pre_doc):{len(pre_doc)}")