mirror of
https://github.com/razzant/ouroboros.git
synced 2026-10-03 04:07:04 +00:00
f3ed795ea1
Keep the same task and live browser while required owner input lends active worker capacity. Restore acknowledged planned continuations without replaying effects or changing the original budget threshold. Align visible Publish admission, root questions, ongoing replies, display fallback names and definite publication failures with existing owners. Co-authored-by: Ouroboros <311266734+ouroboros-agent@users.noreply.github.com>
69 lines
4 KiB
Python
69 lines
4 KiB
Python
"""Chat-promoted roots retain host lineage through the ordinary writers."""
|
|
|
|
from types import SimpleNamespace
|
|
|
|
import pytest
|
|
|
|
from ouroboros.agent import OuroborosAgent
|
|
from ouroboros.agent_task_pipeline import _store_task_result
|
|
from ouroboros.task_results import load_task_result, resolve_task_lineage, write_task_result
|
|
from supervisor import workers
|
|
|
|
|
|
@pytest.mark.parametrize("title,expected", [("", "Write the report"), ("Chosen by the model", "Chosen by the model")])
|
|
def test_promoted_name_survives_admission_receipt_and_live_event(tmp_path, monkeypatch, title, expected):
|
|
from supervisor.events import _handle_promote_chat_to_task
|
|
|
|
pending, named = [], []
|
|
monkeypatch.setattr(workers, "DRIVE_ROOT", tmp_path)
|
|
monkeypatch.setattr(workers, "PENDING", pending)
|
|
monkeypatch.setattr(workers, "RUNNING", {})
|
|
monkeypatch.setattr(workers, "_broadcast_task_named", named.append)
|
|
ctx = SimpleNamespace(
|
|
DRIVE_ROOT=tmp_path, WORKERS={0: SimpleNamespace()}, PENDING=pending, RUNNING={}, bridge=None,
|
|
enqueue_task=lambda task: pending.append(task) or task,
|
|
persist_queue_snapshot=lambda **kwargs: True,
|
|
load_state=lambda: {"owner_chat_id": 1}, append_jsonl=lambda *args: None,
|
|
)
|
|
outcome = _handle_promote_chat_to_task({
|
|
"task_id": "named-root", "routing_token": "name-admission", "chat_id": 1,
|
|
"objective": "Write the report\nFull unshortened working instructions", "title": title,
|
|
"workspace": "none",
|
|
}, ctx)
|
|
assert outcome == {"status": "scheduled", "task_id": "named-root"}
|
|
assert pending[0]["title"] == title
|
|
assert pending[0]["suggested_name"] == expected
|
|
assert pending[0]["description"].endswith("Full unshortened working instructions")
|
|
assert load_task_result(tmp_path, "named-root")["suggested_name"] == expected
|
|
assert named == [{"type": "task_named", "task_id": "named-root", "suggested_name": expected}]
|
|
|
|
|
|
def test_promoted_payload_survives_running_terminal_and_retry(tmp_path, monkeypatch):
|
|
monkeypatch.setattr(workers, "DRIVE_ROOT", tmp_path)
|
|
monkeypatch.setattr(workers, "PENDING", [])
|
|
monkeypatch.setattr(workers, "RUNNING", {})
|
|
enqueued = []
|
|
ctx = SimpleNamespace(enqueue_task=lambda task: enqueued.append(task) or task,
|
|
persist_queue_snapshot=lambda **kwargs: True,
|
|
load_state=lambda: {"owner_chat_id": 1})
|
|
outcome = workers.promote_chat_to_task({"task_id": "promoted", "objective": "Write the report",
|
|
"workspace": "none", "chat_id": 1}, ctx)
|
|
assert outcome["status"] == "scheduled"
|
|
task = enqueued[0]
|
|
assert task["root_task_id"] == "promoted" and task["delegation_role"] == "root"
|
|
assert task.get("actor_id") is None
|
|
write_task_result(tmp_path, "promoted", "scheduled", root_task_id="promoted", delegation_role="root")
|
|
env = SimpleNamespace(drive_root=tmp_path, repo_dir=tmp_path, drive_path=lambda rel: tmp_path / rel)
|
|
OuroborosAgent._persist_running_record(SimpleNamespace(env=env), task)
|
|
running = load_task_result(tmp_path, "promoted")
|
|
assert running["root_task_id"] == "promoted" and running["delegation_role"] == "root"
|
|
_store_task_result(env, task, "The report is ready.", {"rounds": 1, "cost": 0}, {"tool_calls": []})
|
|
terminal = load_task_result(tmp_path, "promoted")
|
|
assert terminal["status"] == "completed"
|
|
assert terminal["root_task_id"] == "promoted" and terminal["delegation_role"] == "root"
|
|
# The reaper copies the admitted task and changes the physical identity.
|
|
retry = {**task, "id": "retried", "original_task_id": "promoted", "timeout_retry_from": "promoted"}
|
|
lineage = resolve_task_lineage(retry["id"], root_task_id=retry["root_task_id"],
|
|
delegation_role=retry["delegation_role"],
|
|
original_task_id=retry["original_task_id"], timeout_retry_from=retry["timeout_retry_from"])
|
|
assert lineage["is_retry_root_attempt"] is True and lineage["is_root_task"] is True
|