Add exception handling when retrieving model info in the Model class constructor.

This commit is contained in:
Paul Gauthier 2024-04-18 09:26:05 -07:00
parent 79170839ee
commit dd000c6147

View file

@ -110,7 +110,11 @@ class Model:
def __init__(self, model):
self.name = model
self.info = litellm.get_model_info(model)
try:
self.info = litellm.get_model_info(model)
except Exception as err:
raise NoModelInfo(str(err))
dump(model, self.info)