From 0bf29a14e89a0cda9facfc5203d01490ae9d3d22 Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Sat, 16 Dec 2023 18:37:00 +0530 Subject: [PATCH] init vertex_vision_models --- litellm/__init__.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/litellm/__init__.py b/litellm/__init__.py index 67e5274a6..36ce31cfa 100644 --- a/litellm/__init__.py +++ b/litellm/__init__.py @@ -111,6 +111,7 @@ cohere_models: List = [] anthropic_models: List = [] openrouter_models: List = [] vertex_language_models: List = [] +vertex_vision_models: List = [] vertex_chat_models: List = [] vertex_code_chat_models: List = [] vertex_text_models: List = [] @@ -139,6 +140,8 @@ for key, value in model_cost.items(): vertex_code_text_models.append(key) elif value.get('litellm_provider') == 'vertex_ai-language-models': vertex_language_models.append(key) + elif value.get('litellm_provider') == 'vertex_ai-vision-models': + vertex_vision_models.append(key) elif value.get('litellm_provider') == 'vertex_ai-chat-models': vertex_chat_models.append(key) elif value.get('litellm_provider') == 'vertex_ai-code-chat-models':