From 98b3722a028a21d18918adc8fa7285991da9a9ab Mon Sep 17 00:00:00 2001 From: Paul Gauthier Date: Tue, 11 Mar 2025 11:33:45 -0700 Subject: [PATCH] refactor: Simplify thinking token budget display logic --- aider/commands.py | 8 ++------ 1 file changed, 2 insertions(+), 6 deletions(-) diff --git a/aider/commands.py b/aider/commands.py index 1232bf057..56295c02c 100644 --- a/aider/commands.py +++ b/aider/commands.py @@ -1421,12 +1421,8 @@ class Commands: model = self.coder.main_model model.set_thinking_tokens(value) - # Try to display the actual value that was set - if model.extra_params and "thinking" in model.extra_params: - budget = model.extra_params["thinking"].get("budget_tokens") - self.io.tool_output(f"Set thinking token budget to {budget:,} tokens.") - else: - self.io.tool_output(f"Set thinking token budget to {value}.") + budget = model.extra_params["thinking"].get("budget_tokens") + self.io.tool_output(f"Set thinking token budget to {budget:,} tokens.") def cmd_copy_context(self, args=None): """Copy the current chat context as markdown, suitable to paste into a web UI"""