From 2880fc98375ceb242d70786009265e88ed3547f1 Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Fri, 18 Jul 2025 18:37:48 -0700 Subject: [PATCH] add list of v0 models by provider --- litellm/__init__.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/litellm/__init__.py b/litellm/__init__.py index 424e6c31985..386d10e4e71 100644 --- a/litellm/__init__.py +++ b/litellm/__init__.py @@ -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