From 3ed16fb796cee9b784c315b853ecd00a167f8918 Mon Sep 17 00:00:00 2001 From: "Paul Gauthier (aider)" Date: Sat, 8 Mar 2025 17:26:20 -0800 Subject: [PATCH] style: fix whitespace and format code --- aider/main.py | 2 +- aider/models.py | 9 +++------ 2 files changed, 4 insertions(+), 7 deletions(-) diff --git a/aider/main.py b/aider/main.py index feb27586e..ac2601009 100644 --- a/aider/main.py +++ b/aider/main.py @@ -776,7 +776,7 @@ def main(argv=None, input=None, output=None, force_git_root=None, return_coder=F # Set reasoning effort if specified if args.reasoning_effort is not None: main_model.set_reasoning_effort(args.reasoning_effort) - + # Set thinking tokens if specified if args.thinking_tokens is not None: main_model.set_thinking_tokens(args.thinking_tokens) diff --git a/aider/models.py b/aider/models.py index bf0b2bf39..148b2b8f7 100644 --- a/aider/models.py +++ b/aider/models.py @@ -590,18 +590,15 @@ class Model(ModelSettings): if "extra_body" not in self.extra_params: self.extra_params["extra_body"] = {} self.extra_params["extra_body"]["reasoning_effort"] = effort - + def set_thinking_tokens(self, num): """Set the thinking token budget for models that support it""" if num is not None: self.use_temperature = False if not self.extra_params: self.extra_params = {} - self.extra_params["thinking"] = { - "type": "enabled", - "budget_tokens": num - } - + self.extra_params["thinking"] = {"type": "enabled", "budget_tokens": num} + def is_deepseek_r1(self): name = self.name.lower() if "deepseek" not in name: