From 1c736161c5b3fe40a544684cdbd3f72edfae4d2a Mon Sep 17 00:00:00 2001 From: "Paul Gauthier (aider)" Date: Tue, 11 Mar 2025 11:37:01 -0700 Subject: [PATCH] feat: Omit decimal point for whole number token budget values --- aider/coders/base_coder.py | 12 ++++++++++-- 1 file changed, 10 insertions(+), 2 deletions(-) diff --git a/aider/coders/base_coder.py b/aider/coders/base_coder.py index cbfda6b70..3eae23c37 100755 --- a/aider/coders/base_coder.py +++ b/aider/coders/base_coder.py @@ -221,9 +221,17 @@ class Coder: budget = main_model.extra_params["thinking"]["budget_tokens"] # Format as xx.yK for thousands, xx.yM for millions if budget >= 1000000: - formatted_budget = f"{budget / 1000000:.1f}M" + value = budget / 1000000 + if value == int(value): + formatted_budget = f"{int(value)}M" + else: + formatted_budget = f"{value:.1f}M" else: - formatted_budget = f"{budget / 1000:.1f}K" + value = budget / 1000 + if value == int(value): + formatted_budget = f"{int(value)}K" + else: + formatted_budget = f"{value:.1f}K" output += f", {formatted_budget} think tokens" lines.append(output)