Files
roboco/roboco/runtime/orchestrator.py
T
0e9f21de69 fix(api): default event loop to asyncio + cancellation-safe commit — kills the CI segfault (#340)
* fix(api): default the event loop to asyncio + cancellation-safe commit

The recurring CI e2e segfault traced to uvloop: the harness's
uvicorn.run() auto-selected it while production's serve() path never
consulted Config.loop (stock asyncio, accidentally safe). Every launch
site now resolves ROBOCO_UVICORN_LOOP (default asyncio; uvloop opt-in),
and DbCommitMiddleware's commit-in-send can no longer be interrupted
mid-wire: on cancellation it gets a bounded grace to finish (committed
data survives the 504), else invalidate-and-reraise.

* feat(runtime): expected-stop breadcrumbs attribute container deaths

Two production exit-143s had no attributable source: every orchestrator
kill path now records a short reason breadcrumb, and the exit monitor
consumes it -- an expected stop logs its reason at info, a genuinely
unexpected one logs none_recorded plus docker-inspect diagnostics
(OOMKilled, timestamps) so the next mystery SIGTERM self-identifies.

---------

Co-authored-by: Renn F <rennf93@users.noreply.github.com>
2026-07-08 16:01:01 +02:00

13696 lines
585 KiB
Python

"""
Agent Orchestrator
Manages Claude Code containers for all RoboCo agents.
Handles spawning, monitoring, health checks, and graceful shutdown.
The orchestrator is the BRAIN of the system:
- Checks for work BEFORE spawning agents (no wasteful spawns)
- Claims tasks on behalf of agents before spawning
- Agents receive their assignment at spawn time
- Agents scan for more work after completing a task
- Agents only call i_am_idle() when truly no work remains
"""
import asyncio
import contextlib
import json
import os
import shutil
import tempfile
import time
from dataclasses import dataclass
from datetime import UTC, datetime, timedelta
from pathlib import Path
from typing import TYPE_CHECKING, Any, ClassVar, cast
import httpx
if TYPE_CHECKING:
from collections.abc import Callable, Coroutine, Iterable
from uuid import UUID
from sqlalchemy.ext.asyncio import AsyncSession
from roboco.llm.providers import AgentProvider, ProviderRegistry
from roboco.services.llm import AgentRoute
from roboco.services.task import TaskService
import structlog
from fastapi import status as http_status
from roboco.agents.factories._base import compose_prompt
from roboco.agents_config import (
ALL_DOCS,
get_agent_role,
get_agent_team,
get_escalation_target,
)
from roboco.config import settings
from roboco.foundation import identity as _foundation
from roboco.foundation.identity import (
CELL_TEAMS,
is_human_only_role,
is_spawnable_agent_slug,
role_for_slug_or_none,
)
from roboco.foundation.policy.agent_loop import DEFAULT_BUDGET as _AGENT_LOOP_BUDGET
from roboco.foundation.policy.batch import is_branchless_coordination
from roboco.foundation.policy.content import markers as _markers
from roboco.models import AgentRole, Team
from roboco.models.base import ModelProvider
from roboco.models.runtime import (
MODEL_MAP,
ROLE_EFFORT_MAP,
ROLE_MODEL_MAP,
AgentInstance,
OrchestratorAgentConfig,
OrchestratorAgentState,
SpawnGitContext,
WaitingRecord,
)
from roboco.models.sandbox import SandboxInfo
from roboco.runtime.sandbox import SandboxProvisioner
from roboco.seeds.initial_data import AGENT_UUIDS
from roboco.services.task import (
PR_REVIEW_SOURCES,
RELEASE_MANAGER_SOURCE,
ROADMAP_SOURCE,
SELF_HEAL_SOURCE,
VIDEO_HELD_SOURCES,
X_FEATURE_EXPLORATION_SOURCE,
X_SOURCES,
)
logger = structlog.get_logger()
# Reverse mapping: UUID -> slug
UUID_TO_SLUG = {uuid: slug for slug, uuid in AGENT_UUIDS.items()}
# Re-export for backwards compatibility
AgentState = OrchestratorAgentState
AgentConfig = OrchestratorAgentConfig
# Docker configuration
AGENT_NETWORK = "roboco_default"
AGENT_BASE_IMAGE = "roboco-agent-base"
# Port on which each agent's Claude Code SDK server listens inside its container.
# Referenced by write-hooks (_finalize_spawn_session, _sweep_token_snapshots,
# _sweep_budget_exceeded) to build the SDK health/usage URL.
SDK_PORT: int = 9000
# Provider-recovery probe: a free, unmetered liveness call confirms a parked
# provider is accepting requests again before parked agents are resumed.
# Listing models / tags costs no tokens; only a 2xx response means recovered
# (a 429 rate limit OR a 5xx overload both keep the provider parked).
_ANTHROPIC_PROBE_BASE = "https://api.anthropic.com"
_PROBE_TIMEOUT_SECONDS = 10.0
# Docker subprocess deadlines for the reaper path. A hung Docker daemon or a
# stuck container FS would otherwise freeze the single asyncio event loop: the
# reaper runs inline before every dispatch tick and shares that loop with every
# background sweeper. Generous enough that a legitimate slow docker call (a
# loaded daemon, a cold-venv ``import httpx, mcp``) is never wrongly aborted;
# short enough that a hang degrades one tick, not the whole fleet.
_DOCKER_INSPECT_TIMEOUT_SECONDS = 10.0
_DOCKER_EXEC_TIMEOUT_SECONDS = 30.0
# Deadline for draining fire-and-forget ``_bg_tasks`` on shutdown. Short DB
# writes (a respawn_tracker upsert, an audit-log row) finish before the
# process exits — preserving the durable PM-respawn counter and the
# metrics-bearing audit trail — while a stuck task can't hang shutdown: past
# this deadline the still-pending tasks are cancelled. Generous enough that a
# legitimate slow write under load commits rather than being dropped (the
# exact data-loss tail the durable tracker exists to prevent).
_SHUTDOWN_DRAIN_TIMEOUT_SECONDS = 5.0
# Attribution breadcrumbs for orchestrator-initiated container stops (see
# _record_expected_stop). A breadcrumb older than this is treated as unrelated
# to whatever exit the monitor is now looking at, rather than mis-attributed.
_EXPECTED_STOP_FRESH_SECONDS = 120.0
_EXPECTED_STOP_MAX_ENTRIES = 200
_HTTP_TOO_MANY_REQUESTS = 429
_HTTP_OK = 200
_HTTP_MULTIPLE_CHOICES = 300 # first non-2xx status; 2xx == [_HTTP_OK, this)
# The orchestrator calls its own write API as a trusted internal actor. Those
# routes require an agent identity (X-Agent-ID); a self-call without it is
# rejected 401, so silent recovery ops (auto-block / auto-resume / auto-recover
# / SLA annotation) no-op and paused/blocked parents wedge. The system identity
# holds TaskAction.ASSIGN, so it is authorized for the audited admin_set_status
# path those routes use. EVERY dispatcher client that can reach the API must
# carry it — header propagation was previously inconsistent across the separate
# AsyncClient call-sites, so only some paths were authenticated.
_SYSTEM_API_HEADERS = {
"X-Agent-ID": "00000000-0000-0000-0000-000000000000",
"X-Agent-Role": "system",
}
def _system_api_headers() -> dict[str, str]:
"""System identity headers for the orchestrator's internal self-API calls.
Wraps ``_SYSTEM_API_HEADERS`` and adds a signed ``X-Agent-Token`` for the
system identity (F038/F039). Without it, arming
``ROBOCO_AGENT_AUTH_REQUIRED=true`` 401s every silent recovery op
(auto-block / auto-resume / auto-recover / SLA annotation) and wedges
paused/blocked parents — the prior self-PATCH 401 fix only carried
``X-Agent-ID`` / ``X-Agent-Role``, so it was incomplete under auth-required.
When the HMAC secret is unset (dev), ``issue_agent_token`` returns the
``UNSIGNED`` sentinel and auth isn't required, so the self-call still
succeeds; the header is present either way so a future arm-when-secret-set
doesn't silently break.
"""
from roboco.agents_config import issue_agent_token
return {
**_SYSTEM_API_HEADERS,
"X-Agent-Token": issue_agent_token(
_SYSTEM_API_HEADERS["X-Agent-ID"], "system", ""
),
}
def _agent_api_headers(agent_uuid: str, role: str) -> dict[str, str]:
"""Headers for the orchestrator's internal self-API calls acting as a
specific agent (the cell-PM auto-submit). Adds the signed ``X-Agent-Token``
+ ``X-Agent-Team`` so the call passes the ``ROBOCO_AGENT_AUTH_REQUIRED``
gate — a hand-built ``{X-Agent-ID, X-Agent-Role}`` dict 401s with
"Missing X-Agent-Token" under auth-required (F038/F039 — the same gap the
system-headers helper closes for the system identity).
The token is attached only when ``ROBOCO_AGENT_AUTH_SECRET`` is set: the
dev-mode middleware rejects a presented-but-unverifiable token (the
``UNSIGNED`` sentinel) with 401 "signature mismatch" while accepting a
missing token, so sending ``UNSIGNED`` would turn a clean dev self-call
into a 401. With the secret armed the token is signed and verifies.
"""
from roboco.agents_config import _auth_secret, issue_agent_token
team = get_agent_team(agent_uuid) or ""
headers = {"X-Agent-ID": agent_uuid, "X-Agent-Role": role}
if team:
headers["X-Agent-Team"] = team
if _auth_secret():
headers["X-Agent-Token"] = issue_agent_token(agent_uuid, role, team)
return headers
# Consecutive failed recovery probes before the CEO is notified once per episode.
_CEO_NOTIFY_THRESHOLD = 10
# Consecutive strategy-engine cycle failures before the CEO is notified once
# per failure episode (#193). Mirrors _CEO_NOTIFY_THRESHOLD so a persistently
# failing assess() (bad DB / goals row) surfaces instead of silently producing
# nothing every tick.
_STRATEGY_FAIL_CEO_NOTIFY_THRESHOLD = 10
# Persistent-probe-failure escape hatch (F094): if the recovery probe keeps
# failing past this threshold, the probe endpoint itself is the problem (a
# misconfigured URL, a removed API key, a network partition to the probe host)
# while the provider may well be fine for real workloads. Hold the park any
# longer and every agent on the provider strands forever with only a one-shot
# CEO notification. Past this threshold, fall back to the same time-expiry
# optimism the unprobeable-provider path uses (``_do_probe`` returns True when
# there is no probe URL): clear the park and resume. If the provider is
# genuinely still down the real workload attempts re-park via the 429/5xx path,
# so this is bounded burn — strictly better than a silent forever-strand. Kept
# above the CEO-notify threshold so the operator gets the notification first.
_PROBE_GIVE_UP_THRESHOLD = 30
# Persistent server-overload parking (HTTP 529 / 500 / 503). The model API's
# SDK already retries transient overloads in-process; only a persistent one
# survives to kill the run. When it does, park the provider like a 429 instead
# of crash-retrying into the overload. These markers are matched (lowercased,
# substring) against the tail of the dead container's own output, so they are
# kept specific to how the API surfaces an overload. Bare "error 529"/"error
# 500"/"error 503" were dropped (F037): an agent that merely writes about an
# HTTP status code in its own notes ("the endpoint returned error 500,
# retrying") would false-match and park the whole Anthropic fleet. The SDK
# error formatter emits "API Error: NNN" + a JSON error type, so the
# ``api error: NNN`` and type-string markers below cover every real overload
# without that false-match surface.
_OVERLOAD_RETRY_AFTER_S = 45.0
_ANTHROPIC_OVERLOAD_MARKERS: tuple[str, ...] = (
"overloaded_error",
"internal_server_error",
"api error: 529",
"api error: 500",
"api error: 503",
)
# Session / usage-limit parking (HTTP 429). The Claude session ("5-hour") limit
# crashes the agent container with a 0-token rejection that is NOT a 5xx
# overload, so without its own markers it falls through to crash-respawn —
# straight back into the limit until the window resets. Park the provider like a
# 429 instead and let the probe-resume loop revive the parked tasks once the
# quota clears. Markers are specific to how the session limit surfaces (matched
# lowercased, substring) so they can't false-match an agent writing about
# limits; the probe (which also hits the same limit) keeps the park until reset.
# Reuses the longer overload retry cadence — probing a multi-hour window every
# few seconds is wasteful, and each probe is itself a rejected call.
_RATE_LIMIT_RETRY_AFTER_S = 300.0
_ANTHROPIC_RATE_LIMIT_MARKERS: tuple[str, ...] = (
"hit your session limit",
"five_hour",
)
# ollama.com HTTP 429 body (the weekly glm-5.2:cloud limit surfaces here).
# Specific to the API error formatter so an agent writing about limits can't
# false-match and park the whole ollama fleet.
_OLLAMA_RATE_LIMIT_MARKERS: tuple[str, ...] = ("rate limit exceeded",)
# ponytail: marker map drives the detector — adding a provider later is a
# table row, not a new branch. Grok is deliberately absent (exit-75 detector).
_RATE_LIMIT_MARKERS_BY_PROVIDER: dict[str, tuple[str, ...]] = {
ModelProvider.ANTHROPIC.value: _ANTHROPIC_RATE_LIMIT_MARKERS,
ModelProvider.OLLAMA_CLOUD.value: _OLLAMA_RATE_LIMIT_MARKERS,
}
_OVERLOAD_MARKERS_BY_PROVIDER: dict[str, tuple[str, ...]] = {
ModelProvider.ANTHROPIC.value: _ANTHROPIC_OVERLOAD_MARKERS,
}
# The intake (prompter) agent: a single seeded, board-adjacent interviewer.
# Unlike delivery agents it is never dispatched and runs ONE persistent
# container at a time (single CEO → one live chat). See the INTAKE section
# below and roboco/agent_sdk/intake_main.py.
INTAKE_AGENT_ID = "intake-1"
# The Secretary agent: a single seeded, persistent chief-of-staff container the
# CEO chats with (like intake), but with gated CEO authority. One container at a
# time. Seeded in identity.AGENTS; see roboco/agent_sdk/secretary_main.py.
SECRETARY_AGENT_ID = "secretary-1"
# Role -> Image mapping
# Specialized images extend the base with role-specific tools
AGENT_IMAGES: dict[str, str] = {
# Backend
"be-dev-1": "roboco-agent-dev-be",
"be-dev-2": "roboco-agent-dev-be",
"be-qa": "roboco-agent-qa-be",
"be-pm": "roboco-agent-pm",
"be-doc": "roboco-agent-doc",
# Frontend
"fe-dev-1": "roboco-agent-dev-fe",
"fe-dev-2": "roboco-agent-dev-fe",
"fe-qa": "roboco-agent-qa-fe",
"fe-pm": "roboco-agent-pm",
"fe-doc": "roboco-agent-doc",
# UX/UI
"ux-dev-1": "roboco-agent-ux",
"ux-dev-2": "roboco-agent-ux",
"ux-qa": "roboco-agent-ux", # Uses same as dev for now
"ux-pm": "roboco-agent-pm",
"ux-doc": "roboco-agent-doc",
# Board
"main-pm": "roboco-agent-pm",
"product-owner": "roboco-agent-pm",
"head-marketing": "roboco-agent-pm",
"auditor": "roboco-agent-pm",
# PR Reviewer — read-only reviewer (diff via API, grep, post one
# change-request; never runs code). Its own image for parity with the other
# agents; built FROM the base, no extra toolchain. The three cell reviewers
# are additional instances of the same role and reuse the same image (as
# be-dev-1/-2 share one dev image) — the in-path gate adds no new image.
"pr-reviewer-1": "roboco-agent-pr-reviewer",
"be-pr-reviewer": "roboco-agent-pr-reviewer",
"fe-pr-reviewer": "roboco-agent-pr-reviewer",
"ux-pr-reviewer": "roboco-agent-pr-reviewer",
# Intake — persistent Agent-SDK driver, not a one-shot `claude -p`.
INTAKE_AGENT_ID: "roboco-agent-prompter",
# Secretary — persistent Agent-SDK driver with gated CEO authority.
SECRETARY_AGENT_ID: "roboco-agent-secretary",
}
def _qualify_agent_image(bare: str) -> str:
"""Apply the configured registry namespace + tag to a bare agent image.
Default (no ``agent_image_registry``, no ``agent_image_tag``) returns the
bare name unchanged — the local build flow. With a registry set the
orchestrator spawns (and ensures) ``{registry}/roboco-agent-*[:tag]``, the
pre-built images the release workflow publishes, instead of building.
"""
registry = settings.agent_image_registry.rstrip("/")
name = f"{registry}/{bare}" if registry else bare
tag = settings.agent_image_tag
return f"{name}:{tag}" if tag else name
def get_agent_image(agent_id: str) -> str:
"""Get the Docker image for an agent (registry-qualified when configured)."""
return _qualify_agent_image(AGENT_IMAGES.get(agent_id, AGENT_BASE_IMAGE))
# When running in a container, we need host paths for volume mounts.
# These can be overridden via environment variables.
CLAUDE_AUTH_HOST_PATH = os.environ.get(
"ROBOCO_HOST_CLAUDE_DIR",
str(Path.home() / ".claude"),
)
PROJECT_HOST_PATH = os.environ.get("ROBOCO_HOST_PROJECT_DIR", "")
DATA_HOST_PATH = os.environ.get("ROBOCO_HOST_DATA_DIR", "")
# In-orchestrator path where each GROK agent's usage capture is visible. The
# agent writes <DATA_HOST_PATH>/grok-usage/<agent_id>/usage.json; the compose file
# mounts the same host dir here so the finalizer can read the captured tokens back
# (the grok analogue of reading the Claude transcript from the mounted ~/.claude).
# Override for local runs.
GROK_USAGE_DATA_DIR = os.environ.get("ROBOCO_GROK_USAGE_DIR", "/data/grok-usage")
# Interactive Grok images (grok-CLI conversation drivers) — selected for the
# intake / secretary roles when their route resolves to GROK, instead of the
# Claude prompter/secretary images. Their dockerfiles build FROM roboco-agent-grok.
GROK_PROMPTER_IMAGE = "roboco-agent-grok-prompter"
GROK_SECRETARY_IMAGE = "roboco-agent-grok-secretary"
_GROK_INTERACTIVE_DOCKERFILES = {
GROK_PROMPTER_IMAGE: "agent-grok-prompter.Dockerfile",
GROK_SECRETARY_IMAGE: "agent-grok-secretary.Dockerfile",
}
# A one-shot Grok container exits with this code (EX_TEMPFAIL) when the run hit
# an xAI 429 (grok-cli-agent-entrypoint.sh detects it). The orchestrator parks the
# grok provider rate-limited instead of crash-retrying, breaking the
# 429 -> exit -> respawn cost loop. The probe-resume loop clears the park after
# the retry window (unknown-provider time-expiry fallback in _probe_target).
_GROK_RATE_LIMIT_EXIT_CODE = 75
_GROK_RATE_LIMIT_RETRY_AFTER_S = 60.0
# Grok has no real recovery probe (the SuperGrok OIDC token is not a valid
# bearer for the metered api.x.ai, so a probe would no-op or strand grok
# parked). The probe loop clears a grok park on a timer; the fresh agent hits
# the still-active xAI 429, exits 75, and re-parks. Back the re-park retry_after
# off exponentially within one episode so the churn dampens (60 -> 120 -> 240
# -> ... capped) instead of spinning flat. The episode gap resets the count
# once the rate limit has actually lifted.
_GROK_REPARK_BACKOFF_CAP = 4 # max 2**4 = 16x base (~16min cycle)
_GROK_REPARK_EPISODE_GAP_S = 1500.0 # 25min — > the capped ~16min cycle
# A one-shot Grok container exits with this code (EX_CONFIG) when the
# entrypoint's `grok_auth --check` backstop found the access token missing or
# expired (it can't be refreshed headlessly, so the CLI would hang at an
# interactive login prompt). Park the provider instead of crash-retrying 3x —
# the agent cannot start without a valid token, so respawning burns tokens for
# zero progress. The probe-resume loop revives the task once
# grok_auth.refresh_if_stale (run once per dispatch tick) mints a fresh token
# from the offline-access refresh token; if still expired, the next exit 78
# re-parks (no token burn). Same shape as the 429 exit-75 path (F041).
_GROK_AUTH_EXIT_CODE = 78
_GROK_AUTH_RETRY_AFTER_S = 60.0
# =============================================================================
# ORCHESTRATOR
# =============================================================================
@dataclass(frozen=True)
class _SlaBreach:
"""Per-(role, state) SLA breach payload for _escalate_sla_breach."""
task_id: str
role: str
status: str
age_seconds: int
sla_seconds: int
@dataclass(frozen=True)
class _IntakeRunSpec:
"""Inputs for ``_build_intake_run_cmd``, bundled to keep the signature small."""
container_name: str
image: str
hosts: dict[str, str | None]
session_id: str
cwd: str
cli_model: str
api_url: str
provider_base_url: str | None
provider_auth_token: str | None
provider_type: str = "anthropic"
model: str = ""
@dataclass
class _StrategyLoopState:
"""Consecutive-failure tracking for ``_strategy_engine_loop`` (#193).
``failures`` counts consecutive cycle exceptions; ``notified`` gates the
one-CEO-alert-per-episode. Both reset on the first success so a fresh
failure episode re-notifies.
"""
failures: int = 0
notified: bool = False
@dataclass
class _SecretaryRunSpec:
"""Inputs for ``_build_secretary_run_cmd`` (mirrors ``_IntakeRunSpec``).
Adds the agent uuid + HMAC token: unlike intake, the Secretary's tools call
the backend, so the container needs an authenticated identity.
"""
container_name: str
image: str
hosts: dict[str, str | None]
session_id: str
cwd: str
cli_model: str
api_url: str
agent_uuid: str
agent_token: str
provider_base_url: str | None
provider_auth_token: str | None
provider_type: str = "anthropic"
model: str = ""
# Roles that always work a concrete task — a spawn row with ``task_id IS NULL``
# for one of these is an unattributed-cost bug (the usage rollup can't tie the
# spend to a task). Intake (prompter), secretary, auditor, and PMs legitimately
# spawn taskless, so they are NOT flagged (#11).
_TASKLESS_SPAWN_SUSPECT_ROLES = frozenset({"developer", "qa", "documenter"})
def is_unattributed_delivery_spawn(role: str, task_id: str | None) -> bool:
"""True when a delivery-role spawn carries no ``task_id`` (#11).
The role string comes from ``get_agent_role`` (lowercase); the comparison is
case-insensitive for safety. Used by ``_record_spawn_session`` to warn on
unattributed usage without noise from the intentional taskless roles.
"""
return task_id is None and role.lower() in _TASKLESS_SPAWN_SUSPECT_ROLES
def _read_project_slug(task: dict[str, Any]) -> str | None:
"""Extract project slug from a task payload shape-tolerantly."""
slug = task.get("project_slug")
if slug:
return str(slug)
project = task.get("project") or {}
inner = project.get("slug") if isinstance(project, dict) else None
return str(inner) if inner else None
def _is_coordination_task(task: dict[str, Any]) -> bool:
"""True for a task that does no git of its own.
Three shapes qualify: a board/fan-out coordination root (carries a product,
no repo — its cell subtasks resolve a real project from the product's
cell->project map), an ad-hoc per-cell map coordination root (carries a
``cell_projects`` map but no project/product — a multi-cell MegaTask
root-subtask), and a MegaTask umbrella (carries a batch_id, top-level — its
root-subtasks each carry their own branch/PR). Such a task has no
project_slug, branch_name, or git token, and must NOT be git-gated at the
spawn-readiness or stuck-detection checks the way a code task is. A task with
none of project / product / cell-map / batch is genuinely unroutable and
stays gated.
"""
return is_branchless_coordination(
project_id=task.get("project_id"),
product_id=task.get("product_id"),
batch_id=task.get("batch_id"),
parent_task_id=task.get("parent_task_id"),
has_cell_projects=bool(task.get("cell_projects")),
)
# A branch is auto-created only at CLAIM (the claimed->in_progress transition).
# Before that — while a task is still pending/backlog awaiting first dispatch —
# it legitimately has no branch_name, so the readiness / stuck / spawn checks
# must NOT treat a missing branch as a defect. These are the only states where
# a code task is expected to already own a branch.
_BRANCH_EXPECTED_STATES: frozenset[str] = frozenset(
{"claimed", "in_progress", "verifying"}
)
def _branch_is_expected(task: dict[str, Any]) -> bool:
"""True iff this task should already have a branch_name.
A branch only exists at/after claim, and a coordination/fan-out task never
gets one (it does no git of its own). Gating the "missing branch_name"
readiness/stuck condition on this predicate stops the orchestrator from
auto-blocking a never-claimed PENDING code task that simply hasn't reached
the claim transition yet (a pending task sat 13min, auto-blocked
every 30s, never dispatched).
"""
if _is_coordination_task(task):
return False
return str(task.get("status") or "") in _BRANCH_EXPECTED_STATES
def _resolve_agent_cli_model(provider_type: str, model: str) -> str:
"""Translate an agent model name to the string Claude Code expects.
For the Anthropic provider, short names (``opus|sonnet|haiku``) are
translated through ``MODEL_MAP`` as they always were. For non-Anthropic
providers (currently Ollama Cloud) the model identifier is passed verbatim
so raw tags like ``kimi-k2.6:cloud`` reach the Ollama-side integration
intact.
Extracted as a module-level function so both the ``--model`` CLI arg
builder and the ``CLAUDE_CODE_SUBAGENT_MODEL`` env-var injector can call
the same logic without referencing the class by name inside a staticmethod.
"""
if provider_type == "anthropic":
return MODEL_MAP.get(model, model)
return model
def _agent_workspace_path(project_slug: str, team: str, agent_id: str) -> str:
"""Per-agent workspace path inside the container.
Mirrors the bind-mount layout: the host's workspaces dir is mounted at
/data/workspaces (orchestrator.py mount args), so each agent's clone lives
at /data/workspaces/<project>/<team>/<agent>. Used by both
_get_role_permissions (Edit/Write allowlist) and _build_mount_args
(docker ``-w`` flag) so the cwd matches the allowlist scope.
"""
return f"/data/workspaces/{project_slug}/{team}/{agent_id}"
def _agent_worktree_path(
project_slug: str, team: str, agent_id: str, task_short_id: str
) -> str:
"""Per-task worktree path inside the container (F123).
Each task with a branch gets its own working tree under the clone root at
``{clone_root}/.worktrees/{task_short_id}/`` so a coordinator PM's parallel
roots (or a dev's parallel tasks) never clobber one shared checkout.
"""
return (
f"/data/workspaces/{project_slug}/{team}/{agent_id}/.worktrees/{task_short_id}"
)
def _agent_cwd_path(
project_slug: str,
team: str,
agent_id: str,
git_context: SpawnGitContext | None,
) -> str:
"""The container cwd + Edit/Write scope for a workspace role (F123).
A task carrying a branch edits in its per-task worktree; a branchless or
no-task spawn stays at the clone root. ONE formula shared by
``_append_workspace_cwd`` (docker ``-w``) and ``_get_role_permissions``
(Edit/Write allowlist via ``_prepare_agent_spawn``) so the cwd and the
allowlist scope can never drift to different paths.
"""
clone_root = _agent_workspace_path(project_slug, team, agent_id)
if git_context and git_context.task_short_id:
return _agent_worktree_path(
project_slug, team, agent_id, git_context.task_short_id
)
return clone_root
def _cell_workspace_path(project_slug: str, team: str) -> str:
"""Cell-level workspace path (documenter scope).
Same rationale as ``_agent_workspace_path``; documenters work at the cell
branch, not a per-agent dev branch.
"""
return f"/data/workspaces/{project_slug}/{team}"
def _resolve_project_slug_from_git_context(
git_context: "SpawnGitContext | None",
) -> str:
"""Extract project_slug from git_context, falling back to 'default'.
Module-level counterpart to the instance method ``_resolve_project_slug``.
Called by static / classmethod contexts (e.g. ``_build_mount_args``) that
cannot access ``self``. The fallback warning is omitted here because the
instance method already logs it when the full spawn path runs; this helper
is only for the mount-args path where the agent_id/task_id context is not
available.
"""
if git_context and git_context.project_slug:
return git_context.project_slug
return "default"
# =============================================================================
# SPAWN MANIFEST — per-developer tool manifest mounting (Phase 1)
# =============================================================================
# Phase 4: every spawned role gets a gateway manifest. The legacy briefing path
# is gone. A role omitted here gets NO manifest and ROBOCO_GATEWAY_ENABLED=false,
# i.e. none of its flow verbs are pre-registered — so it can never claim its work
# and the dispatcher respawns it on the same task forever. The only roles that
# may be absent are the human-only ones (prompter, secretary) that the
# orchestrator never spawns as delivery agents.
GATEWAY_ENABLED_ROLES: frozenset[str] = frozenset(
{
"developer",
"qa",
"documenter",
"cell_pm",
"main_pm",
"product_owner",
"head_marketing",
"auditor",
"pr_reviewer",
}
)
def _build_manifest_for_agent(
agent_id: str, model: str, workspace_path: str | None = None
) -> Path | None:
"""Write a SpawnManifest for developer-role agents; return the host path.
Returns ``None`` for roles outside ``GATEWAY_ENABLED_ROLES`` so callers
can skip the manifest mount entirely without extra branching.
Args:
agent_id: Agent slug (e.g. ``be-dev-1``).
model: Resolved model name passed to ``SpawnInputs.agent_model``.
workspace_path: The task-resolved workspace (project clone or per-task
worktree) — the SAME path the container ``-w`` uses. Without it
the manifest falls back to the agent's roboco-project workspace,
which is WRONG for any other project's task (live 2026-07-02:
be-dev-2's manifest pointed at /data/workspaces/roboco while the
task lived in guard-core-saas-backend).
Returns:
Absolute host path to the written JSON file, or ``None``.
"""
from uuid import UUID
from roboco.runtime.spawn_manifest import (
SpawnInputs,
build_for_role,
write_manifest,
)
role = get_agent_role(agent_id) or "developer"
if role not in GATEWAY_ENABLED_ROLES:
return None
team = get_agent_team(agent_id) or "backend"
# UUID for the agent comes from the seeded AGENT_UUIDS map (slug -> UUID
# string). Fall back to uuid4 for unknown agents so the function stays
# callable in tests without seeded data.
raw_uuid = AGENT_UUIDS.get(agent_id)
agent_uuid = UUID(raw_uuid) if raw_uuid else __import__("uuid").uuid4()
resolved_workspace = (
Path(workspace_path)
if workspace_path
else Path(settings.workspaces_root) / "roboco" / team / agent_id
)
manifest = build_for_role(
SpawnInputs(
agent_id=agent_uuid,
role=role,
team=team,
workspace_path=resolved_workspace,
agent_model=model,
)
)
# Two paths in play:
# - orchestrator-internal: where the file is written inside the
# orchestrator container (settings.manifest_host_dir). The compose
# volume mount makes this dir visible on the host.
# - host-side: what the docker daemon needs for the bind-mount into
# the spawned agent. Computed via DATA_HOST_PATH translation.
write_dir = Path(settings.manifest_host_dir)
write_path = write_dir / f"{agent_id}.json"
write_manifest(manifest, write_path)
if DATA_HOST_PATH:
return Path(f"{DATA_HOST_PATH}/manifests/{agent_id}.json")
return write_path
class AgentReadinessError(Exception):
"""Raised when spawn_agent refuses to spawn because the task isn't ready.
The pre-flight gate auto-blocks the offending task before raising, so the
dispatcher doesn't keep retrying. Callers should log and move on.
"""
class _SpawnAbortedDuringShutdown(Exception):
"""Raised when a non-blocking intake/secretary spawn completes ``docker run``
after the orchestrator began shutting down.
The raiser has already removed the just-started container (so it isn't
orphaned); the guarded wrapper catches this BEFORE its generic
``except Exception`` and closes the live relay silently — shutdown is not a
user-facing failure, so no error is pushed to the SSE stream. The F070
``stop()`` drain awaits the bg spawn coroutine, so this surfaces cleanly
instead of the registration landing a live container into a registry that
``stop()`` has already finished iterating.
"""
def _is_held_ceo_source(task: dict[str, Any]) -> bool:
"""True for sources the PM dispatcher must never route as delivery work.
External-PR review (owned by the PR dispatcher), release proposals, X
posts/replies, and video-post drafts (all CEO-HELD, acted on only by
their own routes), and a self-heal fix task until the CEO's
approve_and_start flips ``confirmed_by_human``. Module-level (not a
method) so the dispatcher's unit tests, which drive it with a
wholesale-mocked ``self``, exercise the real skip logic rather than an
auto-mocked stub.
"""
source = task.get("source")
if source in PR_REVIEW_SOURCES:
return True
if source == RELEASE_MANAGER_SOURCE:
return True
if source in X_SOURCES:
return True
if source in VIDEO_HELD_SOURCES:
return True
return source == SELF_HEAL_SOURCE and not task.get("confirmed_by_human")
def _is_non_dev_dispatch_source(task: dict[str, Any]) -> bool:
"""Sources ``_dispatch_dev_work`` must skip: every CEO-held source plus the
Board exploration cycles (``board_roadmap`` / feature-spotlight exploration)
that ``_dispatch_pm_work`` owns. One flat call keeps the dev loop's skip out
of a long per-source ``if`` chain (xenon budget)."""
if _is_held_ceo_source(task):
return True
return task.get("source") in (ROADMAP_SOURCE, X_FEATURE_EXPLORATION_SOURCE)
# Bounded retry for the video render loop: a failed render (read-clone not yet
# synced to the just-merged composition, or a transient sidecar blip) retries on
# a later cycle; only after this many attempts is a task marked terminally
# failed, so a genuinely broken composition can't re-render forever.
_MAX_VIDEO_RENDER_ATTEMPTS = 5
class AgentOrchestrator:
"""
Manages Claude Code containers for all agents.
Responsibilities:
- Spawn agents as Docker containers
- Monitor health via docker inspect
- Handle waiting states and respawning
- Provide status API
- Cost-efficient on-demand spawning
"""
def __init__(
self,
mcp_config_dir: Path | None = None,
project_root: Path | None = None,
dispatcher_interval: int = 30,
):
self.mcp_config_dir = mcp_config_dir or Path(".mcp")
self.project_root = project_root or Path.cwd()
self.dispatcher_interval = dispatcher_interval
self._instances: dict[str, AgentInstance] = {}
# Sandboxed per-agent-spawn DB/Redis provisioner. Network is threaded
# through explicitly (rather than the provisioner importing
# AGENT_NETWORK itself) so a future network-isolation change only
# has to flip this constant here — sandboxes ride along.
self._sandbox = SandboxProvisioner(network=AGENT_NETWORK)
# Gateway-health grace tracker: agent slug -> first time its gateway was
# seen broken. Tolerates a transient probe miss before the reaper recovers
# a broken-but-alive agent (see _maybe_recover_broken_gateway).
self._gateway_broken_since: dict[str, datetime] = {}
self._waiting_records: dict[str, WaitingRecord] = {}
# Diagnostics only, in-memory: agent_id -> (reason, monotonic ts) for
# the most recent orchestrator-initiated stop/kill, so the exit
# monitor can tell an attributed stop from a truly unexplained one
# (see _record_expected_stop / _consume_expected_stop).
self._expected_stops: dict[str, tuple[str, float]] = {}
# #71: a resumed agent's WaitingRecord is torn down only once liveness is
# confirmed (not on a bare launch) — a container that launches then dies
# immediately would otherwise strand its task until the reaper's TTL.
self._resume_confirm_delay: float = 30.0
self._health_task: asyncio.Task | None = None
self._dispatcher_task: asyncio.Task | None = None
self._sweeper_task: asyncio.Task | None = None
# Last time the transcript-retention prune ran (throttled in the sweep).
self._last_transcript_prune: datetime | None = None
self._last_image_prune: datetime | None = None
# Rate-limit probe loop: 30-second interval, scans Redis for all
# rate-limited providers and resolves waiting agents on success.
self._rate_limit_probe_task: asyncio.Task | None = None
self._strategy_engine_task: asyncio.Task | None = None
self._external_pr_poll_task: asyncio.Task | None = None
self._self_heal_task: asyncio.Task | None = None
self._ci_watch_task: asyncio.Task | None = None
self._dep_update_task: asyncio.Task | None = None
self._release_manager_task: asyncio.Task | None = None
self._x_mentions_task: asyncio.Task | None = None
self._roadmap_engine_task: asyncio.Task | None = None
self._x_feature_spotlight_task: asyncio.Task | None = None
self._video_render_task: asyncio.Task | None = None
# per-engine-loop heartbeat (monotonic last-success, interval) so
# _check_loop_liveness can alert when a cycle task dies silently.
self._loop_heartbeats: dict[str, tuple[float, float]] = {}
# Provider registry: maps a ModelProvider to a dedicated AgentProvider
# backend. Only providers needing a non-Claude-Code runtime are
# registered (currently GROK, which speaks the OpenAI protocol). Agents
# on unregistered providers (Anthropic / Ollama Cloud / self-hosted) use
# the built-in _spawn_container path unchanged. Built lazily.
self._provider_registry: ProviderRegistry | None = None
# Tracks which providers have already received a CEO notification
# during the current rate-limit episode. Cleared when the probe
# succeeds and the rate limit is lifted (tracker.clear() path).
self._rate_limit_ceo_notified: set[str] = set()
# Strong refs for fire-and-forget audit writes. Without this, the
# event loop only weak-refs the Task and may GC it before it
# commits — audit_log was silently empty because of this.
self._bg_tasks: set[asyncio.Task[None]] = set()
# Wake-up signal for the dispatcher. Set() by API routes immediately
# after status transitions so the dispatcher reacts in milliseconds
# instead of waiting for the next 30-second tick.
self._dispatch_wake: asyncio.Event = asyncio.Event()
self._running = False
# Set True once stop() completes — makes the (lifespan + bootstrap
# safety-net) double-call a clean no-op instead of re-stopping already
# stopped agents / re-draining an empty bg-task set.
self._stopped = False
self._lock = asyncio.Lock()
# Serializes CEO supersede calls so a double-click can't pass the
# find_supersede_umbrella dedup check twice and cut two branches /
# spawn two umbrellas for the same PR (the check is read-then-write
# with no DB-level uniqueness).
self._supersede_lock = asyncio.Lock()
# Serialize concurrent live-chat starts for the single-id interactive
# agents (intake / secretary). Each has a fixed agent id, so two
# concurrent starts race on the container name (``docker run --name
# roboco-agent-<id>``) and the ``_instances[<id>]`` write — orphaning a
# container + relay. The lock makes the second start wait for the first
# to fully register (so the second's reap-prior step sees it) instead of
# both clobbering the registry. Distinct from ``self._lock`` (which
# ``stop_agent`` takes) to avoid a reentrancy deadlock: the spawn body
# holds this lock then calls ``stop_agent`` (acquires ``self._lock``) —
# lock order is always ``_intake_spawn_lock`` -> ``self._lock``, never
# the reverse, so there's no cycle.
self._intake_spawn_lock = asyncio.Lock()
self._secretary_spawn_lock = asyncio.Lock()
# Per-tick set of task_ids already handled by an earlier
# dispatcher. Reset at the start of every _dispatch_all_work.
# Consumed via `self._mark_task_handled` / `_is_task_handled`.
self._tick_handled_tasks: set[str] = set()
# Respawn circuit breaker: per (agent_slug, task_id), tracks how
# many times we've spawned without the task status changing. A PM
# that gets re-spawned on the same pending task with no progress
# is in a loop — without this gate the orchestrator re-spawns every
# tick forever (seen in production on 2026-04-22).
self._pm_respawn_tracker: dict[tuple[str, str], dict[str, Any]] = {}
# Dispatcher heartbeat throttle (see _emit_dispatcher_heartbeat).
self._last_dispatch_heartbeat: datetime | None = None
# Serializes the fire-and-forget respawn-tracker upserts so same-key
# persists COMMIT in schedule (logical) order — not whatever order their
# DB transactions resolve in. A respawn loop fires count 1->2->3->4 in
# quick succession, one fire-and-forget persist per increment; without
# serialization a slow stale persist (count=2) can commit AFTER a fast
# fresh one (count=4), leaving the durable row at the stale low count
# and re-burning the strike threshold on restart. The lock is acquired
# as the FIRST await in _persist_respawn_record, so acquisition order
# matches task creation order (FIFO ready queue), which is the logical
# schedule order. Persists are best-effort background writes, so
# serializing them never blocks the dispatcher hot path (the lock lives
# in the bg task, not the caller).
self._respawn_persist_lock = asyncio.Lock()
# Board agents (Product Owner / Head of Marketing) get exactly ONE
# review pass per assigned task: they have no verb to claim, plan,
# delegate, or complete, so a respawn cannot advance the task and would
# just loop. Tracks (agent_slug, task_id) already dispatched.
self._board_dispatched: set[tuple[str, str]] = set()
# Cross-tick damper for notification-triggered spawns (escalation /
# approval / audit / a2a). Those dispatchers carry no task_id, so the
# readiness gate and the PM respawn breaker never see them — without
# this, an unacknowledged notification respawns its recipient every
# dispatch tick, unbounded. One spawn per (agent, notification) per
# cooldown window; the notification stays pending, so the next window
# retries if it is still unacked. In-memory by design (a restart just
# allows one immediate retry — a tick damper, not durable state).
self._notification_spawn_at: dict[tuple[str, str], float] = {}
# Cluster C5: a board review is a two-reviewer gate — BOTH the Product
# Owner and the Head of Marketing must review a board/coordination task
# before it is handed to the CEO for Approve & Start. Once both have
# finished (dispatched-and-no-longer-active), the orchestrator emits ONE
# formal CEO notification per task. Tracks task_ids already notified so
# the signal fires exactly once.
self._board_review_ceo_notified: set[str] = set()
# Stale-claim reaper config, sourced from
# stale_claim_reap_seconds (default 600) rather than
# claim_stale_seconds (default 180) — the reaper gets the longer
# window of the two claim-staleness thresholds.
# Smoke run 3 showed agents reaped at 180s while actively retrying
# rejected verbs — LLM inference routinely exceeds that window.
# Tests bypass `__init__` via `__new__` and set _claim_heartbeat_ttl
# directly; production never uses _task_svc from __init__.
self._claim_heartbeat_ttl: int = settings.stale_claim_reap_seconds
# Short debounce for closure respawn of a recently-paused parent —
# NOT the reaper window. See _is_recently_paused.
self._closure_recently_paused_ttl: int = (
settings.pm_closure_recently_paused_seconds
)
# Longer threshold before a wedged (ACTIVE-yet-idle) GROK container is
# killed + evicted so the reaper can release its task; see
# _maybe_kill_wedged_grok.
self._grok_idle_kill_ttl: int = settings.grok_idle_kill_seconds
# #73: a non-GROK agent stuck in a non-verb loop (alive, no heartbeat
# advance) is killed past this longer window so the reaper can release
# its task; see _maybe_kill_stuck_claude.
self._claude_stuck_kill_ttl: int = settings.claude_stuck_kill_seconds
# Cost ceiling (USD) before a live GROK container is killed — the budget
# kill-switch parity (the grok CLI exposes no live usage hook). 0 disables.
# See _enforce_grok_cost_budget.
self._grok_max_cost_usd: float = settings.grok_max_cost_usd
# Grok re-park backoff state. Track the re-park count within one episode
# so retry_after can back off exponentially (dampening the ~90s
# crash-retry churn), and the last park time so a gap (the rate limit
# actually lifted) resets the count for the next episode.
self._grok_last_park_at: datetime | None = None
self._grok_repark_count: int = 0
def _record_loop_heartbeat(self, name: str, interval: float) -> None:
self._loop_heartbeats[name] = (time.monotonic(), interval)
def _check_loop_liveness(self) -> None:
now = time.monotonic()
heartbeats = getattr(self, "_loop_heartbeats", {})
for name, (last_success, interval) in heartbeats.items():
stall = now - last_success
if stall > 2 * interval:
logger.warning(
"engine loop stalled past 2x interval",
loop=name,
stall_seconds=int(stall),
interval=interval,
)
# =========================================================================
# LIFECYCLE
# =========================================================================
async def start(self) -> None:
"""Start the orchestrator."""
self._running = True
# Ensure agent image is built
await self._ensure_agent_image()
# Restore any WaitingRecord rows left by a prior orchestrator run so
# agents that were WAITING_LONG at shutdown can still be resolved.
await self.restore_waiting_records()
# Restore the PM-respawn loop counter so a task wedged at the strike
# threshold trips immediately after a restart instead of resetting to
# count=1 and re-burning the whole budget. Validates against live tasks
# (drops terminal/missing rows); inert when the table is empty.
await self.restore_respawn_tracker()
# Self-heal: roll back orphan claims left over from a prior crash.
# Tasks that show CLAIMED/IN_PROGRESS but have NO
# branch_name set indicate _finalize_claim flushed the status before
# branch creation failed (before claim-rollback was atomic). Without
# this, the next claim attempt fails non-idempotent on `git checkout -b`.
await self._reconcile_orphan_claims_on_startup()
# Re-adopt agent containers that survived this orchestrator restart, so
# the spawn gate + reaper see them as live immediately (no double-spawn,
# no over-reap). Inert when nothing is running. Must run before the
# dispatcher/reaper loops launch below.
await self._heal_stale_agent_tokens()
await self._readopt_running_agents()
# Close agent_spawn_sessions rows left open by a prior orchestrator
# crash so usage/cost rollups (which filter ended_at IS NOT NULL) count
# their tokens. Running agents stay open for their live finalize.
await self._reconcile_orphan_spawn_sessions()
# Orphan sandbox sweep: a sandbox whose owning agent container didn't
# survive the restart (or a prior crash mid-teardown) is removed here
# rather than lingering until its next reaper-tick sweep.
await self._sandbox_janitor_sweep()
# Note: Per-agent settings are now generated at spawn time
# via _generate_agent_settings() - no shared settings needed
# A restart mid-execute orphans the release mutex in Redis (TTL 3000s,
# no heartbeat after death); sweep stale keys so a CEO retry doesn't
# hit already_in_progress for up to 50 min. Best-effort, inert if Redis
# is down or empty.
from roboco.services.release_proposal import sweep_orphan_release_locks
await sweep_orphan_release_locks()
# Start background tasks
self._health_task = asyncio.create_task(self._health_loop())
self._dispatcher_task = asyncio.create_task(self._dispatcher_loop())
self._sweeper_task = asyncio.create_task(self._sweeper_loop())
self._rate_limit_probe_task = asyncio.create_task(self._rate_limit_probe_loop())
self._strategy_engine_task = asyncio.create_task(self._strategy_engine_loop())
self._external_pr_poll_task = asyncio.create_task(self._external_pr_poll_loop())
self._self_heal_task = asyncio.create_task(self._self_heal_loop())
self._ci_watch_task = asyncio.create_task(self._ci_watch_loop())
self._dep_update_task = asyncio.create_task(self._dep_update_loop())
self._release_manager_task = asyncio.create_task(self._release_manager_loop())
self._x_mentions_task = asyncio.create_task(self._x_mentions_poll_loop())
self._roadmap_engine_task = asyncio.create_task(self._roadmap_engine_loop())
self._x_feature_spotlight_task = asyncio.create_task(
self._x_feature_spotlight_loop()
)
self._video_render_task = asyncio.create_task(self._video_render_loop())
logger.info(
"Orchestrator started",
dispatcher_interval=self.dispatcher_interval,
internal_api_url=self._api_url,
)
async def _cancel_background_task(self, task: asyncio.Task | None) -> None:
"""Cancel one background loop task and await its teardown (idempotent)."""
if task is None:
return
task.cancel()
with contextlib.suppress(asyncio.CancelledError):
await task
async def _drain_bg_tasks(self) -> None:
"""Let fire-and-forget ``_bg_tasks`` finish before the process exits.
Short DB writes (a respawn_tracker upsert, an audit-log row) get a
bounded window to commit — preserving the durable PM-respawn counter
and the metrics-bearing audit trail — while a stuck task can't hang
shutdown: past ``_SHUTDOWN_DRAIN_TIMEOUT_SECONDS`` the still-pending
tasks are cancelled. ``return_exceptions=True`` so one failing bg task
doesn't crash the drain (a failed write already degraded to in-memory;
logging it here would just be noise). No-op when nothing is pending.
"""
pending = [t for t in self._bg_tasks if not t.done()]
if not pending:
return
try:
await asyncio.wait_for(
asyncio.gather(*pending, return_exceptions=True),
timeout=_SHUTDOWN_DRAIN_TIMEOUT_SECONDS,
)
except TimeoutError:
for task in pending:
if not task.done():
task.cancel()
with contextlib.suppress(asyncio.CancelledError):
await asyncio.gather(*pending, return_exceptions=True)
async def _flush_respawn_tracker(self) -> None:
"""Persist the full in-memory PM-respawn snapshot before the process exits.
Fire-and-forget persists (``_schedule_respawn_persist``) are bounded by
the shutdown drain deadline; one cancelled by that deadline leaves the
durable count lagging the in-memory counter, so the next restart
re-burns the strike threshold against a still-wedged task — the exact
re-burn the durable counter exists to stop (#74). Called from ``stop()``
AFTER the bounded drain so it is the last writer (no further gate
mutations fire once the agents and loops are down) and unbounded (a
short upsert must not be dropped on the shutdown path). Best-effort: a
row that fails to persist is logged and skipped, never crashing shutdown
— the in-memory value is gone either way once the process exits.
"""
if not self._pm_respawn_tracker:
return
for agent_slug, task_id in list(self._pm_respawn_tracker.keys()):
record = self._pm_respawn_tracker.get((agent_slug, task_id))
if record is None:
continue
try:
await self._persist_respawn_record(agent_slug, task_id, dict(record))
except Exception:
logger.exception(
"shutdown respawn-tracker flush failed for one row; continuing",
agent_id=agent_slug,
task_id=task_id,
)
async def stop(self) -> None:
"""Stop the orchestrator and all agents."""
if getattr(self, "_stopped", False):
# Idempotent: the lifespan shutdown path stops the orchestrator
# before closing the DB, and bootstrap's finally block re-calls
# stop() as a safety net. The second call must be a no-op, not a
# re-stop of already-stopped agents. ``getattr`` so a ``__new__``-
# constructed instance (unit-test pattern) without ``__init__`` is
# still stoppable.
return
self._running = False
# Cancel every background loop, then stop the agents.
for task in (
self._health_task,
self._dispatcher_task,
self._sweeper_task,
self._rate_limit_probe_task,
self._strategy_engine_task,
self._external_pr_poll_task,
self._self_heal_task,
self._ci_watch_task,
self._dep_update_task,
self._release_manager_task,
self._x_mentions_task,
self._roadmap_engine_task,
self._x_feature_spotlight_task,
self._video_render_task,
):
await self._cancel_background_task(task)
# Stop all agents. One agent's stop error must not skip the drain
# below — that would re-introduce the data-loss tail for every in-flight
# bg write, so log-and-continue rather than propagate.
for agent_id in list(self._instances.keys()):
try:
# release_claim=True: on shutdown the orchestrator is going
# down and no agent will resume its task, so hand claimed
# tasks back to the pool now — they re-dispatch immediately on
# the next start instead of waiting for the reaper's TTL. A
# provider-parked agent is skipped inside stop_agent so its
# claim survives for the probe-resume loop across the restart.
await self.stop_agent(
agent_id, release_claim=True, stop_reason="orchestrator_shutdown"
)
except Exception:
logger.exception(
"stop_agent raised during shutdown; continuing to drain",
agent_id=agent_id,
)
# Drain fire-and-forget bg writes so short DB commits finish before the
# process exits (respawn_tracker upserts, audit-log rows). Bounded so a
# stuck task can't hang shutdown — it is cancelled past the deadline.
await self._drain_bg_tasks()
# #74: flush the authoritative in-memory respawn snapshot AFTER the
# bounded drain so a deadline-cancelled persist can't leave the durable
# count lagging the in-memory counter (and re-burning the strike
# threshold on the next restart). Unbounded — a short upsert must not be
# dropped on the shutdown path.
await self._flush_respawn_tracker()
self._stopped = True
logger.info("Orchestrator stopped")
async def _ensure_agent_image(self, agent_id: str | None = None) -> None:
"""Ensure the agent Docker images are present.
Local mode (no ``agent_image_registry``) builds the base image first,
then the role-specialized image, from ``docker/agent-*.Dockerfile``.
Registry mode pulls the pre-built images instead. Idempotent — skips
anything already present locally.
"""
# Determine build context
if PROJECT_HOST_PATH:
build_context = PROJECT_HOST_PATH
docker_dir = f"{PROJECT_HOST_PATH}/docker"
else:
build_context = str(self.project_root)
docker_dir = str(self.project_root / "docker")
# Always ensure base image exists
await self._ensure_image_present(
AGENT_BASE_IMAGE,
f"{docker_dir}/agent-base.Dockerfile",
build_context,
)
# Ensure the role-specialized image if this agent uses one
if agent_id:
bare = AGENT_IMAGES.get(agent_id, AGENT_BASE_IMAGE)
if bare != AGENT_BASE_IMAGE:
# Map the bare image name to its dockerfile
dockerfile_map = {
"roboco-agent-pm": "agent-pm.Dockerfile",
"roboco-agent-dev-be": "agent-dev-be.Dockerfile",
"roboco-agent-dev-fe": "agent-dev-fe.Dockerfile",
"roboco-agent-qa-be": "agent-qa-be.Dockerfile",
"roboco-agent-qa-fe": "agent-qa-fe.Dockerfile",
"roboco-agent-doc": "agent-doc.Dockerfile",
"roboco-agent-ux": "agent-ux.Dockerfile",
"roboco-agent-prompter": "agent-prompter.Dockerfile",
"roboco-agent-secretary": "agent-secretary.Dockerfile",
"roboco-agent-pr-reviewer": "agent-pr-reviewer.Dockerfile",
}
dockerfile = dockerfile_map.get(bare)
if dockerfile:
await self._ensure_image_present(
bare,
f"{docker_dir}/{dockerfile}",
build_context,
)
async def _ensure_grok_interactive_image(self, image: str) -> None:
"""Ensure a Grok interactive image and its base→runtime chain exist.
The grok-prompter / grok-secretary images build FROM roboco-agent-grok,
which builds FROM the agent base, so the whole chain must be present
before a local build of the interactive image can succeed (on the
registry path each is already pulled and this just verifies presence).
"""
if PROJECT_HOST_PATH:
build_context = PROJECT_HOST_PATH
docker_dir = f"{PROJECT_HOST_PATH}/docker"
else:
build_context = str(self.project_root)
docker_dir = str(self.project_root / "docker")
chain = [
(AGENT_BASE_IMAGE, "agent-base.Dockerfile"),
("roboco-agent-grok", "agent-grok.Dockerfile"),
(image, _GROK_INTERACTIVE_DOCKERFILES[image]),
]
for img, dockerfile in chain:
await self._ensure_image_present(
img, f"{docker_dir}/{dockerfile}", build_context
)
@staticmethod
def _safe_agent_path_segment(agent_id: str) -> str:
"""Return ``agent_id`` if it is safe as a single path segment, else raise.
``agent_id`` reaches the grok usage dir from request-facing call sites, so
it must not be able to traverse the path. Reject every traversal vector —
empty, ``.`` / ``..``, a ``/`` or ``\\`` separator, or an embedded NUL —
rather than stripping it; the orchestrator only ever assigns plain
slug / uuid ids, none of which contain these.
"""
if (
not agent_id
or agent_id in {".", ".."}
or "/" in agent_id
or "\\" in agent_id
or "\x00" in agent_id
):
raise ValueError(f"unsafe agent id for a filesystem path: {agent_id!r}")
return agent_id
@staticmethod
def _grok_usage_root() -> Path:
"""The base dir all per-agent grok usage dirs live under (no agent id).
Branched compose-vs-local: in compose the orchestrator sees the mounted
host dir at ``GROK_USAGE_DATA_DIR``; in local mode usage.json lands under
the shared tempdir. The single fixed anchor the per-agent dir hangs off,
and the safe root a finalize read is checked to stay within.
"""
if PROJECT_HOST_PATH:
return Path(GROK_USAGE_DATA_DIR)
return Path(tempfile.gettempdir()) / "roboco-grok-usage"
@staticmethod
def _grok_usage_dir(agent_id: str) -> Path:
"""Per-agent grok usage dir under :meth:`_grok_usage_root`.
Single source of truth for BOTH the pre-create/mount side
(``_ensure_grok_usage_dir``) and the finalize read side
(``_grok_usage_json``) so they can never drift. ``agent_id`` is validated
as a single safe path segment first — ``_safe_agent_path_segment`` rejects
``.`` / ``..`` / separators / NUL so a bad id raises rather than silently
remapping or traversing. The read side additionally reduces the id to its
final path component (``os.path.basename``) — the CodeQL-recognized
path-injection barrier.
"""
return AgentOrchestrator._grok_usage_root() / (
AgentOrchestrator._safe_agent_path_segment(agent_id)
)
def _ensure_grok_usage_dir(self, agent_id: str) -> None:
"""Pre-create the agent's grok usage dir (world-writable) before the mount.
On Linux, ``docker run -v`` auto-creates a MISSING bind source as
``root:root``, so the non-root ``agent`` user EACCESes when the grok
entrypoint / interactive driver writes ``usage.json`` there. Creating the
dir ``0777`` first makes the mounted dir writable regardless of the agent
uid; the orchestrator (root) can still read it back at finalize.
"""
target = self._grok_usage_dir(agent_id)
try:
target.mkdir(parents=True, exist_ok=True)
target.chmod(0o777)
except OSError as exc:
logger.warning(
"could not pre-create grok usage dir; grok agent may EACCES",
agent_id=agent_id,
path=str(target),
error=str(exc),
)
async def _ensure_image_present(
self, bare_image: str, dockerfile_path: str, build_context: str
) -> None:
"""Ensure one agent image is present locally.
Pulls it (registry mode) or builds it from its Dockerfile (local mode)
when missing; no-op if already present.
"""
image = _qualify_agent_image(bare_image)
# Check if image exists
proc = await asyncio.create_subprocess_exec(
"docker",
"image",
"inspect",
image,
stdout=asyncio.subprocess.DEVNULL,
stderr=asyncio.subprocess.DEVNULL,
)
await proc.wait()
if proc.returncode == 0:
return
if settings.agent_image_registry:
# Registry mode: pull the pre-built image; never build from source
# (a deployment running pre-built images has no build context).
logger.info("Pulling agent image...", image=image)
proc = await asyncio.create_subprocess_exec(
"docker",
"pull",
image,
stdout=asyncio.subprocess.PIPE,
stderr=asyncio.subprocess.PIPE,
)
_, stderr = await proc.communicate()
if proc.returncode != 0:
raise RuntimeError(f"Failed to pull image {image}: {stderr.decode()}")
logger.info("Agent image pulled", image=image)
return
logger.info("Building Docker image...", image=image)
proc = await asyncio.create_subprocess_exec(
"docker",
"build",
"-t",
image,
"-f",
dockerfile_path,
build_context,
stdout=asyncio.subprocess.PIPE,
stderr=asyncio.subprocess.PIPE,
)
_, stderr = await proc.communicate()
if proc.returncode != 0:
raise RuntimeError(f"Failed to build image {image}: {stderr.decode()}")
logger.info("Docker image built successfully", image=image)
# =========================================================================
# PER-AGENT SETTINGS GENERATION
# =========================================================================
def _get_role_permissions(
self, role: str, workspace_path: str, cell_workspace_path: str
) -> dict[str, list[str]]:
"""Get role-specific allow/deny lists for Claude Code tools.
Post-gateway shape: every state-changing operation an agent can
perform routes through ``mcp__roboco-flow__*`` (intent verbs) or
``mcp__roboco-do__*`` (content tools — commit, push, PR, journal,
notify, message), both granted to every role via ``base_allow``.
Role-specific configuration here only governs file IO (Write/Edit
scoping) plus a small handful of legacy native-tool denies that
remain meaningful for weak models. Read-only git lives in
``mcp__roboco-git-readonly__*``.
Args:
role: Agent role (developer, qa, documenter, cell_pm, main_pm, etc.)
workspace_path: Path to agent's own workspace directory
cell_workspace_path: Path to cell's workspace root (for QA/Docs access)
Returns:
Dict with 'allow' and 'deny' lists for Claude Code permissions
"""
# workspace_path: /data/workspaces/{project}/{team}/{agent}
# cell_workspace_path: /data/workspaces/{project}/{team}
configs: dict[str, dict[str, list[str]]] = {
"developer": {
"allow": [
f"Write(/{workspace_path}/**)",
f"Edit(/{workspace_path}/**)",
],
"deny": [],
},
"qa": {
# QA reads code + the open PR via the gateway; never edits.
"allow": [],
"deny": [
"Write(*)",
"Edit(*)",
],
},
"documenter": {
"allow": [
f"Write(/{cell_workspace_path}/**)",
f"Edit(/{cell_workspace_path}/**)",
"Write(//app/docs/**)",
"Edit(//app/docs/**)",
"Write(//app/CHANGELOG.md)",
"Edit(//app/CHANGELOG.md)",
"Write(//app/README.md)",
"Edit(//app/README.md)",
],
"deny": [],
},
"cell_pm": {
# PMs coordinate; they open + merge PRs through the gateway
# but never author code. Edit/Write are denied so weaker
# models can't read the subtask title imperatively and
# start editing source — they have to decompose into a dev
# subtask. Devs are the only role that authors code.
"allow": [],
"deny": [
"Bash(git commit:*)",
"Bash(git push:*)",
"Write(*)",
"Edit(*)",
],
},
"main_pm": {
# Same reasoning as cell_pm — Main PM sits between CEO and
# cell PMs; the work product is coordination + review, not
# commits or edits. Code work routes Main PM → Cell PM →
# Dev only.
"allow": [],
"deny": [
"Bash(git commit:*)",
"Bash(git push:*)",
"Write(*)",
"Edit(*)",
],
},
"product_owner": {
"allow": [
f"Write(/{workspace_path}/**)",
f"Edit(/{workspace_path}/**)",
],
"deny": [],
},
"head_marketing": {
"allow": [
f"Write(/{workspace_path}/**)",
f"Edit(/{workspace_path}/**)",
],
"deny": [],
},
"auditor": {
# Auditor is read-only across the org — observes, never edits.
"allow": [],
"deny": [
"Write(*)",
"Edit(*)",
],
},
"pr_reviewer": {
# PR reviewer reads untrusted external/fork PR diffs and posts a
# change-request via the gateway — it never writes files. Make the
# read-only invariant explicit at the permission layer (it is the
# highest-value prompt-injection target), not just implicit in the
# absence of a writable mount.
"allow": [],
"deny": [
"Write(*)",
"Edit(*)",
],
},
}
if role not in configs:
logger.warning(
"No Claude Code permissions configured for role; "
"agent will be limited to base_allow/base_deny.",
role=role,
)
return configs.get(role, {"allow": [], "deny": []})
def _fable_hook_groups(self) -> dict[str, list[dict[str, Any]]]:
"""Additive Fable-mode hook registrations, keyed by Claude Code event.
Empty when the flag is off, so callers that append these onto the
existing per-event arrays leave settings.json byte-for-byte
unchanged. Appended AFTER RoboCo's own hooks for each event —
stop-hook.sh's mechanical terminal-verb check runs first, the
Fable linguistic check runs second. See
docs/superpowers/plans/2026-07-04-v0.18.0-A-opus-fable-plan.md.
"""
from roboco.config import settings as _settings
if not _settings.fable_mode_enabled:
return {}
return {
"Stop": [
{
"hooks": [
{
"type": "command",
"command": "/app/scripts/fable-stop-gate-hook.sh",
}
]
},
],
"SubagentStop": [
{
"hooks": [
{
"type": "command",
"command": "/app/scripts/fable-stop-gate-hook.sh subagent",
}
]
},
],
"PreToolUse": [
{
"matcher": "Bash",
"hooks": [
{
"type": "command",
"command": "/app/scripts/fable-bash-discipline-hook.sh",
}
],
},
],
"PostToolUse": [
{
"matcher": "Bash",
"hooks": [
{
"type": "command",
"command": "/app/scripts/fable-honesty-nudge-hook.sh",
}
],
},
],
"UserPromptSubmit": [
{
"hooks": [
{
"type": "command",
"command": "/app/scripts/fable-prompt-nudge-hook.sh",
}
]
},
],
"PreCompact": [
{
"matcher": "manual|auto",
"hooks": [
{
"type": "command",
"command": "/app/scripts/fable-precompact-hook.sh",
}
],
},
],
}
def _generate_agent_settings(
self,
agent_id: str,
role: str,
workspace_path: str,
cell_workspace_path: str,
) -> Path:
"""Generate per-agent Claude Code settings file with role-specific permissions.
This replaces the shared settings approach. Each agent gets their own
settings.json with:
- Base MCP tools allowed for all agents
- Role-specific tool permissions
- Explicit deny list blocking native git/file operations
Args:
agent_id: Agent identifier (e.g., "be-dev-1")
role: Agent role (e.g., "developer")
workspace_path: Path to agent's own workspace directory
cell_workspace_path: Path to cell's workspace root (for QA/Docs)
Returns:
Path to the generated settings file
"""
# Base MCP tools for all agents. Post-gateway every role gets the
# full intent-verb + content-tool surface; the orchestrator-side
# API rejects verbs/tools the agent's role isn't authorized for,
# so granting `*` here is safe.
base_allow = [
"mcp__roboco-flow__*",
"mcp__roboco-do__*",
"mcp__roboco-optimal__*",
"mcp__roboco-git-readonly__*",
"Read(*)", # All agents can read any file
]
# Base denials for all agents - block native tools + sensitive reads.
# The Read/Bash denies below are critical: without them an agent can
# read `.git/config` (which, pre-fix, had the PAT embedded in the
# remote URL) or `~/.gitconfig` and exfiltrate project secrets.
# We also block direct curl/wget to github.com — any git-remote op
# must go through the orchestrator's git service, which injects the
# token via bearer header at subprocess time rather than exposing it.
base_deny = [
# Block ALL native git commands - must use roboco_git_* tools
"Bash(git:*)",
# NOTE: Write/Edit are intentionally NOT globally denied here.
# Claude Code evaluates rules deny -> ask -> allow and the first
# match wins, so a deny ALWAYS beats a more-specific allow (the
# glob syntax has no negation). A global Write(*)/Edit(*) here
# therefore unconditionally shadowed the per-role,
# workspace-scoped Write/Edit allows below — every agent (devs
# included) was unable to edit ANY file and fell back to
# destructive bash redirection (clobbering real files). Roles
# that must NOT write (qa, cell_pm, main_pm, auditor) carry
# their own Write(*)/Edit(*) deny in _get_role_permissions.
# Block reads of credential stores, anywhere on the FS
"Read(**/.git/config)",
"Read(**/.gitconfig)",
"Read(/etc/gitconfig)",
"Read(~/.netrc)",
"Read(**/.git-credentials)",
# The host's Claude Code OAuth credential store (`~/.claude`) is
# bind-mounted read-write into EVERY agent container at
# /home/agent/.claude (see _build_mount_args) — it is the shared
# subscription auth every spawned agent uses, so it can't be
# narrowed per-agent. Nothing in any role's job requires the LLM
# to read its own harness's credentials, so block the Read tool
# from the two files that carry them (`.credentials.json` on
# Linux hosts without a keychain; `.claude.json` carries the
# linked account + MCP trust state). Absolute `//` form per the
# #167 gotcha above — a single `/` resolves against the
# settings.json project root, not the container filesystem root.
"Read(//home/agent/.claude/.credentials.json)",
"Read(//home/agent/.claude.json)",
# Block direct GitHub API/wire access — agents must use
# roboco_git_* MCP tools so secrets + traceability stay on the
# orchestrator side.
"Bash(curl:*github.com*)",
"Bash(curl:*api.github.com*)",
"Bash(wget:*github.com*)",
"Bash(wget:*api.github.com*)",
# Same idea for cat-ing credential files in a subshell
"Bash(cat:*.git/config*)",
"Bash(cat:*.gitconfig*)",
"Bash(cat:*.git-credentials*)",
"Bash(cat:*.credentials.json*)",
"Bash(cat:*.claude.json*)",
# Block reading env vars that might leak secrets
"Bash(env:*)",
"Bash(printenv:*)",
]
# Get role-specific permissions
role_config = self._get_role_permissions(
role, workspace_path, cell_workspace_path
)
# Combine base + role-specific.
# defaultMode=bypassPermissions lets unlisted operations proceed
# without an interactive prompt (which would hang a non-TTY agent
# container). Explicit deny rules still apply.
settings: dict[str, Any] = {
"permissions": {
"defaultMode": "bypassPermissions",
"allow": base_allow + role_config["allow"],
"deny": base_deny + role_config["deny"],
},
# Explicit Bash-output cap: a gate/test dump enters the session
# context once and is re-read at cache-read price on every later
# turn. 20K chars (~5K tokens) keeps failures diagnosable without
# relying on the CLI's default ceiling.
"env": {
"BASH_MAX_OUTPUT_LENGTH": "20000",
},
"hooks": {
# Start SDK server on session start (for A2A communication)
"SessionStart": [
{
"hooks": [
{
"type": "command",
"command": "/app/scripts/sdk-startup-hook.sh",
}
]
}
],
# Guard Bash: block shell-level git/curl/wget/env patterns
# that the matcher-based `permissions.deny` can't catch
# (e.g. `cd X && git fetch`). Redirects agents to the MCP
# equivalents instead of bloating prompts with rules.
"PreToolUse": [
{
"matcher": "Bash",
"hooks": [
{
"type": "command",
"command": "/app/scripts/bash-guard-hook.sh",
}
],
},
],
"PostToolUse": [
# Check for incoming A2A messages after each tool use
{
"matcher": "*",
"hooks": [
{
"type": "command",
"command": "/app/scripts/a2a-check-hook.sh",
}
],
},
# Per-session budget counter + loop detector. Shared SDK
# state lets this hook emit [Budget]/[Loop]/[Halt]
# reminders that the orchestrator's kill-switch corroborates.
{
"matcher": "*",
"hooks": [
{
"type": "command",
"command": "/app/scripts/post-tool-budget-hook.sh",
}
],
},
# Sync token usage from the transcript so /usage/status
# (and the cost dashboard) reflect real spend. Idempotent
# absolute set — running it per tool keeps mid-run
# snapshots and reaped-agent sessions accurate.
{
"matcher": "*",
"hooks": [
{
"type": "command",
"command": "/app/scripts/usage-report-hook.sh",
}
],
},
],
# Stop guard: refuse silent exits unless a terminal tool was
# just called (idle/substitute/escalate/pause/...). Second
# attempt auto-substitutes via SDK so the task doesn't rot.
"Stop": [
{
"hooks": [
{
"type": "command",
"command": "/app/scripts/stop-hook.sh",
},
# Final token-usage sync at turn end — guarantees
# the session total is captured before the agent
# idles and the orchestrator finalizes the row.
{
"type": "command",
"command": "/app/scripts/usage-report-hook.sh",
},
]
}
],
# Prompt-injection guard — rejects turns that look like
# another agent's content trying to override our rules.
"UserPromptSubmit": [
{
"hooks": [
{
"type": "command",
"command": "/app/scripts/user-prompt-hook.sh",
}
]
}
],
# Snapshot budget / terminal state before compact so the
# next session resumes with continuity.
"PreCompact": [
{
"hooks": [
{
"type": "command",
"command": "/app/scripts/pre-compact-hook.sh",
}
]
}
],
# Post-mortem: write a reflect-journal entry summarising the
# session (tools called, halt/loop triggered, last tool).
"SessionEnd": [
{
"hooks": [
{
"type": "command",
"command": "/app/scripts/session-end-hook.sh",
}
]
}
],
},
}
for event, groups in self._fable_hook_groups().items():
settings["hooks"].setdefault(event, []).extend(groups)
# Write to per-agent settings file
# When running in container: write to /app/agent-settings (mounted to host)
# When running on host: use temp directory
if DATA_HOST_PATH:
settings_dir = Path("/app/agent-settings")
else:
settings_dir = Path(tempfile.gettempdir()) / "roboco-agent-settings"
settings_dir.mkdir(parents=True, exist_ok=True)
settings_path = settings_dir / f"{agent_id}-settings.json"
# Handle case where Docker auto-created a directory instead of a file
if settings_path.is_dir():
shutil.rmtree(settings_path)
settings_path.write_text(json.dumps(settings, indent=2))
logger.debug(
"Generated per-agent settings",
agent_id=agent_id,
role=role,
settings_path=str(settings_path),
allow_count=len(settings["permissions"]["allow"]),
deny_count=len(settings["permissions"]["deny"]),
)
return settings_path
# =========================================================================
# AGENT SPAWNING
# =========================================================================
def _task_git_context(self, task: dict[str, Any]) -> SpawnGitContext | None:
"""Build SpawnGitContext from a task dict for workspace mounting.
Without this, spawned agents fall back to project_slug="default"
and get a Write/Edit permission lock to /data/workspaces/default/...
which does not exist, so the agent's file tools fail.
"""
project_slug = task.get("project_slug")
if not project_slug:
return None
branch_name = task.get("branch_name")
ctx = SpawnGitContext(project_slug=project_slug, branch_name=branch_name)
# A branch-bearing task edits in a per-task worktree keyed by the short
# id; a branchless coordination root (umbrella / no-project product
# root) has no worktree, so task_short_id stays None and the spawn cwd
# falls back to the clone root.
if branch_name and task.get("id"):
ctx.task_short_id = str(task["id"])[:8]
return ctx
def _fire_audit(
self,
*,
event_type: str,
agent_slug: str,
task_id: str | None = None,
details: dict[str, Any] | None = None,
severity: str = "info",
) -> None:
"""Emit an agent-lifecycle audit event without blocking the caller.
Strong-refs the Task so it isn't garbage-collected before it
commits to `audit_log`. Silently skips if there's no running loop
(e.g. sync unit tests).
"""
import contextlib as _ctx
from roboco.services.audit import get_audit_service
with _ctx.suppress(RuntimeError):
bg = asyncio.get_running_loop().create_task(
get_audit_service().log_agent_event(
event_type=event_type,
agent_slug=agent_slug,
task_id=task_id,
details=details or {},
severity=severity,
)
)
self._bg_tasks.add(bg)
bg.add_done_callback(self._bg_tasks.discard)
async def _git_context_default_project(self) -> SpawnGitContext | None:
"""Return git context for the 'default' project when no task is known.
Used by no-task spawns (idle PM, scanner-only agents). Picks the
first active project in the DB — the common case is a single-project
deployment, where this resolves to the correct slug; for multi-
project deployments the caller should pass task_id to disambiguate.
"""
from sqlalchemy import select
from roboco.db.base import get_db_context
from roboco.db.tables import ProjectTable
try:
async with get_db_context() as db:
result = await db.execute(
select(ProjectTable.slug, ProjectTable.default_branch)
.where(ProjectTable.is_active.is_(True))
.order_by(ProjectTable.created_at.asc())
.limit(1)
)
row = result.first()
if row is None:
return None
slug, default_branch = row
if not slug:
return None
return SpawnGitContext(
project_slug=slug,
branch_name=default_branch,
)
except Exception as e:
logger.warning(
"Could not derive default project git context",
error=str(e),
)
return None
async def _git_context_from_task_id(self, task_id: str) -> SpawnGitContext | None:
"""Load a task by ID and derive git context for spawning.
Used by `spawn_agent` when called without an explicit git_context
(e.g. the /agents/{slug}/spawn API endpoint). Without this, agents
spawned via that endpoint get project_slug="default" and their
workspace mount points at a path that doesn't exist.
"""
from sqlalchemy import select
from roboco.db.base import get_db_context
from roboco.db.tables import ProjectTable, TaskTable
try:
async with get_db_context() as db:
result = await db.execute(
select(TaskTable.branch_name, ProjectTable.slug)
.select_from(TaskTable)
.join(ProjectTable, TaskTable.project_id == ProjectTable.id)
.where(TaskTable.id == task_id)
)
row = result.first()
if row is None:
return None
branch_name, project_slug = row
if not project_slug:
return None
ctx = SpawnGitContext(
project_slug=project_slug, branch_name=branch_name
)
if branch_name and task_id:
ctx.task_short_id = str(task_id)[:8]
return ctx
except Exception as e:
logger.warning(
"Could not derive git context from task_id",
task_id=task_id,
error=str(e),
)
return None
async def _resolve_spawn_git_context(
self,
git_context: SpawnGitContext | None,
task_id: str | None,
) -> SpawnGitContext | None:
"""Auto-derive git context if the caller didn't supply one."""
if git_context is not None and git_context.project_slug:
return git_context
derived: SpawnGitContext | None = None
if task_id:
derived = await self._git_context_from_task_id(task_id)
if derived is None:
derived = await self._git_context_default_project()
return derived if derived is not None else git_context
def _existing_running_instance(self, agent_id: str) -> AgentInstance | None:
"""Return the running instance for agent_id, or None if it can be respawned."""
existing = self._instances.get(agent_id)
if existing is None:
return None
if existing.state in (AgentState.OFFLINE, AgentState.WAITING_LONG):
return None
logger.warning(
"Agent already running",
agent_id=agent_id,
state=existing.state,
)
return existing
def _resolve_project_slug(
self,
git_context: SpawnGitContext | None,
agent_id: str,
task_id: str | None,
) -> str:
"""Pull project_slug from context, or fall back to 'default' with a warning."""
project_slug = (
git_context.project_slug
if git_context and git_context.project_slug
else None
)
if not project_slug:
logger.warning(
"Spawning agent without project_slug; workspace fallback used. "
"Agent file tools will be locked to a nonexistent path.",
agent_id=agent_id,
task_id=task_id,
)
project_slug = "default"
return project_slug
async def _prepare_agent_spawn(
self,
agent_id: str,
task_id: str | None,
model: str | None,
git_context: SpawnGitContext | None,
) -> tuple[AgentConfig, AgentInstance, Path | None]:
"""Build AgentConfig + AgentInstance and surface per-agent settings path."""
project_slug = self._resolve_project_slug(git_context, agent_id, task_id)
ambient = await self._resolve_conventions_ambient(project_slug, task_id)
blueprint_path = self._generate_composed_prompt(agent_id, ambient=ambient)
canonical_role = get_agent_role(agent_id)
team = get_agent_team(agent_id) or "backend"
# Resolve the provider route for this agent. Caller-supplied `model`
# wins (dispatcher overrides, tests). Otherwise the routing service
# resolves (agent_slug | role | global) assignments, falling back
# internally to `ROLE_MODEL_MAP` when no rows exist — so a fresh
# deployment with an empty `model_assignments` table behaves exactly
# as before.
route = await self._resolve_agent_route(agent_id)
if not model:
model = route.model_name
cell_workspace_path = _cell_workspace_path(project_slug, team)
# The agent's edit scope + container cwd: the per-task worktree when
# the task carries a branch (F123), else the clone root. Routed through
# _agent_cwd_path so the Edit/Write allowlist (_generate_agent_settings
# -> _get_role_permissions) and the docker -w (_append_workspace_cwd)
# resolve the SAME path.
cwd_path = _agent_cwd_path(project_slug, team, agent_id, git_context)
# Re-attach the task's worktree before the container launches with -w
# pointing at it (F123). A pruned/evicted worktree would start the
# agent in a missing dir; idempotent re-add, no-op for branchless spawns.
await self._ensure_worktree_before_spawn(
git_context, project_slug, team, agent_id, task_id
)
# Provision this spawn's sandbox DB/Redis (flag + per-project opt-in),
# before `docker run` so its connection info can be injected as env.
# Fail-loud on a provisioning failure (see _maybe_provision_sandbox).
sandbox_info = await self._maybe_provision_sandbox(
agent_id, project_slug, task_id
)
agent_settings_path = self._generate_agent_settings(
agent_id, canonical_role, cwd_path, cell_workspace_path
)
briefing_path = await self._write_agent_briefing(agent_id, task_id, cwd_path)
await self._ensure_agent_image(agent_id)
mcp_config_path = await self._generate_mcp_config(agent_id, git_context)
from uuid import uuid4
config = AgentConfig(
agent_id=agent_id,
blueprint_path=blueprint_path,
model=model,
mcp_config_path=mcp_config_path,
claude_session_id=str(uuid4()),
git_context=git_context,
briefing_path=briefing_path,
provider_type=route.provider_type.value,
provider_base_url=route.base_url,
provider_auth_token=route.auth_token,
sandbox_info=sandbox_info,
)
instance = AgentInstance(
agent_id=agent_id,
state=AgentState.STARTING,
config=config,
current_task_id=task_id,
)
self._instances[agent_id] = instance
return config, instance, agent_settings_path
async def _ensure_worktree_before_spawn(
self,
git_context: SpawnGitContext | None,
project_slug: str,
team: str,
agent_id: str,
task_id: str | None,
) -> None:
"""Re-attach the task's per-task worktree before the container starts.
The container launches with ``-w`` at the worktree; a pruned/evicted
worktree (reaper, disk pressure, manual cleanup while the agent was
down) — or a vanished clone root (disk loss, a redeploy that wiped
``/data/workspaces``) — would start the agent in a missing directory.
Idempotent: a present worktree is a no-op; a pruned worktree is re-added
from the surviving branch ref; a missing clone is re-cloned and the
branch ref recovered from origin (``create_branch`` pushes at claim
time) so the pushed work survives. No-op for branchless / no-task spawns.
The reaper-style claim release preserves ownership + ``branch_name``, so
a re-dispatch is a RESUME, not a fresh claim — ``create_branch`` never
re-runs to re-clone. Without the clone self-heal a vanished clone_root
fatal-looped every tick (``git -C <missing>`` -> release -> re-dispatch
into the same missing clone). A fatal git-state failure
(``WorkspaceError`` — the clone won't re-clone, the token is missing,
or the branch ref is unrecoverable) releases the claim and aborts so
the next dispatch retries the rebuild, never launching the container at
a missing ``-w``. A transient failure (DB/other) aborts without
releasing — the next tick retries the same claim.
"""
if not (git_context and git_context.task_short_id and git_context.branch_name):
return
clone_root = Path(_agent_workspace_path(project_slug, team, agent_id))
worktree = Path(
_agent_worktree_path(
project_slug, team, agent_id, git_context.task_short_id
)
)
from roboco.db.base import get_db_context
from roboco.services.workspace import WorkspaceError, WorkspaceService
try:
async with get_db_context() as db:
ws = WorkspaceService(db)
# Heal a vanished/unhealthy clone first. The reaper-style claim
# release preserves ownership + branch_name, so a re-dispatch is
# a RESUME, not a fresh claim — create_branch never re-runs to
# re-clone, and ensure_worktree_for_resume would ``git -C`` a
# missing directory and fatal-loop every tick. Skipped on a
# healthy clone (no new fetch overhead on the common resume).
if not WorkspaceService._is_workspace_healthy(clone_root):
await ws.ensure_workspace(project_slug, agent_id)
await ws.ensure_worktree_self_heal(
clone_root, worktree, git_context.branch_name, project_slug
)
except WorkspaceError as e:
# Fatal git state (clone won't re-clone, token missing, branch ref
# unrecoverable): release the claim so the next dispatch can retry
# the rebuild, and abort before docker run -w lands on a missing
# path. The release is best-effort (suppressed) so a release
# failure never masks the fatal error.
logger.error(
"worktree ensure failed (fatal); releasing claim for rebuild",
agent_id=agent_id,
task_short_id=git_context.task_short_id,
error=str(e),
)
if task_id:
with contextlib.suppress(Exception):
await self._release_claim_to_pending(task_id)
raise AgentReadinessError(
f"worktree ensure failed for {agent_id}"
f" (task={task_id}, branch={git_context.branch_name}): {e};"
f" claim released for rebuild"
) from e
except Exception as e:
# Transient (DB hiccup, etc.): abort so we don't launch at a
# possibly-missing path, but do NOT release — a fresh claim would
# not help and re-cloning is destructive. Next tick retries.
logger.warning(
"worktree ensure failed (transient); aborting spawn",
agent_id=agent_id,
task_short_id=git_context.task_short_id,
error=str(e),
)
raise AgentReadinessError(
f"worktree ensure failed (transient) for {agent_id}"
f" (task={task_id}): {e}; will retry next tick"
) from e
async def _maybe_provision_sandbox(
self, agent_id: str, project_slug: str, task_id: str | None
) -> SandboxInfo | None:
"""Provision this spawn's sandbox DB/Redis, or None if not opted in.
Off (flag or project) => None, byte-for-byte identical to today (the
legacy `_append_gate_env` prod-creds injection stays active). The
project lookup itself is best-effort (a DB hiccup here degrades to
"no sandbox" rather than blocking every spawn on a transient error —
`_ensure_worktree_before_spawn` already fails loud on a genuine DB
outage). Once a project has opted in, an actual provisioning failure
(container won't start / never becomes ready) IS fail-loud: an agent
whose gate can't run must never spawn.
"""
if not settings.sandbox_db_enabled:
return None
from roboco.db.base import get_db_context
from roboco.services.project import get_project_service
try:
async with get_db_context() as db:
project = await get_project_service(db).get_by_slug(project_slug)
except Exception as e:
logger.warning(
"sandbox project lookup failed; skipping sandbox provisioning",
agent_id=agent_id,
project_slug=project_slug,
error=str(e),
)
return None
services = list(project.sandbox_services or []) if project else []
if not services:
return None
try:
return await self._sandbox.provision(agent_id, services)
except Exception as e:
# str(TimeoutError()) == "" — include the type so a bare timeout
# (a cold image pull exceeding the run deadline) self-diagnoses.
err = f"{type(e).__name__}: {e}"
logger.error(
"sandbox provisioning failed; refusing spawn",
agent_id=agent_id,
task_id=task_id,
services=services,
error=err,
)
raise AgentReadinessError(
f"sandbox provisioning failed for {agent_id} (task={task_id}): {err}"
) from e
async def _launch_spawn(
self,
task_id: str | None,
config: AgentConfig,
instance: AgentInstance,
initial_prompt: str | None,
agent_settings_path: Path | None,
*,
spawned_by: str | None = None,
) -> AgentInstance:
"""Launch the container and emit spawn audit events.
`agent_id` was dropped as a redundant parameter — `config.agent_id`
is the same value and was always the caller's source.
``spawned_by`` names the dispatch loop that requested the spawn; it is
stamped into the spawned/spawn_failed audit details so a rogue spawner
is identifiable from the audit log alone.
"""
agent_slug = config.agent_id
try:
container_id = await self._spawn_container(
config, initial_prompt, agent_settings_path
)
instance.container_id = container_id
instance.state = AgentState.ACTIVE
instance.started_at = datetime.now(UTC)
instance.last_activity = datetime.now(UTC)
logger.info(
"Agent spawned",
agent_id=agent_slug,
container_id=container_id[:12],
model=config.model,
task_id=task_id,
)
self._fire_audit(
event_type="agent.spawned",
agent_slug=agent_slug,
task_id=task_id,
details={
"container_id": container_id[:12],
"model": config.model,
"spawned_by": spawned_by or "unspecified",
},
)
# Record a token-usage session row in the DB and bind its UUID to
# the instance so _finalize_spawn_session can look it up directly.
usage_session_id = await self._record_spawn_session(config, task_id)
if usage_session_id is not None:
instance.usage_session_id = usage_session_id
return instance
except Exception as e:
instance.state = AgentState.OFFLINE
instance.error_count += 1
logger.error(
"Failed to spawn agent",
agent_id=agent_slug,
error=str(e),
)
self._fire_audit(
event_type="agent.spawn_failed",
agent_slug=agent_slug,
task_id=task_id,
details={
"error": str(e),
"spawned_by": spawned_by or "unspecified",
},
severity="error",
)
raise
@staticmethod
def _spawn_preflight_reason(agent_id: str) -> str | None:
"""Refusal reason if this spawn is deterministically futile, else None.
Flag-gated (``spawn_preflight_enabled``, default off). A non-human
delivery role absent from ``GATEWAY_ENABLED_ROLES`` gets no manifest and
``ROBOCO_GATEWAY_ENABLED=false``, so it can never claim its work and the
dispatcher would respawn it on the same task forever. Fail fast instead of
burning the full system prompt on each futile retry.
"""
if not settings.spawn_preflight_enabled:
return None
role = get_agent_role(agent_id)
if role is not None and role not in GATEWAY_ENABLED_ROLES:
return (
f"role {role!r} ({agent_id}) is not gateway-enabled — it could "
f"never claim its work and would respawn forever"
)
return None
async def _refuse_unspawnable(self, agent_id: str, task_id: str | None) -> None:
"""Chokepoint guards for ``spawn_agent``; raise ``AgentReadinessError``.
Three refusals, in order:
1. A traversal-shaped ``agent_id`` (it flows into log dirs, settings and
container names) — rejected before any filesystem op.
2. Human-only roles (ceo / prompter / secretary) are NEVER spawned by a
dispatcher. The CEO is the human operator; intake and secretary are
human-driven chats launched through their own guarded paths
(_spawn_intake_container / _spawn_secretary_container), not this
method. Without this, a dispatcher that spawns "any A2A/notification
target" could resolve a CEO-addressed notification to slug "ceo" and
launch a CEO container — the system acting as the human CEO.
3. Spawn preflight (flag-gated): a non-gateway delivery role could never
claim its work and would respawn forever — refuse + alert once.
"""
AgentOrchestrator._safe_agent_path_segment(agent_id)
task_id_str = str(task_id) if task_id else None
_role = role_for_slug_or_none(agent_id)
if is_human_only_role(_role):
logger.error(
"spawn_agent refused for human-only role — dispatchers must never"
" spawn the CEO / prompter / secretary; these are human-driven",
agent_id=agent_id,
role=str(_role),
task_id=task_id_str,
)
raise AgentReadinessError(
f"refused to spawn human-only role {_role!r} ({agent_id}) — the"
f" CEO is the human operator, not a container; intake and secretary"
f" launch through their dedicated paths, not spawn_agent"
)
preflight_reason = AgentOrchestrator._spawn_preflight_reason(agent_id)
if preflight_reason:
logger.error(
"spawn_agent refused (spawn preflight): role not gateway-enabled",
agent_id=agent_id,
task_id=task_id_str,
)
if task_id_str:
await self._notify_stuck_agent(agent_id, task_id_str, None)
raise AgentReadinessError(preflight_reason)
async def spawn_agent(
self,
agent_id: str,
initial_prompt: str | None = None,
task_id: str | None = None,
model: str | None = None,
git_context: SpawnGitContext | None = None,
*,
spawned_by: str | None = None,
) -> AgentInstance:
"""
Spawn a Claude Code container for an agent.
Args:
agent_id: Agent identifier (e.g., "be-dev-1")
initial_prompt: Optional initial prompt
task_id: Optional task ID being worked on
model: Override model selection
git_context: Optional git context (project_slug, branch_name)
spawned_by: Name of the dispatch loop / entry point requesting
the spawn — stamped into the agent.spawned audit details
Returns:
AgentInstance handle
Raises:
AgentReadinessError: task is not spawn-ready (missing criteria,
missing git token, no branch plan, role mismatch). The task
is auto-blocked before we raise so the dispatcher doesn't
keep retrying.
"""
# Chokepoint guards: path-safe id, never-a-human-role, spawn preflight.
await self._refuse_unspawnable(agent_id, task_id)
# Pre-flight: refuse to spawn if the task isn't ready. Auto-block
# on refusal so the dispatcher doesn't keep spinning a container
# that will immediately fail (wasted image pull + startup tokens).
readiness_reason = await self._readiness_gate(agent_id, task_id)
if readiness_reason:
raise AgentReadinessError(
f"spawn refused for {agent_id} (task={task_id}): {readiness_reason}"
)
# Auto-derive git_context when the caller didn't supply one. Two
# paths:
# (a) task_id present → look up the task's project;
# (b) no task_id → fall back to the sole active project (or
# the first one if there are multiple).
# Without (b), no-task spawns (e.g. idle PM bootstrapping) hit the
# "workspace fallback used" path and get mounted at
# /data/workspaces/default/... which doesn't exist.
git_context = await self._resolve_spawn_git_context(git_context, task_id)
async with self._lock:
existing = self._existing_running_instance(agent_id)
if existing is not None:
return existing
# Provider-parking loop-breaker (cheap pre-check): while this agent's
# provider is parked (rate-limited or overloaded), do NOT run the full
# ``_prepare_agent_spawn`` — which writes the blueprint / settings /
# briefing / MCP-config files, ensures the agent image, and registers a
# STARTING instance — only to bail. The dispatcher re-ticks a parked
# agent every cycle, so running the full prepare each tick wasted all
# that file I/O and left a STARTING instance registered then downgraded
# to OFFLINE. The parked check only needs ``provider_type``, cheaply
# resolvable via ``_resolve_agent_route``. Bailing here returns a
# minimal UNREGISTERED OFFLINE instance (no stale ``_instances`` entry),
# so the next tick re-checks cheaply until the provider recovers. The
# existing-running check above stays first, so a live agent is never
# replaced by this bail. Fail-open: a tracker read error never blocks.
route = await self._resolve_agent_route(agent_id)
if await self._provider_spawn_parked(route.provider_type.value):
self._mark_task_handled(task_id)
logger.info(
"Spawn skipped: provider rate-limited (parked)",
agent_id=agent_id,
task_id=task_id,
provider=route.provider_type.value,
)
return AgentInstance(
agent_id=agent_id,
state=AgentState.OFFLINE,
config=AgentConfig(
agent_id=agent_id,
blueprint_path=Path(), # not launching — no blueprint written
model=route.model_name,
provider_type=route.provider_type.value,
provider_base_url=route.base_url,
provider_auth_token=route.auth_token,
git_context=git_context,
),
current_task_id=task_id,
)
async with self._lock:
# TOCTOU re-check: another tick may have started this agent during
# the unlocked route resolve + parked check above. Re-check before
# the expensive prepare so two concurrent ticks don't double-spawn.
existing = self._existing_running_instance(agent_id)
if existing is not None:
return existing
config, instance, agent_settings_path = await self._prepare_agent_spawn(
agent_id, task_id, model, git_context
)
# Rare-race defense: a park could land during prepare. The every-tick
# parked case is already handled above; this guards the window between
# the pre-check and the launch. Fail-open: a tracker read error never
# blocks spawning.
if await self._provider_spawn_parked(config.provider_type):
self._mark_task_handled(task_id)
instance.state = AgentState.OFFLINE
logger.info(
"Spawn skipped: provider rate-limited (parked)",
agent_id=agent_id,
task_id=task_id,
provider=config.provider_type,
)
return instance
# Record the task as handled so later dispatchers in the same
# tick don't act on it again. Safe even if _launch_spawn fails
# — the next tick starts fresh.
self._mark_task_handled(task_id)
return await self._launch_spawn(
task_id,
config,
instance,
initial_prompt,
agent_settings_path,
spawned_by=spawned_by,
)
def _resolve_host_paths(
self, config: AgentConfig, agent_settings_path: Path | None
) -> dict[str, str | None]:
"""Compute host mount paths for both containerized and host runtime."""
mcp_name = config.mcp_config_path.name if config.mcp_config_path else ""
if PROJECT_HOST_PATH:
return {
"docs": f"{PROJECT_HOST_PATH}/docs",
"workspaces": f"{DATA_HOST_PATH}/workspaces",
"claude": CLAUDE_AUTH_HOST_PATH,
"mcp_config": f"{DATA_HOST_PATH}/mcp-configs/{mcp_name}",
# Per-agent grok usage dir (GROK only); the orchestrator reads the
# captured tokens back at finalize via the shared data volume
# (see GROK_USAGE_DATA_DIR).
"grok_usage": f"{DATA_HOST_PATH}/grok-usage/{config.agent_id}",
"prompt": (
f"{DATA_HOST_PATH}/prompts-generated/{config.agent_id}-prompt.md"
),
"settings": (
f"{DATA_HOST_PATH}/agent-settings/{config.agent_id}-settings.json"
if agent_settings_path
else None
),
"briefing": (
f"{DATA_HOST_PATH}/briefings/{config.agent_id}.md"
if config.briefing_path
else None
),
}
return {
"docs": str((self.project_root / "docs").absolute()),
"workspaces": str(Path(settings.workspaces_root)),
"claude": CLAUDE_AUTH_HOST_PATH,
"mcp_config": str(config.mcp_config_path),
"grok_usage": str(
Path(tempfile.gettempdir()) / "roboco-grok-usage" / config.agent_id
),
"prompt": str(
Path(tempfile.gettempdir())
/ "roboco-prompts"
/ f"{config.agent_id}-prompt.md"
),
"settings": str(agent_settings_path) if agent_settings_path else None,
"briefing": (str(config.briefing_path) if config.briefing_path else None),
}
@staticmethod
def _build_mount_args(
container_name: str, config: AgentConfig, hosts: dict[str, str | None]
) -> list[str]:
"""Compose `docker run -v/-e` mount + env args for the agent."""
cmd: list[str] = [
"docker",
"run",
"-d",
"--name",
container_name,
"--network",
AGENT_NETWORK,
# Mount Claude auth directory (for API keys, etc.)
"-v",
f"{hosts['claude']}:/home/agent/.claude",
]
AgentOrchestrator._append_claude_json_mount(cmd, hosts)
AgentOrchestrator._append_optional_host_mounts(cmd, hosts)
role = get_agent_role(config.agent_id) or "developer"
cmd.extend(AgentOrchestrator._core_volume_and_env_args(config, hosts, role))
AgentOrchestrator._append_provider_env(cmd, config)
subagent_model = _resolve_agent_cli_model(config.provider_type, config.model)
cmd.extend(["-e", f"CLAUDE_CODE_SUBAGENT_MODEL={subagent_model}"])
AgentOrchestrator._append_manifest_args(cmd, config, subagent_model)
AgentOrchestrator._append_workspace_cwd(cmd, config)
return cmd
@staticmethod
def _append_claude_json_mount(cmd: list[str], hosts: dict[str, str | None]) -> None:
"""Mount host's ~/.claude.json sibling FILE if present."""
claude_dir = hosts["claude"]
if not claude_dir:
return
claude_json_host = f"{claude_dir.rstrip('/')}.json"
if Path(claude_json_host).exists():
cmd.extend(["-v", f"{claude_json_host}:/home/agent/.claude.json"])
@staticmethod
def _append_optional_host_mounts(
cmd: list[str], hosts: dict[str, str | None]
) -> None:
"""Mount agent settings.json and briefing.md when their hosts exist."""
settings_host = hosts.get("settings")
if settings_host:
cmd.extend(["-v", f"{settings_host}:/home/agent/.claude/settings.json:ro"])
briefing_host = hosts.get("briefing")
if briefing_host:
cmd.extend(["-v", f"{briefing_host}:/app/briefing.md:ro"])
@staticmethod
def _core_volume_and_env_args(
config: AgentConfig, hosts: dict[str, str | None], role: str
) -> list[str]:
"""The always-on -v/-e block (prompt, docs, workspaces, env)."""
docs_ro = "" if config.agent_id in ALL_DOCS else ":ro"
env = [
"-v",
f"{hosts['prompt']}:/app/system-prompt.md:ro",
"-v",
f"{hosts['docs']}:/app/docs{docs_ro}",
"-v",
f"{hosts['workspaces']}:/data/workspaces",
"-v",
f"{hosts['mcp_config']}:/app/mcp-config.json:ro",
"-e",
# Auth identity is the agent's UUID, not its slug: the MCP servers
# forward X-Agent-ID as the UUID (gateway v1 endpoints parse it as
# Annotated[UUID]) and the HMAC token is signed over the same value,
# so the container env the SDK server inherits must match or its
# direct API calls 401 with "signature mismatch".
f"ROBOCO_AGENT_ID={AGENT_UUIDS.get(config.agent_id, config.agent_id)}",
"-e",
f"ROBOCO_AGENT_ROLE={role}",
"-e",
"ROBOCO_API_URL=http://roboco-orchestrator:8000",
"-e",
"ROBOCO_SDK_PORT=9000",
"-e",
"ROBOCO_SDK_URL=http://localhost:9000",
"-e",
f"ROBOCO_AGENT_TOOL_CALL_WARN={settings.agent_tool_call_warn}",
"-e",
f"ROBOCO_AGENT_TOOL_CALL_HALT={settings.agent_tool_call_halt}",
"-e",
f"ROBOCO_AGENT_LOOP_THRESHOLD={settings.agent_loop_threshold}",
"-e",
f"ROBOCO_AGENT_LOOP_WINDOW={settings.agent_loop_window}",
"-e",
f"ROBOCO_AGENT_STOP_ATTEMPT_ALLOWANCE={settings.agent_stop_attempt_allowance}",
]
return env
@staticmethod
def _append_provider_env(cmd: list[str], config: AgentConfig) -> None:
"""Inject ANTHROPIC_* env only on non-Anthropic providers."""
# Provider routing: only inject ANTHROPIC_* env vars when the
# resolved provider is non-Anthropic (i.e. Ollama Cloud). For the
# Anthropic default path both fields are None and Claude Code
# inside the container continues to use its mounted ~/.claude
# credentials — preserving legacy behaviour byte-for-byte.
if config.provider_base_url:
cmd.extend(["-e", f"ANTHROPIC_BASE_URL={config.provider_base_url}"])
if config.provider_auth_token:
cmd.extend(["-e", f"ANTHROPIC_AUTH_TOKEN={config.provider_auth_token}"])
@staticmethod
def _append_manifest_args(
cmd: list[str], config: AgentConfig, subagent_model: str
) -> None:
"""Write the spawn manifest and flip the gateway flag."""
# Spawn manifest + gateway flag — developer role only in Phase 1.
# _build_manifest_for_agent writes the JSON file to the host and
# returns the path; other roles get None and the gateway flag stays off.
# workspace_path mirrors the container -w (same resolver) so the
# manifest never claims a different directory than the shell.
manifest_host_path = _build_manifest_for_agent(
config.agent_id,
subagent_model,
workspace_path=AgentOrchestrator._resolve_workspace_cwd(config),
)
if manifest_host_path:
cmd.extend(
[
"-v",
f"{manifest_host_path}:/app/tool-manifest.json:ro",
"-e",
"ROBOCO_GATEWAY_ENABLED=true",
"-e",
"ROBOCO_TOOL_MANIFEST_PATH=/app/tool-manifest.json",
]
)
else:
cmd.extend(["-e", "ROBOCO_GATEWAY_ENABLED=false"])
_ROLES_WITH_AGENT_WORKSPACE: ClassVar[frozenset[str]] = frozenset(
{"developer", "product_owner", "head_marketing"}
)
_ROLES_WITH_CELL_WORKSPACE: ClassVar[frozenset[str]] = frozenset({"documenter"})
@staticmethod
def _resolve_workspace_cwd(config: AgentConfig) -> str | None:
"""The task-resolved workspace path for this spawn, or None.
Single source of truth consumed by BOTH the container ``-w`` and the
spawn manifest's ``workspace_path`` — they must agree, or the agent's
prompt claims one directory while its shell sits in another (live
2026-07-02: manifest said the roboco workspace for a guard-core task).
"""
role = get_agent_role(config.agent_id) or "developer"
team = get_agent_team(config.agent_id) or ""
project = _resolve_project_slug_from_git_context(config.git_context)
if role in AgentOrchestrator._ROLES_WITH_AGENT_WORKSPACE:
# Per-task worktree when the task has a branch (F123), else the
# clone root. _agent_cwd_path is the SAME formula the Edit/Write
# allowlist is built from, so -w and the allowlist match exactly.
return _agent_cwd_path(project, team, config.agent_id, config.git_context)
if role in AgentOrchestrator._ROLES_WITH_CELL_WORKSPACE:
return _cell_workspace_path(project, team)
return None
@staticmethod
def _append_workspace_cwd(cmd: list[str], config: AgentConfig) -> None:
"""Set the container -w to the agent or cell workspace by role."""
# Pre-gateway parity: set the container's cwd
# to the agent's task workspace so Edit/Write resolve to paths that
# match _get_role_permissions allowlist, and `git add` operates inside
# the workspace clone. Without this, container WORKDIR (/app from the
# Dockerfile) shadows the workspace and every file op fails.
#
# Workspace selection lives in _resolve_workspace_cwd:
# - developer / product_owner / head_marketing: per-agent workspace
# - documenter: cell workspace
# - qa / cell_pm / main_pm / auditor: no write workspace → omit -w
workspace = AgentOrchestrator._resolve_workspace_cwd(config)
if workspace is not None:
cmd.extend(["-w", workspace])
@staticmethod
def _append_agent_auth_env(cmd: list[str], config: AgentConfig) -> None:
"""Append agent HMAC token env var to the docker run cmd."""
# Agent HMAC auth token — bound to (agent_id, role, team). The
# API middleware refuses requests whose headers don't match the
# token, which stops one agent on the Docker network from
# spoofing another agent's role. Token is stable per agent as
# long as the secret doesn't rotate, so it's fine to compute at
# spawn time and inject once.
from roboco.agents_config import (
get_agent_role as _get_role,
)
from roboco.agents_config import (
get_agent_team as _get_team,
)
from roboco.agents_config import (
issue_agent_token,
)
_role = _get_role(config.agent_id)
_team = _get_team(config.agent_id) or ""
# Sign over the UUID, not the slug: every in-container caller (MCP
# servers via the manifest env, SDK server via this container env)
# sends X-Agent-ID as the UUID, so the HMAC payload must be the UUID
# or the middleware rejects with "signature mismatch". AGENT_UUIDS
# maps slug→UUID; fall back to the slug for custom agents not seeded.
_agent_uuid = AGENT_UUIDS.get(config.agent_id, config.agent_id)
_token = issue_agent_token(
_agent_uuid,
_role,
_team,
ttl_seconds=settings.agent_token_ttl_seconds,
)
cmd.extend(["-e", f"ROBOCO_AGENT_TOKEN={_token}"])
@staticmethod
def _append_git_context_env(cmd: list[str], config: AgentConfig) -> None:
"""Append git-context env vars to the docker run cmd."""
if not config.git_context:
return
if config.git_context.project_slug:
cmd.extend(["-e", f"ROBOCO_PROJECT_SLUG={config.git_context.project_slug}"])
if config.git_context.branch_name:
cmd.extend(["-e", f"ROBOCO_BRANCH={config.git_context.branch_name}"])
@staticmethod
def _append_gate_env(cmd: list[str]) -> None:
"""Inject the test-DB env so an agent's gate runs the real, DB-backed
suite instead of a hollow unit-only subset.
Without a reachable Postgres the conftest skips every integration test,
so coverage collapses far below the gate threshold and a role 'gates'
against a partial run (the failure that made a PM read 71% on a suite
that is ~96% with a DB). The values come from the orchestrator's own DB
settings; agents share the Docker network, so the host resolves. The app
runtime reads ROBOCO_DATABASE_*, never ROBOCO_TEST_DB_*, so this only
feeds the test harness and never changes live behaviour. Gated on the
same faithful-gate flag as interpreter matching — both exist to make an
agent's self-gate trustworthy.
Under DB network isolation (postgres/redis on the data-only compose
network) agents cannot reach these hosts at all, so the injection is
suppressed entirely: creds that dead-end in a connect timeout are
worse than none (the conftest reachability check skips cleanly on a
fast refusal). DB-needing projects opt into `sandbox_services` instead.
"""
if settings.db_network_isolated:
return
if not settings.toolchain_match_enabled:
return
cmd.extend(
[
"-e",
f"ROBOCO_TEST_DB_HOST={settings.database_host}",
"-e",
f"ROBOCO_TEST_DB_PORT={settings.database_port}",
"-e",
f"ROBOCO_TEST_DB_USER={settings.database_user}",
"-e",
f"ROBOCO_TEST_DB_PASSWORD={settings.database_password}",
"-e",
"ROBOCO_TEST_DB_ADMIN_DB=postgres",
]
)
@staticmethod
def _append_sandbox_env(cmd: list[str], config: AgentConfig) -> None:
"""Inject sandbox engine env, in place of the prod-creds gate env.
Called INSTEAD OF `_append_gate_env` whenever a sandbox was provisioned
for this spawn (`config.sandbox_info` set) — sandbox replaces, never
coexists with, the production gate-env injection. Emission is driven by
the engine registry via `SandboxInfo.emit_env`, so a new engine's
`ROBOCO_TEST_*` vars land here with no orchestrator change.
"""
info = config.sandbox_info
if info is None:
return
cmd.extend(info.emit_env())
@staticmethod
def _default_spawn_prompt() -> str:
"""Fallback prompt when the caller provided none."""
return (
"You may have been spawned without a specific task assignment. "
"Follow your standard workflow:\n\n"
"1. Call `give_me_work()` to find work for your role\n"
"2. Begin the assigned task (its details arrive in the "
"response): UNDERSTAND -> PLAN -> EXECUTE -> VERIFY -> HANDOFF\n"
"3. If no tasks available, call `i_am_idle()` "
"to shutdown gracefully\n\n"
"Start now by scanning for work."
)
@classmethod
def _append_image_and_claude_args(
cls, cmd: list[str], config: AgentConfig, initial_prompt: str | None
) -> None:
"""Append the image + Claude Code CLI args to the docker run cmd.
`--tools` explicitly enumerates the built-in tools loaded at session
start. Without it, Claude CLI's default behavior leaves Edit/Write
in the deferred pool, so an agent that doesn't reliably call
ToolSearch (e.g. weaker non-Anthropic models routed via
Ollama-cloud) ends up unable to modify any file. The set below is
the minimum every agent role needs:
- Read/Write/Edit : file IO inside the workspace
- Bash : shell commands (gated by bash-guard hook)
- Grep/Glob : code navigation
- TodoWrite : per-session planning
Permissions still gate *which* paths Edit/Write can touch (see
`_get_role_permissions`), so this is purely about loading vs
denying.
`--disable-slash-commands` closes a separate capability channel
`--tools` doesn't reach: skills/slash-commands resolve independently
of the built-in tool allowlist (Anthropic's own `--bare` flag docs
call this out — skills still resolve via `/skill-name` even with
everything else disabled). The agent's `~/.claude` is the host's
shared Claude Code auth dir, bind-mounted into every container
(`_build_mount_args`); if it ever carries personal
skills/plugins/marketplace installs, this stops them from silently
becoming callable inside the agent's session. No RoboCo role's
workflow uses a Claude Code skill (their surface is the MCP gateway
+ the `--tools` set above), so this has no legitimate flow to break.
"""
claude_args = [
get_agent_image(config.agent_id),
"--model",
cls._resolve_cli_model(config),
"--system-prompt-file",
"/app/system-prompt.md",
"--mcp-config",
"/app/mcp-config.json",
"--strict-mcp-config",
"--tools",
"Read,Write,Edit,Bash,Grep,Glob,TodoWrite",
"--disable-slash-commands",
"--output-format",
"stream-json",
"--verbose",
]
# Per-role reasoning-effort override via Claude Code's `--effort` flag.
# Only set for roles in ROLE_EFFORT_MAP; models without effort support
# ignore it, so passing it for a mapped role is always safe.
_effort_role = get_agent_role(config.agent_id)
_effort = ROLE_EFFORT_MAP.get(_effort_role) if _effort_role else None
if _effort:
claude_args += ["--effort", _effort]
# Pin the Claude session id so the agent's transcript is locatable by id
# at finalize, regardless of which project/cwd dir Claude Code writes it
# to (review/coordinate roles run at /app, not a per-agent workspace).
if config.claude_session_id:
claude_args += ["--session-id", config.claude_session_id]
claude_args += ["-p", initial_prompt or cls._default_spawn_prompt()]
cmd.extend(claude_args)
@staticmethod
def _resolve_cli_model(config: AgentConfig) -> str:
"""Return the string to pass to `claude --model`."""
return _resolve_agent_cli_model(config.provider_type, config.model)
def _ensure_provider_registry(self) -> "ProviderRegistry":
"""Build (once) the registry of dedicated provider backends.
Only providers that need a runtime other than the built-in Claude Code
container are registered. Today that is GROK (xAI, OpenAI protocol).
"""
if self._provider_registry is None:
from roboco.llm.providers import GrokCliProvider, ProviderRegistry
from roboco.models.base import ModelProvider
registry = ProviderRegistry()
# Qualify the grok image with the registry namespace + tag so it
# resolves in both local-build and registry deploys (parity with
# get_agent_image for the Claude path).
registry.register(
ModelProvider.GROK,
GrokCliProvider(self, image=_qualify_agent_image("roboco-agent-grok")),
)
self._provider_registry = registry
return self._provider_registry
def _provider_for(self, provider_type: str) -> "AgentProvider | None":
"""Resolve a dedicated provider for a route's ``provider_type`` string.
Returns ``None`` for providers that use the built-in Claude Code spawn
(Anthropic / Ollama Cloud / self-hosted) or any unrecognised value — the
caller then runs the existing container path unchanged.
"""
from roboco.models.base import ModelProvider
try:
model_provider = ModelProvider(provider_type)
except ValueError:
return None
return self._ensure_provider_registry().get_or_none(model_provider)
async def _spawn_container(
self,
config: AgentConfig,
initial_prompt: str | None = None,
agent_settings_path: Path | None = None,
) -> str:
"""Spawn a Docker container for the agent.
Args:
config: Agent configuration
initial_prompt: Optional initial prompt for the agent
agent_settings_path: Path to per-agent Claude settings file
"""
# Every spawn gets a non-empty user prompt. A prompt-less spawn (e.g. the
# crash auto-restart, which passes no initial_prompt) must still direct the
# agent to scan for work. The Claude body re-applies the same default; doing
# it here single-sources it so dedicated providers (GROK) get it too —
# otherwise grok would launch with an empty `grok -p ""`.
if not initial_prompt:
initial_prompt = self._default_spawn_prompt()
# A dedicated provider backend (e.g. GROK / OpenAI protocol) handles its
# own spawn. Anthropic / Ollama Cloud / self-hosted have no dedicated
# provider registered and fall through to the Claude Code body below,
# byte-for-byte unchanged.
provider = self._provider_for(config.provider_type)
if provider is not None:
result = await provider.spawn(config, initial_prompt, agent_settings_path)
return result.instance_id
container_name = f"roboco-agent-{config.agent_id}"
# teardown_sandbox=False: this spawn's sandbox was provisioned moments
# ago in _build_agent_config — the stale-clear must not destroy it.
# Stale sandboxes from a prior crash are cleared by provision() itself.
await self._remove_container(
container_name, teardown_sandbox=False, stop_reason="pre_spawn_stale_clear"
)
if not config.mcp_config_path:
raise RuntimeError("MCP config path not set")
hosts = self._resolve_host_paths(config, agent_settings_path)
cmd = self._build_mount_args(container_name, config, hosts)
self._append_agent_auth_env(cmd, config)
self._append_git_context_env(cmd, config)
if config.sandbox_info is not None:
self._append_sandbox_env(cmd, config)
else:
self._append_gate_env(cmd)
self._append_image_and_claude_args(cmd, config, initial_prompt)
proc = await asyncio.create_subprocess_exec(
*cmd,
stdout=asyncio.subprocess.PIPE,
stderr=asyncio.subprocess.PIPE,
)
stdout, stderr = await proc.communicate()
if proc.returncode != 0:
raise RuntimeError(f"Failed to start container: {stderr.decode()}")
return stdout.decode().strip()
def _record_expected_stop(self, agent_id: str, reason: str) -> None:
"""Breadcrumb an orchestrator-initiated stop/kill for ``agent_id``.
Diagnostics only (in-memory, no DB): lets the exit monitor tell an
attributed stop from a genuinely unexplained one instead of logging
every death as "unexpectedly". Bounded: past a size threshold, stale
entries are dropped opportunistically rather than growing forever.
``getattr`` defaults the registry so a ``__new__``-constructed test
instance (no ``__init__``) doesn't need to know about it either.
"""
stops: dict[str, tuple[str, float]] | None = getattr(
self, "_expected_stops", None
)
if stops is None:
stops = self._expected_stops = {}
if len(stops) > _EXPECTED_STOP_MAX_ENTRIES:
cutoff = time.monotonic() - _EXPECTED_STOP_FRESH_SECONDS
stops = self._expected_stops = {
k: v for k, v in stops.items() if v[1] >= cutoff
}
stops[agent_id] = (reason, time.monotonic())
def _consume_expected_stop(self, agent_id: str) -> str:
"""Pop and return the breadcrumb reason for ``agent_id``, else "none_recorded".
A breadcrumb older than ``_EXPECTED_STOP_FRESH_SECONDS`` is treated as
stale (not fresh enough to attribute to *this* exit) and reported the
same as no breadcrumb at all. Defensive on a missing registry, like
``_record_expected_stop``.
"""
stops: dict[str, tuple[str, float]] | None = getattr(
self, "_expected_stops", None
)
entry = stops.pop(agent_id, None) if stops else None
if entry is None:
return "none_recorded"
reason, recorded_at = entry
if time.monotonic() - recorded_at > _EXPECTED_STOP_FRESH_SECONDS:
return "none_recorded"
return reason
@staticmethod
async def _inspect_exit_diagnostics(container_name: str) -> dict[str, Any]:
"""Best-effort extra `docker inspect` fields for a dead container's log line.
Cheap (one more inspect the monitor already does one of) and never
raises — any failure/timeout yields {} so the caller's log line still
emits with whatever fields it already had.
"""
try:
proc = await asyncio.create_subprocess_exec(
"docker",
"inspect",
"-f",
"{{.State.OOMKilled}}|{{.State.StartedAt}}|{{.State.FinishedAt}}|"
"{{.State.Error}}",
container_name,
stdout=asyncio.subprocess.PIPE,
stderr=asyncio.subprocess.DEVNULL,
)
stdout, _ = await asyncio.wait_for(
proc.communicate(), timeout=_DOCKER_INSPECT_TIMEOUT_SECONDS
)
oom, started, finished, error = stdout.decode().strip().split("|")
except Exception:
return {}
return {
"oom_killed": oom == "true",
"started_at": started or None,
"finished_at": finished or None,
"state_error": error or None,
}
async def _remove_container(
self,
container_name: str,
*,
teardown_sandbox: bool = True,
stop_reason: str | None = None,
) -> None:
"""Remove a container if it exists, dumping its logs to disk first.
Docker deletes the container's json-file log when we `docker rm`, so
before removal we copy the current log to /data/logs/agents/{slug}/
with a timestamp. That gives us persistent history across respawns
without needing an entrypoint wrapper inside the agent image.
``teardown_sandbox=False`` is passed only by the pre-spawn stale-clear,
whose spawn has already provisioned the sandbox it is about to use.
``stop_reason``, when given, breadcrumbs this removal so the exit
monitor can attribute the death instead of flagging it unexplained.
``None`` (the default) skips it — used by callers (``stop_agent``)
that already recorded their own breadcrumb earlier, before their
docker stop/kill, so this call doesn't clobber it with "unknown".
"""
if stop_reason is not None:
self._record_expected_stop(
container_name.removeprefix("roboco-agent-"), stop_reason
)
# Check the container actually exists before trying to dump logs;
# _remove_container is routinely called pre-spawn to clear stale
# containers, and on first spawn there's nothing to dump.
inspect = await asyncio.create_subprocess_exec(
"docker",
"inspect",
"--format={{.Id}}",
container_name,
stdout=asyncio.subprocess.DEVNULL,
stderr=asyncio.subprocess.DEVNULL,
)
exists = (await inspect.wait()) == 0
if exists:
slug = container_name.removeprefix("roboco-agent-")
try:
# slug builds a path under /data/logs/agents — reject a
# traversal-shaped slug before the join (defense-in-depth;
# spawn_agent already validates the agent_id this container
# name is derived from). A bad slug skips the log dump.
AgentOrchestrator._safe_agent_path_segment(slug)
log_dir = Path("/data/logs/agents") / slug
log_dir.mkdir(parents=True, exist_ok=True)
timestamp = datetime.now(UTC).strftime("%Y%m%dT%H%M%SZ")
log_path = log_dir / f"{timestamp}.log"
with log_path.open("wb") as out:
dump_proc = await asyncio.create_subprocess_exec(
"docker",
"logs",
container_name,
stdout=out,
stderr=out,
)
await dump_proc.wait()
if log_path.stat().st_size == 0:
log_path.unlink(missing_ok=True)
except Exception as e:
logger.warning(
"Could not dump container logs before removal",
container=container_name,
error=str(e),
)
proc = await asyncio.create_subprocess_exec(
"docker",
"rm",
"-f",
container_name,
stdout=asyncio.subprocess.DEVNULL,
stderr=asyncio.subprocess.DEVNULL,
)
await proc.wait()
# Sandbox lifetime tracks the agent container 1:1 — every removal
# path (stop_agent, reaper kills) routes through here. Gated on the
# flag: teardown is idempotent but not free (up to 4 extra docker
# calls), so skip it when the feature was never on. Never raises
# (SandboxProvisioner.teardown contract).
if teardown_sandbox and settings.sandbox_db_enabled:
slug = container_name.removeprefix("roboco-agent-")
await self._sandbox.teardown(slug)
async def _generate_mcp_config(
self,
agent_id: str,
git_context: SpawnGitContext | None = None,
) -> Path:
"""Generate MCP config for an agent.
Post-gateway: every state-changing tool routes through one of two
servers, and read-only views go through two more:
- roboco-flow intent verbs (lifecycle transitions)
- roboco-do content tools (commit, push, PR, journal,
notify, message)
- roboco-git-readonly status, log, diff, branch list
- roboco-optimal knowledge base, RAG, semantic search
- roboco-docs documentation file management (panel docs)
The agent's role is asserted by the orchestrator API on every
verb/tool call, so all roles get the same MCP surface from this
registration; verbs the agent's role can't run return a
not-authorized error rather than 404. Git context is forwarded
only as a fallback for tools that resolve project/branch from env.
"""
# MCP servers run inside agent containers, need to connect via Docker network
if PROJECT_HOST_PATH:
api_url = "http://roboco-orchestrator:8000"
else:
api_url = f"http://127.0.0.1:{settings.port}"
agent_role = get_agent_role(agent_id) or ""
# Gateway v1 endpoints declare X-Agent-ID as Annotated[UUID, Header(...)],
# so the MCP server has to forward the agent's UUID — not the slug — or
# every gateway call 422s on header parse. Resolve via AGENT_UUIDS map;
# if the slug isn't in the map (custom agents), fall back to the slug
# and let the API surface the unknown-agent error.
# Also used as the CLI arg for the three ApiClient-based servers
# (optimal/docs/search) below — their spawn token (issue_agent_token)
# is signed over the UUID, so ApiClient's X-Agent-ID must match or
# verify_agent_token 401s with "signature mismatch" even though
# get_agent_role/get_agent_team resolve either form fine.
agent_uuid = AGENT_UUIDS.get(agent_id, agent_id)
mcp_env: dict[str, str] = {
"ROBOCO_API_URL": api_url,
"ROBOCO_ORCHESTRATOR_URL": api_url,
"ROBOCO_AGENT_ID": agent_uuid,
"ROBOCO_AGENT_ROLE": agent_role,
# Mirrors the server-side FlowVerbTimeoutMiddleware budgets so the
# roboco-flow MCP client's per-verb timeout (flow_server.py, which
# can't read Settings directly) stays coherent with operator
# tuning of either setting.
"ROBOCO_FLOW_VERB_TIMEOUT_SECONDS": str(settings.flow_verb_timeout_seconds),
"ROBOCO_FLOW_VERB_SLOW_TIMEOUT_SECONDS": str(
settings.flow_verb_slow_timeout_seconds
),
# Every MCP server is launched as `uv run python -m
# roboco.mcp.<server>` by Claude Code, with cwd = the agent's
# WORKSPACE (not /app). Without this, `uv run` resolves a
# cwd-relative `.venv` (≠ the pre-baked /app/.venv) and RE-SYNCS
# the full dependency set (torch/lancedb/pyarrow/scipy, ~350MB) into a
# fresh venv on every spawn — masked by a warm uv wheel cache,
# but on a cold cache (first spawn after an image rebuild) the
# download takes minutes and the MCP servers never come up
# before the agent burns its budget. Pinning the project env
# to the pre-baked venv is necessary but NOT sufficient: `uv run`
# still resolves the project from the workspace cwd and re-syncs
# when the clone's uv.lock drifts from the image — leaving the MCP
# servers stuck at status="pending" so the agent gets zero gateway
# verbs. Each server is therefore launched with `uv run --no-sync`
# (below) to use /app/.venv as-is and start instantly.
"UV_PROJECT_ENVIRONMENT": "/app/.venv",
}
# Add git context if available
if git_context:
if git_context.project_slug:
mcp_env["ROBOCO_PROJECT_SLUG"] = git_context.project_slug
if git_context.branch_name:
mcp_env["ROBOCO_BRANCH"] = git_context.branch_name
mcp_servers: dict[str, dict[str, Any]] = {
# Intent verbs — every role-scoped lifecycle transition.
"roboco-flow": {
"command": "uv",
"args": ["run", "--no-sync", "python", "-m", "roboco.mcp.flow_server"],
"env": mcp_env,
},
# Content tools — commit, push, PR, journal, notify, message.
"roboco-do": {
"command": "uv",
"args": ["run", "--no-sync", "python", "-m", "roboco.mcp.do_server"],
"env": mcp_env,
},
# Read-only git views — status, log, diff, branches.
"roboco-git-readonly": {
"command": "uv",
"args": ["run", "--no-sync", "python", "-m", "roboco.mcp.git_readonly"],
"env": mcp_env,
},
# Knowledge base — RAG / semantic search / ask_mentor.
"roboco-optimal": {
"command": "uv",
"args": [
"run",
"--no-sync",
"python",
"-m",
"roboco.mcp.optimal_server",
agent_uuid,
],
"env": mcp_env,
},
}
# Docs server — documentation file management. Registered only for
# roles that touch panel docs; handlers still enforce per-role
# access so the surface is fail-closed.
docs_roles = (
"documenter",
"cell_pm",
"main_pm",
"product_owner",
"head_marketing",
)
if agent_role in docs_roles:
mcp_servers["roboco-docs"] = {
"command": "uv",
"args": [
"run",
"--no-sync",
"python",
"-m",
"roboco.mcp.docs_server",
agent_uuid,
],
"env": mcp_env,
}
# Web research — external search/fetch for Board + PM roles. The
# provider key stays server-side (the route holds it); the agent only
# ever talks to the backend, so the container needs no external egress.
research_roles = (
"cell_pm",
"main_pm",
"product_owner",
"head_marketing",
)
if settings.research_enabled and agent_role in research_roles:
mcp_servers["roboco-search"] = {
"command": "uv",
"args": [
"run",
"--no-sync",
"python",
"-m",
"roboco.mcp.search_server",
agent_uuid,
],
"env": mcp_env,
}
config: dict[str, Any] = {"mcpServers": mcp_servers}
# Write to shared config directory (mounted in both orchestrator and agents)
# When running in container: /app/mcp-configs -> host's ./data/mcp-configs
# When running on host: use temp directory
if DATA_HOST_PATH:
# Running in container - use shared mounted directory
config_dir = Path("/app/mcp-configs")
config_dir.mkdir(parents=True, exist_ok=True)
else:
# Running on host - use temp directory
config_dir = Path(tempfile.gettempdir())
config_path = config_dir / f"roboco-mcp-{agent_id}.json"
config_path.write_text(json.dumps(config, indent=2))
return config_path
def _generate_composed_prompt(
self, agent_id: str, ambient: str | None = None
) -> Path:
"""Generate composed system prompt for an agent.
Uses the layered prompt composition system:
base.md + roles/{role}.md + teams/{team}.md + identities/{agent}.md
plus an optional ``ambient`` layer (the project's architectural
standard, resolved by the async spawn path).
Returns:
Path to the generated prompt file
"""
# Get role and team from canonical config
role_str = get_agent_role(agent_id)
team_str = get_agent_team(agent_id)
# Convert to enums
role_enum = AgentRole(role_str) if role_str else None
team_enum = Team(team_str) if team_str else None
if not role_enum:
raise ValueError(f"Unknown role for agent: {agent_id}")
# Compose the prompt from layers
prompt_content = compose_prompt(role_enum, team_enum, agent_id, ambient=ambient)
# Determine output directory
if PROJECT_HOST_PATH:
# Running in container - use shared directory that maps to host
config_dir = Path("/app/prompts-generated")
config_dir.mkdir(parents=True, exist_ok=True)
else:
# Running directly on host
config_dir = Path(tempfile.gettempdir()) / "roboco-prompts"
config_dir.mkdir(parents=True, exist_ok=True)
# Write to file
prompt_path = config_dir / f"{agent_id}-prompt.md"
prompt_path.write_text(prompt_content)
logger.debug(
"Generated composed prompt",
agent_id=agent_id,
role=role_str,
team=team_str,
path=str(prompt_path),
size=len(prompt_content),
)
return prompt_path
async def _resolve_conventions_ambient(
self,
project_slug: str | None,
task_id: str | None = None,
product_id: str | None = None,
project_ids: list[str] | None = None,
) -> str | None:
"""Resolve the architectural-standard ambient block for the spawn.
Covers a delivery role's single project (via ``project_slug``), a PO /
Intake working a product (per-cell projects resolved from the task's
``product_id`` or a directly-supplied ``product_id``), AND a MegaTask
intake's explicit ``project_ids`` scope. Best-effort + flag-gated:
returns None (no ambient layer) when the subsystem is off, no project is
in scope, or anything fails — a prompt compose must never be blocked by
conventions resolution.
"""
from roboco.config import settings
if not settings.conventions_enabled:
return None
try:
from roboco.agents.factories._base import conventions_ambient_layer
from roboco.db.base import get_session_factory
factory = get_session_factory()
async with factory() as db:
projects = await self._resolve_ambient_projects(
db,
project_slug=project_slug,
task_id=task_id,
product_id=product_id,
project_ids=project_ids,
)
return await conventions_ambient_layer(db, projects)
except Exception as exc:
logger.warning(
"Conventions ambient resolution failed (non-fatal)",
project_slug=project_slug,
error=str(exc),
)
return None
async def _resolve_ambient_projects(
self,
db: Any,
*,
project_slug: str | None,
task_id: str | None,
product_id: str | None,
project_ids: list[str] | None = None,
) -> list[Any]:
"""The in-scope projects for the ambient block: single repo, product,
an explicit MegaTask ``project_ids`` set, or an ad-hoc cell map."""
if project_ids:
return await self._projects_by_ids(db, project_ids)
if product_id is not None:
return await self._ambient_product_projects(db, product_id)
if task_id is not None:
projects = await self._ambient_projects_for_task(db, task_id)
if projects:
return projects
if project_slug:
from roboco.services.project import get_project_service
project = await get_project_service(db).get_by_slug(project_slug)
return [project] if project is not None else []
return []
@staticmethod
async def _projects_by_ids(db: Any, project_ids: list[str]) -> list[Any]:
"""Resolve an explicit id list to project rows, in order, skipping any
that don't resolve — best-effort ambient resolution, not the hard
clone-scope resolver (which fails loud on a missing id)."""
from uuid import UUID
from roboco.services.project import get_project_service
project_svc = get_project_service(db)
out = []
for pid in project_ids:
p = await project_svc.get(UUID(pid))
if p is not None:
out.append(p)
return out
@staticmethod
async def _ambient_projects_for_task(db: Any, task_id: str) -> list[Any]:
"""The in-scope projects for a task's ambient block, from its product OR
its ad-hoc ``cell_projects`` map. Empty for a plain project task (the
project_slug branch handles those) or a not-yet-mapped coordination root.
"""
from uuid import UUID
from roboco.services.project import get_project_service
from roboco.services.task import get_task_service
task = await get_task_service(db).get(UUID(task_id))
if task is None:
return []
project_service = get_project_service(db)
if task.product_id is not None:
from roboco.services.product import get_product_service
ids = await get_product_service(db).distinct_project_ids(
UUID(str(task.product_id))
)
resolved = [await project_service.get(pid) for pid in ids]
return [p for p in resolved if p is not None]
# Ad-hoc per-cell map: resolve the distinct projects the map spans (de-dupe
# by project_id — a monorepo mapped across cells yields one project).
distinct_ids: dict[Any, None] = {}
for mapping in sorted(task.cell_projects, key=lambda m: m.team.value):
distinct_ids.setdefault(UUID(str(mapping.project_id)), None)
resolved = [await project_service.get(pid) for pid in distinct_ids]
return [p for p in resolved if p is not None]
@staticmethod
async def _ambient_product_projects(db: Any, product_id: str) -> list[Any]:
from uuid import UUID
from roboco.services.product import get_product_service
from roboco.services.project import get_project_service
project_service = get_project_service(db)
ids = await get_product_service(db).distinct_project_ids(UUID(product_id))
resolved = [await project_service.get(pid) for pid in ids]
return [p for p in resolved if p is not None]
async def _resolve_history_digest_ambient(
self,
project_slug: str | None,
product_id: str | None = None,
project_ids: list[str] | None = None,
) -> str | None:
"""Resolve the prompter's task-history-digest ambient block for this scope.
Covers all three intake scopes including ``project_ids`` (a MegaTask) —
the digest is meant to span every project the intake agent is reading.
Best-effort: returns None on any failure or empty scope so history
resolution can never block a spawn.
"""
try:
from roboco.db.base import get_session_factory
from roboco.services.prompter import history_digest_layer
factory = get_session_factory()
async with factory() as db:
projects = await self._resolve_history_digest_projects(
db,
project_slug=project_slug,
product_id=product_id,
project_ids=project_ids,
)
return await history_digest_layer(db, projects)
except Exception as exc:
logger.warning(
"History digest ambient resolution failed (non-fatal)",
project_slug=project_slug,
error=str(exc),
)
return None
@staticmethod
async def _resolve_history_digest_projects(
db: Any,
*,
project_slug: str | None,
product_id: str | None,
project_ids: list[str] | None,
) -> list[Any]:
"""The in-scope ProjectTable rows for the history digest — single repo,
product (all cell projects), or an explicit MegaTask project_ids set."""
if project_ids:
return await AgentOrchestrator._projects_by_ids(db, project_ids)
if product_id is not None:
return await AgentOrchestrator._ambient_product_projects(db, product_id)
if project_slug:
from roboco.services.project import get_project_service
project = await get_project_service(db).get_by_slug(project_slug)
return [project] if project is not None else []
return []
async def _resolve_intake_ambient(
self,
project_slug: str | None,
*,
product_id: str | None,
project_ids: list[str] | None,
) -> str | None:
"""The intake spawn's full ambient block: conventions + history digest,
joined with ``compose_prompt``'s own layer separator."""
conventions_ambient = await self._resolve_conventions_ambient(
project_slug, product_id=product_id, project_ids=project_ids
)
history_ambient = await self._resolve_history_digest_ambient(
project_slug, product_id=product_id, project_ids=project_ids
)
return (
"\n\n---\n\n".join(
part for part in (conventions_ambient, history_ambient) if part
)
or None
)
async def _readiness_gate(self, agent_id: str, task_id: str | None) -> str | None:
"""Return a reason string if the spawn must be refused, else None.
Checks run only when a task is being spawned for. No-task spawns
(idle PM bootstrap, etc.) are always ready. On any refusal that
represents a persistent problem we auto-block the task so the
dispatcher stops retrying — the PM sees the block notification.
"""
if not task_id:
return None
try:
async with httpx.AsyncClient(
timeout=5.0, headers=_system_api_headers()
) as client:
task_or_reason = await self._readiness_fetch_task(client, task_id)
if isinstance(task_or_reason, str):
return task_or_reason
task = task_or_reason
# Universal dependency gate: refuse to spawn an agent of ANY role
# onto a task whose cross-task dependencies are not yet terminal.
# This check previously lived only on the dev dispatch path, so
# cell-PM, Main-PM and board agents were spawned onto
# dependency-blocked tasks and flailed unblock / escalate / notify
# against an unfinished upstream. Auto-block so the task leaves the
# pending pool (no per-tick spawn-refusal that would starve
# siblings); `_unblock_dependents` revives it the moment the
# upstream reaches a terminal state.
if dep_reason := await self._check_dependencies_terminal(client, task):
return await self._readiness_block(client, task_id, dep_reason)
persistent = self._readiness_check_task(agent_id, task)
# Skip the git-token gate for coordination tasks — they have no
# project of their own, so there's no token to require.
if persistent is None and not _is_coordination_task(task):
project_slug = _read_project_slug(task)
persistent = await self._readiness_check_git_token(project_slug)
if persistent is not None:
return await self._readiness_block(client, task_id, persistent)
except httpx.HTTPError as e:
# Transient — retry on next dispatch without auto-blocking.
return f"readiness check HTTP error: {e}"
return None
async def _readiness_fetch_task(
self, client: httpx.AsyncClient, task_id: str
) -> dict[str, Any] | str:
"""Fetch the task or return a reason string.
404 → "task not found" (caller should auto-block).
Other non-200s → transient; caller returns the reason verbatim
without auto-blocking so the dispatcher can retry next tick.
"""
resp = await client.get(f"{self._api_url}/tasks/{task_id}")
if resp.status_code == http_status.HTTP_404_NOT_FOUND:
await self._readiness_block(client, task_id, "task not found")
return "task not found"
if resp.status_code != http_status.HTTP_200_OK:
return f"task-fetch returned {resp.status_code}"
task = resp.json()
return task if isinstance(task, dict) else "task payload not an object"
@staticmethod
@staticmethod
def _readiness_check_acceptance_criteria(task: dict[str, Any]) -> str | None:
"""Return blocker reason for missing acceptance criteria, else None."""
criteria = task.get("acceptance_criteria") or []
if isinstance(criteria, str):
criteria = [criteria] if criteria.strip() else []
if not criteria:
return "missing acceptance_criteria"
return None
@staticmethod
def _readiness_check_role_for_status(
agent_id: str,
role: str,
status: str,
*,
is_coordination: bool = False,
owner_is_pm: bool = False,
) -> str | None:
"""Verify agent role matches the role expected for the task status.
Handoff states are role-specific. Dev-owned states (in_progress,
verifying, needs_revision, paused, blocked) are restricted to
developer/documenter to defang the bug where QA got
respawned on a `needs_revision` task via the crash-restart path
and immediately hit ``role 'qa' may not claim from status
'needs_revision'`` at the gateway. The exception is a PM-OWNED revision:
a coordination root (no code; product fan-out owned by a PM, a CEO-reject
returning to its PM) AND a gate-failed assembled PR (the PR-review gate's
``pr_fail`` sends a cell->root / root->master PR back to needs_revision,
still owned by the cell/main PM). In both the owner is a PM, so the
dev-owned states also accept the PM roles when ``owner_is_pm`` — matching
``_dispatch_revision_coordination_roots``, which re-spawns exactly those.
A pure widening; nothing currently allowed is blocked, and QA stays out.
"""
role_mismatch: dict[str, str | set[str]] = {
"awaiting_qa": "qa",
"awaiting_documentation": "documenter",
"awaiting_pr_review": "pr_reviewer",
"awaiting_pm_review": {"cell_pm", "main_pm"},
"awaiting_ceo_approval": "ceo",
# Dev-owned states — only developer/documenter may claim or
# resume work here. PMs / QA spawning on these is a misroute.
"needs_revision": {"developer", "documenter"},
"verifying": {"developer", "documenter"},
}
required = role_mismatch.get(status)
if required is None:
return None
if status in ("needs_revision", "verifying") and (
is_coordination or owner_is_pm
):
required = set(required) | {"cell_pm", "main_pm"}
ok = role in required if isinstance(required, set) else role == required
if ok:
return None
return (
f"state={status} requires role in {required!r} "
f"but agent {agent_id} is {role!r}"
)
def _readiness_check_task(self, agent_id: str, task: dict[str, Any]) -> str | None:
"""Return a persistent blocker reason on the task itself, else None."""
status = task.get("status", "")
role = get_agent_role(agent_id) or ""
if reason := self._readiness_check_acceptance_criteria(task):
return reason
# A coordination task (product, no repo of its own) does no git: skip the
# project-slug and branch-name gates that only apply to code tasks.
if not _is_coordination_task(task):
if not _read_project_slug(task):
return "task has no project"
# Branch is auto-created at claim, so only states at/after claim are
# expected to own one. _branch_is_expected centralizes this gate so
# the readiness and stuck-detection paths agree.
if _branch_is_expected(task) and not task.get("branch_name"):
return f"state={status} but branch_name is unset"
owner = task.get("assigned_to") or task.get("claimed_by")
owner_role = get_agent_role(self._resolve_agent_slug(owner)) if owner else None
return self._readiness_check_role_for_status(
agent_id,
role,
status,
is_coordination=_is_coordination_task(task),
owner_is_pm=owner_role in ("cell_pm", "main_pm"),
)
@staticmethod
async def _readiness_check_git_token(project_slug: str | None) -> str | None:
"""Ensure the project has a decryptable git token, else blocker reason."""
if not project_slug:
return "task has no project"
from roboco.db.base import get_session_factory
from roboco.services.project import get_project_service
session_factory = get_session_factory()
async with session_factory() as db:
project_svc = get_project_service(db)
try:
token = await project_svc.get_decrypted_token_by_slug(project_slug)
except Exception as e:
return f"project '{project_slug}' git-token decrypt failed: {e}"
if not token:
return f"project '{project_slug}' has no git token configured"
return None
async def _readiness_block(
self, client: httpx.AsyncClient, task_id: str, reason: str
) -> str:
"""Auto-block the task and return the human-readable reason."""
await self._auto_block_task(client, task_id, f"readiness: {reason}")
return reason
async def _resolve_agent_route(self, agent_id: str) -> "AgentRoute":
"""Resolve (provider, model) for `agent_id` via `ModelRoutingService`.
Errors are contained: any DB/session failure degrades to a legacy
Anthropic-default AgentRoute so spawn never stalls on routing.
"""
from roboco.db.base import get_session_factory
from roboco.models.base import ModelProvider
from roboco.models.runtime import MODEL_MAP
from roboco.services.llm import (
AgentRoute,
get_model_routing_service,
)
try:
factory = get_session_factory()
async with factory() as db:
router = get_model_routing_service(db)
return await router.resolve_for_agent(agent_id)
except Exception as e: # pragma: no cover
role = get_agent_role(agent_id) or ""
short = ROLE_MODEL_MAP.get(role, "sonnet")
logger.warning(
"Model routing resolve failed; using legacy Anthropic path",
agent_id=agent_id,
error=str(e),
)
return AgentRoute(
provider_id=None,
provider_type=ModelProvider.ANTHROPIC,
base_url=None,
auth_token=None,
model_name=MODEL_MAP.get(short, short),
)
_TOOL_LOAD_CACHE: ClassVar[dict[str, str]] = {}
# Per-role built-in tools, enumerated in the briefing so the agent
# knows exactly what it has. These are pre-loaded at spawn via the
# Claude Code `--tools` flag and gated only by the per-role
# permission rules — NOT by ToolSearch (MCP-only; never gates
# built-ins). Mirrors the system-prompt layer's _ROLE_BUILTIN_TOOLS
# in roboco/agents/factories/_base.py — kept in sync because the
# briefing and the system prompt are independent code paths.
_COMMON_BUILTIN_TOOLS: ClassVar[tuple[str, ...]] = (
"Read",
"Bash",
"Grep",
"Glob",
"TodoWrite",
)
_ROLE_BUILTIN_TOOLS: ClassVar[dict[str, tuple[str, ...]]] = {
"developer": (*_COMMON_BUILTIN_TOOLS, "Edit", "Write"),
"documenter": (*_COMMON_BUILTIN_TOOLS, "Edit", "Write"),
"qa": _COMMON_BUILTIN_TOOLS,
"main_pm": _COMMON_BUILTIN_TOOLS,
"cell_pm": _COMMON_BUILTIN_TOOLS,
"product_owner": _COMMON_BUILTIN_TOOLS,
"head_marketing": _COMMON_BUILTIN_TOOLS,
"auditor": _COMMON_BUILTIN_TOOLS,
"pr_reviewer": _COMMON_BUILTIN_TOOLS,
}
def _build_tool_load_block(self, role: str) -> str:
"""Briefing block affirming the role's built-in tools are ready.
Built-in tools are pre-loaded at spawn via the Claude Code
`--tools` flag and gated only by the per-role permission rules.
ToolSearch is MCP-only and never gates built-ins — an earlier
revision instructed agents to "run ToolSearch to activate
Edit/Write", which was false (ToolSearch is not even callable
here), so weak models chased a nonexistent tool and fell back to
destructive shell file-writes. This states the tools are live and
steers away from that failure. Cached per role.
"""
if role in self._TOOL_LOAD_CACHE:
return self._TOOL_LOAD_CACHE[role]
tools = self._ROLE_BUILTIN_TOOLS.get(role)
if not tools:
block = ""
else:
tool_list = ", ".join(tools)
edit_line = (
"Make file changes with Edit/Write — never rewrite a "
"whole file via shell redirection (>, heredoc, tee); "
"that destroys content and is unnecessary.\n"
if "Edit" in tools
else "You read and review; you do not author files.\n"
)
block = (
"## Your tools are ready\n"
"\n"
f"Loaded and available now: {tool_list}. Use them "
"directly. Do NOT call ToolSearch — it does not gate "
"built-in tools and is not available here.\n"
f"{edit_line}"
"\n"
)
self._TOOL_LOAD_CACHE[role] = block
return block
@staticmethod
def _format_task_briefing_block(task_id: str, task: dict[str, Any]) -> str:
"""Build the ``## Current task`` markdown block from a fetched task."""
criteria_list = task.get("acceptance_criteria") or []
if isinstance(criteria_list, str):
criteria_list = [criteria_list]
criteria = (
"\n".join(f"- {c}" for c in criteria_list)
if criteria_list
else "- (none listed — ask PM before proceeding)"
)
branch = task.get("branch_name") or "(to be created)"
project_slug = task.get("project_slug") or "(unset — ask PM)"
return (
"\n## Current task\n"
f"- **ID:** `{task.get('id', task_id)}`\n"
f"- **Title:** {task.get('title', '(untitled)')}\n"
f"- **Status:** {task.get('status', 'unknown')}\n"
f"- **Type:** {task.get('task_type', 'unknown')}\n"
f"- **Project slug:** `{project_slug}` "
"(pass this as `project_slug=` on every git/task tool)\n"
f"- **Branch:** `{branch}`\n"
"\n### Acceptance criteria\n"
f"{criteria}\n"
)
async def _fetch_task_for_briefing(
self, agent_id: str, task_id: str
) -> dict[str, Any] | None:
"""Best-effort GET /tasks/{id}; returns task dict or None on failure."""
try:
async with httpx.AsyncClient(
timeout=5.0, headers=_system_api_headers()
) as client:
resp = await client.get(f"{self._api_url}/tasks/{task_id}")
if resp.status_code == http_status.HTTP_200_OK:
payload: dict[str, Any] = resp.json()
return payload
except Exception as e:
logger.debug(
"Briefing task-fetch failed — falling back to role-only",
agent_id=agent_id,
task_id=task_id,
error=str(e),
)
return None
async def _write_agent_briefing(
self,
agent_id: str,
task_id: str | None,
workspace_path: str,
) -> Path | None:
"""Write a compact task briefing to be read by SessionStart hook.
The briefing saves the agent from burning its first 2-3 tool calls on
`give_me_work` (whose Envelope already carries the task details). If
`task_id` is known we fetch
the task and include title, status, branch, and acceptance criteria.
On fetch failure we still emit the role-level part (role, escalation
target, terminal tools, workspace path) — strictly better than nothing.
"""
role = get_agent_role(agent_id) or "agent"
team = get_agent_team(agent_id) or "-"
escalate_to = get_escalation_target(agent_id) or "main-pm"
tool_load_block = self._build_tool_load_block(role)
task_block = ""
if task_id:
task = await self._fetch_task_for_briefing(agent_id, task_id)
if task is not None:
task_block = self._format_task_briefing_block(task_id, task)
content = (
f"# Session briefing — {agent_id}\n"
"\n"
f"{tool_load_block}"
"## You are\n"
f"- **Agent:** `{agent_id}`\n"
f"- **Role:** {role}\n"
f"- **Team:** {team}\n"
f"- **Escalate to:** `{escalate_to}`\n"
f"- **Workspace:** `{workspace_path}`\n"
f"{task_block}"
"\n## Terminal tools (how to exit cleanly)\n"
"- `i_am_idle()` — no work remaining (every role)\n"
"- `i_am_blocked(task_id, reason, ...)` — stuck (developer)\n"
"- `unclaim(task_id)` — release a claim back to the pool\n"
"- Role handoffs:\n"
" - developer → `i_am_done(task_id, notes)` (submit for QA)\n"
" - qa → `pass(task_id, notes)` / `fail(task_id, issues)`\n"
" - documenter → `i_documented(task_id, notes, files)`\n"
" - cell_pm → `complete(task_id, notes)` / `submit_up(...)`"
" / `escalate_up(...)`\n"
" - main_pm → `complete(...)` / `escalate_to_ceo(...)`\n"
"\n"
"A Stop without a terminal tool will be rejected; a second Stop\n"
"auto-substitutes the task so it can be picked up elsewhere.\n"
"\n"
"## Budget\n"
f"Soft-warn at {settings.agent_tool_call_warn} tool calls, "
f"hard cap at {settings.agent_tool_call_halt}. Loops — same "
f"tool+args {settings.agent_loop_threshold}x within "
f"{settings.agent_loop_window} calls — are flagged; stop and "
"escalate instead of retrying.\n"
)
if PROJECT_HOST_PATH:
briefings_dir = Path("/app/briefings")
else:
briefings_dir = Path(tempfile.gettempdir()) / "roboco-briefings"
briefings_dir.mkdir(parents=True, exist_ok=True)
path = briefings_dir / f"{agent_id}.md"
path.write_text(content)
logger.debug(
"Wrote agent briefing",
agent_id=agent_id,
path=str(path),
has_task=bool(task_block),
)
return path
# Slug -> team string for ROUTING purposes. Derived from
# foundation.AGENTS so adding/renaming an agent edits exactly one
# file (foundation/identity.py). The dispatcher relies on this for
# task assignment routing categories.
_AGENT_TEAM_MAP: ClassVar[dict[str, str]] = {
slug: row.team.value for slug, row in _foundation.AGENTS.items()
}
def _get_agent_team(self, agent_id: str) -> str | None:
"""Get team from agent_id. Returns None for unknown slugs."""
try:
return _foundation.team_for_slug(agent_id).value
except KeyError:
return None
def _resolve_agent_slug(self, agent_id_or_uuid: str) -> str:
"""Resolve agent UUID to slug. Returns input if already a slug."""
# Check if it's a known UUID and convert to slug
if agent_id_or_uuid in UUID_TO_SLUG:
return UUID_TO_SLUG[agent_id_or_uuid]
# Already a slug or unknown UUID
return agent_id_or_uuid
def _mark_task_handled(self, task_id: str | None) -> None:
"""Record that `task_id` was acted on earlier in this dispatch tick."""
if task_id:
self._tick_handled_tasks.add(task_id)
def _is_task_handled_this_tick(self, task_id: str | None) -> bool:
"""True if a prior dispatcher already handled this task this tick."""
return bool(task_id and task_id in self._tick_handled_tasks)
_NOTIFICATION_COOLDOWN_PRUNE_AT = 512
def _notification_spawn_cooled(
self, agent_slug: str, notification_id: str | None
) -> bool:
"""True when this (agent, notification) spawned within the cooldown.
Returns False — and stamps the pair — when a spawn is allowed. A
notification with no id is never damped (fail-open: better one extra
spawn than a silently dropped escalation).
"""
if not notification_id:
return False
# Lazy init keeps the damper working on partially-constructed
# instances (tests build the orchestrator via __new__).
store: dict[tuple[str, str], float] = self.__dict__.setdefault(
"_notification_spawn_at", {}
)
key = (agent_slug, str(notification_id))
now = time.monotonic()
cooldown = settings.notification_spawn_cooldown_seconds
last = store.get(key)
if last is not None and (now - last) < cooldown:
return True
store[key] = now
if len(store) > self._NOTIFICATION_COOLDOWN_PRUNE_AT:
cutoff = now - cooldown
self._notification_spawn_at = {
k: v for k, v in store.items() if v >= cutoff
}
return False
def _is_parallel_phase_claim(
self, task: dict[str, Any], dev_uuid: str | None
) -> bool:
"""True if a `claimed` task is actually in the doc/PR parallel phase.
The `original_developer:` quick_context marker is set pre-QA by
`open_pr`, so it alone cannot distinguish a QA-claimed
awaiting_qa task (wrong) from a doc-claimed awaiting_documentation
task (right). Require the claimant to be a documenter.
"""
if not dev_uuid:
return False
claimed_by = task.get("claimed_by")
if not claimed_by:
return False
claimed_slug = self._resolve_agent_slug(claimed_by)
return bool(claimed_slug) and "doc" in claimed_slug
async def _respawn_dev_for_pr_half(
self, task: dict[str, Any], dev_uuid: str | None
) -> None:
"""Respawn the original developer if they still owe the PR half.
`pr_number` is set by the PR-create handler as soon as GitHub
confirms the PR, even if the status-gated `pr_created` flag never
flips — without that second check we'd respawn the dev forever
after they've already created the PR (the handler refuses to set
pr_created=True when the doc's claim moved status out of
awaiting_documentation).
"""
if not dev_uuid or task.get("pr_created") or task.get("pr_number"):
return
dev_slug = self._resolve_agent_slug(dev_uuid)
if dev_slug and await self._pm_respawn_should_gate(dev_slug, task):
# Respawn circuit breaker — the PR-half respawn loops exactly like
# the doc half when the dev can never finish (progress resets it).
return
if not dev_slug or self._is_agent_active(dev_slug):
return
await self.spawn_agent(
agent_id=dev_slug,
task_id=task["id"],
initial_prompt=self._build_dev_prompt(task),
git_context=self._task_git_context(task),
spawned_by="_respawn_dev_for_pr_half",
)
# =========================================================================
# INTAKE (PROMPTER) LIVE SESSION
#
# The intake agent is not task-driven and is never dispatched. It is a
# persistent Claude-Agent-SDK driver the CEO chats with live (the container
# entrypoint is roboco.agent_sdk.intake_main). One fixed container —
# `intake-1`, the seeded board-adjacent interviewer — serves one live
# session at a time (single CEO; one-session-per-CEO).
#
# This spawn is a DELIBERATELY separate path from spawn_agent: no task, no
# readiness gate, no `claude -p` CLI args (the image ENTRYPOINT is the
# driver), no settings.json/hook mount (the driver owns the receiver on
# port 9000, not the inbox sidecar), and no MCP/gateway surface (the live
# agent reads code with Read/Grep/Glob and talks only to the human).
# =========================================================================
@staticmethod
def _require_one_intake_scope(
project_slug: str | None,
product_id: str | None,
project_ids: list[str] | None,
) -> None:
"""Exactly one intake scope: a single project, a product, or a MegaTask's
explicit project set."""
chosen = sum(1 for scope in (project_slug, product_id, project_ids) if scope)
if chosen != 1:
raise ValueError(
"intake scope requires exactly one of project_slug / product_id"
" / project_ids"
)
async def start_intake_session(
self,
session_id: str,
*,
project_slug: str | None = None,
product_id: str | None = None,
project_ids: list[str] | None = None,
initial_message: str | None = None,
) -> None:
"""Non-blocking start: open the relay now, spawn the container in the bg.
The panel's ``POST /live/start`` returns immediately rather than blocking
on the workspace clone + first-time image build + ``docker run`` (which
can exceed the HTTP timeout — the cause of the "Request timed out" the
panel showed). The panel opens the SSE stream right away; the agent's
first reply arrives once the container is up. A spawn failure is pushed
onto the relay as an ``error`` event and closes the session, so the panel
shows it instead of hanging. Exactly one of ``project_slug`` /
``product_id`` / ``project_ids`` (a MegaTask) must be given.
"""
self._require_one_intake_scope(project_slug, product_id, project_ids)
self._open_intake_relay(session_id)
self._schedule_bg(
self._spawn_intake_container_guarded(
session_id,
project_slug=project_slug,
product_id=product_id,
project_ids=project_ids,
initial_message=initial_message,
)
)
async def spawn_intake_session(
self,
session_id: str,
*,
project_slug: str | None = None,
product_id: str | None = None,
project_ids: list[str] | None = None,
initial_message: str | None = None,
) -> AgentInstance:
"""Spawn the intake container for one live chat, **synchronously**.
Opens the relay then clones + launches the container, awaiting the whole
thing. Prefer ``start_intake_session`` on the request path; this blocking
variant is for direct/internal callers and tests. Exactly one of
``project_slug`` / ``product_id`` / ``project_ids`` (a MegaTask) must be
given.
"""
self._require_one_intake_scope(project_slug, product_id, project_ids)
self._open_intake_relay(session_id)
return await self._spawn_intake_container(
session_id,
project_slug=project_slug,
product_id=product_id,
project_ids=project_ids,
initial_message=initial_message,
)
@staticmethod
def _open_intake_relay(session_id: str) -> None:
"""Register the live relay session so the SSE stream connects immediately."""
from roboco.services.prompter_live import get_live_registry
get_live_registry().open(session_id, INTAKE_AGENT_ID)
async def _spawn_intake_container_guarded(
self,
session_id: str,
*,
project_slug: str | None,
product_id: str | None,
project_ids: list[str] | None = None,
initial_message: str | None,
) -> None:
"""Background container spawn; surface failures on the relay, not silently."""
from roboco.services.prompter_live import get_live_registry
try:
await self._spawn_intake_container(
session_id,
project_slug=project_slug,
product_id=product_id,
project_ids=project_ids,
initial_message=initial_message,
)
except _SpawnAbortedDuringShutdown:
# Shutdown began mid-spawn; the just-started container was already
# removed by the raiser. Close the relay silently — shutdown is not a
# user-facing failure, so no error is pushed to the SSE stream.
get_live_registry().close(session_id)
return
except Exception as exc:
logger.error(
"Intake container spawn failed", session_id=session_id, error=str(exc)
)
registry = get_live_registry()
registry.push(
session_id,
{"kind": "error", "text": f"Couldn't start the intake agent: {exc}"},
)
registry.close(session_id)
async def _spawn_intake_container(
self,
session_id: str,
*,
project_slug: str | None,
product_id: str | None,
project_ids: list[str] | None = None,
initial_message: str | None,
) -> AgentInstance:
"""Clone the scope, launch the SDK-driver container, track the instance.
The relay must already be open (``_open_intake_relay``). Heavy + slow
(clone + first-time image build + docker run) — keep it off the request
path via ``start_intake_session``.
Serialized by ``_intake_spawn_lock``: the intake agent id is a single
fixed id, so two concurrent starts would race on the container name and
the ``_instances`` write, orphaning a container + relay. The lock makes a
concurrent start wait for the in-flight one to finish (reap + register)
before it begins its own reap-prior check.
"""
async with self._intake_spawn_lock:
# Single live session: reap any prior intake container before spawning.
if INTAKE_AGENT_ID in self._instances:
await self.stop_agent(
INTAKE_AGENT_ID,
graceful=False,
stop_reason="intake_respawn_guard",
)
from roboco.models.base import ModelProvider
cwd, cloned = await self._clone_intake_scope(
project_slug, product_id, project_ids
)
ambient = await self._resolve_intake_ambient(
project_slug, product_id=product_id, project_ids=project_ids
)
prompt_path = self._generate_composed_prompt(
INTAKE_AGENT_ID, ambient=ambient
)
route = await self._resolve_agent_route(INTAKE_AGENT_ID)
cli_model = _resolve_agent_cli_model(
route.provider_type.value, route.model_name
)
api_url = (
"http://roboco-orchestrator:8000"
if PROJECT_HOST_PATH
else f"http://127.0.0.1:{settings.port}"
)
# GROK runs the interactive driver on its own grok-CLI prompter image;
# every other provider uses the Claude SDK-driver prompter image.
is_grok = route.provider_type == ModelProvider.GROK
image = GROK_PROMPTER_IMAGE if is_grok else get_agent_image(INTAKE_AGENT_ID)
if is_grok:
await self._ensure_grok_interactive_image(image)
self._ensure_grok_usage_dir(INTAKE_AGENT_ID)
else:
await self._ensure_agent_image(INTAKE_AGENT_ID)
container_name = f"roboco-agent-{INTAKE_AGENT_ID}"
await self._remove_container(
container_name, stop_reason="pre_spawn_stale_clear"
)
cmd = self._build_intake_run_cmd(
_IntakeRunSpec(
container_name=container_name,
image=image,
hosts=self._resolve_intake_host_paths(),
session_id=session_id,
cwd=cwd,
cli_model=cli_model,
api_url=api_url,
provider_base_url=route.base_url,
provider_auth_token=route.auth_token,
provider_type=route.provider_type.value,
model=route.model_name,
)
)
container_id = await self._run_container_cmd(cmd)
# Shutdown may have begun while this (non-blocking) spawn was in flight
# — the bg coroutine runs concurrently with stop(). If so, remove the
# just-started container and abort WITHOUT registering: stop()'s
# _instances iteration has already run (or is running), so a registration
# now would land a live container nothing tears down (the orphan). The
# stop() drain awaits this coroutine, so the abort surfaces cleanly.
if not self._running:
await self._remove_container(
container_name, stop_reason="spawn_aborted_shutdown"
)
raise _SpawnAbortedDuringShutdown(INTAKE_AGENT_ID)
config = AgentConfig(
agent_id=INTAKE_AGENT_ID,
blueprint_path=prompt_path,
model=route.model_name,
git_context=None,
provider_type=route.provider_type.value,
)
instance = AgentInstance(
agent_id=INTAKE_AGENT_ID,
state=AgentState.ACTIVE,
config=config,
current_task_id=None,
)
instance.container_id = container_id
instance.started_at = datetime.now(UTC)
instance.last_activity = datetime.now(UTC)
self._instances[INTAKE_AGENT_ID] = instance
# Record a usage session (task_id=None) and pin its id on the instance
# so the reap finalizer can look up token usage — without this an
# interactive session finalizes at 0 tokens / $0 (the GROK path reads the
# captured usage.json; the Claude path reads the transcript). Mirrors
# _launch_spawn.
usage_session_id = await self._record_spawn_session(config, None)
if usage_session_id is not None:
instance.usage_session_id = usage_session_id
# The relay was already opened on the request path
# (start_intake_session / spawn_intake_session) BEFORE the panel
# connected its SSE stream. Do NOT re-open here: a second open would
# swap in a fresh queue and orphan that already-connected stream (the
# agent's replies would push to the new queue while the browser keeps
# reading the old one). open() is idempotent now as a guard, but the
# redundant call is gone regardless.
logger.info(
"Intake session spawned",
session_id=session_id,
container_id=container_id[:12],
cwd=cwd,
repos=len(cloned),
)
self._fire_audit(
event_type="agent.spawned",
agent_slug=INTAKE_AGENT_ID,
details={"session_id": session_id, "cwd": cwd, "repos": cloned},
)
if initial_message:
self._schedule_intake_first_message(session_id, initial_message)
return instance
async def reap_intake_session(self, session_id: str) -> None:
"""End a live chat: close the relay stream and stop the container."""
from roboco.services.prompter_live import get_live_registry
get_live_registry().close(session_id)
await self.stop_agent(
INTAKE_AGENT_ID, graceful=True, stop_reason="intake_session_reaped"
)
logger.info("Intake session reaped", session_id=session_id)
# ------------------------------------------------------------------ #
# Secretary live session (mirrors intake; no scope clone; auth token)
# ------------------------------------------------------------------ #
async def start_secretary_session(
self, session_id: str, *, initial_message: str | None = None
) -> None:
"""Non-blocking start: open the relay now, spawn the container in the bg."""
from roboco.services.prompter_live import get_live_registry
get_live_registry().open(session_id, SECRETARY_AGENT_ID)
self._schedule_bg(
self._spawn_secretary_container_guarded(
session_id, initial_message=initial_message
)
)
async def spawn_secretary_session(
self, session_id: str, *, initial_message: str | None = None
) -> AgentInstance:
"""Spawn the Secretary container synchronously (internal callers/tests)."""
from roboco.services.prompter_live import get_live_registry
get_live_registry().open(session_id, SECRETARY_AGENT_ID)
return await self._spawn_secretary_container(
session_id, initial_message=initial_message
)
async def _spawn_secretary_container_guarded(
self, session_id: str, *, initial_message: str | None
) -> None:
"""Background spawn; surface failures on the relay, not silently."""
from roboco.services.prompter_live import get_live_registry
try:
await self._spawn_secretary_container(
session_id, initial_message=initial_message
)
except _SpawnAbortedDuringShutdown:
# Shutdown began mid-spawn; the just-started container was already
# removed by the raiser. Close the relay silently — shutdown is not
# a user-facing failure, so no error is pushed to the SSE stream.
get_live_registry().close(session_id)
return
except Exception as exc:
logger.error(
"Secretary container spawn failed",
session_id=session_id,
error=str(exc),
)
registry = get_live_registry()
registry.push(
session_id,
{"kind": "error", "text": f"Couldn't start the Secretary: {exc}"},
)
registry.close(session_id)
async def _spawn_secretary_container(
self, session_id: str, *, initial_message: str | None
) -> AgentInstance:
"""Launch the Secretary SDK-driver container and track the instance.
Unlike intake there is no workspace scope to clone — the Secretary reads
company state through the API, so its cwd is the baked ``/app`` tree. It
gets an HMAC agent token so its directive tools authenticate as the
Secretary role.
Serialized by ``_secretary_spawn_lock`` for the same reason intake is
serialized by ``_intake_spawn_lock``: a single fixed agent id, so two
concurrent starts would race on the container name and the ``_instances``
write. See ``_spawn_intake_container`` for the deadlock-ordering note.
"""
async with self._secretary_spawn_lock:
from roboco.agents_config import issue_agent_token
from roboco.foundation.identity import AGENTS
from roboco.models.base import ModelProvider
if SECRETARY_AGENT_ID in self._instances:
await self.stop_agent(
SECRETARY_AGENT_ID,
graceful=False,
stop_reason="secretary_respawn_guard",
)
prompt_path = self._generate_composed_prompt(SECRETARY_AGENT_ID)
route = await self._resolve_agent_route(SECRETARY_AGENT_ID)
cli_model = _resolve_agent_cli_model(
route.provider_type.value, route.model_name
)
api_url = (
"http://roboco-orchestrator:8000"
if PROJECT_HOST_PATH
else f"http://127.0.0.1:{settings.port}"
)
is_grok = route.provider_type == ModelProvider.GROK
image = (
GROK_SECRETARY_IMAGE if is_grok else get_agent_image(SECRETARY_AGENT_ID)
)
if is_grok:
await self._ensure_grok_interactive_image(image)
self._ensure_grok_usage_dir(SECRETARY_AGENT_ID)
else:
await self._ensure_agent_image(SECRETARY_AGENT_ID)
container_name = f"roboco-agent-{SECRETARY_AGENT_ID}"
await self._remove_container(
container_name, stop_reason="pre_spawn_stale_clear"
)
agent_uuid = str(AGENTS[SECRETARY_AGENT_ID].uuid)
cmd = self._build_secretary_run_cmd(
_SecretaryRunSpec(
container_name=container_name,
image=image,
hosts=self._resolve_secretary_host_paths(),
session_id=session_id,
cwd="/app",
cli_model=cli_model,
api_url=api_url,
agent_uuid=agent_uuid,
agent_token=issue_agent_token(
agent_uuid,
"secretary",
get_agent_team(SECRETARY_AGENT_ID) or "",
),
provider_base_url=route.base_url,
provider_auth_token=route.auth_token,
provider_type=route.provider_type.value,
model=route.model_name,
)
)
container_id = await self._run_container_cmd(cmd)
# Shutdown may have begun while this (non-blocking) spawn was in flight
# — see the matching guard in _spawn_intake_container. Remove the
# just-started container and abort WITHOUT registering, so it isn't
# orphaned by a stop() that has already iterated _instances.
if not self._running:
await self._remove_container(
container_name, stop_reason="spawn_aborted_shutdown"
)
raise _SpawnAbortedDuringShutdown(SECRETARY_AGENT_ID)
config = AgentConfig(
agent_id=SECRETARY_AGENT_ID,
blueprint_path=prompt_path,
model=route.model_name,
git_context=None,
provider_type=route.provider_type.value,
)
instance = AgentInstance(
agent_id=SECRETARY_AGENT_ID,
state=AgentState.ACTIVE,
config=config,
current_task_id=None,
)
instance.container_id = container_id
instance.started_at = datetime.now(UTC)
instance.last_activity = datetime.now(UTC)
self._instances[SECRETARY_AGENT_ID] = instance
# Pin a usage session id so the reap finalizer can attribute token
# usage (else $0); see the matching note in _spawn_intake_container.
usage_session_id = await self._record_spawn_session(config, None)
if usage_session_id is not None:
instance.usage_session_id = usage_session_id
logger.info(
"Secretary session spawned",
session_id=session_id,
container_id=container_id[:12],
)
self._fire_audit(
event_type="agent.spawned",
agent_slug=SECRETARY_AGENT_ID,
details={"session_id": session_id},
)
if initial_message:
self._schedule_intake_first_message(session_id, initial_message)
return instance
async def reap_secretary_session(self, session_id: str) -> None:
"""End a live Secretary chat: close the relay and stop the container."""
from roboco.services.prompter_live import get_live_registry
get_live_registry().close(session_id)
await self.stop_agent(
SECRETARY_AGENT_ID, graceful=True, stop_reason="secretary_session_reaped"
)
logger.info("Secretary session reaped", session_id=session_id)
async def _reap_idle_interactive_sessions(self) -> None:
"""Retire live intake/secretary chats idle past the configured threshold.
An abandoned chat (the human closed the tab without confirming or
stopping) otherwise leaks its container until the orchestrator restarts.
Idle is measured by time-since-last-turn (push/deliver), NOT connection
state, so an active or page-reloaded chat that keeps exchanging turns is
never reaped; board-review-parked sessions are exempt. Provider-agnostic
(Claude + Grok interactive). Disabled when the threshold is 0.
"""
from roboco.services.prompter_live import get_live_registry
threshold = float(settings.interactive_idle_reap_seconds)
for session_id, agent_id in get_live_registry().idle_session_ids(threshold):
try:
if agent_id == INTAKE_AGENT_ID:
await self.reap_intake_session(session_id)
elif agent_id == SECRETARY_AGENT_ID:
await self.reap_secretary_session(session_id)
else:
continue
logger.info(
"Reaped idle interactive session",
session_id=session_id,
agent_id=agent_id,
idle_threshold_s=threshold,
)
except Exception as exc:
logger.warning(
"Idle interactive reap failed",
session_id=session_id,
error=str(exc),
)
def _resolve_secretary_host_paths(self) -> dict[str, str | None]:
"""Host paths for the Secretary container's mounts (claude + prompt).
No workspaces mount: the Secretary reads company state via the API and
runs from the baked ``/app`` tree.
"""
if PROJECT_HOST_PATH:
return {
"claude": CLAUDE_AUTH_HOST_PATH,
"prompt": (
f"{DATA_HOST_PATH}/prompts-generated/{SECRETARY_AGENT_ID}-prompt.md"
),
"grok_usage": f"{DATA_HOST_PATH}/grok-usage/{SECRETARY_AGENT_ID}",
}
return {
"claude": CLAUDE_AUTH_HOST_PATH,
"prompt": str(
Path(tempfile.gettempdir())
/ "roboco-prompts"
/ f"{SECRETARY_AGENT_ID}-prompt.md"
),
"grok_usage": str(
Path(tempfile.gettempdir()) / "roboco-grok-usage" / SECRETARY_AGENT_ID
),
}
@staticmethod
def _build_secretary_run_cmd(spec: _SecretaryRunSpec) -> list[str]:
"""Compose the `docker run` argv for the persistent Secretary container."""
cmd: list[str] = [
"docker",
"run",
"-d",
"--name",
spec.container_name,
"--network",
AGENT_NETWORK,
"-v",
f"{spec.hosts['claude']}:/home/agent/.claude",
]
AgentOrchestrator._append_claude_json_mount(cmd, spec.hosts)
cmd.extend(
[
"-v",
f"{spec.hosts['prompt']}:/app/system-prompt.md:ro",
"-e",
f"ROBOCO_AGENT_ID={spec.agent_uuid}",
"-e",
"ROBOCO_AGENT_ROLE=secretary",
"-e",
f"ROBOCO_AGENT_TOKEN={spec.agent_token}",
"-e",
f"ROBOCO_API_URL={spec.api_url}",
"-e",
f"ROBOCO_SECRETARY_SESSION_ID={spec.session_id}",
"-e",
f"ROBOCO_WORKSPACE={spec.cwd}",
"-e",
f"CLAUDE_CODE_SUBAGENT_MODEL={spec.cli_model}",
]
)
AgentOrchestrator._append_interactive_provider_env(cmd, spec)
cmd.append(spec.image)
return cmd
async def _clone_intake_scope(
self,
project_slug: str | None,
product_id: str | None,
project_ids: list[str] | None = None,
) -> tuple[str, list[str]]:
"""Clone the chat scope's repo(s); return (container cwd, all paths).
``project`` → one repo; ``product`` → each distinct cell project (the
Main-PM-style distinct-repo set, kept in its deterministic team order so
the primary is stable); ``project_ids`` → a MegaTask's explicit set of
(possibly unrelated) projects, in the order given. The agent's cwd is the
primary project's intake workspace; for a multi-repo scope the sibling
repos sit alongside it under ``/data/workspaces`` and are readable via
Grep/Glob/Read.
"""
from roboco.db.base import get_session_factory
from roboco.services.workspace import WorkspaceService
team = get_agent_team(INTAKE_AGENT_ID) or "board"
factory = get_session_factory()
async with factory() as db:
slugs = await self._intake_scope_slugs(
db, project_slug, product_id, project_ids
)
ws = WorkspaceService(db)
for slug in slugs:
await ws.ensure_workspace(slug, INTAKE_AGENT_ID)
# Container-side paths (the workspaces tree is mounted at
# /data/workspaces inside the container, regardless of the host root).
paths = [_agent_workspace_path(slug, team, INTAKE_AGENT_ID) for slug in slugs]
return paths[0], paths
@staticmethod
async def _intake_scope_slugs(
db: Any,
project_slug: str | None,
product_id: str | None,
project_ids: list[str] | None = None,
) -> list[str]:
"""Resolve the chat scope to the project slug(s) to clone."""
if project_slug:
return [project_slug]
if project_ids:
return await AgentOrchestrator._slugs_for_project_ids(db, project_ids)
if product_id:
return await AgentOrchestrator._slugs_for_product(db, product_id)
raise ValueError(
"intake scope requires project_slug, product_id, or project_ids"
)
@staticmethod
async def _slugs_for_project_ids(db: Any, project_ids: list[str]) -> list[str]:
"""MegaTask scope: the slugs of an explicit set of (unrelated) projects."""
from uuid import UUID
from roboco.services.project import get_project_service
project_svc = get_project_service(db)
slugs: list[str] = []
for pid in project_ids:
project = await project_svc.get(UUID(pid))
# Fail loud on ANY unresolvable id (matching the single-project route's
# 404) rather than silently cloning fewer repos — a partial scope would
# let the agent draft against an incomplete workspace with no signal.
if not (project and project.slug):
raise ValueError(f"MegaTask scope: project {pid} not found")
slugs.append(project.slug)
if not slugs:
raise ValueError("MegaTask scope resolves to no projects")
return slugs
@staticmethod
async def _slugs_for_product(db: Any, product_id: str) -> list[str]:
"""Product scope: the distinct cell-project slugs, in deterministic order."""
from uuid import UUID
from roboco.services.product import ProductService
from roboco.services.project import get_project_service
project_ids = await ProductService(db).distinct_project_ids(UUID(product_id))
project_svc = get_project_service(db)
slugs: list[str] = []
for pid in project_ids:
project = await project_svc.get(pid)
if project and project.slug:
slugs.append(project.slug)
if not slugs:
raise ValueError(f"product {product_id} resolves to no projects")
return slugs
def _resolve_intake_host_paths(self) -> dict[str, str | None]:
"""Host paths for the intake container's three mounts (claude/prompt/ws).
Mirrors ``_resolve_host_paths`` but only for what the driver needs —
there is no settings.json, MCP config, or briefing for the intake agent.
"""
if PROJECT_HOST_PATH:
return {
"claude": CLAUDE_AUTH_HOST_PATH,
"prompt": (
f"{DATA_HOST_PATH}/prompts-generated/{INTAKE_AGENT_ID}-prompt.md"
),
"workspaces": f"{DATA_HOST_PATH}/workspaces",
"grok_usage": f"{DATA_HOST_PATH}/grok-usage/{INTAKE_AGENT_ID}",
}
return {
"claude": CLAUDE_AUTH_HOST_PATH,
"prompt": str(
Path(tempfile.gettempdir())
/ "roboco-prompts"
/ f"{INTAKE_AGENT_ID}-prompt.md"
),
"workspaces": str(Path(settings.workspaces_root)),
"grok_usage": str(
Path(tempfile.gettempdir()) / "roboco-grok-usage" / INTAKE_AGENT_ID
),
}
@staticmethod
def _append_interactive_provider_env(
cmd: list[str], spec: "_IntakeRunSpec | _SecretaryRunSpec"
) -> None:
"""Inject the per-provider LLM env for an interactive container.
GROK runs on the official ``grok`` CLI, exactly like the one-shot path:
the subscription auth (``~/.grok/auth.json``) is mounted read-only, no
metered xAI key is used, the per-agent data dir is mounted so the driver's
per-turn usage capture lands a ``usage.json`` the finalizer reads back, and
the per-role permissions / reasoning come from the grok flags the driver
computes (``grok_cli_config``) — not env. Every other provider uses the
Claude path's ``ANTHROPIC_*`` injection (or the mounted ``~/.claude``
default when the route carries no creds).
"""
from roboco.llm.providers.grok import GrokCliProvider
from roboco.models.base import ModelProvider
base_url = spec.provider_base_url
auth_token = spec.provider_auth_token
if spec.provider_type == ModelProvider.GROK.value:
GrokCliProvider._append_grok_auth_mount(cmd)
GrokCliProvider._append_usage_mount(cmd, spec.hosts)
cmd.extend(
[
"-e",
"ROBOCO_AGENT_MODEL=grok-build",
"-e",
"ROBOCO_GROK_USAGE_FILE=/home/agent/.grok-usage/usage.json",
]
)
return
if base_url:
cmd.extend(["-e", f"ANTHROPIC_BASE_URL={base_url}"])
if auth_token:
cmd.extend(["-e", f"ANTHROPIC_AUTH_TOKEN={auth_token}"])
@staticmethod
def _build_intake_run_cmd(spec: _IntakeRunSpec) -> list[str]:
"""Compose the `docker run` argv for the persistent intake container.
No claude CLI args (the image ENTRYPOINT is the SDK driver), no
settings.json/hook mount (the driver owns port 9000), no MCP config.
The driver reads ``/app/system-prompt.md`` and the env below.
"""
cmd: list[str] = [
"docker",
"run",
"-d",
"--name",
spec.container_name,
"--network",
AGENT_NETWORK,
"-v",
f"{spec.hosts['claude']}:/home/agent/.claude",
]
AgentOrchestrator._append_claude_json_mount(cmd, spec.hosts)
cmd.extend(
[
"-v",
f"{spec.hosts['prompt']}:/app/system-prompt.md:ro",
"-v",
f"{spec.hosts['workspaces']}:/data/workspaces",
"-e",
f"ROBOCO_AGENT_ID={INTAKE_AGENT_ID}",
"-e",
f"ROBOCO_AGENT_ROLE={get_agent_role(INTAKE_AGENT_ID) or 'prompter'}",
"-e",
f"ROBOCO_API_URL={spec.api_url}",
"-e",
f"ROBOCO_PROMPTER_SESSION_ID={spec.session_id}",
"-e",
f"ROBOCO_WORKSPACE={spec.cwd}",
"-e",
f"CLAUDE_CODE_SUBAGENT_MODEL={spec.cli_model}",
]
)
# GROK mounts the subscription auth + usage dir; other providers use the
# ANTHROPIC_* injection or the mounted ~/.claude default.
AgentOrchestrator._append_interactive_provider_env(cmd, spec)
cmd.append(spec.image)
return cmd
async def _run_container_cmd(self, cmd: list[str]) -> str:
"""Run a detached `docker run` and return the container id."""
proc = await asyncio.create_subprocess_exec(
*cmd,
stdout=asyncio.subprocess.PIPE,
stderr=asyncio.subprocess.PIPE,
)
stdout, stderr = await proc.communicate()
if proc.returncode != 0:
raise RuntimeError(f"Failed to start intake container: {stderr.decode()}")
return stdout.decode().strip()
def _schedule_bg(self, coro: "Coroutine[Any, Any, None]") -> None:
"""Fire-and-forget a coroutine, strong-reffed so it isn't GC'd mid-flight.
Silently no-ops when there's no running loop (sync unit tests); the coro
is closed to avoid a "never awaited" warning.
"""
import contextlib as _ctx
try:
loop = asyncio.get_running_loop()
except RuntimeError:
with _ctx.suppress(Exception):
coro.close()
return
bg = loop.create_task(coro)
self._bg_tasks.add(bg)
bg.add_done_callback(self._bg_tasks.discard)
def _schedule_respawn_persist(
self, agent_slug: str, task_id: str, record: dict[str, Any]
) -> None:
"""Fire-and-forget a write-through of one PM-respawn counter row.
Copies ``record`` so a later in-place mutation can't race the background
write, then schedules it on the strong-ref ``_bg_tasks`` set — the
dispatcher hot path never blocks on the DB, and a write failure degrades
to in-memory-only (today's behaviour).
"""
self._schedule_bg(
self._persist_respawn_record(agent_slug, task_id, dict(record))
)
def _schedule_intake_first_message(self, session_id: str, text: str) -> None:
"""Fire-and-forget the opening message once the container is reachable."""
self._schedule_bg(self._deliver_when_ready(session_id, text))
async def _deliver_when_ready(
self,
session_id: str,
text: str,
*,
attempts: int = 30,
delay: float = 1.0,
) -> None:
"""Retry-deliver the first message until the container receiver is up."""
from roboco.services.prompter_live import get_live_registry
registry = get_live_registry()
for _ in range(attempts):
if await registry.deliver(session_id, text):
return
await asyncio.sleep(delay)
logger.warning(
"Intake first message never delivered (receiver never came up)",
session_id=session_id,
)
# =========================================================================
# AGENT STOPPING
# =========================================================================
async def stop_agent(
self,
agent_id: str,
graceful: bool = True,
exit_reason: str = "stopped",
release_claim: bool = False,
stop_reason: str = "stop_agent",
) -> None:
"""Stop an agent container.
Finalization (the HTTP call to the agent SDK's /usage/status endpoint)
is performed BEFORE acquiring self._lock so that the network I/O does
not block other operations that need the lock.
When ``release_claim`` is True the caller declares the stopped agent
will not continue its task (budget kill, orchestrator shutdown) and the
agent's claimed/in_progress task is handed back to the pool immediately
instead of waiting up to ``stale_claim_reap_seconds`` for the
stale-claim reaper to notice the dead heartbeat — closing the
SIGTERM-mid-verb gap where a task sat CLAIMED/IN_PROGRESS with no
running agent. Default False: the provider-park / waiting path
(``mark_waiting_long``) and interactive stops manage their own claim
lifecycle, so they opt out and the claim survives for the probe-resume
loop. A provider-parked agent (``rate_limit_lifted`` WaitingRecord) is
always skipped even when a caller opts in — its claim must survive so
probe-success revives the same agent on the same task.
``stop_reason`` breadcrumbs the container as an expected stop (see
``_record_expected_stop``) BEFORE the docker stop/kill is issued —
``_check_health`` polls without holding ``self._lock``, so it can
observe the container already gone while this call is still mid-flight;
recording early (not just at ``_remove_container``) closes that race.
"""
# Finalize the spawn-session row before the container is removed so we
# can still query the SDK's /usage/status endpoint. This must happen
# outside self._lock — the HTTP round-trip would otherwise hold the
# lock for the full network timeout.
instance = self._instances.get(agent_id)
if instance is None:
return
if instance.container_id:
await self._finalize_spawn_session(agent_id, exit_reason=exit_reason)
# Capture the task the agent was working on before the instance state
# is mutated, so a release_claim stop can hand it back to the pool once
# the container is gone. Only relevant when the caller opted in.
stopped_task_id = instance.current_task_id if release_claim else None
async with self._lock:
if agent_id not in self._instances:
return
instance = self._instances[agent_id]
if instance.container_id:
self._record_expected_stop(agent_id, stop_reason)
instance.state = AgentState.STOPPING
container_name = f"roboco-agent-{agent_id}"
if graceful:
# Graceful stop with timeout
proc = await asyncio.create_subprocess_exec(
"docker",
"stop",
"-t",
"10",
container_name,
stdout=asyncio.subprocess.DEVNULL,
stderr=asyncio.subprocess.DEVNULL,
)
await proc.wait()
else:
# Force kill
proc = await asyncio.create_subprocess_exec(
"docker",
"kill",
container_name,
stdout=asyncio.subprocess.DEVNULL,
stderr=asyncio.subprocess.DEVNULL,
)
await proc.wait()
# Remove container
await self._remove_container(container_name)
instance.state = AgentState.OFFLINE
instance.container_id = None
logger.info("Agent stopped", agent_id=agent_id)
# Hand the stopped agent's claimed task back to the pool now, instead
# of leaving it CLAIMED/IN_PROGRESS with no running agent for the
# reaper's full heartbeat TTL. Done outside self._lock (DB I/O) and
# best-effort: a failure logs a warning and the stale-claim reaper
# remains the backstop. Skipped for a provider-parked agent — the
# probe-resume loop owns its recovery and the claim must survive.
if stopped_task_id and not self._is_rate_limit_parked(agent_id):
await self._release_stopped_agent_claim(agent_id, stopped_task_id)
def _is_rate_limit_parked(self, agent_id: str) -> bool:
"""True if the agent is provider-parked on a rate limit.
Mirrors the reaper's ``_assignee_is_provider_parked`` guard but keyed
by slug directly (no task row needed): a ``rate_limit_lifted``
WaitingRecord means the probe-resume loop owns this agent's recovery
and its claim must survive a stop. Defensive on a missing registry.
"""
records = getattr(self, "_waiting_records", None)
if not records:
return False
record = records.get(agent_id)
return record is not None and record.waiting_for == "rate_limit_lifted"
async def _release_stopped_agent_claim(
self, agent_id: str, task_id_str: str
) -> None:
"""Force a stopped agent's claimed/in_progress task back to pending.
Reuses the hardened, idempotent, status-checked
``TaskService.unclaim_for_reaper`` (the same path the stale-claim
reaper uses) so a task that already moved on (e.g. submitted to QA
before the stop) is a clean no-op. Opens its own short-lived session
outside ``self._lock``. Best-effort: a DB failure logs and the reaper
backstops on the next tick.
"""
from roboco.db.base import get_session_factory
from roboco.services.task import TaskService
from roboco.utils.converters import InvalidIdentifierError, require_uuid
try:
task_id = require_uuid(task_id_str)
except InvalidIdentifierError as exc:
# A malformed task_id_str is a bad identifier, not a transient
# failure — log it so the drop is visible instead of swallowed,
# then no-op (nothing to release). Other exceptions still fall
# through to the broad catch below (#25).
logger.warning(
"stopped agent claim had malformed task id",
task_id_str=task_id_str,
error=str(exc),
)
return
try:
factory = get_session_factory()
async with factory() as db:
svc = TaskService(db)
await svc.unclaim_for_reaper(task_id)
await db.commit()
logger.info(
"stopped agent claim released to pool",
agent_id=agent_id,
task_id=task_id_str,
)
except Exception as exc:
logger.warning(
"stop_agent claim release failed; reaper will backstop",
agent_id=agent_id,
task_id=task_id_str,
error=str(exc),
)
# =========================================================================
# WAITING STATE MANAGEMENT
# =========================================================================
async def mark_waiting_long(
self,
agent_id: str,
waiting_for: str,
task_id: str | None = None,
context: dict[str, Any] | None = None,
) -> None:
"""
Mark an agent as WAITING_LONG and terminate.
The agent will be respawned when the wait condition is resolved.
The record is mirrored to `waiting_records` in Postgres so a later
orchestrator restart can still resolve the wait.
"""
record = WaitingRecord(
agent_id=agent_id,
task_id=task_id,
waiting_for=waiting_for,
waiting_since=datetime.now(UTC),
context=context or {},
)
self._waiting_records[agent_id] = record
await self._persist_waiting_record(record)
# Stop the agent
await self.stop_agent(agent_id, stop_reason=f"waiting_long_{waiting_for}")
# Update state
if agent_id in self._instances:
self._instances[agent_id].state = AgentState.WAITING_LONG
self._instances[agent_id].waiting_for = waiting_for
self._instances[agent_id].waiting_context = context or {}
logger.info(
"Agent marked as waiting_long",
agent_id=agent_id,
waiting_for=waiting_for,
task_id=task_id,
)
async def _persist_waiting_record(self, record: WaitingRecord) -> None:
"""Upsert a WaitingRecord into the waiting_records table."""
try:
from uuid import UUID as _UUID
from sqlalchemy import delete
from roboco.db.base import get_session_factory
from roboco.db.tables import WaitingRecordTable
session_factory = get_session_factory()
async with session_factory() as db:
# One record per agent; delete prior then insert.
await db.execute(
delete(WaitingRecordTable).where(
WaitingRecordTable.agent_id == record.agent_id
)
)
row = WaitingRecordTable(
agent_id=record.agent_id,
task_id=(_UUID(record.task_id) if record.task_id else None),
waiting_for=record.waiting_for,
waiting_since=record.waiting_since,
context=record.context,
)
db.add(row)
await db.commit()
except Exception as e:
logger.error(
"Failed to persist waiting record",
agent_id=record.agent_id,
error=str(e),
)
async def _delete_waiting_record(self, agent_id: str) -> None:
"""Delete a persisted waiting record when its wait resolves."""
try:
from sqlalchemy import delete
from roboco.db.base import get_session_factory
from roboco.db.tables import WaitingRecordTable
session_factory = get_session_factory()
async with session_factory() as db:
await db.execute(
delete(WaitingRecordTable).where(
WaitingRecordTable.agent_id == agent_id
)
)
await db.commit()
except Exception as e:
logger.error(
"Failed to delete waiting record",
agent_id=agent_id,
error=str(e),
)
async def _persist_respawn_record(
self, agent_slug: str, task_id: str, record: dict[str, Any]
) -> None:
"""Write-through one PM-respawn counter row (atomic upsert).
Best-effort, mirroring ``_persist_waiting_record``: a persistence failure
must never gate or un-gate a spawn, so any error is logged and swallowed.
The counter stays authoritative in memory regardless.
Unlike ``_persist_waiting_record`` (inline-awaited, one row per agent),
this is scheduled fire-and-forget per gate mutation, and a respawn loop
fires several persists for the same ``(agent_slug, task_id)`` in quick
succession. A delete-then-insert raced under that concurrency: two
transactions for the same key overlapped, the loser's INSERT hit
``pk_respawn_tracker`` UniqueViolation, the durable count stuck at the
first INSERT's value, and a restart re-burned the strike threshold — the
exact re-burn this feature was built to stop (2026-06-27 live meltdown).
The single ``ON CONFLICT DO UPDATE`` upsert is race-free at row level, BUT
fire-and-forget tasks can still COMMIT out of order: a slow stale persist
(count=2) scheduled first can resolve AFTER a fast fresh one (count=4)
scheduled second, leaving the durable row at the stale low count (same
re-burn on restart). The ``_respawn_persist_lock`` is acquired as the
FIRST await below, so acquisition order = task creation order = logical
schedule order, and commits land in that order — the durable row always
ends at the latest logical value.
"""
async with self._respawn_persist_lock:
try:
from uuid import UUID as _UUID
from sqlalchemy.dialects.postgresql import insert as pg_insert
from roboco.db.base import get_session_factory
from roboco.db.tables import RespawnTrackerTable
tid = _UUID(task_id)
now = datetime.now(UTC)
stmt = pg_insert(RespawnTrackerTable).values(
agent_slug=agent_slug,
task_id=tid,
count=int(record["count"]),
last_status=record.get("last_status"),
last_check=record["last_check"],
tracing_resets=int(record.get("tracing_resets", 0)),
revisit_resets=int(record.get("revisit_resets", 0)),
notified=bool(record.get("notified", False)),
updated_at=now,
)
stmt = stmt.on_conflict_do_update(
index_elements=[
RespawnTrackerTable.agent_slug,
RespawnTrackerTable.task_id,
],
set_={
"count": stmt.excluded.count,
"last_status": stmt.excluded.last_status,
"last_check": stmt.excluded.last_check,
"tracing_resets": stmt.excluded.tracing_resets,
"revisit_resets": stmt.excluded.revisit_resets,
"notified": stmt.excluded.notified,
"updated_at": stmt.excluded.updated_at,
},
)
session_factory = get_session_factory()
async with session_factory() as db:
await db.execute(stmt)
await db.commit()
except Exception as e:
logger.error(
"Failed to persist respawn record",
agent_id=agent_slug,
task_id=task_id,
error=str(e),
)
async def _clear_respawn_record(self, agent_slug: str, task_id: str) -> None:
"""Delete one PM-respawn counter row (best-effort).
Used by the startup loader to evict a row whose task is gone or
terminal, so a stale counter never resurrects against a fixed task.
"""
try:
from uuid import UUID as _UUID
from sqlalchemy import delete
from roboco.db.base import get_session_factory
from roboco.db.tables import RespawnTrackerTable
session_factory = get_session_factory()
async with session_factory() as db:
await db.execute(
delete(RespawnTrackerTable).where(
RespawnTrackerTable.agent_slug == agent_slug,
RespawnTrackerTable.task_id == _UUID(task_id),
)
)
await db.commit()
except Exception as e:
logger.error(
"Failed to clear respawn record",
agent_id=agent_slug,
task_id=task_id,
error=str(e),
)
# =========================================================================
# PROVIDER QUERY HELPERS (used by the choreographer rate-limit path)
# =========================================================================
def get_provider_for_agent(self, agent_slug: str) -> str | None:
"""Return the ``provider_type`` for a currently-tracked agent, or None.
Reads the in-memory ``_instances`` dict so this is synchronous and
O(1). Returns None when the agent is not tracked or has no config.
Args:
agent_slug: The agent slug (e.g. ``"be-dev-1"``).
"""
instance = self._instances.get(agent_slug)
if instance is None or instance.config is None:
return None
return instance.config.provider_type
def get_active_agent_slugs_for_provider(self, provider: str) -> list[str]:
"""Return slugs of all active agents currently using ``provider``.
"Active" means the instance's state is ACTIVE or STARTING (i.e.
the container is running or spinning up — not IDLE, WAITING_LONG,
STOPPING, or OFFLINE).
Args:
provider: Provider type string, e.g. ``"anthropic"`` or
``"ollama_cloud"``.
"""
active_states = {AgentState.ACTIVE, AgentState.STARTING}
return [
slug
for slug, inst in self._instances.items()
if inst.state in active_states
and inst.config is not None
and inst.config.provider_type == provider
]
# =========================================================================
# TOKEN USAGE INSTRUMENTATION
# =========================================================================
async def _record_spawn_session(
self,
config: "OrchestratorAgentConfig",
task_id: str | None,
) -> "UUID | None":
"""Insert a row into agent_spawn_sessions after a successful spawn.
Returns the UUID of the created row so the caller can store it on
the AgentInstance for later direct-by-id lookup in
_finalize_spawn_session. Returns None when the insert fails; a
missing session row must never block the spawn path.
"""
try:
from uuid import uuid4 as _uuid4
from roboco.db.base import get_session_factory
from roboco.db.tables import AgentSpawnSessionTable
agent_slug = config.agent_id
team = get_agent_team(agent_slug) or "backend"
role = get_agent_role(agent_slug) or "developer"
# A delivery-role spawn with no task_id is unattributed usage (#11) —
# the rollup can't tie the spend to a task. Intake/secretary/PM spawns
# legitimately carry no task and are not flagged.
if is_unattributed_delivery_spawn(role, task_id):
logger.warning(
"Spawn session has no task_id for a delivery role — "
"unattributed usage",
agent_slug=agent_slug,
role=role,
)
session_id = _uuid4()
session_factory = get_session_factory()
async with session_factory() as db:
row = AgentSpawnSessionTable(
id=session_id,
agent_slug=agent_slug,
team=team,
role=role,
model=config.model or "unknown",
task_id=task_id,
started_at=datetime.now(UTC),
)
db.add(row)
await db.commit()
logger.debug(
"Spawn session recorded",
agent_slug=agent_slug,
session_id=str(session_id),
task_id=task_id,
)
return session_id
except Exception as exc:
logger.warning(
"Failed to record spawn session",
agent_slug=config.agent_id,
error=str(exc),
)
return None
def _claude_session_id_for(self, agent_id: str) -> str | None:
"""The orchestrator-assigned Claude session id for a running agent."""
instance = self._instances.get(agent_id)
return (
instance.config.claude_session_id if instance and instance.config else None
)
@staticmethod
def _usage_from_transcript(
agent_id: str, claude_session_id: str | None = None
) -> tuple[int, int, int, int, int]:
"""Sum token usage + turn count from the agent's Claude Code transcript.
The host ``~/.claude`` is mounted into the orchestrator, so transcripts
are readable here under ``projects/<cwd-dir>/<session-id>.jsonl``. When
the orchestrator-assigned ``claude_session_id`` is known we locate the
exact transcript by id across ANY project dir — review/coordinate roles
run at cwd ``/app`` so theirs lands in ``projects/-app``, not in a
per-agent ``projects/*-{slug}`` dir. Without an id we fall back to the
newest transcript in the agent's own workspace dir. Durable fallback for
the live SDK ``/usage/status`` fetch, which misses for short-lived or
torn-down agents. Returns zeros when no transcript is found.
"""
from roboco.agent_sdk.transcript_usage import sum_transcript_usage
projects = Path.home() / ".claude" / "projects"
try:
if claude_session_id:
by_id = list(projects.glob(f"*/{claude_session_id}.jsonl"))
if by_id:
return sum_transcript_usage(by_id[0])
jsonl = [
f
for d in projects.glob(f"*-{agent_id}")
if d.is_dir()
for f in d.glob("*.jsonl")
]
if not jsonl:
return (0, 0, 0, 0, 0)
newest = max(jsonl, key=lambda f: f.stat().st_mtime)
return sum_transcript_usage(newest)
except OSError:
return (0, 0, 0, 0, 0)
def _grok_usage_json(self, agent_id: str) -> dict[str, Any] | None:
"""Read a GROK agent's ``usage.json`` (``{model, total_tokens, cost_usd}``).
Written to the per-agent data dir by the grok-CLI entrypoint (one-shot,
post-run) and the interactive driver (per-turn); read back from the same
branched dir the writers mount (``_grok_usage_dir``). Returns ``None`` when
absent / unreadable.
"""
# os.path.basename keeps only the final path component of the agent id
# before the path is built — the path-injection sanitizer CodeQL models,
# applied here in the read's own scope. _grok_usage_dir's guard rejects
# '.' / '..' / separators / NUL upstream (a bad id raises -> None here).
try:
usage_json = self._grok_usage_dir(os.path.basename(agent_id)) / "usage.json"
data = json.loads(usage_json.read_text(encoding="utf-8"))
except (OSError, ValueError, json.JSONDecodeError):
return None
return data if isinstance(data, dict) else None
def _grok_usage_tokens(self, agent_id: str) -> tuple[int, int, int, int]:
"""A GROK agent's token usage from its ``usage.json``.
grok reports a single cumulative total with no input/output split, so it
folds into output (it bills at the output rate, matching
``calculate_cost``). A WARNING is logged on a missing/zero read because a
silent mount/uid failure is otherwise indistinguishable from a genuine
zero-cost run. Returns ``(input, output, cache_read, cache_write)``.
"""
data = self._grok_usage_json(agent_id)
total = 0
if data:
try:
total = int(data.get("total_tokens", 0))
except (TypeError, ValueError):
total = 0
if not total:
logger.warning(
"GROK agent finalized with no readable usage "
"(0 tokens / $0) — check the data dir mount",
agent_id=agent_id,
)
return (0, total, 0, 0)
def _grok_cost_usd(self, agent_id: str) -> float:
"""A GROK agent's captured notional cost from its ``usage.json`` (0 if none)."""
data = self._grok_usage_json(agent_id)
if not data:
return 0.0
try:
return float(data.get("cost_usd", 0.0))
except (TypeError, ValueError):
return 0.0
async def _enforce_grok_cost_budget(self) -> None:
"""Kill a live GROK container whose captured cost exceeds the cap.
The grok CLI exposes no live token/budget hook, so the budget kill-switch
(Claude Code parity for runaway token burn — a loop that keeps firing
verbs evades the idle watchdog but still burns cost) reads each ACTIVE
GROK container's captured cost from its ``usage.json`` and kills + evicts
it past ``ROBOCO_GROK_MAX_COST_USD``. The reaper then releases the freed
task. This bites on the interactive sessions (the driver rewrites
usage.json every turn, so a runaway chat is caught between turns); a
one-shot ``grok -p`` writes usage.json only post-run and is bounded by its
``--max-turns`` cap instead. Disabled (no-op) when the cap is <= 0.
"""
cap = getattr(self, "_grok_max_cost_usd", 0.0)
if cap <= 0:
return
from roboco.models.base import ModelProvider
for agent_id, instance in list(self._instances.items()):
config = instance.config
if (
config is None
or config.provider_type != ModelProvider.GROK.value
or instance.state != AgentState.ACTIVE
):
continue
cost = self._grok_cost_usd(agent_id)
if cost <= cap:
continue
try:
await self._remove_container(
f"roboco-agent-{agent_id}", stop_reason="grok_cost_cap"
)
except Exception as exc:
logger.error(
"grok cost-cap kill failed; will retry next tick",
agent_id=agent_id,
error=str(exc),
)
continue
# Finalize the spawn session BEFORE popping the instance so the
# captured usage/cost is recorded; popping first would lose the
# model + usage_session_id and leave the session row open.
with contextlib.suppress(Exception):
await self._finalize_spawn_session(agent_id, exit_reason="cost_cap")
self._instances.pop(agent_id, None)
# Interactive roles (intake/secretary) have an open panel relay; a
# raw kill would leave the SSE hanging (frozen chat). Close it with a
# reason so the panel reports why the chat ended.
if agent_id in (INTAKE_AGENT_ID, SECRETARY_AGENT_ID):
from roboco.services.prompter_live import get_live_registry
get_live_registry().close_by_agent(
agent_id, error="Chat ended: the Grok cost cap was exceeded."
)
logger.warning(
"grok container killed: cost ceiling exceeded",
agent_id=agent_id,
cost_usd=round(cost, 4),
cap_usd=cap,
)
async def _resolve_final_token_usage(
self, agent_id: str
) -> tuple[int, int, int, int]:
"""Resolve final token counts for a stopping agent.
For a GROK agent, reads the captured ``usage.json`` (no SDK server /
Claude transcript exists). Otherwise tries the live SDK ``/usage/status``
first; if that misses — the SDK's in-memory counts race container teardown
for short-lived agents — it falls back to the agent's Claude Code
transcript, which is durable and mounted into this container. Returns
``(input, output, cache_read, cache_write)``.
"""
from roboco.models.base import ModelProvider
if self.get_provider_for_agent(agent_id) == ModelProvider.GROK.value:
return self._grok_usage_tokens(agent_id)
tokens = (0, 0, 0, 0)
sdk_url = f"http://roboco-agent-{agent_id}:{SDK_PORT}/usage/status"
try:
async with httpx.AsyncClient(
timeout=3.0, headers=_system_api_headers()
) as client:
resp = await client.get(sdk_url)
if resp.status_code == http_status.HTTP_200_OK:
data = resp.json()
tokens = (
data.get("tokens_input", 0),
data.get("tokens_output", 0),
data.get("tokens_cache_read", 0),
data.get("tokens_cache_write", 0),
)
except Exception as sdk_exc:
logger.debug(
"Could not fetch final token counts from SDK",
agent_id=agent_id,
error=str(sdk_exc),
)
if not tokens[0] and not tokens[1]:
tin, tout, cr, cw, _turns = self._usage_from_transcript(
agent_id, self._claude_session_id_for(agent_id)
)
if tin or tout:
tokens = (tin, tout, cr, cw)
return tokens
async def _resolve_final_turns_tools(self, agent_id: str) -> tuple[int, int]:
"""Resolve final ``(turns, tool_calls)`` for a stopping agent.
Primary source is the live SDK ``/usage/status`` (which carries both).
For ``turns`` only there is a durable Claude-transcript fallback (unique
assistant-message count) for short-lived agents whose SDK counts race
teardown; ``tool_calls`` has no transcript equivalent and stays 0 ("n/a")
when the SDK misses. Grok agents have neither — returns ``(0, 0)``.
Best-effort: any failure degrades to zeros, never blocks finalize.
"""
from roboco.models.base import ModelProvider
if self.get_provider_for_agent(agent_id) == ModelProvider.GROK.value:
return (0, 0)
turns = tool_calls = 0
sdk_url = f"http://roboco-agent-{agent_id}:{SDK_PORT}/usage/status"
try:
async with httpx.AsyncClient(
timeout=3.0, headers=_system_api_headers()
) as client:
resp = await client.get(sdk_url)
if resp.status_code == http_status.HTTP_200_OK:
data = resp.json()
turns = int(data.get("turns", 0) or 0)
tool_calls = int(data.get("tool_calls", 0) or 0)
except Exception as sdk_exc:
logger.debug(
"Could not fetch final turns/tool_calls from SDK",
agent_id=agent_id,
error=str(sdk_exc),
)
if not turns:
*_tokens, t = self._usage_from_transcript(
agent_id, self._claude_session_id_for(agent_id)
)
turns = t
return turns, tool_calls
async def _finalize_spawn_session(
self,
agent_id: str,
exit_reason: str = "stopped",
) -> None:
"""Close the open agent_spawn_sessions row for this agent.
Resolves final token counts (live SDK, with a durable transcript
fallback), calculates cost via the pricing module, then updates the DB
row with ended_at, token totals, exit_reason, and estimated_cost_usd.
Errors are caught and logged — finalization must never block stop_agent.
"""
try:
from roboco.billing.pricing import calculate_cost
from roboco.db.base import get_session_factory
from roboco.db.tables import AgentSpawnSessionTable
# Resolve final token counts (live SDK, with transcript fallback).
(
tokens_input,
tokens_output,
tokens_cache_read,
tokens_cache_write,
) = await self._resolve_final_token_usage(agent_id)
# Resolve LLM iterations + tool calls (live SDK; turns has a
# transcript fallback). Separate from the token tuple so the live
# snapshot helpers keep their 4-tuple contract.
turns, tool_calls = await self._resolve_final_turns_tools(agent_id)
# Look up the model and usage_session_id from the running instance config.
model = "unknown"
instance = self._instances.get(agent_id)
if instance and instance.config:
model = instance.config.model or "unknown"
usage_session_id = instance.usage_session_id if instance else None
cost = calculate_cost(
model=model,
tokens_input=tokens_input,
tokens_output=tokens_output,
tokens_cache_read=tokens_cache_read,
tokens_cache_write=tokens_cache_write,
)
session_factory = get_session_factory()
async with session_factory() as db:
from sqlalchemy import select, update
# Prefer a direct lookup by the session UUID captured at spawn
# time; fall back to the (agent_slug, ended_at IS NULL) query
# for instances that pre-date the usage_session_id field.
if usage_session_id is not None:
result = await db.execute(
select(AgentSpawnSessionTable).where(
AgentSpawnSessionTable.id == usage_session_id
)
)
else:
result = await db.execute(
select(AgentSpawnSessionTable)
.where(
AgentSpawnSessionTable.agent_slug == agent_id,
AgentSpawnSessionTable.ended_at.is_(None),
)
.order_by(AgentSpawnSessionTable.started_at.desc())
.limit(1)
)
session_row = result.scalar_one_or_none()
if session_row is not None:
await db.execute(
update(AgentSpawnSessionTable)
.where(AgentSpawnSessionTable.id == session_row.id)
.values(
ended_at=datetime.now(UTC),
tokens_input=tokens_input,
tokens_output=tokens_output,
tokens_cache_read=tokens_cache_read,
tokens_cache_write=tokens_cache_write,
turns=turns,
tool_calls=tool_calls,
exit_reason=exit_reason,
estimated_cost_usd=cost,
)
)
await db.commit()
logger.debug(
"Spawn session finalized",
agent_id=agent_id,
session_id=str(session_row.id),
tokens_input=tokens_input,
tokens_output=tokens_output,
estimated_cost_usd=cost,
)
except Exception as exc:
logger.warning(
"Failed to finalize spawn session",
agent_id=agent_id,
error=str(exc),
)
@staticmethod
async def _fetch_agent_tokens(
client: httpx.AsyncClient, agent_id: str
) -> tuple[int, int, int, int] | None:
"""Fetch cumulative token counts from an agent's SDK usage endpoint.
Returns ``(input, output, cache_read, cache_write)`` or ``None`` when the
agent returns a non-200 status or has not accrued any tokens yet.
"""
sdk_url = f"http://roboco-agent-{agent_id}:{SDK_PORT}/usage/status"
resp = await client.get(sdk_url)
if resp.status_code != http_status.HTTP_200_OK:
return None
data = resp.json()
tokens = (
data.get("tokens_input", 0),
data.get("tokens_output", 0),
data.get("tokens_cache_read", 0),
data.get("tokens_cache_write", 0),
)
if sum(tokens) == 0:
return None
return tokens
async def _resolve_active_tokens(
self, client: httpx.AsyncClient, agent_id: str
) -> tuple[int, int, int, int] | None:
"""Resolve live token counts for an active agent.
Tries the agent SDK's ``/usage/status`` first; on a zero/miss falls
back to the durable transcript (the SDK can report zero mid-run, the
same race the finalize path handles). Returns ``None`` when neither
source has any usage yet. GROK has no SDK server or Claude transcript,
so it routes to its ``usage.json`` — the same early return the finalize
path uses, so live USAGE_SNAPSHOT reflects grok agents mid-run too.
"""
instance = self._instances.get(agent_id)
is_grok = (
instance is not None
and instance.config is not None
and instance.config.provider_type == ModelProvider.GROK.value
)
if is_grok:
grok_tokens = self._grok_usage_tokens(agent_id)
return grok_tokens if any(grok_tokens) else None
tokens = await self._fetch_agent_tokens(client, agent_id)
if tokens is not None:
return tokens
tin, tout, cr, cw, _turns = self._usage_from_transcript(
agent_id, self._claude_session_id_for(agent_id)
)
token_counts = (tin, tout, cr, cw)
return token_counts if any(token_counts) else None
@staticmethod
async def _persist_token_snapshot(
session_factory: Any,
agent_id: str,
instance: AgentInstance,
tokens: tuple[int, int, int, int],
) -> bool:
"""Insert a token_usage_snapshots row and refresh the open session totals.
Returns True when a snapshot was written; False when the agent has no
open spawn-session row to attach it to.
"""
from uuid import uuid4
from sqlalchemy import select, update
from roboco.db.tables import AgentSpawnSessionTable, TokenUsageSnapshotTable
tokens_input, tokens_output, tokens_cache_read, tokens_cache_write = tokens
async with session_factory() as db:
# Prefer a direct lookup by the session UUID captured at spawn time;
# fall back to the agent_slug heuristic for instances that pre-date
# the usage_session_id field.
if instance.usage_session_id is not None:
result = await db.execute(
select(AgentSpawnSessionTable).where(
AgentSpawnSessionTable.id == instance.usage_session_id
)
)
else:
result = await db.execute(
select(AgentSpawnSessionTable)
.where(
AgentSpawnSessionTable.agent_slug == agent_id,
AgentSpawnSessionTable.ended_at.is_(None),
)
.order_by(AgentSpawnSessionTable.started_at.desc())
.limit(1)
)
session_row = result.scalar_one_or_none()
if session_row is None:
return False
db.add(
TokenUsageSnapshotTable(
id=uuid4(),
agent_spawn_session_id=session_row.id,
snapshotted_at=datetime.now(UTC),
tokens_input=tokens_input,
tokens_output=tokens_output,
tokens_cache_read=tokens_cache_read,
tokens_cache_write=tokens_cache_write,
)
)
await db.execute(
update(AgentSpawnSessionTable)
.where(AgentSpawnSessionTable.id == session_row.id)
.values(
tokens_input=tokens_input,
tokens_output=tokens_output,
tokens_cache_read=tokens_cache_read,
tokens_cache_write=tokens_cache_write,
)
)
await db.commit()
return True
async def _sweep_token_snapshots(self) -> None:
"""Write a token_usage_snapshots row for each active agent with non-zero tokens.
Called from _run_sweep() every ~60 s. Also updates the cumulative
token counts on the open agent_spawn_sessions row so the DB reflects
current progress without waiting for session close.
Errors per-agent are caught so one bad agent doesn't abort the whole sweep.
Also publishes a USAGE_SNAPSHOT aggregate event after the loop so the
/ws/system dashboard updates live for active agents.
"""
if not self._instances:
return
try:
from roboco.db.base import get_session_factory
except ImportError:
return
session_factory = get_session_factory()
# Accumulators for the post-loop USAGE_SNAPSHOT event.
_usage_by_agent: list[dict[str, Any]] = []
_usage_total_input = 0
_usage_total_output = 0
_usage_total_cost = 0.0
async with httpx.AsyncClient(
timeout=3.0, headers=_system_api_headers()
) as client:
for agent_id, instance in list(self._instances.items()):
if instance.state not in (
AgentState.ACTIVE,
AgentState.WAITING_SHORT,
):
continue
try:
tokens = await self._resolve_active_tokens(client, agent_id)
if tokens is None:
continue
persisted = await self._persist_token_snapshot(
session_factory, agent_id, instance, tokens
)
if not persisted:
continue
tokens_input = tokens[0]
tokens_output = tokens[1]
tokens_cache_read = tokens[2]
tokens_cache_write = tokens[3]
model = instance.config.model if instance.config else "unknown"
# Accumulate per-agent data for the aggregate snapshot.
with contextlib.suppress(Exception):
from roboco.billing.pricing import calculate_cost
agent_cost = calculate_cost(
model=model,
tokens_input=tokens_input,
tokens_output=tokens_output,
tokens_cache_read=tokens_cache_read,
tokens_cache_write=tokens_cache_write,
)
_usage_by_agent.append(
{
"agent_id": agent_id,
"input_tokens": tokens_input,
"output_tokens": tokens_output,
"cache_read_tokens": tokens_cache_read,
"cache_write_tokens": tokens_cache_write,
"model": model,
"cost_estimate": agent_cost,
}
)
_usage_total_input += tokens_input
_usage_total_output += tokens_output
_usage_total_cost += agent_cost
except Exception as agent_exc:
logger.debug(
"Token snapshot failed for agent",
agent_id=agent_id,
error=str(agent_exc),
)
# Publish a USAGE_SNAPSHOT aggregate if any active agents had token data.
if _usage_by_agent:
with contextlib.suppress(Exception):
from roboco.events import get_event_bus
from roboco.services.usage_events import (
UsageSnapshot,
publish_usage_snapshot,
)
await publish_usage_snapshot(
get_event_bus(),
UsageSnapshot(
period="live",
totals={
"input_tokens": _usage_total_input,
"output_tokens": _usage_total_output,
},
cost_estimate=_usage_total_cost,
by_agent=_usage_by_agent,
),
)
async def _sweep_daily_rollup(self) -> None:
"""Upsert daily_usage_rollups from closed agent_spawn_sessions.
Groups ended sessions by (date, agent_slug, team, model) and sums
their token counts + cost. Uses a Python-side upsert to stay
compatible with asyncpg / SQLAlchemy without raw INSERT ... ON CONFLICT
dialect-specific SQL.
Errors are caught so a bad rollup doesn't abort the sweeper.
"""
try:
from roboco.db.base import get_session_factory
from roboco.db.tables import AgentSpawnSessionTable
except ImportError:
return
try:
from uuid import uuid4 as _uuid4
from sqlalchemy import func, select
session_factory = get_session_factory()
async with session_factory() as db:
# Aggregate closed sessions by (date, agent_slug, team, model).
# Limit to the last 7 days to avoid re-aggregating all-time
# history on every sweep — older days are already stable.
rollup_window_start = datetime.now(UTC) - timedelta(days=7)
result = await db.execute(
select(
func.date(AgentSpawnSessionTable.started_at).label("date"),
AgentSpawnSessionTable.agent_slug,
AgentSpawnSessionTable.team,
AgentSpawnSessionTable.model,
func.sum(AgentSpawnSessionTable.tokens_input).label(
"tokens_input"
),
func.sum(AgentSpawnSessionTable.tokens_output).label(
"tokens_output"
),
func.sum(AgentSpawnSessionTable.tokens_cache_read).label(
"tokens_cache_read"
),
func.sum(AgentSpawnSessionTable.tokens_cache_write).label(
"tokens_cache_write"
),
func.sum(AgentSpawnSessionTable.estimated_cost_usd).label(
"total_cost_usd"
),
func.count(AgentSpawnSessionTable.id).label("session_count"),
)
.where(
AgentSpawnSessionTable.ended_at.isnot(None),
AgentSpawnSessionTable.started_at >= rollup_window_start,
)
.group_by(
func.date(AgentSpawnSessionTable.started_at),
AgentSpawnSessionTable.agent_slug,
AgentSpawnSessionTable.team,
AgentSpawnSessionTable.model,
)
)
rows = result.fetchall()
for row in rows:
await self._upsert_rollup_row(db, row, _uuid4)
await db.commit()
logger.debug("Daily usage rollup complete", rows_processed=len(rows))
except Exception as exc:
logger.warning("Daily usage rollup failed", error=str(exc))
async def _upsert_rollup_row(self, db: Any, row: Any, uuid4: Any) -> None:
"""Insert or update a single daily_usage_rollups row from an aggregate.
Looks up the existing rollup for (date, agent_slug, team, model) and
either updates its summed columns or inserts a fresh row.
"""
from sqlalchemy import select, update
from roboco.db.tables import DailyUsageRollupTable
key = {
"date": row.date,
"agent_slug": row.agent_slug,
"team": row.team,
"model": row.model,
}
values = {
"tokens_input": int(row.tokens_input or 0),
"tokens_output": int(row.tokens_output or 0),
"tokens_cache_read": int(row.tokens_cache_read or 0),
"tokens_cache_write": int(row.tokens_cache_write or 0),
"total_cost_usd": float(row.total_cost_usd or 0.0),
"session_count": int(row.session_count or 0),
}
existing_result = await db.execute(
select(DailyUsageRollupTable).where(
DailyUsageRollupTable.date == key["date"],
DailyUsageRollupTable.agent_slug == key["agent_slug"],
DailyUsageRollupTable.team == key["team"],
DailyUsageRollupTable.model == key["model"],
)
)
existing = existing_result.scalar_one_or_none()
if existing is not None:
await db.execute(
update(DailyUsageRollupTable)
.where(DailyUsageRollupTable.id == existing.id)
.values(**values)
)
else:
db.add(DailyUsageRollupTable(id=uuid4(), **key, **values))
# =========================================================================
# MEMBER-PERFORMANCE ROLLUP (granular per-member scorecards)
# =========================================================================
async def _sweep_member_performance(self) -> None:
"""Upsert member_performance_daily from spawn sessions + audit_log.
Mirrors _sweep_daily_rollup: a trailing 7-day, overwrite-upsert sweep
(idempotent — re-running overwrites, never accumulates). Aggregates each
metric with one query keyed (date, agent_slug) into an accumulator, then
upserts one row per member per day plus one CEO row per day. Wrapped in
its own try/except so a bad member rollup never aborts the sweeper.
"""
try:
from roboco.db.base import get_session_factory
except ImportError:
return
try:
window_start = datetime.now(UTC) - timedelta(days=7)
session_factory = get_session_factory()
async with session_factory() as db:
acc: dict[tuple[Any, str], dict[str, Any]] = {}
await self._msweep_spawn(db, window_start, acc)
await self._msweep_delivery(db, window_start, acc)
await self._msweep_caused(db, window_start, acc)
await self._msweep_qa(db, window_start, acc)
await self._msweep_escalations(db, window_start, acc)
await self._msweep_blocked_others(db, window_start, acc)
await self._msweep_idle(db, window_start, acc)
await self._msweep_blocked_seconds(db, window_start, acc)
for (day, slug), fields in acc.items():
await self._upsert_member_perf_row(db, day, "agent", slug, fields)
await self._msweep_ceo(db, window_start)
await db.commit()
logger.debug("Member performance rollup complete", rows=len(acc))
except Exception as exc:
logger.warning("Member performance rollup failed", error=str(exc))
@staticmethod
def _merge_member(
acc: dict[tuple[Any, str], dict[str, Any]],
day: Any,
slug: str,
**fields: Any,
) -> None:
"""Merge one metric's aggregate into the (date, slug) accumulator entry."""
if not slug:
return
entry = acc.setdefault((day, slug), {})
for key, value in fields.items():
entry[key] = value
async def _msweep_spawn(
self,
db: Any,
window_start: datetime,
acc: dict[tuple[Any, str], dict[str, Any]],
) -> None:
"""Effort / turns / tool_calls / tokens / cost from closed spawn sessions."""
from sqlalchemy import text
sql = text(
"""
SELECT date(started_at) AS d, agent_slug AS slug, team, role,
COALESCE(SUM(EXTRACT(epoch FROM
(COALESCE(ended_at, now()) - started_at))), 0) AS active_s,
COALESCE(SUM(turns), 0) AS turns,
COALESCE(SUM(tool_calls), 0) AS tool_calls,
COALESCE(SUM(tokens_input + tokens_output
+ tokens_cache_read + tokens_cache_write), 0) AS tokens,
COALESCE(SUM(estimated_cost_usd), 0) AS cost
FROM agent_spawn_sessions
WHERE ended_at IS NOT NULL AND started_at >= :ws
GROUP BY date(started_at), agent_slug, team, role
"""
)
for r in (await db.execute(sql, {"ws": window_start})).all():
self._merge_member(
acc,
r.d,
r.slug,
team=r.team,
role=r.role,
active_runtime_seconds=int(r.active_s or 0),
turns=int(r.turns or 0),
tool_calls=int(r.tool_calls or 0),
tokens=int(r.tokens or 0),
cost_usd=float(r.cost or 0.0),
)
async def _msweep_delivery(
self,
db: Any,
window_start: datetime,
acc: dict[tuple[Any, str], dict[str, Any]],
) -> None:
"""Completed / first-pass / revisions-received per task owner per day."""
from sqlalchemy import text
sql = text(
"""
SELECT date(t.completed_at) AS d, ag.slug AS slug,
COUNT(*) AS completed,
COUNT(*) FILTER (WHERE COALESCE(t.revision_count, 0) = 0) AS first_pass,
COALESCE(SUM(t.revision_count), 0) AS received
FROM tasks t JOIN agents ag ON ag.id = t.assigned_to
WHERE t.status = 'completed' AND t.completed_at >= :ws
AND t.assigned_to IS NOT NULL
GROUP BY date(t.completed_at), ag.slug
"""
)
for r in (await db.execute(sql, {"ws": window_start})).all():
self._merge_member(
acc,
r.d,
r.slug,
tasks_completed=int(r.completed or 0),
tasks_first_pass=int(r.first_pass or 0),
revisions_received=int(r.received or 0),
)
async def _msweep_caused(
self,
db: Any,
window_start: datetime,
acc: dict[tuple[Any, str], dict[str, Any]],
) -> None:
"""Revisions caused — qa/pr fail events attributed to the rejector."""
from sqlalchemy import text
sql = text(
"""
SELECT date(al.timestamp) AS d, ag.slug AS slug, COUNT(*) AS caused
FROM audit_log al JOIN agents ag ON ag.id = al.agent_id
WHERE al.event_type IN ('task.qa_fail', 'task.pr_fail')
AND al.timestamp >= :ws
GROUP BY date(al.timestamp), ag.slug
"""
)
for r in (await db.execute(sql, {"ws": window_start})).all():
self._merge_member(acc, r.d, r.slug, revisions_caused=int(r.caused or 0))
async def _msweep_qa(
self,
db: Any,
window_start: datetime,
acc: dict[tuple[Any, str], dict[str, Any]],
) -> None:
"""QA pass-rate — passed (awaiting_documentation by qa) + failed (qa_fail)."""
from sqlalchemy import text
sql = text(
"""
SELECT date(al.timestamp) AS d, ag.slug AS slug,
COUNT(*) FILTER (
WHERE al.event_type = 'task.awaiting_documentation') AS passed,
COUNT(*) FILTER (WHERE al.event_type = 'task.qa_fail') AS failed
FROM audit_log al JOIN agents ag ON ag.id = al.agent_id
WHERE al.timestamp >= :ws AND (
(al.event_type = 'task.awaiting_documentation'
AND (al.details->>'agent_role') = 'qa')
OR al.event_type = 'task.qa_fail'
)
GROUP BY date(al.timestamp), ag.slug
"""
)
for r in (await db.execute(sql, {"ws": window_start})).all():
passed = int(r.passed or 0)
failed = int(r.failed or 0)
self._merge_member(
acc,
r.d,
r.slug,
qa_reviews_passed=passed,
qa_reviews_total=passed + failed,
)
async def _msweep_escalations(
self,
db: Any,
window_start: datetime,
acc: dict[tuple[Any, str], dict[str, Any]],
) -> None:
"""Escalations raised per member (keyed on details.escalator_slug)."""
from sqlalchemy import text
sql = text(
"""
SELECT date(timestamp) AS d,
(details->>'escalator_slug') AS slug, COUNT(*) AS n
FROM audit_log
WHERE event_type = 'task.escalated' AND timestamp >= :ws
AND (details->>'escalator_slug') IS NOT NULL
GROUP BY date(timestamp), (details->>'escalator_slug')
"""
)
for r in (await db.execute(sql, {"ws": window_start})).all():
self._merge_member(acc, r.d, r.slug, escalations=int(r.n or 0))
async def _msweep_blocked_others(
self,
db: Any,
window_start: datetime,
acc: dict[tuple[Any, str], dict[str, Any]],
) -> None:
"""Downstream tasks a member's completed task was blocking."""
from sqlalchemy import text
sql = text(
"""
SELECT date(al.timestamp) AS d, ag.slug AS slug,
COALESCE(SUM((al.details->>'count')::int), 0) AS n
FROM audit_log al
JOIN tasks t ON t.id = al.target_id
JOIN agents ag ON ag.id = t.assigned_to
WHERE al.event_type = 'task.unblocked_dependents' AND al.timestamp >= :ws
GROUP BY date(al.timestamp), ag.slug
"""
)
for r in (await db.execute(sql, {"ws": window_start})).all():
self._merge_member(acc, r.d, r.slug, blocked_others=int(r.n or 0))
async def _msweep_idle(
self,
db: Any,
window_start: datetime,
acc: dict[tuple[Any, str], dict[str, Any]],
) -> None:
"""Idle seconds — each idle mark to the member's next spawn (else now)."""
from sqlalchemy import text
sql = text(
"""
WITH idle AS (
SELECT date(al.timestamp) AS d,
(al.details->>'agent_slug') AS slug, al.timestamp AS idle_at
FROM audit_log al
WHERE al.event_type = 'agent.idle' AND al.timestamp >= :ws
AND (al.details->>'agent_slug') IS NOT NULL
)
SELECT i.d, i.slug,
COALESCE(SUM(EXTRACT(epoch FROM (
COALESCE((SELECT MIN(s.started_at) FROM agent_spawn_sessions s
WHERE s.agent_slug = i.slug AND s.started_at > i.idle_at),
now()) - i.idle_at))), 0) AS idle_s
FROM idle i GROUP BY i.d, i.slug
"""
)
for r in (await db.execute(sql, {"ws": window_start})).all():
self._merge_member(acc, r.d, r.slug, idle_seconds=int(r.idle_s or 0))
async def _msweep_blocked_seconds(
self,
db: Any,
window_start: datetime,
acc: dict[tuple[Any, str], dict[str, Any]],
) -> None:
"""Wall-clock a member's tasks spent in `blocked`, per owner per day."""
from sqlalchemy import text
sql = text(
"""
WITH ordered AS (
SELECT a.target_id, a.timestamp AS entered,
(a.details->>'to_status') AS status,
LEAD(a.timestamp) OVER (
PARTITION BY a.target_id ORDER BY a.timestamp) AS exited
FROM audit_log a
WHERE a.event_type LIKE 'task.%'
AND a.event_type = 'task.' || (a.details->>'to_status')
AND a.timestamp >= :ws
)
SELECT date(o.entered) AS d, ag.slug AS slug,
COALESCE(SUM(EXTRACT(epoch FROM
(COALESCE(o.exited, now()) - o.entered))), 0) AS blocked_s
FROM ordered o
JOIN tasks t ON t.id = o.target_id
JOIN agents ag ON ag.id = t.assigned_to
WHERE o.status = 'blocked'
GROUP BY date(o.entered), ag.slug
"""
)
for r in (await db.execute(sql, {"ws": window_start})).all():
self._merge_member(acc, r.d, r.slug, blocked_seconds=int(r.blocked_s or 0))
async def _msweep_ceo(self, db: Any, window_start: datetime) -> None:
"""Upsert one CEO row per day: approval/unblock dwell + god-mode count."""
from sqlalchemy import text
sql = text(
"""
WITH events AS (
SELECT target_id, timestamp, date(timestamp) AS d,
(details->>'to_status') AS to_status,
(details->>'agent_role') AS role
FROM audit_log
WHERE event_type LIKE 'task.%' AND timestamp >= :ws
),
approvals AS (
SELECT e.d, EXTRACT(epoch FROM ((
SELECT MIN(x.timestamp) FROM events x
WHERE x.target_id = e.target_id AND x.timestamp > e.timestamp
AND x.role = 'ceo'
AND x.to_status IN
('completed', 'needs_revision', 'cancelled', 'pending')
) - e.timestamp)) AS latency
FROM events e WHERE e.to_status = 'awaiting_ceo_approval'
),
unblocks AS (
SELECT e.d, EXTRACT(epoch FROM ((
SELECT MIN(x.timestamp) FROM events x
WHERE x.target_id = e.target_id AND x.timestamp > e.timestamp
AND x.role = 'ceo' AND x.to_status IN ('in_progress', 'pending')
) - e.timestamp)) AS latency
FROM events e WHERE e.to_status = 'blocked'
)
SELECT d,
COALESCE(SUM(approval_latency), 0) AS approval_s,
COALESCE(SUM(unblock_latency), 0) AS unblock_s,
COALESCE(SUM(godmode), 0) AS godmode
FROM (
SELECT d, latency AS approval_latency, 0 AS unblock_latency, 0 AS godmode
FROM approvals WHERE latency IS NOT NULL
UNION ALL
SELECT d, 0, latency, 0 FROM unblocks WHERE latency IS NOT NULL
UNION ALL
SELECT d, 0, 0, 1 FROM events WHERE role = 'ceo'
) u GROUP BY d
"""
)
for r in (await db.execute(sql, {"ws": window_start})).all():
await self._upsert_member_perf_row(
db,
r.d,
"ceo",
"",
{
"ceo_approval_dwell_seconds": int(r.approval_s or 0),
"ceo_unblock_dwell_seconds": int(r.unblock_s or 0),
"godmode_actions": int(r.godmode or 0),
},
)
async def _upsert_member_perf_row(
self, db: Any, day: Any, member_kind: str, slug: str, fields: dict[str, Any]
) -> None:
"""Overwrite-upsert one member_performance_daily row on the natural key."""
from uuid import uuid4 as _uuid4
from sqlalchemy import select, update
from roboco.db.tables import MemberPerformanceDailyTable
existing = (
await db.execute(
select(MemberPerformanceDailyTable).where(
MemberPerformanceDailyTable.date == day,
MemberPerformanceDailyTable.member_kind == member_kind,
MemberPerformanceDailyTable.agent_slug == slug,
)
)
).scalar_one_or_none()
if existing is not None:
await db.execute(
update(MemberPerformanceDailyTable)
.where(MemberPerformanceDailyTable.id == existing.id)
.values(**fields)
)
else:
db.add(
MemberPerformanceDailyTable(
id=_uuid4(),
date=day,
member_kind=member_kind,
agent_slug=slug,
**fields,
)
)
async def restore_waiting_records(self) -> int:
"""Load persisted waiting records into memory on orchestrator start.
Call this from `start()` so agents marked WAITING_LONG before the
previous orchestrator exited can still be resolved.
"""
try:
from sqlalchemy import select
from roboco.db.base import get_session_factory
from roboco.db.tables import WaitingRecordTable
session_factory = get_session_factory()
async with session_factory() as db:
rows = await db.execute(select(WaitingRecordTable))
count = 0
for row in rows.scalars().all():
self._waiting_records[row.agent_id] = WaitingRecord(
agent_id=row.agent_id,
task_id=str(row.task_id) if row.task_id else None,
waiting_for=row.waiting_for,
waiting_since=row.waiting_since,
context=dict(row.context or {}),
)
count += 1
if count:
logger.info(
"Restored waiting records from database",
count=count,
)
return count
except Exception as e:
logger.error("Failed to restore waiting records", error=str(e))
return 0
@staticmethod
def _partition_respawn_rows(
rows: "Iterable[Any]",
status_by_id: dict[Any, Any],
now: datetime | None = None,
) -> tuple[dict[tuple[str, str], dict[str, Any]], list[tuple[str, Any]]]:
"""Split persisted respawn rows into (restorable entries, stale keys).
Pure: a row is **stale** when its task is missing from ``status_by_id``
or terminal (completed/cancelled) — a stale counter must never resurrect
against a fixed/deleted task. Restorable entries are keyed
``(agent_slug, str(task_id))`` to match the in-memory dict; stale keys
carry the raw ``task_id`` for deletion.
F034: ``last_check`` is re-stamped to ``now`` (the restore time) on
every restorable entry. ``_pm_made_rule_following_retry`` reads
``since = record.get("last_check")`` to bound its tracing_gap audit
lookup; a stale pre-restart ``last_check`` would match a pre-restart
tracing_gap row and falsely reset the breaker on the first post-restart
spawn. Re-stamping bounds the lookup to post-restart gaps only.
"""
from roboco.models.base import TaskStatus
restore_now = now or datetime.now(UTC)
terminal = {TaskStatus.COMPLETED.value, TaskStatus.CANCELLED.value}
restored: dict[tuple[str, str], dict[str, Any]] = {}
stale: list[tuple[str, Any]] = []
for r in rows:
status = status_by_id.get(r.task_id)
norm = getattr(status, "value", status)
if status is None or norm in terminal:
stale.append((r.agent_slug, r.task_id))
continue
restored[(r.agent_slug, str(r.task_id))] = {
"count": r.count,
# Re-stamp to the LIVE status (mirrors the last_check re-stamp
# above): a pre-restart last_status is as stale w.r.t. post-restart
# reality, and a status mismatch across the restart gap would
# otherwise disarm the breaker on the first post-restart spawn and
# re-burn the whole strike threshold against a still-wedged task.
"last_status": norm,
"last_check": restore_now,
"tracing_resets": r.tracing_resets,
"revisit_resets": r.revisit_resets,
"notified": r.notified,
}
return restored, stale
async def restore_respawn_tracker(self) -> int:
"""Load the persisted PM-respawn counter into memory on startup.
Mirrors ``restore_waiting_records``: read every ``respawn_tracker`` row,
keep only those whose task is still live and non-terminal, evict the
rest, and populate ``_pm_respawn_tracker`` so a wedged-task counter trips
at its persisted threshold instead of resetting to 1 and re-burning the
whole budget. Best-effort — any failure starts with an empty tracker
(exactly today's behaviour) and never blocks startup.
"""
try:
from sqlalchemy import select
from roboco.db.base import get_session_factory
from roboco.db.tables import RespawnTrackerTable, TaskTable
session_factory = get_session_factory()
async with session_factory() as db:
rows = (await db.execute(select(RespawnTrackerTable))).scalars().all()
if not rows:
return 0
ids = [r.task_id for r in rows]
live = (
await db.execute(
select(TaskTable.id, TaskTable.status).where(
TaskTable.id.in_(ids)
)
)
).all()
status_by_id = {row.id: row.status for row in live}
restored, stale = self._partition_respawn_rows(rows, status_by_id)
self._pm_respawn_tracker.update(restored)
for agent_slug, task_id in stale:
await self._clear_respawn_record(agent_slug, str(task_id))
if restored:
logger.info(
"Restored PM-respawn records from database",
count=len(restored),
evicted=len(stale),
)
return len(restored)
except Exception as e:
logger.error("Failed to restore respawn records", error=str(e))
return 0
async def resolve_wait(
self,
agent_id: str,
resolution: dict[str, Any],
) -> AgentInstance | None:
"""
Resolve a wait condition and respawn the agent.
Args:
agent_id: The waiting agent
resolution: Details about the resolution
Returns:
Respawned AgentInstance or None
"""
if agent_id not in self._waiting_records:
return None
# #71: a lingering record (a prior resume whose liveness confirmation
# hasn't torn it down yet) must not double-spawn an already-active agent.
if self._is_agent_active(agent_id):
return None
record = self._waiting_records[agent_id]
# Generate resume prompt
resume_prompt = self._generate_resume_prompt(record, resolution)
# Preserve the original git_context from the prior instance so the
# respawned agent keeps the same workspace mount path.
prior = self._instances.get(agent_id)
prior_git_context = prior.config.git_context if prior and prior.config else None
# Respawn FIRST, then tear down the record only once a container actually
# launched. The old order deleted the record (in-memory + durable) before
# the spawn: a re-park during the resume window — the provider's rate limit
# lifts then immediately re-limits, or a second provider limit lands —
# bails spawn with an OFFLINE instance (the parked-provider short-circuit),
# and deleting the record first orphaned the agent. With no record the
# probe-resume loop can never revive it and the spawn gate bails every
# tick, so the agent is lost until the operator intervenes. Keeping the
# record through a bail lets the next probe-success re-attempt the resume.
try:
instance = await self.spawn_agent(
agent_id=agent_id,
initial_prompt=resume_prompt,
task_id=record.task_id,
git_context=prior_git_context,
spawned_by="resolve_wait",
)
except Exception:
# Spawn failed (e.g. readiness refused → task auto-blocked). Tear
# down the record so the probe loop doesn't keep re-resuming a task
# that has moved to a different state; the blocked-task path takes
# over. This matches the pre-fix behavior where the record was
# deleted before the spawn attempt.
del self._waiting_records[agent_id]
await self._delete_waiting_record(agent_id)
raise
if instance is None or instance.state == AgentState.OFFLINE:
# Spawn bailed without launching (provider re-parked). Keep the record
# so the probe-resume loop re-attempts on the next clear.
return instance
if record.waiting_for == "rate_limit_lifted":
# #71: don't tear down the record on a bare launch — a container that
# launches then dies immediately would orphan the task until the
# reaper's TTL. Keep the record past the launch and confirm liveness
# in the background; if the container dies the probe-resume orphan
# fallback re-resumes within a tick instead of waiting the full TTL.
self._schedule_bg(self._confirm_resume_liveness(agent_id))
return instance
del self._waiting_records[agent_id]
await self._delete_waiting_record(agent_id)
return instance
async def _confirm_resume_liveness(self, agent_id: str) -> None:
"""Tear down a resumed agent's WaitingRecord once it is confirmed alive.
A container that launches then dies immediately must not strand its task
until the reaper's TTL: the record is kept past the launch (``resolve_wait``
schedules this) and deleted only once the agent is still active past a
short confirmation window. If the container died, the record survives so
the probe-resume orphan fallback re-resumes on the next tick (#71). The
confirmation reads ``_is_agent_active`` — the same signal the spawn gate
trusts — so a container the health loop has marked dead keeps its record.
Best-effort: a delete error is swallowed (the in-memory record is gone
either way once the process exits, and the orphan fallback is in-memory).
"""
if agent_id not in self._waiting_records:
return
await asyncio.sleep(self._resume_confirm_delay)
if not self._is_agent_active(agent_id):
return # container died — keep the record for the orphan fallback
del self._waiting_records[agent_id]
try:
await self._delete_waiting_record(agent_id)
except Exception:
logger.warning(
"resume-liveness confirm failed to delete the durable record",
agent_id=agent_id,
)
def _generate_resume_prompt(
self,
record: WaitingRecord,
resolution: dict[str, Any],
) -> str:
"""Generate a resume prompt for a respawning agent."""
if record.waiting_for == "blocker_resolution":
return f"""
You were working on TASK-{record.task_id} and got blocked.
The blocker has been resolved: {resolution.get("details", "Resolved")}
Resume by:
1. Reading your checkpoint from .tasks/active/TASK-{record.task_id}/
2. Call unblock("{record.task_id}")
3. Continue from where you left off
"""
elif record.waiting_for == "qa_result":
if resolution.get("passed"):
return f"""
TASK-{record.task_id} has passed QA review.
The task is now awaiting documentation.
You may return to scanning for new work with give_me_work().
"""
else:
return f"""
TASK-{record.task_id} needs revision based on QA feedback.
QA notes: {resolution.get("notes", "See task for details")}
Resume by:
1. Reading the QA feedback
2. Updating your TODOs to address each issue
3. Making the fixes
4. Re-submitting for QA
"""
elif record.waiting_for == "answer":
return f"""
You asked a question about TASK-{record.task_id}:
Your question: {record.context.get("question", "Unknown")}
Answer received: {resolution.get("answer", "Unknown")}
Resume by incorporating this information and continuing from where you stopped.
"""
elif record.waiting_for == "assignment":
return f"""
You have been assigned a new task: TASK-{resolution.get("task_id")}
Start by:
1. Review the task details provided in your briefing / context_briefing
2. Follow the standard workflow: UNDERSTAND → PLAN → EXECUTE → VERIFY → NOTES
"""
else:
return f"Resuming. Wait condition '{record.waiting_for}' resolved."
# =========================================================================
# HEALTH MONITORING
# =========================================================================
async def _health_loop(self) -> None:
"""Background health check loop."""
while self._running:
try:
await asyncio.sleep(30) # Check every 30 seconds
await self._check_health()
except asyncio.CancelledError:
break
except Exception as e:
logger.error("Health check error", error=str(e))
async def _sweeper_loop(self) -> None:
"""Background sweeper for stale notifications + runtime maintenance.
Addresses a silent-failure surface (NotificationTable.expires_at existed
but no job ever acted on it) and drives the budget kill-switch, token
rollups, transcript retention, and dangling-image pruning.
Runs on its own interval so a slow sweep can't delay agent dispatch.
"""
sweep_interval = 60 # seconds
while self._running:
try:
await asyncio.sleep(sweep_interval)
await self._run_sweep()
except asyncio.CancelledError:
break
except Exception as e:
logger.error("Sweeper loop error", error=str(e))
async def _run_sweep(self) -> None:
"""Run one pass of the notification sweeper + runtime maintenance."""
from roboco.db.base import get_session_factory
from roboco.services.notification_delivery import (
get_notification_delivery_service,
)
session_factory = get_session_factory()
async with session_factory() as db:
deliv_svc = get_notification_delivery_service(db)
try:
expired = await deliv_svc.sweep_expired_notifications()
if expired:
await db.commit()
except Exception as e:
await db.rollback()
logger.warning("Notification sweep failed", error=str(e))
# Retire abandoned live intake/secretary chats (idle past the threshold)
# so a closed-tab session doesn't leak its container until restart.
await self._reap_idle_interactive_sessions()
# Budget kill-switch — runs every sweep. Any agent whose SDK reports
# halt=true has breached its per-session tool-call cap; terminate the
# container so the next dispatcher tick doesn't waste tokens on the
# same session.
await self._sweep_budget_exceeded()
# Token-usage instrumentation: snapshot active agents and roll up
# closed sessions into the daily aggregation table.
await self._sweep_token_snapshots()
await self._sweep_daily_rollup()
# Granular per-member performance rollup (own try/except inside).
await self._sweep_member_performance()
# Prune old agent transcripts (throttled internally to ~hourly) so the
# operator's bind-mounted ~/.claude doesn't grow without bound.
await self._sweep_transcript_retention()
# Prune dangling (<none>) Docker images left by agent-image rebuilds
# (throttled internally to ~6h) so deploys don't pile up orphaned layers.
await self._sweep_dangling_images()
# Close-on-land for landed supersedes — runs here (always-on sweeper)
# rather than the default-off external-PR poll loop, so a supersede that
# lands after external_pr_enabled is toggled off is still reconciled.
await self._sweep_superseded_prs()
async def _sweep_superseded_prs(self) -> None:
"""Retire the contributor PR for any supersede umbrella that landed.
Dormant in a standard deployment: when no ``external_pr_supersede``
umbrellas exist the lookup returns nothing and no GitHub call is made,
so this is safe to run unconditionally on every sweep.
"""
from roboco.db.base import get_session_factory
from roboco.services.git import GitService
from roboco.services.task import get_task_service
system_id = _foundation.AGENTS["system"].uuid
session_factory = get_session_factory()
async with session_factory() as db:
try:
git = GitService(db)
task_service = get_task_service(db)
closed = await self._close_superseded_prs(git, task_service, system_id)
if closed:
await db.commit()
except Exception as e:
await db.rollback()
logger.warning("Supersede close-on-land sweep failed", error=str(e))
async def _sweep_dangling_images(self) -> None:
"""Prune dangling (<none>) Docker images left by agent-image rebuilds.
Each rebuild of an agent image orphans the prior build's layers as an
untagged ``<none>`` image; over many deploys these pile up (the operator
saw ~80). Pruning only DANGLING images is safe — a tagged image, or one
backing a running container, is never dangling. Throttled to
``settings.image_prune_interval_seconds`` (default 6h) and gated by
``settings.image_prune_enabled`` (default on). Best-effort: any failure
is logged, never raised into the sweeper.
"""
if not settings.image_prune_enabled:
return
now = datetime.now(UTC)
last = self._last_image_prune
if (
last is not None
and (now - last).total_seconds() < settings.image_prune_interval_seconds
):
return
self._last_image_prune = now
try:
proc = await asyncio.create_subprocess_exec(
"docker",
"image",
"prune",
"-f",
"--filter",
"dangling=true",
stdout=asyncio.subprocess.PIPE,
stderr=asyncio.subprocess.DEVNULL,
)
stdout, _ = await proc.communicate()
if proc.returncode == 0:
summary = stdout.decode().strip().splitlines()[-1:] if stdout else []
logger.info(
"pruned dangling images", reclaimed=summary[0] if summary else ""
)
else:
logger.warning("dangling-image prune returned non-zero")
except Exception as e:
logger.warning("dangling-image prune failed (best-effort)", error=str(e))
async def _sweep_transcript_retention(self) -> None:
"""Prune agent transcripts older than the retention window.
Throttled to ``settings.transcript_prune_interval_seconds``. Reads the
window from the ``system_settings`` store (panel-editable), falling back
to ``settings.transcript_retention_days``. Only agent-owned project dirs
(``-app`` + per-workspace dirs) are touched — never the operator's own
Claude sessions. Best-effort: any failure is logged, never raised.
"""
if not settings.transcript_prune_enabled:
return
now = datetime.now(UTC)
last = self._last_transcript_prune
if (
last is not None
and (now - last).total_seconds()
< settings.transcript_prune_interval_seconds
):
return
self._last_transcript_prune = now
retention_days = settings.transcript_retention_days
with contextlib.suppress(Exception):
from roboco.db.base import get_session_factory
from roboco.services.settings import get_settings_service
session_factory = get_session_factory()
async with session_factory() as db:
retention_days = await get_settings_service(db).get_int(
"transcript_retention_days", settings.transcript_retention_days
)
from roboco.runtime.transcript_retention import select_prunable_transcripts
projects_root = Path.home() / ".claude" / "projects"
cutoff = (now - timedelta(days=retention_days)).timestamp()
prunable = select_prunable_transcripts(
projects_root, settings.workspaces_root, cutoff
)
pruned = 0
for transcript in prunable:
try:
transcript.unlink()
pruned += 1
except OSError as exc:
logger.debug(
"Transcript prune failed", path=str(transcript), error=str(exc)
)
if pruned:
logger.info(
"Pruned old agent transcripts",
count=pruned,
retention_days=retention_days,
)
@staticmethod
async def _fetch_budget_status(
client: httpx.AsyncClient, url: str, agent_id: str
) -> dict[str, Any] | None:
"""Read an agent's SDK budget status; None if unreachable/not-JSON.
The SDK being unreachable is benign (container not yet started, already
gone, or a transient blip) and the health loop covers genuine failures,
so the failure is swallowed — but logged at debug so it is observable
rather than silent (the bare try/except/continue it replaced was not).
"""
try:
resp = await client.get(url)
except httpx.HTTPError as exc:
logger.debug(
"Budget status unreachable; skipping agent this sweep",
agent_id=agent_id,
error=str(exc),
)
return None
if resp.status_code != http_status.HTTP_200_OK:
return None
try:
data = resp.json()
except ValueError as exc:
logger.debug(
"Budget status not JSON; skipping agent this sweep",
agent_id=agent_id,
error=str(exc),
)
return None
return data if isinstance(data, dict) else None
async def _sweep_budget_exceeded(self) -> None:
"""Stop agents whose per-session SDK budget reports halt=true.
Each agent's SDK server is reachable at
`http://roboco-agent-{agent_id}:9000/budget/status` on the shared
agent network. A budget-exceeded agent gets a forced stop with a
`budget_exceeded` reason; the task is already being auto-substituted
by the post-tool hook on the agent side.
"""
if not self._instances:
return
async with httpx.AsyncClient(
timeout=3.0, headers=_system_api_headers()
) as client:
for agent_id, instance in list(self._instances.items()):
if instance.state not in (
AgentState.ACTIVE,
AgentState.WAITING_SHORT,
):
continue
url = f"http://roboco-agent-{agent_id}:{SDK_PORT}/budget/status"
data = await self._fetch_budget_status(client, url, agent_id)
if data is None or not data.get("halt"):
continue
logger.warning(
"Agent budget exceeded; terminating container",
agent_id=agent_id,
total_calls=data.get("total"),
halt_threshold=data.get("halt_threshold"),
)
try:
# release_claim=True: a budget-exceeded agent is terminated
# for cost overruns and will not continue its task, so hand
# the claim back to the pool now instead of waiting for the
# reaper's TTL.
await self.stop_agent(
agent_id,
graceful=True,
release_claim=True,
stop_reason="budget_sweep",
)
except Exception as e:
logger.warning(
"Failed to stop budget-exceeded agent",
agent_id=agent_id,
error=str(e),
)
@staticmethod
async def _inspect_container_state(
container_name: str,
) -> tuple[bool, int | None]:
"""Return (is_running, exit_code) from `docker inspect`.
exit_code is None when the output is missing or unparseable; the
caller treats None as a crash for safety.
"""
proc = await asyncio.create_subprocess_exec(
"docker",
"inspect",
"-f",
"{{.State.Running}} {{.State.ExitCode}}",
container_name,
stdout=asyncio.subprocess.PIPE,
stderr=asyncio.subprocess.DEVNULL,
)
try:
stdout, _ = await asyncio.wait_for(
proc.communicate(), timeout=_DOCKER_INSPECT_TIMEOUT_SECONDS
)
except TimeoutError:
proc.kill()
raise
parts = stdout.decode().strip().split()
is_running = bool(parts) and parts[0] == "true"
try:
exit_code = int(parts[1]) if len(parts) > 1 and parts[1] else None
except ValueError:
exit_code = None
return is_running, exit_code
@staticmethod
async def _resolve_container_id(container_name: str) -> str | None:
"""Return the Docker container id for ``container_name`` via `docker inspect`.
Used at startup re-adoption (F033) so a re-adopted ACTIVE instance
carries the real container id — ``_check_health`` skips
``container_id is None`` instances, so without it a later container exit
is invisible to the health loop and the task strands. Returns ``None``
when the id can't be resolved (caller treats that as best-effort
degraded re-adoption, still covered by the reaper's liveness fallback).
"""
proc = await asyncio.create_subprocess_exec(
"docker",
"inspect",
"-f",
"{{.Id}}",
container_name,
stdout=asyncio.subprocess.PIPE,
stderr=asyncio.subprocess.DEVNULL,
)
try:
stdout, _ = await asyncio.wait_for(
proc.communicate(), timeout=_DOCKER_INSPECT_TIMEOUT_SECONDS
)
except TimeoutError:
proc.kill()
raise
cid = stdout.decode().strip()
return cid or None
@staticmethod
async def _probe_gateway_health(slug: str) -> bool | None:
"""Probe an agent container's gateway out-of-band: healthy / broken / unknown.
The heartbeat only proves a verb fired recently; it cannot tell a quiet-
but-healthy agent from one whose MCP gateway is broken (e.g. a corrupted
``/app/.venv`` so every gateway tool import raises) yet whose container is
still up. This asks the container directly whether the gateway venv imports
its core deps. Returns True (healthy), False (the import failed => broken
gateway), or None when the probe itself could not run (no docker, container
gone) so the caller declines to act on an inconclusive probe.
"""
try:
proc = await asyncio.create_subprocess_exec(
"docker",
"exec",
f"roboco-agent-{slug}",
"/app/.venv/bin/python",
"-c",
"import httpx, mcp",
stdout=asyncio.subprocess.DEVNULL,
stderr=asyncio.subprocess.DEVNULL,
)
except Exception:
return None
try:
rc = await asyncio.wait_for(
proc.wait(), timeout=_DOCKER_EXEC_TIMEOUT_SECONDS
)
except TimeoutError:
# A hung docker exec is inconclusive (the probe could not run to
# completion): kill the child and decline to act, matching the
# existing probe-failure contract. The next grace tick retries.
proc.kill()
return None
except Exception:
return None
return rc == 0
async def _maybe_park_for_exit_error(
self, agent_id: str, instance: Any, graceful: bool
) -> bool:
"""Park the provider on a session/usage limit or a server overload detected
in the dead container's output, instead of crash-retrying into it. Returns
True when parked (caller returns); False to proceed with normal handling.
The probe-resume loop revives the task when the limit lifts / overload clears.
"""
if graceful:
return False
rate_limited_provider = await self._provider_rate_limit_park_target(
agent_id, instance
)
if rate_limited_provider is not None:
logger.warning(
"Session/usage limit detected in agent output; parking provider",
agent_id=agent_id,
provider=rate_limited_provider,
task_id=instance.current_task_id,
)
await self._park_provider_unavailable(
agent_id,
instance,
provider=rate_limited_provider,
retry_after=_RATE_LIMIT_RETRY_AFTER_S,
kind="rate_limited",
)
return True
overloaded_provider = await self._provider_overload_park_target(
agent_id, instance
)
if overloaded_provider is not None:
logger.warning(
"Provider overload detected in agent output; parking provider",
agent_id=agent_id,
provider=overloaded_provider,
task_id=instance.current_task_id,
)
await self._park_provider_unavailable(
agent_id,
instance,
provider=overloaded_provider,
retry_after=_OVERLOAD_RETRY_AFTER_S,
kind="overloaded",
)
return True
return False
async def _handle_stopped_container(
self, agent_id: str, instance: Any, exit_code: int | None
) -> None:
"""Update state + auto-restart only when the exit was non-zero.
Graceful exits (exit 0 — agent called i_am_idle)
were treated as crashes by the old logic. The health check bumped
error_count and respawned the agent with the prior task_id even if
the task had since moved into a state the role can't claim from
(e.g. QA → needs_revision). Now: clean exits reset error_count and
do nothing; non-zero exits keep the existing crash-retry behaviour.
"""
cid = instance.container_id[:12] if instance.container_id else None
# Grok 429 parking (B4): a one-shot grok run that hit an xAI 429 exits
# 75 (set by grok-cli-agent-entrypoint.sh). Park the provider instead of
# crash-retrying so the spawn guard suppresses the respawn loop; the
# probe-resume loop revives the task when the limit lifts.
if self._is_grok_rate_limit_exit(instance, exit_code):
await self._park_grok_rate_limited(agent_id, instance)
return
# Grok auth-missing parking (F041): a one-shot grok run whose entrypoint
# found the token missing/expired exits 78 (EX_CONFIG). Park the provider
# instead of crash-retrying — the agent can't start without a valid token,
# so respawning burns tokens for zero progress. The probe-resume loop
# revives the task once grok_auth.refresh_if_stale mints a fresh token.
if self._is_grok_auth_exit(instance, exit_code):
await self._park_grok_auth_unavailable(agent_id, instance)
return
graceful = exit_code == 0
# Park the provider on a session/usage limit or a server overload detected
# in the dead container's output instead of crash-retrying into it. The
# probe-resume loop revives the task when the limit lifts / overload clears.
if await self._maybe_park_for_exit_error(agent_id, instance, graceful):
return
if graceful:
logger.info(
"Agent container exited gracefully",
agent_id=agent_id,
container_id=cid,
exit_code=exit_code,
)
else:
await self._log_stopped_container(agent_id, cid, exit_code)
# The agent self-exited (a graceful i_am_idle shutdown, or a crash), so
# stop_agent() — which normally finalizes — was never called. Finalize
# here to capture token usage from the transcript; otherwise the
# spawn-session row is left open with zero tokens.
await self._finalize_spawn_session(
agent_id, exit_reason="completed" if graceful else "crashed"
)
instance.state = AgentState.OFFLINE
instance.container_id = None
if graceful:
instance.error_count = 0
return
await self._crash_retry_or_escalate(agent_id, instance)
async def _log_stopped_container(
self, agent_id: str, container_id: str | None, exit_code: int | None
) -> None:
"""Log a non-graceful exit, attributed via the expected-stop breadcrumb.
A fresh breadcrumb (recorded by an orchestrator-initiated stop/kill
path — see ``_record_expected_stop``) means this death is explained:
log it at info as "(expected)" so the warning line stays meaningful
for genuinely unattributed SIGTERMs/crashes. Inspect diagnostics are
best-effort and never block the log line.
"""
reason = self._consume_expected_stop(agent_id)
diagnostics = await self._inspect_exit_diagnostics(f"roboco-agent-{agent_id}")
expected = reason != "none_recorded"
log = logger.info if expected else logger.warning
log(
"Agent container stopped (expected)"
if expected
else "Agent container stopped unexpectedly",
agent_id=agent_id,
container_id=container_id,
exit_code=exit_code,
expected_stop_reason=reason,
**diagnostics,
)
async def _crash_retry_or_escalate(self, agent_id: str, instance: Any) -> None:
"""A crashed (non-graceful) agent: auto-restart up to a cap, then escalate.
Bumps error_count and respawns while under the cap; at exactly the cap
escalates once to humans (subsequent crashes stay quiet to avoid spam).
"""
instance.error_count += 1
max_retries = 3
if instance.error_count < max_retries:
logger.info("Auto-restarting crashed agent", agent_id=agent_id)
await self.spawn_agent(
agent_id=agent_id,
task_id=instance.current_task_id,
git_context=(instance.config.git_context if instance.config else None),
spawned_by="_crash_retry_or_escalate",
)
elif instance.error_count == max_retries:
# Exactly at the threshold — escalate once to humans so a
# stranded agent doesn't die silently. Subsequent crashes
# stay quiet to avoid notification spam.
logger.error(
"Agent exceeded max restart attempts; escalating",
agent_id=agent_id,
error_count=instance.error_count,
task_id=instance.current_task_id,
)
await self._notify_agent_stranded(
agent_id=agent_id,
error_count=instance.error_count,
task_id=instance.current_task_id,
)
async def _check_health(self) -> None:
"""Check health of all running agents."""
for agent_id, instance in list(self._instances.items()):
if instance.state not in (AgentState.ACTIVE, AgentState.WAITING_SHORT):
continue
if instance.container_id is None:
continue
# A per-agent docker-inspect timeout (or any docker error) must skip
# THIS agent, not abort the whole sweep — otherwise one hung daemon
# call means no agent gets health-checked this tick. The reaper's
# own liveness fallback still covers a genuinely-stopped container
# next tick; skipping is the safe fail-direction.
try:
is_running, exit_code = await self._inspect_container_state(
f"roboco-agent-{agent_id}"
)
except Exception as exc:
logger.debug(
"container inspect failed; skipping agent this tick",
agent_id=agent_id,
error=str(exc),
)
continue
if not is_running:
await self._handle_stopped_container(agent_id, instance, exit_code)
self._check_loop_liveness()
async def _notify_agent_stranded(
self,
agent_id: str,
error_count: int,
task_id: str | None,
) -> None:
"""Create a notification for humans when an agent can't be restarted.
Posts a high-priority notification addressed to the auditor and CEO.
Fire-and-forget: the agent is already dead; don't let our own failure
stop the health loop.
"""
try:
from sqlalchemy import select
from roboco.db.base import get_session_factory
from roboco.db.tables import AgentTable, NotificationTable
from roboco.models.base import (
AgentRole,
NotificationPriority,
NotificationType,
)
from roboco.services.notification_delivery import (
get_notification_delivery_service,
)
from roboco.utils.converters import require_uuid
session_factory = get_session_factory()
async with session_factory() as db:
orch_agent = await db.execute(
select(AgentTable).where(AgentTable.role == AgentRole.AUDITOR)
)
auditor = orch_agent.scalar_one_or_none()
ceo_result = await db.execute(
select(AgentTable).where(AgentTable.role == AgentRole.CEO)
)
ceo = ceo_result.scalar_one_or_none()
recipients = [a.id for a in (auditor, ceo) if a is not None]
if not recipients:
logger.warning(
"No auditor/ceo found for stranded-agent notification",
agent_id=agent_id,
)
return
# recipients is non-empty (guarded above) and already holds the
# non-None ids in auditor-then-ceo order — its first entry is the
# same value as `auditor.id if auditor else ceo.id`, without the
# union-narrowing mypy can't prove.
from_agent = recipients[0]
notification = NotificationTable(
type=NotificationType.ALERT,
priority=NotificationPriority.HIGH,
from_agent=from_agent,
to_agents=recipients,
subject=f"Agent stranded: {agent_id}",
body=(
f"Agent '{agent_id}' exceeded max restart attempts "
f"({error_count}) and will not auto-recover. "
f"Task: {task_id or 'none'}. Manual intervention needed."
),
requires_ack=True,
)
db.add(notification)
await db.flush()
delivery = get_notification_delivery_service(db)
await delivery.deliver(require_uuid(notification.id))
await db.commit()
except Exception as e:
logger.error(
"Failed to send stranded-agent notification",
agent_id=agent_id,
error=str(e),
)
# =========================================================================
# RATE-LIMIT PROBE LOOP
# =========================================================================
async def _strategy_engine_loop(self) -> None:
"""Engine 2: periodically surface goal drift / idle / stranded work.
Dormant by default — returns immediately unless ``strategy_engine_enabled``
is set, so it adds zero behaviour to a standard deployment. Notify-only;
it never spends or builds. A persistently failing cycle surfaces to the
CEO once per failure episode (#193) instead of silently logging forever.
"""
if not settings.strategy_engine_enabled:
return
state = self._new_strategy_loop_state()
interval = settings.strategy_engine_interval_seconds
while self._running:
try:
await asyncio.sleep(interval)
await self._strategy_engine_cycle(state)
except asyncio.CancelledError:
break
@staticmethod
def _new_strategy_loop_state() -> _StrategyLoopState:
return _StrategyLoopState()
async def _strategy_engine_cycle(self, state: _StrategyLoopState) -> None:
"""Run one strategy-engine pass; track consecutive failures (#193).
On success the failure state resets (a fresh failure episode later
re-notifies). On a non-cancel failure, count it and notify the CEO once
per episode past ``_STRATEGY_FAIL_CEO_NOTIFY_THRESHOLD``.
"""
from roboco.db import get_db_context
from roboco.services.strategy_engine import get_strategy_engine
try:
async with get_db_context() as db:
await get_strategy_engine(db).run_cycle()
state.failures = 0
state.notified = False
except asyncio.CancelledError:
raise
except Exception:
logger.exception("strategy engine cycle failed")
state.failures += 1
if (
state.failures >= _STRATEGY_FAIL_CEO_NOTIFY_THRESHOLD
and not state.notified
):
state.notified = True
await self._notify_strategy_engine_failure(state.failures)
async def _notify_strategy_engine_failure(self, fail_count: int) -> None:
"""Send one CEO alert that the strategy engine is persistently failing."""
try:
from roboco.services.notification import NotificationService
await NotificationService().send_ack_notification(
from_agent="system",
to_agent="ceo",
body=(
"[strategy engine] persistently failing: the last "
f"{fail_count} cycles raised and produced no "
"observations. Check the orchestrator logs."
),
)
except Exception:
logger.exception("strategy engine failure-notify dropped")
async def _external_pr_poll_loop(self) -> None:
"""Engine 3: discover inbound PRs and open review tasks.
Dormant by default — returns immediately unless ``external_pr_enabled``
OR ``internal_pr_enabled``, so a standard deployment makes no inbound
GitHub call. This only lists open PRs and records a review task per
newly-seen reviewable one (external/fork PRs, and — when internal review
is on — org-repo PRs not tied to an active task); it never fetches or
runs contributor code (that waits on a human confirmation downstream).
New review tasks wake the dispatcher.
"""
if not (settings.external_pr_enabled or settings.internal_pr_enabled):
return
from roboco.db import get_db_context
interval = settings.external_pr_poll_interval_seconds
while self._running:
try:
await asyncio.sleep(interval)
async with get_db_context() as db:
ingested = await self._poll_external_prs_once(db)
if ingested:
self._dispatch_wake.set()
except asyncio.CancelledError:
break
except Exception:
logger.exception("external-PR poll cycle failed")
async def _self_heal_loop(self) -> None:
"""Engine 4: watch RoboCo's OWN CI, surface regressions, open fix tasks.
Dormant by default — returns immediately unless ``self_heal_enabled``, so
a standard deployment makes no CI call and adds zero behaviour. It only
NOTIFIES the CEO and (behind ``self_heal_originate_enabled``) opens a
PENDING fix task into RoboCo's own lifecycle; it never starts, merges, or
deploys. The per-cycle session commits any opened task here.
"""
if not settings.self_heal_enabled:
return
from roboco.db import get_db_context
from roboco.services.self_heal_engine import get_self_heal_engine
# Operability: self-heal is armed but has no target → it will silently
# no-op every cycle. Say so once at startup so a misconfiguration (unset
# or wrong ROBOCO_SELF_HEAL_PROJECT_SLUG) isn't mistaken for "all green".
if not settings.self_heal_project_slug.strip():
logger.warning(
"self-heal enabled but self_heal_project_slug is unset — the loop "
"will not detect anything until the target project is configured"
)
interval = settings.self_heal_interval_seconds
self._record_loop_heartbeat("self_heal", interval)
while self._running:
try:
await asyncio.sleep(interval)
async with get_db_context() as db:
await get_self_heal_engine(db).run_cycle()
await db.commit()
self._record_loop_heartbeat("self_heal", interval)
except asyncio.CancelledError:
break
except Exception:
logger.exception("self-heal cycle failed")
async def _ci_watch_loop(self) -> None:
"""Multi-repo CI-watch: watch every opted-in project's CI, open fix tasks.
Dormant by default — returns immediately unless ``ci_watch_enabled``, so
a standard deployment adds zero behaviour. It generalizes the single-repo
self-heal loop (which is untouched) to every project with
``ci_watch_enabled`` set; like self-heal it only OPENS a fix task and
never starts / approves / merges / deploys. The per-cycle session commits
any opened task here.
"""
if not settings.ci_watch_enabled:
return
interval = settings.ci_watch_interval_seconds
self._record_loop_heartbeat("ci_watch", interval)
while self._running:
try:
await asyncio.sleep(interval)
await self._run_ci_watch_cycle()
self._record_loop_heartbeat("ci_watch", interval)
except asyncio.CancelledError:
break
except Exception:
logger.exception("ci-watch cycle failed")
async def _run_ci_watch_cycle(self) -> None:
"""One CI-watch pass: load the watch set, run the engine, commit.
Extracted from the loop so it is testable without the sleep. A loud
warning fires when CI-watch is armed but no project opted in (so a
misconfiguration isn't mistaken for "all green").
"""
from roboco.db import get_db_context
from roboco.services.ci_watch_engine import get_ci_watch_engine
async with get_db_context() as db:
watch_set = await self._load_ci_watch_set(db)
if not watch_set:
logger.warning(
"ci-watch enabled but no project has ci_watch_enabled — "
"nothing to watch"
)
return
await get_ci_watch_engine(db).run_cycle(watch_set)
await db.commit()
async def _load_ci_watch_set(self, db: Any) -> list[Any]:
"""Opted-in projects (``ci_watch_enabled`` + a git_url), one per
(repo, workflow).
A monorepo's several cell-projects can each carry their OWN
``ci_watch_workflow`` (e.g. a backend CI workflow distinct from the
frontend's). Collapsing to one canonical project per REPO would watch
only the canonical cell's workflow and miss a red on the others (the
under-count). Collapse to one canonical project per (repo, effective
workflow) instead — every distinct workflow is sampled once, and the
engine's per-``git_url`` fix-task dedup still prevents a duplicate fix
task for the same repo. The effective workflow is the project override
or ``ci_watch_default_workflow`` (matching ``MultiProjectCITelemetrySource``).
"""
from roboco.services.project import get_project_service
projects = await get_project_service(db).list_all(active_only=True)
watched = [
p
for p in projects
if getattr(p, "ci_watch_enabled", False) and getattr(p, "git_url", None)
]
return self._projects_one_per_key(
watched,
key_fn=lambda p: (
self._repo_key(str(getattr(p, "git_url", "") or "")),
self._effective_ci_watch_workflow(p),
),
)
@staticmethod
def _effective_ci_watch_workflow(project: Any) -> str | None:
"""The workflow that will actually be polled for ``project``.
Mirrors ``MultiProjectCITelemetrySource._sample_for``: the project's
``ci_watch_workflow`` override, falling back to the global
``ci_watch_default_workflow``. Used as the per-(repo, workflow) collapse
key so two cells sharing a workflow still collapse to one sample.
"""
workflow = str(
getattr(project, "ci_watch_workflow", None)
or settings.ci_watch_default_workflow
).strip()
return workflow or None
async def _dep_update_loop(self) -> None:
"""Dependency-update bot: probe opted-in projects, open update tasks.
Dormant by default — returns immediately unless ``dep_update_enabled``.
Each interval (default weekly) it loads projects with a
``dep_update_command``, collapses to one per repo, and runs
``DepUpdateEngine.run_cycle``; it only OPENS a task and never starts /
approves / merges / deploys. Separate from the self-heal and CI-watch
loops.
"""
if not settings.dep_update_enabled:
return
interval = settings.dep_update_interval_seconds
self._record_loop_heartbeat("dep_update", interval)
while self._running:
try:
await asyncio.sleep(interval)
await self._run_dep_update_cycle()
self._record_loop_heartbeat("dep_update", interval)
except asyncio.CancelledError:
break
except Exception:
logger.exception("dep-update cycle failed")
async def _run_dep_update_cycle(self) -> None:
"""One dep-update pass: load eligible projects, run the engine, commit.
Extracted from the loop so it is testable without the sleep. Warns when
the bot is armed but no project has a ``dep_update_command`` set.
"""
from roboco.db import get_db_context
from roboco.services.dep_update_engine import get_dep_update_engine
async with get_db_context() as db:
projects = await self._load_dep_update_set(db)
if not projects:
logger.warning(
"dep-update enabled but no project has a dep_update_command — "
"nothing to probe"
)
return
await get_dep_update_engine(db).run_cycle(projects)
await db.commit()
async def _release_manager_loop(self) -> None:
"""Gated release manager: at a logical point, propose a CEO-gated release.
Dormant by default — returns immediately unless ``release_manager_enabled``,
so a standard deployment adds zero behaviour. Each interval it runs the
deterministic readiness sweep and originates at most one HELD proposal for
the CEO; it NEVER publishes, merges, or deploys. The per-cycle session
commits any opened proposal here.
"""
if not settings.release_manager_enabled:
return
interval = settings.release_manager_interval_seconds
self._record_loop_heartbeat("release_manager", interval)
while self._running:
try:
await asyncio.sleep(interval)
await self._run_release_manager_cycle()
self._record_loop_heartbeat("release_manager", interval)
except asyncio.CancelledError:
break
except Exception:
logger.exception("release-manager cycle failed")
async def _run_release_manager_cycle(self) -> None:
"""One release-manager pass: run the engine, commit. Testable w/o the sleep."""
from roboco.db import get_db_context
from roboco.services.release_manager_engine import get_release_manager_engine
async with get_db_context() as db:
await get_release_manager_engine(db).run_cycle()
await db.commit()
async def _x_mentions_poll_loop(self) -> None:
"""X engine: poll mentions on an interval, hold meaningful ones as draft
replies.
Gated by ``x_replies_enabled`` (default off) on top of the engine
master switch — release posting does not need this loop (those drafts
are originated event-driven from the release-proposal approve hook), so
a standard X deployment posts about releases without ever polling
mentions. It never posts or replies — every draft is held for the CEO.
"""
if not (settings.x_engine_enabled and settings.x_replies_enabled):
return
interval = settings.x_mentions_interval_seconds
self._record_loop_heartbeat("x_mentions", interval)
while self._running:
try:
await asyncio.sleep(interval)
await self._run_x_mentions_cycle()
self._record_loop_heartbeat("x_mentions", interval)
except asyncio.CancelledError:
break
except Exception:
logger.exception("x-mentions poll cycle failed")
async def _run_x_mentions_cycle(self) -> None:
"""One mentions-poll pass: run the engine, commit. Testable w/o the sleep."""
from roboco.db import get_db_context
from roboco.services.x_engine import get_x_engine
async with get_db_context() as db:
await get_x_engine(db).run_cycle()
await db.commit()
async def _roadmap_engine_loop(self) -> None:
"""Board roadmap engine: on an interval, open ONE held exploration cycle.
Dormant by default — returns immediately unless ``roadmap_engine_enabled``,
so a standard deployment originates nothing. The engine itself only opens
the held exploration task; the Product Owner authors the themed cycle
(``propose_roadmap``) once the board dispatcher spawns it, and approved
items land in BACKLOG only via the CEO's per-item approve — this loop
never starts anything.
"""
if not settings.roadmap_engine_enabled:
return
interval = settings.roadmap_interval_seconds
self._record_loop_heartbeat("roadmap_engine", interval)
while self._running:
try:
await asyncio.sleep(interval)
await self._run_roadmap_engine_cycle()
self._record_loop_heartbeat("roadmap_engine", interval)
except asyncio.CancelledError:
break
except Exception:
logger.exception("roadmap-engine cycle failed")
async def _run_roadmap_engine_cycle(self) -> None:
"""One roadmap-engine pass: run the engine, commit. Testable w/o the sleep."""
from roboco.db import get_db_context
from roboco.services.roadmap_engine import get_roadmap_engine
async with get_db_context() as db:
await get_roadmap_engine(db).run_cycle()
await db.commit()
async def _x_feature_spotlight_loop(self) -> None:
"""X engine: on an interval, open ONE held feature-spotlight exploration
for the Head of Marketing.
Dormant by default — returns immediately unless BOTH x_engine_enabled and
x_feature_spotlight_enabled, so a standard deployment (or one running only
release posts / mention replies) never spawns HoM for this.
"""
if not (settings.x_engine_enabled and settings.x_feature_spotlight_enabled):
return
interval = settings.x_feature_spotlight_interval_seconds
self._record_loop_heartbeat("x_feature_spotlight", interval)
while self._running:
try:
await asyncio.sleep(interval)
await self._run_x_feature_spotlight_cycle()
self._record_loop_heartbeat("x_feature_spotlight", interval)
except asyncio.CancelledError:
break
except Exception:
logger.exception("x-feature-spotlight cycle failed")
async def _run_x_feature_spotlight_cycle(self) -> None:
"""One feature-spotlight pass: run the engine, commit. Testable w/o sleep."""
from roboco.db import get_db_context
from roboco.services.x_engine import get_x_engine
async with get_db_context() as db:
await get_x_engine(db).open_feature_spotlight_exploration()
await db.commit()
async def _video_render_loop(self) -> None:
"""Video engine: on an interval, render merged compositions to MP4 and
materialize held video_post drafts.
Dormant by default — returns immediately unless video_engine_enabled,
so a standard deployment never scans for completed video tasks or
reaches the rendering sidecar.
"""
if not settings.video_engine_enabled:
return
interval = settings.video_render_interval_seconds
self._record_loop_heartbeat("video_render", interval)
while self._running:
try:
await asyncio.sleep(interval)
await self._run_video_render_cycle()
self._record_loop_heartbeat("video_render", interval)
except asyncio.CancelledError:
break
except Exception:
logger.exception("video-render cycle failed")
async def _run_video_render_cycle(self) -> None:
"""One render pass: render every completed authoring task carrying an
unrendered composition. Testable w/o the sleep.
commit per-task so a raise mid-cycle no longer rolls back prior
renders (and the next cycle no longer re-renders + re-originates a
second held video_post draft). The committed ``render_status=
"rendered"`` is the idempotency key the next scan skips.
"""
from roboco.db import get_db_context
from roboco.services.task import get_task_service
async with get_db_context() as db:
tasks = await get_task_service(db).list_completed_video_tasks()
for task in tasks:
await self._render_video_task(db, task)
await db.commit()
async def _render_video_task(self, db: Any, task: Any) -> None:
"""Render one completed authoring task's composition, or skip/retry/fail.
Skips silently when the dev hasn't called ``propose_video`` yet (no
``composition_id``) or the task already reached a terminal render state
(``rendered`` — idempotent re-run; ``failed`` — retries exhausted). A
render failure (read-clone not yet synced to the just-merged
composition, transient sidecar blip, bad response) is caught here so one
broken task never blocks the cycle, and is RETRIED on later cycles up to
``_MAX_VIDEO_RENDER_ATTEMPTS`` before being marked terminally failed.
"""
from roboco.foundation.policy.content import markers
draft = markers.get_video_draft(task) or {}
composition_id = draft.get("composition_id")
if not composition_id or draft.get("render_status") in ("rendered", "failed"):
return
try:
mp4_paths = await self._render_both_cuts(
db, draft, composition_id, str(task.id)
)
await self._materialize_video_post(db, task, draft, mp4_paths)
except Exception as exc:
attempts = int(draft.get("render_attempts", 0)) + 1
terminal = attempts >= _MAX_VIDEO_RENDER_ATTEMPTS
payload = {**draft, "render_attempts": attempts}
if terminal:
payload["render_status"] = "failed"
markers.set_video_draft(task, payload)
logger.warning(
"video-render: render attempt failed",
task_id=str(task.id),
attempts=attempts,
terminal=terminal,
error=str(exc),
)
if terminal:
await self._notify_video_render_failure(task, str(exc))
async def _notify_video_render_failure(self, task: Any, last_error: str) -> None:
"""Send one CEO alert that a video render exhausted its retries.
Best-effort, mirroring ``_notify_strategy_engine_failure`` — a
notification-send failure must never raise out of the render loop.
"""
try:
from roboco.services.notification import NotificationService
await NotificationService().send_ack_notification(
from_agent="system",
to_agent="ceo",
body=(
f"[video engine] render terminally failed for task "
f"{task.title!r} ({_MAX_VIDEO_RENDER_ATTEMPTS} attempts "
f"exhausted): {last_error}"
),
task_id=task.id,
)
except Exception:
logger.exception(
"video-render failure-notify dropped", task_id=str(task.id)
)
async def _render_both_cuts(
self, db: Any, draft: dict[str, Any], composition_id: str, render_key: str
) -> dict[str, str]:
"""Render the vertical + square cuts from the roboco project's merged
read-clone's motion/ dir; returns {"vertical": path, "square": path}.
``render_key`` (the source task id) scopes each cut's output path."""
from roboco.services.video_renderer_client import get_video_renderer
from roboco.services.workspace import get_workspace_service
slug = (settings.self_heal_project_slug or "roboco-api").strip()
workspace = await get_workspace_service(db).ensure_read_clone(slug)
motion_dir = str(workspace / "motion")
input_props = draft.get("input_props") or {}
renderer = get_video_renderer()
cuts: dict[str, str] = {}
for orientation in ("vertical", "square"):
cuts[orientation] = await renderer.render(
source_dir=motion_dir,
composition_id=composition_id,
input_props=input_props,
orientation=orientation,
render_key=render_key,
)
return cuts
async def _materialize_video_post(
self, db: Any, task: Any, draft: dict[str, Any], mp4_paths: dict[str, str]
) -> None:
"""Materialize the held video_post draft, then mark the source task
rendered — the idempotency key the next cycle's scan checks."""
from roboco.foundation.policy.content import markers
from roboco.services.video_engine import get_video_engine
await get_video_engine(db)._originate_video_post(
source_task=task,
mp4_paths=mp4_paths,
captions={
"x": draft.get("x_caption", ""),
"tiktok": draft.get("tiktok_caption", ""),
},
platforms=draft.get("platforms") or [],
)
markers.set_video_draft(task, {**draft, "render_status": "rendered"})
async def _load_dep_update_set(self, db: Any) -> list[Any]:
"""Projects with a ``dep_update_command`` + a git_url, one per
(repo, command).
A monorepo's several cell-projects can each carry their OWN
``dep_update_command`` (different ecosystems → different lockfiles,
e.g. ``uv lock --upgrade`` vs ``pnpm update -L``). Collapsing to one
canonical project per REPO would probe only the canonical cell's
lockfile and miss the others' drift (the under-count). Collapse to one
canonical project per (repo, command) instead — every distinct command
is probed once, and the engine's per-``git_url`` open-task dedup still
prevents a duplicate update task for the same repo.
"""
from roboco.services.project import get_project_service
projects = await get_project_service(db).list_all(active_only=True)
eligible = [
p
for p in projects
if str(getattr(p, "dep_update_command", None) or "").strip()
and getattr(p, "git_url", None)
]
return self._projects_one_per_key(
eligible,
key_fn=lambda p: (
self._repo_key(str(getattr(p, "git_url", "") or "")),
str(getattr(p, "dep_update_command", None) or "").strip(),
),
)
@staticmethod
def _repo_key(git_url: str) -> str:
"""Normalized repo identity (case/.git/trailing-slash insensitive).
Delegates to :func:`roboco.utils.converters.repo_key` so the dedupe
queries and the poll-set collapse share one source of truth (#1267).
"""
from roboco.utils.converters import repo_key
return repo_key(git_url)
@classmethod
def _projects_one_per_repo(cls, projects: list[Any]) -> list[Any]:
"""One canonical project per distinct repo.
Many projects can point at the SAME repo — a monorepo product's
backend/frontend/ux cells each have their own Project mapping to one
git_url. Polling per-project would then ingest one review task per cell
for a single external PR (the per-(project,pr) dedup can't see across
projects). Collapse to one canonical project per repo (deterministic by
slug so the pick is stable across polls); genuinely separate repos
(multi-repo) each keep their own. Projects without a git_url are skipped.
Used by the external-PR discovery path (one review per PR per repo). The
CI-watch and dep-update loaders use :meth:`_projects_one_per_key` with a
finer (repo, workflow) / (repo, command) key so a monorepo's per-cell
workflow / lockfile-command overrides are each sampled once instead of
collapsing to the canonical cell's value.
"""
return cls._projects_one_per_key(
projects,
key_fn=lambda p: (cls._repo_key(str(getattr(p, "git_url", "") or "")),),
)
@classmethod
def _projects_one_per_key(
cls, projects: list[Any], *, key_fn: "Callable[[Any], tuple[Any, ...]]"
) -> list[Any]:
"""One canonical project per distinct key (deterministic by slug).
``key_fn`` defines what distinguishes a duplicate: repo identity for
external-PR discovery (one review per PR per repo); ``(repo, workflow)``
for CI-watch and ``(repo, command)`` for dep-update so a monorepo's
several cell-projects — each potentially carrying its OWN workflow /
lockfile command — are each sampled once instead of collapsing to the
canonical cell's value (the under-count fixed by F115). The first
project (by slug) per key is the canonical pick; the engine's
per-``git_url`` fix-task dedup still prevents duplicate fix tasks for the
same repo. Projects without a git_url are skipped.
"""
seen: set[tuple[Any, ...]] = set()
canonical: list[Any] = []
for project in sorted(projects, key=lambda p: str(getattr(p, "slug", ""))):
git_url = getattr(project, "git_url", None)
if not git_url:
continue
key = key_fn(project)
if key in seen:
continue
seen.add(key)
canonical.append(project)
return canonical
async def _poll_external_prs_once(self, db: "AsyncSession") -> int:
"""One discovery pass across active repos; returns tasks ingested.
Repo-aware: collapses active projects to one canonical project per
distinct repo (so a monorepo product yields ONE review per PR, not one
per cell-project), lists each repo's open PRs, and ingests a de-duped
review task for each reviewable one — external/fork PRs, and (when
internal review is on) org-repo PRs not tied to an active task. Commits
once at the end.
"""
from roboco.services.git import GitService
from roboco.services.project import get_project_service
from roboco.services.task import get_task_service
git = GitService(db)
task_service = get_task_service(db)
projects = await get_project_service(db).list_all(active_only=True)
system_id = _foundation.AGENTS["system"].uuid
allowlist = {a.lower() for a in settings.external_pr_author_allowlist}
ingested = 0
for project in self._projects_one_per_repo(projects):
for pr in await git.list_open_prs(project.slug):
if await self._ingest_pr_if_reviewable(
task_service, project, pr, system_id, allowlist
):
ingested += 1
await db.commit()
return ingested
async def _ingest_pr_if_reviewable(
self,
task_service: "TaskService",
project: Any,
pr: dict[str, Any],
system_id: "UUID",
allowlist: set[str],
) -> bool:
"""Ingest a review task for one open PR if it qualifies; True if ingested.
External/fork PRs (when external review is on and the author is allowed)
are ingested as ``external_pr``. Org-repo PRs whose head branch no active
task owns (when internal review is on) are ingested as ``internal_pr`` —
the org's own in-flight integration PRs are skipped, since a live task
owns their branch and they already pass QA + PM review.
"""
if pr.get("number") is None:
return False
# The reviewer reviews PRs the org did NOT author. Skip PRs opened by the
# repo-owner account: a self-review can't post REQUEST_CHANGES (GitHub
# 422), and re-reviewing the org's own in-flight PRs every poll is noise.
if pr.get("author_is_owner"):
return False
if self._is_external_pr(pr):
if not settings.external_pr_enabled or not self._pr_author_allowed(
pr, allowlist
):
return False
source = "external_pr"
else:
if not settings.internal_pr_enabled:
return False
if await task_service.active_task_owns_branch(
str(pr.get("head_ref") or ""),
cast("UUID", project.id),
):
return False
source = "internal_pr"
created = await task_service.ingest_external_pr(
project_id=cast("UUID", project.id),
pr=pr,
created_by=system_id,
team=Team.BOARD,
source=source,
)
return created is not None
async def _close_superseded_prs(
self, git: Any, task_service: Any, system_id: "UUID"
) -> int:
"""Close + link the contributor PR for each landed supersede umbrella.
Idempotent: each umbrella is marked ``closed=1`` after its contributor PR
is closed, so it is processed once. ``delete_branch=False`` — the
contributor's branch lives on their fork; we never touch it. Caller
commits.
"""
closed = 0
for umbrella in await task_service.supersede_umbrellas_pending_close():
pr_number = self._parse_supersede_pr(umbrella.quick_context or "")
if pr_number is None:
continue
try:
await git.close_pull_request(
pr_number,
comment=(
"Superseded by the roboco team's own PR — the work was "
"finished and hardened to our standards. Thanks for the "
"contribution!"
),
delete_branch=False,
actor_agent_id=system_id,
# PR numbers are per-repo — scope the close to THIS
# umbrella's project so a same-numbered PR in another
# project's repo is never resolved (and closed) by mistake.
project_id=cast("UUID", umbrella.project_id),
)
except Exception:
# A permanent close failure (deleted PR, revoked PAT) would
# otherwise re-fire + re-log every tick forever; keep it a single
# warning rather than a per-tick stack trace.
logger.warning("close-on-land failed", pr_number=pr_number)
continue
await task_service.mark_supersede_pr_closed(cast("UUID", umbrella.id))
closed += 1
return closed
@staticmethod
def _parse_supersede_pr(quick_context: str) -> int | None:
"""Extract the contributor PR number from a supersede umbrella marker.
Anchored to the marker line so a CEO note containing ``pr=`` on a later
line of the multi-writer ``quick_context`` can't be misread as the PR.
"""
for raw in quick_context.splitlines():
line = raw.strip()
if not line.startswith("external_pr_supersede"):
continue
for part in line.split():
if part.startswith("pr="):
try:
return int(part[3:])
except ValueError:
return None
return None
return None
@staticmethod
def _pr_author_allowed(pr: dict[str, Any], allowlist: set[str]) -> bool:
"""With a non-empty allowlist, only those GitHub authors are reviewed.
An empty allowlist (the default) reviews every external PR — the review
is read-only, so it is safe; the ``confirmed_by_human`` gate still
protects any later supersede that would run the contributor's code.
"""
if not allowlist:
return True
return (pr.get("user_login") or "").lower() in allowlist
@staticmethod
def _is_external_pr(pr: dict[str, Any]) -> bool:
"""A PR the org did not author: a fork head or a non-member author."""
if pr.get("is_fork"):
return True
trusted = {"OWNER", "MEMBER", "COLLABORATOR"}
assoc = (pr.get("author_association") or "").upper()
return assoc not in trusted
async def supersede_external_pr(self, review_task_id: "UUID") -> dict[str, Any]:
"""CEO-authorized takeover of a reviewed external PR.
Confirms the review task (this CEO action is the human confirmation that
authorizes running the contributor's code), cuts a roboco-owned branch
off the contributor's fork head (refs/pull/{n}/head — the only point
untrusted code enters a roboco branch), and creates the supersede
umbrella for Main PM to delegate to a cell. Returns a status dict.
"""
from roboco.db import get_db_context
from roboco.models.base import TaskStatus
from roboco.services.git import GitService
from roboco.services.project import get_project_service
from roboco.services.task import get_task_service
# Serialize concurrent CEO calls (double-click) — the dedup check and
# the umbrella/branch creation are not atomic across DB sessions.
async with self._supersede_lock, get_db_context() as db:
task_service = get_task_service(db)
review = await task_service.get(review_task_id)
if review is None or getattr(review, "source", "") not in PR_REVIEW_SOURCES:
return {"ok": False, "error": "not a PR-review task"}
if not review.project_id or not review.pr_number:
return {
"ok": False,
"error": "review task missing project or pr_number",
}
# Review-first: only supersede a PR the org has actually reviewed.
if review.status != TaskStatus.COMPLETED:
return {
"ok": False,
"error": "review not complete — review the PR first",
}
project = await get_project_service(db).get(cast("UUID", review.project_id))
if project is None:
return {"ok": False, "error": "project not found"}
pr_number = int(review.pr_number)
project_id = cast("UUID", review.project_id)
# Idempotent: a repeat call returns the existing umbrella — no second
# branch cut, no duplicate cell takeover.
existing = await task_service.find_supersede_umbrella(project_id, pr_number)
if existing is not None:
return {
"ok": True,
"supersede_task_id": str(existing.id),
"branch": existing.branch_name,
"already_superseded": True,
}
system_id = _foundation.AGENTS["system"].uuid
branch_name = f"feature/main_pm/supersede-pr-{pr_number}"
# The CEO authorized fetching + finishing the contributor's code.
review.confirmed_by_human = True
# Create the umbrella BEFORE the push: a create failure then can't
# orphan a pushed branch. Only a commit failure after the push could
# (rare) — the branch is logged so an orphan stays discoverable.
umbrella = await task_service.create_supersede_umbrella(
review_task_id=review_task_id,
branch_name=branch_name,
created_by=system_id,
)
umbrella_id = str(umbrella.id) if umbrella is not None else None
git = GitService(db)
workspace = await git.get_workspace(project.slug, agent_id=system_id)
logger.warning(
"supersede: cutting roboco branch off untrusted fork PR head",
branch=branch_name,
pr_number=pr_number,
project=project.slug,
)
await git.create_branch_from_pr_head(
workspace, project.slug, pr_number, branch_name
)
await db.commit()
self._dispatch_wake.set()
return {"ok": True, "supersede_task_id": umbrella_id, "branch": branch_name}
async def _rate_limit_probe_loop(self) -> None:
"""Background loop: probe rate-limited providers every ~30 seconds.
Runs independently of the 60-second session/notification sweeper so
rate limits can be cleared on their own cadence without blocking
other sweep work.
"""
probe_interval = 30 # seconds
while self._running:
try:
await asyncio.sleep(probe_interval)
await self._sweep_rate_limit_probes()
except asyncio.CancelledError:
break
except Exception as e:
logger.error("Rate-limit probe loop error", error=str(e))
async def _sweep_rate_limit_probes(self) -> None:
"""One probe pass: check every rate-limited provider.
For each provider whose estimated_lift_at has passed:
- Call ``_do_probe(provider)`` to test connectivity.
- **Success**: clear the tracker, resolve all parked agents, publish
``RATE_LIMIT_LIFTED``.
- **Failure**: increment probe_failures; if the count reaches 10 and
we haven't already sent a CEO notification for this episode, send
one now.
F045: the loop is tracker-driven, but an ``activate()`` failure in the
in-verb ``i_am_blocked(rate_limited)`` path (or a Redis hiccup) can
leave agents parked in ``_waiting_records`` for a provider the tracker
never learned about — so the tracker-listed loop above never probes it
and the parked agents strand in WAITING_LONG forever. After probing the
tracker-listed set, scan the in-memory records for any
``rate_limit_lifted`` provider the loop did NOT cover and probe it via
the time-expiry fallback (empty state → ``_too_early_to_probe`` returns
False → probe now) so ``_on_probe_success`` can resume them. The
fallback reads only local memory, so it works even when Redis is down.
"""
from roboco.services.gateway.rate_limit_tracker import RateLimitStateTracker
try:
providers = await RateLimitStateTracker.list_rate_limited_providers()
except Exception as e:
logger.warning("Failed to list rate-limited providers", error=str(e))
providers = []
probed_providers: set[str] = set()
for provider, state in providers:
probed_providers.add(provider)
try:
await self._probe_one_provider(provider, state)
except Exception as e:
logger.error(
"Unhandled error probing provider",
provider=provider,
error=str(e),
)
# Orphan fallback: resume agents parked for a provider the
# tracker-listed loop above did not cover (activate failed silently or
# Redis was down at park time). On probe success ``_on_probe_success``
# clears the tracker (self-healing) and resumes the parked agents.
orphan_providers: set[str] = set()
for record in self._waiting_records.values():
if record.waiting_for != "rate_limit_lifted":
continue
prov = record.context.get("provider")
if prov and prov not in probed_providers:
orphan_providers.add(prov)
for provider in orphan_providers:
try:
await self._probe_one_provider(provider, {})
except Exception as e:
logger.error(
"Unhandled error probing orphaned rate-limited provider",
provider=provider,
error=str(e),
)
def _make_tracker(self, provider: str) -> Any:
"""Return a RateLimitStateTracker for *provider*.
Extracted as its own method so unit tests can monkeypatch it to
return an async mock without needing to intercept lazy imports.
"""
from roboco.services.gateway.rate_limit_tracker import RateLimitStateTracker
return RateLimitStateTracker(provider)
async def _provider_spawn_parked(self, provider_type: str | None) -> bool:
"""True when *provider_type*'s provider is parked (rate-limited/overloaded).
The spawn loop-breaker consults this before launching any container.
Fail-open: any error reading the tracker returns False so a Redis hiccup
can never block spawning.
"""
if provider_type is None:
return False
try:
tracker = self._make_tracker(provider_type)
return bool(await tracker.is_rate_limited())
except Exception as exc:
logger.warning(
"provider rate-limit check failed; allowing spawn",
provider=provider_type,
error=str(exc),
)
return False
@staticmethod
def _is_grok_rate_limit_exit(instance: Any, exit_code: int | None) -> bool:
"""True for a one-shot grok container that exited 75 (xAI 429)."""
from roboco.models.base import ModelProvider
return (
exit_code == _GROK_RATE_LIMIT_EXIT_CODE
and instance.config is not None
and instance.config.provider_type == ModelProvider.GROK.value
)
@staticmethod
def _is_grok_auth_exit(instance: Any, exit_code: int | None) -> bool:
"""True for a one-shot grok container that exited 78 (auth missing/expired).
The entrypoint runs ``grok_auth --check`` as a backstop and exits 78
(EX_CONFIG) when the access token is missing or expired — the CLI cannot
refresh it headlessly and would otherwise hang at an interactive login
prompt. See ``_GROK_AUTH_EXIT_CODE`` for the full rationale (F041).
"""
from roboco.models.base import ModelProvider
return (
exit_code == _GROK_AUTH_EXIT_CODE
and instance.config is not None
and instance.config.provider_type == ModelProvider.GROK.value
)
@staticmethod
async def _tail_container_logs(container_name: str, lines: int = 80) -> str:
"""Return the last ``lines`` of a container's combined output, '' on error.
The container is still present at exit (agents run detached, not
``--rm``), so ``docker logs`` can read what the dead run printed.
"""
try:
proc = await asyncio.create_subprocess_exec(
"docker",
"logs",
"--tail",
str(lines),
container_name,
stdout=asyncio.subprocess.PIPE,
stderr=asyncio.subprocess.STDOUT,
)
out, _ = await proc.communicate()
except Exception:
return ""
return out.decode(errors="replace")
def _transcript_tail_text(self, agent_id: str, lines: int = 80) -> str:
"""Return the last ``lines`` of the newest Claude transcript for *agent_id*.
The SDK server redirects its runtime log to ``/tmp/sdk-server.log`` inside
the container, so session-limit markers such as "hit your session limit"
and "five_hour" do not reach ``docker logs``. The durable Claude
transcript on the host (mounted into the orchestrator at ``~/.claude``)
contains those same events, so we search it as a fallback when deciding
whether to park the provider. Returns "" when no transcript is found or it
cannot be read.
"""
from pathlib import Path
projects = Path.home() / ".claude" / "projects"
try:
jsonl = [
f
for d in projects.glob(f"*-{agent_id}")
if d.is_dir()
for f in d.glob("*.jsonl")
]
if not jsonl:
return ""
newest = max(jsonl, key=lambda f: f.stat().st_mtime)
text = newest.read_text(encoding="utf-8", errors="replace")
return "\n".join(text.splitlines()[-lines:])
except OSError:
return ""
async def _provider_overload_park_target(
self, agent_id: str, instance: Any
) -> str | None:
"""Provider to park if this dead run hit a persistent overload, else None.
Data-driven by ``_OVERLOAD_MARKERS_BY_PROVIDER``: only providers with
specific overload markers are candidates, so grok (its own exit-75
detector) and unhandled providers stay on crash-retry. Returns the
matched provider value, or None when the feature is disabled, the
provider has no overload markers, or the output holds no marker.
"""
if not settings.overload_break_enabled:
return None
provider_type = instance.config.provider_type if instance.config else None
markers = (
_OVERLOAD_MARKERS_BY_PROVIDER.get(provider_type) if provider_type else None
)
if markers is None:
return None
tail = await self._tail_container_logs(f"roboco-agent-{agent_id}")
# The SDK server writes model-API errors to /tmp/sdk-server.log, not
# stdout, so the overload marker may appear only in the durable Claude
# transcript; without it an overload is missed and the agent
# crash-respawns straight back into it.
transcript_tail = self._transcript_tail_text(agent_id)
lowered = (tail + "\n" + transcript_tail).lower()
if any(marker in lowered for marker in markers):
return provider_type
return None
async def _provider_rate_limit_park_target(
self, agent_id: str, instance: Any
) -> str | None:
"""Provider to park if this dead run hit a session/usage limit, else None.
Mirrors ``_provider_overload_park_target`` but matches the Claude
session ("5-hour") limit AND the ollama.com weekly limit
(``glm-5.2:cloud``), both of which surface as a 429 the SDK does not
retry. Data-driven by ``_RATE_LIMIT_MARKERS_BY_PROVIDER``; returns the
matched provider value or None. Gated so a misfire can be turned off
without a redeploy.
"""
if not settings.overload_break_enabled:
return None
provider_type = instance.config.provider_type if instance.config else None
markers = (
_RATE_LIMIT_MARKERS_BY_PROVIDER.get(provider_type)
if provider_type
else None
)
if markers is None:
return None
tail = await self._tail_container_logs(f"roboco-agent-{agent_id}")
# The SDK server writes to /tmp/sdk-server.log, not stdout, so the
# session-limit markers may not appear in docker logs. Search the durable
# Claude transcript on the host as well.
transcript_tail = self._transcript_tail_text(agent_id)
lowered = (tail + "\n" + transcript_tail).lower()
if any(marker in lowered for marker in markers):
return provider_type
return None
async def _park_provider_unavailable(
self,
agent_id: str,
instance: Any,
*,
provider: str,
retry_after: float,
kind: str,
) -> None:
"""Park an agent whose run ended because its provider is unavailable.
Covers both a 429 rate limit and a persistent 5xx overload. Finalize
the session for usage capture, mark the instance OFFLINE WITHOUT
counting a crash (so it isn't escalated as stranded), and activate the
provider's tracker so the spawn guard suppresses re-spawns until the
probe-resume loop clears it. The task stays claimed/in_progress and is
retried when the provider recovers.
"""
await self._finalize_spawn_session(agent_id, exit_reason=kind)
instance.state = AgentState.OFFLINE
instance.container_id = None
instance.error_count = 0 # provider unavailability is not a crash
try:
await self._make_tracker(provider).activate(
retry_after=retry_after,
affected_agents=[agent_id],
kind=kind,
)
except Exception as exc:
logger.warning(
"failed to park provider-unavailable state",
provider=provider,
kind=kind,
error=str(exc),
)
# Register a WaitingRecord so the probe-resume loop can revive this
# agent when the provider recovers; without it recovery falls to the
# 600s stale-claim reaper instead of the probe-success path the parking
# design relies on. Persisted so a restart still resolves the wait.
# We do NOT call ``mark_waiting_long`` — the container is already dead,
# and parking keeps OFFLINE so the reaper's live-skip / health loop
# ignore it.
task_id = str(instance.current_task_id) if instance.current_task_id else None
record = WaitingRecord(
agent_id=agent_id,
task_id=task_id,
waiting_for="rate_limit_lifted",
waiting_since=datetime.now(UTC),
context={"provider": provider, "kind": kind},
)
self._waiting_records[agent_id] = record
with contextlib.suppress(Exception):
await self._persist_waiting_record(record)
logger.warning(
"Provider unavailable; parked (task retried when it recovers)",
provider=provider,
kind=kind,
agent_id=agent_id,
task_id=instance.current_task_id,
)
async def _park_grok_rate_limited(self, agent_id: str, instance: Any) -> None:
"""Park a grok agent whose run hit an xAI 429 (entrypoint exit 75).
F097: grok has no real recovery probe, so the probe loop clears a grok
park optimistically on a timer — a cleared park dispatches a fresh
grok agent that hits the still-active xAI 429, exits 75, and re-parks.
Without a backoff this is a flat ~90s crash-retry cycle for the whole
xAI rate-limit window. Back the re-park retry_after off exponentially
within one episode (60 -> 120 -> 240 -> ... capped) so the churn
dampens. A gap past ``_GROK_REPARK_EPISODE_GAP_S`` (no re-park for that
long => the rate limit actually lifted) starts a fresh episode at the
base retry_after, so recovery latency isn't penalized across episodes.
"""
from roboco.models.base import ModelProvider
now = datetime.now(UTC)
last = self._grok_last_park_at
if (
last is not None
and (now - last).total_seconds() < _GROK_REPARK_EPISODE_GAP_S
):
self._grok_repark_count += 1
else:
self._grok_repark_count = 0
self._grok_last_park_at = now
backoff = 2 ** min(self._grok_repark_count, _GROK_REPARK_BACKOFF_CAP)
retry_after = _GROK_RATE_LIMIT_RETRY_AFTER_S * backoff
await self._park_provider_unavailable(
agent_id,
instance,
provider=ModelProvider.GROK.value,
retry_after=retry_after,
kind="rate_limited",
)
async def _park_grok_auth_unavailable(self, agent_id: str, instance: Any) -> None:
"""Park a grok agent whose token was missing/expired (entrypoint exit 78).
Same park-and-probe shape as the 429 exit-75 path, but with
``kind="auth_missing"``: the agent cannot start without a valid token, so
crash-retrying burns tokens for zero progress. The probe-resume loop
revives the task once ``grok_auth.refresh_if_stale`` mints a fresh token
(run once per dispatch tick); if still expired, the next exit 78 re-parks
(no token burn). See ``_GROK_AUTH_EXIT_CODE`` (F041).
"""
from roboco.models.base import ModelProvider
await self._park_provider_unavailable(
agent_id,
instance,
provider=ModelProvider.GROK.value,
retry_after=_GROK_AUTH_RETRY_AFTER_S,
kind="auth_missing",
)
@staticmethod
def _too_early_to_probe(state: dict[str, Any]) -> bool:
"""True while the estimated lift time (activated_at + retry_after) is future.
Missing or malformed timestamps fall through to allow the probe.
"""
activated_at_raw = state.get("activated_at")
retry_after = state.get("retry_after")
if not activated_at_raw or retry_after is None:
return False
try:
activated_at = datetime.fromisoformat(activated_at_raw)
except (ValueError, TypeError):
return False
return datetime.now(UTC) < activated_at + timedelta(seconds=retry_after)
def _parked_agents_for(self, provider: str) -> list[str]:
"""Agent slugs parked waiting for *provider*'s rate limit to lift."""
return [
agent_id
for agent_id, record in list(self._waiting_records.items())
if record.waiting_for == "rate_limit_lifted"
and record.context.get("provider") == provider
]
async def _on_probe_success(self, provider: str, tracker: Any) -> None:
"""Clear the limit, resume parked agents, publish RATE_LIMIT_LIFTED."""
logger.info("Rate-limit probe succeeded; clearing provider", provider=provider)
await tracker.clear()
# New episodes should get a fresh CEO notification.
self._rate_limit_ceo_notified.discard(provider)
resumed = self._parked_agents_for(provider)
for agent_id in resumed:
with contextlib.suppress(Exception):
await self.resolve_wait(
agent_id,
{
"reason": "rate_limit_lifted",
"provider": provider,
"lifted_at": datetime.now(UTC).isoformat(),
},
)
with contextlib.suppress(Exception):
from roboco.events import get_event_bus
from roboco.models.events import Event, EventType
await get_event_bus().publish(
Event(
type=EventType.RATE_LIMIT_LIFTED,
data={
"provider": provider,
"resumedAgents": resumed,
"timestamp": datetime.now(UTC).isoformat(),
},
)
)
logger.info(
"RATE_LIMIT_LIFTED published",
provider=provider,
resumed_agents=len(resumed),
)
async def _on_probe_failure(
self, provider: str, tracker: Any, activated_at_raw: str | None
) -> None:
"""Count a failed probe; notify the CEO once at the failure threshold.
F094 escape hatch: past ``_PROBE_GIVE_UP_THRESHOLD`` persistent failures
the probe endpoint itself is the problem (misconfigured URL / removed API
key / network partition to the probe host) while the provider may be fine
for real workloads. Holding the park any longer strands every agent on
the provider forever with only a one-shot CEO notification. Fall back to
the same time-expiry optimism the unprobeable-provider path uses: clear
the park and resume. If the provider is genuinely still down the real
workload attempts re-park via the 429/5xx path, so this is bounded burn —
strictly better than a silent forever-strand. ``_on_probe_success``
clears the tracker (so the loop won't probe this provider again until a
real 429 re-parks) and discards the CEO-notified flag (a fresh episode
later gets a fresh notification).
"""
failure_count = await tracker.increment_probe_failures()
logger.debug(
"Rate-limit probe failed", provider=provider, probe_failures=failure_count
)
if (
failure_count >= _CEO_NOTIFY_THRESHOLD
and provider not in self._rate_limit_ceo_notified
):
self._rate_limit_ceo_notified.add(provider)
await self._notify_rate_limit_ceo(
provider=provider,
activated_at_str=activated_at_raw or "unknown",
paused_agent_count=len(self._parked_agents_for(provider)),
)
if failure_count >= _PROBE_GIVE_UP_THRESHOLD:
logger.warning(
"Rate-limit probe persistently failing; giving up on the probe "
"and falling back to time-expiry optimism (clearing the park + "
"resuming parked agents). If the provider is genuinely still down "
"they will re-park via the real 429/5xx path.",
provider=provider,
probe_failures=failure_count,
)
await self._on_probe_success(provider, tracker)
async def _probe_one_provider(self, provider: str, state: dict[str, Any]) -> None:
"""Probe a single rate-limited provider and handle the outcome."""
if self._too_early_to_probe(state):
return # Wait until after the estimated lift time.
tracker = self._make_tracker(provider)
if await self._do_probe(provider):
await self._on_probe_success(provider, tracker)
else:
await self._on_probe_failure(provider, tracker, state.get("activated_at"))
@staticmethod
def _probe_target(provider: str) -> tuple[str | None, dict[str, str]]:
"""Resolve the (url, headers) for a free liveness probe of ``provider``.
Returns ``(None, {})`` when the provider can't be probed — an unknown
provider, or Anthropic with no API key configured. The caller then
falls back to time-expiry optimism rather than parking forever.
"""
p = provider.lower()
if p == "anthropic":
key = settings.anthropic_api_key
if not key:
return None, {}
return (
f"{_ANTHROPIC_PROBE_BASE}/v1/models",
{"x-api-key": key, "anthropic-version": "2023-06-01"},
)
if p.startswith("ollama"):
return f"{settings.ollama_base_url.rstrip('/')}/api/tags", {}
return None, {}
async def _do_probe(self, provider: str) -> bool:
"""Return True if ``provider`` is accepting requests again.
Makes a free, unmetered liveness call — Anthropic ``GET /v1/models``
or Ollama ``GET /api/tags`` — and treats only a 2xx response as
recovered. Any error status keeps the provider parked: a 429 (still
rate-limited) **and** a 5xx (still overloaded) alike — resuming on a
non-2xx would march parked agents straight back into the failure. A
network error stays parked too (retry next sweep). When the provider
can't be probed (no key / unknown), fall back to time-expiry optimism:
the caller only reaches this after ``estimated_lift_at`` has passed.
Injectable boundary — tests monkeypatch this to force outcomes.
"""
url, headers = self._probe_target(provider)
if url is None:
return True # cannot probe — trust the elapsed retry_after window
try:
async with httpx.AsyncClient(timeout=_PROBE_TIMEOUT_SECONDS) as client:
resp = await client.get(url, headers=headers)
except httpx.HTTPError as exc:
logger.debug(
"Provider-recovery probe request failed",
provider=provider,
error=str(exc),
)
return False # unreachable — stay parked, retry on the next sweep
return _HTTP_OK <= resp.status_code < _HTTP_MULTIPLE_CHOICES
async def _notify_rate_limit_ceo(
self,
provider: str,
activated_at_str: str,
paused_agent_count: int,
) -> None:
"""Send a high-priority notification to the CEO about a persistent rate limit.
Fires once per rate-limit episode. Follows the same pattern as
``_notify_stranded_agent`` — direct DB insert + delivery.deliver().
"""
try:
from sqlalchemy import select as _select
from roboco.db.base import get_session_factory
from roboco.db.tables import AgentTable, NotificationTable
from roboco.models.base import (
AgentRole,
NotificationPriority,
NotificationType,
)
from roboco.services.notification_delivery import (
get_notification_delivery_service,
)
from roboco.utils.converters import require_uuid
# Compute human-friendly duration
duration_desc = "unknown duration"
try:
activated_at = datetime.fromisoformat(activated_at_str)
elapsed = datetime.now(UTC) - activated_at
total_minutes = int(elapsed.total_seconds() / 60)
if total_minutes < 60: # noqa: PLR2004
duration_desc = f"{total_minutes} minute(s)"
else:
duration_desc = f"{total_minutes // 60}h {total_minutes % 60}m"
except (ValueError, TypeError):
pass
session_factory = get_session_factory()
async with session_factory() as db:
ceo_result = await db.execute(
_select(AgentTable).where(AgentTable.role == AgentRole.CEO)
)
ceo = ceo_result.scalar_one_or_none()
if ceo is None:
logger.warning(
"CEO agent not found; skipping rate-limit CEO notification",
provider=provider,
)
return
notification = NotificationTable(
type=NotificationType.ALERT,
priority=NotificationPriority.HIGH,
from_agent=ceo.id,
to_agents=[ceo.id],
subject=f"Rate limit persisting: {provider}",
body=(
f"Provider '{provider}' has been rate-limited for "
f"{duration_desc}. "
f"{paused_agent_count} agent(s) are currently paused. "
f"10 consecutive probe attempts have failed. "
f"Manual intervention may be required."
),
requires_ack=True,
)
db.add(notification)
await db.flush()
delivery = get_notification_delivery_service(db)
await delivery.deliver(require_uuid(notification.id))
await db.commit()
logger.info(
"Rate-limit CEO notification sent",
provider=provider,
paused_agents=paused_agent_count,
)
except Exception as e:
logger.error(
"Failed to send rate-limit CEO notification",
provider=provider,
error=str(e),
)
# =========================================================================
# STATUS API
# =========================================================================
def get_state(self, agent_id: str) -> AgentState:
"""Get current state of an agent."""
if agent_id not in self._instances:
return AgentState.OFFLINE
return self._instances[agent_id].state
def get_instance(self, agent_id: str) -> AgentInstance | None:
"""Get instance for an agent."""
return self._instances.get(agent_id)
def get_waiting_agents(self) -> dict[str, WaitingRecord]:
"""Get all waiting agents."""
return dict(self._waiting_records)
def get_status_summary(self) -> dict[str, Any]:
"""Get summary of all agent states."""
by_state: dict[str, int] = {}
agents: list[dict[str, Any]] = []
for state in AgentState:
count = sum(1 for i in self._instances.values() if i.state == state)
if count > 0:
by_state[state.value] = count
for agent_id, instance in self._instances.items():
cid = instance.container_id[:12] if instance.container_id else None
agents.append(
{
"agent_id": agent_id,
"state": instance.state.value,
"container_id": cid,
"task_id": instance.current_task_id,
"error_count": instance.error_count,
"started_at": instance.started_at.isoformat()
if instance.started_at
else None,
}
)
return {
"total": len(self._instances),
"by_state": by_state,
"waiting_count": len(self._waiting_records),
"agents": agents,
}
# =========================================================================
# SMART DISPATCHER - API HELPERS
# =========================================================================
@property
def _api_url(self) -> str:
"""Get the internal API URL for task/notification queries."""
return settings.internal_api_url
def _is_agent_active(self, agent_id: str) -> bool:
"""Check if an agent is currently running."""
if agent_id not in self._instances:
return False
return self._instances[agent_id].state == AgentState.ACTIVE
async def _check_parent_branch_ready(
self, client: httpx.AsyncClient, task_id: str, parent_id: str
) -> str | None:
"""Verify the parent task has a branch; auto-block + return msg if not.
Race window: the PM's `i_will_plan` claims the parent (transitions
status -> in_progress, sets assigned_to) and then `_finalize_claim`
creates the branch via `_ensure_branch_for_task`. Both actions land
in the same DB transaction but a child dev's spawn dispatch can fire
microseconds before that transaction commits and see branch_name=None.
Without retry we'd auto-block the child unnecessarily.
When the parent is clearly mid-claim (in_progress + assigned_to set)
re-fetch up to 3 times with a 250ms delay before giving up. Total
worst-case wait is 750ms — well inside the dispatcher's tick budget
and only paid when the race actually triggers. Real misses (parent
still pending or unassigned) auto-block immediately as before.
"""
parent_resp = await client.get(f"{self._api_url}/tasks/{parent_id}")
if not parent_resp.is_success:
return None
parent = parent_resp.json()
if parent.get("branch_name"):
return None
# A coordination/fan-out parent (product, no repo of its own) never gets
# a branch: the child resolves its own real project and cuts from that
# project's default branch, not from the parent. Blocking the child on a
# branch the parent will never have wedges the cell↔Main-PM loop.
if _is_coordination_task(parent):
return None
if parent.get("status") == "in_progress" and parent.get("assigned_to"):
for _ in range(3):
await asyncio.sleep(0.25)
parent_resp = await client.get(f"{self._api_url}/tasks/{parent_id}")
if not parent_resp.is_success:
continue
parent = parent_resp.json()
if parent.get("branch_name"):
return None
await self._auto_block_task(
client,
task_id,
"Parent task must be claimed first to create its branch",
)
return f"Task {task_id} waiting for parent branch"
async def _check_dev_needs_subtasks(
self, client: httpx.AsyncClient, task: dict[str, Any]
) -> str | None:
"""Block non-trivial root tasks routed to a dev without subtasks."""
complexity = task.get("estimated_complexity", "low")
parent_task_id = task.get("parent_task_id")
if complexity not in ("medium", "high") or parent_task_id:
return None
task_id = task.get("id")
try:
resp = await client.get(f"{self._api_url}/tasks/{task_id}/subtasks")
subtasks = resp.json() if resp.is_success else []
except Exception:
subtasks = []
if subtasks:
return None
await self._auto_block_task(
client,
str(task_id),
f"Task complexity is {complexity} but no subtasks. "
"Cell PM must break down work first.",
)
return (
f"Task {task_id} is {complexity} complexity "
"without subtasks - Cell PM must break it down"
)
async def _validate_task_for_spawn(
self,
client: httpx.AsyncClient,
task: dict,
agent_slug: str,
) -> str | None:
"""
Validate task is ready for agent spawn.
Returns None if valid, or error message if task cannot proceed.
This prevents spawning agents on tasks that are missing prerequisites.
"""
from roboco.agents_config import get_agent_role
if shape_err := await self._check_spawn_task_shape(client, task):
return shape_err
if dep_err := await self._check_dependencies_terminal(client, task):
return dep_err
# _check_spawn_task_shape guarantees a non-empty id past this point.
task_id = str(task.get("id"))
parent_id = task.get("parent_task_id")
if parent_id:
err = await self._check_parent_branch_ready(client, task_id, parent_id)
if err:
return err
logger.info("Task ready for hierarchical branch creation", task_id=task_id)
if get_agent_role(agent_slug) == "developer":
err = await self._check_dev_needs_subtasks(client, task)
if err:
return err
return None # All validations passed
async def _check_spawn_task_shape(
self, client: httpx.AsyncClient, task: dict[str, Any]
) -> str | None:
"""Reject a task that is structurally unroutable (id/description/repo)."""
task_id = task.get("id")
if not task_id:
return "Task missing ID"
min_description_len = 10
description = (task.get("description") or "").strip()
if len(description) < min_description_len:
return (
f"Task {task_id} has inadequate description ({len(description)} chars)"
)
# A coordination task carries a product or an ad-hoc cell map instead of a
# repo; only a task with neither is genuinely unroutable.
if not task.get("project_id") and not _is_coordination_task(task):
await self._auto_block_task(
client, task_id, "Task needs a project_id, product_id, or cell map"
)
return f"Task {task_id} needs a project, product, or cell map"
return None
async def _check_dependencies_terminal(
self, client: httpx.AsyncClient, task: dict[str, Any]
) -> str | None:
"""Hold a pre-assigned task whose dependencies are not yet terminal.
A dev subtask is always pre-assigned, so it never passes through the
unassigned claim pool's dependency filter. Without this gate the
dispatcher would spawn the dev container while a cross-cell dependency
(e.g. the UX/UI design the frontend dev waits on) is still open. Return
a skip reason while ANY dependency is non-terminal; allow the spawn
once every dependency reaches completed/cancelled.
"""
dependency_ids = task.get("dependency_ids") or []
if not dependency_ids:
return None
terminal = ("completed", "cancelled")
for dep_id in dependency_ids:
dep_resp = await client.get(f"{self._api_url}/tasks/{dep_id}")
# A dependency we cannot read is treated as unmet — fail closed
# rather than spawn ahead of work whose state is unknown.
if not dep_resp.is_success or dep_resp.json().get("status") not in terminal:
return (
f"Task {task.get('id')} waiting on non-terminal dependency {dep_id}"
)
return None
async def _auto_block_task(
self, client: httpx.AsyncClient, task_id: str, reason: str
) -> None:
"""Auto-block a task that cannot proceed due to missing prerequisites."""
try:
await client.patch(
f"{self._api_url}/tasks/{task_id}",
json={
"status": "blocked",
"dev_notes": f"[AUTO-BLOCKED] {reason}",
},
)
logger.info(
"Auto-blocked task with missing prerequisites",
task_id=task_id,
reason=reason,
)
except Exception as e:
logger.error(
"Failed to auto-block task",
task_id=task_id,
error=str(e),
)
async def _auto_resume_paused_parent(
self, client: httpx.AsyncClient, task_id: str
) -> None:
"""Resume a paused parent right before its PM is respawned for closure.
A PM auto-pauses its owned parent on i_am_idle (by design,
so the closure dispatcher knows to respawn it). Pre-gateway the
parent was resumed at respawn so the PM landed actionable; the
gateway refactor dropped that, so the respawned PM had to issue
``resume()`` itself — which weak models reliably fail,
wedging the whole chain. Restore the auto-resume:
paused -> in_progress before spawn so the PM can directly
submit_up / complete / escalate. Best-effort; a resume failure
must not block the spawn (the PM can still resume manually).
"""
try:
await client.patch(
f"{self._api_url}/tasks/{task_id}",
json={"status": "in_progress"},
)
logger.info(
"Auto-resumed paused parent for PM closure respawn",
task_id=task_id,
)
except Exception as e:
logger.error(
"Failed to auto-resume paused parent",
task_id=task_id,
error=str(e),
)
async def _auto_recover_blocked_parent(
self, client: httpx.AsyncClient, task_id: str
) -> None:
"""Recover a blocked parent right before its PM is respawned for closure.
Symmetric to ``_auto_resume_paused_parent``. The
closure dispatcher only reaches this point once every descendant
is terminal, so a parent still ``blocked`` here is an errant /
stale block (e.g. a child's i_am_blocked propagated, or a PM
blocked it and never unblocked) — the real dependency is already
done. That resume path handled only ``paused`` parents, so a ``blocked``
one wedged the whole chain forever: the respawned PM cannot
submit_up / complete a blocked parent and must first ``unblock``
it (needs journal:decision), which weak models never reliably do
(a dogfood run wedged exactly here). ``blocked -> in_progress``
is lifecycle-valid — it is precisely what ``unblock(restore=True)``
performs. Best-effort; a failure must not block the spawn (the PM
can still ``unblock`` manually).
"""
try:
await client.patch(
f"{self._api_url}/tasks/{task_id}",
json={"status": "in_progress"},
)
logger.info(
"Auto-recovered blocked parent for PM closure respawn",
task_id=task_id,
)
except Exception as e:
logger.error(
"Failed to auto-recover blocked parent",
task_id=task_id,
error=str(e),
)
def _select_agent_for_cell(self, cell: str, role: str) -> str | None:
"""
Select the best available agent for a cell and role.
Prefers agents that are not currently active.
For developers, uses round-robin among candidates.
"""
prefix_map = {"backend": "be", "frontend": "fe", "ux_ui": "ux"}
prefix = prefix_map.get(cell)
if not prefix:
return None
# Build candidate list based on role
if role == "dev":
candidates = [f"{prefix}-dev-1", f"{prefix}-dev-2"]
elif role == "qa":
candidates = [f"{prefix}-qa"]
elif role == "doc":
candidates = [f"{prefix}-doc"]
elif role == "pm":
candidates = [f"{prefix}-pm"]
elif role == "pr_reviewer":
candidates = [f"{prefix}-pr-reviewer"]
else:
return None
# Prefer non-active agents
for agent_id in candidates:
if not self._is_agent_active(agent_id):
return agent_id
# All active - return first (task will queue for them via scan)
return candidates[0]
async def _claim_task_for_agent(
self,
client: httpx.AsyncClient,
task_id: str,
agent_id: str,
) -> bool:
"""Claim a task on behalf of an agent before spawning."""
try:
resp = await client.post(
f"{self._api_url}/tasks/{task_id}/claim",
json={"agent_id": agent_id},
)
if resp.status_code == http_status.HTTP_200_OK:
logger.info(
"Task claimed for agent",
task_id=task_id,
agent_id=agent_id,
)
return True
logger.warning(
"Failed to claim task",
task_id=task_id,
agent_id=agent_id,
status=resp.status_code,
)
except Exception as e:
logger.error("Claim task error", task_id=task_id, error=str(e))
return False
async def _fetch_tasks(
self,
client: httpx.AsyncClient,
status: str | list[str],
team: str | None = None,
) -> list[dict[str, Any]]:
"""Fetch tasks by status and optional team filter."""
# If multiple statuses, make separate requests and combine results
statuses = status if isinstance(status, list) else [status]
all_tasks: list[dict[str, Any]] = []
for single_status in statuses:
params: dict[str, Any] = {"status": single_status}
if team:
params["team"] = team
try:
resp = await client.get(f"{self._api_url}/tasks", params=params)
if resp.status_code == http_status.HTTP_200_OK:
tasks: list[dict[str, Any]] = resp.json()
all_tasks.extend(tasks)
except Exception as e:
logger.error(
"Fetch tasks error", status=single_status, team=team, error=str(e)
)
return all_tasks
async def _fetch_notifications(
self,
client: httpx.AsyncClient,
notification_type: str,
unacknowledged: bool = True,
) -> list[dict[str, Any]]:
"""Fetch notifications by type."""
params: dict[str, Any] = {
"type_filter": notification_type,
"pending_ack_only": str(unacknowledged).lower(),
}
try:
resp = await client.get(
f"{self._api_url}/notifications",
params=params,
)
if resp.status_code == http_status.HTTP_200_OK:
data = resp.json()
items: list[dict[str, Any]] = data.get("items", [])
return items
except Exception as e:
logger.error(
"Fetch notifications error",
notification_type=notification_type,
error=str(e),
)
return []
# =========================================================================
# SMART ROUTING - TASK CLASSIFICATION
# =========================================================================
# Keywords that indicate strategic/board-level tasks
_BOARD_KEYWORDS = frozenset(
{
"roadmap",
"architecture",
"security",
"budget",
"hiring",
"strategy",
"vision",
"milestone",
"release",
"launch",
}
)
# Keywords that indicate PM coordination is needed
_PM_KEYWORDS = frozenset(
{
"coordinate",
"integration",
"cross-team",
"sync",
"planning",
"milestone",
"dependencies",
"review",
}
)
# Keywords that indicate cross-cell work (requires Main PM)
_CROSS_CELL_KEYWORDS = frozenset(
{
"all teams",
"all cells",
"every team",
"every cell",
"all departments",
"cross-cell",
"company-wide",
"organization-wide",
"backend and frontend",
"frontend and backend",
"all three",
}
)
def _has_board_keywords(self, text: str) -> bool:
"""Check if text contains board-level keywords."""
return any(kw in text for kw in self._BOARD_KEYWORDS)
def _has_pm_keywords(self, text: str) -> bool:
"""Check if text contains PM coordination keywords."""
return any(kw in text for kw in self._PM_KEYWORDS)
def _has_cross_cell_keywords(self, text: str) -> bool:
"""Check if text indicates work spanning multiple cells."""
return any(kw in text for kw in self._CROSS_CELL_KEYWORDS)
# Direct team-to-routing mappings (explicit assignments bypass keyword analysis)
_TEAM_ROUTING_MAP: ClassVar[dict[str, str]] = {
"main_pm": "main_pm",
"board": "board",
"marketing": "marketing",
}
@staticmethod
def _route_by_task_type(task_type: str, team: str | None) -> str | None:
"""Route based on task_type field alone; returns None if no match."""
cell_teams = tuple(
sorted(t.value for t in CELL_TEAMS)
) # ("backend", "frontend", "ux_ui")
if task_type in ("planning", "research", "administrative"):
return "cell_pm" if team in cell_teams else "main_pm"
if task_type == "design" and team not in ("backend", "frontend"):
return "cell_pm"
return None
def _classify_cell_code_task(self, text: str, complexity: str) -> str:
"""Route a cell-owned code task WITHIN its cell (dev or cell_pm).
Implementation work that belongs to a CELL never escalates to the
board or main_pm by keyword — a dev task whose description says
"Create & Launch" or "auth/security" is still a dev task. Letting the
board/main_pm keyword heuristics fire on it is how a cell code task
ended up "reviewed" by the board and a PM ended up owning (and
deadlocking) a dev code task.
"""
if self._has_pm_keywords(text) or complexity == "high":
return "cell_pm"
return "dev"
def _classify_strategic_code_task(
self, text: str, team: str | None, complexity: str
) -> str:
"""Route a team-less / "all" top-level code task by strategic heuristics."""
if self._has_board_keywords(text):
return "board"
if (
self._has_cross_cell_keywords(text)
or complexity == "high"
or not team
or team == "all"
):
return "main_pm"
if self._has_pm_keywords(text) or complexity == "medium":
return "cell_pm"
return "dev"
def _classify_code_task(self, task: dict[str, Any]) -> str:
"""Classify a generic `code` task via keyword/complexity heuristics."""
team = task.get("team")
title = (task.get("title") or "").lower()
description = (task.get("description") or "").lower()
text = f"{title} {description}"
complexity = task.get("estimated_complexity", "medium").lower()
cell_teams = frozenset(t.value for t in CELL_TEAMS)
if team in cell_teams:
return self._classify_cell_code_task(text, complexity)
return self._classify_strategic_code_task(text, team, complexity)
def _classify_task_routing(self, task: dict[str, Any]) -> str:
"""
Classify a task for routing based on task_type, team, complexity, and keywords.
Returns one of: "board", "main_pm", "cell_pm", "dev", "marketing"
"""
team = task.get("team")
task_type = task.get("task_type", "code")
# Task type takes precedence for non-code work
by_type = self._route_by_task_type(task_type, team)
if by_type:
return by_type
if team in self._TEAM_ROUTING_MAP:
return self._TEAM_ROUTING_MAP[team]
return self._classify_code_task(task)
# Team to PM mapping for routing
_TEAM_PM_MAP: ClassVar[dict[str, str]] = {
"backend": "be-pm",
"frontend": "fe-pm",
"ux_ui": "ux-pm",
}
def _get_routing_target(self, routing: str, task: dict[str, Any]) -> str | None:
"""
Resolve a routing decision to a specific agent slug.
Args:
routing: One of "board", "main_pm", "cell_pm", "dev", "marketing"
task: The task being routed
Returns:
Agent slug (e.g., "main-pm", "be-pm", "be-dev-1") or None
"""
team = task.get("team")
# Static routing targets
static_targets = {
"board": "product-owner",
"main_pm": "main-pm",
"marketing": "head-marketing",
}
if routing in static_targets:
return static_targets[routing]
# Cell PM routing - requires team lookup
if routing == "cell_pm":
return self._TEAM_PM_MAP.get(team, "main-pm") if team else "main-pm"
# Dev routing - select a cell agent.
if routing == "dev":
agent = self._select_agent_for_cell(team, "dev") if team else None
if agent:
return agent
# No cell agent — team is missing or a non-cell team (fullstack /
# system). Fall back to main-pm to triage rather than leaving the
# task ownerless-and-dormant: the dispatcher never re-spawns an
# unrouted pending task, so a None here strands it. Mirrors the
# cell_pm / escalation `... or "main-pm"` default.
logger.warning(
"dev routing found no cell agent; falling back to main-pm",
task_id=task.get("id"),
team=team,
)
return "main-pm"
# Unrecognized routing classification — never strand the task; main-pm
# triages it instead of it going dormant.
logger.warning(
"unrecognized routing classification; falling back to main-pm",
routing=routing,
task_id=task.get("id"),
)
return "main-pm"
def _build_main_pm_triage_prompt(self, task: dict[str, Any]) -> str:
"""Build prompt for MAIN PM to triage and distribute to Cell PMs."""
task_id = task.get("id", "unknown")
title = task.get("title", "Untitled")
complexity = task.get("complexity", "medium")
description = task.get("description", "")
return f"""You are the MAIN PM at RoboCo. This task is assigned to YOU.
TASK: {task_id}
TITLE: {title}
COMPLEXITY: {complexity}
DESCRIPTION: {description[:500]}
YOUR JOB: Break this down and delegate to Cell PMs. You do NOT implement
code. You do NOT assign directly to developers — Cell PMs manage their
teams. For purely-PM work (validation, announcements, cross-cell sync) you
may keep the task and work it via your gateway verbs.
== DELEGATION TARGETS ==
- Backend work → be-pm (who delegates to be-dev-1 / be-dev-2)
- Frontend work → fe-pm (who delegates to fe-dev-1 / fe-dev-2)
- UX/UI work → ux-pm (who delegates to ux-dev-1 / ux-dev-2)
NEVER assign to a dev slug from this seat — only Cell PM slugs.
== TOOLS ==
Gateway verbs (already loaded):
- evidence(task_id="{task_id}") — inspect the task
- triage_all() — see what's pending across cells
- note(text, scope='decision', task_id="{task_id}")
REQUIRED before i_will_plan / complete / escalate
- i_will_plan(task_id="{task_id}", plan="<your detailed plan as a string>")
claim + record plan + start your own root task
- delegate(parent_task_id="{task_id}", title=..., description=...,
assigned_to=<one of "be-pm" / "fe-pm" / "ux-pm">,
team=<one of "backend" / "frontend" / "ux_ui">,
task_type=<one of "code" / "documentation" / "research" /
"planning" / "design" / "administrative">,
acceptance_criteria=[...],
estimated_complexity=<one of "low" / "medium" / "high">)
creates a subtask under your root and assigns it to a Cell PM.
Use the EXACT enum strings above — invented values like
"development" or "small" are rejected by the gateway. Repeat
once per cell that needs work.
- unblock(task_id, restore=True)
- complete(task_id="{task_id}", notes=...) for root awaiting_pm_review
- escalate_to_ceo(task_id="{task_id}", reason=...) for root tasks
- dm(recipient, text), read_a2a()
- i_am_idle() — when delegated and waiting
== WORKFLOW ==
1. evidence(task_id="{task_id}")
2. note(scope='decision', task_id="{task_id}",
text="<plan summary: cells X/Y get subtasks A/B>")
3. i_will_plan(task_id="{task_id}",
plan="<detailed plan: scope, cell breakdown, sequencing, risks>")
4. delegate(parent_task_id="{task_id}", title="Backend slice of <root>",
description="What be-pm should coordinate.",
assigned_to="be-pm", team="backend", task_type="code",
acceptance_criteria=["c1", "c2"], estimated_complexity="medium")
— repeat per cell that needs work. ONE subtask per cell; the Cell PM
breaks it down further.
5. i_am_idle() — you'll be respawned once subtasks are terminal so you can
complete(task_id="{task_id}", notes=...) or escalate_to_ceo on the root.
== RULES ==
- Never `commit`, never write code, never run `git`. PMs coordinate.
- Never assign a code subtask directly to a developer slug — always to a Cell PM.
- delegate / complete / escalate will fail unless you've logged a journal
decision for this task — read the `remediate` field on errors.
Start now: evidence(task_id="{task_id}")
"""
def _build_pm_triage_prompt(self, task: dict[str, Any]) -> str:
"""Build prompt for CELL PM to triage and delegate a task."""
task_id = task.get("id", "unknown")
title = task.get("title", "Untitled")
complexity = task.get("complexity", "medium")
team = task.get("team", "unknown")
# Build team-specific info
dev_map = {
"backend": ("be-dev-1", "be-dev-2"),
"frontend": ("fe-dev-1", "fe-dev-2"),
"ux_ui": ("ux-dev-1", "ux-dev-2"),
}
devs = dev_map.get(team, ("be-dev-1",))
primary_dev = devs[0]
dev_options = " or ".join(devs)
return f"""You are the PM for {team} team. This task is assigned to YOU.
TASK: {task_id}
TITLE: {title}
COMPLEXITY: {complexity}
TEAM: {team}
YOUR JOB: Break this down into concrete subtasks and delegate each to a
developer in your cell. You do NOT code. You do NOT run git. You coordinate.
Available developers in your cell: {dev_options}
== TOOLS ==
Gateway verbs (already loaded):
- evidence(task_id="{task_id}") — read PR + commits + diff
- triage() — see what your cell needs next
- note(text, scope='decision', task_id="{task_id}")
REQUIRED before i_will_plan / unblock / complete / escalate
- i_will_plan(task_id="{task_id}", plan="<detailed plan as a string>")
claim + record plan + start your own cell-PM task
- delegate(parent_task_id="{task_id}", title=..., description=...,
assigned_to=<dev slug in your cell, e.g. "be-dev-1">,
team="{team}",
task_type=<one of "code" / "documentation" / "research" /
"planning" / "design" / "administrative">,
acceptance_criteria=[...],
estimated_complexity=<one of "low" / "medium" / "high">)
creates a subtask under your cell-PM task and assigns it to a developer.
Use the EXACT enum strings above — invented values like
"development" or "small" are rejected by the gateway.
Repeat 2 to 5 times for focused subtasks.
- unblock(task_id, restore=True)
when a dev signals i_am_blocked
- complete(task_id, notes)
review a SUBTASK in awaiting_pm_review (auto-merges its leaf PR)
- submit_up(task_id="{task_id}", notes=...)
when YOUR OWN cell-PM task's subtasks are all terminal: opens cell-level
PR up to Main PM's branch and transitions to awaiting_pm_review.
- escalate_up(task_id, reason) — to Main PM
- dm(recipient, text), read_a2a()
- i_am_idle() — when delegated and waiting
== WORKFLOW ==
1. evidence(task_id="{task_id}")
2. note(scope='decision', task_id="{task_id}",
text="<approach>; subtasks: A→{primary_dev}, B→...")
3. i_will_plan(task_id="{task_id}",
plan="<detailed plan: scope, subtask breakdown, sequencing, risks>")
4. delegate(parent_task_id="{task_id}", title="Add login endpoint",
description="Implement POST /login that issues a session token.",
assigned_to="{primary_dev}", team="{team}", task_type="code",
acceptance_criteria=["c1", "c2"], estimated_complexity="medium")
— repeat 2 to 5 times for focused subtasks under your cell-PM task.
5. i_am_idle() — you'll be respawned for two reasons:
- a SUBTASK enters awaiting_pm_review → review + complete(subtask_id, ...)
- all subtasks terminal → submit_up(task_id="{task_id}", notes=...) on YOUR task
== RULES ==
- Never `commit`, never write code, never run `git`. PMs coordinate.
- Subtasks MUST go to a developer slug in YOUR cell, not another cell's PM.
- delegate / complete / submit_up / escalate will fail unless you've logged
a journal decision for the relevant task — read the `remediate` field.
Start now: evidence(task_id="{task_id}")
"""
# =========================================================================
# SMART DISPATCHER - MAIN LOOP
# =========================================================================
def trigger_dispatch(self) -> None:
"""Wake the dispatcher up immediately for a single pass.
Called by API routes right after a task status transition so the
orchestrator reacts in milliseconds — e.g. a PM creates a subtask
and the assignee spawns within a second instead of after the next
30-second poll. Safe to call multiple times; the Event coalesces.
"""
self._dispatch_wake.set()
# Heartbeat cadence: one dispatcher.alive audit row per window. 300s keeps
# audit_log growth trivial (~288 rows/day) while making a dead loop visible
# within minutes (the 2026-07-01 outage was 4h25m of undetectable silence).
_DISPATCH_HEARTBEAT_SECONDS = 300
async def _emit_dispatcher_heartbeat(self) -> None:
"""Periodic dispatcher.alive audit row — a dead loop becomes detectable.
The dispatch loop can die silently (task cancelled, unhandled exit) and
its stdout dies with the container; audit_log survives both. Absence of
a fresh heartbeat row = loop dead, distinguishable from "no work".
"""
now = datetime.now(UTC)
last = getattr(self, "_last_dispatch_heartbeat", None)
if last is not None and (now - last).total_seconds() < (
self._DISPATCH_HEARTBEAT_SECONDS
):
return
self._last_dispatch_heartbeat = now
self._fire_audit(
event_type="dispatcher.alive",
agent_slug="orchestrator",
details={"interval_seconds": self._DISPATCH_HEARTBEAT_SECONDS},
)
async def _dispatcher_loop(self) -> None:
"""
Main dispatcher loop - periodically checks for work and spawns agents.
This is the BRAIN of the orchestrator. It:
1. Queries for tasks needing work (pending, awaiting_qa, etc.)
2. Queries for events needing attention (blockers, escalations)
3. Spawns appropriate agents with task assignments
Hybrid timing: a poll (dispatcher_interval) guarantees progress even
without external signals, while `_dispatch_wake` lets API routes
kick the loop for immediate reactions after status transitions.
"""
while self._running:
try:
# Wait either for an explicit wake signal or the poll timeout,
# whichever comes first. asyncio.wait_for re-raises
# TimeoutError when the poll window expires, which we treat
# as "run dispatch anyway".
import contextlib
with contextlib.suppress(TimeoutError):
await asyncio.wait_for(
self._dispatch_wake.wait(),
timeout=self.dispatcher_interval,
)
self._dispatch_wake.clear()
await self._refresh_grok_auth()
await self._dispatch_all_work()
await self._emit_dispatcher_heartbeat()
except asyncio.CancelledError:
break
except Exception as e:
logger.error("Dispatcher loop error", error=str(e))
async def _refresh_grok_auth(self) -> None:
"""Keep the host SuperGrok token live so grok agents never mount a dead one.
The grok access token has a ~6h server-set TTL and headless grok cannot
self-refresh — on an expired token it hangs at an interactive login
prompt. The per-agent mount is read-only, so the orchestrator refreshes
the host ``auth.json`` itself (refresh-token grant) before expiry; agents
then mount a fresh credential. Best-effort, throttled, and serial (run
once per dispatch tick) so concurrent refreshes can't rotate the
refresh-token out from under each other. Never breaks the loop.
"""
now = datetime.now(UTC)
next_check = getattr(self, "_grok_auth_next_check", None)
if next_check is not None and now < next_check:
return
self._grok_auth_next_check = now + timedelta(seconds=60)
try:
from roboco.llm.providers import grok_auth
from roboco.llm.providers.grok import GROK_AUTH_HOST_PATH
auth_path = Path(GROK_AUTH_HOST_PATH) / "auth.json"
status = await asyncio.to_thread(grok_auth.refresh_if_stale, auth_path)
if status == "refreshed":
logger.info("grok auth token refreshed")
elif status == "failed":
logger.warning(
"grok auth refresh failed; agents may hit an expired token"
)
except Exception as exc:
logger.error("grok auth refresh hook error", error=str(exc))
async def _reconcile_orphan_claims_on_startup(self) -> None:
"""Roll back tasks left in CLAIMED/IN_PROGRESS without a branch.
A task in CLAIMED/IN_PROGRESS with ``branch_name IS NULL`` is an
orphan: ``_finalize_claim`` flushed the status before branch creation
failed (or before claim rollback became atomic). The next claim then
fails non-idempotent on ``git checkout -b`` because the on-disk
branch may exist while the DB state is stale.
Opens its own session via the factory; the logic itself lives in
``_reconcile_with_service`` so tests can drive it against an
injected session without the factory dance. Best-effort: if
reconciliation fails, log and continue — startup must not be
blocked by a single bad row.
"""
from roboco.db.base import get_session_factory
from roboco.services.task import TaskService
factory = get_session_factory()
try:
async with factory() as db:
svc = TaskService(db)
await self._reconcile_with_service(svc)
await db.commit()
except Exception as exc:
logger.error("startup reconcile failed; continuing", error=str(exc))
async def _reconcile_orphan_spawn_sessions(self) -> int:
"""Close agent_spawn_sessions rows left open by a prior crash.
usage.get_summary / get_time_series filter ``ended_at IS NOT NULL``,
so an open row whose container is gone is permanently excluded from
usage/cost rollups. Close each open session whose agent slug is NOT
in ``self._instances`` (the re-adopted running set) with
``ended_at=now`` and ``exit_reason='abandoned'``. Running agents
stay open for their live finalize. Best-effort; never blocks startup.
"""
try:
from sqlalchemy import select, update
from roboco.db.base import get_session_factory
from roboco.db.tables import AgentSpawnSessionTable
except ImportError:
return 0
try:
running = set(self._instances.keys())
session_factory = get_session_factory()
async with session_factory() as db:
rows = (
(
await db.execute(
select(AgentSpawnSessionTable).where(
AgentSpawnSessionTable.ended_at.is_(None)
)
)
)
.scalars()
.all()
)
orphans = [r for r in rows if r.agent_slug not in running]
if not orphans:
return 0
now = datetime.now(UTC)
await db.execute(
update(AgentSpawnSessionTable)
.where(AgentSpawnSessionTable.id.in_([r.id for r in orphans]))
.values(ended_at=now, exit_reason="abandoned")
)
await db.commit()
return len(orphans)
except Exception as exc:
logger.warning(
"Failed to reconcile orphan spawn sessions",
error=str(exc),
)
return 0
async def _reconcile_with_service(self, svc: "TaskService") -> None:
"""Inner reconcile loop, parameterised by the TaskService to use.
Same shape as ``_reap_with_service`` — extracted so tests can
bypass ``get_session_factory`` and drive the logic directly.
"""
from roboco.utils.converters import require_uuid
candidates = await svc.list_in_progress_or_claimed()
orphans = [t for t in candidates if not t.branch_name]
if not orphans:
logger.info("startup reconcile: no orphan claims")
return
for t in orphans:
task_id = require_uuid(t.id)
try:
await svc.unclaim_for_reaper(task_id)
logger.warning(
"startup reconcile: orphan claim rolled back",
task_id=str(task_id),
had_status=str(t.status),
)
except Exception as exc:
logger.error(
"startup reconcile: rollback failed",
task_id=str(t.id),
error=str(exc),
)
async def _reap_stale_claims(self) -> None:
"""Release claimed/in_progress tasks whose holder hasn't heart-beat in TTL.
Closes the "dead container squats task forever" failure mode that
the schema hinted at (``last_heartbeat_at`` since migration 006) but
no code enforced. The runtime decision (cutoff, iteration) lives
here in the orchestrator; the actual UPDATE statements live in
``TaskService.unclaim_for_reaper``.
Opens a fresh per-tick session — short-lived because the reaper
runs on every dispatch cycle and the work is cheap (one SELECT
plus N UPDATEs for the typically-empty stale set). Tests that
need to inject a mock service do so by building an instance via
``__new__`` (bypassing this method) and calling
``_reap_with_service`` directly.
"""
from roboco.db.base import get_session_factory
from roboco.services.task import TaskService
factory = get_session_factory()
async with factory() as db:
svc = TaskService(db)
await self._reap_with_service(svc)
await db.commit()
await self._sandbox_janitor_sweep()
async def _sandbox_janitor_sweep(self) -> None:
"""Best-effort: remove sandbox containers whose owner agent is gone.
Cheap (a couple of docker calls) and error-isolated — the provisioner
itself never raises out of ``janitor_sweep``, so a hiccup here never
blocks the reaper tick it rides alongside.
"""
if not settings.sandbox_db_enabled:
return
with contextlib.suppress(Exception):
await self._sandbox.janitor_sweep()
def _assignee_has_active_instance(self, task: Any) -> bool:
"""True if the task's assignee currently holds a live (ACTIVE) container.
The heartbeat only approximates liveness. A developer deep in an
edit/test cycle can go longer than the heartbeat TTL between gateway
calls, so a heartbeat-only reaper releases claims out from under agents
that are alive and working — churning the task (and risking a double
spawn against the still-running container). The agent-instance registry
is the ground truth; defer to it when present. Defensive on missing
fields so a heartbeat-only caller (and the reaper's own unit tests)
behave exactly as before.
"""
owner = getattr(task, "assigned_to", None) or getattr(task, "claimed_by", None)
if not owner:
return False
instances = getattr(self, "_instances", None)
if not instances:
return False
instance = instances.get(self._resolve_agent_slug(str(owner)))
return instance is not None and instance.state == AgentState.ACTIVE
def _assignee_is_provider_parked(self, task: Any) -> bool:
"""True if the task's assignee is parked waiting for a provider to recover.
A provider-parked agent (session-limit / overload / grok-429) is OFFLINE
with a dead container and a ``rate_limit_lifted`` WaitingRecord; the
probe-resume loop owns its recovery. The stale-claim reaper must skip it
so the claim survives until the probe revives the agent — reaping would
release the claim to pending and probe-success would then respawn the
agent on a task it no longer owns. Defensive on a missing registry.
"""
owner = getattr(task, "assigned_to", None) or getattr(task, "claimed_by", None)
if not owner:
return False
records = getattr(self, "_waiting_records", None)
if not records:
return False
slug = self._resolve_agent_slug(str(owner))
record = records.get(slug)
return record is not None and record.waiting_for == "rate_limit_lifted"
async def _agent_holds_live_claim(self, slug: str) -> bool | None:
"""Whether ``slug`` currently owns a non-terminal task.
Used by ``_readopt_running_agents`` to tell a still-useful running
container (the agent is mid-task) from a zombie left over after a prior
orchestrator released the claim: registering a zombie ACTIVE would block
the spawn gate from re-dispatching that slug until the stale container is
eventually noticed (#72). Returns True when the slug owns a non-terminal
task, False when it owns nothing (zombie), and None on a lookup error
(indeterminate — the caller falls back to today's register behaviour so a
startup DB hiccup can't regress the cold-start double-spawn protection).
"""
from sqlalchemy import select
from roboco.db.base import get_db_context
from roboco.db.tables import TaskTable
from roboco.models.base import TaskStatus
agent_uuid = AGENT_UUIDS.get(slug)
if agent_uuid is None:
return False # unknown slug owns nothing by definition
try:
async with get_db_context() as db:
result = await db.execute(
select(TaskTable.id)
.where(
TaskTable.assigned_to == agent_uuid,
TaskTable.status.notin_(
(TaskStatus.COMPLETED, TaskStatus.CANCELLED)
),
)
.limit(1)
)
return result.first() is not None
except Exception:
logger.warning(
"readopt live-claim lookup failed; falling back to register",
slug=slug,
)
return None
async def _read_container_auth_env(
self, container_name: str
) -> tuple[str, str, str] | None:
"""Read (token, agent_id, role) from a running agent container's env.
Returns None on any probe failure or missing var so the caller can
skip the container (best-effort — the reaper still covers it).
"""
try:
proc = await asyncio.create_subprocess_exec(
"docker",
"exec",
container_name,
"printenv",
"ROBOCO_AGENT_TOKEN",
"ROBOCO_AGENT_ID",
"ROBOCO_AGENT_ROLE",
stdout=asyncio.subprocess.PIPE,
stderr=asyncio.subprocess.DEVNULL,
)
stdout, _ = await proc.communicate()
except Exception:
return None
if proc.returncode != 0:
return None
token: str | None = None
agent_id_env: str | None = None
role_env: str | None = None
for line in stdout.decode("utf-8", "replace").splitlines():
if line.startswith("ROBOCO_AGENT_TOKEN="):
token = line[len("ROBOCO_AGENT_TOKEN=") :]
elif line.startswith("ROBOCO_AGENT_ID="):
agent_id_env = line[len("ROBOCO_AGENT_ID=") :]
elif line.startswith("ROBOCO_AGENT_ROLE="):
role_env = line[len("ROBOCO_AGENT_ROLE=") :]
if not token or not agent_id_env or not role_env:
return None
return token, agent_id_env, role_env
async def _heal_stale_agent_tokens(self) -> int:
"""Kill running agent containers whose ROBOCO_AGENT_TOKEN no longer
verifies against the current ``ROBOCO_AGENT_AUTH_SECRET``.
A token is baked into the container env at spawn (``_append_agent_auth_env``
signs with the orchestrator's secret at that moment). If the secret later
drifts — a `.env` change, a compose recreate that reloads the
orchestrator's env without recreating the agent containers, an image
redeploy — the surviving agent keeps sending its old token and the
middleware 401s every verb with "signature mismatch". The container stays
alive (heartbeating), so the reaper never reclaims it and no fresh agent
spawns: the fleet stalls. This self-heals it at startup by killing each
stale-token container so the normal dispatch re-spawns it with a freshly
signed token.
Inert when the secret is unset (dev): ``verify_agent_token`` fails for
every token without a secret, so the heal would kill the whole fleet —
gated to prod-only. Best-effort: a probe failure leaves the container
alone (the reaper's own liveness path still covers it).
"""
from roboco.agents_config import _auth_secret, verify_agent_token
if not _auth_secret():
return 0
killed = 0
for slug in AGENT_IMAGES:
try:
is_running, _ = await self._inspect_container_state(
f"roboco-agent-{slug}"
)
except Exception:
continue
if not is_running:
continue
env = await self._read_container_auth_env(f"roboco-agent-{slug}")
if env is None:
continue
token, agent_id_env, role_env = env
team = get_agent_team(agent_id_env) or ""
# Verify against the UUID the MCP servers actually send as
# X-Agent-ID, not the container-env ROBOCO_AGENT_ID (a slug on
# pre-fix containers). A stale container spawned before the
# slug→UUID fix carries a slug-signed token + a slug
# ROBOCO_AGENT_ID, so verifying against the slug would PASS and
# leave the stale container running (its MCP server still 401s
# sending the UUID). Resolving to the UUID makes the heal reject
# the slug-signed token and kill the container so it respawns
# with a UUID-signed one. AGENT_UUIDS is slug→UUID keyed, so a
# UUID input falls back to itself.
agent_uuid = AGENT_UUIDS.get(agent_id_env, agent_id_env)
if verify_agent_token(token, agent_uuid, role_env, team):
continue
logger.warning(
"Killing agent with a stale auth token at startup; the reaper "
"will re-spawn it with a freshly signed token",
slug=slug,
)
await self._remove_container(
f"roboco-agent-{slug}",
teardown_sandbox=False,
stop_reason="stale_token_heal",
)
killed += 1
if killed:
logger.info("Healed stale agent tokens at startup", count=killed)
return killed
async def _readopt_running_agents(self) -> int:
"""Re-adopt still-running agent containers into ``_instances`` at startup.
An orchestrator restart loses the in-memory ``_instances`` registry while
the agent containers keep running. The reaper has a Docker-liveness
fallback for that (``_assignee_container_running``), but the spawn gate's
``_is_agent_active`` does NOT — so after a restart it sees a live agent as
inactive and can double-spawn it onto work its forgotten-but-running
container is already doing. Probe each known agent slug's container (the
same ``docker inspect`` the reaper uses) and register a minimal ACTIVE
instance for any that is running, not already tracked, AND still holds a
live (non-terminal) claim — a running container whose claim a prior
orchestrator already released is a zombie and is skipped so it can't
block the spawn gate from re-dispatching that slug (#72). Inert when
nothing is running (degrades to today's cold start) and best-effort: a
probe or claim-lookup error leaves that slot untracked / falls back to
registering (the reaper's own fallback still covers it). Returns the
number re-adopted.
"""
readopted = 0
for slug in AGENT_IMAGES:
if slug in self._instances:
continue
try:
is_running, _ = await self._inspect_container_state(
f"roboco-agent-{slug}"
)
except Exception:
continue
if not is_running:
continue
# Capture the real container id: ``_check_health`` skips instances
# with ``container_id is None``, so a re-adopted instance without
# the id would be invisible to the health loop and strand the task
# under a phantom ACTIVE instance. Best-effort — a probe failure
# degrades to None (reaper's Docker-liveness fallback covers it).
container_id: str | None = None
try:
container_id = await self._resolve_container_id(f"roboco-agent-{slug}")
except Exception:
container_id = None
# #72: a running container whose slug no longer holds a live claim is
# a zombie from a prior orchestrator that released the claim — skip it
# so it can't block re-dispatch of the slug. ``None`` (lookup error)
# falls back to registering: a startup DB hiccup must not regress the
# cold-start double-spawn protection this readopt exists to provide.
holds_claim = await self._agent_holds_live_claim(slug)
if holds_claim is False:
logger.info(
"readopt skipped a running zombie container (no live claim)",
slug=slug,
)
continue
self._instances[slug] = AgentInstance(
agent_id=slug,
state=AgentState.ACTIVE,
container_id=container_id,
)
readopted += 1
if readopted:
logger.info(
"re-adopted running agent containers at startup", count=readopted
)
return readopted
async def _assignee_container_running(self, task: Any) -> bool:
"""Docker-liveness fallback for the reaper on an instance-registry MISS.
``_assignee_has_active_instance`` reads the in-memory ``_instances``
registry, which is lost on an orchestrator restart while the agent's
container keeps running. Without a fallback the heartbeat-stale reaper
then releases a task out from under a live agent the orchestrator has
merely forgotten — registry amnesia, the over-reap that hit be-dev-1.
This asks Docker directly, but ONLY on a true registry miss: a known
instance (ACTIVE or stopped) is authoritative and not second-guessed,
and an uninitialised registry (``None`` — e.g. a unit-test harness) is
left to the existing behaviour. Any error (no docker binary, inspect
fails) yields False, so non-Docker test/dev contexts are unaffected.
"""
instances = getattr(self, "_instances", None)
if instances is None:
return False
owner = getattr(task, "assigned_to", None) or getattr(task, "claimed_by", None)
if not owner:
return False
slug = self._resolve_agent_slug(str(owner))
if slug in instances:
return False
try:
is_running, _ = await self._inspect_container_state(f"roboco-agent-{slug}")
except Exception:
return False
return is_running
def _wedged_grok_slug(
self, task: Any, last_heartbeat: "datetime | None"
) -> str | None:
"""Slug of an ACTIVE GROK container holding ``task`` and idle past the kill TTL.
``_assignee_has_active_instance`` shields a live container from the
reaper — correct for a Claude agent quiet during a long edit/test cycle.
A wedged GROK container is the one case that breaks: ACTIVE *and*
silent (an idle model call fires no gateway verb), so its heartbeat never
advances and the skip would protect it forever. Returns the slug only for
a GROK instance idle past the grok-kill TTL — a recent heartbeat, no
owner, a non-GROK provider, or a non-ACTIVE instance all yield ``None``.
"""
from roboco.models.base import ModelProvider
cutoff = datetime.now(UTC) - timedelta(
seconds=getattr(self, "_grok_idle_kill_ttl", 900)
)
if last_heartbeat is not None and last_heartbeat >= cutoff:
return None
owner = getattr(task, "assigned_to", None) or getattr(task, "claimed_by", None)
if not owner:
return None
slug = self._resolve_agent_slug(str(owner))
instance = (getattr(self, "_instances", None) or {}).get(slug)
config = getattr(instance, "config", None)
is_active_grok = (
instance is not None
and instance.state == AgentState.ACTIVE
and config is not None
and config.provider_type == ModelProvider.GROK.value
)
return slug if is_active_grok else None
async def _maybe_kill_wedged_grok(
self, task: Any, last_heartbeat: "datetime | None"
) -> bool:
"""Kill + evict a wedged GROK container so this tick's reaper frees its task.
On a kill the container is removed (its logs dumped to disk first) and
dropped from ``_instances``. Returns True only when a container was
actually killed; see :meth:`_wedged_grok_slug` for the eligibility rule.
"""
slug = self._wedged_grok_slug(task, last_heartbeat)
if slug is None:
return False
try:
await self._remove_container(
f"roboco-agent-{slug}", stop_reason="reaper_wedged_grok"
)
except Exception as exc:
logger.error(
"wedged-grok kill failed; will retry next tick",
agent_id=slug,
error=str(exc),
)
return False
self._instances.pop(slug, None)
logger.warning(
"wedged grok container killed and evicted",
agent_id=slug,
task_id=str(getattr(task, "id", "")),
)
return True
def _stuck_claude_slug(
self, task: Any, last_heartbeat: "datetime | None"
) -> str | None:
"""Slug of an ACTIVE non-GROK container holding ``task``, stuck past the TTL.
The reaper's live-container skip shields a quiet agent during a long
edit/test cycle — correct for a working agent (it fires gateway verbs
every few minutes, advancing its heartbeat). A non-GROK agent stuck in a
non-verb loop is ACTIVE yet silent, so the skip would protect its claim
forever (#73). Returns the slug only for a non-GROK ACTIVE instance whose
heartbeat has been stale longer than ``claude_stuck_kill_seconds`` — a
recent heartbeat, no owner, a GROK provider (handled by the wedged-grok
path), or a non-ACTIVE instance all yield ``None``.
"""
from roboco.models.base import ModelProvider
cutoff = datetime.now(UTC) - timedelta(
seconds=getattr(self, "_claude_stuck_kill_ttl", 3600)
)
if last_heartbeat is not None and last_heartbeat >= cutoff:
return None
owner = getattr(task, "assigned_to", None) or getattr(task, "claimed_by", None)
if not owner:
return None
slug = self._resolve_agent_slug(str(owner))
instance = (getattr(self, "_instances", None) or {}).get(slug)
config = getattr(instance, "config", None)
is_active_non_grok = (
instance is not None
and instance.state == AgentState.ACTIVE
and config is not None
and config.provider_type != ModelProvider.GROK.value
)
return slug if is_active_non_grok else None
async def _maybe_kill_stuck_claude(
self, task: Any, last_heartbeat: "datetime | None"
) -> bool:
"""Kill + evict a stuck non-GROK container so the reaper frees its task.
On a kill the container is removed and dropped from ``_instances``.
Returns True only when a container was actually killed; see
:meth:`_stuck_claude_slug` for the eligibility rule (#73).
"""
slug = self._stuck_claude_slug(task, last_heartbeat)
if slug is None:
return False
try:
await self._remove_container(
f"roboco-agent-{slug}", stop_reason="reaper_stuck_claude"
)
except Exception as exc:
logger.error(
"stuck-claude kill failed; will retry next tick",
agent_id=slug,
error=str(exc),
)
return False
self._instances.pop(slug, None)
logger.warning(
"stuck non-grok container killed and evicted",
agent_id=slug,
task_id=str(getattr(task, "id", "")),
)
return True
async def _maybe_recover_broken_gateway(self, task: Any) -> bool:
"""Kill + evict a live agent whose gateway is broken past the grace window.
The reaper's live-skip protects a running container from a stale-heartbeat
reap — right for a healthy agent quiet during a long edit/test cycle, but
it would shield a broken-but-alive agent (a corrupted gateway firing no
verb) forever. This probes the gateway out-of-band and, once it has been
broken longer than ``gateway_health_grace_seconds`` (so a transient probe
miss is tolerated), kills + evicts the container so the reaper falls
through to release + respawn. Returns True only on a kill; a healthy
gateway, an inconclusive probe, or a still-within-grace breakage returns
False (the live container is spared). Gated by ``gateway_health_enabled``.
"""
if not settings.gateway_health_enabled:
return False
owner = getattr(task, "assigned_to", None) or getattr(task, "claimed_by", None)
if not owner:
return False
slug = self._resolve_agent_slug(str(owner))
if not await self._gateway_broken_past_grace(slug):
return False
try:
await self._remove_container(
f"roboco-agent-{slug}", stop_reason="gateway_health_recovery"
)
except Exception as exc:
logger.error(
"broken-gateway kill failed; will retry next tick",
agent_id=slug,
error=str(exc),
)
return False
self._instances.pop(slug, None)
self._gateway_broken_since.pop(slug, None)
logger.warning(
"broken-gateway agent killed and evicted",
agent_id=slug,
task_id=str(getattr(task, "id", "")),
)
return True
async def _gateway_broken_past_grace(self, slug: str) -> bool:
"""True when ``slug``'s gateway has probed broken longer than the grace.
Probe-inconclusive (None) or healthy clears the grace mark and returns
False; the first broken sighting records the mark and returns False (one
grace tick); a breakage older than ``gateway_health_grace_seconds`` (or a
test-injected ``_gateway_health_grace``) returns True.
"""
healthy = await self._probe_gateway_health(slug)
if healthy is None or healthy:
self._gateway_broken_since.pop(slug, None)
return False
now = datetime.now(UTC)
first_seen = self._gateway_broken_since.get(slug)
if first_seen is None:
self._gateway_broken_since[slug] = now
return False
grace = getattr(self, "_gateway_health_grace", None)
if grace is None:
grace = settings.gateway_health_grace_seconds
return (now - first_seen).total_seconds() >= grace
async def _should_skip_live_reap(self, t: Any, ts: Any) -> bool:
"""True when a live container should be spared from reaping.
A live container normally protects its task; on a registry MISS (e.g. the
orchestrator restarted and forgot a still-running container) fall back to
asking Docker. A live container is spared UNLESS it is wedged (grok) or
its gateway is broken-but-alive past the grace window — both checks kill +
evict it (returning False here) so the caller falls through to release +
respawn. Short-circuits like the original ``and``: when not live, neither
kill nor recovery check is awaited.
"""
live = self._assignee_has_active_instance(
t
) or await self._assignee_container_running(t)
return (
live
and not await self._maybe_kill_wedged_grok(t, ts)
and not await self._maybe_kill_stuck_claude(t, ts)
and not await self._maybe_recover_broken_gateway(t)
)
async def _reap_with_service(self, svc: "TaskService") -> None:
"""Inner reap loop, parameterized by the TaskService to use.
Wraps each ``unclaim_for_reaper`` in try/except so a single bad row
doesn't abort the dispatch tick — the reaper must keep ticking even
if one task's release somehow fails. A claim whose assignee still has
a live container is skipped: the heartbeat is a stale proxy there, and
reaping a working agent only churns the task.
"""
from roboco.utils.converters import require_uuid
cutoff = datetime.now(UTC) - timedelta(seconds=self._claim_heartbeat_ttl)
candidates = await svc.list_in_progress_or_claimed()
for t in candidates:
ts = t.last_heartbeat_at
if ts is None or ts < cutoff:
# A live container is spared unless it is wedged (grok) or its
# gateway is broken-but-alive past the grace window — see
# _should_skip_live_reap, which kills + evicts those so we fall
# through to release + respawn.
if await self._should_skip_live_reap(t, ts):
continue
# A provider-parked agent (session-limit / overload / grok-429)
# is OFFLINE with a dead container and a ``rate_limit_lifted``
# WaitingRecord. The probe-resume loop owns its recovery — do
# NOT reap the claim, or probe-success would respawn the agent
# on a task it no longer owns.
if self._assignee_is_provider_parked(t):
continue
task_id = require_uuid(t.id)
try:
await svc.unclaim_for_reaper(task_id)
logger.warning(
"stale claim reaped",
task_id=str(task_id),
last_heartbeat=ts.isoformat() if ts else None,
)
except Exception as exc:
logger.error(
"stale-claim reap failed; continuing",
task_id=str(task_id),
error=str(exc),
)
async def _dispatch_all_work(self) -> None:
"""Run all dispatchers to check for and assign work.
Each dispatcher is isolated: if one raises (e.g., a transient API
error), the rest still run in this tick instead of waiting for the
next one.
`_tick_handled_tasks` gives downstream dispatchers a way to
skip tasks that an earlier dispatcher already acted on this
tick. Order-dependent bugs (like the Fix-B scenario where
`_dispatch_qa_work` claimed for QA and the next dispatcher
re-spawned the dev on the same claimed row) are defanged by
early dispatchers marking the task handled.
The stale-claim reaper runs first, before any dispatcher tries to
spawn an agent for a task whose previous holder is dead. Without
this ordering, the spawn pass could race against a stale claim and
skip work the reaper would have freed in the same tick.
"""
self._tick_handled_tasks = set()
# Free any tasks whose claim went stale before the spawn pass runs.
# Wrapped because a reaper failure must not block dispatch — the
# next tick will retry.
try:
await self._reap_stale_claims()
except Exception as e:
logger.error("Stale-claim reaper failed; continuing tick", error=str(e))
# Enforce the GROK cost ceiling (budget kill-switch parity). Wrapped so a
# failure never blocks dispatch; the next tick retries.
try:
await self._enforce_grok_cost_budget()
except Exception as e:
logger.error("Grok cost-budget sweep failed; continuing tick", error=str(e))
dispatchers: list[tuple[str, Any]] = []
async with httpx.AsyncClient(
timeout=30.0, headers=_system_api_headers()
) as client:
dispatchers = [
("pm_work", self._dispatch_pm_work(client)),
("pm_closure_work", self._dispatch_pm_closure_work(client)),
(
"revision_coordination",
self._dispatch_revision_coordination_roots(client),
),
("dev_work", self._dispatch_dev_work(client)),
("qa_work", self._dispatch_qa_work(client)),
("pr_review_work", self._dispatch_pr_review_work(client)),
("pr_gate_work", self._dispatch_pr_gate_work(client)),
("doc_work", self._dispatch_doc_work(client)),
("pm_review_work", self._dispatch_pm_review_work(client)),
("marketing_work", self._dispatch_marketing_work(client)),
("blocker_work", self._dispatch_blocker_work(client)),
(
"claimed_without_agent",
self._dispatch_claimed_without_agent(client),
),
("escalation_work", self._dispatch_escalation_work(client)),
("approval_work", self._dispatch_approval_work(client)),
("a2a_work", self._dispatch_a2a_work(client)),
("audit_work", self._dispatch_audit_work(client)),
("detect_stuck_tasks", self._detect_stuck_tasks(client)),
]
for name, coro in dispatchers:
try:
await coro
except Exception as e:
logger.error(
"Dispatcher raised; continuing with next dispatcher",
dispatcher=name,
error=str(e),
)
# =========================================================================
# SMART DISPATCHER - TASK-BASED DISPATCHERS
# =========================================================================
_PM_AGENTS: ClassVar[frozenset[str]] = frozenset(
{
"main-pm",
"be-pm",
"fe-pm",
"ux-pm",
}
)
# Board reviewers. They advise — review + record requirements + escalate —
# but do not build or delegate. Dispatched once per assigned board task.
_BOARD_AGENTS: ClassVar[frozenset[str]] = frozenset(
{
"product-owner",
"head-marketing",
}
)
# Use foundation's default; keep the local name for back-compat.
_PM_RESPAWN_MAX_UNPRODUCTIVE = _AGENT_LOOP_BUDGET.pm_respawn_max_unproductive
_PM_RESPAWN_MAX_TRACING_RESETS = _AGENT_LOOP_BUDGET.pm_respawn_max_tracing_resets
_PM_RESPAWN_MAX_REVISIT_RESETS = _AGENT_LOOP_BUDGET.pm_respawn_max_revisit_resets
_PM_RESPAWN_TRIP_COOLDOWN_SECONDS = (
_AGENT_LOOP_BUDGET.pm_respawn_trip_cooldown_seconds
)
def _respawn_status_change_resets(
self,
key: tuple[str, Any],
record: dict[str, Any],
current_status: Any,
now: datetime,
) -> bool:
"""Handle a status CHANGE; True when it resets the strike counter.
A status never seen on this (agent, task) is genuine forward progress
and fully resets, exactly as before. A REVISITED status — the A<->B
oscillation (blocked <-> in_progress) that changes status on every
spawn while advancing nothing (2026-07-02: a dev looped 2h/8 spawns
without tripping the gate) — gets a bounded reset budget mirroring
tracing_resets, after which strikes accrue. seen_statuses is
in-memory only (not a tracker column): after a restart it rebuilds
from observed statuses, which can only under-gate briefly — never
over-gate.
"""
agent_slug, task_id = key
seen = record.get("seen_statuses") or [record.get("last_status")]
if current_status not in seen:
self._pm_respawn_tracker[key] = {
"count": 1,
"last_status": current_status,
"last_check": now,
"seen_statuses": [*seen, current_status],
}
self._schedule_respawn_persist(
agent_slug, str(task_id), self._pm_respawn_tracker[key]
)
return True
record["last_status"] = current_status
revisits = record.get("revisit_resets", 0)
if revisits < self._PM_RESPAWN_MAX_REVISIT_RESETS:
record["revisit_resets"] = revisits + 1
record["count"] = 1
record["last_check"] = now
record["notified"] = False
self._schedule_respawn_persist(agent_slug, str(task_id), record)
return True
logger.warning(
"PM respawn status ping-pong budget exhausted — "
"revisited statuses no longer reset the strike counter",
agent_id=agent_slug,
task_id=str(task_id),
task_status=current_status,
revisit_resets=revisits,
)
return False
async def _pm_tracing_gap_reset(
self,
agent_slug: str,
task_id: Any,
record: dict[str, Any],
current_status: Any,
now: datetime,
) -> bool:
"""Reset the strike counter when the PM made a rule-following retry.
A tracing_gap normally means the agent is advancing through a verb
chain, so reset — but only up to ``_PM_RESPAWN_MAX_TRACING_RESETS``.
A task whose every respawn trips the same gap is wedged, not
progressing, so cap the resets and let strikes accrue once the
budget is exhausted. Returns True when the counter was reset.
"""
if not await self._pm_made_rule_following_retry(agent_slug, task_id, record):
return False
resets = record.get("tracing_resets", 0)
if resets < self._PM_RESPAWN_MAX_TRACING_RESETS:
record["tracing_resets"] = resets + 1
record["count"] = 1
record["last_check"] = now
record["notified"] = False
self._schedule_respawn_persist(agent_slug, str(task_id), record)
return True
logger.warning(
"PM respawn tracing_gap reset budget exhausted — "
"treating recurring gap as a stuck loop",
agent_id=agent_slug,
task_id=task_id,
task_status=current_status,
tracing_resets=resets,
)
return False
def _pm_cooldown_gate(
self,
agent_slug: str,
task_id: Any,
record: dict[str, Any],
now: datetime,
) -> bool | None:
"""Self-heal a previously-tripped gate after a cooldown.
Returns True to keep gating, False to let the spawn through after a
cooldown reset, or None when the gate hasn't tripped yet (caller
continues to the increment path). last_check is frozen at the trip
tick; this branch returns before the increment below updates it.
"""
if not (
record["count"] > self._PM_RESPAWN_MAX_UNPRODUCTIVE
and record.get("notified")
):
return None
elapsed: bool = (now - record["last_check"]).total_seconds() > (
self._PM_RESPAWN_TRIP_COOLDOWN_SECONDS
)
if elapsed:
record["count"] = 1
record["last_check"] = now
record["notified"] = False
self._schedule_respawn_persist(agent_slug, str(task_id), record)
return not elapsed
async def _pm_respawn_should_gate(
self, agent_slug: str, task: dict[str, Any]
) -> bool:
"""Return True when the respawn should be skipped (loop detected).
Tracks (agent_slug, task_id) -> count of consecutive spawns where
the task's status did not advance. When the task status changes,
the counter resets. Once the count hits the threshold, the spawn
is skipped and a warning logged; operators must intervene.
Tracing-gap reset
-----------------
With the gateway claim-time gates installed, a rule-following PM
will hit ``PARENT_NOT_CLAIMED`` (a ``tracing_gap`` envelope) and
the prompt will tell it to call the prerequisite verb first.
Each retry leaves the task status unchanged but the agent IS
making progress through the verb chain. Counting that as a
strike kills rule-followers.
Solution: before incrementing on a same-status spawn, check
``audit_log`` for a ``gateway.rejected`` row tagged
``reason == "tracing_gap"`` from this (agent, task) since the
last check. If found, reset the counter — the agent followed
the rules, not stuck.
Audit lookup is best-effort: any failure falls through to the
legacy strike behavior so audit problems don't break the gate.
"""
task_id = task.get("id")
if not task_id:
return False
key = (agent_slug, task_id)
current_status = task.get("status")
record = self._pm_respawn_tracker.get(key)
now = datetime.now(UTC)
if record is None:
self._pm_respawn_tracker[key] = {
"count": 1,
"last_status": current_status,
"last_check": now,
"seen_statuses": [current_status],
}
self._schedule_respawn_persist(
agent_slug, str(task_id), self._pm_respawn_tracker[key]
)
return False
if record.get("last_status") != current_status and (
self._respawn_status_change_resets(key, record, current_status, now)
):
return False
# ponytail: helpers hold the two resettable sub-loops (tracing-gap,
# cooldown); main fn just routes. Inline again if either grows a
# second distinct reset path.
if await self._pm_tracing_gap_reset(
agent_slug, task_id, record, current_status, now
):
return False
# Already tripped on a PREVIOUS tick (notified flipped): the count is
# frozen past the threshold and last_check is frozen at the trip tick,
# so a deploy that fixed the underlying loop (auth/prompt/schema) can
# self-heal after a cooldown instead of wedging until manual DB
# surgery. A still-wedged task re-trips after the threshold (bounded
# re-burn: ~3 spawns per cooldown window); a fixed one advances and
# the status-change path fully resets the counter.
gate = self._pm_cooldown_gate(agent_slug, task_id, record, now)
if gate is not None:
return gate
record["count"] += 1
record["last_check"] = now
self._schedule_respawn_persist(agent_slug, str(task_id), record)
tripped: bool = record["count"] > self._PM_RESPAWN_MAX_UNPRODUCTIVE
if tripped:
logger.warning(
"PM respawn loop detected — skipping spawn",
agent_id=agent_slug,
task_id=task_id,
task_status=current_status,
spawn_attempts=record["count"],
threshold=self._PM_RESPAWN_MAX_UNPRODUCTIVE,
hint=(
"Agent repeatedly spawned without advancing task state. "
"Investigate prompt/schema drift or escalate manually."
),
)
# A skipped spawn pauses the loop but can't advance the task; alert
# an overseer once so a wedged agent isn't silently stranded.
if not record.get("notified"):
record["notified"] = True
self._schedule_respawn_persist(agent_slug, str(task_id), record)
await self._notify_stuck_agent(agent_slug, task_id, current_status)
return tripped
async def _notify_stuck_agent(
self, agent_slug: str, task_id: str, task_status: str | None
) -> None:
"""One-shot alert to the CEO that an agent is wedged in a respawn loop.
Best-effort: a notification failure must not wedge dispatch, so any
error is logged and swallowed.
"""
from roboco.services.notification import NotificationService
try:
await NotificationService().send_stuck_agent_notification(
task_id=task_id,
agent_slug=agent_slug,
task_status=task_status or "unknown",
to_agent="ceo",
)
except Exception as exc:
logger.warning(
"Failed to send stuck-agent notification",
agent_id=agent_slug,
task_id=task_id,
error=str(exc),
)
async def _pm_made_rule_following_retry(
self,
agent_slug: str,
task_id: str,
record: dict[str, Any],
) -> bool:
"""Did the agent emit a ``tracing_gap`` envelope since the last check?
Returns ``False`` for unknown slugs (defensive — the audit query
needs an agent UUID, and we'd rather fall through to the legacy
strike behavior than crash). Returns ``False`` if the audit
lookup raises — observability must never block the gate.
"""
agent_uuid_str = AGENT_UUIDS.get(agent_slug)
if not agent_uuid_str:
return False
from uuid import UUID
try:
agent_uuid = UUID(agent_uuid_str)
task_uuid = UUID(task_id)
except (ValueError, TypeError):
return False
since = record.get("last_check") or datetime.now(UTC)
from roboco.services.audit import get_audit_service
audit = get_audit_service()
try:
return await audit.has_recent_tracing_gap(
agent_id=agent_uuid,
task_id=task_uuid,
since=since,
)
except Exception as exc:
logger.debug(
"audit.has_recent_tracing_gap failed; falling back to strike count",
agent_slug=agent_slug,
task_id=task_id,
error=str(exc),
)
return False
async def _handle_pm_assigned_task(
self, task: dict[str, Any], assigned_to: str
) -> None:
"""Spawn an already-assigned PM agent if it isn't running."""
agent_slug = self._resolve_agent_slug(assigned_to)
if agent_slug not in self._PM_AGENTS or self._is_agent_active(agent_slug):
return
if await self._pm_respawn_should_gate(agent_slug, task):
return
logger.info(
"Spawning assigned PM agent",
task_id=task.get("id"),
agent_id=agent_slug,
)
pm_prompt = (
self._build_main_pm_triage_prompt(task)
if agent_slug == "main-pm"
else self._build_pm_triage_prompt(task)
)
await self.spawn_agent(
agent_id=agent_slug,
task_id=task["id"],
initial_prompt=pm_prompt,
git_context=self._task_git_context(task),
spawned_by="_handle_pm_assigned_task",
)
async def _handle_board_assigned_task(
self, task: dict[str, Any], assigned_to: str
) -> None:
"""Review an assigned board task with the FULL board (PO + HoM), ONCE each.
A board/coordination task — especially one with a UI / user-facing
dimension — must be reviewed by BOTH the Product Owner AND the Head of
Marketing before it is handed to the CEO. The task is assigned to one
board agent, but the review is a two-reviewer gate, so this dispatches
both regardless of which one ``assigned_to`` names.
Board roles advise: they can triage, record notes, and discuss, but have
NO verb to claim, plan, delegate, or complete. A respawn cannot advance
the task — it would just loop — so dispatch is one-shot per (agent, task).
The board reviews and records requirements; the CEO then approves and
hands the task to Main PM for delegation to the cells.
Once BOTH reviewers have finished (each dispatched and no longer active),
the board-review handoff fires: the task is flagged board-reviewed and a
single formal CEO notification is emitted so Approve & Start is an
actionable signal rather than buried chatter.
"""
# `assigned_to` only gates that this IS a board task; the review itself
# always involves the whole board, not just the named assignee.
if self._resolve_agent_slug(assigned_to) not in self._BOARD_AGENTS:
return
task_id = str(task.get("id"))
for board_slug in sorted(self._BOARD_AGENTS):
await self._dispatch_board_reviewer(board_slug, task_id, task)
await self._maybe_handoff_board_review_to_ceo(task_id)
async def _dispatch_board_reviewer(
self, board_slug: str, task_id: str, task: dict[str, Any]
) -> None:
"""One-shot spawn of a single board reviewer for a board task.
Skips when the reviewer is already running or has already been
dispatched for this task (board roles have no progression verb, so a
respawn would loop). Records the (agent, task) pair so the
review-completion detector can tell which reviewers have run.
"""
if self._is_agent_active(board_slug):
return
key = (board_slug, task_id)
if key in self._board_dispatched:
return
# Respawn circuit breaker — parity with every other task-keyed path.
if await self._pm_respawn_should_gate(board_slug, task):
return
self._board_dispatched.add(key)
logger.info(
"Spawning board agent for review",
task_id=task_id,
agent_id=board_slug,
)
await self.spawn_agent(
agent_id=board_slug,
task_id=task["id"],
initial_prompt=self._build_board_prompt(task),
git_context=self._task_git_context(task),
spawned_by="_dispatch_board_reviewer",
)
async def _dispatch_roadmap_exploration(self, task: dict[str, Any]) -> None:
"""One-shot Product-Owner spawn to author a themed roadmap cycle.
Unlike ``_handle_board_assigned_task`` (the two-reviewer board-review
gate), a roadmap cycle is Product-Owner-solo in v1 (see the roadmap
spec's non-goals — HoM co-authoring is out of scope), so this bypasses
the review-pair machinery and its board-review-complete/Approve & Start
handoff entirely: HoM is never spawned for this task, and no CEO
"Approve & Start" notification fires. ``propose_roadmap`` (not this
dispatcher) marks the cycle authored (a ``roadmap_cycle`` marker); this
only ever spawns once per task while that marker is absent, reusing the
same one-shot ``_board_dispatched`` tracker + respawn breaker every
other board dispatch uses.
"""
task_id = str(task.get("id"))
markers_dict = task.get("orchestration_markers") or {}
if markers_dict.get(_markers.ROADMAP_CYCLE) is not None:
return # already authored — the CEO roadmap queue owns the rest
po_slug = "product-owner"
if self._is_agent_active(po_slug):
return
key = (po_slug, task_id)
if key in self._board_dispatched:
return
if await self._pm_respawn_should_gate(po_slug, task):
return
self._board_dispatched.add(key)
logger.info("Spawning Product Owner for roadmap exploration", task_id=task_id)
await self.spawn_agent(
agent_id=po_slug,
task_id=task["id"],
initial_prompt=self._build_roadmap_prompt(task),
git_context=self._task_git_context(task),
spawned_by="_dispatch_roadmap_exploration",
)
async def _dispatch_feature_spotlight_exploration(
self, task: dict[str, Any]
) -> None:
"""One-shot Head-of-Marketing spawn to investigate + author a spotlight.
Simpler than _dispatch_roadmap_exploration: no "already authored" marker
pre-check is needed here, because a successful propose_feature_spotlight()
completes this task atomically (it stops matching the PENDING fetch on the
next tick) — unlike the roadmap cycle, which stays open across the CEO's
per-item decisions and needs the marker check to avoid re-spawning the PO
after authoring. Reuses the same one-shot _board_dispatched tracker +
respawn breaker every other board dispatch uses.
"""
task_id = str(task.get("id"))
hom_slug = "head-marketing"
if self._is_agent_active(hom_slug):
return
key = (hom_slug, task_id)
if key in self._board_dispatched:
return
if await self._pm_respawn_should_gate(hom_slug, task):
return
self._board_dispatched.add(key)
logger.info(
"Spawning Head of Marketing for feature-spotlight exploration",
task_id=task_id,
)
await self.spawn_agent(
agent_id=hom_slug,
task_id=task["id"],
initial_prompt=self._build_feature_spotlight_prompt(task),
git_context=self._task_git_context(task),
spawned_by="_dispatch_feature_spotlight_exploration",
)
def _board_review_complete(self, task_id: str) -> bool:
"""True once EVERY board reviewer has reviewed and gone idle.
A reviewer has finished when it was dispatched for this task
(``_board_dispatched``) and is no longer running (``_is_agent_active``).
Both PO and HoM must satisfy this before the task is handoff-ready.
"""
return all(
(board_slug, task_id) in self._board_dispatched
and not self._is_agent_active(board_slug)
for board_slug in self._BOARD_AGENTS
)
async def _maybe_handoff_board_review_to_ceo(self, task_id: str) -> None:
"""Unlock the CEO's Approve & Start gate when the board review is done.
Two one-shot effects fire once BOTH board reviewers have finished:
1. Persist ``board_review_complete`` on the task. The task stays
pending (its pending state is what hands it to Main PM on approval),
so this flag is the only thing that makes the CEO's Approve & Start
button appear — it never shows on a board task the board hasn't
finished reviewing.
2. Emit an ack-required APPROVAL notification to the CEO. Board agents
only record journal notes during review, which
left the CEO with no actionable signal; this is that signal.
Fires at most once per task; a failure clears the guard so a later tick
retries, and never blocks the dispatch loop.
"""
if task_id in self._board_review_ceo_notified:
return
if not self._board_review_complete(task_id):
return
self._board_review_ceo_notified.add(task_id)
from uuid import UUID
from roboco.db.base import get_db_context
from roboco.services.notification import NotificationService
from roboco.services.task import TaskService
try:
async with get_db_context() as db:
await TaskService(db).mark_board_review_complete(UUID(task_id))
await db.commit()
await NotificationService().send_board_review_complete_notification(
task_id=task_id,
)
except Exception as exc:
# Don't wedge dispatch on a failure; allow a retry by clearing the
# one-shot guard so a later tick can re-run the handoff.
self._board_review_ceo_notified.discard(task_id)
logger.warning(
"Failed to hand board-review completion to CEO",
task_id=task_id,
error=str(exc),
)
return
logger.info(
"Board review complete — CEO Approve & Start unlocked",
task_id=task_id,
)
# Keep-alive re-draft: if an intake chat is parked awaiting this review,
# inject the board's feedback so the still-resident prompter re-drafts
# in-context. Best-effort; the cold "Re-draft" path covers the rest.
await self._inject_board_brief_into_parked_intake(task_id)
async def _inject_board_brief_into_parked_intake(self, task_id: str) -> None:
"""Inject the board's review into a parked intake session, if one exists.
No-op when no session is parked for this task (it was reaped, the
container died, or the draft never used the board route) — the CEO then
re-drafts via the cold ``/re-interview`` path instead. Never raises.
"""
from roboco.services.prompter_live import get_live_registry
session = get_live_registry().find_by_task(task_id)
if session is None:
return
from uuid import UUID
from roboco.db.base import get_db_context
from roboco.services.journal import get_journal_service
from roboco.services.prompter import compose_redraft_message
from roboco.services.task import get_task_service
try:
async with get_db_context() as db:
task = await get_task_service(db).get(UUID(task_id))
if task is None:
return
entries = await get_journal_service(db).board_review_brief(
UUID(task_id)
)
message = compose_redraft_message(task, entries)
delivered = await get_live_registry().deliver(session.session_id, message)
logger.info(
"Injected board feedback into parked intake",
task_id=task_id,
delivered=delivered,
)
except Exception as exc:
logger.warning(
"Failed to inject board feedback into parked intake",
task_id=task_id,
error=str(exc),
)
def _pm_spawn_prompt(
self, routing: str, agent_id: str, task: dict[str, Any]
) -> str:
"""Pick the correct prompt for a classified spawn."""
if routing == "dev":
return self._build_dev_prompt(task)
if routing == "main_pm" or agent_id == "main-pm":
return self._build_main_pm_triage_prompt(task)
return self._build_pm_triage_prompt(task)
async def _route_unassigned_pm_task(
self, client: httpx.AsyncClient, task: dict[str, Any]
) -> None:
"""Classify and route an unassigned pending task to its target agent."""
routing = self._classify_task_routing(task)
agent_id = self._get_routing_target(routing, task)
if not agent_id:
logger.warning(
"No routing target found",
task_id=task.get("id"),
routing=routing,
)
return
# Board work is a two-reviewer gate (PO + Head of Marketing), not a
# single-assignee claim. Routing only ever names one board agent
# (product-owner), so claiming + spawning that one here would leave the
# Head of Marketing out (finding #4). Delegate to the board handler,
# which dispatches BOTH reviewers one-shot and leaves the task pending
# for the CEO's Approve & Start. ``agent_id`` is the routed board slug.
if routing == "board":
await self._handle_board_assigned_task(task, agent_id)
return
# Don't auto-claim back to the creator. A PM that just created this
# task is about to assign it (e.g. be-pm creating a code subtask to
# hand to be-dev-1 one tool-call later). Racing in and claiming for
# the PM hijacks the delegation — the PM ends up owning a code task
# it never intended to work on itself. Skip this tick and let the
# next dispatch pick it up once assigned_to is set, OR re-evaluate
# when we have a clearer signal the creator won't route it.
created_by = task.get("created_by")
if created_by:
creator_slug = self._resolve_agent_slug(str(created_by))
if creator_slug == agent_id:
logger.info(
"Skipping auto-claim: routing target is the creator",
task_id=task.get("id"),
creator=creator_slug,
routing=routing,
)
return
logger.info(
"Routing task",
task_id=task.get("id"),
routing=routing,
agent_id=agent_id,
)
if self._is_agent_active(agent_id):
await self._claim_task_for_agent(client, task["id"], agent_id)
return
if await self._claim_task_for_agent(client, task["id"], agent_id):
prompt = self._pm_spawn_prompt(routing, agent_id, task)
await self.spawn_agent(
agent_id=agent_id,
task_id=task["id"],
initial_prompt=prompt,
git_context=self._task_git_context(task),
spawned_by="_route_unassigned_pm_task",
)
async def _dispatch_pm_work(self, client: httpx.AsyncClient) -> None:
"""
Dispatch PM triage work - routes new tasks to appropriate level.
This is the FIRST dispatcher called - it classifies unassigned tasks
and routes them to Board, Main PM, Cell PM, or directly to devs.
Also handles already-assigned pending tasks for PM agents.
Monitors: pending tasks (both assigned and unassigned)
Spawns: product-owner, main-pm, be-pm, fe-pm, ux-pm (or devs for simple)
"""
tasks = await self._fetch_tasks(client, "pending")
for task in tasks:
if self._is_task_handled_this_tick(task.get("id")):
continue
# CEO-HELD / externally-owned sources are never PM delivery work
# (external-PR review, release proposals, X posts/replies, and a
# not-yet-confirmed self-heal fix task) — see _is_held_ceo_source.
if _is_held_ceo_source(task):
continue
assigned_to = task.get("assigned_to")
if assigned_to:
if task.get("source") == ROADMAP_SOURCE:
# PO-solo (v1) — bypasses the two-reviewer board-review gate;
# never rides _handle_board_assigned_task (that would also
# spawn Head of Marketing and fire the Approve & Start
# handoff, both wrong for a roadmap cycle).
await self._dispatch_roadmap_exploration(task)
elif task.get("source") == X_FEATURE_EXPLORATION_SOURCE:
# HoM-solo (mirrors the ROADMAP_SOURCE branch above) —
# bypasses the two-reviewer board-review gate; never rides
# _handle_board_assigned_task (that would also spawn the
# Product Owner and fire the Approve & Start handoff, both
# wrong for a feature-spotlight cycle).
await self._dispatch_feature_spotlight_exploration(task)
elif self._resolve_agent_slug(assigned_to) in self._BOARD_AGENTS:
await self._handle_board_assigned_task(task, assigned_to)
else:
await self._handle_pm_assigned_task(task, assigned_to)
continue
await self._route_unassigned_pm_task(client, task)
async def _dispatch_revision_coordination_roots(
self, client: httpx.AsyncClient
) -> None:
"""Re-spawn the owning PM for a PM-owned needs_revision task.
Two cases land a task in ``needs_revision`` owned by a PM rather than a
developer: a CEO-rejected coordination root (team=main_pm, product-linked,
no repo), and a gate-failed assembled task (a cell→root or root→master PR
the in-path reviewer sent back via pr_fail). The dev dispatcher only
spawns developers and the closure path only handles paused parents, so
without this such a task would sit in needs_revision forever — the
deadlock. The PM-ownership filter below scopes this to exactly those: a
leaf dev revision stays owned by its developer and is left to the dev
dispatcher.
"""
tasks = await self._fetch_tasks(client, "needs_revision")
for task in tasks:
if self._is_task_handled_this_tick(task.get("id")):
continue
owner = task.get("assigned_to") or task.get("claimed_by")
agent_slug = self._resolve_agent_slug(owner) if owner else None
if not agent_slug or self._is_agent_active(agent_slug):
continue
if get_agent_role(agent_slug) not in ("cell_pm", "main_pm"):
continue
# Respawn circuit breaker — a revision the PM can never land must
# stop respawning the coordinator (progress resets the strikes).
if await self._pm_respawn_should_gate(agent_slug, task):
continue
await self.spawn_agent(
agent_id=agent_slug,
task_id=task["id"],
initial_prompt=self._get_prompt_for_agent(agent_slug, task),
git_context=self._task_git_context(task),
spawned_by="_dispatch_revision_coordination_roots",
)
@staticmethod
def _all_descendants_terminal(descendants: list[dict[str, Any]]) -> bool:
"""Every descendant in a closure-complete state?"""
return all(st.get("status") in ("completed", "cancelled") for st in descendants)
@staticmethod
def _already_promoted_for_closure(task: dict[str, Any]) -> bool:
"""Skip closure respawn when PR+status show task has moved up."""
return bool(
task.get("pr_number")
and task.get("status")
in ("awaiting_pm_review", "awaiting_ceo_approval", "completed")
)
@staticmethod
def _coerce_heartbeat(value: Any) -> datetime | None:
"""Normalize ``last_heartbeat_at`` to an aware UTC datetime.
The dispatcher reads tasks via the HTTP API, which serializes
datetimes as ISO-8601 strings; direct service callers (and tests)
may pass ``datetime`` objects. Anything else is treated as
absent so a malformed value can't accidentally arm the gate.
"""
if value is None:
return None
if isinstance(value, datetime):
return value if value.tzinfo else value.replace(tzinfo=UTC)
if isinstance(value, str):
try:
parsed = datetime.fromisoformat(value)
except ValueError:
return None
return parsed if parsed.tzinfo else parsed.replace(tzinfo=UTC)
return None
def _is_recently_paused(self, task: dict[str, Any]) -> bool:
"""A paused task whose heartbeat is fresher than the closure debounce.
Closes the ``i_am_idle`` vs closure-respawn race:
``i_am_idle`` auto-pauses in-flight tasks and then sets the agent
IDLE. If the dispatcher ticks between those two writes it sees a
paused parent and would spawn the closure PM against a session
that is mid-shutdown. A fresh ``last_heartbeat_at`` (newer than
``settings.pm_closure_recently_paused_seconds``) is the signal that
the agent was alive moments ago and a respawn now would race the
existing session. Genuinely-stale paused tasks (or tasks with no
heartbeat recorded) fall through and follow the regular closure path.
This debounce is deliberately SHORT (a few dispatch ticks). It is
NOT the reaper window (``_claim_heartbeat_ttl`` /
``stale_claim_reap_seconds``, 600s default and 1800s on the NAS):
binding it there delayed every cell/main closure by up to 10-30
minutes, because a paused parent's heartbeat reflects when the PM
last *worked*, so a PM that worked right up to idle leaves a fresh
heartbeat. The live-session case is already covered separately by
the ``_is_agent_active`` check in ``_maybe_spawn_pm_closure``.
"""
if task.get("status") != "paused":
return False
last_hb = self._coerce_heartbeat(task.get("last_heartbeat_at"))
if last_hb is None:
return False
cutoff = datetime.now(UTC) - timedelta(
seconds=self._closure_recently_paused_ttl
)
return last_hb > cutoff
def _closure_pm_for_team(self, team: str | None) -> str:
"""Pick the PM that owns closure for a given team."""
if team in ("backend", "frontend", "ux_ui"):
return self._TEAM_PM_MAP.get(team, "be-pm")
return "main-pm"
# Which flow route + verb submits an assembled parent, per PM role.
_AUTO_SUBMIT_VERB_BY_ROLE: ClassVar[dict[str, tuple[str, str]]] = {
"cell_pm": ("cell_pm", "submit_up"),
"main_pm": ("main_pm", "submit_root"),
}
def _auto_submit_target(
self, task: dict[str, Any], pm_slug: str
) -> tuple[str, str, str, str] | None:
"""(role, route, verb, pm_uuid) when this parent is auto-submittable.
None when the flag is off, the parent is branchless coordination (a
MegaTask umbrella assembles no PR), the role has no submit verb, or
no PM identity can be resolved.
"""
role = get_agent_role(pm_slug) or ""
pair = self._AUTO_SUBMIT_VERB_BY_ROLE.get(role)
pm_uuid = str(task.get("assigned_to") or AGENT_UUIDS.get(pm_slug) or "")
if (
not settings.pr_gate_auto_submit_enabled
or not task.get("branch_name")
or not task.get("project_id")
or pair is None
or not pm_uuid
):
return None
return (role, pair[0], pair[1], pm_uuid)
async def _try_auto_submit(
self, client: httpx.AsyncClient, task: dict[str, Any], pm_slug: str
) -> bool:
"""Submit an assembled, all-children-terminal parent to the PR gate
WITHOUT spawning its PM — the turn's substance (freshness rebase,
integrity check, PR open) is deterministic gate code, so the real
submit verb is run through the internal API as the owning PM.
Returns True when the gate accepted (the reviewer dispatch takes it
from awaiting_pr_review); False on ANY refusal — flag off, a
branchless coordination parent (a MegaTask umbrella assembles no
PR), an unmapped role, a gate rejection (freshness/integrity — the
PM turn is then genuinely needed), or a transport error — and the
caller falls back to the classic PM closure spawn.
"""
target = self._auto_submit_target(task, pm_slug)
if target is None:
return False
role, role_path, verb, pm_uuid = target
task_id = str(task.get("id"))
notes = (
"Auto-submitted for gate review: every child task is terminal and "
"the assembled branch is ready. Freshness and integrity are "
"enforced by the submit gate itself; the in-path PR reviewer "
"takes it from here."
)
try:
resp = await client.post(
f"{self._api_url}/v1/flow/{role_path}/{verb}",
headers=_agent_api_headers(pm_uuid, role),
json={"task_id": task_id, "notes": notes},
)
body = resp.json()
except Exception as e:
logger.warning(
"Auto-submit transport failure; falling back to PM closure spawn",
task_id=task_id,
error=str(e),
)
return False
if not isinstance(body, dict) or body.get("error"):
logger.info(
"Auto-submit rejected by the gate; PM closure spawn proceeds",
task_id=task_id,
error=(body or {}).get("error") if isinstance(body, dict) else body,
message=(body or {}).get("message") if isinstance(body, dict) else None,
)
return False
logger.info(
"Assembled parent auto-submitted to the PR gate (PM turn skipped)",
task_id=task_id,
verb=verb,
pm=pm_slug,
)
self._fire_audit(
event_type="task.auto_submitted",
agent_slug=pm_slug,
task_id=task_id,
details={"verb": verb, "auto": True},
)
self._mark_task_handled(task_id)
return True
async def _closure_handled_without_pm(
self,
client: httpx.AsyncClient,
task: dict[str, Any],
task_id: str,
pm_id: str,
) -> bool:
"""Recover the parent's status, then try the submit turn cut.
The parent auto-paused when its PM idled (by design) — resume it
before anything else so whoever acts next (the auto-submit or the
spawned PM) lands on an actionable in_progress parent; an errant
`blocked` at closure is recovered symmetrically. Then the turn cut:
an assembled parent whose children are all terminal is submitted to
the PR gate system-side (True => the PM spawn is skipped); parents
past the gate (awaiting_pm_review — the merge turn) always spawn.
"""
parent_status = task.get("status")
if parent_status == "paused":
await self._auto_resume_paused_parent(client, task_id)
elif parent_status == "blocked":
await self._auto_recover_blocked_parent(client, task_id)
return parent_status in (
"claimed",
"in_progress",
"paused",
) and await self._try_auto_submit(client, task, pm_id)
async def _maybe_spawn_pm_closure(
self, client: httpx.AsyncClient, task: dict[str, Any]
) -> None:
"""If this parent task is ready for closure, spawn its PM."""
task_id = task.get("id")
if not task_id:
return
if self._is_recently_paused(task):
logger.debug(
"Skipping closure spawn for recently-paused parent",
task_id=task_id,
last_heartbeat_at=task.get("last_heartbeat_at"),
)
return
descendants = await self._fetch_all_descendants(client, task_id)
if not descendants or not self._all_descendants_terminal(descendants):
return
if self._already_promoted_for_closure(task):
return
pm_id = self._closure_pm_for_team(task.get("team"))
if self._is_agent_active(pm_id):
return
logger.info(
"Parent task ready for closure",
task_id=task_id,
descendants_count=len(descendants),
pm_id=pm_id,
)
# The parent auto-paused when its PM idled (by design). Resume
# it before respawn so the PM lands actionable (in_progress) and can
# directly submit_up / complete / escalate — pre-gateway behaviour the
# gateway refactor dropped, which wedged a dogfood run (the model
# never issued resume() itself).
# A parent that is `blocked` at closure (all descendants
# terminal) is an errant/stale block — recover it symmetrically so
# the chain can't wedge forever waiting for a PM to manually unblock.
if await self._closure_handled_without_pm(client, task, task_id, pm_id):
return
prompt = self._build_pm_closure_prompt(task, descendants)
await self.spawn_agent(
agent_id=pm_id,
task_id=task_id,
initial_prompt=prompt,
git_context=self._task_git_context(task),
spawned_by="_maybe_spawn_pm_closure",
)
async def _dispatch_pm_closure_work(self, client: httpx.AsyncClient) -> None:
"""
Dispatch PM closure work - check parent tasks ready to close.
When all subtasks of a parent task are completed, spawn the PM
to review and close the parent task.
Monitors: tasks with completed subtasks but parent still open
Spawns: be-pm, fe-pm, ux-pm, main-pm (based on parent team)
"""
# Find parent tasks that might have children ready for closure
# Include "paused" - PM pauses while waiting, respawned when subtasks done
# Include "awaiting_pm_review" - parent awaiting review when children done
parent_statuses = ["claimed", "in_progress", "paused", "awaiting_pm_review"]
for status in parent_statuses:
tasks = await self._fetch_tasks(client, status)
for task in tasks:
await self._maybe_spawn_pm_closure(client, task)
async def _fetch_subtasks(
self, client: httpx.AsyncClient, parent_id: str
) -> list[dict[str, Any]]:
"""Fetch direct subtasks for a parent task."""
try:
resp = await client.get(
f"{self._api_url}/tasks",
params={"parent_task_id": parent_id},
)
if resp.status_code == http_status.HTTP_200_OK:
data = resp.json()
tasks = data.get("tasks", data) if isinstance(data, dict) else data
return list(tasks) if tasks else []
except Exception as e:
logger.warning(
"Failed to fetch subtasks", parent_id=parent_id, error=str(e)
)
return []
async def _fetch_all_descendants(
self, client: httpx.AsyncClient, task_id: str
) -> list[dict[str, Any]]:
"""Fetch ALL descendants (children, grandchildren, etc.) recursively.
Uses the /tasks/{id}/descendants endpoint which does BFS traversal.
"""
try:
resp = await client.get(f"{self._api_url}/tasks/{task_id}/descendants")
if resp.status_code == http_status.HTTP_200_OK:
data = resp.json()
# Endpoint returns list directly
return list(data) if data else []
except Exception as e:
logger.warning("Failed to fetch descendants", task_id=task_id, error=str(e))
return []
def _build_pm_closure_prompt(
self, task: dict[str, Any], subtasks: list[dict[str, Any]]
) -> str:
"""Prompt for PM closing their own parent task (subtasks terminal)."""
task_id = task.get("id", "unknown")
title = task.get("title", "Untitled")
team = task.get("team", "unknown")
subtask_summary = "\n".join(
f" - {st.get('title', 'Untitled')} ({st.get('status', 'unknown')})"
for st in subtasks
)
is_root = not task.get("parent_task_id")
project_slug = task.get("project_slug", "")
if is_root:
target_line = (
"submit_up promotes to awaiting_ceo_approval; the CEO reviews "
"and merges to master. You do NOT merge to master yourself."
)
submit_step = (
f'4. submit_up(task_id="{task_id}",\n'
' notes="<aggregate summary: what shipped across the '
'cells, evidence, risk callouts>")\n'
" — promotes to awaiting_ceo_approval. "
"CEO is the final approver."
)
else:
target_line = (
"submit_up opens your cell-level PR into the parent task's "
"branch and transitions you to awaiting_pm_review for the "
"parent PM."
)
submit_step = (
f'4. submit_up(task_id="{task_id}",\n'
' notes="<cell summary: what your cell shipped, '
'evidence>")\n'
" — opens cell-level PR up to the parent's branch and "
"transitions to awaiting_pm_review."
)
return f"""You are closing YOUR OWN parent task. All subtasks are
terminal — promote the merged work one level up the hierarchy.
TASK: {task_id}
TITLE: {title}
TEAM: {team}
PROJECT: {project_slug}
ROOT TASK: {"yes" if is_root else "no"}
SUBTASK SUMMARY:
{subtask_summary}
PROMOTION TARGET: {target_line}
== PM CLOSURE WORKFLOW ==
1. evidence(task_id="{task_id}")
— review aggregate state, every acceptance criterion, and each
subtask's terminal status. Returns the inline diff for your branch
(all merged subtask work).
2. If any subtask is still in awaiting_pm_review, review + close it FIRST:
- APPROVE leaf: complete(task_id="<subtask_id>",
notes="<merge rationale>")
(auto-merges the leaf PR into your cell branch).
- NEEDS REWORK: leave a clear note(scope='decision',
task_id="<subtask_id>", text="...") and rely on the dispatcher to
respawn the dev for revision.
3. note(scope='decision', task_id="{task_id}",
text="Closure: {title} — <rationale, AC coverage, risks>")
— REQUIRED before submit_up().
{submit_step}
5. i_am_idle()
Never `commit`, never write code, never run `git`. PMs coordinate.
"""
def _get_prompt_for_agent(self, agent_slug: str, task: dict[str, Any]) -> str:
"""Get the prompt appropriate to the agent's ACTUAL role.
A respawn must hand each role the prompt it can act on — a PM or board
agent handed the developer prompt is told to write code and call verbs
it does not own. Reuses the same per-role prompt builders the role
dispatchers use so a respawn matches a fresh dispatch:
developer → dev prompt
qa → QA prompt
documenter → doc prompt
cell_pm → cell-PM triage prompt
main_pm → main-PM triage prompt
product_owner → board-review prompt
head_marketing → marketing prompt for a marketing task, else board
auditor → audit prompt
Unknown roles fall back to the dev prompt (safe default for an
executable task).
"""
role = get_agent_role(agent_slug)
# head_marketing is the one role whose prompt depends on the task, so it
# is resolved before the static role→builder table.
if role == "head_marketing":
if task.get("team") == "marketing":
return self._build_marketing_prompt(task)
return self._build_board_prompt(task)
builders: dict[str, Callable[[dict[str, Any]], str]] = {
"developer": self._build_dev_prompt,
"qa": self._build_qa_prompt,
"documenter": self._build_doc_prompt,
"cell_pm": self._build_pm_triage_prompt,
"main_pm": self._build_main_pm_triage_prompt,
"product_owner": self._build_board_prompt,
"auditor": lambda _task: self._build_audit_prompt(),
"pr_reviewer": self._build_pr_review_prompt,
}
builder = builders.get(role, self._build_dev_prompt)
return builder(task)
async def _dispatch_dev_work(self, client: httpx.AsyncClient) -> None:
"""
Dispatch assigned work to the assigned agent.
NOTE: This handles PRE-ASSIGNED tasks (assigned by PM),
needs_revision tasks, and in_progress tasks where agent is not active
(e.g., after unblock). New unassigned pending tasks are handled by
_dispatch_pm_work() which routes them through the PM hierarchy.
Monitors: assigned pending tasks, needs_revision tasks, orphaned in_progress
Spawns: Any assigned agent (dev, doc, qa) with appropriate prompt
"""
# Get tasks needing attention. Includes:
# - `claimed` — PM-delegated claims where the assignee was never spawned
# - `blocked` — but only when another agent can resolve (see below)
# `pending`, `needs_revision`, `in_progress` are the classic cases.
tasks = await self._fetch_tasks(
client,
["pending", "claimed", "needs_revision", "in_progress", "blocked"],
)
for task in tasks:
if self._is_task_handled_this_tick(task.get("id")):
continue
# Held CEO artifacts + Board exploration cycles belong to other
# dispatchers (their own routes / _dispatch_pm_work), never a dev.
if _is_non_dev_dispatch_source(task):
continue
await self._dev_dispatch_one(client, task)
@staticmethod
def _resolve_dev_owner_uuid(task: dict[str, Any]) -> str | None:
"""Pick the right owner UUID for dev dispatch based on status.
Always falls back to ``claimed_by`` when ``assigned_to`` is missing, so
a task left half-reaped (assigned_to nulled but still claimed) still
dispatches to its rightful owner instead of going dormant — the
orchestrator knows who to call even when one ownership field was cleared.
"""
status = task.get("status")
if status in ("claimed", "blocked"):
return task.get("claimed_by") or task.get("assigned_to")
return task.get("assigned_to") or task.get("claimed_by")
async def _respawn_dev_if_inactive(
self, task: dict[str, Any], agent_slug: str
) -> None:
"""Respawn a dev agent on an existing task when it isn't running."""
if self._is_agent_active(agent_slug):
return
await self.spawn_agent(
agent_id=agent_slug,
task_id=task["id"],
initial_prompt=self._build_dev_prompt(task),
git_context=self._task_git_context(task),
spawned_by="_respawn_dev_if_inactive",
)
async def _spawn_pending_dev(
self,
client: httpx.AsyncClient,
task: dict[str, Any],
agent_slug: str,
) -> None:
"""Validate and spawn a dev agent for a pending, pre-assigned task."""
if self._is_agent_active(agent_slug):
return
# Per-dev queue order: hold a dev's higher-sequence code leaf while it
# still has an earlier non-terminal code sibling under the same parent,
# so the dev works its queue one task at a time, in order. Loop-free —
# just not dispatched this tick.
if await self._blocked_by_earlier_lane_sibling(task):
return
# Respawn circuit breaker — a dev leaf that respawns without the task
# advancing (wedged workspace, unclaimable state) stops after strikes.
if await self._pm_respawn_should_gate(agent_slug, task):
return
validation_issue = await self._validate_task_for_spawn(client, task, agent_slug)
if validation_issue:
logger.warning(
"Skipping spawn due to validation failure",
task_id=task["id"],
agent=agent_slug,
reason=validation_issue,
)
return
await self.spawn_agent(
agent_id=agent_slug,
task_id=task["id"],
initial_prompt=self._get_prompt_for_agent(agent_slug, task),
git_context=self._task_git_context(task),
spawned_by="_spawn_pending_dev",
)
@staticmethod
def _is_hitl_blocked(task: dict[str, Any]) -> bool:
"""HITL-blocked tasks wait for human resolution; skip respawn."""
return (
task.get("status") == "blocked"
and task.get("blocker_resolver_type") == "human"
)
async def _handle_dev_existing_owner(
self, task: dict[str, Any], status: str, agent_slug: str
) -> None:
"""Respawn existing dev for needs_revision / in_progress / claimed."""
# A `blocked` task is waiting for its blocker to clear (PM / dependency);
# the owner has no legal move from `blocked`, so respawning it does
# nothing but churn. It is revived only when unblocked back to
# in_progress, or released to the pool (unclaim) for re-delegation.
if status == "blocked":
return
if status in (
"in_progress",
"claimed",
) and not self._is_agent_active(agent_slug):
logger.info(
"Respawning agent for orphaned task",
task_id=task["id"],
agent=agent_slug,
status=status,
)
await self._respawn_dev_if_inactive(task, agent_slug)
async def _dev_dispatch_one(
self, client: httpx.AsyncClient, task: dict[str, Any]
) -> None:
"""Dispatch a single task from `_dispatch_dev_work`'s fetch set."""
team = task.get("team")
if team not in ["backend", "frontend", "ux_ui"]:
return
if self._is_hitl_blocked(task):
logger.debug(
"Skipping HITL-blocked task; waiting for human",
task_id=task["id"],
)
return
status = task.get("status")
owner_uuid = self._resolve_dev_owner_uuid(task)
agent_slug = self._resolve_agent_slug(owner_uuid) if owner_uuid else None
# Role/task_type mismatch guard. The dispatcher
# previously trusted whatever ``assigned_to`` named, so a
# documentation task accidentally assigned to a developer agent
# would silently spawn the dev. Reject the dispatch if the
# assignee's role doesn't match the task type — the PM that
# mis-assigned needs to fix it before any agent runs.
# Tasks owned by PM/board/QA roles aren't this dispatcher's lane;
# `_dispatch_pm_work` and the QA-pool path own them. Silently skip
# so the warning only fires on actual dev/doc misassignments.
if agent_slug:
assignee_role = get_agent_role(agent_slug)
if assignee_role not in ("developer", "documenter", "unknown"):
return
if not self._dev_dispatch_role_matches(task, agent_slug):
logger.warning(
"dev dispatch: role/task_type mismatch — skipping spawn",
task_id=task.get("id"),
task_type=task.get("task_type"),
assignee_slug=agent_slug,
assignee_role=assignee_role,
)
return
if agent_slug and status in (
"needs_revision",
"in_progress",
"claimed",
"blocked",
):
await self._handle_dev_existing_owner(task, status, agent_slug)
return
# Pending tasks pre-assigned by PM.
if agent_slug:
await self._spawn_pending_dev(client, task, agent_slug)
@staticmethod
def _dev_dispatch_role_matches(task: dict[str, Any], agent_slug: str) -> bool:
"""Return True if the assignee role matches the task's task_type.
Dev dispatcher only spawns developer-role agents. A doc/qa task
assigned to a dev (or vice versa) should be flagged, not silently
spawned. Returns True when the type is unknown or the assignee role
is unknown — the validation runs as a guard, not a strict gate, so
an unknown classification doesn't block work that would otherwise
proceed.
"""
role = get_agent_role(agent_slug)
if role == "unknown":
return True
task_type = task.get("task_type")
if task_type == "documentation":
return role == "documenter"
# `code` / `research` / `planning` / `administrative` / `design` all
# route through dev or PM; only the doc-task case is unambiguous.
return role == "developer"
async def _spawn_assigned_qa(self, task: dict[str, Any], assigned_to: str) -> bool:
"""If task.assigned_to is a QA slug, spawn/skip-if-running; else False.
Returns True when the dispatch decision for this task was
handled at the assignee level (spawned or already running).
Returns False when the assigned_to is NOT a QA agent — caller
then falls through to the unassigned-select path.
"""
assigned_slug = self._resolve_agent_slug(assigned_to)
if not assigned_slug or "qa" not in assigned_slug:
logger.warning(
"awaiting_qa task assigned to non-QA slug; reassigning via QA pool",
task_id=task["id"],
assigned_slug=assigned_slug,
)
return False
if self._is_agent_active(assigned_slug):
return True
# Respawn circuit breaker — same progress-aware gate as every other
# task-keyed spawn path; notifies the CEO once it trips.
if await self._pm_respawn_should_gate(assigned_slug, task):
return True
await self.spawn_agent(
agent_id=assigned_slug,
task_id=task["id"],
initial_prompt=self._build_qa_prompt(task),
git_context=self._task_git_context(task),
spawned_by="_spawn_assigned_qa",
)
return True
async def _dispatch_qa_work(self, client: httpx.AsyncClient) -> None:
"""
Dispatch QA work to QA agents.
Monitors: awaiting_qa tasks
Spawns: be-qa, fe-qa, ux-qa
"""
tasks = await self._fetch_tasks(client, "awaiting_qa")
for task in tasks:
if self._is_task_handled_this_tick(task.get("id")):
continue
team = task.get("team")
if team not in ["backend", "frontend", "ux_ui"]:
continue
assigned_to = task.get("assigned_to")
if assigned_to and await self._spawn_assigned_qa(task, assigned_to):
continue
# Unassigned task - select QA agent for this team
agent_id = self._select_agent_for_cell(team, "qa")
if not agent_id:
continue
if self._is_agent_active(agent_id):
# QA already running, they'll pick up on scan
continue
# Respawn circuit breaker — same progress-aware gate as every
# other task-keyed spawn path.
if await self._pm_respawn_should_gate(agent_id, task):
continue
# NO pre-claim (matches _spawn_assigned_qa and the external-PR
# reviewer dispatch): the transitioning claim moved the task to
# 'claimed' before the agent existed, stranding the QA whose own
# claim_review/pass_review demand awaiting_qa (live 2026-07-02,
# ba7b751c). The agent claims itself via claim_review; the
# _is_agent_active guard prevents a double-spawn across ticks.
await self.spawn_agent(
agent_id=agent_id,
task_id=task["id"],
initial_prompt=self._build_qa_prompt(task),
git_context=self._task_git_context(task),
spawned_by="_dispatch_qa_work",
)
# Only spawn one QA at a time per cell
break
async def _dispatch_pr_review_work(self, client: httpx.AsyncClient) -> None:
"""Dispatch inbound external-PR review tasks to the PR reviewer.
Monitors: pending tasks with ``source='external_pr'``.
Spawns: the single global reviewer ``pr-reviewer-1`` (one review at a
time). No pre-claim — the task stays PENDING until the reviewer claims
it itself via ``claim_pr_review``; the prompt carries the task id. The
``is_agent_active`` guard prevents a double-spawn across ticks.
"""
reviewer = "pr-reviewer-1"
if self._is_agent_active(reviewer):
return
tasks = await self._fetch_tasks(client, "pending")
for task in tasks:
if task.get("source") not in PR_REVIEW_SOURCES:
continue
if self._is_task_handled_this_tick(task.get("id")):
continue
if task.get("assigned_to"):
continue
# Respawn circuit breaker — parity with the in-path gate dispatcher.
if await self._pm_respawn_should_gate(reviewer, task):
continue
await self.spawn_agent(
agent_id=reviewer,
task_id=task["id"],
initial_prompt=self._build_pr_review_prompt(task),
git_context=self._task_git_context(task),
spawned_by="_dispatch_pr_review_work",
)
break
async def _dispatch_pr_gate_work(self, client: httpx.AsyncClient) -> None:
"""Dispatch in-path PR-review-gate tasks (awaiting_pr_review) to reviewers.
Routes by level: a cell→root task (team backend/frontend/ux_ui) goes to
that cell's reviewer (be/fe/ux-pr-reviewer); the root→master task goes to
the main reviewer (pr-reviewer-1). The reviewer claims the task itself via
``claim_gate_review`` (no pre-claim — mirrors the external-PR dispatcher);
the ``is_agent_active`` guard + one-reviewer-per-cell prevent a
double-spawn, and ``spawned`` bounds each reviewer to one task per tick.
"""
tasks = await self._fetch_tasks(client, "awaiting_pr_review")
spawned: set[str] = set()
for task in tasks:
if self._is_task_handled_this_tick(task.get("id")):
continue
team = task.get("team")
if team in ("backend", "frontend", "ux_ui"):
reviewer = self._select_agent_for_cell(team, "pr_reviewer")
else:
reviewer = "pr-reviewer-1"
if not reviewer or reviewer in spawned or self._is_agent_active(reviewer):
continue
# Respawn circuit breaker — a gate task that keeps re-surfacing
# without advancing must stop respawning the reviewer.
if await self._pm_respawn_should_gate(reviewer, task):
continue
spawned.add(reviewer)
await self.spawn_agent(
agent_id=reviewer,
task_id=task["id"],
initial_prompt=self._build_pr_gate_prompt(task),
git_context=self._task_git_context(task),
spawned_by="_dispatch_pr_gate_work",
)
async def _dispatch_doc_work(self, client: httpx.AsyncClient) -> None:
"""
Dispatch documentation + developer work during the parallel
awaiting_documentation phase.
`awaiting_documentation` requires BOTH docs_complete=True AND
pr_created=True to advance to awaiting_pm_review. Doc writes the
docs; original developer pushes and creates the PR. Whoever
finishes last triggers the state transition. Previously this
dispatcher only spawned the documenter — if the documenter
finished first, the task would sit indefinitely with pr_created=
False and nothing would spawn the dev to finish the other half.
Monitors: awaiting_documentation tasks
Spawns:
- documenter (be-doc, fe-doc, ux-doc) if docs_complete=False
- original_developer if pr_created=False (tracked in
quick_context as "original_developer:<uuid>")
"""
# Fetch both `awaiting_documentation` and `claimed` because the
# doc's claim transitions status from awaiting_documentation →
# claimed. Without including `claimed` we'd miss tasks where doc
# already grabbed it but pr_created is still false (dev hasn't
# pushed/created PR yet). The `original_developer:` marker in
# quick_context identifies tasks that are actually in the parallel
# phase vs unrelated claimed tasks.
tasks = await self._fetch_tasks(client, ["awaiting_documentation", "claimed"])
for task in tasks:
if self._is_task_handled_this_tick(task.get("id")):
continue
await self._doc_dispatch_one(client, task)
async def _auto_assign_doc(
self, client: httpx.AsyncClient, task: dict[str, Any], team: str
) -> None:
"""
Auto-select and spawn a documenter for an unassigned awaiting_documentation task
"""
agent_id = self._select_agent_for_cell(team, "doc")
if not agent_id or self._is_agent_active(agent_id):
return
# Respawn circuit breaker — before claiming, so a wedged doc task
# doesn't churn claims while the gate is open.
if await self._pm_respawn_should_gate(agent_id, task):
return
if not await self._claim_task_for_agent(client, task["id"], agent_id):
logger.warning(
"Failed to claim awaiting_documentation task for doc",
task_id=task["id"],
agent_id=agent_id,
)
return
await self.spawn_agent(
agent_id=agent_id,
task_id=task["id"],
initial_prompt=self._build_doc_prompt(task),
git_context=self._task_git_context(task),
spawned_by="_auto_assign_doc",
)
async def _doc_dispatch_one(
self,
client: httpx.AsyncClient,
task: dict[str, Any],
) -> None:
"""Process a single task for `_dispatch_doc_work`."""
team = task.get("team")
if team not in ["backend", "frontend", "ux_ui"]:
return
dev_uuid = (task.get("orchestration_markers") or {}).get("original_developer")
status = task.get("status")
# Only consider `claimed` tasks actually in the doc/PR parallel
# phase. See `_is_parallel_phase_claim` docstring for the why.
if status == "claimed" and not self._is_parallel_phase_claim(task, dev_uuid):
return
# Developer half: push + create PR
await self._respawn_dev_for_pr_half(task, dev_uuid)
# Documenter half: write docs
if task.get("docs_complete"):
return
if await self._respawn_doc_if_assigned(task):
return
# Auto-assign a documenter only when still in awaiting_documentation.
if status != "awaiting_documentation":
return
await self._auto_assign_doc(client, task, team)
async def _respawn_doc_if_assigned(self, task: dict[str, Any]) -> bool:
"""If task is assigned to an inactive documenter, respawn them.
Returns True when the task is already assigned (whether or not a
respawn happened) so the caller can stop processing. Returns
False when the task is unassigned so the caller can auto-select
a documenter for it.
"""
assigned_to = task.get("assigned_to")
if not assigned_to:
return False
assigned_slug = self._resolve_agent_slug(assigned_to)
if self._is_agent_active(assigned_slug):
return True
if assigned_slug and "doc" in assigned_slug:
# Respawn circuit breaker — the fe-doc 26-respawn loop ran on this
# exact path unguarded; the gate notifies the CEO once it trips.
if await self._pm_respawn_should_gate(assigned_slug, task):
return True
await self.spawn_agent(
agent_id=assigned_slug,
task_id=task["id"],
initial_prompt=self._build_doc_prompt(task),
git_context=self._task_git_context(task),
spawned_by="_respawn_doc_if_assigned",
)
return True
async def _blocked_by_earlier_sibling(self, task: dict[str, Any]) -> bool:
"""True if a lower-sequence, same-team sibling is not yet terminal.
Sequence-ordered merge: leaf siblings share one cell branch, so merging
a later sibling before an earlier one diverges the branch and wedges the
loser. Hold a higher-sequence sibling's review/merge dispatch until the
earlier ones land (or are cancelled). Loop-free: the task simply isn't
dispatched this tick — no reject, no respawn churn.
Only same-team siblings block (they target the same branch). Terminal
siblings (completed/cancelled) never block, so a cancelled sibling can't
deadlock the rest. Best-effort: any lookup failure falls through to
dispatch — the ordering check must never wedge the dispatcher.
"""
parent_id = task.get("parent_task_id")
seq = task.get("sequence")
team = task.get("team")
if not parent_id or seq is None:
return False
from uuid import UUID
from roboco.db.base import get_session_factory
from roboco.models.base import TaskStatus
from roboco.services.task import get_task_service
terminal = {TaskStatus.COMPLETED, TaskStatus.CANCELLED}
try:
session_factory = get_session_factory()
async with session_factory() as db:
task_svc = get_task_service(db)
siblings = await task_svc.get_subtasks(UUID(str(parent_id)))
except Exception as exc:
logger.debug(
"sibling-order check failed; dispatching anyway",
task_id=task.get("id"),
error=str(exc),
)
return False
for sib in siblings:
sib_seq = getattr(sib, "sequence", 0) or 0
sib_team = getattr(sib, "team", None)
sib_status = getattr(sib, "status", None)
sib_team_val = getattr(sib_team, "value", sib_team)
if (
str(sib_team_val) == str(team)
and sib_seq < seq
and sib_status not in terminal
):
return True
return False
async def _blocked_by_earlier_lane_sibling(self, task: dict[str, Any]) -> bool:
"""True if the SAME dev has an earlier non-terminal code sibling.
Per-dev sequenced queues (Spec 3): a PM delegates a full queue of code
subtasks to each cell dev up front. This BUILD/dispatch barrier holds a
dev's higher-sequence code leaf until its own lower-sequence code
siblings under the same parent are terminal, so the dev works its queue
one live task at a time, in order — while the other dev's lane runs
concurrently (true two-dev parallelism).
Distinct from :meth:`_blocked_by_earlier_sibling` (the MERGE barrier,
keyed on team): this is keyed on the assignee and only gates ``code``.
Loop-free (skip this tick — no reject, no respawn churn) and best-effort
(any lookup failure falls through to dispatch so the check never wedges).
"""
if str(task.get("task_type") or "") != "code":
return False
parent_id = task.get("parent_task_id")
seq = task.get("sequence")
owner = task.get("assigned_to") or task.get("claimed_by")
if not parent_id or seq is None or not owner:
return False
from uuid import UUID
from roboco.db.base import get_session_factory
from roboco.models.base import TaskStatus
from roboco.services.task import get_task_service
terminal = {TaskStatus.COMPLETED, TaskStatus.CANCELLED}
try:
session_factory = get_session_factory()
async with session_factory() as db:
task_svc = get_task_service(db)
siblings = await task_svc.get_subtasks(UUID(str(parent_id)))
except Exception as exc:
logger.debug(
"lane-order check failed; dispatching anyway",
task_id=task.get("id"),
error=str(exc),
)
return False
task_id = str(task.get("id"))
return any(
self._is_earlier_live_lane_sibling(
sib, task_id=task_id, owner=str(owner), seq=seq, terminal=terminal
)
for sib in siblings
)
@staticmethod
def _is_earlier_live_lane_sibling(
sib: Any, *, task_id: str, owner: str, seq: int, terminal: set[Any]
) -> bool:
"""True if ``sib`` is a lower-sequence non-terminal code task for ``owner``."""
if str(sib.id) == task_id:
return False
sib_type = getattr(sib, "task_type", None)
sib_type_val = getattr(sib_type, "value", sib_type)
return (
str(getattr(sib, "assigned_to", None)) == owner
and str(sib_type_val) == "code"
and (getattr(sib, "sequence", 0) or 0) < seq
and getattr(sib, "status", None) not in terminal
)
async def _dispatch_pm_review_work(self, client: httpx.AsyncClient) -> None:
"""
Dispatch PM review work to cell PMs or Main PM.
Monitors: awaiting_pm_review tasks
Spawns: be-pm, fe-pm, ux-pm, main-pm
"""
tasks = await self._fetch_tasks(client, "awaiting_pm_review")
for task in tasks:
team = task.get("team")
assigned_to = task.get("assigned_to")
# Sequence-ordered merge: don't review/merge a leaf until its
# earlier same-team siblings have landed, so they merge into the
# shared cell branch in order instead of racing and wedging.
if await self._blocked_by_earlier_sibling(task):
continue
# If already assigned, check if that agent is running
if assigned_to:
assigned_slug = self._resolve_agent_slug(assigned_to)
# Human-only roles (CEO / prompter / secretary) are never
# containers — there is no reviewer agent to respawn. Leave
# the task for the human (the CEO approves via the panel).
# A stale/ex-human slug is also skipped: is_spawnable_agent_slug
# is False for it, so a renamed secretary slug can't slip past
# the layered guard to a doomed spawn (#49). Mirrors the
# spawn_agent human-role guard; a skip here keeps a mis-assigned
# human task from aborting this dispatcher's whole tick.
if not is_spawnable_agent_slug(assigned_slug):
continue
if self._is_agent_active(assigned_slug):
continue
# Loop guard: a review task that keeps re-surfacing without
# advancing (e.g. an unmergeable PR that re-blocks every cycle)
# must stop respawning the reviewer, else it burns tokens
# forever. The gate notifies the CEO once it trips.
if await self._pm_respawn_should_gate(assigned_slug, task):
continue
# Agent not running - spawn them to continue
await self.spawn_agent(
agent_id=assigned_slug,
task_id=task["id"],
initial_prompt=self._build_pm_review_prompt(task),
git_context=self._task_git_context(task),
spawned_by="_dispatch_pm_review_work",
)
continue
# Unassigned task - select PM based on team
# Cell tasks go to Cell PM, cross-cell/main_pm tasks go to Main PM
if team in ["backend", "frontend", "ux_ui"]:
pm_id = self._TEAM_PM_MAP.get(team, "be-pm")
else:
# main_pm, board, or no team → Main PM handles it
pm_id = "main-pm"
if self._is_agent_active(pm_id):
continue
# Claim the task for PM BEFORE spawning
if not await self._claim_task_for_agent(client, task["id"], pm_id):
logger.warning(
"Failed to claim awaiting_pm_review task for PM",
task_id=task["id"],
agent_id=pm_id,
)
continue
await self.spawn_agent(
agent_id=pm_id,
task_id=task["id"],
initial_prompt=self._build_pm_review_prompt(task),
git_context=self._task_git_context(task),
spawned_by="_dispatch_pm_review_work",
)
break
async def _dispatch_marketing_work(self, client: httpx.AsyncClient) -> None:
"""
Dispatch marketing work to head-marketing.
Monitors: pending tasks with team=marketing
Spawns: head-marketing
"""
tasks = await self._fetch_tasks(client, "pending", team="marketing")
for task in tasks:
# Skip already claimed/assigned tasks
if task.get("assigned_to"):
continue
if self._is_agent_active("head-marketing"):
# Already running, they'll pick up on scan
continue
await self.spawn_agent(
agent_id="head-marketing",
task_id=task["id"],
initial_prompt=self._build_marketing_prompt(task),
git_context=self._task_git_context(task),
spawned_by="_dispatch_marketing_work",
)
break
# =========================================================================
# SMART DISPATCHER - EVENT-BASED DISPATCHERS
# =========================================================================
def _blocker_resolver_slug(self, task: dict[str, Any]) -> str | None:
"""Pick the agent that should be dispatched to unblock ``task``.
The unblock content gate (note/unblock) is assignee-only: the
dispatched agent must be the task's CURRENT ``assigned_to``, or its
required pre-unblock decision note returns not_authorized and the
orchestrator respawns it forever (a livelock — a task escalated to
Main PM kept respawning the ex-assignee cell PM, which could not author
the note). So whenever the blocked task carries an assignee that is a
PM role, dispatch THAT assignee. Only a task with no PM assignee
(e.g. still held by the dev who raised i_am_blocked) falls back to the
cell PM for its team.
A BOARD/advisory assignee (product-owner / head-marketing) is the one
case we must NOT dispatch: a board role has no ``unblock`` verb at all
— its only moves are notify/note/triage/i_am_idle — so dispatching it
to "resolve" a blocker is a futile catch-22. It cannot unblock, cannot
hand the task off (the assignee-only gate also forbids any PM from
unblocking a task it does not own), and so it spam-notifies the CEO and
the orchestrator respawns it forever (observed: 6400+ tool calls burned
on a single delivery root mis-assigned to product-owner). Return None so
the blocker dispatch SKIPS it — the task is mis-owned and must be
re-routed / surfaced to the CEO out-of-band, never auto-respawned onto a
role that physically cannot act. (The upstream cure is to never assign a
board role as the owner of an executable delivery/coordination root.)
"""
assignee_uuid = task.get("assigned_to") or task.get("claimed_by")
if assignee_uuid:
assignee_slug = self._resolve_agent_slug(str(assignee_uuid))
if assignee_slug in self._BOARD_AGENTS:
return None
if assignee_slug in self._PM_AGENTS:
return assignee_slug
team = task.get("team")
if team not in ("backend", "frontend", "ux_ui"):
return None
return self._select_agent_for_cell(team, "pm")
async def _dispatch_blocker_work(self, client: httpx.AsyncClient) -> None:
"""
Dispatch blocker resolution to the task's current unblock authority.
Monitors: blocked tasks
Spawns: the task's current PM/board assignee, else the cell PM
"""
tasks = await self._fetch_tasks(client, "blocked")
for task in tasks:
# HITL-blocked tasks wait for a human; never spawn an agent on them.
if self._is_hitl_blocked(task):
continue
agent_id = self._blocker_resolver_slug(task)
if not agent_id:
continue
if self._is_agent_active(agent_id):
continue
# Loop guard: a blocked task whose unblock can never succeed (e.g.
# a cold-respawned PM that can't satisfy the unblock decision gate,
# or an unresolvable merge conflict) must stop respawning the
# resolver. The gate notifies the CEO once it trips so the wedged
# task surfaces instead of silently burning tokens.
if await self._pm_respawn_should_gate(agent_id, task):
continue
await self.spawn_agent(
agent_id=agent_id,
task_id=task["id"],
initial_prompt=self._build_pm_blocker_prompt(task),
git_context=self._task_git_context(task),
spawned_by="_dispatch_blocker_work",
)
break
def _claimed_task_needs_agent(self, task: dict[str, Any]) -> str | None:
"""Return the assignee slug to (re)spawn for an agentless claimed task.
A task left CLAIMED/IN_PROGRESS with an assignee but no running
container (e.g. a reassignment that didn't spawn) is invisibly stuck —
only PENDING tasks get fresh dispatch, and the heartbeat reaper can't
see it because the claim seeded a fresh heartbeat. Returns the assignee
slug when the task has sat past the grace window with no active agent;
``None`` when it is healthy, too fresh, or HITL-blocked.
"""
if self._is_hitl_blocked(task):
return None
owner_uuid = task.get("assigned_to") or task.get("claimed_by")
if not owner_uuid:
return None
agent_slug = self._resolve_agent_slug(str(owner_uuid))
# Human-only roles (CEO / prompter / secretary) are never containers —
# there is no agent to respawn. Leave the task as-is for the human to
# act on through the panel; do NOT release it to pending (that would
# re-route a human-owned task to a PM). A stale slug (None role) is NOT
# skipped here — a stale-slug claim SHOULD be released to pending so a
# real agent can reclaim it (recovery, not spawning). See spawn_agent's
# human-role guard for the structural backstop.
if is_human_only_role(role_for_slug_or_none(agent_slug)):
return None
# The assignee is running, and on THIS task — healthy.
instance = self._instances.get(agent_slug)
if instance is not None and instance.state == AgentState.ACTIVE:
return None
# Grace window: a just-claimed task whose spawn is still in flight must
# not be churned. _time_in_state under-counts (any update bumps it),
# which biases toward "agent is working" — exactly the safe direction.
age = self._time_in_state(task)
grace = settings.claimed_no_agent_grace_seconds
if age is None or age.total_seconds() < grace:
return None
return agent_slug
async def _dispatch_claimed_without_agent(self, client: httpx.AsyncClient) -> None:
"""(Re)spawn or release claimed/in_progress tasks that have no agent.
Net for the invisible-stuck case the other dispatchers miss: a task
held CLAIMED/IN_PROGRESS by an assignee with no running container. If
the assignee is a known spawnable agent, respawn it on the task; if not
(unknown slug — e.g. a stale UUID), release the claim to PENDING so the
normal routing reclaims it with a role match.
Throttle: spawns at most ONE container per tick (``break`` after the
first respawn), matching every sibling dispatcher. A restart leaves
many agentless claims at once; without the cap this single tick would
burst-spawn a container for every one of them. The release-to-pending
path spawns nothing, so it does not consume the per-tick spawn budget
and keeps draining stale claims.
"""
tasks = await self._fetch_tasks(client, ["claimed", "in_progress"])
for task in tasks:
task_id = task.get("id")
if self._is_task_handled_this_tick(task_id):
continue
agent_slug = self._claimed_task_needs_agent(task)
if agent_slug is None:
continue
if get_agent_role(agent_slug) in (None, "unknown"):
# Unknown assignee — no agent to spawn; release for re-dispatch.
await self._release_claim_to_pending(str(task_id))
continue
logger.warning(
"Claimed/in_progress task has no running agent; respawning assignee",
task_id=task_id,
agent=agent_slug,
status=task.get("status"),
)
await self.spawn_agent(
agent_id=agent_slug,
task_id=str(task_id),
initial_prompt=self._get_prompt_for_agent(agent_slug, task),
git_context=self._task_git_context(task),
spawned_by="_dispatch_claimed_without_agent",
)
break
async def _release_claim_to_pending(self, task_id: str) -> None:
"""Release a stuck claim back to PENDING via the lifecycle-safe path.
Reuses ``TaskService.unclaim_for_reaper`` (claimed/in_progress ->
pending, clears assignee + work session) so the state machine records
the transition rather than a raw status PATCH. Opens its own short-lived
session, mirroring ``_reap_stale_claims``.
"""
from roboco.db.base import get_session_factory
from roboco.services.task import TaskService
from roboco.utils.converters import require_uuid
try:
factory = get_session_factory()
async with factory() as db:
svc = TaskService(db)
await svc.unclaim_for_reaper(require_uuid(task_id))
await db.commit()
logger.warning(
"Released agentless claim to pending for re-dispatch",
task_id=task_id,
)
except Exception as exc:
logger.error(
"Failed to release agentless claim; will retry next tick",
task_id=task_id,
error=str(exc),
)
async def _dispatch_escalation_work(self, client: httpx.AsyncClient) -> None:
"""
Dispatch escalations to appropriate managers.
Monitors: escalation notifications (unacknowledged)
Spawns: be-pm, fe-pm, ux-pm, main-pm, product-owner, head-marketing
"""
notifications = await self._fetch_notifications(client, "blocker_escalation")
for notif in notifications:
targets = notif.get("to_agents", [])
for agent_id in targets:
# Resolve UUID to slug - to_agents contains UUIDs from database
agent_slug = self._resolve_agent_slug(str(agent_id))
valid_targets = [
"be-pm",
"fe-pm",
"ux-pm",
"main-pm",
"product-owner",
"head-marketing",
]
if agent_slug not in valid_targets:
continue
if self._is_agent_active(agent_slug):
continue
if self._notification_spawn_cooled(agent_slug, notif.get("id")):
continue
await self.spawn_agent(
agent_id=agent_slug,
initial_prompt=self._build_escalation_prompt(notif),
spawned_by="_dispatch_escalation_work",
)
break
async def _dispatch_approval_work(self, client: httpx.AsyncClient) -> None:
"""
Dispatch approval requests to approvers.
Monitors: approval notifications (unacknowledged)
Spawns: product-owner, head-marketing, main-pm
"""
notifications = await self._fetch_notifications(client, "approval")
for notif in notifications:
targets = notif.get("to_agents", [])
for agent_id in targets:
# Resolve UUID to slug - to_agents contains UUIDs from database
agent_slug = self._resolve_agent_slug(str(agent_id))
if agent_slug not in ["product-owner", "head-marketing", "main-pm"]:
continue
if self._is_agent_active(agent_slug):
continue
if self._notification_spawn_cooled(agent_slug, notif.get("id")):
continue
await self.spawn_agent(
agent_id=agent_slug,
initial_prompt=self._build_approval_prompt(notif),
spawned_by="_dispatch_approval_work",
)
break
async def _dispatch_audit_work(self, client: httpx.AsyncClient) -> None:
"""
Dispatch audit work to the auditor.
Monitors: quality alert notifications
Spawns: auditor
Note: Periodic scheduled audits can be added here in the future.
"""
alerts = await self._fetch_notifications(client, "alert")
for alert in alerts:
targets = alert.get("to_agents", [])
# Resolve UUIDs to slugs and check if auditor is a target
target_slugs = [self._resolve_agent_slug(str(t)) for t in targets]
if "auditor" in target_slugs and not self._is_agent_active("auditor"):
if self._notification_spawn_cooled("auditor", alert.get("id")):
continue
await self.spawn_agent(
agent_id="auditor",
initial_prompt=self._build_audit_prompt(alert),
spawned_by="_dispatch_audit_work",
)
return
# TODO: Add scheduled periodic audits
# Check last audit time, spawn if overdue
async def _detect_stuck_tasks(self, client: httpx.AsyncClient) -> None:
"""
Detect and auto-block tasks that are stuck.
This is a proactive enforcement mechanism that finds tasks which
have been pending without progress and have prerequisite issues.
Runs every dispatcher cycle but only takes action on truly stuck tasks.
CEO-approved timeout: 10 minutes
"""
STUCK_THRESHOLD_MINUTES = 10 # CEO-approved threshold
tasks = await self._fetch_tasks(client, "pending")
for task in tasks:
# never auto-block a CEO-held artifact (release_manager / x_post /
# video_post / ...); it sits PENDING by design until the CEO acts
if _is_held_ceo_source(task):
continue
age = self._get_task_age(task)
if age is None or age < timedelta(minutes=STUCK_THRESHOLD_MINUTES):
continue
issues = self._check_stuck_conditions(task)
issues.extend(await self._check_dev_subtask_issue(client, task))
if issues:
task_id = task.get("id")
if not task_id:
continue
age_mins = int(age.total_seconds() // 60)
reason = f"Task stuck for {age_mins} minutes: " + ", ".join(issues)
await self._auto_block_task(client, task_id, reason)
logger.warning(
"Auto-blocked stuck task",
task_id=task_id,
age_minutes=age_mins,
issues=issues,
)
# Per-(role, state) SLA check. Independent from the pending-task
# sweep above — different states, different action (escalate vs
# auto-block).
await self._detect_sla_exceeded(client)
async def _check_sla_for_task(
self,
client: httpx.AsyncClient,
task: dict[str, Any],
status: str,
) -> None:
"""Check one task's SLA; escalate if exceeded. No-ops on missing data."""
from roboco.enforcement.task_lifecycle import sla_seconds_for
assigned = task.get("assigned_to")
if not assigned:
return
assigned_slug = self._resolve_agent_slug(assigned)
role = get_agent_role(assigned_slug or "")
sla = sla_seconds_for(role, status)
if sla is None:
return
age = self._time_in_state(task)
if age is None or age.total_seconds() < sla:
return
task_id = task.get("id")
if not task_id:
return
await self._escalate_sla_breach(
client,
_SlaBreach(
task_id=str(task_id),
role=role or "",
status=status,
age_seconds=int(age.total_seconds()),
sla_seconds=sla,
),
)
async def _detect_sla_exceeded(self, client: httpx.AsyncClient) -> None:
"""Auto-escalate tasks that exceeded their per-role SLA.
Uses ROLE_STATE_SLA_KEYS in enforcement/task_lifecycle.py. Dev tasks
stuck in `in_progress`/`verifying`, QA tasks in `claimed`, doc tasks
in `claimed`, and cell-PM tasks in `claimed` all get a soft bump so
work doesn't silently rot.
"""
from roboco.enforcement.task_lifecycle import ROLE_STATE_SLA_KEYS
# Fetch each (role, state) combo we care about. One API call per
# unique status so we don't fan out pointlessly.
statuses = sorted({state for _, state in ROLE_STATE_SLA_KEYS})
for status in statuses:
try:
tasks = await self._fetch_tasks(client, status)
except Exception as e:
logger.debug(
"SLA sweep fetch failed; skipping status",
status=status,
error=str(e),
)
continue
for task in tasks:
await self._check_sla_for_task(client, task, status)
def _time_in_state(self, task: dict[str, Any]) -> timedelta | None:
"""Approximate time in current state via task.updated_at.
Not perfect — any field update bumps `updated_at`, not just status
changes — but it's the coarse signal we have, and it under-counts
(biased toward "agent is working") rather than over-counts, which
matches the soft-SLA intent.
"""
updated_at = task.get("updated_at") or task.get("created_at")
if not updated_at:
return None
try:
if updated_at.endswith("Z"):
updated_at = updated_at[:-1] + "+00:00"
parsed = datetime.fromisoformat(updated_at)
if parsed.tzinfo is None:
parsed = parsed.replace(tzinfo=UTC)
return datetime.now(UTC) - parsed
except (ValueError, TypeError):
return None
async def _escalate_sla_breach(
self, client: httpx.AsyncClient, breach: _SlaBreach
) -> None:
"""Record SLA breach in dev_notes and nudge state forward.
We don't force a state transition here — the MCP lifecycle rules are
still authoritative. We log, annotate the task, and notify the
assignee's escalation target. The agent's next spawn picks up the
updated notes and usually self-escalates.
"""
age_mins = breach.age_seconds // 60
sla_mins = breach.sla_seconds // 60
note = (
f"[SLA] role={breach.role} status={breach.status} "
f"time_in_state={age_mins}m sla={sla_mins}m. "
"Escalating — agent should call escalate_up() "
"or unclaim()."
)
try:
await client.patch(
f"{self._api_url}/tasks/{breach.task_id}",
json={"dev_notes": note},
)
logger.warning(
"SLA breach noted on task",
task_id=breach.task_id,
role=breach.role,
status=breach.status,
age_minutes=age_mins,
sla_minutes=sla_mins,
)
except Exception as e:
logger.debug(
"SLA breach annotation failed",
task_id=breach.task_id,
error=str(e),
)
def _get_task_age(self, task: dict[str, Any]) -> timedelta | None:
"""Parse task created_at and return age, or None if unparseable."""
created_at_str = task.get("created_at")
if not created_at_str:
return None
try:
if created_at_str.endswith("Z"):
created_at_str = created_at_str[:-1] + "+00:00"
created_at = datetime.fromisoformat(created_at_str)
if created_at.tzinfo is None:
created_at = created_at.replace(tzinfo=UTC)
return datetime.now(UTC) - created_at
except (ValueError, TypeError):
return None
_MIN_DESCRIPTION_LEN = 10
def _check_stuck_conditions(self, task: dict[str, Any]) -> list[str]:
"""Check for common stuck conditions (git, description)."""
issues: list[str] = []
# A branch only exists once a task is claimed; a coordination task does
# no git at all. A pending, never-claimed code task therefore has no
# branch by design — flagging that here auto-blocked tasks before their
# first dispatch. Only flag a missing branch when the task is in a
# state where it should already own one.
if not task.get("branch_name") and _branch_is_expected(task):
issues.append("Task missing branch_name")
description = (task.get("description") or "").strip()
if len(description) < self._MIN_DESCRIPTION_LEN:
issues.append("Empty or inadequate description")
return issues
async def _check_dev_subtask_issue(
self, client: httpx.AsyncClient, task: dict[str, Any]
) -> list[str]:
"""Check if complex dev task is missing subtasks."""
from roboco.agents_config import get_agent_role
assigned_to = task.get("assigned_to")
if not assigned_to:
return []
agent_slug = self._resolve_agent_slug(assigned_to)
if not agent_slug or get_agent_role(agent_slug) != "developer":
return []
complexity = task.get("estimated_complexity", "low")
is_low_complexity = complexity not in ("medium", "high")
if is_low_complexity or task.get("parent_task_id"):
return []
try:
resp = await client.get(f"{self._api_url}/tasks/{task.get('id')}/subtasks")
subtasks = resp.json() if resp.is_success else []
except Exception:
subtasks = []
if not subtasks:
return [f"{complexity} complexity task without subtasks"]
return []
async def _dispatch_a2a_work(self, client: httpx.AsyncClient) -> None:
"""
Dispatch A2A (Agent-to-Agent) requests to target agents.
Monitors: a2a_request notifications (unacknowledged)
Spawns: Any agent that is the target of an A2A request
This is a fallback mechanism - primary A2A routing happens via events.
If the event-based spawn fails, these notifications will be picked up here.
"""
notifications = await self._fetch_notifications(client, "a2a_request")
for notif in notifications:
targets = notif.get("to_agents", [])
for agent_id in targets:
# Resolve UUID to slug - to_agents contains UUIDs from database
agent_slug = self._resolve_agent_slug(str(agent_id))
# Human-only roles (CEO / prompter / secretary) are never
# dispatched — the CEO is the human operator and intake/
# secretary are human-driven chats with their own launch
# paths. Spawning a container for one is a trust violation
# (the system acting as the human CEO). A stale/ex-human slug
# is skipped too (is_spawnable_agent_slug is False for it) so
# a renamed secretary slug can't slip past to a spawn (#49).
# The CEO being a notification target (board-review handoff,
# escalation, etc.) is expected; it is NOT a spawn signal.
# Skip — the notification stays for the human to read. #75:
# surface the skip for a human-only target (vs a silent stale
# slug) so an a2a expecting a human-side action (a CEO sign-off
# relay) is visible in the dispatch log, not silently dropped.
if is_human_only_role(role_for_slug_or_none(agent_slug)):
logger.info(
"a2a request targets a human-only role; left as a "
"notification for the human (not spawned)",
target_slug=agent_slug,
)
continue
if not is_spawnable_agent_slug(agent_slug):
continue
if self._is_agent_active(agent_slug):
# Agent is online - SDK handles A2A delivery directly
# No action needed here, SDK server receives messages
continue
# Agent is offline - spawn them with A2A context
if self._notification_spawn_cooled(agent_slug, notif.get("id")):
continue
await self.spawn_agent(
agent_id=agent_slug,
initial_prompt=self._build_a2a_prompt(notif),
spawned_by="_dispatch_a2a_work",
)
break
# =========================================================================
# SMART DISPATCHER - PROMPT BUILDERS
# =========================================================================
def _get_workflow_state(
self,
status: str,
has_plan: bool,
) -> str:
"""Determine developer workflow state from task attributes.
Args:
status: Task status (claimed, in_progress, needs_revision, etc.)
has_plan: Whether task has a plan submitted
Returns:
Workflow state string (NEEDS_PLAN, READY_TO_START, EXECUTING, etc.)
"""
# Direct status mappings
status_map = {
"in_progress": "EXECUTING",
"needs_revision": "REVISION_REQUIRED",
"verifying": "VERIFYING",
}
if status in status_map:
return status_map[status]
# Handle claimed status with sub-states
if status == "claimed":
if not has_plan:
return "NEEDS_PLAN"
return "READY_TO_START"
return status.upper()
def _get_workflow_instructions(self, state: str, task_id: str) -> str:
"""Get workflow instructions for the given state.
Args:
state: Workflow state (NEEDS_PLAN, READY_TO_START, etc.)
task_id: Task ID for tool call examples
Returns:
Markdown-formatted instructions for the current state
"""
instructions = {
"NEEDS_PLAN": f"""## NEXT STEP: Claim + Plan + Start
Call i_will_work_on(task_id="{task_id}",
plan="<approach, ordered steps, risks, open questions>").
This single verb claims the task, records your plan, and transitions
to in_progress.
""",
"READY_TO_START": f"""## NEXT STEP: Start Work
Call i_will_work_on(task_id="{task_id}", plan="<your plan as a string>")
to begin.
""",
"EXECUTING": """## IN PROGRESS
Continue development. Required gates before i_am_done() will succeed
(enforced server-side — `remediate` tells you what's missing):
1. commit("<type(scope): subject, >=20 chars>")
— makes the git commit, auto-prefixes task ID, records progress.
Repeat per meaningful chunk.
2. note(scope='decision'|'learning'|'reflect', task_id="...", text=...)
as you make trade-offs.
When acceptance criteria are met, call
open_pr(task_id="...") to push your branch and open the PR,
then i_am_done(task_id="...", notes="<self-verification summary>")
to submit for QA review.
If you hit something you can't unblock yourself:
i_am_blocked(task_id="...",
reason="<blocked_external|low_context|...>").
""",
"REVISION_REQUIRED": f"""## REVISION REQUESTED
QA or PM requested changes:
1. evidence(task_id="{task_id}") — read qa_notes / pm_notes / inline diff
2. i_will_work_on(task_id="{task_id}",
plan="<revised plan addressing each issue>")
3. commit() the fixes, then
i_am_done(task_id="{task_id}", notes="<what was fixed>")
""",
"VERIFYING": f"""## SELF-VERIFICATION
Run the project's quality checks against acceptance criteria:
1. Run tests, lint, type checks in your workspace.
2. evidence(task_id="{task_id}") — sanity-check inline diff + commits.
3. If everything passes:
i_am_done(task_id="{task_id}", notes="<verification summary>")
— chains submit_verification + push + create_pr + submit_qa.
4. If issues found: commit() the fixes and retry.
""",
}
return instructions.get(
state, f'Call evidence(task_id="{task_id}") to check status.'
)
def _build_dev_prompt(self, task: dict[str, Any]) -> str:
"""Build state-aware initial prompt for a developer."""
task_id = task.get("id", "unknown")
title = task.get("title", "Untitled")
status = task.get("status", "unknown")
# Determine workflow state based on task attributes
has_plan = bool(task.get("plan"))
workflow_state = self._get_workflow_state(status, has_plan)
instructions = self._get_workflow_instructions(workflow_state, task_id)
return f"""You have been assigned a development task.
TASK ID: {task_id}
TITLE: {title}
STATUS: {status}
WORKFLOW STATE: {workflow_state}
{instructions}
Start by calling evidence(task_id="{task_id}") for full details and acceptance criteria.
When out of work: i_am_idle().
"""
def _build_qa_prompt(self, task: dict[str, Any]) -> str:
"""Build initial prompt for a QA agent."""
task_id = task.get("id", "unknown")
title = task.get("title", "Untitled")
assigned_to = task.get("assigned_to", "unknown")
team = task.get("team", "unknown")
return f"""A task is ready for QA review.
TASK ID: {task_id}
TITLE: {title}
DEVELOPER: {assigned_to}
TEAM: {team}
== QA WORKFLOW ==
1. claim_review(task_id="{task_id}")
— assigns the QA seat; returns inline diff + PR + commits as evidence.
The PR is already open (dev opened it before submitting QA);
review on GitHub if you need more context.
2. Review the implementation against EVERY acceptance criterion.
Run/read tests; sanity-check the diff for regressions, security,
and scope creep.
3. Decide:
- PASS: pass(task_id="{task_id}",
notes="<>=80 chars: what you verified, which AC, evidence>")
— transitions awaiting_qa → awaiting_documentation.
- FAIL: fail(task_id="{task_id}",
issues=["concrete issue 1", "concrete issue 2", ...])
— transitions to needs_revision; each issue must be specific and
actionable.
4. note(scope='reflect'|'learning', task_id="{task_id}", text=...)
for anything worth flagging.
5. give_me_work() to pick up the next QA item,
or i_am_idle() if the queue is empty.
"""
def _build_pr_review_prompt(self, task: dict[str, Any]) -> str:
"""Build the initial prompt for the PR reviewer on an external PR."""
task_id = task.get("id", "unknown")
title = task.get("title", "Untitled")
pr_number = task.get("pr_number", "?")
pr_url = task.get("pr_url", "")
return f"""An external contributor opened a pull request. Review it.
TASK ID: {task_id}
TITLE: {title}
EXTERNAL PR: #{pr_number} {pr_url}
== TRUST BOUNDARY ==
This PR is from OUTSIDE the org — the code is untrusted. The review is
READ-ONLY: you read the diff, you do NOT fetch, check out, build, or run the
contributor's code. Do not push to their fork. You never merge.
== REVIEW WORKFLOW ==
1. claim_pr_review(task_id="{task_id}")
— starts the review; returns the contributor's unified diff inline.
2. Review the diff adversarially: correctness, security (injection, secret
leaks, supply-chain/dependency risk), scope, and the codebase's standards.
Reason about it from the diff alone — do not run it.
3. note(scope="learning", task_id="{task_id}", text="<what the review surfaced>")
— required before you can post.
4. post_pr_review(task_id="{task_id}",
body="<one complete change-request: per-finding file + line + expected
vs actual; be specific and actionable>",
event="REQUEST_CHANGES")
— posts ONE complete review to the PR and finishes the task. Use
event="APPROVE" only if the PR is genuinely ready as-is.
5. i_am_idle() when done.
"""
def _build_pr_gate_prompt(self, task: dict[str, Any]) -> str:
"""Build the prompt for a reviewer on an in-path assembled-PR gate task."""
task_id = task.get("id", "unknown")
title = task.get("title", "Untitled")
team = task.get("team", "unknown")
pr_number = task.get("pr_number", "?")
pr_url = task.get("pr_url", "")
criteria = task.get("acceptance_criteria") or []
crit_block = (
"\n".join(f" - {c}" for c in criteria) if criteria else " (none recorded)"
)
return f"""\
An assembled pull request is ready for review before the PM merges it.
TASK ID: {task_id}
TITLE: {title}
TEAM: {team}
ASSEMBLED PR: #{pr_number} {pr_url}
== WHAT YOU ARE REVIEWING ==
This is the gate BEFORE the merge — the merge-level review QA does not do. You
review the ASSEMBLED diff (the whole cell→root or root→master PR), not a single
leaf, against the original intent and the contract between cells. The bug class
this catches lives in the seam (e.g. a frontend that sends a string where the
backend requires a UUID) — invisible to any single-cell QA. Read-only: you
never push or merge.
ACCEPTANCE CRITERIA (the assembled work must satisfy ALL of these):
{crit_block}
== REVIEW WORKFLOW ==
1. claim_gate_review(task_id="{task_id}")
— claims the review; returns the assembled diff + acceptance criteria inline.
2. Review the diff against the objective + every acceptance criterion + the
FE↔BE / cross-cell contract. Do not lose scope: the assembled thing must
actually do what was asked.
3. note(scope="learning", task_id="{task_id}", text="<what the review surfaced>")
— required before you pass or fail.
4a. pr_pass(task_id="{task_id}", notes="<how you verified the assembled work>")
— if correct and complete: moves it to the PM to merge.
4b. pr_fail(task_id="{task_id}", issues=["<concrete, actionable gap>", ...])
— if anything is wrong: sends it back to the PM for revision, like a QA fail.
5. i_am_idle() when done.
"""
def _build_doc_prompt(self, task: dict[str, Any]) -> str:
"""Build initial prompt for a documenter."""
task_id = task.get("id", "unknown")
title = task.get("title", "Untitled")
team = task.get("team", "unknown")
return f"""A task is ready for documentation. The dev's PR is already open
— you're documenting alongside the QA-passed branch.
TASK ID: {task_id}
TITLE: {title}
TEAM: {team}
== DOC WORKFLOW ==
1. claim_doc_task(task_id="{task_id}")
— assigns the doc seat and opens your workspace on the task's branch.
2. evidence(task_id="{task_id}") — read dev handoff notes, qa_notes,
and the inline diff so the docs reflect what actually shipped.
3. Write/update docs in your workspace: README sections, API references,
code comments, migration notes, or new docs files as the change requires.
4. commit("docs(scope): <subject, >=20 chars>") per logical doc chunk
— auto-prefixes the task ID and stages tracked changes.
5. i_documented(task_id="{task_id}",
notes="<>=20 chars: what you documented and where>",
files=["docs/foo.md", "README.md", ...])
— transitions awaiting_documentation → awaiting_pm_review.
6. give_me_work() for the next doc item,
or i_am_idle() if the queue is empty.
"""
def _build_pm_review_prompt(self, task: dict[str, Any]) -> str:
"""Prompt for PM reviewing a SUBTASK in awaiting_pm_review."""
task_id = task.get("id", "unknown")
title = task.get("title", "Untitled")
team = task.get("team", "unknown")
return f"""A SUBTASK in your cell is awaiting your PM review.
It has passed QA and documentation; the leaf PR is open and ready to merge.
TASK ID: {task_id}
TITLE: {title}
TEAM: {team}
== PM REVIEW WORKFLOW (leaf subtask) ==
1. evidence(task_id="{task_id}")
— review PR, commits, inline diff, dev_notes, qa_notes, doc files.
2. Spot-check that:
- every acceptance criterion is satisfied,
- QA's pass notes line up with the actual diff,
- docs reflect what shipped.
3. note(scope='decision', task_id="{task_id}",
text="<approve rationale or rejection reason>")
— REQUIRED before complete().
4. Decide:
- APPROVE: complete(task_id="{task_id}", notes="<merge rationale>")
— auto-merges the leaf PR and finalizes the subtask.
- NEEDS REWORK: leave a clear note(scope='decision', text="...") and
rely on the dispatcher to respawn the dev for revision.
Use escalate_up only if the issue is truly outside your cell.
5. give_me_work() / triage() for the next item, or i_am_idle().
Never `commit`, never write code, never run `git`. PMs coordinate.
"""
def _build_board_prompt(self, task: dict[str, Any]) -> str:
"""Prompt for a board agent (Product Owner / Head of Marketing) to
review and SHAPE a strategic task. Board roles advise — they do not
build, code, or delegate."""
task_id = task.get("id", "unknown")
title = task.get("title", "Untitled")
description = task.get("description", "No description")
return f"""\
You are on the Board. This strategic task is under board review.
TASK: {task_id}
TITLE: {title}
DESCRIPTION: {description}
THE BOARD REVIEWS AS A PAIR: the Product Owner AND the Head of Marketing both
review every board task before it reaches the CEO. The Product Owner owns
product requirements + acceptance scope; the Head of Marketing owns the UX /
user-facing / positioning dimension. The CEO only gets the handoff after BOTH
of you have recorded a review.
YOUR ROLE: review and shape this work. You do NOT build, code, claim, or
delegate — those verbs are not yours. Your deliverable is a recorded review.
== WHAT TO DO ==
1. triage()
— see your board-level work and context.
2. note(text="<the product requirements and acceptance criteria you expect, the
scope, the must-haves, and what 'done' looks like — Head of Marketing:
the UX, user-facing impact, and how the feature is positioned>",
scope='decision', task_id="{task_id}")
— this recorded review is how the CEO and Main PM act on your input.
3. dm(...) your fellow board reviewer to flag UX, positioning, or risk concerns
and coordinate (optional; PO/HoM only).
4. i_am_idle()
— when your review is recorded. Once both board reviewers are done, the
CEO is notified the task is ready for Approve & Start, then routes it to
Main PM for delegation to the cells; you do NOT hand it off yourself.
Do NOT attempt to claim, plan, complete, or delegate — the gateway will reject
those, and a substantive recorded note IS your job here.
"""
def _build_roadmap_prompt(self, task: dict[str, Any]) -> str:
"""Prompt for the Product Owner's one-shot roadmap-exploration cycle.
Unlike the two-reviewer board-review prompt, this is PO-solo (v1 —
see the roadmap spec's non-goals): explore, author ONE themed cycle,
then idle. No claim/plan/delegate/complete — those verbs aren't the
Product Owner's."""
task_id = task.get("id", "unknown")
min_items = settings.roadmap_min_items_per_cycle
max_items = settings.roadmap_max_items_per_cycle
return f"""\
You are the Product Owner. It's time for your periodic roadmap exploration.
TASK: {task_id}
Explore the company's projects and propose ONE themed cycle of roadmap items
for the CEO to review — you author this alone. The Head of Marketing is not
involved in this cycle.
== WHAT TO DO ==
1. triage() — see your board-level context.
2. Explore: read the company charter, recent releases, metrics, and each
project's current state (read-only git). Check the knowledge base for open
threads. Optionally run web research for market/competitive signal.
3. Pick ONE theme/goal for this cycle — a one-line focus that ties the items
together (e.g. "close onboarding friction" or "harden the payments path").
4. propose_roadmap(cycle_goal="<the theme>", items=[...])
— call this EXACTLY ONCE with {min_items}-{max_items} item drafts. Each
item is an object with: title, description, acceptance_criteria (list
of strings), project_slug, team ('backend'|'frontend'|'ux_ui'),
priority (1-4, default 2), rationale (why this, why now).
5. i_am_idle() — once proposed. The CEO reviews and approves/rejects each
item individually in the roadmap queue; an approved item lands in BACKLOG
for normal PM activation — nothing here auto-starts.
Do NOT claim, plan, delegate, or attempt to start any of the items yourself —
that is not your job here, and the gateway will reject those verbs.
"""
def _build_feature_spotlight_prompt(self, task: dict[str, Any]) -> str:
"""Prompt for the Head of Marketing's one-shot feature-spotlight cycle."""
task_id = task.get("id", "unknown")
markers_dict = task.get("orchestration_markers") or {}
seen = markers_dict.get(_markers.X_SEEN_FEATURES) or []
seen_line = ", ".join(seen) if seen else "(none yet — this is the first cycle)"
return f"""\
You are the Head of Marketing. It's time for your periodic feature-spotlight cycle.
TASK: {task_id}
RoboCo markets its own capabilities, not just releases. Investigate what the
company has actually shipped and draft ONE marketing post about a genuinely
useful, under-publicized capability — something a user or prospect would not
already know from the last release announcement.
ALREADY COVERED — do not repeat: {seen_line}
== WHAT TO DO ==
1. triage() — see your board-level context.
2. Investigate (read-only, you have full repo read access): CHANGELOG.md (what
has actually shipped), the feature-flags ledger (panel/src/components/
settings/feature-flags-card.tsx and roboco/services/settings.py's
FEATURE_FLAGS — the enumerated subsystems), docs/map/ (the exhaustive
codebase map — each slice's Purpose section is marketing-readable), the
company charter (already in your briefing), and the knowledge base
(roboco_ask_mentor / roboco_kb_search).
3. Pick ONE feature that is real, currently shipped (or shipped behind a flag
the CEO can enable), not in the already-covered list above, and worth
telling people about.
4. Draft ONE post in your voice (see your identity's VOICE GUIDE) — plain
text, no markdown, no thread, max 280 characters, and never invent a
capability that doesn't exist.
5. propose_feature_spotlight(feature_slug="<a short stable slug>",
feature_title="<human-readable feature name>", body="<the post>")
— call this EXACTLY ONCE.
6. i_am_idle() — once proposed. The CEO reviews, edits, approves, or rejects
the draft in the X post queue; nothing posts without that explicit
approval.
Do NOT claim, plan, delegate, or attempt to post anything yourself — that is
not your job here, and the gateway will reject those.
"""
def _build_marketing_prompt(self, task: dict[str, Any]) -> str:
"""Build initial prompt for head-marketing with a marketing task."""
task_id = task.get("id", "unknown")
title = task.get("title", "Untitled")
description = task.get("description", "No description")
return f"""You have been assigned a marketing task.
TASK ID: {task_id}
TITLE: {title}
DESCRIPTION: {description}
Begin work:
1. Review the task details above (full acceptance criteria arrive in your
briefing / the give_me_work response)
2. Execute the marketing task (content, campaigns, research, etc.)
3. Coordinate with Product Owner or Main PM if needed
4. Call i_am_done() when done
5. Call give_me_work() to check for more marketing work
6. If no more work, call i_am_idle() to shutdown gracefully
"""
def _build_pm_blocker_prompt(self, task: dict[str, Any]) -> str:
"""Build initial prompt for a Cell PM handling a blocker."""
task_id = task.get("id", "unknown")
title = task.get("title", "Untitled")
assigned_to = task.get("assigned_to", "unknown")
blocker = task.get("blocker", {})
reason = blocker.get("reason", "Unknown")
what_needed = blocker.get("what_needed", "Unknown")
return f"""A task in your cell is BLOCKED and needs your attention.
TASK ID: {task_id}
TITLE: {title}
ASSIGNED TO: {assigned_to}
BLOCKER REASON: {reason}
WHAT'S NEEDED: {what_needed}
Your job:
1. Understand the blocker by reviewing task details
2. Communicate with the blocked developer if needed
3. Resolve the blocker (coordinate resources, make decisions, escalate if needed)
4. Once resolved, call unblock("{task_id}") to release the task back to the developer
5. Call triage() to check for other blocked tasks in your cell
6. If no more blockers, call i_am_idle() to shutdown gracefully
"""
def _build_escalation_prompt(self, notification: dict[str, Any]) -> str:
"""Build initial prompt for handling an escalation."""
notif_id = notification.get("id", "unknown")
from_agent = notification.get("from_agent", "unknown")
subject = notification.get("subject", "No subject")
priority = notification.get("priority", "normal")
body = notification.get("body", "No details provided")
return f"""You have received an ESCALATION that requires your attention.
FROM: {from_agent}
SUBJECT: {subject}
PRIORITY: {priority}
DETAILS:
{body}
Your job:
1. Acknowledge the notification with notify_ack("{notif_id}")
2. Assess the escalation and determine action needed
3. Communicate decisions via dm / notify
4. If this requires further escalation, use escalate_up()
5. When resolved, call triage() for other work
6. If no more work, call i_am_idle() to shutdown gracefully
"""
def _build_approval_prompt(self, notification: dict[str, Any]) -> str:
"""Build initial prompt for handling an approval request."""
notif_id = notification.get("id", "unknown")
from_agent = notification.get("from_agent", "unknown")
subject = notification.get("subject", "No subject")
related_task_id = notification.get("related_task_id", "None")
body = notification.get("body", "No details provided")
return f"""You have received an APPROVAL REQUEST.
FROM: {from_agent}
SUBJECT: {subject}
RELATED TASK: {related_task_id}
REQUEST:
{body}
Your job:
1. Review the approval request carefully
2. If related to a task, use the task context provided in your briefing
3. Make your decision and communicate it
4. Acknowledge with notify_ack("{notif_id}")
5. Call triage() for other work
6. If no more work, call i_am_idle() to shutdown gracefully
"""
def _build_audit_prompt(self, alert: dict[str, Any] | None = None) -> str:
"""Build initial prompt for the auditor."""
if alert:
subject = alert.get("subject", "Quality issue detected")
body = alert.get("body", "Review system quality metrics")
return f"""QUALITY ALERT triggered your attention.
ALERT: {subject}
DETAILS: {body}
Your job:
1. Investigate the quality issue
2. Review relevant tasks and history (you have read access to all)
3. Compile your findings
4. Report to CEO via your journal (note scope='reflect')
5. Call i_am_idle() when complete
"""
return """Periodic AUDIT requested.
Your job:
1. Review recent activity across all cells
2. Check quality metrics (QA pass/fail rates, blocker frequency, etc.)
3. Identify any concerns or patterns
4. Compile audit report for CEO
5. Call i_am_idle() when complete
"""
def _build_a2a_prompt(self, notification: dict[str, Any]) -> str:
"""Build initial prompt for handling an A2A (Agent-to-Agent) request.
Reads `priority` directly off the notification row (set by
NotificationService.send_a2a_notification). Pre-Phase-3 this
consumed a non-existent `metadata.urgent` and always rendered
urgency_note=False; the column-level priority is now the source
of truth.
"""
notif_id = notification.get("id", "unknown")
from_agent = notification.get("from_agent", "unknown")
body = notification.get("body", "No message provided")
related_task_id = notification.get("related_task_id")
metadata = notification.get("metadata", {})
skill = metadata.get("skill", "general")
priority_raw = notification.get("priority", "normal")
# URGENT gets the bold attention-grabber; HIGH gets a quieter
# "higher priority" hint; NORMAL gets no prefix.
if priority_raw == "urgent":
urgency_note = "**URGENT** - This request has priority.\n\n"
elif priority_raw == "high":
urgency_note = "**HIGH PRIORITY** - Please handle promptly.\n\n"
else:
urgency_note = ""
task_note = f"RELATED TASK: {related_task_id}\n" if related_task_id else ""
return f"""You have received an A2A (Agent-to-Agent) REQUEST.
{urgency_note}FROM: {from_agent}
SKILL: {skill}
{task_note}
REQUEST:
{body}
Your job:
1. Acknowledge the notification with notify_ack("{notif_id}")
2. Process the request using your {skill} capabilities
3. Respond to {from_agent} using dm("{from_agent}", ...)
4. If you need task context, it is provided in your briefing for the related task
5. When done, call give_me_work() for other work
6. If no more work, call i_am_idle() to shutdown gracefully
"""