enhance search similarity

2023-11-07 14:30:01 +08:00 · 2023-11-07 14:30:01 +08:00 · 4cdd2a5e79
parent 526c4b52a8
commit 4cdd2a5e79
1 changed files with 1 additions and 1 deletions
--- 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)}")