File size: 2,061 Bytes
cb1b811 6fe947e cb1b811 6fe947e cb1b811 6fe947e cb1b811 6fe947e cb1b811 6fe947e cb1b811 6fe947e cb1b811 6fe947e cb1b811 6fe947e cb1b811 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 | """Tests for graph wiring, routing, and the judge evaluation loop."""
from types import SimpleNamespace
from unittest.mock import MagicMock, patch
from gaia_agent.nodes import (
judge,
route_after_judge,
route_after_research,
)
from gaia_agent.schemas import JudgeVerdict
def test_graph_compiles():
from gaia_agent.graph import graph
assert graph.name == "GAIA Agent"
nodes = set(graph.get_graph().nodes.keys())
assert {"ingest_file", "research", "tools", "solver", "judge", "formatter"} <= nodes
def test_route_after_research_tools_vs_solver():
with_calls = SimpleNamespace(tool_calls=[{"name": "tavily_search"}])
without = SimpleNamespace(tool_calls=[])
assert route_after_research({"messages": [with_calls]}) == "tools"
assert route_after_research({"messages": [without]}) == "solver"
def test_route_after_judge_pass():
state = {"verdict": JudgeVerdict(verdict="PASS"), "attempts": 0}
assert route_after_judge(state) == "formatter"
def test_route_after_judge_revise_within_budget():
state = {"verdict": JudgeVerdict(verdict="REVISE"), "attempts": 0}
with patch("gaia_agent.nodes.get_settings", return_value=SimpleNamespace(max_judge_retries=1)):
assert route_after_judge(state) == "research"
def test_route_after_judge_revise_budget_exhausted():
state = {"verdict": JudgeVerdict(verdict="REVISE"), "attempts": 1}
with patch("gaia_agent.nodes.get_settings", return_value=SimpleNamespace(max_judge_retries=1)):
assert route_after_judge(state) == "formatter"
def test_judge_increments_attempts_on_revise():
structured = MagicMock()
structured.invoke.return_value = JudgeVerdict(
verdict="REVISE", feedback="wrong format", missing=["units"]
)
with patch("gaia_agent.nodes.get_structured_llm", return_value=structured):
out = judge({"question": "q", "candidate": None, "attempts": 0})
assert out["attempts"] == 1
assert out["verdict"].verdict == "REVISE"
assert "messages" in out # feedback injected back into the loop
|