浏览代码

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

E Chen 2 月之前
父节点
当前提交
a97d73ab05
共有 1 个文件被更改,包括 1 次插入0 次删除
  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)