From 93fe9007f0670102938e98264c02a553bc34ce87 Mon Sep 17 00:00:00 2001 From: "Paul Gauthier (aider)" Date: Tue, 20 Aug 2024 09:36:04 -0700 Subject: [PATCH] feat: Update `calculate_and_show_tokens_and_cost` to format token counts --- aider/coders/base_coder.py | 16 ++++++++++++---- 1 file changed, 12 insertions(+), 4 deletions(-) diff --git a/aider/coders/base_coder.py b/aider/coders/base_coder.py index 8663b2718..50e21af89 100755 --- a/aider/coders/base_coder.py +++ b/aider/coders/base_coder.py @@ -1502,13 +1502,21 @@ class Coder: self.message_tokens_received += completion_tokens - tokens_report = f"Tokens: {self.message_tokens_sent:,} sent" + def format_tokens(count): + if count < 1000: + return f"{count}" + elif count < 10000: + return f"{count/1000:.1f}k" + else: + return f"{count//1000}k" + + tokens_report = f"Tokens: {format_tokens(self.message_tokens_sent)} sent" if cache_write_tokens: - tokens_report += f", {cache_write_tokens:,} cache write" + tokens_report += f", {format_tokens(cache_write_tokens)} cache write" if cache_hit_tokens: - tokens_report += f", {cache_hit_tokens:,} cache hit" - tokens_report += f", {self.message_tokens_received:,} received." + tokens_report += f", {format_tokens(cache_hit_tokens)} cache hit" + tokens_report += f", {format_tokens(self.message_tokens_received)} received." if not self.main_model.info.get("input_cost_per_token"): self.usage_report = tokens_report