phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

init vertex_vision_models

Browse source
This commit is contained in:
ishaan-jaff 2023-12-16 18:37:00 +05:30
parent 55e2beeaf1
commit 0bf29a14e8
1 changed files with 3 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

3
litellm/__init__.py
View file

@ -111,6 +111,7 @@ cohere_models: List = []
anthropic_models: List = [] anthropic_models: List = []
openrouter_models: List = [] openrouter_models: List = []
vertex_language_models: List = [] vertex_language_models: List = []
vertex_vision_models: List = []
vertex_chat_models: List = [] vertex_chat_models: List = []
vertex_code_chat_models: List = [] vertex_code_chat_models: List = []
vertex_text_models: List = [] vertex_text_models: List = []
@ -139,6 +140,8 @@ for key, value in model_cost.items():
vertex_code_text_models.append(key) vertex_code_text_models.append(key)
elif value.get('litellm_provider') == 'vertex_ai-language-models': elif value.get('litellm_provider') == 'vertex_ai-language-models':
vertex_language_models.append(key) 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': elif value.get('litellm_provider') == 'vertex_ai-chat-models':
vertex_chat_models.append(key) vertex_chat_models.append(key)
elif value.get('litellm_provider') == 'vertex_ai-code-chat-models': elif value.get('litellm_provider') == 'vertex_ai-code-chat-models':