From b9f34f679f0fa5ec8d101861fe111200fc16cdaf Mon Sep 17 00:00:00 2001 From: takatost Date: Mon, 26 Aug 2024 22:00:17 +0800 Subject: [PATCH] fix: iteration start node id --- api/core/workflow/nodes/iteration/iteration_node.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/api/core/workflow/nodes/iteration/iteration_node.py b/api/core/workflow/nodes/iteration/iteration_node.py index a988bcf06d..377fd27664 100644 --- a/api/core/workflow/nodes/iteration/iteration_node.py +++ b/api/core/workflow/nodes/iteration/iteration_node.py @@ -64,7 +64,7 @@ class IterationNode(BaseNode): ): start_node_ids.append(node_config.get('id')) - if len(start_node_ids) > 1: + if len(start_node_ids) > 0: # add new fake iteration start node that connect to all start nodes root_node_id = f"{self.node_id}-start" graph_config['nodes'].append({ @@ -81,6 +81,9 @@ class IterationNode(BaseNode): "target": start_node_id }) else: + if not self.node_data.start_node_id: + raise ValueError(f'field start_node_id in iteration {self.node_id} not found') + root_node_id = self.node_data.start_node_id # init graph