From 617ea4b3b8b77eb40f19d570615f048a28993e90 Mon Sep 17 00:00:00 2001 From: takatost Date: Tue, 20 Aug 2024 22:16:41 +0800 Subject: [PATCH] fix(workflow): fix parallel bug --- api/core/workflow/graph_engine/entities/graph.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/api/core/workflow/graph_engine/entities/graph.py b/api/core/workflow/graph_engine/entities/graph.py index 52e0803d90..4061075ba5 100644 --- a/api/core/workflow/graph_engine/entities/graph.py +++ b/api/core/workflow/graph_engine/entities/graph.py @@ -319,8 +319,8 @@ class Graph(BaseModel): end_to_node_id: Optional[str] = None for node_id in node_parallel_mapping: - if not end_to_node_id and edge_mapping.get(node_id): - node_edges = edge_mapping[node_id] + node_edges = edge_mapping.get(node_id) + if not end_to_node_id and node_edges and len(node_edges) == 1: target_node_id = node_edges[0].target_node_id if node_parallel_mapping.get(target_node_id) == parent_parallel_id: end_to_node_id = target_node_id