fix: iteration total tokens calculate error (#15813)

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

View File

@@ -873,11 +873,12 @@ class GraphEngine:
def create_copy(self):
"""
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.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.total_tokens = 0
return new_instance
def _handle_continue_on_error(