dearwolf/aider
1
0
Fork 0
mirror of https://github.com/Aider-AI/aider.git synced 2025-05-30 17:24:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 3

refactor: Separate raw thinking token retrieval and fix self access

Browse source
This commit is contained in:
Paul Gauthier 2025-03-28 20:09:01 -10:00 • committed by Paul Gauthier (aider)
parent a4c9c10029
commit 96aa77288b
1 changed files with 7 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

9
aider/models.py
View file

@ -688,11 +688,11 @@ class Model(ModelSettings):
else:
self.extra_params["thinking"] = {"type": "enabled", "budget_tokens": num_tokens}
def get_thinking_tokens(self, model):
def get_raw_thinking_tokens(self):
"""Get formatted thinking token budget if available"""
budget = None
if model.extra_params:
if self.extra_params:
# Check for OpenRouter reasoning format
if (
"reasoning" in model.extra_params
@ -706,6 +706,11 @@ class Model(ModelSettings):
):
budget = model.extra_params["thinking"]["budget_tokens"]
return budget
def get_thinking_tokens(self):
budget = self.get_raw_thinking_tokens()
if budget is not None:
# Format as xx.yK for thousands, xx.yM for millions
if budget >= 1024 * 1024: