codee-agent 0.4.1__tar.gz → 0.4.2__tar.gz

{codee_agent-0.4.1 → codee_agent-0.4.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: codee-agent
-Version: 0.4.1
+Version: 0.4.2
 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.4.1 → codee_agent-0.4.2}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "codee-agent"
-version = "0.4.1"
+version = "0.4.2"
 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.4.1 → codee_agent-0.4.2}/src/codee_agent_github_copilot/provider.py

@@ -1,4 +1,5 @@
 import json
+import os
 import queue
 import subprocess
 import threading
@@ -25,13 +26,31 @@ ACP_TIMEOUT_SECONDS = 60
 _INITIALIZE_ID = 1
 _SESSION_NEW_ID = 2
 
+# Ceiling on what one run may spend. AI credits bill at $0.04 each, so the
+# default is the same $20 cap the Claude Code agent puts on a run.
+MAX_AI_CREDITS_ENV_VAR = "CODEE_COPILOT_MAX_AI_CREDITS"
+DEFAULT_MAX_AI_CREDITS = 1000
+# The CLI rejects anything lower outright, which reads as a broken agent rather
+# than a misconfigured cap, so a smaller override is raised to it instead.
+MIN_AI_CREDITS = 30
+
+
+def _max_ai_credits() -> str:
+    """The per-run credit cap, read fresh so a changed env applies to the next run."""
+    raw = os.environ.get(MAX_AI_CREDITS_ENV_VAR, "").strip()
+    if not raw:
+        return str(DEFAULT_MAX_AI_CREDITS)
+    try:
+        return str(max(MIN_AI_CREDITS, int(raw)))
+    except ValueError:
+        log.warning("%s=%r is not a number, using %d",
+                    MAX_AI_CREDITS_ENV_VAR, raw, DEFAULT_MAX_AI_CREDITS)
+        return str(DEFAULT_MAX_AI_CREDITS)
+
 
 class GitHubCopilotAgent(AbstractCodingAgent):
     """Runs the ``copilot`` CLI in a fresh session, under the id the caller supplies."""
 
-    # Ceiling on what one run may spend. AI credits bill at $0.04 each, so this
-    # is the same $20 cap the Claude Code agent puts on a run. Minimum is 30.
-    MAX_AI_CREDITS = "500"
     TIMEOUT_SECONDS = 7200  # 2 hours
 
     def __init__(self, settings: Settings, cwd: Path):
@@ -52,7 +71,7 @@ class GitHubCopilotAgent(AbstractCodingAgent):
             "copilot",
             "-p", user_message,
             "--session-id", session_id,
-            "--max-ai-credits", self.MAX_AI_CREDITS,
+            "--max-ai-credits", _max_ai_credits(),
             "--output-format", "json",
             # Tools, paths and URLs: the equivalent of claude's bypassPermissions.
             "--allow-all",
@@ -95,7 +114,8 @@ class GitHubCopilotAgent(AbstractCodingAgent):
         if outcome is None:
             # Not the JSONL we know how to read — hand back whatever it printed
             # rather than failing a run that the CLI itself called successful.
-            log.warning("copilot produced no result event; returning raw output")
+            log.warning(
+                "copilot produced no result event; returning raw output")
             return result.stdout
         if outcome.get("exitCode"):
             raise RuntimeError(
@@ -240,7 +260,8 @@ def _parse_events(stdout: str) -> tuple[str, dict | None, list[str]]:
                 reply = content.strip()
         elif kind == SESSION_ERROR:
             data = event.get("data") or {}
-            message = data.get("message") or data.get("errorType") or "unknown error"
+            message = data.get("message") or data.get(
+                "errorType") or "unknown error"
             errors.append(str(message))
         elif kind == RESULT:
             outcome = event

{codee_agent-0.4.1 → codee_agent-0.4.2}/src/codee_agent_github_copilot/test.py

@@ -1,11 +1,13 @@
 import json
+import os
 import queue
 import subprocess
 import unittest
 from pathlib import Path
 from unittest.mock import Mock, patch
 
-from codee_agent_github_copilot.provider import GitHubCopilotAgent, _await_result
+from codee_agent_github_copilot.provider import (
+    MAX_AI_CREDITS_ENV_VAR, GitHubCopilotAgent, _await_result)
 from codee_main_context.context import Settings
 
 SESSION = "82232f47-df60-4cb3-8c3a-de12074c9205"
@@ -123,6 +125,36 @@ class CopilotRunTest(unittest.TestCase):
                 self.agent.run("/do-it CORE-1", SESSION)
 
 
+class CopilotCreditCapTest(unittest.TestCase):
+    def _credits(self, value: str | None) -> str:
+        stdout = _stream(_event("assistant.message", {"content": "ok"}),
+                         _event("result", exitCode=0))
+        with patch.dict(os.environ, {}, clear=False):
+            if value is None:
+                os.environ.pop(MAX_AI_CREDITS_ENV_VAR, None)
+            else:
+                os.environ[MAX_AI_CREDITS_ENV_VAR] = value
+            with patch("subprocess.run", return_value=_completed(stdout)) as run:
+                GitHubCopilotAgent(Settings(), Path("/repo")).run("/do-it", SESSION)
+        cmd = run.call_args.args[0]
+        return cmd[cmd.index("--max-ai-credits") + 1]
+
+    def test_an_unset_env_var_keeps_the_default_cap(self) -> None:
+        self.assertEqual(self._credits(None), "1000")
+
+    def test_the_env_var_sets_the_cap(self) -> None:
+        self.assertEqual(self._credits("250"), "250")
+
+    def test_a_blank_env_var_keeps_the_default_cap(self) -> None:
+        self.assertEqual(self._credits("  "), "1000")
+
+    def test_a_cap_under_the_cli_minimum_is_raised_to_it(self) -> None:
+        self.assertEqual(self._credits("5"), "30")
+
+    def test_a_non_numeric_cap_falls_back_to_the_default(self) -> None:
+        self.assertEqual(self._credits("lots"), "1000")
+
+
 class CopilotModelCatalogTest(unittest.TestCase):
     def _queue(self, *lines: str) -> "queue.Queue[str]":
         lines_queue: queue.Queue[str] = queue.Queue()