feat: Use ModelSettings to set defaults in Model.__init__

This commit is contained in:
Paul Gauthier (aider) 2024-08-01 17:31:13 -03:00
parent 38b411a6cb
commit 32d82b3175

View file

@ -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)