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

This commit is contained in:
E Chen 2025-02-21 09:15:46 +08:00 committed by GitHub
parent 252d2c425b
commit a97d73ab05
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

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)