monoco-toolkit 0.3.2__py3-none-any.whl → 0.3.3__py3-none-any.whl

monoco/features/issue/validator.py

@@ -444,6 +444,17 @@ class IssueValidator:
         if not all_ids:
             return diagnostics
 
+        # Logic: Epics must have a parent (unless it is the Sink Root EPIC-0000)
+        if meta.type == "epic" and meta.id != "EPIC-0000" and not meta.parent:
+            line = self._get_field_line(content, "parent")
+            diagnostics.append(
+                self._create_diagnostic(
+                    "Hierarchy Violation: Epics must have a parent (e.g., 'EPIC-0000').",
+                    DiagnosticSeverity.Error,
+                    line=line,
+                )
+            )
+
         if (
             meta.parent
             and meta.parent not in all_ids
@@ -575,6 +586,7 @@ class IssueValidator:
         has_domains_field = False
         lines = content.splitlines()
         in_fm = False
+        field_line = 0
         for i, line_content in enumerate(lines):
             stripped = line_content.strip()
             if stripped == "---":
@@ -585,6 +597,7 @@ class IssueValidator:
             elif in_fm:
                 if stripped.startswith("domains:"):
                     has_domains_field = True
+                    field_line = i
                     break
 
         # Governance Maturity Check
@@ -607,6 +620,40 @@ class IssueValidator:
                     )
                 )
 
+        # Domain Content Validation
+        from .domain_service import DomainService
+
+        service = DomainService()
+
+        if hasattr(meta, "domains") and meta.domains:
+            for domain in meta.domains:
+                if service.is_alias(domain):
+                    canonical = service.get_canonical(domain)
+                    diagnostics.append(
+                        self._create_diagnostic(
+                            f"Domain Alias: '{domain}' is an alias for '{canonical}'. Preference: Canonical.",
+                            DiagnosticSeverity.Warning,
+                            line=field_line,
+                        )
+                    )
+                elif not service.is_defined(domain):
+                    if service.config.strict:
+                        diagnostics.append(
+                            self._create_diagnostic(
+                                f"Unknown Domain: '{domain}' is not defined in domain ontology.",
+                                DiagnosticSeverity.Error,
+                                line=field_line,
+                            )
+                        )
+                    else:
+                        diagnostics.append(
+                            self._create_diagnostic(
+                                f"Unknown Domain: '{domain}' is not defined in domain ontology.",
+                                DiagnosticSeverity.Warning,
+                                line=field_line,
+                            )
+                        )
+
         return diagnostics
 
     def _validate_checkbox_logic_blocks(

monoco/features/scheduler/__init__.py

@@ -0,0 +1,19 @@
+from .models import RoleTemplate, SchedulerConfig
+from .worker import Worker
+from .config import load_scheduler_config
+from .defaults import DEFAULT_ROLES
+from .session import Session, RuntimeSession
+from .manager import SessionManager
+from .reliability import ApoptosisManager
+
+__all__ = [
+    "RoleTemplate",
+    "SchedulerConfig",
+    "Worker",
+    "load_scheduler_config",
+    "DEFAULT_ROLES",
+    "Session",
+    "RuntimeSession",
+    "SessionManager",
+    "ApoptosisManager",
+]

monoco/features/scheduler/cli.py

@@ -0,0 +1,204 @@
+import typer
+import time
+from pathlib import Path
+from typing import Optional
+from monoco.core.output import print_output
+from monoco.core.config import get_config
+from monoco.features.scheduler import SessionManager, load_scheduler_config
+
+app = typer.Typer(name="agent", help="Manage agent sessions")
+
+
+@app.command()
+def run(
+    target: str = typer.Argument(
+        ..., help="Issue ID (e.g. FEAT-101) or a Task Description in quotes."
+    ),
+    role: Optional[str] = typer.Option(
+        None,
+        help="Specific role to use (crafter/builder/auditor). Default: intelligent selection.",
+    ),
+    detach: bool = typer.Option(
+        False, "--detach", "-d", help="Run in background (Daemon)"
+    ),
+    fail: bool = typer.Option(
+        False, "--fail", help="Simulate a crash for testing Apoptosis."
+    ),
+):
+    """
+    Start an agent session.
+    - If TARGET is an Issue ID: Work on that issue.
+    - If TARGET is a text description: Create a new issue (Crafter).
+    """
+    settings = get_config()
+    project_root = Path(settings.paths.root).resolve()
+
+    # 1. Smart Intent Recognition
+    import re
+
+    is_id = re.match(r"^[a-zA-Z]+-\d+$", target)
+
+    if is_id:
+        issue_id = target.upper()
+        role_name = role or "builder"
+        description = None
+    else:
+        issue_id = "NEW_TASK"
+        role_name = role or "crafter"
+        description = target
+
+    # 2. Load Roles
+    roles = load_scheduler_config(project_root)
+    selected_role = roles.get(role_name)
+
+    if not selected_role:
+        from monoco.core.output import print_error
+
+        print_error(f"Role '{role_name}' not found. Available: {list(roles.keys())}")
+        raise typer.Exit(code=1)
+
+    print_output(
+        f"Starting Agent Session for '{target}' as {role_name}...",
+        title="Agent Scheduler",
+    )
+
+    # 3. Initialize Session
+    manager = SessionManager()
+    session = manager.create_session(issue_id, selected_role)
+
+    if detach:
+        print_output(
+            "Background mode not fully implemented yet. Running in foreground."
+        )
+
+    try:
+        # Pass description if it's a new task
+        context = {"description": description} if description else None
+
+        if fail:
+            from monoco.core.output import rprint
+
+            rprint("[bold yellow]DEBUG: Simulating immediate crash...[/bold yellow]")
+            session.model.status = "failed"
+        else:
+            session.start(context=context)
+
+        # Monitoring Loop
+        while session.model.status == "running":
+            time.sleep(1)
+
+        if session.model.status == "failed":
+            from monoco.core.output import print_error
+
+            print_error(
+                f"Session {session.model.id} FAILED. Use 'monoco agent autopsy {session.model.id}' for analysis."
+            )
+        else:
+            print_output(
+                f"Session finished with status: {session.model.status}",
+                title="Agent Scheduler",
+            )
+
+    except KeyboardInterrupt:
+        print("\nStopping...")
+        session.terminate()
+        print_output("Session terminated.")
+
+
+@app.command()
+def kill(session_id: str):
+    """
+    Terminate a session.
+    """
+    manager = SessionManager()
+    session = manager.get_session(session_id)
+    if session:
+        session.terminate()
+        print_output(f"Session {session_id} terminated.")
+    else:
+        print_output(f"Session {session_id} not found.", style="red")
+
+
+@app.command()
+def autopsy(
+    target: str = typer.Argument(..., help="Session ID or Issue ID to analyze."),
+):
+    """
+    Execute Post-Mortem analysis on a failed session or target Issue.
+    """
+    from .reliability import ApoptosisManager
+
+    manager = SessionManager()
+
+    print_output(f"Initiating Autopsy for '{target}'...", title="Coroner")
+
+    # Try to find session
+    session = manager.get_session(target)
+    if not session:
+        # Fallback: Treat target as Issue ID and create a dummy failed session context
+        import re
+
+        if re.match(r"^[a-zA-Z]+-\d+$", target):
+            print_output(f"Session not in memory. Analyzing Issue {target} directly.")
+            # We create a transient session just to trigger the coroner
+            from .defaults import DEFAULT_ROLES
+
+            builder_role = next(r for r in DEFAULT_ROLES if r.name == "builder")
+            session = manager.create_session(target.upper(), builder_role)
+            session.model.status = "failed"
+        else:
+            print_output(
+                f"Could not find session or valid Issue ID for '{target}'", style="red"
+            )
+            raise typer.Exit(code=1)
+
+    apoptosis = ApoptosisManager(manager)
+    apoptosis.trigger_apoptosis(session.model.id)
+
+
+@app.command(name="list")
+def list_sessions():
+    """
+    List active agent sessions.
+    """
+    manager = SessionManager()
+    sessions = manager.list_sessions()
+
+    output = []
+    for s in sessions:
+        output.append(
+            {
+                "id": s.model.id,
+                "issue": s.model.issue_id,
+                "role": s.model.role_name,
+                "status": s.model.status,
+                "branch": s.model.branch_name,
+            }
+        )
+
+    print_output(
+        output
+        or "No active sessions found (Note: Persistence not implemented in CLI list yet).",
+        title="Active Sessions",
+    )
+
+
+@app.command()
+def logs(session_id: str):
+    """
+    Stream logs for a session.
+    """
+    print_output(f"Streaming logs for {session_id}...", title="Session Logs")
+    # Placeholder
+    print("[12:00:00] Session started")
+    print("[12:00:01] Worker initialized")
+
+
+@app.command()
+def kill(session_id: str):
+    """
+    Terminate a session.
+    """
+    print_output(f"Killing session {session_id}...", title="Kill Session")
+    # Placeholder
+    print("Signal sent.")

monoco/features/scheduler/config.py

@@ -0,0 +1,32 @@
+import yaml
+from pathlib import Path
+from typing import Dict
+from .models import RoleTemplate, SchedulerConfig
+from .defaults import DEFAULT_ROLES
+
+
+def load_scheduler_config(project_root: Path) -> Dict[str, RoleTemplate]:
+    """
+    Load scheduler configuration from .monoco/scheduler.yaml
+    Merges with default roles.
+    """
+    roles = {role.name: role for role in DEFAULT_ROLES}
+
+    config_path = project_root / ".monoco" / "scheduler.yaml"
+    if config_path.exists():
+        try:
+            with open(config_path, "r") as f:
+                data = yaml.safe_load(f) or {}
+
+            # Use Pydantic to validate the whole config if possible, or just the roles list
+            # Depending on file structure. Assuming the file has a 'roles' key.
+            if "roles" in data:
+                # We can validate using SchedulerConfig
+                config = SchedulerConfig(roles=data["roles"])
+                for role in config.roles:
+                    roles[role.name] = role
+        except Exception as e:
+            # For now, just log or print. Ideally use a logger.
+            print(f"Warning: Failed to load scheduler config: {e}")
+
+    return roles

monoco/features/scheduler/defaults.py

@@ -0,0 +1,54 @@
+from .models import RoleTemplate
+
+DEFAULT_ROLES = [
+    RoleTemplate(
+        name="crafter",
+        description="Responsible for initial design, research, and drafting issues from descriptions.",
+        trigger="task.received",
+        goal="Produce a structured Issue file and/or detailed design document.",
+        tools=[
+            "create_issue_file",
+            "read_file",
+            "search_web",
+            "view_file_outline",
+            "write_to_file",
+        ],
+        system_prompt=(
+            "You are a Crafter agent. Your goal is to turn vague ideas into structured engineering plans.\n"
+            "If the user provides a description, use 'monoco issue create' and 'monoco issue update' to build the task.\n"
+            "If the user provides an existing Issue, analyze the context and provide a detailed design or implementation plan."
+        ),
+        engine="gemini",
+    ),
+    RoleTemplate(
+        name="builder",
+        description="Responsible for implementation.",
+        trigger="design.approved",
+        goal="Implement code and tests",
+        tools=["read_file", "write_to_file", "run_command", "git"],
+        system_prompt="You are a Builder agent. Your job is to implement the code based on the design.",
+        engine="gemini",
+    ),
+    RoleTemplate(
+        name="auditor",
+        description="Responsible for code review.",
+        trigger="implementation.submitted",
+        goal="Review code and provide feedback",
+        tools=[
+            "read_file",
+            "read_terminal",
+            "run_command",
+        ],  # Assumed read_diff and lint are via run_command
+        system_prompt="You are an Auditor agent. Your job is to review the code for quality and correctness.",
+        engine="gemini",
+    ),
+    RoleTemplate(
+        name="coroner",
+        description="Responsible for analyzing failure root causes (Autopsy).",
+        trigger="session.crashed",
+        goal="Produce a post-mortem report",
+        tools=["read_file", "read_terminal", "git_log"],
+        system_prompt="You are a Coroner agent. Your job is to analyze why the previous session failed and write a post-mortem report.",
+        engine="gemini",
+    ),
+]

monoco/features/scheduler/manager.py

@@ -0,0 +1,49 @@
+from typing import Dict, List, Optional
+import uuid
+from .models import RoleTemplate
+from .worker import Worker
+from .session import Session, RuntimeSession
+
+
+class SessionManager:
+    """
+    Manages the lifecycle of sessions.
+    Responsible for creating, tracking, and retrieving sessions.
+    """
+
+    def __init__(self):
+        # In-memory storage for now. In prod, this might be a DB or file-backed.
+        self._sessions: Dict[str, RuntimeSession] = {}
+
+    def create_session(self, issue_id: str, role: RoleTemplate) -> RuntimeSession:
+        session_id = str(uuid.uuid4())
+        branch_name = (
+            f"agent/{issue_id}/{session_id[:8]}"  # Simple branch naming strategy
+        )
+
+        session_model = Session(
+            id=session_id,
+            issue_id=issue_id,
+            role_name=role.name,
+            branch_name=branch_name,
+        )
+
+        worker = Worker(role, issue_id)
+        runtime = RuntimeSession(session_model, worker)
+        self._sessions[session_id] = runtime
+        return runtime
+
+    def get_session(self, session_id: str) -> Optional[RuntimeSession]:
+        return self._sessions.get(session_id)
+
+    def list_sessions(self, issue_id: Optional[str] = None) -> List[RuntimeSession]:
+        if issue_id:
+            return [s for s in self._sessions.values() if s.model.issue_id == issue_id]
+        return list(self._sessions.values())
+
+    def terminate_session(self, session_id: str):
+        session = self.get_session(session_id)
+        if session:
+            session.terminate()
+            # We might want to keep the record for a while, so don't delete immediately
+            # del self._sessions[session_id]

monoco/features/scheduler/models.py

@@ -0,0 +1,24 @@
+from typing import List
+from pydantic import BaseModel, Field
+
+
+class RoleTemplate(BaseModel):
+    name: str = Field(
+        ..., description="Unique identifier for the role (e.g., 'crafter')"
+    )
+    description: str = Field(..., description="Human-readable description of the role")
+    trigger: str = Field(
+        ..., description="Event that triggers this agent (e.g., 'issue.created')"
+    )
+    goal: str = Field(..., description="The primary goal/output of this agent")
+    tools: List[str] = Field(default_factory=list, description="List of allowed tools")
+    system_prompt: str = Field(
+        ..., description="The system prompt template for this agent"
+    )
+    engine: str = Field(
+        default="gemini", description="CLI agent engine (gemini/claude)"
+    )
+
+
+class SchedulerConfig(BaseModel):
+    roles: List[RoleTemplate] = Field(default_factory=list)

monoco/features/scheduler/reliability.py

@@ -0,0 +1,99 @@
+from .manager import SessionManager
+from .session import RuntimeSession
+from .defaults import DEFAULT_ROLES
+
+
+class ApoptosisManager:
+    """
+    Handles the 'Apoptosis' (Programmed Cell Death) lifecycle for agents.
+    Ensures that failing agents are killed, analyzed, and the environment is reset.
+    """
+
+    def __init__(self, session_manager: SessionManager):
+        self.session_manager = session_manager
+        # Find coroner role
+        self.coroner_role = next(
+            (r for r in DEFAULT_ROLES if r.name == "coroner"), None
+        )
+        if not self.coroner_role:
+            raise ValueError("Coroner role not defined!")
+
+    def check_health(self, session: RuntimeSession) -> bool:
+        """
+        Check if a session is healthy.
+        In a real implementation, this would check heartbeat, CPU usage, or token limits.
+        """
+        # Placeholder logic: Random failure or external flag?
+        # For now, always healthy unless explicitly marked 'crashed' (which we can simulate)
+        if hasattr(session, "simulate_crash") and session.simulate_crash:
+            return False
+        return True
+
+    def trigger_apoptosis(self, session_id: str):
+        """
+        Execute the full death and rebirth cycle.
+        """
+        session = self.session_manager.get_session(session_id)
+        if not session:
+            print(f"Session {session_id} not found for apoptosis.")
+            return
+
+        print(f"💀 [Apoptosis] Starting lifecycle for Session {session_id}")
+
+        # 1. Kill
+        self._kill(session)
+
+        # 2. Autopsy
+        try:
+            self._perform_autopsy(session)
+        except Exception as e:
+            print(f"⚠️ Autopsy failed: {e}")
+
+        # 3. Reset
+        self._reset_environment(session)
+
+        print(
+            f"✅ [Apoptosis] Task {session.model.issue_id} has been reset and analyzed."
+        )
+
+    def _kill(self, session: RuntimeSession):
+        print(f"🔪 Killing worker process for {session.model.id}...")
+        session.terminate()
+        session.model.status = "crashed"
+
+    def _perform_autopsy(self, victim_session: RuntimeSession):
+        print(
+            f"🔍 Performing autopsy on {victim_session.model.id} via Coroner agent..."
+        )
+
+        # Start a Coroner session
+        coroner_session = self.session_manager.create_session(
+            victim_session.model.issue_id, self.coroner_role
+        )
+
+        # Context for the coroner
+        context = {
+            "description": f"The previous agent session ({victim_session.model.id}) for role '{victim_session.model.role_name}' crashed. Please analyze the environment and the Issue {victim_session.model.issue_id}, then write a ## Post-mortem section in the issue file."
+        }
+
+        coroner_session.start(context=context)
+        print("📄 Coroner agent finished analysis.")
+
+    def _reset_environment(self, session: RuntimeSession):
+        print("🧹 Resetting environment (simulated git reset --hard)...")
+        # In real impl:
+        # import subprocess
+        # subprocess.run(["git", "reset", "--hard"], check=True)
+        pass
+
+    def _retry(self, session: RuntimeSession):
+        print("🔄 Reincarnating session...")
+        # Create a new session with the same role and issue
+        new_session = self.session_manager.create_session(
+            session.model.issue_id,
+            # We need to find the original role object.
+            # Simplified: assuming we can find it by name or pass it.
+            # For now, just placeholder.
+            session.worker.role,
+        )
+        new_session.start()

monoco/features/scheduler/session.py

@@ -0,0 +1,87 @@
+from datetime import datetime
+from typing import Optional
+from pydantic import BaseModel, Field
+from .worker import Worker
+
+
+class Session(BaseModel):
+    """
+    Represents a runtime session of a worker.
+    Persisted state of the session.
+    """
+
+    id: str = Field(..., description="Unique session ID (likely UUID)")
+    issue_id: str = Field(..., description="The Issue ID this session is working on")
+    role_name: str = Field(..., description="Name of the role employed")
+    status: str = Field(
+        default="pending", description="pending, running, suspended, terminated"
+    )
+    branch_name: str = Field(
+        ..., description="Git branch name associated with this session"
+    )
+    created_at: datetime = Field(default_factory=datetime.now)
+    updated_at: datetime = Field(default_factory=datetime.now)
+    # History could be a list of logs or pointers to git commits
+    # For now, let's keep it simple. The git log IS the history.
+
+    class Config:
+        arbitrary_types_allowed = True
+
+
+class RuntimeSession:
+    """
+    The in-memory wrapper around the Session model and the active Worker.
+    """
+
+    def __init__(self, session_model: Session, worker: Worker):
+        self.model = session_model
+        self.worker = worker
+
+    def start(self, context: Optional[dict] = None):
+        print(
+            f"Session {self.model.id}: Starting worker on branch {self.model.branch_name}"
+        )
+        # In real impl, checking out branch happening here
+        self.model.status = "running"
+        self.model.updated_at = datetime.now()
+
+        try:
+            self.worker.start(context)
+            # Async mode: we assume it started running.
+            # Use poll or refresh_status to check later.
+            self.model.status = "running"
+        except Exception:
+            self.model.status = "failed"
+            raise
+        finally:
+            self.model.updated_at = datetime.now()
+
+    def refresh_status(self) -> str:
+        """
+        Polls the worker and updates the session model status.
+        """
+        worker_status = self.worker.poll()
+        self.model.status = worker_status
+        self.model.updated_at = datetime.now()
+        return worker_status
+
+    def suspend(self):
+        print(f"Session {self.model.id}: Suspending worker")
+        self.worker.stop()
+        self.model.status = "suspended"
+        self.model.updated_at = datetime.now()
+        # In real impl, ensure git commit of current state?
+
+    def resume(self):
+        print(f"Session {self.model.id}: Resuming worker")
+        self.worker.start()  # In real impl, might need to re-init process
+
+        # Async mode: assume running
+        self.model.status = "running"
+        self.model.updated_at = datetime.now()
+
+    def terminate(self):
+        print(f"Session {self.model.id}: Terminating")
+        self.worker.stop()
+        self.model.status = "terminated"
+        self.model.updated_at = datetime.now()