mirror of
https://github.com/Aider-AI/aider.git
synced 2025-06-01 10:14:59 +00:00
style: fix whitespace and format code
This commit is contained in:
parent
6f99392eda
commit
3ed16fb796
2 changed files with 4 additions and 7 deletions
|
@ -776,7 +776,7 @@ def main(argv=None, input=None, output=None, force_git_root=None, return_coder=F
|
||||||
# Set reasoning effort if specified
|
# Set reasoning effort if specified
|
||||||
if args.reasoning_effort is not None:
|
if args.reasoning_effort is not None:
|
||||||
main_model.set_reasoning_effort(args.reasoning_effort)
|
main_model.set_reasoning_effort(args.reasoning_effort)
|
||||||
|
|
||||||
# Set thinking tokens if specified
|
# Set thinking tokens if specified
|
||||||
if args.thinking_tokens is not None:
|
if args.thinking_tokens is not None:
|
||||||
main_model.set_thinking_tokens(args.thinking_tokens)
|
main_model.set_thinking_tokens(args.thinking_tokens)
|
||||||
|
|
|
@ -590,18 +590,15 @@ class Model(ModelSettings):
|
||||||
if "extra_body" not in self.extra_params:
|
if "extra_body" not in self.extra_params:
|
||||||
self.extra_params["extra_body"] = {}
|
self.extra_params["extra_body"] = {}
|
||||||
self.extra_params["extra_body"]["reasoning_effort"] = effort
|
self.extra_params["extra_body"]["reasoning_effort"] = effort
|
||||||
|
|
||||||
def set_thinking_tokens(self, num):
|
def set_thinking_tokens(self, num):
|
||||||
"""Set the thinking token budget for models that support it"""
|
"""Set the thinking token budget for models that support it"""
|
||||||
if num is not None:
|
if num is not None:
|
||||||
self.use_temperature = False
|
self.use_temperature = False
|
||||||
if not self.extra_params:
|
if not self.extra_params:
|
||||||
self.extra_params = {}
|
self.extra_params = {}
|
||||||
self.extra_params["thinking"] = {
|
self.extra_params["thinking"] = {"type": "enabled", "budget_tokens": num}
|
||||||
"type": "enabled",
|
|
||||||
"budget_tokens": num
|
|
||||||
}
|
|
||||||
|
|
||||||
def is_deepseek_r1(self):
|
def is_deepseek_r1(self):
|
||||||
name = self.name.lower()
|
name = self.name.lower()
|
||||||
if "deepseek" not in name:
|
if "deepseek" not in name:
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue