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

feat: support interrupt protocol

Browse Source
This commit is contained in:
He Tao
2025-04-15 16:36:02 +08:00
parent 83a4b668b7
commit 92606fbd4c
4 changed files with 37 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
src/graph/nodes.py
View File

@@ -75,10 +75,10 @@ def human_feedback_node(
# check if the plan is auto accepted
auto_accepted_plan = state.get("auto_accepted_plan", False)
if not auto_accepted_plan:
feedback = interrupt(current_plan)
feedback = interrupt("Please Review the Plan.")
# if the feedback is not accepted, return the planner node
if feedback and str(feedback).upper() != "[ACCEPTED]":
if feedback and str(feedback).upper().startswith("[EDIT_PLAN]"):
return Command(
update={
"messages": [
@@ -87,7 +87,7 @@ def human_feedback_node(
},
goto="planner",
)
elif feedback and str(feedback).upper() == "[ACCEPTED]":
elif feedback and str(feedback).upper().startswith("[ACCEPTED]"):
logger.info("Plan is accepted by user.")
else:
raise TypeError(f"Interrupt value of {feedback} is not supported.")