From 21eb00278d59ef97ba0420fe1b5cb3b4bbda1b5e Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Wed, 23 Aug 2023 14:26:10 -0700 Subject: [PATCH] undo provider list formatting --- litellm/__init__.py | 42 +++++++++++++++++++++--------------------- 1 file changed, 21 insertions(+), 21 deletions(-) diff --git a/litellm/__init__.py b/litellm/__init__.py index a82b888312..06c1a9a38e 100644 --- a/litellm/__init__.py +++ b/litellm/__init__.py @@ -229,30 +229,30 @@ model_list = ( ) provider_list = [ - "OpenAI", - "Azure", - "Cohere", - "Anthropic", - "Replicate", - "Huggingface", - "Together_ai", - "Openrouter", - "Vertex_ai", - "AI21", - "Baseten", + "open", + "azure", + "cohere", + "anthropic", + "replicate", + "huggingface", + "together_ai", + "openrouter", + "vertex_ai", + "ai21", + "baseten", ] models_by_provider = { - "OpenAI": open_ai_chat_completion_models + open_ai_text_completion_models, - "Cohere": cohere_models, - "Anthropic": anthropic_models, - "Replicate": replicate_models, - "Huggingface": huggingface_models, - "Together_ai": together_ai_models, - "Baseten": baseten_models, - "Openrouter": openrouter_models, - "Vertex_ai": vertex_chat_models + vertex_text_models, - "AI21": ai21_models, + "openai": open_ai_chat_completion_models + open_ai_text_completion_models, + "cohere": cohere_models, + "anthropic": anthropic_models, + "replicate": replicate_models, + "huggingface": huggingface_models, + "together_ai": together_ai_models, + "baseten": baseten_models, + "openrouter": openrouter_models, + "vertex_ai": vertex_chat_models + vertex_text_models, + "ai21": ai21_models, } ####### EMBEDDING MODELS ###################