diff --git a/aider/models.py b/aider/models.py index cc7ce2220..df82ec454 100644 --- a/aider/models.py +++ b/aider/models.py @@ -602,7 +602,6 @@ class Model(ModelSettings): self.max_chat_history_tokens = 1024 self.weak_model = None self.junior_model = None - self.junior_edit_format = junior_edit_format self.info = self.get_model_info(model) @@ -626,7 +625,7 @@ class Model(ModelSettings): if junior_model is False: self.junior_model_name = None else: - self.get_junior_model(junior_model) + self.get_junior_model(junior_model, junior_edit_format) def get_model_info(self, model): return get_model_info(model) @@ -699,10 +698,12 @@ class Model(ModelSettings): def commit_message_models(self): return [self.weak_model, self] - def get_junior_model(self, provided_junior_model_name): + def get_junior_model(self, provided_junior_model_name, junior_edit_format): # If junior_model_name is provided, override the model settings if provided_junior_model_name: self.junior_model_name = provided_junior_model_name + if junior_edit_format: + self.junior_edit_format = junior_edit_format if not self.junior_model_name: self.junior_model = self @@ -714,14 +715,9 @@ class Model(ModelSettings): self.junior_model = Model( self.junior_model_name, - weak_model=False, junior_model=False, ) - # Use the provided junior_edit_format if available, otherwise use the ModelSettings value - if self.junior_edit_format is None: - self.junior_edit_format = self.junior_edit_format or self.junior_model.edit_format - return self.junior_model def tokenizer(self, text):