jack 2 달 전
부모
커밋
05bd26eed9
1개의 변경된 파일5개의 추가작업 그리고 3개의 파일을 삭제
  1. 5 3
      app/core/tools/file_search_tool.py

+ 5 - 3
app/core/tools/file_search_tool.py

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