jack 22 小时之前
父节点
当前提交
07dea6d4eb
共有 1 个文件被更改,包括 7 次插入9 次删除
  1. 7 9
      app/core/tools/file_search_tool.py

+ 7 - 9
app/core/tools/file_search_tool.py

@@ -105,19 +105,17 @@ class FileSearchTool(BaseTool):
         print(self.__keys)
         print(self.__dirkeys)
         files = []
-        if full_text_search:
-            files = FileService.list_chunks(ids=self.__keys)
-            pass
-        else:
-            ## 必须有总结的内容query和才能触发
-            if self.index == 0 and query:
-                try:
+        try:
+            if full_text_search:
+                files = FileService.list_chunks(ids=self.__keys)
+            elif self.index == 0 and query:
+                ## 必须有总结的内容query和才能触发
                 files = FileService.search_in_files(
                     query=query, file_keys=self.__keys, folder_keys=self.__dirkeys
                 )
                 self.index = 1
-            except Exception as e:
-                print(e)
+        except Exception as e:
+            print(e)
         # print(files)
         return files