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

refactor: Improve model info retrieval logic in ModelInfoManager

Browse source
This commit is contained in:
Paul Gauthier 2024-11-12 18:37:33 -08:00 • committed by Paul Gauthier (aider)
parent 2e97fcc47f
commit 1f39c7ef0d
1 changed files with 13 additions and 11 deletions
Download patch file Download diff file Expand all files Collapse all files

24
aider/models.py
View file

@ -775,18 +775,20 @@ class ModelInfoManager:
return dict()
def get_model_info(self, model):
if not litellm._lazy_module:
info = self.get_model_from_cached_json_db(model)
if info:
return info
cached_info = self.get_model_from_cached_json_db(model)
# If all else fails, do it the slow way...
try:
return litellm.get_model_info(model)
except Exception as ex:
if "model_prices_and_context_window.json" not in str(ex):
print(str(ex))
return dict()
litellm_info = None
if litellm._lazy_module or not cached_info:
try:
litellm_info = litellm.get_model_info(model)
except Exception as ex:
if "model_prices_and_context_window.json" not in str(ex):
print(str(ex))
if litellm_info:
return litellm_info
return cached_info
model_info_manager = ModelInfoManager()