wanwu/deer-flow
1
0
Fork 0
mirror of https://gitee.com/wanwujie/deer-flow synced 2026-04-19 12:24:46 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: remove the unnessary conditional edge. (#671)

Browse Source
This commit is contained in:
Willem Jiang
2025-10-29 10:12:32 +08:00
committed by GitHub
parent 0415f622da
commit 6ae4bc588a
2 changed files with 2 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
src/graph/builder.py
View File

@@ -63,12 +63,6 @@ def _build_base_graph():
["planner", "researcher", "coder"], ["planner", "researcher", "coder"],
) )
builder.add_edge("reporter", END) 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 return builder

4
tests/unit/graph/test_builder.py
View File

@@ -96,8 +96,8 @@ def test_build_base_graph_adds_nodes_and_edges(MockStateGraph):
# Check that all nodes and edges are added # Check that all nodes and edges are added
assert mock_builder.add_edge.call_count >= 2 assert mock_builder.add_edge.call_count >= 2
assert mock_builder.add_node.call_count >= 8 assert mock_builder.add_node.call_count >= 8
# Now we have 2 conditional edges: research_team and coordinator # Now we have 1 conditional edges: research_team
assert mock_builder.add_conditional_edges.call_count == 2 assert mock_builder.add_conditional_edges.call_count == 1
@patch("src.graph.builder._build_base_graph") @patch("src.graph.builder._build_base_graph")