diff --git a/litellm/__init__.py b/litellm/__init__.py index 865a90cc1f3..bb844c302b6 100644 --- a/litellm/__init__.py +++ b/litellm/__init__.py @@ -469,6 +469,7 @@ friendliai_models: List = [] palm_models: List = [] groq_models: List = [] azure_models: List = [] +azure_text_models: List = [] anyscale_models: List = [] cerebras_models: List = [] galadriel_models: List = [] @@ -481,6 +482,8 @@ def add_known_models(): open_ai_chat_completion_models.append(key) elif value.get("litellm_provider") == "text-completion-openai": open_ai_text_completion_models.append(key) + elif value.get("litellm_provider") == "azure_text": + azure_text_models.append(key) elif value.get("litellm_provider") == "cohere": cohere_models.append(key) elif value.get("litellm_provider") == "cohere_chat": @@ -847,6 +850,7 @@ model_list = ( + cerebras_models + galadriel_models + sambanova_models + + azure_text_models ) @@ -893,7 +897,8 @@ models_by_provider: dict = { "friendliai": friendliai_models, "palm": palm_models, "groq": groq_models, - "azure": azure_models, + "azure": azure_models + azure_text_models, + "azure_text": azure_text_models, "anyscale": anyscale_models, "cerebras": cerebras_models, "galadriel": galadriel_models,