[F038/F039] orchestrator: sign X-Agent-Token on self-API calls

The prior self-PATCH 401 fix only carried X-Agent-ID/X-Agent-Role. Arming
ROBOCO_AGENT_AUTH_REQUIRED=true made the middleware require a signed
X-Agent-Token, so every orchestrator self-call (auto-block / auto-resume /
auto-recover / SLA annotation) 401'd and silently no-op'd — wedging
paused/blocked parents.

Add _system_api_headers() that wraps the base headers with a signed token
for the system identity (issue_agent_token); switch all six self-call sites.
Dev fallback: no secret set => UNSIGNED sentinel + auth not required.
This commit is contained in:
Renn F
2026-06-28 11:07:56 +02:00
parent 20f290ef29
commit f8d31bf940
2 changed files with 60 additions and 7 deletions
+30 -6
View File
@@ -102,6 +102,30 @@ _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", ""
),
}
# Consecutive failed recovery probes before the CEO is notified once per episode.
_CEO_NOTIFY_THRESHOLD = 10
@@ -2744,7 +2768,7 @@ class AgentOrchestrator:
try:
async with httpx.AsyncClient(
timeout=5.0, headers=_SYSTEM_API_HEADERS
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):
@@ -3043,7 +3067,7 @@ class AgentOrchestrator:
"""Best-effort GET /tasks/{id}; returns task dict or None on failure."""
try:
async with httpx.AsyncClient(
timeout=5.0, headers=_SYSTEM_API_HEADERS
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:
@@ -4518,7 +4542,7 @@ class AgentOrchestrator:
sdk_url = f"http://roboco-agent-{agent_id}:{SDK_PORT}/usage/status"
try:
async with httpx.AsyncClient(
timeout=3.0, headers=_SYSTEM_API_HEADERS
timeout=3.0, headers=_system_api_headers()
) as client:
resp = await client.get(sdk_url)
if resp.status_code == http_status.HTTP_200_OK:
@@ -4775,7 +4799,7 @@ class AgentOrchestrator:
_usage_total_cost = 0.0
async with httpx.AsyncClient(
timeout=3.0, headers=_SYSTEM_API_HEADERS
timeout=3.0, headers=_system_api_headers()
) as client:
for agent_id, instance in list(self._instances.items()):
if instance.state not in (
@@ -5439,7 +5463,7 @@ Start by:
if not self._instances:
return
async with httpx.AsyncClient(
timeout=3.0, headers=_SYSTEM_API_HEADERS
timeout=3.0, headers=_system_api_headers()
) as client:
for agent_id, instance in list(self._instances.items()):
if instance.state not in (
@@ -8071,7 +8095,7 @@ Start now: evidence(task_id="{task_id}")
dispatchers: list[tuple[str, Any]] = []
async with httpx.AsyncClient(
timeout=30.0, headers=_SYSTEM_API_HEADERS
timeout=30.0, headers=_system_api_headers()
) as client:
dispatchers = [
("pm_work", self._dispatch_pm_work(client)),
+30 -1
View File
@@ -9,10 +9,14 @@ system identity; these tests lock that the identity is both *present* and
*authorized* for task writes (otherwise the self-call would 403 instead of act).
"""
import secrets
import pytest
from roboco.agents_config import verify_agent_token
from roboco.foundation import identity as _foundation
from roboco.models import AgentRole
from roboco.models.permissions import TASK_PERMISSIONS, TaskAction
from roboco.runtime.orchestrator import _SYSTEM_API_HEADERS
from roboco.runtime.orchestrator import _SYSTEM_API_HEADERS, _system_api_headers
def test_system_api_headers_match_the_system_identity() -> None:
@@ -26,3 +30,28 @@ def test_system_identity_is_authorized_for_task_writes() -> None:
# auto-recover / auto-resume drive — is gated behind TaskAction.ASSIGN.
# The identity the orchestrator sends must hold it.
assert TaskAction.ASSIGN in TASK_PERMISSIONS[AgentRole.SYSTEM]
def test_system_api_headers_carry_signed_token(monkeypatch: pytest.MonkeyPatch) -> None:
# F038/F039: the orchestrator's self-API calls must carry a signed
# X-Agent-Token for the system identity, or 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 self-PATCH 401 fix is incomplete without it.
monkeypatch.setenv("ROBOCO_AGENT_AUTH_SECRET", secrets.token_hex(32))
headers = _system_api_headers()
token = headers["X-Agent-Token"]
assert token and token != "UNSIGNED"
assert verify_agent_token(token, headers["X-Agent-ID"], "system", "")
def test_system_api_headers_unsigned_when_secret_unset(
monkeypatch: pytest.MonkeyPatch,
) -> None:
# Dev fallback: with no secret set, the token is the UNSIGNED sentinel and
# auth is not required, so the self-call still succeeds. The header is
# present either way so a future arm-when-secret-set doesn't silently break.
monkeypatch.delenv("ROBOCO_AGENT_AUTH_SECRET", raising=False)
headers = _system_api_headers()
assert headers["X-Agent-Token"] == "UNSIGNED"