Sfoglia il codice sorgente

Fix increase_usage of total_price in agent_runner (#6688)

tags/0.6.16
Pedro Gomes 1 anno fa
parent
commit
20d3e1d297
Nessun account collegato all'indirizzo email del committer
2 ha cambiato i file con 2 aggiunte e 0 eliminazioni
  1. 1
    0
      api/core/agent/cot_agent_runner.py
  2. 1
    0
      api/core/agent/fc_agent_runner.py

+ 1
- 0
api/core/agent/cot_agent_runner.py Vedi File

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


+ 1
- 0
api/core/agent/fc_agent_runner.py Vedi File

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


Loading…
Annulla
Salva