refactor: delegate to ui_state

Author: damassi

src/agent_chat_cli/core/actions.py

@@ -4,7 +4,6 @@
 from agent_chat_cli.components.messages import RoleType
 from agent_chat_cli.components.chat_history import ChatHistory
 from agent_chat_cli.components.tool_permission_prompt import ToolPermissionPrompt
-from agent_chat_cli.components.model_selection_menu import ModelSelectionMenu
 from agent_chat_cli.utils.logger import log_json
 from agent_chat_cli.utils.save_conversation import save_conversation
 
@@ -16,9 +15,6 @@ class Actions:
     def __init__(self, app: "AgentChatCLIApp") -> None:
         self.app = app
 
-    def quit(self) -> None:
-        self.app.exit()
-
     async def post_user_message(self, message: str) -> None:
         await self.app.renderer.add_message(RoleType.USER, message)
         await self._query(message)
@@ -46,6 +42,9 @@ async def new(self) -> None:
         await self.app.agent_loop.query_queue.put(ControlCommand.NEW_CONVERSATION)
         await self.clear()
 
+    def quit(self) -> None:
+        self.app.exit()
+
     async def respond_to_tool_permission(self, response: str) -> None:
         log_json(
             {
@@ -74,8 +73,7 @@ async def save(self) -> None:
         )
 
     def show_model_menu(self) -> None:
-        model_menu = self.app.query_one(ModelSelectionMenu)
-        model_menu.show()
+        self.app.ui_state.show_model_menu()
 
     async def change_model(self, model: str) -> None:
         await self.app.agent_loop.change_model(model)

src/agent_chat_cli/core/ui_state.py

@@ -7,6 +7,7 @@
 from agent_chat_cli.components.thinking_indicator import ThinkingIndicator
 from agent_chat_cli.components.tool_permission_prompt import ToolPermissionPrompt
 from agent_chat_cli.components.user_input import UserInput
+from agent_chat_cli.components.model_selection_menu import ModelSelectionMenu
 
 if TYPE_CHECKING:
     from agent_chat_cli.app import AgentChatCLIApp
@@ -76,3 +77,7 @@ async def scroll_to_bottom(self) -> None:
         await asyncio.sleep(0.1)
         container = self.app.query_one(VerticalScroll)
         container.scroll_end(animate=False, immediate=True)
+
+    def show_model_menu(self) -> None:
+        model_menu = self.app.query_one(ModelSelectionMenu)
+        model_menu.show()

tests/core/test_actions.py

@@ -254,3 +254,15 @@ async def test_does_not_trigger_thinking(
 
             thinking_indicator = app.query_one(ThinkingIndicator)
             assert thinking_indicator.is_thinking is False
+
+
+class TestActionsShowModelMenu:
+    async def test_delegates_to_ui_state(self, mock_agent_loop, mock_config):
+        from agent_chat_cli.components.model_selection_menu import ModelSelectionMenu
+
+        app = AgentChatCLIApp()
+        async with app.run_test():
+            app.actions.show_model_menu()
+
+            model_menu = app.query_one(ModelSelectionMenu)
+            assert model_menu.is_visible is True

tests/core/test_ui_state.py

@@ -7,6 +7,7 @@
 from agent_chat_cli.components.thinking_indicator import ThinkingIndicator
 from agent_chat_cli.components.tool_permission_prompt import ToolPermissionPrompt
 from agent_chat_cli.components.user_input import UserInput
+from agent_chat_cli.components.model_selection_menu import ModelSelectionMenu
 
 
 @pytest.fixture
@@ -163,3 +164,13 @@ async def test_clear_input(self, mock_agent_loop, mock_config):
             app.ui_state.clear_input()
 
             assert text_area.text == ""
+
+
+class TestUIStateModelMenu:
+    async def test_show_model_menu_makes_visible(self, mock_agent_loop, mock_config):
+        app = AgentChatCLIApp()
+        async with app.run_test():
+            app.ui_state.show_model_menu()
+
+            model_menu = app.query_one(ModelSelectionMenu)
+            assert model_menu.is_visible is True