From 32d82b3175f9131d2044c74c9a39079e7532d128 Mon Sep 17 00:00:00 2001 From: "Paul Gauthier (aider)" Date: Thu, 1 Aug 2024 17:31:13 -0300 Subject: [PATCH] feat: Use ModelSettings to set defaults in Model.__init__ --- aider/models.py | 22 +++++++--------------- 1 file changed, 7 insertions(+), 15 deletions(-) diff --git a/aider/models.py b/aider/models.py index 9077c1eac..46d8e4a4a 100644 --- a/aider/models.py +++ b/aider/models.py @@ -369,23 +369,15 @@ MODEL_SETTINGS = [ class Model: - name = None - - edit_format = "whole" - use_repo_map = False - send_undo_reply = False - accepts_images = False - weak_model_name = None - lazy = False - reminder_as_sys_msg = False - examples_as_sys_msg = False - can_prefill = False - - max_chat_history_tokens = 1024 - weak_model = None - def __init__(self, model, weak_model=None): + # Set defaults from ModelSettings + default_settings = ModelSettings(name="") + for field in fields(ModelSettings): + setattr(self, field.name, getattr(default_settings, field.name)) + self.name = model + self.max_chat_history_tokens = 1024 + self.weak_model = None self.info = self.get_model_info(model)