codee-agent 0.6.5__tar.gz → 0.6.6__tar.gz

{codee_agent-0.6.5 → codee_agent-0.6.6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: codee-agent
-Version: 0.6.5
+Version: 0.6.6
 Summary: A virtual co-worker that picks up Jira and Azure DevOps tasks and solves them with coding agents.
 Keywords: ai,agent,jira,azure-devops,claude-code,github-copilot,codex,automation
 Author: Denis Kibalko

{codee_agent-0.6.5 → codee_agent-0.6.6}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "codee-agent"
-version = "0.6.5"
+version = "0.6.6"
 description = "A virtual co-worker that picks up Jira and Azure DevOps tasks and solves them with coding agents."
 readme = "README.md"
 license = "MIT"

{codee_agent-0.6.5 → codee_agent-0.6.6}/src/codee/admin.py

@@ -367,6 +367,10 @@ class AdminState(rx.State):
     agent_test_messages: list[ConversationMessage] = []
     agent_test_sending: bool = False
     agent_test_generation: int = 0
+    agent_test_model: str = ""
+    agent_test_models: list[ModelOption] = []
+    agent_test_model_query: str = ""
+    agent_test_models_loading: bool = False
     # Whether the executor runs Claude Code on the connected accounts instead
     # of leaving ~/.claude/.credentials.json alone.
     claude_code_rotate_keys: bool = False
@@ -475,6 +479,32 @@ class AdminState(rx.State):
             return ""
         return query
 
+    @rx.var
+    def filtered_agent_test_models(self) -> list[ModelOption]:
+        query = self.agent_test_model_query.strip().lower()
+        return [
+            model for model in self.agent_test_models
+            if not query or query in f"{model.name} {model.id}".lower()
+        ]
+
+    @rx.var
+    def agent_test_model_label(self) -> str:
+        if not self.agent_test_model:
+            return "Agent default"
+        for model in self.agent_test_models:
+            if model.id == self.agent_test_model:
+                return model.name
+        return self.agent_test_model
+
+    @rx.var
+    def custom_agent_test_model_query(self) -> str:
+        query = self.agent_test_model_query.strip()
+        if not query or any(
+            model.id == query for model in self.agent_test_models
+        ):
+            return ""
+        return query
+
     @rx.var
     def active_route(self) -> str:
         return self.router.url.path.rstrip("/") or "/"
@@ -1527,13 +1557,17 @@ class AdminState(rx.State):
         error = self._persist_settings()
         return rx.toast.error(error) if error else rx.toast.success("Settings saved")
 
-    def open_agent_test(self) -> None:
+    def open_agent_test(self) -> Any:
         self.agent_test_input = ""
         self.agent_test_session_id = ""
         self.agent_test_messages = []
         self.agent_test_sending = False
         self.agent_test_generation += 1
+        self.agent_test_model = ""
+        self.agent_test_models = []
+        self.agent_test_model_query = ""
         self.agent_test_open = True
+        return AdminState.load_agent_test_models
 
     def set_agent_test_open(self, open_: bool) -> None:
         self.agent_test_open = open_
@@ -1544,6 +1578,28 @@ class AdminState(rx.State):
     def set_agent_test_input(self, value: str) -> None:
         self.agent_test_input = value
 
+    def set_agent_test_model_query(self, value: str) -> None:
+        self.agent_test_model_query = value
+
+    def choose_agent_test_model(self, model_id: str) -> None:
+        self.agent_test_model = model_id.strip()
+        self.agent_test_model_query = ""
+
+    @rx.event(background=True)
+    async def load_agent_test_models(self) -> None:
+        async with self:
+            agent = self.coding_agent
+            self.agent_test_models_loading = True
+        try:
+            models = await asyncio.to_thread(SERVICE.list_agent_models, agent)
+        except Exception:
+            models = []
+        async with self:
+            if self.coding_agent != agent:
+                return
+            self.agent_test_models = [ModelOption(**model) for model in models]
+            self.agent_test_models_loading = False
+
     @rx.event(background=True)
     async def send_agent_test_message(self) -> Any:
         async with self:
@@ -1552,6 +1608,7 @@ class AdminState(rx.State):
                 return
             agent = self.coding_agent
             session_id = self.agent_test_session_id
+            model = self.agent_test_model
             generation = self.agent_test_generation
             self.agent_test_input = ""
             self.agent_test_sending = True
@@ -1565,6 +1622,7 @@ class AdminState(rx.State):
                 agent,
                 message,
                 session_id,
+                model,
             )
         except Exception as error:
             async with self:
@@ -2131,7 +2189,7 @@ def model_menu_item(button: rx.Component) -> rx.Component:
     return rx.popover.close(rx.flex(button, width="100%"), width="100%")
 
 
-def model_option_row(option: ModelOption) -> rx.Component:
+def _model_option_row(option: ModelOption, choose_model: Any) -> rx.Component:
     """One row of the model picker: friendly name left, model code right."""
     return model_menu_item(
         rx.button(
@@ -2143,7 +2201,15 @@ def model_option_row(option: ModelOption) -> rx.Component:
                       align="center", spacing="2", width="100%"),
             variant="ghost", color_scheme="gray", width="100%",
             justify_content="start", padding="0.45rem 0.6rem",
-            on_click=AdminState.choose_model(option.id)))
+            on_click=choose_model(option.id)))
+
+
+def model_option_row(option: ModelOption) -> rx.Component:
+    return _model_option_row(option, AdminState.choose_model)
+
+
+def agent_test_model_option_row(option: ModelOption) -> rx.Component:
+    return _model_option_row(option, AdminState.choose_agent_test_model)
 
 
 def agent_picker() -> rx.Component:
@@ -3309,6 +3375,72 @@ def agent_test_dialog() -> rx.Component:
             rx.dialog.description(
                 "Chat with the selected coding agent in the Codee project.",
                 color=MUTED),
+            field(
+                "Model",
+                rx.popover.root(
+                    rx.popover.trigger(
+                        rx.button(
+                            rx.hstack(
+                                rx.text(AdminState.agent_test_model_label),
+                                rx.spacer(),
+                                rx.icon("chevrons-up-down", size=14),
+                                align="center", width="100%"),
+                            variant="surface", color_scheme="gray",
+                            width="100%", type="button")),
+                    rx.popover.content(
+                        rx.vstack(
+                            rx.input(
+                                placeholder=(
+                                    "Search models, or type a model code"),
+                                value=AdminState.agent_test_model_query,
+                                on_change=AdminState.set_agent_test_model_query,
+                                auto_focus=True, width="100%"),
+                            rx.cond(
+                                AdminState.custom_agent_test_model_query != "",
+                                model_menu_item(
+                                    rx.button(
+                                        rx.hstack(
+                                            rx.icon("plus", size=14),
+                                            rx.text("Use "),
+                                            rx.code(
+                                                AdminState.custom_agent_test_model_query),
+                                            align="center", spacing="2"),
+                                        variant="soft", width="100%",
+                                        justify_content="start",
+                                        padding="0.45rem 0.6rem",
+                                        on_click=AdminState.choose_agent_test_model(
+                                            AdminState.custom_agent_test_model_query)))),
+                            rx.scroll_area(
+                                rx.vstack(
+                                    model_menu_item(
+                                        rx.button(
+                                            "Agent default", variant="ghost",
+                                            color_scheme="gray", width="100%",
+                                            justify_content="start",
+                                            padding="0.45rem 0.6rem",
+                                            on_click=AdminState.choose_agent_test_model(""))),
+                                    rx.foreach(
+                                        AdminState.filtered_agent_test_models,
+                                        agent_test_model_option_row),
+                                    rx.cond(
+                                        AdminState.agent_test_models_loading,
+                                        rx.text(
+                                            "Loading models from the coding agent…",
+                                            color=MUTED, font_size="0.8rem",
+                                            padding="0.5rem")),
+                                    spacing="1", width="100%"),
+                                type="auto", scrollbars="vertical",
+                                max_height="15rem", width="100%"),
+                            spacing="2", width="100%"),
+                        width="24rem", max_width="calc(100vw - 3rem)")),
+                rx.text(
+                    rx.cond(
+                        AdminState.agent_test_model == "",
+                        "Runs on whatever that agent defaults to.",
+                        rx.fragment("Uses ",
+                                    rx.code(AdminState.agent_test_model),
+                                    " for this conversation.")),
+                    color=MUTED, font_size="0.82rem")),
             rx.scroll_area(
                 rx.vstack(
                     rx.cond(
@@ -3374,7 +3506,7 @@ def settings_page() -> rx.Component:
             rx.heading("Coding agent", size="4", margin_bottom="1rem"),
             rx.vstack(
                 field("Default agent",
-                      rx.hstack(
+                      rx.grid(
                           rx.select(
                               ["claude_code", "github_copilot", "codex"],
                               value=AdminState.coding_agent,
@@ -3384,7 +3516,10 @@ def settings_page() -> rx.Component:
                                     "Test conversation", variant="outline",
                                     white_space="nowrap",
                                     on_click=AdminState.open_agent_test),
-                          width="100%", spacing="3")),
+                          grid_template_columns=rx.breakpoints(
+                              initial="minmax(0, 1fr)",
+                              md="minmax(0, 1fr) auto"),
+                          width="100%", gap="0.75rem")),
                 field("Max parallel tasks",
                       rx.input(value=AdminState.max_parallel_agents,
                                on_change=AdminState.set_max_parallel_agents,

{codee_agent-0.6.5 → codee_agent-0.6.6}/src/codee/admin_service.py

@@ -2444,7 +2444,8 @@ class AdminService:
             raise RuntimeError(str(exc)) from exc
 
     def test_agent_conversation(
-        self, agent_code: str, user_message: str, session_id: str = ""
+        self, agent_code: str, user_message: str, session_id: str = "",
+        model: str = "",
     ) -> tuple[str, str]:
         """Run one turn with the selected agent and return reply plus session id."""
         selected = resolve_agent_code(agent_code)
@@ -2459,10 +2460,10 @@ class AdminService:
 
         if session_id:
             response = agent.continue_conversation(
-                user_message, session_id, on_session_id=opened)
+                user_message, session_id, model, on_session_id=opened)
         else:
             response = agent.run(
-                user_message, actual_session_id, on_session_id=opened)
+                user_message, actual_session_id, model, on_session_id=opened)
         return response, actual_session_id
 
     def setup_tasks_mcp(

{codee_agent-0.6.5 → codee_agent-0.6.6}/src/codee/test_admin_service.py

@@ -182,23 +182,27 @@ class TestAgentConversationTest(unittest.TestCase):
         agent.run.side_effect = run
 
         response, session_id = self.service.test_agent_conversation(
-            "codex", "Hello")
+            "codex", "Hello", model="gpt-6-astra")
 
         self.assertEqual(response, "Final response")
         self.assertEqual(session_id, "agent-thread")
         build_agent.assert_called_once_with(
             self.service.context.settings, Path("/repo"), CodingAgent.CODEX)
+        self.assertEqual(agent.run.call_args.args[2], "gpt-6-astra")
 
     @patch("codee.admin_service.build_coding_agent")
     def test_later_turn_resumes_the_same_session(self, build_agent) -> None:
         build_agent.return_value.continue_conversation.return_value = "Still here"
 
         response, session_id = self.service.test_agent_conversation(
-            "github_copilot", "What did I ask?", "same-thread")
+            "github_copilot", "What did I ask?", "same-thread",
+            "claude-opus-5")
 
         self.assertEqual((response, session_id), ("Still here", "same-thread"))
         call = build_agent.return_value.continue_conversation.call_args
-        self.assertEqual(call.args[:2], ("What did I ask?", "same-thread"))
+        self.assertEqual(
+            call.args[:3],
+            ("What did I ask?", "same-thread", "claude-opus-5"))
 
 
 class AdminServiceWorkItemsTest(unittest.TestCase):