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

Add usage reporting for prompt and completion tokens, and session cost.

Browse source
This commit is contained in:
Paul Gauthier 2024-07-29 08:24:22 -03:00 • committed by Paul Gauthier (aider)
parent f898d7e4d7
commit effa5236c6
1 changed files with 4 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

8
aider/coders/base_coder.py
View file

@ -1255,15 +1255,15 @@ class Coder:
prompt_tokens = self.main_model.token_count(messages) prompt_tokens = self.main_model.token_count(messages)
completion_tokens = self.main_model.token_count(self.partial_response_content) completion_tokens = self.main_model.token_count(self.partial_response_content)
tokens = f"{prompt_tokens:,} prompt tokens, {completion_tokens:,} completion tokens" self.usage_report = f"Tokens: {prompt_tokens:,} sent, {completion_tokens:,} received."
cost_report = ""
if self.main_model.info.get("input_cost_per_token"): if self.main_model.info.get("input_cost_per_token"):
cost += prompt_tokens * 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"): if self.main_model.info.get("output_cost_per_token"):
cost += completion_tokens * 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 self.total_cost += cost
self.usage_report += f" Cost: ${cost:.6f} request, ${self.total_cost:.6f} session."
self.usage_report = f"{tokens}, session cost: ${self.total_cost:.6f}"
def get_multi_response_content(self, final=False): def get_multi_response_content(self, final=False):
cur = self.multi_response_content cur = self.multi_response_content