fix(workflow): both parallel and single branch errors occur in if-else (

langgenius#8378)
cuiks · Sep 26, 2024 · a1ac104 · a1ac104
1 parent 3d4521e
commit a1ac104
Showing 1 changed file with 15 additions and 14 deletions.
diff --git a/api/core/workflow/graph_engine/entities/graph.py b/api/core/workflow/graph_engine/entities/graph.py
@@ -405,21 +405,22 @@ def _recursively_add_parallels(
 
             if condition_edge_mappings:
                 for condition_hash, graph_edges in condition_edge_mappings.items():
-                    current_parallel = cls._get_current_parallel(
-                        parallel_mapping=parallel_mapping,
-                        graph_edge=graph_edge,
-                        parallel=condition_parallels.get(condition_hash),
-                        parent_parallel=parent_parallel,
-                    )
+                    for graph_edge in graph_edges:
+                        current_parallel: GraphParallel | None = cls._get_current_parallel(
+                            parallel_mapping=parallel_mapping,
+                            graph_edge=graph_edge,
+                            parallel=condition_parallels.get(condition_hash),
+                            parent_parallel=parent_parallel,
+                        )
 
-                    cls._recursively_add_parallels(
-                        edge_mapping=edge_mapping,
-                        reverse_edge_mapping=reverse_edge_mapping,
-                        start_node_id=graph_edge.target_node_id,
-                        parallel_mapping=parallel_mapping,
-                        node_parallel_mapping=node_parallel_mapping,
-                        parent_parallel=current_parallel,
-                    )
+                        cls._recursively_add_parallels(
+                            edge_mapping=edge_mapping,
+                            reverse_edge_mapping=reverse_edge_mapping,
+                            start_node_id=graph_edge.target_node_id,
+                            parallel_mapping=parallel_mapping,
+                            node_parallel_mapping=node_parallel_mapping,
+                            parent_parallel=current_parallel,
+                        )
             else:
                 for graph_edge in target_node_edges:
                     current_parallel = cls._get_current_parallel(