From c38cdef2207080aadbbbff10171b7d77a2032e8f Mon Sep 17 00:00:00 2001 From: "Paul Gauthier (aider)" Date: Tue, 11 Mar 2025 11:35:31 -0700 Subject: [PATCH] style: Format code with linter and improve readability --- aider/coders/base_coder.py | 12 +++++++----- 1 file changed, 7 insertions(+), 5 deletions(-) diff --git a/aider/coders/base_coder.py b/aider/coders/base_coder.py index ddd03db18..806b095f1 100755 --- a/aider/coders/base_coder.py +++ b/aider/coders/base_coder.py @@ -211,11 +211,13 @@ class Coder: output += ", prompt cache" if main_model.info.get("supports_assistant_prefill"): output += ", infinite output" - + # Check for thinking token budget - if (main_model.extra_params and - "thinking" in main_model.extra_params and - "budget_tokens" in main_model.extra_params["thinking"]): + if ( + main_model.extra_params + and "thinking" in main_model.extra_params + and "budget_tokens" in main_model.extra_params["thinking"] + ): budget = main_model.extra_params["thinking"]["budget_tokens"] # Format as xx.yK for thousands, xx.yM for millions if budget >= 1000000: @@ -223,7 +225,7 @@ class Coder: else: formatted_budget = f"{budget/1000:.1f}K" output += f", {formatted_budget} think tokens" - + lines.append(output) if self.edit_format == "architect":