dearwolf/aider
1
0
Fork 0
mirror of https://github.com/Aider-AI/aider.git synced 2025-06-02 02:34:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 5

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