Fix increase_usage of total_price in agent_runner (#6688)

2 years ago · 20d3e1d297
parent 79715345ef
commit 20d3e1d297
2 changed files with 2 additions and 0 deletions
--- 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

--- 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