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: add self.event() to calculate_and_show_tokens_and_cost

Browse source
This commit is contained in:
Paul Gauthier (aider) 2024-08-12 16:17:11 -07:00
parent 4129065d6c
commit 57ce0dca67
1 changed files with 8 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

8
aider/coders/base_coder.py
View file

@ -1411,6 +1411,14 @@ class Coder:
f" Cost: ${format_cost(cost)} request, ${format_cost(self.total_cost)} session." f" Cost: ${format_cost(cost)} request, ${format_cost(self.total_cost)} session."
) )
self.event("token_usage",
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): def get_multi_response_content(self, final=False):
cur = self.multi_response_content or "" cur = self.multi_response_content or ""
new = self.partial_response_content or "" new = self.partial_response_content or ""