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"]