opensource
/
dify
mirror of https://github.com/langgenius/dify.git
2
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

refactor(graph_engine): Remove `node_id` from SkipPropagator.skip_branch_paths 

Signed-off-by: -LAN- <laipz8200@outlook.com>
This commit is contained in:
-LAN- 2025-08-28 04:43:56 +08:00
parent 98b25c0bbc
commit c82697f267
No known key found for this signature in database
GPG Key ID: 6BA0D108DED011FF
2 changed files with 2 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
api/core/workflow/graph_engine/graph_traversal/branch_handler.py
View File

@ -65,7 +65,7 @@ class BranchHandler:
selected_edges, unselected_edges = self.edge_state_manager.categorize_branch_edges(node_id, selected_handle)
# Skip all unselected paths
self.skip_propagator.skip_branch_paths(node_id, unselected_edges)
self.skip_propagator.skip_branch_paths(unselected_edges)
# Process selected edges and get ready nodes and streaming events
return self.edge_processor.process_node_success(node_id, selected_handle)

3
api/core/workflow/graph_engine/graph_traversal/skip_propagator.py
View File

@ -83,12 +83,11 @@ class SkipPropagator:
# Recursively propagate skip
self.propagate_skip_from_edge(edge.id)
def skip_branch_paths(self, node_id: str, unselected_edges: list) -> None:
def skip_branch_paths(self, unselected_edges: Sequence[Edge]) -> None:
"""
Skip all paths from unselected branch edges.
Args:
node_id: The ID of the branch node
unselected_edges: List of edges not taken by the branch
"""
for edge in unselected_edges: