Fixed bug with cost computations when --no-steam in effect #475

This commit is contained in:
Paul Gauthier 2024-02-11 05:46:27 -08:00
parent a1a2cf7ac2
commit 72e6cb901c
2 changed files with 3 additions and 2 deletions

View file

@ -706,7 +706,8 @@ class Coder:
tokens = f"{prompt_tokens} prompt tokens, {completion_tokens} completion tokens"
if self.main_model.prompt_price:
cost = prompt_tokens * self.main_model.prompt_price / 1000
cost += completion_tokens * self.main_model.completion_price / 1000
if self.main_model.completion_price:
cost += completion_tokens * self.main_model.completion_price / 1000
tokens += f", ${cost:.6f} cost"
self.total_cost += cost