|
@@ -105,19 +105,17 @@ class FileSearchTool(BaseTool):
|
|
|
print(self.__keys)
|
|
|
print(self.__dirkeys)
|
|
|
files = []
|
|
|
- if full_text_search:
|
|
|
- files = FileService.list_chunks(ids=self.__keys)
|
|
|
- pass
|
|
|
- else:
|
|
|
- ## 必须有总结的内容query和才能触发
|
|
|
- if self.index == 0 and query:
|
|
|
- try:
|
|
|
+ try:
|
|
|
+ if full_text_search:
|
|
|
+ files = FileService.list_chunks(ids=self.__keys)
|
|
|
+ elif self.index == 0 and query:
|
|
|
+ ## 必须有总结的内容query和才能触发
|
|
|
files = FileService.search_in_files(
|
|
|
query=query, file_keys=self.__keys, folder_keys=self.__dirkeys
|
|
|
)
|
|
|
self.index = 1
|
|
|
- except Exception as e:
|
|
|
- print(e)
|
|
|
+ except Exception as e:
|
|
|
+ print(e)
|
|
|
# print(files)
|
|
|
return files
|
|
|
|