From eb0331baed14c2ce80911ffc3b1765feeac137e8 Mon Sep 17 00:00:00 2001 From: "Paul Gauthier (aider)" Date: Fri, 27 Sep 2024 13:01:20 -0700 Subject: [PATCH] feat: wrap max_tokens in extra_params for ModelSettings --- aider/models.py | 20 +++++++++++++++----- 1 file changed, 15 insertions(+), 5 deletions(-) diff --git a/aider/models.py b/aider/models.py index 2825b3497..b5dd7382c 100644 --- a/aider/models.py +++ b/aider/models.py @@ -342,7 +342,9 @@ MODEL_SETTINGS = [ use_repo_map=True, examples_as_sys_msg=True, accepts_images=True, - max_tokens=8192, + extra_params={ + "max_tokens": 8192, + }, reminder="user", ), ModelSettings( @@ -431,7 +433,9 @@ MODEL_SETTINGS = [ use_repo_map=True, examples_as_sys_msg=True, reminder="sys", - max_tokens=8192, + extra_params={ + "max_tokens": 8192, + }, ), ModelSettings( "deepseek/deepseek-coder", @@ -440,7 +444,9 @@ MODEL_SETTINGS = [ examples_as_sys_msg=True, reminder="sys", caches_by_default=True, - max_tokens=8192, + extra_params={ + "max_tokens": 8192, + }, ), ModelSettings( "deepseek-chat", @@ -448,7 +454,9 @@ MODEL_SETTINGS = [ use_repo_map=True, examples_as_sys_msg=True, reminder="sys", - max_tokens=8192, + extra_params={ + "max_tokens": 8192, + }, ), ModelSettings( "deepseek-coder", @@ -457,7 +465,9 @@ MODEL_SETTINGS = [ examples_as_sys_msg=True, reminder="sys", caches_by_default=True, - max_tokens=8192, + extra_params={ + "max_tokens": 8192, + }, ), ModelSettings( "openrouter/deepseek/deepseek-coder",