phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-27 19:54:13 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

provider: model_list

Browse source
This commit is contained in:
ishaan-jaff 2023-08-21 11:04:11 -07:00
parent 506c9122cc
commit cf826cb314
1 changed files with 30 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

30
litellm/__init__.py
View file

@ -185,6 +185,18 @@ huggingface_models = [
ai21_models = ["j2-ultra", "j2-mid", "j2-light"] ai21_models = ["j2-ultra", "j2-mid", "j2-light"]
together_ai_models = [
"togethercomputer/llama-2-70b-chat",
"togethercomputer/Llama-2-7B-32K-Instruct",
"togethercomputer/llama-2-7b"
]
baseten_models = [
"qvv0xeq", # FALCON 7B
"q841o8w", # WizardLM
"31dxrj3" # Mosaic ML
]
model_list = ( model_list = (
open_ai_chat_completion_models open_ai_chat_completion_models
+ open_ai_text_completion_models + open_ai_text_completion_models
@ -196,6 +208,8 @@ model_list = (
+ vertex_chat_models + vertex_chat_models
+ vertex_text_models + vertex_text_models
+ ai21_models + ai21_models
+ together_ai_models
+ baseten_models
) )
provider_list = [ provider_list = [
@ -208,7 +222,23 @@ provider_list = [
"openrouter", "openrouter",
"vertex_ai", "vertex_ai",
"ai21", "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,
}
####### EMBEDDING MODELS ################### ####### EMBEDDING MODELS ###################
open_ai_embedding_models = ["text-embedding-ada-002"] open_ai_embedding_models = ["text-embedding-ada-002"]