fix: 重命名查询参数为 query_text

This commit is contained in:
Wenjie Zhang 2025-06-16 23:07:28 +08:00
parent e104e07cac
commit 22c473d370
2 changed files with 5 additions and 1 deletions

View File

@ -66,7 +66,7 @@ def regist_tool(
class KnowledgeRetrieverModel(BaseModel):
query: str = Field(
query_text: str = Field(
description=(
"查询的关键词,查询的时候,应该尽量以可能帮助回答这个问题的关键词进行查询,"
"不要直接使用用户的原始输入去查询。"

View File

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