From 03faad3110ed01171af7aba79c5509efff89b357 Mon Sep 17 00:00:00 2001 From: "Paul Gauthier (aider)" Date: Wed, 2 Oct 2024 11:13:44 -0700 Subject: [PATCH] refactor: improve sanity_check_models to include editor model check --- aider/models.py | 12 ++++++++++-- 1 file changed, 10 insertions(+), 2 deletions(-) diff --git a/aider/models.py b/aider/models.py index b5dd7382c..60cb5d0f4 100644 --- a/aider/models.py +++ b/aider/models.py @@ -915,11 +915,19 @@ def validate_variables(vars): def sanity_check_models(io, main_model): + problem_main = sanity_check_model(io, main_model) + problem_weak = None - problem_strong = sanity_check_model(io, main_model) if main_model.weak_model and main_model.weak_model is not main_model: problem_weak = sanity_check_model(io, main_model.weak_model) - return problem_strong or problem_weak + + problem_editor = None + if (main_model.editor_model and + main_model.editor_model is not main_model and + main_model.editor_model is not main_model.weak_model): + problem_editor = sanity_check_model(io, main_model.editor_model) + + return problem_main or problem_weak or problem_editor def sanity_check_model(io, model):