From 1765fe2a299437733aa1f47ee8b5b5013891cb44 Mon Sep 17 00:00:00 2001 From: Novice <857526207@qq.com> Date: Wed, 11 Dec 2024 14:23:01 +0800 Subject: [PATCH] fix: iteration node in parallel mode token count error (#11539) Co-authored-by: Novice Lee --- api/core/workflow/nodes/iteration/iteration_node.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/api/core/workflow/nodes/iteration/iteration_node.py b/api/core/workflow/nodes/iteration/iteration_node.py index 5b3853c5d6..cbabe7a3c5 100644 --- a/api/core/workflow/nodes/iteration/iteration_node.py +++ b/api/core/workflow/nodes/iteration/iteration_node.py @@ -182,7 +182,6 @@ class IterationNode(BaseNode[IterationNodeData]): future.add_done_callback(thread_pool.task_done_callback) futures.append(future) succeeded_count = 0 - empty_count = 0 while True: try: event = q.get(timeout=1) @@ -593,3 +592,4 @@ class IterationNode(BaseNode[IterationNodeData]): parallel_mode_run_id=parallel_mode_run_id, ): q.put(event) + graph_engine.graph_runtime_state.total_tokens += graph_engine_copy.graph_runtime_state.total_tokens