From 75a51025a05a4a648ec9c8ae51dff4b05ed46974 Mon Sep 17 00:00:00 2001 From: "Paul Gauthier (aider)" Date: Mon, 26 Aug 2024 15:42:10 -0700 Subject: [PATCH] style: Improve formatting of cache warming message --- aider/coders/base_coder.py | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/aider/coders/base_coder.py b/aider/coders/base_coder.py index cb30eed81..ad52b35dc 100755 --- a/aider/coders/base_coder.py +++ b/aider/coders/base_coder.py @@ -1013,7 +1013,7 @@ class Coder: cache_hit_tokens = getattr( completion.usage, "prompt_cache_hit_tokens", 0 ) or getattr(completion.usage, "cache_read_input_tokens", 0) - + def format_tokens(count): if count < 1000: return f"{count}" @@ -1021,8 +1021,11 @@ class Coder: return f"{count / 1000:.1f}k" else: return f"{round(count / 1000)}k" - - self.io.tool_output(f"Warmed {format_tokens(cache_hit_tokens)} cached tokens. ({i+1}/{self.num_cache_warming_pings})") + + self.io.tool_output( + f"Warmed {format_tokens(cache_hit_tokens)} cached tokens." + f" ({i+1}/{self.num_cache_warming_pings})" + ) self.io.tool_output("Stopped warming.")