ai-workspace-services/litellm
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki 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
Show Stats 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 = []
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':