|
|
@@ -109,18 +109,21 @@ class FileSearchTool(BaseTool):
|
|
|
print(self.__keys)
|
|
|
|
|
|
def run(self, indexes: List[int], query: str) -> dict:
|
|
|
+
|
|
|
file_keys = []
|
|
|
|
|
|
for index in indexes:
|
|
|
- file_key = self.__keys[index]
|
|
|
- file_keys.append(file_key)
|
|
|
+ if index is not None:
|
|
|
+ file_key = self.__keys[index]
|
|
|
+ file_keys.append(file_key)
|
|
|
|
|
|
print(file_key)
|
|
|
-
|
|
|
- # self.loop = asyncio.get_event_loop()
|
|
|
- files = asyncio.run(
|
|
|
- FileService.search_in_files(query=query, file_keys=file_keys)
|
|
|
- )
|
|
|
+ files = []
|
|
|
+ if len(file_key) > 0:
|
|
|
+ # self.loop = asyncio.get_event_loop()
|
|
|
+ files = asyncio.run(
|
|
|
+ FileService.search_in_files(query=query, file_keys=file_keys)
|
|
|
+ )
|
|
|
return files
|
|
|
|
|
|
def instruction_supplement(self) -> str:
|
