From da257c33a0cd058ae09369b7c4758d682e684824 Mon Sep 17 00:00:00 2001 From: Paul Gauthier Date: Sun, 28 Jul 2024 15:54:02 -0300 Subject: [PATCH] Add token count calculation for messages in base_coder.py --- aider/coders/base_coder.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/aider/coders/base_coder.py b/aider/coders/base_coder.py index e1783a604..e9cbdc4df 100755 --- a/aider/coders/base_coder.py +++ b/aider/coders/base_coder.py @@ -1236,7 +1236,7 @@ class Coder: def render_incremental_response(self, final): 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 completion_tokens = 0 cost = 0 @@ -1245,6 +1245,7 @@ class Coder: prompt_tokens = completion.usage.prompt_tokens completion_tokens = completion.usage.completion_tokens else: + prompt_tokens = self.main_model.token_count(messages) completion_tokens = self.main_model.token_count(self.partial_response_content) tokens = f"{prompt_tokens} prompt tokens, {completion_tokens} completion tokens"