jack 4 месяцев назад
Родитель
Сommit
c056c5f91d
1 измененных файлов с 10 добавлено и 6 удалено
  1. 10 6
      app/core/tools/file_search_tool.py

+ 10 - 6
app/core/tools/file_search_tool.py

@@ -14,11 +14,6 @@ import nest_asyncio
 nest_asyncio.apply()
 
 # asyncio.set_event_loop_policy(asyncio.WindowsSelectorEventLoopPolicy())
-try:
-    loop = asyncio.get_running_loop()  # 检查是否有运行的事件循环
-except RuntimeError:
-    loop = asyncio.get_event_loop()
-    print("事件循环未运行,手动启动")
 
 
 class FileSearchToolInput(BaseModel):
@@ -53,7 +48,11 @@ class FileSearchTool(BaseTool):
         """
         置当前 Retrieval 涉及文件信息
         """
-
+        try:
+            loop = asyncio.get_running_loop()  # 检查是否有运行的事件循环
+        except RuntimeError:
+            loop = asyncio.get_event_loop()
+            print("事件循环未运行,手动启动")
         # 获取当前事件循环
         document_id = []
         file_key = []
@@ -140,6 +139,11 @@ class FileSearchTool(BaseTool):
         print(file_keys)
         files = []
         if len(file_keys) > 0:
+            try:
+                loop = asyncio.get_running_loop()  # 检查是否有运行的事件循环
+            except RuntimeError:
+                loop = asyncio.get_event_loop()
+            print("事件循环未运行,手动启动")
             # self.loop = asyncio.get_event_loop()
             files = loop.run_until_complete(
                 FileService.search_in_files(query=query, file_keys=file_keys)