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

Add token count calculation for messages in base_coder.py

Browse source
This commit is contained in:
Paul Gauthier 2024-07-28 15:54:02 -03:00 • committed by Paul Gauthier (aider)
parent 57964f1e79
commit da257c33a0
1 changed files with 2 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

3
aider/coders/base_coder.py
View file

@ -1236,7 +1236,7 @@ class Coder:
def render_incremental_response(self, final): def render_incremental_response(self, final):
return self.get_multi_response_content() return self.get_multi_response_content()
def calculate_and_show_tokens_and_cost(self, completion=None): def calculate_and_show_tokens_and_cost(self, messages, completion=None):
prompt_tokens = 0 prompt_tokens = 0
completion_tokens = 0 completion_tokens = 0
cost = 0 cost = 0
@ -1245,6 +1245,7 @@ class Coder:
prompt_tokens = completion.usage.prompt_tokens prompt_tokens = completion.usage.prompt_tokens
completion_tokens = completion.usage.completion_tokens completion_tokens = completion.usage.completion_tokens
else: else:
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" tokens = f"{prompt_tokens} prompt tokens, {completion_tokens} completion tokens"