Chào @minhkhacoder! Tôi là Dosu, một bot có thể giúp bạn giải quyết lỗi, trả lời câu hỏi và trở thành một người đóng góp. Tôi ở đây để giúp bạn trong khi đợi người duy trì hệ thống. 🤖

To fix the error in calculating the correct total in the workflow by removing the number of tokens in the loop, you need to modify the section where the total tokens are being incremented. Specifically, you should remove or adjust the part where self.graph_runtime_state.total_tokens is being updated within the loop.

Here is the relevant part of the code that needs to be modified:

if run_result.status == WorkflowNodeExecutionStatus.SUCCEEDED:
    if run_result.metadata and run_result.metadata.get(NodeRunMetadataKey.TOTAL_TOKENS):
        # plus state total_tokens
        self.graph_runtime_state.total_tokens += int(
            run_result.metadata.get(NodeRunMetadataKey.TOTAL_TOKENS)  # type: ignore[arg-type]
        )

To remove the number of tokens in the loop, you can simply comment out or delete the lines that update self.graph_runtime_state.total_tokens:

if run_result.status == WorkflowNodeExecutionStatus.SUCCEEDED:
    if run_result.metadata and run_result.metadata.get(NodeRunMetadataKey.TOTAL_TOKENS):
        # Comment out or remove the following lines to stop updating total_tokens
        # self.graph_runtime_state.total_tokens += int(
        #     run_result.metadata.get(NodeRunMetadataKey.TOTAL_TOKENS)  # type: ignore[arg-type]
        # )

By doing this, the total tokens will no longer be incremented within the loop, which should fix the error in calculating the correct total in the workflow ^[1].

To continue talking to Dosu, mention @dosu.