|
|
@@ -103,12 +103,12 @@ class FileSearchTool(BaseTool):
|
|
|
for fid in asst_folder_ids:
|
|
|
future = asyncio.run_coroutine_threadsafe(
|
|
|
FileService.list_documents(id=fid, offset=0, limit=100),
|
|
|
- self.loop
|
|
|
+ self.loop,
|
|
|
)
|
|
|
folder_fileinfo += future.result()
|
|
|
- #folder_fileinfo += asyncio.run(
|
|
|
+ # folder_fileinfo += asyncio.run(
|
|
|
# FileService.list_documents(id=fid, offset=0, limit=100)
|
|
|
- #)
|
|
|
+ # )
|
|
|
print(folder_fileinfo)
|
|
|
for file in folder_fileinfo:
|
|
|
self.__filenames.append(file.get("title"))
|
|
|
@@ -138,13 +138,12 @@ class FileSearchTool(BaseTool):
|
|
|
if len(file_keys) > 0:
|
|
|
# self.loop = asyncio.get_event_loop()
|
|
|
future = asyncio.run_coroutine_threadsafe(
|
|
|
- FileService.search_in_files(query=query, file_keys=file_keys)
|
|
|
- self.loop
|
|
|
- )
|
|
|
+ FileService.search_in_files(query=query, file_keys=file_keys), self.loop
|
|
|
+ )
|
|
|
files += future.result()
|
|
|
- #files = asyncio.run(
|
|
|
+ # files = asyncio.run(
|
|
|
# FileService.search_in_files(query=query, file_keys=file_keys)
|
|
|
- #)
|
|
|
+ # )
|
|
|
return files
|
|
|
|
|
|
def instruction_supplement(self) -> str:
|
