update

app/core/tools/file_search_tool.py

@@ -47,7 +47,7 @@ class FileSearchTool(BaseTool):
         self.__keys = []
         self.__dirkeys = []
         self.loop = None
-
+'''
     def configure(self, session: Session, run: Run, **kwargs):
         # 获取当前事件循环
         # document_id = []
@@ -166,6 +166,25 @@ class FileSearchTool(BaseTool):
         return files
         """
 
+    def instruction_supplement(self) -> str:
+        """
+        为 Retrieval 提供文件选择信息，用于 llm 调用抉择
+        """
+        # if len(self.__filenames) == 0:
+        #    return ""
+        # else:
+        filenames_info = [
+            f"({index}){filename}" for index, filename in enumerate(self.__filenames)
+        ]
+        return (
+            'You can use the "retrieval" tool to retrieve relevant context from the following attached files. '
+            + 'Each line represents a file in the format "(index)filename":\n'
+            + "\n".join(filenames_info)
+            + "\nMake sure to be extremely concise when using attached files. "
+        )
+
+
+
     def instruction_supplement(self) -> str:
         """
         为 Retrieval 提供文件选择信息，用于 llm 调用抉择
@@ -180,9 +199,9 @@ class FileSearchTool(BaseTool):
         return 'You can use the "retrieval" tool to search for relevant information. Please specify the keywords or context you are looking for to retrieve the most relevant content.'
 
     #'You can use the "retrieval" to search for relevant information within the attached files. Please specify the keywords or context you are looking for to retrieve the most relevant content.'
+'''
 
 
-'''
     def configure(self, session: Session, run: Run, **kwargs):
         # 获取当前事件循环
         document_id = []
@@ -295,4 +314,4 @@ class FileSearchTool(BaseTool):
                 + "\n".join(filenames_info)
                 + "\nMake sure to be extremely concise when using attached files. "
             )
-'''
+