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

feat: Accumulate tokens sent and received in calculate_and_show_tokens_and_cost

Browse source
This commit is contained in:
Paul Gauthier (aider) 2024-08-13 06:42:12 -07:00
parent 26305a975d
commit 935bc2d2a6
1 changed files with 8 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

9
aider/coders/base_coder.py
View file

@ -74,6 +74,8 @@ class Coder:
partial_response_content = "" partial_response_content = ""
commit_before_message = [] commit_before_message = []
message_cost = 0.0 message_cost = 0.0
message_tokens_sent = 0
message_tokens_received = 0
@classmethod @classmethod
def create( def create(
@ -1387,7 +1389,10 @@ 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_report = f"Tokens: {prompt_tokens:,} sent, {completion_tokens:,} received." self.message_tokens_sent += prompt_tokens
self.message_tokens_received += completion_tokens
tokens_report = f"Tokens: {self.message_tokens_sent:,} sent, {self.message_tokens_received:,} received."
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")
@ -1417,6 +1422,8 @@ class Coder:
if self.usage_report: if self.usage_report:
self.io.tool_output(self.usage_report) self.io.tool_output(self.usage_report)
self.message_cost = 0.0 self.message_cost = 0.0
self.message_tokens_sent = 0
self.message_tokens_received = 0
def get_multi_response_content(self, final=False): def get_multi_response_content(self, final=False):
cur = self.multi_response_content or "" cur = self.multi_response_content or ""