From c82697f26714465e19ceb6963b4d571e8fcfa5cf Mon Sep 17 00:00:00 2001 From: -LAN- Date: Thu, 28 Aug 2025 04:43:56 +0800 Subject: [PATCH] refactor(graph_engine): Remove `node_id` from SkipPropagator.skip_branch_paths Signed-off-by: -LAN- --- .../workflow/graph_engine/graph_traversal/branch_handler.py | 2 +- .../workflow/graph_engine/graph_traversal/skip_propagator.py | 3 +-- 2 files changed, 2 insertions(+), 3 deletions(-) diff --git a/api/core/workflow/graph_engine/graph_traversal/branch_handler.py b/api/core/workflow/graph_engine/graph_traversal/branch_handler.py index deddd86eb8..36f047b2b7 100644 --- a/api/core/workflow/graph_engine/graph_traversal/branch_handler.py +++ b/api/core/workflow/graph_engine/graph_traversal/branch_handler.py @@ -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) diff --git a/api/core/workflow/graph_engine/graph_traversal/skip_propagator.py b/api/core/workflow/graph_engine/graph_traversal/skip_propagator.py index ef0e5e3273..572ed4deb6 100644 --- a/api/core/workflow/graph_engine/graph_traversal/skip_propagator.py +++ b/api/core/workflow/graph_engine/graph_traversal/skip_propagator.py @@ -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: