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