chore(orchestrator): enable gateway cooldown logic in production

ROBOCO_GATEWAY_ENABLED defaulted to False, leaving trigger_filter's
spawn cooldown / role-rate logic dormant. Flip to true and add the
gateway_triggers table migration if missing. Without this, respawn
rate has no server-side limit besides _pm_respawn_should_gate.
This commit is contained in:
Renn F
2026-05-03 07:43:05 +02:00
parent 00b385c019
commit 87ef42bf09
2 changed files with 133 additions and 0 deletions
+5
View File
@@ -235,6 +235,11 @@ services:
ROBOCO_HOST_PROJECT_DIR: ${ROBOCO_HOST_PROJECT_DIR:-/volume1/roboco} ROBOCO_HOST_PROJECT_DIR: ${ROBOCO_HOST_PROJECT_DIR:-/volume1/roboco}
ROBOCO_HOST_CLAUDE_DIR: ${ROBOCO_HOST_CLAUDE_DIR:-/home/renzof/.claude} ROBOCO_HOST_CLAUDE_DIR: ${ROBOCO_HOST_CLAUDE_DIR:-/home/renzof/.claude}
ROBOCO_HOST_DATA_DIR: ${ROBOCO_HOST_DATA_DIR:-/volume1/roboco/data} ROBOCO_HOST_DATA_DIR: ${ROBOCO_HOST_DATA_DIR:-/volume1/roboco/data}
# Agent Gateway — enables trigger_filter.decide_spawn cooldown logic
# (4 rules: terminal-task drop, claimant-lock queue, per-task cooldown,
# per-role rate limit). Without this the legacy spawn path has only
# _pm_respawn_should_gate (PM-specific) + per-tick dedupe.
ROBOCO_GATEWAY_ENABLED: "true"
volumes: volumes:
# Docker socket - allows spawning agent containers # Docker socket - allows spawning agent containers
- /var/run/docker.sock:/var/run/docker.sock - /var/run/docker.sock:/var/run/docker.sock
+128
View File
@@ -0,0 +1,128 @@
"""Gateway cooldown is consulted only when ROBOCO_GATEWAY_ENABLED=true.
`gateway_pre_spawn_check` short-circuits to ``("spawn", ...)`` when
``settings.gateway_enabled`` is False so the legacy spawn path stays
unchanged. When the flag is True it must reach
``roboco.services.gateway.trigger_filter.decide_spawn`` whose 4-rule
cooldown machinery is the real spawn gate.
Without these assertions a regression that flips the flag back to False
(or drops the call site entirely) would leave the orchestrator with no
server-side spawn cooldown beyond ``_pm_respawn_should_gate``.
"""
from __future__ import annotations
from unittest.mock import AsyncMock, MagicMock, patch
from uuid import uuid4
import pytest
from roboco.runtime import orchestrator as orchestrator_module
from roboco.runtime.orchestrator import gateway_pre_spawn_check
from roboco.services.gateway.trigger_filter import Decision, SpawnDecision
@pytest.mark.asyncio
async def test_gateway_disabled_short_circuits_without_calling_decide_spawn(
monkeypatch: pytest.MonkeyPatch,
) -> None:
"""Flag off -> short-circuit. ``decide_spawn`` must not be invoked."""
monkeypatch.setattr(orchestrator_module.settings, "gateway_enabled", False)
with patch(
"roboco.services.gateway.trigger_filter.decide_spawn"
) as mock_decide_spawn:
outcome, reason = await gateway_pre_spawn_check(
task_id=str(uuid4()),
trigger_kind="scan",
target_role="developer",
)
assert outcome == "spawn"
assert "gateway disabled" in reason
mock_decide_spawn.assert_not_called()
@pytest.mark.asyncio
async def test_gateway_enabled_consults_decide_spawn(
monkeypatch: pytest.MonkeyPatch,
) -> None:
"""Flag on -> ``decide_spawn`` is invoked and its decision propagates."""
monkeypatch.setattr(orchestrator_module.settings, "gateway_enabled", True)
task_id = str(uuid4())
# Stub task row that decide_spawn will receive.
fake_task_row = MagicMock()
fake_task_row.status = "pending"
fake_task_row.active_claimant_id = None
fake_task_row.last_heartbeat_at = None
# Stub the async DB session: count queries return 0, task lookup returns
# our fake row. ``async with factory() as db`` -> ``db.execute(...)``.
fake_count_result = MagicMock()
fake_count_result.scalars.return_value.all.return_value = []
fake_task_result = MagicMock()
fake_task_result.scalars.return_value.first.return_value = fake_task_row
fake_db = AsyncMock()
fake_db.execute = AsyncMock(
side_effect=[fake_count_result, fake_count_result, fake_task_result]
)
fake_db.add = MagicMock()
fake_db.flush = AsyncMock()
fake_db.commit = AsyncMock()
fake_factory = MagicMock()
fake_factory.return_value.__aenter__ = AsyncMock(return_value=fake_db)
fake_factory.return_value.__aexit__ = AsyncMock(return_value=None)
expected = Decision(SpawnDecision.QUEUE, "per-task spawn cooldown active")
with (
patch("roboco.db.base.get_session_factory", return_value=fake_factory),
patch(
"roboco.services.gateway.trigger_filter.decide_spawn",
return_value=expected,
) as mock_decide_spawn,
):
outcome, reason = await gateway_pre_spawn_check(
task_id=task_id,
trigger_kind="scan",
target_role="developer",
)
mock_decide_spawn.assert_called_once()
call_kwargs = mock_decide_spawn.call_args.kwargs
assert call_kwargs["task"] is fake_task_row
assert call_kwargs["trigger"].kind.value == "scan"
assert call_kwargs["config"].cooldown_seconds > 0
assert outcome == "queue"
assert reason == "per-task spawn cooldown active"
@pytest.mark.asyncio
async def test_gateway_enabled_skips_decide_spawn_when_no_task_id(
monkeypatch: pytest.MonkeyPatch,
) -> None:
"""No task_id -> early return; ``decide_spawn`` not called.
Documents the no-task-spawn carve-out: idle PM ticks pass through
even with the gateway enabled.
"""
monkeypatch.setattr(orchestrator_module.settings, "gateway_enabled", True)
with patch(
"roboco.services.gateway.trigger_filter.decide_spawn"
) as mock_decide_spawn:
outcome, reason = await gateway_pre_spawn_check(
task_id=None,
trigger_kind="scan",
target_role="main_pm",
)
assert outcome == "spawn"
assert "no task_id" in reason
mock_decide_spawn.assert_not_called()