From 57964f1e797de928e74a04c0195b838d57da7cf0 Mon Sep 17 00:00:00 2001 From: Paul Gauthier Date: Sun, 28 Jul 2024 15:53:34 -0300 Subject: [PATCH] Add token and cost calculation to message completion --- aider/coders/base_coder.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/aider/coders/base_coder.py b/aider/coders/base_coder.py index 90e9fe3b1..e1783a604 100755 --- a/aider/coders/base_coder.py +++ b/aider/coders/base_coder.py @@ -1137,6 +1137,8 @@ class Coder: if interrupted: raise KeyboardInterrupt + self.calculate_and_show_tokens_and_cost(messages, completion) + def show_send_output(self, completion): if self.verbose: print(completion) @@ -1169,8 +1171,6 @@ class Coder: self.io.tool_error(show_content_err) raise Exception("No data found in LLM response!") - self.calculate_and_show_tokens_and_cost(completion) - show_resp = self.render_incremental_response(True) if self.show_pretty(): show_resp = Markdown( @@ -1228,7 +1228,6 @@ class Coder: sys.stdout.flush() yield text - self.calculate_and_show_tokens_and_cost() def live_incremental_response(self, final): show_resp = self.render_incremental_response(final)