diff --git a/litellm/__init__.py b/litellm/__init__.py index ef7bfe91d68..68de303b637 100644 --- a/litellm/__init__.py +++ b/litellm/__init__.py @@ -198,6 +198,10 @@ bedrock_models: List = [ "ai21.j2-grande-instruct" ] +ollama_models = [ + "llama2" +] + model_list = ( open_ai_chat_completion_models + open_ai_text_completion_models @@ -213,6 +217,7 @@ model_list = ( + baseten_models + aleph_alpha_models + nlp_cloud_models + + ollama_models ) provider_list: List = [ @@ -234,6 +239,7 @@ provider_list: List = [ "bedrock", "petals", "oobabooga", + "ollama", "custom", # custom apis ] @@ -250,6 +256,7 @@ models_by_provider: dict = { "ai21": ai21_models, "bedrock": bedrock_models, "petals": petals_models, + "ollama": ollama_models, } ####### EMBEDDING MODELS ###################