wanwu/deer-flow
1
0
Fork 0
mirror of https://gitee.com/wanwujie/deer-flow synced 2026-04-17 19:44:45 +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

13
src/graph/builder.py
View File

@@ -22,14 +22,23 @@ def continue_to_running_research_team(state: State):
current_plan = state.get("current_plan")
if not current_plan or not current_plan.steps:
return "planner"
if all(step.execution_res for step in current_plan.steps):
return "planner"
# Find first incomplete step
incomplete_step = None
for step in current_plan.steps:
if not step.execution_res:
incomplete_step = step
break
if step.step_type and step.step_type == StepType.RESEARCH:
if not incomplete_step:
return "planner"
if incomplete_step.step_type == StepType.RESEARCH:
return "researcher"
if step.step_type and step.step_type == StepType.PROCESSING:
if incomplete_step.step_type == StepType.PROCESSING:
return "coder"
return "planner"