From 61aaeff4134e6aa8b0b3292d1e1b12d6c01cdaa4 Mon Sep 17 00:00:00 2001 From: waltcow Date: Wed, 3 Jan 2024 17:44:41 +0800 Subject: [PATCH] Fix variable name in AgentApplicationRunner (#1884) --- api/core/app_runner/agent_app_runner.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/api/core/app_runner/agent_app_runner.py b/api/core/app_runner/agent_app_runner.py index 02bdae900d..344a19d1b7 100644 --- a/api/core/app_runner/agent_app_runner.py +++ b/api/core/app_runner/agent_app_runner.py @@ -237,8 +237,8 @@ class AgentApplicationRunner(AppRunner): all_message_tokens = 0 all_answer_tokens = 0 for agent_thought in agent_thoughts: - all_message_tokens += agent_thought.message_tokens - all_answer_tokens += agent_thought.answer_tokens + all_message_tokens += agent_thought.message_token + all_answer_tokens += agent_thought.answer_token model_type_instance = model_config.provider_model_bundle.model_type_instance model_type_instance = cast(LargeLanguageModel, model_type_instance)