cleanup --model-metadata-file

This commit is contained in:
Paul Gauthier 2024-06-11 13:29:05 -07:00
parent 2f6e360188
commit fb3cc98da5
8 changed files with 60 additions and 49 deletions

View file

@ -426,12 +426,13 @@ class Model:
return res
def register_models(model_def_fnames):
model_files_loaded = []
model_metadata_files_loaded = []
for model_def_fname in model_def_fnames:
if not os.path.exists(model_def_fname):
continue
model_files_loaded.append(model_def_fname)
model_metadata_files_loaded.append(model_def_fname)
try:
with open(model_def_fname, "r") as model_def_file:
model_def = json.load(model_def_file)
@ -440,7 +441,8 @@ def register_models(model_def_fnames):
litellm.register_model(model_def)
return model_files_loaded
return model_metadata_files_loaded
def validate_variables(vars):
missing = []