dearwolf/aider
1
0
Fork 0
mirror of https://github.com/Aider-AI/aider.git synced 2025-05-29 00:35:00 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 3

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:
io.tool_output("Did you mean one of these?")
for match in possible_matches:
io.tool_output(f"- {model}")
io.tool_output(f"- {match}")
if show:
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
matching_models = [m for m in chat_models if name in m]
if matching_models:
return matching_models
return sorted(set(matching_models))
# 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)
return sorted(matching_models)
return sorted(set(matching_models))
def print_matching_models(io, search):