From 8b811c610aecdd658f0111486e6274427fd2443d Mon Sep 17 00:00:00 2001 From: "Paul Gauthier (aider)" Date: Fri, 28 Mar 2025 20:12:04 -1000 Subject: [PATCH] refactor: Use get_raw_thinking_tokens in cmd_think_tokens --- aider/commands.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/aider/commands.py b/aider/commands.py index 1aab5b00d..3f4e2fc5b 100644 --- a/aider/commands.py +++ b/aider/commands.py @@ -1500,7 +1500,7 @@ class Commands: if formatted_budget is None: self.io.tool_output("Thinking tokens are not currently set.") else: - budget = model.extra_params["thinking"].get("budget_tokens") + budget = model.get_raw_thinking_tokens() self.io.tool_output( f"Current thinking token budget: {budget:,} tokens ({formatted_budget})." ) @@ -1510,7 +1510,7 @@ class Commands: model.set_thinking_tokens(value) formatted_budget = model.get_thinking_tokens() - budget = model.extra_params["thinking"].get("budget_tokens") + budget = model.get_raw_thinking_tokens() self.io.tool_output(f"Set thinking token budget to {budget:,} tokens ({formatted_budget}).") self.io.tool_output()