AI/dify
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/langgenius/dify.git synced 2025-08-15 11:15:54 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix(iteration): iterator_length not correct

Browse Source
This commit is contained in:
takatost 2024-09-03 12:01:56 +08:00
parent 3431b19f9a
commit 36d95e49b0
1 changed files with 7 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
api/core/workflow/nodes/iteration/iteration_node.py
View File

@ -41,7 +41,12 @@ class IterationNode(BaseNode):
Run the node. Run the node.
""" """
self.node_data = cast(IterationNodeData, self.node_data) self.node_data = cast(IterationNodeData, self.node_data)
iterator_list_value = self.graph_runtime_state.variable_pool.get_any(self.node_data.iterator_selector) iterator_list_segment = self.graph_runtime_state.variable_pool.get(self.node_data.iterator_selector)
if not iterator_list_segment:
raise ValueError(f"Iterator variable {self.node_data.iterator_selector} not found")
iterator_list_value = iterator_list_segment.to_object()
if not isinstance(iterator_list_value, list): if not isinstance(iterator_list_value, list):
raise ValueError(f"Invalid iterator value: {iterator_list_value}, please provide a list.") raise ValueError(f"Invalid iterator value: {iterator_list_value}, please provide a list.")
@ -138,7 +143,7 @@ class IterationNode(BaseNode):
start_at=start_at, start_at=start_at,
inputs=inputs, inputs=inputs,
metadata={ metadata={
"iterator_length": 1 "iterator_length": len(iterator_list_value)
}, },
predecessor_node_id=self.previous_node_id predecessor_node_id=self.previous_node_id
) )