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

Merge pull request #5326 from BerriAI/litellm_Add_vertex_multimodal_embedding

Browse source 
[Feat] add vertex multimodal embedding support
This commit is contained in:
Ishaan Jaff 2024-08-21 17:06:43 -07:00 • committed by GitHub
commit dd524a4f50
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
5 changed files with 242 additions and 16 deletions
Download patch file Download diff file Expand all files Collapse all files

2
litellm/utils.py
View file

@ -541,7 +541,7 @@ def function_setup(
call_type == CallTypes.embedding.value
or call_type == CallTypes.aembedding.value
):
messages = args[1] if len(args) > 1 else kwargs["input"]
messages = args[1] if len(args) > 1 else kwargs.get("input", None)
elif (
call_type == CallTypes.image_generation.value
or call_type == CallTypes.aimage_generation.value