From 22c473d3702f4f75db1574540969304ccffbc389 Mon Sep 17 00:00:00 2001
From: Wenjie Zhang <xerrors@163.com>
Date: Mon, 16 Jun 2025 23:07:28 +0800
Subject: [PATCH] =?UTF-8?q?fix:=20=E9=87=8D=E5=91=BD=E5=90=8D=E6=9F=A5?=
 =?UTF-8?q?=E8=AF=A2=E5=8F=82=E6=95=B0=E4=B8=BA=20query=5Ftext?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/agents/tools_factory.py | 2 +-
 src/core/knowledgebase.py   | 4 ++++
 2 files changed, 5 insertions(+), 1 deletion(-)

diff --git a/src/agents/tools_factory.py b/src/agents/tools_factory.py
index 5bdc3ef7..71d40f95 100644
--- a/src/agents/tools_factory.py
+++ b/src/agents/tools_factory.py
@@ -66,7 +66,7 @@ def regist_tool(
 
 
 class KnowledgeRetrieverModel(BaseModel):
-    query: str = Field(
+    query_text: str = Field(
         description=(
             "查询的关键词，查询的时候，应该尽量以可能帮助回答这个问题的关键词进行查询，"
             "不要直接使用用户的原始输入去查询。"
diff --git a/src/core/knowledgebase.py b/src/core/knowledgebase.py
index 0cde0921..3e93f6e7 100644
--- a/src/core/knowledgebase.py
+++ b/src/core/knowledgebase.py
@@ -629,6 +629,10 @@ class KnowledgeBase:
         }
 
         def retriever(query_text): # Renamed query to query_text
+            """
+            retriever_params: 检索参数
+            query_text: 查询文本
+            """
             response = self.query(query_text, db_id, **retriever_params) # Use query_text
             return response["results"]