dearwolf/aider
1
0
Fork 0
mirror of https://github.com/Aider-AI/aider.git synced 2025-05-31 17:55:01 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 9

style: Format code according to linter rules

Browse source
This commit is contained in:
Paul Gauthier (aider) 2025-03-21 07:27:57 -07:00
parent 245ee84c2d
commit 649742f02a
1 changed files with 12 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

18
aider/models.py
View file

@ -663,9 +663,9 @@ class Model(ModelSettings):
self.use_temperature = False
if not self.extra_params:
self.extra_params = {}
# OpenRouter models use 'reasoning' instead of 'thinking'
if self.name.startswith('openrouter/'):
if self.name.startswith("openrouter/"):
self.extra_params["reasoning"] = {"max_tokens": num_tokens}
else:
self.extra_params["thinking"] = {"type": "enabled", "budget_tokens": num_tokens}
@ -673,15 +673,21 @@ class Model(ModelSettings):
def get_thinking_tokens(self, model):
"""Get formatted thinking token budget if available"""
budget = None
if model.extra_params:
# Check for OpenRouter reasoning format
if "reasoning" in model.extra_params and "max_tokens" in model.extra_params["reasoning"]:
if (
"reasoning" in model.extra_params
and "max_tokens" in model.extra_params["reasoning"]
):
budget = model.extra_params["reasoning"]["max_tokens"]
# Check for standard thinking format
elif "thinking" in model.extra_params and "budget_tokens" in model.extra_params["thinking"]:
elif (
"thinking" in model.extra_params
and "budget_tokens" in model.extra_params["thinking"]
):
budget = model.extra_params["thinking"]["budget_tokens"]
if budget is not None:
# Format as xx.yK for thousands, xx.yM for millions
if budget >= 1024 * 1024: