Browse Source

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

E Chen 2 months ago
parent
commit
a97d73ab05
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

@@ -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)