Inicio Explorar Axuda
Iniciar sesión
biheng
/
dify
1
0
Fork 0
Ficheiros Incidencias 0 Pull Requests 0 Wiki
Explorar o código

Fix increase_usage of total_price in agent_runner (#6688)

Pedro Gomes hai 8 meses
pai
79715345ef
achega
20d3e1d297
Modificáronse 2 ficheiros con 2 adicións e 0 borrados
Dividir vista Mostrar estatísticas de Diff
  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
Ver ficheiro 

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

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