fix: 重命名查询参数为 query_text
This commit is contained in:
parent
e104e07cac
commit
22c473d370
@ -66,7 +66,7 @@ def regist_tool(
|
||||
|
||||
|
||||
class KnowledgeRetrieverModel(BaseModel):
|
||||
query: str = Field(
|
||||
query_text: str = Field(
|
||||
description=(
|
||||
"查询的关键词,查询的时候,应该尽量以可能帮助回答这个问题的关键词进行查询,"
|
||||
"不要直接使用用户的原始输入去查询。"
|
||||
|
||||
@ -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"]
|
||||
|
||||
|
||||
Loading…
Reference in New Issue
Block a user