本地知识库搜索字段修改

2026-01-31 03:03:22 +08:00 · 2024-01-25 19:25:04 +08:00 · 2024-01-25 19:25:04 +08:00 · 54ffdcab8b
commit 54ffdcab8b
parent 89b0d467ea
1 changed files with 1 additions and 1 deletions
--- a/server/agent/tools_factory/search_local_knowledgebase.py
+++ b/server/agent/tools_factory/search_local_knowledgebase.py
@ -8,7 +8,7 @@ from configs import TOOL_CONFIG
 def search_knowledgebase(query: str, database: str, config: dict):
    docs = search_docs(
        query=query,
-        knowledge_base_name="samples",
+        knowledge_base_name=database,
        top_k=config["top_k"],
        score_threshold=config["score_threshold"])
    context = ""