From 8082cbed988fee8a0be178aa6608fa3a3455e46e Mon Sep 17 00:00:00 2001 From: Paul Gauthier Date: Wed, 30 Oct 2024 13:21:19 -0700 Subject: [PATCH] refactor: rename get_model_flexible to get_model_from_cached_json_db --- aider/models.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/aider/models.py b/aider/models.py index e3d74dd04..080a15011 100644 --- a/aider/models.py +++ b/aider/models.py @@ -635,6 +635,7 @@ MODEL_SETTINGS = [ class ModelInfoManager: MODEL_INFO_URL = "https://raw.githubusercontent.com/BerriAI/litellm/main/model_prices_and_context_window.json" CACHE_TTL = 60 * 60 * 24 # 24 hours + content = None def __init__(self): self.cache_dir = Path.home() / ".aider" / "caches" @@ -667,7 +668,7 @@ class ModelInfoManager: except Exception as ex: print(str(ex)) - def get_model_flexible(self, model): + def get_model_from_cached_json_db(self, model): if not self.content: self._update_cache() @@ -687,7 +688,7 @@ class ModelInfoManager: return dict() def get_model_info(self, model): - info = self.get_model_flexible(model) + info = self.get_model_from_cached_json_db(model) if info: return info