dearwolf/aider
1
0
Fork 0
mirror of https://github.com/Aider-AI/aider.git synced 2025-06-04 11:45:00 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 4

feat: Add flexible model lookup in get_model_info

Browse source
This commit is contained in:
Paul Gauthier 2024-08-27 06:08:24 -07:00 • committed by Paul Gauthier (aider)
parent 790ebc35a7
commit ec6e23a852
1 changed files with 16 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

20
aider/models.py
View file

@ -427,6 +427,20 @@ model_info_url = (
) )
def get_model_flexible(model, content):
info = content.get(model, dict())
if info:
return info
pieces = model.split("/")
if len(pieces) == 2:
info = content.get(pieces[1])
if info and info.get("litellm_provider") == pieces[0]:
return info
return dict()
def get_model_info(model): def get_model_info(model):
if not litellm._lazy_module: if not litellm._lazy_module:
cache_dir = Path.home() / ".aider" / "caches" cache_dir = Path.home() / ".aider" / "caches"
@ -441,8 +455,7 @@ def get_model_info(model):
if cache_age < 60 * 60 * 24: if cache_age < 60 * 60 * 24:
try: try:
content = json.loads(cache_file.read_text()) content = json.loads(cache_file.read_text())
info = content.get(model, dict()) return get_model_flexible(model, content)
return info
except Exception as ex: except Exception as ex:
print(str(ex)) print(str(ex))
@ -453,8 +466,7 @@ def get_model_info(model):
if response.status_code == 200: if response.status_code == 200:
content = response.json() content = response.json()
cache_file.write_text(json.dumps(content, indent=4)) cache_file.write_text(json.dumps(content, indent=4))
info = content.get(model, dict()) return get_model_flexible(model, content)
return info
except Exception as ex: except Exception as ex:
print(str(ex)) print(str(ex))