From 5554c45b9c937a01067834c6c54e597d42b4a750 Mon Sep 17 00:00:00 2001 From: Your Name Date: Sun, 26 Apr 2026 00:28:28 -0700 Subject: [PATCH 01/13] feat: Save and load MCP, tool, and skill states in sessions Co-authored-by: cecli (openai/gemini_cli_local/gemini-2.5-pro) --- cecli/commands/load_session.py | 2 +- cecli/main.py | 2 +- cecli/sessions.py | 68 ++++++++++++++++++++++++++++++++-- 3 files changed, 67 insertions(+), 5 deletions(-) diff --git a/cecli/commands/load_session.py b/cecli/commands/load_session.py index 1d5676d97e9..f3c38396a8b 100644 --- a/cecli/commands/load_session.py +++ b/cecli/commands/load_session.py @@ -18,7 +18,7 @@ async def execute(cls, io, coder, args, **kwargs): from cecli import sessions session_manager = sessions.SessionManager(coder, io) - session_manager.load_session(args.strip()) + await session_manager.load_session(args.strip()) return format_command_result(io, "load-session", f"Loaded session: {args.strip()}") diff --git a/cecli/main.py b/cecli/main.py index 76b92cb0870..077fb114b34 100644 --- a/cecli/main.py +++ b/cecli/main.py @@ -1301,7 +1301,7 @@ def apply_model_overrides(model_name): from cecli.sessions import SessionManager session_manager = SessionManager(coder, io) - session_manager.load_session( + await session_manager.load_session( args.auto_save_session_name if args.auto_save_session_name else "auto-save" ) except Exception: diff --git a/cecli/sessions.py b/cecli/sessions.py index 5d8447d5213..cb361810696 100644 --- a/cecli/sessions.py +++ b/cecli/sessions.py @@ -7,6 +7,7 @@ from cecli import models from cecli.helpers.conversation import ConversationService, MessageTag +from cecli.hooks.manager import HookManager class SessionManager: @@ -88,7 +89,7 @@ def list_sessions(self) -> List[Dict]: return sessions - def load_session(self, session_identifier: str) -> bool: + async def load_session(self, session_identifier: str) -> bool: """Load a saved session by name or file path.""" if not session_identifier: self.io.tool_error("Please provide a session name or file path.") @@ -112,7 +113,17 @@ def load_session(self, session_identifier: str) -> bool: return False # Apply session data - return self._apply_session_data(session_data, session_file) + applied = await self._apply_session_data(session_data, session_file) + if applied: + from cecli.commands import SwitchCoderSignal + + raise SwitchCoderSignal( + edit_format=self.coder.edit_format, + from_coder=self.coder, + summarize_from_coder=False, + show_announcements=True, + ) + return applied def _build_session_data(self, session_name) -> Dict: """Build session data dictionary from current coder state.""" @@ -140,6 +151,20 @@ def _build_session_data(self, session_name) -> Dict: self.io.tool_warning(f"Could not read todo list file: {e}") # Get CUR and DONE messages from ConversationManager + connected_mcps = [] + if hasattr(self.coder, "mcp_manager") and self.coder.mcp_manager: + connected_mcps = [server.name for server in self.coder.mcp_manager.connected_servers] + + hook_manager = HookManager() + enabled_hooks = [hook.name for hook in hook_manager.get_hooks()] + + skills_data = None + if hasattr(self.coder, "skills_manager") and self.coder.skills_manager: + skills_data = { + "include": self.coder.skills_manager.include_list, + "exclude": self.coder.skills_manager.exclude_list, + } + return { "version": 1, "session_name": session_name, @@ -168,6 +193,9 @@ def _build_session_data(self, session_name) -> Dict: "auto_test": self.coder.auto_test, }, "todo_list": todo_content, + "mcps": connected_mcps, + "hooks": enabled_hooks, + "skills": skills_data, } def _find_session_file(self, session_identifier: str) -> Optional[Path]: @@ -194,7 +222,7 @@ def _find_session_file(self, session_identifier: str) -> Optional[Path]: self.io.tool_output("Use /list-sessions to see available sessions.") return None - def _apply_session_data(self, session_data: Dict, session_file: Path) -> bool: + async def _apply_session_data(self, session_data: Dict, session_file: Path) -> bool: """Apply session data to current coder state.""" try: # Clear current state @@ -303,6 +331,40 @@ def _apply_session_data(self, session_data: Dict, session_file: Path) -> bool: ) self.io.tool_output(f"Loaded {num_messages} messages and {num_files} files") + # Load MCPs + saved_mcps = session_data.get("mcps", []) + if hasattr(self.coder, "mcp_manager") and self.coder.mcp_manager: + current_mcps = {server.name for server in self.coder.mcp_manager.connected_servers} + saved_mcps_set = set(saved_mcps) + + to_disconnect = current_mcps - saved_mcps_set + for mcp_name in to_disconnect: + await self.coder.mcp_manager.disconnect_server(mcp_name) + + to_connect = saved_mcps_set - current_mcps + for mcp_name in to_connect: + await self.coder.mcp_manager.connect_server(mcp_name) + + # Load hooks + hook_manager = HookManager() + saved_hooks = session_data.get("hooks", []) + # Disable all hooks first + for hook in hook_manager.get_hooks(): + hook_manager.disable_hook(hook.name) + # Enable saved hooks + for hook_name in saved_hooks: + hook_manager.enable_hook(hook_name) + + # Load skills + skills_data = session_data.get("skills") + if ( + skills_data + and hasattr(self.coder, "skills_manager") + and self.coder.skills_manager + ): + self.coder.skills_manager.include_list = skills_data.get("include", []) + self.coder.skills_manager.exclude_list = skills_data.get("exclude", []) + return True except Exception as e: From 9213b2f026dbc6ab7d1b7e62f0ac77d8e8cc43e8 Mon Sep 17 00:00:00 2001 From: Your Name Date: Sun, 26 Apr 2026 14:46:20 -0700 Subject: [PATCH 02/13] fix: Correctly retrieve and manage enabled hooks during session save/load Co-authored-by: cecli (openai/gemini_cli_local/gemini-2.5-pro) --- cecli/sessions.py | 12 +++++++++--- 1 file changed, 9 insertions(+), 3 deletions(-) diff --git a/cecli/sessions.py b/cecli/sessions.py index cb361810696..8bb650b18f5 100644 --- a/cecli/sessions.py +++ b/cecli/sessions.py @@ -156,7 +156,11 @@ def _build_session_data(self, session_name) -> Dict: connected_mcps = [server.name for server in self.coder.mcp_manager.connected_servers] hook_manager = HookManager() - enabled_hooks = [hook.name for hook in hook_manager.get_hooks()] + all_hooks_by_type = hook_manager.get_all_hooks() + enabled_hooks = [] + for hook_type in all_hooks_by_type: + for hook in hook_manager.get_hooks(hook_type): + enabled_hooks.append(hook.name) skills_data = None if hasattr(self.coder, "skills_manager") and self.coder.skills_manager: @@ -349,8 +353,10 @@ async def _apply_session_data(self, session_data: Dict, session_file: Path) -> b hook_manager = HookManager() saved_hooks = session_data.get("hooks", []) # Disable all hooks first - for hook in hook_manager.get_hooks(): - hook_manager.disable_hook(hook.name) + all_hooks_by_type = hook_manager.get_all_hooks() + for hook_type in all_hooks_by_type: + for hook in hook_manager.get_hooks(hook_type): + hook_manager.disable_hook(hook.name) # Enable saved hooks for hook_name in saved_hooks: hook_manager.enable_hook(hook_name) From 999c121e624656ead9bea4a0caaf0a0f32eca55f Mon Sep 17 00:00:00 2001 From: Your Name Date: Sun, 26 Apr 2026 21:09:04 -0700 Subject: [PATCH 03/13] refactor: Save and load agent tools configuration in session Co-authored-by: cecli (openai/gemini_cli_local/gemini-2.5-pro) --- cecli/sessions.py | 42 ++++++++++++++++++++++-------------------- 1 file changed, 22 insertions(+), 20 deletions(-) diff --git a/cecli/sessions.py b/cecli/sessions.py index 8bb650b18f5..5e9ddbc4c11 100644 --- a/cecli/sessions.py +++ b/cecli/sessions.py @@ -7,7 +7,6 @@ from cecli import models from cecli.helpers.conversation import ConversationService, MessageTag -from cecli.hooks.manager import HookManager class SessionManager: @@ -155,12 +154,10 @@ def _build_session_data(self, session_name) -> Dict: if hasattr(self.coder, "mcp_manager") and self.coder.mcp_manager: connected_mcps = [server.name for server in self.coder.mcp_manager.connected_servers] - hook_manager = HookManager() - all_hooks_by_type = hook_manager.get_all_hooks() - enabled_hooks = [] - for hook_type in all_hooks_by_type: - for hook in hook_manager.get_hooks(hook_type): - enabled_hooks.append(hook.name) + # Get CUR and DONE messages from ConversationManager + connected_mcps = [] + if hasattr(self.coder, "mcp_manager") and self.coder.mcp_manager: + connected_mcps = [server.name for server in self.coder.mcp_manager.connected_servers] skills_data = None if hasattr(self.coder, "skills_manager") and self.coder.skills_manager: @@ -169,6 +166,14 @@ def _build_session_data(self, session_name) -> Dict: "exclude": self.coder.skills_manager.exclude_list, } + agent_config_data = None + if hasattr(self.coder, "agent_config"): + agent_config_data = { + "tools_paths": self.coder.agent_config.get("tools_paths", []), + "tools_includelist": self.coder.agent_config.get("tools_includelist", []), + "tools_excludelist": self.coder.agent_config.get("tools_excludelist", []), + } + return { "version": 1, "session_name": session_name, @@ -198,8 +203,8 @@ def _build_session_data(self, session_name) -> Dict: }, "todo_list": todo_content, "mcps": connected_mcps, - "hooks": enabled_hooks, "skills": skills_data, + "agent_config": agent_config_data, } def _find_session_file(self, session_identifier: str) -> Optional[Path]: @@ -349,18 +354,6 @@ async def _apply_session_data(self, session_data: Dict, session_file: Path) -> b for mcp_name in to_connect: await self.coder.mcp_manager.connect_server(mcp_name) - # Load hooks - hook_manager = HookManager() - saved_hooks = session_data.get("hooks", []) - # Disable all hooks first - all_hooks_by_type = hook_manager.get_all_hooks() - for hook_type in all_hooks_by_type: - for hook in hook_manager.get_hooks(hook_type): - hook_manager.disable_hook(hook.name) - # Enable saved hooks - for hook_name in saved_hooks: - hook_manager.enable_hook(hook_name) - # Load skills skills_data = session_data.get("skills") if ( @@ -371,6 +364,15 @@ async def _apply_session_data(self, session_data: Dict, session_file: Path) -> b self.coder.skills_manager.include_list = skills_data.get("include", []) self.coder.skills_manager.exclude_list = skills_data.get("exclude", []) + # Load agent_config for tools + agent_config_data = session_data.get("agent_config") + if agent_config_data and hasattr(self.coder, "agent_config"): + self.coder.agent_config.update(agent_config_data) + from cecli.tools.utils.registry import ToolRegistry + + ToolRegistry.build_registry(agent_config=self.coder.agent_config) + self.coder.loaded_custom_tools = ToolRegistry.loaded_custom_tools + return True except Exception as e: From 25294e8432d31d6e94d394947d3b9c5d0af910ea Mon Sep 17 00:00:00 2001 From: Your Name Date: Mon, 27 Apr 2026 11:05:45 -0700 Subject: [PATCH 04/13] fix: Convert skill lists to serializable format for session saving Co-authored-by: cecli (openai/gemini_cli_local/gemini-2.5-pro) --- cecli/sessions.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/cecli/sessions.py b/cecli/sessions.py index 5e9ddbc4c11..d2a6e785d1e 100644 --- a/cecli/sessions.py +++ b/cecli/sessions.py @@ -162,8 +162,8 @@ def _build_session_data(self, session_name) -> Dict: skills_data = None if hasattr(self.coder, "skills_manager") and self.coder.skills_manager: skills_data = { - "include": self.coder.skills_manager.include_list, - "exclude": self.coder.skills_manager.exclude_list, + "include": list(self.coder.skills_manager.include_list), + "exclude": list(self.coder.skills_manager.exclude_list), } agent_config_data = None @@ -361,8 +361,8 @@ async def _apply_session_data(self, session_data: Dict, session_file: Path) -> b and hasattr(self.coder, "skills_manager") and self.coder.skills_manager ): - self.coder.skills_manager.include_list = skills_data.get("include", []) - self.coder.skills_manager.exclude_list = skills_data.get("exclude", []) + self.coder.skills_manager.include_list = set(skills_data.get("include", [])) + self.coder.skills_manager.exclude_list = set(skills_data.get("exclude", [])) # Load agent_config for tools agent_config_data = session_data.get("agent_config") From a3ac3b50b327a5748d6f32d79b32d3e7e4923a48 Mon Sep 17 00:00:00 2001 From: Your Name Date: Mon, 27 Apr 2026 13:58:14 -0700 Subject: [PATCH 05/13] fix: Handle None skill lists when saving sessions Co-authored-by: cecli (openai/gemini_cli_local/gemini-2.5-pro) --- cecli/sessions.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/cecli/sessions.py b/cecli/sessions.py index d2a6e785d1e..1d9fa5177c4 100644 --- a/cecli/sessions.py +++ b/cecli/sessions.py @@ -162,8 +162,12 @@ def _build_session_data(self, session_name) -> Dict: skills_data = None if hasattr(self.coder, "skills_manager") and self.coder.skills_manager: skills_data = { - "include": list(self.coder.skills_manager.include_list), - "exclude": list(self.coder.skills_manager.exclude_list), + "include": list(self.coder.skills_manager.include_list) + if self.coder.skills_manager.include_list is not None + else [], + "exclude": list(self.coder.skills_manager.exclude_list) + if self.coder.skills_manager.exclude_list is not None + else [], } agent_config_data = None From 59789b52f8baf427aa201f2e1bc13b2e81478ff5 Mon Sep 17 00:00:00 2001 From: Your Name Date: Mon, 27 Apr 2026 16:25:58 -0700 Subject: [PATCH 06/13] refactor: Update session to save skills paths and lists Co-authored-by: cecli (openai/gemini_cli_local/gemini-2.5-pro) --- cecli/sessions.py | 14 ++++++++++---- 1 file changed, 10 insertions(+), 4 deletions(-) diff --git a/cecli/sessions.py b/cecli/sessions.py index 1d9fa5177c4..5469fa76fa2 100644 --- a/cecli/sessions.py +++ b/cecli/sessions.py @@ -162,10 +162,11 @@ def _build_session_data(self, session_name) -> Dict: skills_data = None if hasattr(self.coder, "skills_manager") and self.coder.skills_manager: skills_data = { - "include": list(self.coder.skills_manager.include_list) + "skills_paths": self.coder.skills_manager.directory_paths, + "skills_includelist": list(self.coder.skills_manager.include_list) if self.coder.skills_manager.include_list is not None else [], - "exclude": list(self.coder.skills_manager.exclude_list) + "skills_excludelist": list(self.coder.skills_manager.exclude_list) if self.coder.skills_manager.exclude_list is not None else [], } @@ -365,8 +366,13 @@ async def _apply_session_data(self, session_data: Dict, session_file: Path) -> b and hasattr(self.coder, "skills_manager") and self.coder.skills_manager ): - self.coder.skills_manager.include_list = set(skills_data.get("include", [])) - self.coder.skills_manager.exclude_list = set(skills_data.get("exclude", [])) + self.coder.skills_manager.directory_paths = skills_data.get("skills_paths", []) + self.coder.skills_manager.include_list = set( + skills_data.get("skills_includelist", []) + ) + self.coder.skills_manager.exclude_list = set( + skills_data.get("skills_excludelist", []) + ) # Load agent_config for tools agent_config_data = session_data.get("agent_config") From a435f07162bd20078238767fef322c50d271fc95 Mon Sep 17 00:00:00 2001 From: Your Name Date: Mon, 27 Apr 2026 18:14:15 -0700 Subject: [PATCH 07/13] fix: Convert WindowsPath to string for JSON serialization Co-authored-by: cecli (openai/gemini_cli_local/gemini-2.5-pro) --- cecli/sessions.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/cecli/sessions.py b/cecli/sessions.py index 5469fa76fa2..f31897cd562 100644 --- a/cecli/sessions.py +++ b/cecli/sessions.py @@ -162,7 +162,7 @@ def _build_session_data(self, session_name) -> Dict: skills_data = None if hasattr(self.coder, "skills_manager") and self.coder.skills_manager: skills_data = { - "skills_paths": self.coder.skills_manager.directory_paths, + "skills_paths": [str(p) for p in self.coder.skills_manager.directory_paths], "skills_includelist": list(self.coder.skills_manager.include_list) if self.coder.skills_manager.include_list is not None else [], From 6e4f9eb5d828a774dd3f4cda73f09a19aeda691c Mon Sep 17 00:00:00 2001 From: Your Name Date: Mon, 27 Apr 2026 20:07:56 -0700 Subject: [PATCH 08/13] fix: Rename agent_config to tools in session data Co-authored-by: cecli (openai/gemini_cli_local/gemini-2.5-pro) --- cecli/sessions.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/cecli/sessions.py b/cecli/sessions.py index f31897cd562..17489c9f623 100644 --- a/cecli/sessions.py +++ b/cecli/sessions.py @@ -209,7 +209,7 @@ def _build_session_data(self, session_name) -> Dict: "todo_list": todo_content, "mcps": connected_mcps, "skills": skills_data, - "agent_config": agent_config_data, + "tools": agent_config_data, } def _find_session_file(self, session_identifier: str) -> Optional[Path]: @@ -374,8 +374,8 @@ async def _apply_session_data(self, session_data: Dict, session_file: Path) -> b skills_data.get("skills_excludelist", []) ) - # Load agent_config for tools - agent_config_data = session_data.get("agent_config") + # Load tools config + agent_config_data = session_data.get("tools") if agent_config_data and hasattr(self.coder, "agent_config"): self.coder.agent_config.update(agent_config_data) from cecli.tools.utils.registry import ToolRegistry From d7b6d3b1d6440ed388a2e0a645d4a5b39f55d6bb Mon Sep 17 00:00:00 2001 From: Your Name Date: Mon, 27 Apr 2026 20:19:43 -0700 Subject: [PATCH 09/13] test: add tests for session save/load agent config Co-authored-by: cecli (openai/gemini_cli_local/gemini-2.5-pro) --- tests/basic/test_sessions.py | 115 ++++++++++++++++++++++++++++++++++- 1 file changed, 114 insertions(+), 1 deletion(-) diff --git a/tests/basic/test_sessions.py b/tests/basic/test_sessions.py index adb5a01a907..aafc8105756 100644 --- a/tests/basic/test_sessions.py +++ b/tests/basic/test_sessions.py @@ -7,7 +7,7 @@ from unittest import TestCase, mock from cecli.coders import Coder -from cecli.commands import Commands +from cecli.commands import Commands, SwitchCoderSignal from cecli.helpers.file_searcher import handle_core_files from cecli.io import InputOutput from cecli.models import Model @@ -196,3 +196,116 @@ async def test_preserve_todo_list_deprecated(self): self.assertTrue( any("deprecated" in call[0][0] for call in mock_tool_warning.call_args_list) ) + + async def test_cmd_save_load_session_agent_config(self): + """Test session save/load for agent-specific configs (mcp, skills, tools).""" + with GitTemporaryDirectory(): + # Mock args for AgentCoder + mock_args = mock.MagicMock() + mock_args.agent_config = json.dumps( + { + "tools_paths": ["/test/tools/path"], + "tools_includelist": ["included_tool"], + "tools_excludelist": ["excluded_tool"], + } + ) + # This is needed for the skills manager to be created + mock_args.skills_paths = ["/test/skills/path"] + mock_args.mcp_servers = json.dumps([{"name": "mock_mcp"}]) + mock_args.mcp_servers_files = [] + mock_args.verbose = False + mock_args.debug = False + mock_args.tui = False + mock_args.auto_save_session_name = "auto-save" + mock_args.auto_save = False + mock_args.auto_load = False + mock_args.yes_always_commands = True + mock_args.command_prefix = None + mock_args.file_diffs = True + mock_args.max_reflections = 3 + mock_args.model = "gpt-3.5-turbo" + mock_args.weak_model = None + mock_args.editor_model = None + mock_args.agent_model = None + mock_args.editor_edit_format = None + mock_args.retries = None + mock_args.reasoning_effort = None + mock_args.thinking_tokens = None + mock_args.check_model_accepts_settings = True + mock_args.copy_paste = False + mock_args.hooks = None + + io = InputOutput(pretty=False, fancy_input=False, yes=True) + + # === SAVE SESSION === + coder_to_save = await Coder.create( + self.GPT35, "agent", io, args=mock_args, repo=mock.MagicMock() + ) + commands_to_save = Commands(io, coder_to_save, args=mock_args) + + # Configure state to be saved + await coder_to_save.mcp_manager.connect_server("mock_mcp") + coder_to_save.skills_manager.include_list = {"included_skill"} + coder_to_save.skills_manager.exclude_list = {"excluded_skill"} + coder_to_save.skills_manager.directory_paths = ["/test/skills/path/saved"] + + session_name = "agent_session" + await commands_to_save.execute("save-session", session_name) + + session_file = Path(handle_core_files(".cecli")) / "sessions" / f"{session_name}.json" + self.assertTrue(session_file.exists()) + + with open(session_file, "r", encoding="utf-8") as f: + saved_data = json.load(f) + + # Assert saved data is correct + self.assertEqual(saved_data["mcps"], ["mock_mcp"]) + self.assertEqual(saved_data["skills"]["skills_paths"], ["/test/skills/path/saved"]) + self.assertEqual(saved_data["skills"]["skills_includelist"], ["included_skill"]) + self.assertEqual(saved_data["skills"]["skills_excludelist"], ["excluded_skill"]) + self.assertEqual(saved_data["tools"]["tools_paths"], ["/test/tools/path"]) + self.assertEqual(saved_data["tools"]["tools_includelist"], ["included_tool"]) + self.assertEqual(saved_data["tools"]["tools_excludelist"], ["excluded_tool"]) + + # === LOAD SESSION === + # Create a new coder to load into, ensuring it's a clean slate + coder_to_load_initial = await Coder.create( + self.GPT35, "agent", io, args=mock_args, repo=mock.MagicMock() + ) + commands_to_load = Commands(io, coder_to_load_initial, args=mock_args) + + # Mock ToolRegistry.build_registry to check if it's called + with mock.patch( + "cecli.tools.utils.registry.ToolRegistry.build_registry" + ) as mock_build_registry: + coder_after_load = None + try: + await commands_to_load.execute("load-session", session_name) + except SwitchCoderSignal as e: + # The SwitchCoderSignal is expected, we need to get the new coder from it + coder_after_load = await Coder.create(**e.kwargs) + + self.assertIsNotNone(coder_after_load) + + # Assert loaded state is correct in the new coder instance + connected_mcps = {s.name for s in coder_after_load.mcp_manager.connected_servers} + self.assertIn("mock_mcp", connected_mcps) + + self.assertEqual( + coder_after_load.skills_manager.directory_paths, ["/test/skills/path/saved"] + ) + self.assertEqual(coder_after_load.skills_manager.include_list, {"included_skill"}) + self.assertEqual(coder_after_load.skills_manager.exclude_list, {"excluded_skill"}) + + self.assertEqual( + coder_after_load.agent_config["tools_paths"], ["/test/tools/path"] + ) + self.assertEqual( + coder_after_load.agent_config["tools_includelist"], ["included_tool"] + ) + self.assertEqual( + coder_after_load.agent_config["tools_excludelist"], ["excluded_tool"] + ) + + # Assert that the tool registry was rebuilt + mock_build_registry.assert_called_with(agent_config=coder_after_load.agent_config) From 390fb25f97dd0013e4c92b398ab18c45ee32d0cc Mon Sep 17 00:00:00 2001 From: Your Name Date: Mon, 27 Apr 2026 22:10:33 -0700 Subject: [PATCH 10/13] fixed formatting --- cecli/coders/base_coder.py | 5 ++--- cecli/sessions.py | 37 ++++++++++++++++++------------------- 2 files changed, 20 insertions(+), 22 deletions(-) diff --git a/cecli/coders/base_coder.py b/cecli/coders/base_coder.py index 0b7f847d436..0dd6203e298 100755 --- a/cecli/coders/base_coder.py +++ b/cecli/coders/base_coder.py @@ -1131,9 +1131,8 @@ def _include_in_map(abs_path): "other_files": other_files, "mentioned_fnames": mentioned_fnames, "all_abs_files": all_abs_files, - "read_only_count": ( - len(set(self.abs_read_only_fnames)) - + len(set(self.abs_read_only_stubs_fnames)) + "read_only_count": len(set(self.abs_read_only_fnames)) + len( + set(self.abs_read_only_stubs_fnames) ), } ) diff --git a/cecli/sessions.py b/cecli/sessions.py index 17489c9f623..a70bc098c8d 100644 --- a/cecli/sessions.py +++ b/cecli/sessions.py @@ -71,10 +71,9 @@ def list_sessions(self) -> List[Dict]: "file": session_file, "model": session_data.get("model", "unknown"), "edit_format": session_data.get("edit_format", "unknown"), - "num_messages": ( - len(session_data.get("chat_history", {}).get("done_messages", [])) - + len(session_data.get("chat_history", {}).get("cur_messages", [])) - ), + "num_messages": len( + session_data.get("chat_history", {}).get("done_messages", []) + ) + len(session_data.get("chat_history", {}).get("cur_messages", [])), "num_files": ( len(session_data.get("files", {}).get("editable", [])) + len(session_data.get("files", {}).get("read_only", [])) @@ -163,12 +162,16 @@ def _build_session_data(self, session_name) -> Dict: if hasattr(self.coder, "skills_manager") and self.coder.skills_manager: skills_data = { "skills_paths": [str(p) for p in self.coder.skills_manager.directory_paths], - "skills_includelist": list(self.coder.skills_manager.include_list) - if self.coder.skills_manager.include_list is not None - else [], - "skills_excludelist": list(self.coder.skills_manager.exclude_list) - if self.coder.skills_manager.exclude_list is not None - else [], + "skills_includelist": ( + list(self.coder.skills_manager.include_list) + if self.coder.skills_manager.include_list is not None + else [] + ), + "skills_excludelist": ( + list(self.coder.skills_manager.exclude_list) + if self.coder.skills_manager.exclude_list is not None + else [] + ), } agent_config_data = None @@ -189,11 +192,11 @@ def _build_session_data(self, session_name) -> Dict: "editor_edit_format": self.coder.main_model.editor_edit_format, "edit_format": self.coder.edit_format, "chat_history": { - "done_messages": ( - ConversationService.get_manager(self.coder).get_messages_dict(MessageTag.DONE) + "done_messages": ConversationService.get_manager(self.coder).get_messages_dict( + MessageTag.DONE ), - "cur_messages": ( - ConversationService.get_manager(self.coder).get_messages_dict(MessageTag.CUR) + "cur_messages": ConversationService.get_manager(self.coder).get_messages_dict( + MessageTag.CUR ), }, "files": { @@ -361,11 +364,7 @@ async def _apply_session_data(self, session_data: Dict, session_file: Path) -> b # Load skills skills_data = session_data.get("skills") - if ( - skills_data - and hasattr(self.coder, "skills_manager") - and self.coder.skills_manager - ): + if skills_data and hasattr(self.coder, "skills_manager") and self.coder.skills_manager: self.coder.skills_manager.directory_paths = skills_data.get("skills_paths", []) self.coder.skills_manager.include_list = set( skills_data.get("skills_includelist", []) From fe1b37de636b72cd300de282a29e3545382001d2 Mon Sep 17 00:00:00 2001 From: Your Name Date: Tue, 28 Apr 2026 12:57:12 -0700 Subject: [PATCH 11/13] cli-15: fixed formatting --- cecli/tools/edit_text.py | 18 +++++++++--------- 1 file changed, 9 insertions(+), 9 deletions(-) diff --git a/cecli/tools/edit_text.py b/cecli/tools/edit_text.py index f03fc6d96df..5b4d64f7c3c 100644 --- a/cecli/tools/edit_text.py +++ b/cecli/tools/edit_text.py @@ -56,9 +56,9 @@ class Tool(BaseTool): "type": "string", "enum": ["replace", "delete", "insert"], "description": ( - "The type of operation: 'replace' (replace range with text), " - "'delete' (remove range), or 'insert' (insert text after start_line). " - "Defaults to 'replace'." + "The type of operation: 'replace' (replace range with" + " text), 'delete' (remove range), or 'insert' (insert text" + " after start_line). Defaults to 'replace'." ), }, "text": { @@ -78,8 +78,8 @@ class Tool(BaseTool): "end_line": { "type": "string", "description": ( - 'Hashline format for end line: "{4 char hash}" (without the ' - "braces)" + 'Hashline format for end line: "{4 char hash}" (without the' + " braces)" ), }, }, @@ -179,8 +179,8 @@ def execute( if operation in ("replace", "delete"): if edit_start_line is None: raise ToolError( - f"Edit {edit_index + 1}: 'start_line' parameter is required " - f"for '{operation}' operation" + f"Edit {edit_index + 1}: 'start_line' parameter is required" + f" for '{operation}' operation" ) if edit_end_line is None: raise ToolError( @@ -190,8 +190,8 @@ def execute( if operation == "insert": if edit_start_line is None: raise ToolError( - f"Edit {edit_index + 1}: 'start_line' parameter is required " - "for 'insert' operation" + f"Edit {edit_index + 1}: 'start_line' parameter is required" + " for 'insert' operation" ) # For insert, end_line defaults to start_line edit_end_line = edit_end_line or edit_start_line From 5a613b6473d7364b208facd3dbe5efcce8c164b8 Mon Sep 17 00:00:00 2001 From: Your Name Date: Tue, 28 Apr 2026 20:13:06 -0700 Subject: [PATCH 12/13] cli-15: fixed formatting --- cecli/coders/base_coder.py | 5 +++-- cecli/sessions.py | 15 ++++++++------- 2 files changed, 11 insertions(+), 9 deletions(-) diff --git a/cecli/coders/base_coder.py b/cecli/coders/base_coder.py index 0dd6203e298..0b7f847d436 100755 --- a/cecli/coders/base_coder.py +++ b/cecli/coders/base_coder.py @@ -1131,8 +1131,9 @@ def _include_in_map(abs_path): "other_files": other_files, "mentioned_fnames": mentioned_fnames, "all_abs_files": all_abs_files, - "read_only_count": len(set(self.abs_read_only_fnames)) + len( - set(self.abs_read_only_stubs_fnames) + "read_only_count": ( + len(set(self.abs_read_only_fnames)) + + len(set(self.abs_read_only_stubs_fnames)) ), } ) diff --git a/cecli/sessions.py b/cecli/sessions.py index a70bc098c8d..c1e9fbdc5f3 100644 --- a/cecli/sessions.py +++ b/cecli/sessions.py @@ -71,9 +71,10 @@ def list_sessions(self) -> List[Dict]: "file": session_file, "model": session_data.get("model", "unknown"), "edit_format": session_data.get("edit_format", "unknown"), - "num_messages": len( - session_data.get("chat_history", {}).get("done_messages", []) - ) + len(session_data.get("chat_history", {}).get("cur_messages", [])), + "num_messages": ( + len(session_data.get("chat_history", {}).get("done_messages", [])) + + len(session_data.get("chat_history", {}).get("cur_messages", [])) + ), "num_files": ( len(session_data.get("files", {}).get("editable", [])) + len(session_data.get("files", {}).get("read_only", [])) @@ -192,11 +193,11 @@ def _build_session_data(self, session_name) -> Dict: "editor_edit_format": self.coder.main_model.editor_edit_format, "edit_format": self.coder.edit_format, "chat_history": { - "done_messages": ConversationService.get_manager(self.coder).get_messages_dict( - MessageTag.DONE + "done_messages": ( + ConversationService.get_manager(self.coder).get_messages_dict(MessageTag.DONE) ), - "cur_messages": ConversationService.get_manager(self.coder).get_messages_dict( - MessageTag.CUR + "cur_messages": ( + ConversationService.get_manager(self.coder).get_messages_dict(MessageTag.CUR) ), }, "files": { From 2566f6c4cee68da3479e047c37a3b579aae2ab10 Mon Sep 17 00:00:00 2001 From: Your Name Date: Tue, 28 Apr 2026 20:21:20 -0700 Subject: [PATCH 13/13] cli-15: fixed formatting --- tests/basic/test_sessions.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/tests/basic/test_sessions.py b/tests/basic/test_sessions.py index aafc8105756..aa26f4f5a26 100644 --- a/tests/basic/test_sessions.py +++ b/tests/basic/test_sessions.py @@ -297,9 +297,7 @@ async def test_cmd_save_load_session_agent_config(self): self.assertEqual(coder_after_load.skills_manager.include_list, {"included_skill"}) self.assertEqual(coder_after_load.skills_manager.exclude_list, {"excluded_skill"}) - self.assertEqual( - coder_after_load.agent_config["tools_paths"], ["/test/tools/path"] - ) + self.assertEqual(coder_after_load.agent_config["tools_paths"], ["/test/tools/path"]) self.assertEqual( coder_after_load.agent_config["tools_includelist"], ["included_tool"] )