be careful logging main_models that are not in the model db

This commit is contained in:
Paul Gauthier 2024-08-12 18:20:35 -07:00
parent a7a626423c
commit aa840f0e28
3 changed files with 16 additions and 4 deletions

View file

@ -1412,13 +1412,13 @@ class Coder:
)
self.event(
"token_usage",
"message_send",
main_model=self.main_model,
prompt_tokens=prompt_tokens,
completion_tokens=completion_tokens,
total_tokens=prompt_tokens + completion_tokens,
cost=cost,
total_cost=self.total_cost,
model=self.main_model.name,
)
def get_multi_response_content(self, final=False):