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!
VERBOSE = True
VERBOSE = False
class LazyLiteLLM:
_lazy_module = None
def __getattr__(self, name):
dump(name)
if name == "_lazy_module":
return super()
self._load_litellm()

View file

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