add list of v0 models by provider

This commit is contained in:
Ishaan Jaff 2025-07-18 18:37:48 -07:00
parent 1ff766e738
commit 2880fc9837

View file

@ -501,7 +501,7 @@ deepgram_models: List = []
elevenlabs_models: List = []
dashscope_models: List = []
moonshot_models: List = []
v0_models: List = []
def is_bedrock_pricing_only_model(key: str) -> bool:
"""
@ -680,6 +680,8 @@ def add_known_models():
dashscope_models.append(key)
elif value.get("litellm_provider") == "moonshot":
moonshot_models.append(key)
elif value.get("litellm_provider") == "v0":
v0_models.append(key)
add_known_models()
@ -709,7 +711,6 @@ ollama_models = ["llama2"]
maritalk_models = ["maritalk"]
model_list = (
open_ai_chat_completion_models
+ open_ai_text_completion_models
@ -765,6 +766,7 @@ model_list = (
+ elevenlabs_models
+ dashscope_models
+ moonshot_models
+ v0_models
)
model_list_set = set(model_list)
@ -832,6 +834,7 @@ models_by_provider: dict = {
"elevenlabs": elevenlabs_models,
"dashscope": dashscope_models,
"moonshot": moonshot_models,
"v0": v0_models,
}
# mapping for those models which have larger equivalents