diff --git a/src/serena/tools/jetbrains_tools.py b/src/serena/tools/jetbrains_tools.py index 6615206e..9f49efe5 100644 --- a/src/serena/tools/jetbrains_tools.py +++ b/src/serena/tools/jetbrains_tools.py @@ -28,6 +28,8 @@ class JetBrainsFindSymbolTool(Tool, ToolMarkerSymbolicRead, ToolMarkerOptional, Performs a global (or local) search for symbols using the JetBrains backend """ + symbol_dict_grouper = JetBrainsApi.find_symbol_grouper_ + def apply( self, name_path_pattern: str, @@ -215,6 +217,8 @@ class JetBrainsFindReferencingSymbolsTool(Tool, ToolMarkerSymbolicRead, ToolMark Finds symbols that reference the given symbol using the JetBrains backend """ + symbol_dict_grouper = JetBrainsApi.references_grouper_ + def apply( self, name_path: str, @@ -241,6 +245,8 @@ class JetBrainsGetSymbolsOverviewTool(Tool, ToolMarkerSymbolicRead, ToolMarkerOp Retrieves an overview of the top-level symbols within a specified file using the JetBrains backend """ + symbol_dict_grouper = JetBrainsApi.overview_grouper_ + def apply( self, relative_path: str, diff --git a/src/serena/tools/symbol_tools.py b/src/serena/tools/symbol_tools.py index 63115f9f..b6df9c2c 100644 --- a/src/serena/tools/symbol_tools.py +++ b/src/serena/tools/symbol_tools.py @@ -43,6 +43,8 @@ class GetSymbolsOverviewTool(Tool, ToolMarkerSymbolicRead, LspApiMixin): Gets an overview of the top-level symbols defined in a given file. """ + symbol_dict_grouper = LspApi.overview_grouper_ + def apply(self, relative_path: str, depth: int = -1, max_answer_chars: int = -1) -> str: """ Use this tool to get a high-level understanding of the code symbols in a file. @@ -65,6 +67,8 @@ class FindSymbolTool(Tool, ToolMarkerSymbolicRead, LspApiMixin): Performs a global (or local) search using the language server backend. """ + symbol_dict_grouper = LspApi.find_symbol_dict_grouper_ + def apply( self, name_path_pattern: str, @@ -144,6 +148,8 @@ class FindReferencingSymbolsTool(Tool, ToolMarkerSymbolicRead, LspApiMixin): Finds symbols that reference the given symbol """ + symbol_dict_grouper = LspApi.references_grouper_ + def apply( self, name_path: str,