@prateek_ai/agents-maker 1.0.0 → 1.0.2

package/tools/init_project.py

@@ -11,10 +11,7 @@ Usage:
 from __future__ import annotations
 
 import argparse
-import hashlib
-import os
 import sys
-import tempfile
 from datetime import date
 from pathlib import Path
 
@@ -28,12 +25,6 @@ SCRIPT_DIR = Path(__file__).resolve().parent       # agents-maker/tools/
 KIT_DIR = SCRIPT_DIR.parent                        # agents-maker/
 sys.path.insert(0, str(KIT_DIR))
 
-try:
-    import yaml
-except ImportError:
-    print("[ERROR] pyyaml is required: pip install pyyaml", file=sys.stderr)
-    sys.exit(1)
-
 try:
     from context_loaders.project_summary import build_summary
     from context_loaders.repo_tree import walk_tree
@@ -43,10 +34,10 @@ except ImportError as e:
     sys.exit(1)
 
 try:
-    from tools.domain_utils import _load_yaml
+    from tools._core import atomic_write, atomic_write_yaml, load_yaml, source_hash
     from tools.domain_utils import detect_domain as _detect_domain
 except ImportError:
-    from domain_utils import _load_yaml
+    from _core import atomic_write, atomic_write_yaml, load_yaml, source_hash
     from domain_utils import detect_domain as _detect_domain
 
 
@@ -82,46 +73,6 @@ def _sanitize_yaml_str(value: str, field: str) -> str:
     return value
 
 
-# ---------------------------------------------------------------------------
-# Source hash — lets validate_kit.py detect stale system_prompt.md
-# ---------------------------------------------------------------------------
-
-def _compute_source_hash(kit_dir: Path) -> str:
-    h = hashlib.sha256()
-    agent_files = sorted((kit_dir / "agents").glob("*.md")) if (kit_dir / "agents").is_dir() else []
-    skill_files = sorted((kit_dir / "skills").glob("*.md")) if (kit_dir / "skills").is_dir() else []
-    for path in agent_files + skill_files:
-        try:
-            h.update(path.read_bytes().replace(b"\r\n", b"\n"))
-        except (OSError, PermissionError):
-            pass
-    return h.hexdigest()[:16]
-
-
-# ---------------------------------------------------------------------------
-# Atomic write helpers (write to temp then os.replace — crash-safe)
-# ---------------------------------------------------------------------------
-
-def _atomic_write_text(path: Path, content: str) -> None:
-    path.parent.mkdir(parents=True, exist_ok=True)
-    with tempfile.NamedTemporaryFile(
-        "w", dir=path.parent, delete=False, suffix=".tmp", encoding="utf-8"
-    ) as f:
-        f.write(content)
-        tmp = f.name
-    os.replace(tmp, path)
-
-
-def _atomic_write_yaml(path: Path, data: dict) -> None:
-    path.parent.mkdir(parents=True, exist_ok=True)
-    with tempfile.NamedTemporaryFile(
-        "w", dir=path.parent, delete=False, suffix=".tmp", encoding="utf-8"
-    ) as f:
-        yaml.dump(data, f, default_flow_style=False, allow_unicode=True, sort_keys=False)
-        tmp = f.name
-    os.replace(tmp, path)
-
-
 # ---------------------------------------------------------------------------
 # system_prompt.md builder
 # ---------------------------------------------------------------------------
@@ -130,6 +81,8 @@ def build_system_prompt(project_name: str, domain: str, stack: list[str]) -> str
     agents_dir = KIT_DIR / "agents"
     agent_order = [
         "orchestrator.md",
+        "brain.md",
+        "planpro.md",
         "architect_agent.md",
         "code_agent.md",
         "execution_agent.md",
@@ -139,10 +92,10 @@ def build_system_prompt(project_name: str, domain: str, stack: list[str]) -> str
         "compression_agent.md",
     ]
     skill_count = len(list((KIT_DIR / "skills").glob("*.md"))) if (KIT_DIR / "skills").exists() else 0
-    source_hash = _compute_source_hash(KIT_DIR)
+    src_hash = source_hash(KIT_DIR)
     version_header = (
         f"# agents-maker system_prompt.md\n"
-        f"# Version: 1.0 | Generated: {date.today().isoformat()} | Source hash: {source_hash}\n"
+        f"# Version: 1.0 | Generated: {date.today().isoformat()} | Source hash: {src_hash}\n"
         f"# Regenerate: python agents-maker/tools/init_project.py --update\n"
         f"# Contains: {len(agent_order)} agents + {skill_count} skills\n"
         f"#\n"
@@ -332,7 +285,7 @@ def main() -> None:
     project_yaml_path = config_dir / "project.yaml"
 
     # Preserve session_count if updating
-    existing_cfg = _load_yaml(project_yaml_path)
+    existing_cfg = load_yaml(project_yaml_path)
     session_count = existing_cfg.get("session_count", 0) if args.update else 0
     created_at = existing_cfg.get("created_at", date.today().isoformat()) if args.update else date.today().isoformat()
 
@@ -349,7 +302,7 @@ def main() -> None:
         "last_session": existing_cfg.get("last_session") if args.update else None,
     }
     try:
-        _atomic_write_yaml(project_yaml_path, project_cfg)
+        atomic_write_yaml(project_yaml_path, project_cfg)
     except OSError as e:
         print(f"[ERROR] Could not write project.yaml: {e}", file=sys.stderr)
         sys.exit(1)
@@ -359,7 +312,7 @@ def main() -> None:
     if not system_prompt_path.exists() or args.update:
         system_prompt_text = build_system_prompt(project_name, final_domain, stack)
         try:
-            _atomic_write_text(system_prompt_path, system_prompt_text)
+            atomic_write(system_prompt_path, system_prompt_text)
         except OSError as e:
             print(f"[ERROR] Could not write system_prompt.md: {e}", file=sys.stderr)
             sys.exit(1)
@@ -375,7 +328,7 @@ def main() -> None:
     state_path = KIT_DIR / "project_state.md"
     if not state_path.exists():
         try:
-            _atomic_write_text(state_path, STATE_TEMPLATE)
+            atomic_write(state_path, STATE_TEMPLATE)
             state_created = True
         except OSError as e:
             print(f"[ERROR] Could not write project_state.md: {e}", file=sys.stderr)
@@ -437,7 +390,7 @@ def main() -> None:
         )
         claude_md_path = project_root / "CLAUDE.md"
         try:
-            _atomic_write_text(claude_md_path, claude_md_content)
+            atomic_write(claude_md_path, claude_md_content)
             print(f"  [DONE] CLAUDE.md written to {claude_md_path}")
             print("         Claude Code will auto-load domain/phase/stack on every session.")
             print("         Commit CLAUDE.md to git — it is project config, not private state.")

package/tools/routing.py

@@ -0,0 +1,103 @@
+"""
+tools/routing.py
+
+Single source of truth for lifecycle routing metadata: which specialist
+agents are active for a given (phase, domain), plus phase labels and agent
+role labels used across the generators.
+
+Previously this mapping was hardcoded independently in three places
+(generate_prompt.py, generate_claude_md.py, generate_platform_configs.py),
+and the three copies had drifted apart. Centralizing them here makes the
+routing consistent across every generated artifact (the pasted prompt,
+CLAUDE.md, Copilot/Cursor rules, and the agkit config).
+
+Two views are exposed because the tools present routing differently:
+  * phase_agents(phase, domain)  -> the phase's agent list as declared in the
+    table (used where the orchestrator is mentioned separately, e.g. CLAUDE.md).
+  * active_agents(phase, domain) -> orchestrator-first, deduplicated (used for
+    the "Active agents" line of a generated prompt).
+"""
+
+from __future__ import annotations
+
+# Phase -> domain -> agents. "_all" is the default for domains without an override.
+PHASE_AGENTS: dict[str, dict[str, list[str]]] = {
+    "task_framing":     {"_all": ["orchestrator"]},
+    "requirements":     {"_all": ["orchestrator", "architect_agent"]},
+    "solution_design":  {
+        "_all":          ["architect_agent"],
+        "software":      ["architect_agent", "ui_agent"],
+        "product_design": ["architect_agent", "ui_agent", "ux_agent"],
+        "marketing":     ["architect_agent", "ux_agent"],
+    },
+    "implementation":   {
+        "_all":          ["execution_agent"],
+        "software":      ["code_agent"],
+        "data_analytics": ["code_agent"],
+    },
+    "review_refinement": {
+        "_all":          ["reviewer_agent"],
+        "software":      ["reviewer_agent", "code_agent"],
+        "product_design": ["reviewer_agent", "ui_agent", "ux_agent"],
+        "marketing":     ["reviewer_agent", "ux_agent"],
+    },
+    "handoff":          {"_all": ["orchestrator", "execution_agent"]},
+}
+
+PHASE_LABELS: dict[str, str] = {
+    "task_framing":     "Task Framing",
+    "requirements":     "Requirements",
+    "solution_design":  "Solution Design",
+    "implementation":   "Implementation",
+    "review_refinement": "Review & Refinement",
+    "handoff":          "Handoff",
+    # aliases accepted by generate_prompt.py
+    "framing":          "Task Framing",
+    "design":           "Solution Design",
+    "implement":        "Implementation",
+    "review":           "Review & Refinement",
+}
+
+AGENT_ROLES: dict[str, str] = {
+    "orchestrator":      "routing",
+    "architect_agent":   "design",
+    "code_agent":        "implementation",
+    "execution_agent":   "execution",
+    "ui_agent":          "UI",
+    "ux_agent":          "UX",
+    "reviewer_agent":    "QA",
+    "compression_agent": "compression",
+    "brain":             "brainstorm",
+    "planpro":           "planning",
+}
+
+
+def phase_agents(phase: str, domain: str) -> list[str]:
+    """Agents declared for (phase, domain), falling back to the phase's `_all` list."""
+    phase_map = PHASE_AGENTS.get(phase, {"_all": ["orchestrator"]})
+    return list(phase_map.get(domain, phase_map["_all"]))
+
+
+def active_agents(phase: str, domain: str) -> list[str]:
+    """Orchestrator-first, de-duplicated agent list for a generated prompt."""
+    agents = ["orchestrator"]
+    for a in phase_agents(phase, domain):
+        if a not in agents:
+            agents.append(a)
+    return agents
+
+
+def domain_agents(domain: str) -> list[str]:
+    """Every agent this domain uses across all phases (union), orchestrator first.
+
+    Used to scope a self-contained/Direct-Task-Mode system prompt: it must carry
+    the domain's implementation + review specialists (and their skills), not just
+    the phase-0 orchestrator — otherwise the skills that add value get dropped.
+    Still excludes agents from other domains, so it stays smaller than the full kit.
+    """
+    seen: list[str] = ["orchestrator"]
+    for phase in PHASE_AGENTS:
+        for a in active_agents(phase, domain):
+            if a not in seen:
+                seen.append(a)
+    return seen