dearwolf/aider
1
0
Fork 0
mirror of https://github.com/Aider-AI/aider.git synced 2025-05-31 17:55:01 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 9

Show the correct did-you-mean models #917

Browse source
This commit is contained in:
Paul Gauthier 2024-07-22 17:24:30 +02:00
parent 5edde3aaea
commit b7eaa0fe7c
1 changed files with 5 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

9
aider/models.py
View file

@ -672,7 +672,7 @@ def sanity_check_model(io, model):
if possible_matches: if possible_matches:
io.tool_output("Did you mean one of these?") io.tool_output("Did you mean one of these?")
for match in possible_matches: for match in possible_matches:
io.tool_output(f"- {model}") io.tool_output(f"- {match}")
if show: if show:
io.tool_output(f"For more info, see: {urls.model_warnings}\n") io.tool_output(f"For more info, see: {urls.model_warnings}\n")
@ -708,12 +708,13 @@ def fuzzy_match_models(name):
# Check for model names containing the name # Check for model names containing the name
matching_models = [m for m in chat_models if name in m] matching_models = [m for m in chat_models if name in m]
if matching_models: if matching_models:
return matching_models return sorted(set(matching_models))
# Check for slight misspellings # Check for slight misspellings
models = list(chat_models) models = set(chat_models)
matching_models = difflib.get_close_matches(name, models, n=3, cutoff=0.8) matching_models = difflib.get_close_matches(name, models, n=3, cutoff=0.8)
return sorted(matching_models)
return sorted(set(matching_models))
def print_matching_models(io, search): def print_matching_models(io, search):