本地知识库搜索字段修改

This commit is contained in:
glide-the 2024-01-25 19:25:04 +08:00 committed by liunux4odoo
parent 89b0d467ea
commit 54ffdcab8b

View File

@ -8,7 +8,7 @@ from configs import TOOL_CONFIG
def search_knowledgebase(query: str, database: str, config: dict): def search_knowledgebase(query: str, database: str, config: dict):
docs = search_docs( docs = search_docs(
query=query, query=query,
knowledge_base_name="samples", knowledge_base_name=database,
top_k=config["top_k"], top_k=config["top_k"],
score_threshold=config["score_threshold"]) score_threshold=config["score_threshold"])
context = "" context = ""