wanwu/deer-flow
1
0
Fork 0
mirror of https://gitee.com/wanwujie/deer-flow synced 2026-04-18 20:14:44 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: clean up the builder code (#417)

Browse Source 
* fix: clean up the builder code

* fix:reformat the code
This commit is contained in:
Willem Jiang
2025-07-15 17:22:50 +08:00
committed by GitHub
parent ae30517f48
commit 0f118fda92
2 changed files with 22 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -66,6 +66,17 @@ def test_continue_to_running_research_team_next_coder(mock_state):
assert builder_mod.continue_to_running_research_team(state) == "coder"
def test_continue_to_running_research_team_next_coder_withresult(mock_state):
Step = mock_state["Step"]
Plan = mock_state["Plan"]
steps = [
Step(execution_res=True),
Step(execution_res=True, step_type=builder_mod.StepType.PROCESSING),
]
state = {"current_plan": Plan(steps=steps)}
assert builder_mod.continue_to_running_research_team(state) == "planner"
def test_continue_to_running_research_team_default_planner(mock_state):
Step = mock_state["Step"]
Plan = mock_state["Plan"]