mirror of
https://github.com/Aider-AI/aider.git
synced 2025-05-30 17:24:59 +00:00
refactor: improve sanity_check_models to include editor model check
This commit is contained in:
parent
992600fcb7
commit
03faad3110
1 changed files with 10 additions and 2 deletions
|
@ -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):
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue