codee-agent 0.5.3__tar.gz → 0.5.4__tar.gz

{codee_agent-0.5.3 → codee_agent-0.5.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: codee-agent
-Version: 0.5.3
+Version: 0.5.4
 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,automation
 Author: Denis Kibalko

{codee_agent-0.5.3 → codee_agent-0.5.4}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "codee-agent"
-version = "0.5.3"
+version = "0.5.4"
 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.5.3 → codee_agent-0.5.4}/src/codee/admin_service.py

@@ -1397,7 +1397,9 @@ class AdminService:
                           "fill in every field above.")
         try:
             agent = self._build_coding_agent()
-            response = agent.run(_mcp_check_prompt(server_name, steps), session_id)
+            response = agent.run(
+                _mcp_check_prompt(server_name, steps), session_id,
+                agent.best_model())
         except Exception as exc:
             return _check(MCP_CHECK, False, f"The coding agent failed: {exc}")
         return _mcp_check_result(server_name, response)

{codee_agent-0.5.3 → codee_agent-0.5.4}/src/codee/test_admin_service.py

@@ -1436,6 +1436,7 @@ class VerifyTasksMcpCheckTest(unittest.TestCase):
                 "jira", self.JIRA_CREDENTIALS))
         self.assertEqual([check["name"] for check in checks],
                          [TASKS_CHECK, MCP_CHECK])
+        self.agent = agent
         self.prompt = agent.run.call_args.args[0] if agent.run.called else ""
         return checks[1]
 
@@ -1462,6 +1463,42 @@ class VerifyTasksMcpCheckTest(unittest.TestCase):
             self.assertIn("CORE-42", check["message"])
             self.assertIn("Done", check["message"])
 
+    def test_azure_devops_mcp_check_uses_the_agents_best_model(self) -> None:
+        class AzureProvider:
+            MCP_SERVER_NAME = "ado"
+
+            def mcp_check_steps(self, summary: str) -> list[str]:
+                return [f"Create {summary}"]
+
+        with tempfile.TemporaryDirectory() as temporary_directory:
+            service = self._service(Path(temporary_directory))
+            agent = Mock(**{
+                "best_model.return_value": "claude-opus-5",
+                "run.return_value": '{"ok": true, "task": "42"}',
+            })
+
+            with patch.object(service, "tasks_mcp_configured",
+                              return_value=True), \
+                    patch.object(AdminService, "_build_coding_agent",
+                                 return_value=agent):
+                check = service._check_tasks_mcp(
+                    "azure_devops", AzureProvider(), blocked=False)
+
+            self.assertTrue(check["ok"], check["message"])
+            agent.best_model.assert_called_once_with()
+            self.assertEqual(agent.run.call_args.args[2], "claude-opus-5")
+
+    def test_jira_mcp_check_uses_the_agents_best_model(self) -> None:
+        with tempfile.TemporaryDirectory() as temporary_directory:
+            service = self._service(Path(temporary_directory))
+
+            self._run(service, '{"ok": true, "task": "CORE-1"}')
+
+            self.agent.best_model.assert_called_once_with()
+            self.assertEqual(
+                self.agent.run.call_args.args[2],
+                self.agent.best_model.return_value)
+
     def test_the_prompt_forbids_every_route_other_than_mcp(self) -> None:
         with tempfile.TemporaryDirectory() as temporary_directory:
             service = self._service(Path(temporary_directory))