aicortexpy 0.1.0__py3-none-any.whl

aicortexpy/process.py

@@ -0,0 +1,168 @@
+"""Background process helpers for Windows / POSIX."""
+
+from __future__ import annotations
+
+import os
+import subprocess
+import sys
+import time
+from pathlib import Path
+
+from .config import Config
+from .health import fetch_health, request_shutdown
+
+
+def pid_path(cfg: Config) -> Path:
+    return cfg.state_dir / "python-daemon.pid"
+
+
+def log_path(cfg: Config) -> Path:
+    return cfg.state_dir / "python-daemon.log"
+
+
+def write_pid(cfg: Config, pid: int) -> None:
+    cfg.state_dir.mkdir(parents=True, exist_ok=True)
+    pid_path(cfg).write_text(str(pid), encoding="utf-8")
+
+
+def read_pid(cfg: Config) -> int | None:
+    p = pid_path(cfg)
+    if not p.is_file():
+        return None
+    try:
+        return int(p.read_text(encoding="utf-8").strip())
+    except ValueError:
+        return None
+
+
+def clear_pid(cfg: Config) -> None:
+    try:
+        pid_path(cfg).unlink(missing_ok=True)
+    except OSError:
+        pass
+
+
+def pid_alive(pid: int) -> bool:
+    if pid <= 0:
+        return False
+    if sys.platform == "win32":
+        import ctypes
+
+        PROCESS_QUERY_LIMITED_INFORMATION = 0x1000
+        handle = ctypes.windll.kernel32.OpenProcess(
+            PROCESS_QUERY_LIMITED_INFORMATION, False, pid
+        )
+        if not handle:
+            return False
+        ctypes.windll.kernel32.CloseHandle(handle)
+        return True
+    try:
+        os.kill(pid, 0)
+        return True
+    except OSError:
+        return False
+
+
+def spawn_background(cfg: Config, extra_args: list[str] | None = None) -> int:
+    """Start `python -m aicortexpy run` detached; return child PID."""
+    cfg.state_dir.mkdir(parents=True, exist_ok=True)
+    log_file = log_path(cfg)
+    args = [
+        sys.executable,
+        "-m",
+        "aicortexpy",
+        "run",
+    ]
+    if cfg.profile:
+        args.extend(["--profile", cfg.profile])
+    if extra_args:
+        args.extend(extra_args)
+
+    log_fh = open(log_file, "a", encoding="utf-8")
+    popen_kwargs: dict = {
+        "stdin": subprocess.DEVNULL,
+        "stdout": log_fh,
+        "stderr": subprocess.STDOUT,
+        "cwd": str(Path.cwd()),
+        "close_fds": True,
+    }
+
+    if sys.platform == "win32":
+        # DETACHED_PROCESS | CREATE_NEW_PROCESS_GROUP | CREATE_BREAKAWAY_FROM_JOB
+        CREATE_BREAKAWAY_FROM_JOB = 0x01000000
+        flags = (
+            subprocess.DETACHED_PROCESS
+            | subprocess.CREATE_NEW_PROCESS_GROUP
+            | CREATE_BREAKAWAY_FROM_JOB
+        )
+        try:
+            proc = subprocess.Popen(args, creationflags=flags, **popen_kwargs)
+        except OSError:
+            # Parent job may disallow breakaway — fall back.
+            flags = subprocess.DETACHED_PROCESS | subprocess.CREATE_NEW_PROCESS_GROUP
+            proc = subprocess.Popen(args, creationflags=flags, **popen_kwargs)
+    else:
+        popen_kwargs["start_new_session"] = True
+        proc = subprocess.Popen(args, **popen_kwargs)
+
+    write_pid(cfg, proc.pid)
+    # Give health server a moment to bind
+    for _ in range(20):
+        time.sleep(0.25)
+        if fetch_health(cfg.health_port):
+            break
+    return proc.pid
+
+
+def stop_daemon(cfg: Config) -> str:
+    health = fetch_health(cfg.health_port)
+    if health:
+        if request_shutdown(cfg.health_port):
+            for _ in range(40):
+                time.sleep(0.25)
+                if not fetch_health(cfg.health_port):
+                    clear_pid(cfg)
+                    return "stopped via /shutdown"
+        return "shutdown requested but process still responding"
+
+    pid = read_pid(cfg)
+    if pid and pid_alive(pid):
+        if sys.platform == "win32":
+            subprocess.run(
+                ["taskkill", "/PID", str(pid), "/T", "/F"],
+                capture_output=True,
+                check=False,
+            )
+        else:
+            os.kill(pid, 15)
+            time.sleep(1)
+            if pid_alive(pid):
+                os.kill(pid, 9)
+        clear_pid(cfg)
+        return f"killed pid {pid}"
+
+    clear_pid(cfg)
+    return "daemon not running"
+
+
+def status_text(cfg: Config) -> str:
+    health = fetch_health(cfg.health_port)
+    if health:
+        lines = [
+            "status: running",
+            f"pid: {health.get('pid')}",
+            f"uptime: {health.get('uptime')}",
+            f"daemon_id: {health.get('daemon_id')}",
+            f"server: {health.get('server_url')}",
+            f"version: {health.get('cli_version')}",
+            f"active_tasks: {health.get('active_task_count')}",
+            f"agents: {', '.join(health.get('agents') or [])}",
+            f"workspaces: {len(health.get('workspaces') or [])}",
+            f"health: http://127.0.0.1:{cfg.health_port}/health",
+        ]
+        return "\n".join(lines)
+
+    pid = read_pid(cfg)
+    if pid and pid_alive(pid):
+        return f"status: pid {pid} alive but health port {cfg.health_port} not responding"
+    return "status: not running"

aicortexpy/procutil.py

@@ -0,0 +1,61 @@
+"""Cross-platform process-tree helpers."""
+
+from __future__ import annotations
+
+import logging
+import os
+import signal
+import subprocess
+import sys
+import time
+
+log = logging.getLogger("aicortexpy.procutil")
+
+
+def popen_kwargs_for_killable_group() -> dict:
+    """Flags so the child (and its tree) can be killed as a unit."""
+    if sys.platform == "win32":
+        return {
+            "creationflags": subprocess.CREATE_NEW_PROCESS_GROUP
+            | getattr(subprocess, "CREATE_NO_WINDOW", 0),
+        }
+    return {"start_new_session": True}
+
+
+def kill_process_tree(pid: int, *, graceful_seconds: float = 2.0) -> None:
+    if pid <= 0:
+        return
+    if sys.platform == "win32":
+        subprocess.run(
+            ["taskkill", "/PID", str(pid), "/T", "/F"],
+            capture_output=True,
+            check=False,
+        )
+        return
+
+    try:
+        os.killpg(pid, signal.SIGTERM)
+    except ProcessLookupError:
+        return
+    except PermissionError:
+        try:
+            os.kill(pid, signal.SIGTERM)
+        except ProcessLookupError:
+            return
+
+    deadline = time.time() + graceful_seconds
+    while time.time() < deadline:
+        try:
+            os.kill(pid, 0)
+        except ProcessLookupError:
+            return
+        time.sleep(0.1)
+
+    try:
+        os.killpg(pid, signal.SIGKILL)
+    except (ProcessLookupError, PermissionError):
+        try:
+            os.kill(pid, signal.SIGKILL)
+        except ProcessLookupError:
+            pass
+    log.info("killed process tree pid=%s", pid)

aicortexpy/prompt.py

@@ -0,0 +1,235 @@
+"""Task prompt builders (ported from Go daemon prompt.go)."""
+
+from __future__ import annotations
+
+from .client import Task
+from .runtime_config import build_comment_reply_instructions
+
+
+def build_prompt(task: Task, provider: str = "") -> str:
+    if task.design_mode:
+        return build_design_prompt(task)
+    if task.chat_session_id:
+        return build_chat_prompt(task)
+    if task.trigger_comment_id or task.trigger_comment_content:
+        return build_comment_prompt(task, provider)
+    if task.autopilot_run_id:
+        return build_autopilot_prompt(task)
+    if task.quick_create_prompt:
+        return build_quick_create_prompt(task)
+    return build_issue_prompt(task)
+
+
+def build_issue_prompt(task: Task) -> str:
+    parts = [
+        "You are running as a local coding agent for a AICortex workspace.\n\n",
+        f"Your assigned issue ID is: {task.issue_id}\n\n",
+    ]
+    if task.project_title or task.project_prompt:
+        if task.project_title:
+            parts.append(f"Project: {task.project_title}\n")
+        if task.project_prompt:
+            parts.append(f"Project instructions:\n{task.project_prompt}\n\n")
+    parts.append(
+        f"Start by running `aicortex issue get {task.issue_id} --output json` "
+        "to understand your task, then complete it.\n"
+        f"If you need comment history, `aicortex issue comment list {task.issue_id} "
+        "--output json` returns all comments for the issue (server caps at 2000). "
+        "Pass `--since <RFC3339>` to fetch only comments newer than a known cursor.\n"
+    )
+    return "".join(parts)
+
+
+def build_comment_prompt(task: Task, provider: str) -> str:
+    parts = [
+        "You are running as a local coding agent for a AICortex workspace.\n\n",
+        f"Your assigned issue ID is: {task.issue_id}\n\n",
+    ]
+    if task.trigger_comment_content:
+        author = "A user"
+        if task.trigger_author_type == "agent":
+            name = task.trigger_author_name or "another agent"
+            author = f"Another agent ({name})"
+        parts.append(
+            f"[NEW COMMENT] {author} just left a new comment. "
+            "Focus on THIS comment — do not confuse it with previous ones:\n\n"
+            f"> {task.trigger_comment_content}\n\n"
+        )
+        if task.trigger_author_type == "agent":
+            parts.append(
+                "The triggering comment was posted by another agent. "
+                "If you produced real work this turn, post the result as a normal reply. "
+                "If it was only acknowledgment/thanks and you produced no work, do NOT reply "
+                "and do NOT @mention the other agent (that re-triggers a loop).\n\n"
+            )
+    parts.append(
+        f"Start by running `aicortex issue get {task.issue_id} --output json` "
+        "to understand your task, then decide how to proceed.\n\n"
+        f"If you need comment history, `aicortex issue comment list {task.issue_id} "
+        "--output json`.\n\n"
+    )
+    parent = task.trigger_comment_id
+    if parent:
+        parts.append(build_comment_reply_instructions(provider, task.issue_id, parent))
+    else:
+        parts.append(
+            f"Final results MUST be delivered via `aicortex issue comment add {task.issue_id}`.\n"
+        )
+    return "".join(parts)
+
+
+def build_chat_prompt(task: Task) -> str:
+    parts = [
+        "You are running as a chat assistant for a AICortex workspace.\n",
+        "A user is chatting with you directly. Respond to their message.\n\n",
+    ]
+    if task.chat_history:
+        parts.append("Recent conversation (context only; answer the latest user message below):\n")
+        for m in task.chat_history:
+            role = str(m.get("role") or "unknown").strip() or "unknown"
+            content = str(m.get("content") or "")
+            parts.append(f"{role}:\n{content}\n\n")
+    parts.append(f"User message:\n{task.chat_message}\n")
+    atts = task.chat_message_attachments
+    if atts:
+        parts.append("\nAttachments on this message:\n")
+        for a in atts:
+            aid = a.get("id")
+            fn = a.get("filename")
+            ct = a.get("content_type") or ""
+            if ct:
+                parts.append(f'- id={aid} filename="{fn}" content_type={ct}\n')
+            else:
+                parts.append(f'- id={aid} filename="{fn}"\n')
+        incoming = task.design_incoming_rel_paths or []
+        if incoming:
+            parts.append("\nThese files are already copied into the work directory:\n")
+            for rel in incoming:
+                parts.append(f"- `{rel}`\n")
+            parts.append(
+                "Read them directly from the work directory. "
+                "Use `aicortex attachment download <id>` only if you need another copy.\n"
+            )
+        else:
+            parts.append(
+                "Use `aicortex attachment download <id>` to fetch each file locally "
+                "before referring to it.\n"
+            )
+    return "".join(parts)
+
+
+def build_autopilot_prompt(task: Task) -> str:
+    parts = [
+        "You are running as a local coding agent for a AICortex workspace.\n\n",
+        "This task was triggered by an Autopilot in run-only mode. "
+        "There is no assigned AICortex issue for this run.\n\n",
+        f"Autopilot run ID: {task.autopilot_run_id}\n",
+    ]
+    if task.autopilot_id:
+        parts.append(f"Autopilot ID: {task.autopilot_id}\n")
+    if task.autopilot_title:
+        parts.append(f"Autopilot title: {task.autopilot_title}\n")
+    if task.autopilot_source:
+        parts.append(f"Trigger source: {task.autopilot_source}\n")
+    if task.autopilot_trigger_payload.strip():
+        parts.append(f"Trigger payload:\n{task.autopilot_trigger_payload.strip()}\n")
+    parts.append("\nAutopilot instructions:\n")
+    if task.autopilot_description.strip():
+        parts.append(task.autopilot_description + "\n\n")
+    elif task.autopilot_title:
+        parts.append(task.autopilot_title + "\n\n")
+    else:
+        parts.append(
+            "No additional autopilot instructions were provided. "
+            "Inspect the autopilot configuration before proceeding.\n\n"
+        )
+    if task.autopilot_id:
+        parts.append(
+            f"Start by running `aicortex autopilot get {task.autopilot_id} --output json` "
+            "if you need the full autopilot configuration, then complete the instructions above.\n"
+        )
+    else:
+        parts.append("Complete the instructions above.\n")
+    parts.append("Do not run `aicortex issue get`; this run does not have an issue ID.\n")
+    return "".join(parts)
+
+
+def build_quick_create_prompt(task: Task) -> str:
+    parts = [
+        "You are running as a quick-create assistant for a AICortex workspace.\n\n",
+        "A user captured the following input via the quick-create modal. "
+        "There is NO existing issue. Your job is to create a well-formed issue "
+        "from this input with a single `aicortex issue create` command.\n\n",
+        f"User input:\n> {task.quick_create_prompt}\n\n",
+        "Field rules:\n\n",
+        "- **title**: required. Concise but semantically rich summary.\n\n",
+        "- **description**: High fidelity restatement of user intent "
+        "(User request + optional Context sections).\n\n",
+        "- **priority**: urgent|high|medium|low or omit.\n\n",
+        "- **assignee**: resolve names via workspace members/agents/squads JSON; "
+        "prefer `--assignee-id <uuid>`.\n",
+    ]
+    agent = task.agent or {}
+    agent_id = str(agent.get("id") or "")
+    agent_name = str(agent.get("name") or "")
+    if task.squad_id:
+        label = task.squad_name or task.squad_id
+        parts.append(
+            f'    - When the user did NOT name an assignee, default to the picker SQUAD "{label}": '
+            f'pass `--assignee-id "{task.squad_id}"`.\n\n'
+        )
+    elif agent_id:
+        parts.append(
+            f'    - When the user did NOT name an assignee, default to YOURSELF: '
+            f'pass `--assignee-id "{agent_id}"`.\n\n'
+        )
+    elif agent_name:
+        parts.append(
+            f'    - When the user did NOT name an assignee, default to YOURSELF: '
+            f'pass `--assignee "{agent_name}"`.\n\n'
+        )
+    if task.project_id:
+        title = task.project_title or task.project_id
+        parts.append(
+            f'- **project**: required. Pass `--project "{task.project_id}"` '
+            f'(project "{title}").\n'
+        )
+    else:
+        parts.append("- **project**: omit.\n")
+    parts.append(
+        "- **status**: omit (defaults to todo).\n"
+        "- **attachments**: do NOT pass `--attachment` for remote URLs.\n\n"
+        "Output format:\n"
+        "- Run exactly one `aicortex issue create --output json`.\n"
+        "- Print exactly one line: `Created <identifier-or-id>: <title>` and exit.\n"
+        "- Do NOT call `aicortex issue get` or comment add.\n"
+    )
+    return "".join(parts)
+
+
+def build_design_prompt(task: Task) -> str:
+    parts = [
+        "You are running in AICortex Design Studio mode.\n\n",
+        f"Design mode: {task.design_mode}\n",
+    ]
+    if task.design_system_name:
+        parts.append(f"Design system: {task.design_system_name}\n")
+    if task.design_system_content:
+        parts.append(f"\nDesign system content:\n{task.design_system_content}\n")
+    if task.design_example_hint:
+        parts.append(f"\nExample hint: {task.design_example_hint}\n")
+    if task.artifact_entry:
+        parts.append(f"\nPrimary artifact entry: {task.artifact_entry}\n")
+    if task.project_prompt:
+        parts.append(f"\nProject instructions:\n{task.project_prompt}\n")
+    parts.append(f"\nUser message:\n{task.chat_message}\n")
+    incoming = task.design_incoming_rel_paths or []
+    if incoming:
+        parts.append("\nIncoming files already in the work directory:\n")
+        for rel in incoming:
+            parts.append(f"- `{rel}`\n")
+    parts.append(
+        "\nProduce the design artifacts in the work directory. "
+        "Prefer HTML under design/ when applicable.\n"
+    )
+    return "".join(parts)