From 54ffdcab8b0df572e5348309c295a24b4a54b3a4 Mon Sep 17 00:00:00 2001
From: glide-the <2533736852@qq.com>
Date: Thu, 25 Jan 2024 19:25:04 +0800
Subject: [PATCH] =?UTF-8?q?=E6=9C=AC=E5=9C=B0=E7=9F=A5=E8=AF=86=E5=BA=93?=
 =?UTF-8?q?=E6=90=9C=E7=B4=A2=E5=AD=97=E6=AE=B5=E4=BF=AE=E6=94=B9?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 server/agent/tools_factory/search_local_knowledgebase.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/server/agent/tools_factory/search_local_knowledgebase.py b/server/agent/tools_factory/search_local_knowledgebase.py
index c57a1cf4..eca3709c 100644
--- 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 = ""