|
@@ -18,6 +18,10 @@ nest_asyncio.apply()
|
|
|
# asyncio.set_event_loop_policy(asyncio.WindowsSelectorEventLoopPolicy())
|
|
|
# query: str = Field(..., description="query to look up in retrieval")
|
|
|
class FileSearchToolInput(BaseModel):
|
|
|
+ # query: str = Field(
|
|
|
+ # ...,
|
|
|
+ # description="query to look up in retrieval",
|
|
|
+ # )
|
|
|
query: str = Field(
|
|
|
...,
|
|
|
description="query to look up in retrieval",
|
|
@@ -26,9 +30,7 @@ class FileSearchToolInput(BaseModel):
|
|
|
|
|
|
class FileSearchTool(BaseTool):
|
|
|
name: str = "file_search"
|
|
|
- description: str = (
|
|
|
- "Useful for answering questions related to documents, data analysis, specific file content, upload file content or file content."
|
|
|
- )
|
|
|
+ description: str = "根据用户输入的问题从文件中检索相关的内容."
|
|
|
|
|
|
args_schema: Type[BaseModel] = FileSearchToolInput
|
|
|
|