mirror of
https://github.com/openswarm-ai/openswarm.git
synced 2026-09-14 05:37:40 +02:00
ciregenzandClaude Fable 5.1
e6b3865851
Co-Authored-By: Claude Fable 5.1 <noreply@anthropic.com> Claude-Session: https://claude.ai/code/session_012G8kyALnPjsA7aJFmMBq3R
49 lines
2.8 KiB
Python
49 lines
2.8 KiB
Python
"""The unwedge envelope must say what the children looked like at the kill.
|
|
|
|
Haik's install: 154 CreateBrowserAgent kills in 14 days at a 150 s floor, and the envelope carried
|
|
only tool, seconds and pids, so nobody could tell "child finished, result lost" (recovery by design)
|
|
from "child died first" (a different bug). One shared helper feeds the watchdog's settled test AND the
|
|
envelope, so the two can never disagree about which children count."""
|
|
import datetime
|
|
from backend.apps.agents.agent_manager import agent_manager
|
|
from backend.apps.agents.core.models import AgentSession
|
|
from backend.apps.agents.manager.streaming import unwedge_sidecar as us
|
|
from backend.apps.agents.manager.streaming.delegation_watchdog import delegation_children_settled
|
|
from backend.apps.service import client as svc
|
|
|
|
|
|
def p_child(parent_id: str, status: str, born_ago_s: float) -> AgentSession:
|
|
s = AgentSession(name="Browser Agent", model="sonnet", mode="browser-agent", status=status)
|
|
s.parent_session_id = parent_id
|
|
s.created_at = datetime.datetime.now() - datetime.timedelta(seconds=born_ago_s)
|
|
return s
|
|
|
|
|
|
def test_the_envelope_carries_each_child_status_and_age(monkeypatch) -> None:
|
|
parent = AgentSession(name="p", model="sonnet")
|
|
agent_manager.sessions.clear(); agent_manager.sessions[parent.id] = parent
|
|
for kid in (p_child(parent.id, "completed", 30), p_child(parent.id, "error", 10), p_child(parent.id, "running", 5), p_child(parent.id, "completed", 900)):
|
|
agent_manager.sessions[kid.id] = kid
|
|
sent: list = []
|
|
monkeypatch.setattr(svc, "submit_diagnostic", lambda d: sent.append(d))
|
|
monkeypatch.setattr(us, "find_sidecar_pids", lambda sid: [424242])
|
|
monkeypatch.setattr(us.subprocess, "run", lambda *a, **k: None)
|
|
monkeypatch.setattr(us.time, "sleep", lambda s: None)
|
|
us.unwedge(parent.id, "mcp__openswarm-core__CreateBrowserAgent", 150.0)
|
|
env = [d for d in sent if isinstance(d, dict) and d.get("kind") == "mcp_sidecar_unwedged"]
|
|
assert len(env) == 1
|
|
kids = env[0]["children"]
|
|
assert sorted(k["status"] for k in kids) == ["completed", "error", "running"], "only children born after the call started; the 900 s one is an earlier delegation's"
|
|
assert all(isinstance(k["age_s"], float) for k in kids)
|
|
|
|
|
|
def test_the_watchdog_and_the_envelope_count_the_same_children() -> None:
|
|
parent = AgentSession(name="p", model="sonnet")
|
|
agent_manager.sessions.clear(); agent_manager.sessions[parent.id] = parent
|
|
agent_manager.sessions.update({k.id: k for k in (p_child(parent.id, "completed", 20), p_child(parent.id, "completed", 600))})
|
|
import time
|
|
since = time.time() - 100
|
|
assert len(us.delegation_children_born_after(parent.id, since)) == 1
|
|
assert delegation_children_settled(parent.id, since) is True
|
|
assert len(us.children_summary(parent.id, since)) == 1
|