From 6ae4bc588a0022d067c61da21f16d50980519d4a Mon Sep 17 00:00:00 2001 From: Willem Jiang Date: Wed, 29 Oct 2025 10:12:32 +0800 Subject: [PATCH] fix: remove the unnessary conditional edge. (#671) --- src/graph/builder.py | 6 ------ tests/unit/graph/test_builder.py | 4 ++-- 2 files changed, 2 insertions(+), 8 deletions(-) diff --git a/src/graph/builder.py b/src/graph/builder.py index aeedaff..45576b7 100644 --- a/src/graph/builder.py +++ b/src/graph/builder.py @@ -63,12 +63,6 @@ def _build_base_graph(): ["planner", "researcher", "coder"], ) builder.add_edge("reporter", END) - # Add conditional edges for coordinator to handle clarification flow - builder.add_conditional_edges( - "coordinator", - lambda state: state.get("goto", "planner"), - ["planner", "background_investigator", "coordinator", END], - ) return builder diff --git a/tests/unit/graph/test_builder.py b/tests/unit/graph/test_builder.py index 31a35df..9c10284 100644 --- a/tests/unit/graph/test_builder.py +++ b/tests/unit/graph/test_builder.py @@ -96,8 +96,8 @@ def test_build_base_graph_adds_nodes_and_edges(MockStateGraph): # Check that all nodes and edges are added assert mock_builder.add_edge.call_count >= 2 assert mock_builder.add_node.call_count >= 8 - # Now we have 2 conditional edges: research_team and coordinator - assert mock_builder.add_conditional_edges.call_count == 2 + # Now we have 1 conditional edges: research_team + assert mock_builder.add_conditional_edges.call_count == 1 @patch("src.graph.builder._build_base_graph")