codee-agent 0.2.0__tar.gz → 0.3.0__tar.gz

{codee_agent-0.2.0 → codee_agent-0.3.0}/PKG-INFO

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

@@ -1,6 +1,6 @@
 [project]
 name = "codee-agent"
-version = "0.2.0"
+version = "0.3.0"
 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.2.0 → codee_agent-0.3.0}/src/codee/admin.py

@@ -335,6 +335,11 @@ class AdminState(rx.State):
         return _save_toast(*SERVICE.save_agents(self.agents_content))
 
     def load_memories(self) -> None:
+        self.coding_agent = SERVICE.load_settings().coding_agent.value
+        if self.coding_agent == "github_copilot":
+            self.memories = []
+            self.selected_memory = ""
+            return
         self.memories = [MemoryEntry(**entry)
                          for entry in SERVICE.list_memories()]
 
@@ -605,6 +610,7 @@ class AdminState(rx.State):
 
 
 ACCENT = "var(--codee-accent)"
+ACCENT_DEEP = "var(--codee-accent-deep)"
 BORDER = "1px solid var(--codee-border)"
 MUTED = "var(--codee-muted)"
 SURFACE = "var(--codee-surface)"
@@ -706,23 +712,26 @@ def shell(content: rx.Component) -> rx.Component:
         color=TEXT,
         font_family="IBM Plex Sans, sans-serif",
         style={
-            "--codee-accent": rx.color_mode_cond("#167d5a", "#4abd85"),
-            "--codee-border": rx.color_mode_cond("#dfe4df", "#354039"),
-            "--codee-muted": rx.color_mode_cond("#68716b", "#a1ada5"),
-            "--codee-surface": rx.color_mode_cond("#ffffff", "#18201c"),
-            "--codee-page-background": rx.color_mode_cond("#f3f5f3", "#101512"),
-            "--codee-nav-background": rx.color_mode_cond("#f7f9f7", "#141b17"),
-            "--codee-text": rx.color_mode_cond("#202721", "#edf3ef"),
-            "--codee-hover": rx.color_mode_cond("#edf5f0", "#213029"),
-            "--codee-active": rx.color_mode_cond("#e2efe9", "#23372e"),
-            "--codee-grid": rx.color_mode_cond("#e4e9e5", "#2e3932"),
-            "--codee-subtle-icon": rx.color_mode_cond("#91a097", "#718078"),
-            "--codee-warning-background": rx.color_mode_cond("#fffbeb", "#332a13"),
-            "--codee-warning-border": rx.color_mode_cond("#d97706", "#f59e0b"),
+            "--codee-accent": rx.color_mode_cond("#2f6fd6", "#7aa9f0"),
+            # The darker half of the two-blue brand pairing, for filled marks
+            # that should recede behind the interactive blue (step markers).
+            "--codee-accent-deep": rx.color_mode_cond("#232c9b", "#5866de"),
+            "--codee-border": rx.color_mode_cond("#dbe1ec", "#2b3350"),
+            "--codee-muted": rx.color_mode_cond("#5d687f", "#a3adc4"),
+            "--codee-surface": rx.color_mode_cond("#ffffff", "#171d33"),
+            "--codee-page-background": rx.color_mode_cond("#f2f5fb", "#0e1224"),
+            "--codee-nav-background": rx.color_mode_cond("#f7f9fd", "#121729"),
+            "--codee-text": rx.color_mode_cond("#1b2440", "#e9edf7"),
+            "--codee-hover": rx.color_mode_cond("#edf3fc", "#1e2742"),
+            "--codee-active": rx.color_mode_cond("#e0eafb", "#24304f"),
+            "--codee-grid": rx.color_mode_cond("#e2e7f1", "#2a3350"),
+            "--codee-subtle-icon": rx.color_mode_cond("#8e99b0", "#6e7a94"),
+            "--codee-warning-background": rx.color_mode_cond("#fff5ed", "#33200f"),
+            "--codee-warning-border": rx.color_mode_cond("#e26128", "#f5854a"),
             # Tint reserved for in-flight work, so a live run reads at a glance.
-            "--codee-running-background": rx.color_mode_cond("#effaf4", "#15271f"),
+            "--codee-running-background": rx.color_mode_cond("#eff4fd", "#14203a"),
             "--codee-running-glow": rx.color_mode_cond(
-                "rgba(22, 125, 90, 0.16)", "rgba(74, 189, 133, 0.18)"),
+                "rgba(47, 111, 214, 0.16)", "rgba(122, 169, 240, 0.18)"),
             "color_scheme": "light dark",
         },
     )
@@ -888,13 +897,13 @@ def skill_card(skill: SkillSummary) -> rx.Component:
     return rx.box(
         rx.vstack(
             rx.hstack(rx.heading(skill.name, size="4"), rx.spacer(),
-                      rx.badge(skill.type, color_scheme="green", variant="soft"), width="100%"),
+                      rx.badge(skill.type, color_scheme="blue", variant="soft"), width="100%"),
             rx.text(skill.description, color=MUTED, font_size="0.9rem", min_height="2.7rem",
                     overflow="hidden"),
             rx.cond(
                 skill.issue_status != "",
                 rx.hstack(
-                    rx.badge(skill.issue_type, color_scheme="green",
+                    rx.badge(skill.issue_type, color_scheme="blue",
                              variant="outline"),
                     rx.icon("circle-dot", size=14, color=SUBTLE_ICON),
                     rx.text(skill.issue_status, color=MUTED,
@@ -1193,8 +1202,12 @@ def memory_page() -> rx.Component:
                                 memory_row), spacing="3", width="100%"),
                       empty_state("notebook-text", "No memories yet."))
     return shell(rx.vstack(page_header("Memory", "Review and maintain durable project context."),
-                           rx.cond(AdminState.selected_memory ==
-                                   "", listing, memory_editor()),
+                           rx.cond(AdminState.coding_agent == "github_copilot",
+                                   empty_state(
+                                       "notebook-text",
+                                       "Github Copilot currently does not support local memory"),
+                                   rx.cond(AdminState.selected_memory ==
+                                           "", listing, memory_editor())),
                            align="start", width="100%"))
 
 
@@ -1239,7 +1252,7 @@ def workflow_warning(message: rx.Var) -> rx.Component:
     return rx.callout(
         message,
         icon="triangle-alert",
-        color_scheme="amber",
+        color_scheme="orange",
         width="100%",
     )
 
@@ -1393,7 +1406,7 @@ def sessions_page() -> rx.Component:
 
 def azure_step(number: int, title: str, detail: rx.Component | str) -> rx.Component:
     return rx.hstack(
-        rx.box(str(number), color="white", background=ACCENT, min_width="1.4rem",
+        rx.box(str(number), color="white", background=ACCENT_DEEP, min_width="1.4rem",
                height="1.4rem", display="grid", place_items="center",
                border_radius="50%", font_size="0.75rem", font_weight="700"),
         rx.vstack(rx.text(title, font_weight="600", font_size="0.9rem"),
@@ -1635,10 +1648,10 @@ app = rx.App(
             "width": "6px",
         },
         ".workflow-route-node--forward .react-flow__handle": {
-            "background": "#167d5a",
+            "background": "#2f6fd6",
         },
         ".workflow-route-node--return .react-flow__handle": {
-            "background": "#d97706",
+            "background": "#e26128",
         },
     },
     stylesheets=[

{codee_agent-0.2.0 → codee_agent-0.3.0}/src/codee/admin_cli.py

@@ -22,8 +22,8 @@ config = rx.Config(
     plugins=[
         rx.plugins.RadixThemesPlugin(
             theme=rx.theme(
-                accent_color="green",
-                gray_color="sage",
+                accent_color="indigo",
+                gray_color="slate",
                 radius="small",
                 appearance="inherit",
             )

{codee_agent-0.2.0 → codee_agent-0.3.0}/src/codee/admin_service.py

@@ -1,4 +1,5 @@
 """Framework-independent operations for the Codee admin UI."""
+import hashlib
 import json
 import re
 import os
@@ -74,6 +75,9 @@ _CODING_AGENTS: dict[CodingAgent, type[AbstractCodingAgent]] = {
 }
 WORKFLOW_NODE_SPACING = 440
 WORKFLOW_NODE_CENTER_OFFSET = 110
+# Inferring the workflow costs a coding-agent run, so the graph is kept in the
+# data directory and reused by later admin processes.
+WORKFLOW_CACHE_FILE = "workflow.json"
 
 # Port the admin UI listens on unless ``codee-admin --port`` says otherwise.
 # The OAuth redirect URI is built from it, and Entra ID matches redirect URIs
@@ -199,6 +203,18 @@ def _transition_values(value: Any) -> list[dict[str, str]]:
     return transitions
 
 
+def _is_workflow(value: Any) -> bool:
+    """Whether a stored graph still has the shape the workflow page reads."""
+    if not isinstance(value, dict):
+        return False
+    return all(
+        isinstance(value.get(issue_type), dict)
+        and all(isinstance(value[issue_type].get(key), list)
+                for key in ("nodes", "edges", "warnings"))
+        for issue_type in ISSUE_TYPES
+    )
+
+
 def _remove_redundant_skill_transitions(
     transitions: list[dict[str, str]],
 ) -> list[dict[str, str]]:
@@ -247,7 +263,8 @@ class AdminService:
         self.data_dir.mkdir(parents=True, exist_ok=True)
         self.context = CodeeMainContext(data_dir=self.data_dir)
         self.context.settings = load_settings(self.data_dir)
-        self._workflow_cache: dict[str, Any] | None = None
+        # Cached graph plus the skill fingerprint it was generated from.
+        self._workflow_cache: tuple[str, dict[str, Any]] | None = None
         self._workflow_lock = threading.Lock()
         # Asking an agent for its catalog can mean spawning its CLI, so the
         # answer is cached per agent for the life of the process.
@@ -343,16 +360,73 @@ class AdminService:
         return ""
 
     def generate_workflow(self, force: bool = False) -> dict[str, Any]:
-        """Return cached workflows by issue type, regenerating when requested."""
+        """Return cached workflows by issue type, regenerating when requested.
+
+        The cache outlives the admin process: it is stored in the data
+        directory under the fingerprint of the skill documents the graph was
+        inferred from, so a restart reuses it while an edited skill still
+        regenerates it.
+        """
         workflow_lock = getattr(self, "_workflow_lock", None)
         if workflow_lock is None:
             workflow_lock = self._workflow_lock = threading.Lock()
         with workflow_lock:
-            workflow_cache = getattr(self, "_workflow_cache", None)
-            if not force and workflow_cache is not None:
-                return workflow_cache
-            self._workflow_cache = self._generate_workflow()
-            return self._workflow_cache
+            fingerprint = self._workflow_fingerprint()
+            if not force:
+                cached = self._cached_workflow(fingerprint)
+                if cached is not None:
+                    return cached
+            workflow = self._generate_workflow()
+            self._workflow_cache = (fingerprint, workflow)
+            self._store_workflow(fingerprint, workflow)
+            return workflow
+
+    def _workflow_fingerprint(self) -> str:
+        """Digest the skill documents the workflow graph is inferred from."""
+        digest = hashlib.sha256()
+        for skill in find_issue_triggered_skills(self.skills_dir):
+            digest.update(skill.slug.encode())
+            digest.update(b"\0")
+            digest.update(skill.path.read_bytes())
+            digest.update(b"\0")
+        return digest.hexdigest()
+
+    def _workflow_cache_path(self) -> Path:
+        return self.data_dir / WORKFLOW_CACHE_FILE
+
+    def _cached_workflow(self, fingerprint: str) -> dict[str, Any] | None:
+        """This process's graph, else the one an earlier process stored."""
+        cached = getattr(self, "_workflow_cache", None)
+        if cached is not None and cached[0] == fingerprint:
+            return cached[1]
+        try:
+            stored = json.loads(self._workflow_cache_path().read_text())
+        except FileNotFoundError:
+            return None
+        except (OSError, ValueError) as error:
+            print(f"[admin] Ignoring unreadable workflow cache: {error}")
+            return None
+        if not isinstance(stored, dict) or stored.get("fingerprint") != fingerprint:
+            return None
+        workflow = stored.get("workflow")
+        if not _is_workflow(workflow):
+            return None
+        self._workflow_cache = (fingerprint, workflow)
+        return workflow
+
+    def _store_workflow(self, fingerprint: str, workflow: dict[str, Any]) -> None:
+        """Persist the graph for the next admin process; failing is not fatal."""
+        path = self._workflow_cache_path()
+        pending = path.with_suffix(".tmp")
+        try:
+            path.parent.mkdir(parents=True, exist_ok=True)
+            pending.write_text(json.dumps(
+                {"fingerprint": fingerprint, "workflow": workflow}))
+            # Replace in one step so a crash mid-write cannot leave a
+            # half-written cache behind.
+            pending.replace(path)
+        except OSError as error:
+            print(f"[admin] Failed to store workflow cache: {error}")
 
     def _generate_workflow(self) -> dict[str, Any]:
         """Infer separate status graphs for story and task skills."""

{codee_agent-0.2.0 → codee_agent-0.3.0}/src/codee/test_admin_service.py

@@ -15,6 +15,25 @@ from codee_main_context.context import (
     CodeeMainContext, CodingAgent, Settings, TasksProvider, save_settings)
 
 
+def _empty_workflow() -> dict:
+    return {issue_type: {"nodes": [], "edges": [], "warnings": []}
+            for issue_type in ("story", "task")}
+
+
+def _write_issue_skill(root: Path) -> Path:
+    """Create one issue-trigger skill, the input the workflow cache is keyed on."""
+    skills_dir = root / ".claude" / "skills"
+    skill_dir = skills_dir / "develop"
+    skill_dir.mkdir(parents=True)
+    (skill_dir / "SKILL.md").write_text(
+        "---\nname: develop\ndisable-model-invocation: true\n"
+        "x-codee-trigger: issue\nx-codee-issue-status: [Ready]\n"
+        "x-codee-issue-type: story\n---\n"
+        "After implementation, move the issue to Review.\n"
+    )
+    return skills_dir
+
+
 class AdminServiceIssueTriggerTest(unittest.TestCase):
     def test_list_skills_includes_issue_status(self) -> None:
         with tempfile.TemporaryDirectory() as temporary_directory:
@@ -227,6 +246,7 @@ class AdminServiceIssueTriggerTest(unittest.TestCase):
             service = AdminService.__new__(AdminService)
             service.root = root
             service.skills_dir = skills_dir
+            service.data_dir = root / ".codee"
             service.context = Mock(settings=Settings(
                 coding_agent=CodingAgent.CLAUDE_CODE))
 
@@ -315,6 +335,7 @@ class AdminServiceIssueTriggerTest(unittest.TestCase):
             service = AdminService.__new__(AdminService)
             service.root = root
             service.skills_dir = skills_dir
+            service.data_dir = root / ".codee"
             service.context = Mock(
                 settings=Settings(coding_agent=CodingAgent.CLAUDE_CODE))
 
@@ -356,6 +377,7 @@ class AdminServiceIssueTriggerTest(unittest.TestCase):
             service = AdminService.__new__(AdminService)
             service.root = root
             service.skills_dir = skills_dir
+            service.data_dir = root / ".codee"
             service.context = Mock(
                 settings=Settings(coding_agent=CodingAgent.CLAUDE_CODE))
 
@@ -406,6 +428,7 @@ class AdminServiceIssueTriggerTest(unittest.TestCase):
             service = AdminService.__new__(AdminService)
             service.root = root
             service.skills_dir = skills_dir
+            service.data_dir = root / ".codee"
             service.context = Mock(
                 settings=Settings(coding_agent=CodingAgent.CLAUDE_CODE))
 
@@ -462,6 +485,7 @@ class AdminServiceIssueTriggerTest(unittest.TestCase):
             service = AdminService.__new__(AdminService)
             service.root = root
             service.skills_dir = skills_dir
+            service.data_dir = root / ".codee"
             service.context = Mock(
                 settings=Settings(coding_agent=CodingAgent.CLAUDE_CODE))
 
@@ -501,6 +525,7 @@ class AdminServiceIssueTriggerTest(unittest.TestCase):
             service = AdminService.__new__(AdminService)
             service.root = root
             service.skills_dir = skills_dir
+            service.data_dir = root / ".codee"
             service.context = Mock(
                 settings=Settings(coding_agent=CodingAgent.CLAUDE_CODE))
 
@@ -518,8 +543,10 @@ class AdminServiceIssueTriggerTest(unittest.TestCase):
 
     def test_generate_workflow_returns_empty_without_issue_skills(self) -> None:
         with tempfile.TemporaryDirectory() as temporary_directory:
+            root = Path(temporary_directory)
             service = AdminService.__new__(AdminService)
-            service.skills_dir = Path(temporary_directory)
+            service.skills_dir = root
+            service.data_dir = root / ".codee"
 
             self.assertEqual(service.generate_workflow(), {
                 "story": {"nodes": [], "edges": [], "warnings": []},
@@ -527,15 +554,92 @@ class AdminServiceIssueTriggerTest(unittest.TestCase):
             })
 
     def test_generate_workflow_is_cached_until_forced(self) -> None:
-        service = AdminService.__new__(AdminService)
-        generated = {"nodes": [], "edges": [], "warnings": []}
+        with tempfile.TemporaryDirectory() as temporary_directory:
+            root = Path(temporary_directory)
+            service = AdminService.__new__(AdminService)
+            service.skills_dir = root / ".claude" / "skills"
+            service.data_dir = root / ".codee"
+            generated = _empty_workflow()
+
+            with patch.object(service, "_generate_workflow",
+                              return_value=generated) as generate:
+                self.assertIs(service.generate_workflow(), generated)
+                self.assertIs(service.generate_workflow(), generated)
+                self.assertIs(service.generate_workflow(force=True), generated)
+
+            self.assertEqual(generate.call_count, 2)
+
+    def test_generate_workflow_reuses_graph_stored_by_an_earlier_process(self) -> None:
+        with tempfile.TemporaryDirectory() as temporary_directory:
+            root = Path(temporary_directory)
+            skills_dir = _write_issue_skill(root)
+            generated = _empty_workflow()
+            generated["story"]["warnings"] = ["stored"]
+
+            first = AdminService.__new__(AdminService)
+            first.skills_dir = skills_dir
+            first.data_dir = root / ".codee"
+            with patch.object(first, "_generate_workflow", return_value=generated):
+                first.generate_workflow()
+
+            # A restart starts from an empty in-memory cache, so only the file
+            # written above can spare it another coding-agent run.
+            restarted = AdminService.__new__(AdminService)
+            restarted.skills_dir = skills_dir
+            restarted.data_dir = root / ".codee"
+            with patch.object(restarted, "_generate_workflow") as generate:
+                workflow = restarted.generate_workflow()
 
-        with patch.object(service, "_generate_workflow", return_value=generated) as generate:
-            self.assertIs(service.generate_workflow(), generated)
-            self.assertIs(service.generate_workflow(), generated)
-            self.assertIs(service.generate_workflow(force=True), generated)
+            generate.assert_not_called()
+            self.assertEqual(workflow["story"]["warnings"], ["stored"])
+
+    def test_generate_workflow_ignores_stored_graph_after_a_skill_changes(self) -> None:
+        with tempfile.TemporaryDirectory() as temporary_directory:
+            root = Path(temporary_directory)
+            skills_dir = _write_issue_skill(root)
+
+            first = AdminService.__new__(AdminService)
+            first.skills_dir = skills_dir
+            first.data_dir = root / ".codee"
+            with patch.object(first, "_generate_workflow",
+                              return_value=_empty_workflow()):
+                first.generate_workflow()
+
+            (skills_dir / "develop" / "SKILL.md").write_text(
+                "---\nname: develop\ndisable-model-invocation: true\n"
+                "x-codee-trigger: issue\nx-codee-issue-status: [Ready]\n"
+                "x-codee-issue-type: story\n---\n"
+                "After implementation, move the issue to Done.\n"
+            )
+            regenerated = _empty_workflow()
+            regenerated["story"]["warnings"] = ["regenerated"]
+            restarted = AdminService.__new__(AdminService)
+            restarted.skills_dir = skills_dir
+            restarted.data_dir = root / ".codee"
+            with patch.object(restarted, "_generate_workflow",
+                              return_value=regenerated) as generate:
+                workflow = restarted.generate_workflow()
+
+            generate.assert_called_once_with()
+            self.assertEqual(workflow["story"]["warnings"], ["regenerated"])
+
+    def test_generate_workflow_ignores_a_corrupt_stored_graph(self) -> None:
+        with tempfile.TemporaryDirectory() as temporary_directory:
+            root = Path(temporary_directory)
+            skills_dir = _write_issue_skill(root)
+            data_dir = root / ".codee"
+            data_dir.mkdir()
+            (data_dir / "workflow.json").write_text("{ not json")
+
+            service = AdminService.__new__(AdminService)
+            service.skills_dir = skills_dir
+            service.data_dir = data_dir
+            generated = _empty_workflow()
+            with patch.object(service, "_generate_workflow",
+                              return_value=generated) as generate:
+                self.assertIs(service.generate_workflow(), generated)
 
-        self.assertEqual(generate.call_count, 2)
+            generate.assert_called_once_with()
 
 
 class AdminServiceSkillModelTest(unittest.TestCase):