diff --git a/api/core/agent/cot_agent_runner.py b/api/core/agent/cot_agent_runner.py index 9bd8f37d85..06492bb12f 100644 --- a/api/core/agent/cot_agent_runner.py +++ b/api/core/agent/cot_agent_runner.py @@ -79,6 +79,7 @@ class CotAgentRunner(BaseAgentRunner, ABC): llm_usage.completion_tokens += usage.completion_tokens llm_usage.prompt_price += usage.prompt_price llm_usage.completion_price += usage.completion_price + llm_usage.total_price += usage.total_price model_instance = self.model_instance diff --git a/api/core/agent/fc_agent_runner.py b/api/core/agent/fc_agent_runner.py index 7019b5e39f..3ee6e47742 100644 --- a/api/core/agent/fc_agent_runner.py +++ b/api/core/agent/fc_agent_runner.py @@ -62,6 +62,7 @@ class FunctionCallAgentRunner(BaseAgentRunner): llm_usage.completion_tokens += usage.completion_tokens llm_usage.prompt_price += usage.prompt_price llm_usage.completion_price += usage.completion_price + llm_usage.total_price += usage.total_price model_instance = self.model_instance