Will
il y a 5 mois
2 fichiers modifiés avec 2 ajouts et 0 suppressions
+ 1
- 0
api/core/agent/cot_agent_runner.py
Voir le fichier
| @@ -80,6 +80,7 @@ class CotAgentRunner(BaseAgentRunner, ABC): | |||
| llm_usage = final_llm_usage_dict["usage"] | |||
| llm_usage.prompt_tokens += usage.prompt_tokens | |||
| llm_usage.completion_tokens += usage.completion_tokens | |||
| llm_usage.total_tokens += usage.total_tokens | |||
| llm_usage.prompt_price += usage.prompt_price | |||
| llm_usage.completion_price += usage.completion_price | |||
| llm_usage.total_price += usage.total_price | |||
+ 1
- 0
api/core/agent/fc_agent_runner.py
Voir le fichier
| @@ -65,6 +65,7 @@ class FunctionCallAgentRunner(BaseAgentRunner): | |||
| llm_usage = final_llm_usage_dict["usage"] | |||
| llm_usage.prompt_tokens += usage.prompt_tokens | |||
| llm_usage.completion_tokens += usage.completion_tokens | |||
| llm_usage.total_tokens += usage.total_tokens | |||
| llm_usage.prompt_price += usage.prompt_price | |||
| llm_usage.completion_price += usage.completion_price | |||
| llm_usage.total_price += usage.total_price | |||
Chargement…