mirror of
https://github.com/rennf93/roboco.git
synced 2026-08-03 07:23:24 +02:00
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:
@@ -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
|
||||||
|
|||||||
@@ -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()
|
||||||
Reference in New Issue
Block a user