from __future__ import annotations

from schemas.agent.runtime_models import RouterAgentOutput, WorkerAgentOutputLite


def test_router_agent_output_parses_simplified_contract() -> None:
    payload = {
        "objective": "查询今天的日程安排",
        "context_summary": "用户询问天气",
        "requires_tool_evidence": True,
    }

    model = RouterAgentOutput.model_validate(payload)

    assert model.objective == "查询今天的日程安排"
    assert model.requires_tool_evidence is True


def test_worker_agent_output_lite_keeps_suggested_actions() -> None:
    payload = {
        "status": "success",
        "answer": "done",
        "suggested_actions": ["要不要我继续帮你查明天的安排？"],
    }

    model = WorkerAgentOutputLite.model_validate(payload)

    assert model.suggested_actions == ["要不要我继续帮你查明天的安排？"]