dearwolf/aider
1
0
Fork 0
mirror of https://github.com/Aider-AI/aider.git synced 2025-05-24 14:25:00 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 9

token pricing

Browse source
This commit is contained in:
Paul Gauthier 2024-04-17 15:34:59 -07:00
parent c9bb22d6d5
commit 3662a4680b
3 changed files with 20 additions and 16 deletions
Download patch file Download diff file Expand all files Collapse all files

10
aider/coders/base_coder.py
View file

@ -508,7 +508,7 @@ class Coder:
messages += self.cur_messages
# Add the reminder prompt if we still have room to include it.
if total_tokens < self.main_model.max_context_tokens:
if total_tokens < self.main_model.info.get("max_input_tokens", 0):
messages += reminder_message
return messages
@ -707,10 +707,10 @@ class Coder:
completion_tokens = completion.usage.completion_tokens
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
if self.main_model.completion_price:
cost += completion_tokens * self.main_model.completion_price / 1000
if self.main_model.info.get("input_cost_per_token"):
cost = prompt_tokens * self.main_model.info.get("input_cost_per_token")
if self.main_model.info.get("output_cost_per_token"):
cost += completion_tokens * self.main_model.info.get("output_cost_per_token")
tokens += f", ${cost:.6f} cost"
self.total_cost += cost