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

refactor: preserve original case in model name matching

Browse source
This commit is contained in:
Paul Gauthier (aider) 2025-01-23 11:34:16 -08:00
parent 16c4374f7a
commit a4b7236289
1 changed files with 5 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

11
aider/models.py
View file

@ -1348,9 +1348,8 @@ def fuzzy_match_models(name):
name = name.lower()
chat_models = set()
for model, attrs in litellm.model_cost.items():
# it's fine to lowercase for fuzzy searching, but we need to return the original case version ai!
model = model.lower()
for orig_model, attrs in litellm.model_cost.items():
model = orig_model.lower()
if attrs.get("mode") != "chat":
continue
provider = attrs.get("litellm_provider", "").lower()
@ -1359,12 +1358,12 @@ def fuzzy_match_models(name):
provider += "/"
if model.startswith(provider):
fq_model = model
fq_model = orig_model
else:
fq_model = provider + model
fq_model = provider + orig_model
chat_models.add(fq_model)
chat_models.add(model)
chat_models.add(orig_model)
chat_models = sorted(chat_models)
# exactly matching model