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