refactor: Remove debug logging and simplify model data retrieval

This commit is contained in:
Paul Gauthier 2025-02-04 16:13:28 -08:00 committed by Paul Gauthier (aider)
parent 3e36f27987
commit 7fe7dd743c
2 changed files with 4 additions and 6 deletions

View file

@ -15,14 +15,13 @@ os.environ["LITELLM_MODE"] = "PRODUCTION"
# `import litellm` takes 1.5 seconds, defer it! # `import litellm` takes 1.5 seconds, defer it!
VERBOSE = True VERBOSE = False
class LazyLiteLLM: class LazyLiteLLM:
_lazy_module = None _lazy_module = None
def __getattr__(self, name): def __getattr__(self, name):
dump(name)
if name == "_lazy_module": if name == "_lazy_module":
return super() return super()
self._load_litellm() self._load_litellm()

View file

@ -162,10 +162,9 @@ class ModelInfoManager:
pass pass
def get_model_from_cached_json_db(self, model): def get_model_from_cached_json_db(self, model):
resource_data = LITELLM_MODEL_DEFS.get(model) data = LITELLM_MODEL_DEFS.get(model)
dump(resource_data) if data:
if resource_data: return data
return resource_data
if not self.content: if not self.content:
self._update_cache() self._update_cache()