|
|
@@ -55,9 +55,10 @@ class FileSearchTool(BaseTool):
|
|
|
folder_fileinfo = []
|
|
|
if asst_folder_ids:
|
|
|
for fid in asst_folder_ids:
|
|
|
- folder_fileinfo += self.loop.run_until_complete(
|
|
|
+ fut = self.loop.create_task(
|
|
|
FileService.list_documents(id=fid, offset=0, limit=100)
|
|
|
)
|
|
|
+ folder_fileinfo += self.loop.run_until_complete(fut)
|
|
|
print(folder_fileinfo)
|
|
|
for file in folder_fileinfo:
|
|
|
self.__filenames.append(file.get("title"))
|
|
|
@@ -81,10 +82,12 @@ class FileSearchTool(BaseTool):
|
|
|
file_keys.append(file_key)
|
|
|
|
|
|
print(file_key)
|
|
|
- self.loop = asyncio.get_event_loop()
|
|
|
- files = self.loop.run_until_complete(
|
|
|
+
|
|
|
+ # self.loop = asyncio.get_event_loop()
|
|
|
+ fut = self.loop.create_task(
|
|
|
FileService.search_in_files(query=query, file_keys=file_keys)
|
|
|
)
|
|
|
+ files = self.loop.run_until_complete(fut)
|
|
|
return files
|
|
|
|
|
|
def instruction_supplement(self) -> str:
|
