fix(proxy_server.py): ensure /models returns unique model names

This commit is contained in:
Krrish Dholakia 2023-11-28 17:32:20 -08:00
parent 686cebeb7a
commit 6162d32b5a

View file

@ -634,8 +634,7 @@ def model_list():
if general_settings.get("infer_model_from_keys", False): if general_settings.get("infer_model_from_keys", False):
all_models = litellm.utils.get_valid_models() all_models = litellm.utils.get_valid_models()
if llm_model_list: if llm_model_list:
print(f"llm model list: {llm_model_list}") all_models = list(set(all_models + [m["model_name"] for m in llm_model_list]))
all_models += [m["model_name"] for m in llm_model_list]
if user_model is not None: if user_model is not None:
all_models += [user_model] all_models += [user_model]
print(f"all_models: {all_models}") print(f"all_models: {all_models}")