From d7e4f7c6895ec8a85b747e5fc373ea56004b1cb7 Mon Sep 17 00:00:00 2001 From: Dominik Jain Date: Mon, 12 Jan 2026 19:31:59 +0100 Subject: [PATCH] JetBrainsGetSymbolsOverviewTool: Allow to disable compact format for testing --- src/serena/tools/jetbrains_tools.py | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/src/serena/tools/jetbrains_tools.py b/src/serena/tools/jetbrains_tools.py index 6407de7c..aad111c1 100644 --- a/src/serena/tools/jetbrains_tools.py +++ b/src/serena/tools/jetbrains_tools.py @@ -101,6 +101,8 @@ class JetBrainsGetSymbolsOverviewTool(Tool, ToolMarkerSymbolicRead, ToolMarkerOp Retrieves an overview of the top-level symbols within a specified file using the JetBrains backend """ + USE_COMPACT_FORMAT = True + @staticmethod def _transform_symbols_to_compact_format(symbols: list[dict[str, Any]]) -> dict[str, list]: """ @@ -154,6 +156,9 @@ class JetBrainsGetSymbolsOverviewTool(Tool, ToolMarkerSymbolicRead, ToolMarkerOp with JetBrainsPluginClient.from_project(self.project) as client: response_dict = client.get_symbols_overview(relative_path=relative_path, depth=depth) symbols = response_dict["symbols"] - compact_symbols = self._transform_symbols_to_compact_format(symbols) - result = self._to_json(compact_symbols) + if self.USE_COMPACT_FORMAT: + compact_symbols = self._transform_symbols_to_compact_format(symbols) + result = self._to_json(compact_symbols) + else: + result = self._to_json(symbols) return self._limit_length(result, max_answer_chars)