Browse Source

Update iteration_node.py to fix #14098: Token count increases exponen… (#14100)

tags/1.0.0
E Chen 8 months ago
parent
commit
a97d73ab05
No account linked to committer's email address
1 changed files with 1 additions and 0 deletions
  1. 1
    0
      api/core/workflow/nodes/iteration/iteration_node.py

+ 1
- 0
api/core/workflow/nodes/iteration/iteration_node.py View File

@@ -590,6 +590,7 @@ class IterationNode(BaseNode[IterationNodeData]):
with flask_app.app_context():
parallel_mode_run_id = uuid.uuid4().hex
graph_engine_copy = graph_engine.create_copy()
graph_engine_copy.graph_runtime_state.total_tokens = 0
variable_pool_copy = graph_engine_copy.graph_runtime_state.variable_pool
variable_pool_copy.add([self.node_id, "index"], index)
variable_pool_copy.add([self.node_id, "item"], item)

Loading…
Cancel
Save