fix: iteration invalid output selector doesn't throw an error (#10544)

langgenius · Nov 11, 2024 · 4b45ef6 · 4b45ef6
1 parent a1543b7
commit 4b45ef6
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/api/core/workflow/nodes/iteration/iteration_node.py b/api/core/workflow/nodes/iteration/iteration_node.py
@@ -489,7 +489,10 @@ def _run_single_iter(
                             )
                     yield metadata_event
 
-            current_iteration_output = variable_pool.get(self.node_data.output_selector).value
+            current_output_segment = variable_pool.get(self.node_data.output_selector)
+            if current_output_segment is None:
+                raise IterationNodeError("iteration output selector not found")
+            current_iteration_output = current_output_segment.value
             outputs[current_index] = current_iteration_output
             # remove all nodes outputs from variable pool
             for node_id in iteration_graph.node_ids: