estel/dify
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

fix: iteration total tokens calculate error (#15813)

Browse Source 
Co-authored-by: 刘江波 <jiangbo721@163.com>
This commit is contained in:
jiangbo721
2025-03-14 17:44:24 +08:00
committed by GitHub
parent 1a6298b6ea
commit 5646442931
2 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
api/core/workflow/graph_engine/graph_engine.py
View File

@@ -873,11 +873,12 @@ class GraphEngine:
def create_copy(self): def create_copy(self):
""" """
create a graph engine copy create a graph engine copy
:return: with a new variable pool instance of graph engine :return: graph engine with a new variable pool and initialized total tokens
""" """
new_instance = copy(self) new_instance = copy(self)
new_instance.graph_runtime_state = copy(self.graph_runtime_state) new_instance.graph_runtime_state = copy(self.graph_runtime_state)
new_instance.graph_runtime_state.variable_pool = deepcopy(self.graph_runtime_state.variable_pool) new_instance.graph_runtime_state.variable_pool = deepcopy(self.graph_runtime_state.variable_pool)
new_instance.graph_runtime_state.total_tokens = 0
return new_instance return new_instance
def _handle_continue_on_error( def _handle_continue_on_error(

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

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