scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-17 23:51:30 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 67

fix: use correct parameter name 'dimension' for Vertex AI multimodalembedding

Browse source 
The Vertex AI multimodalembedding predict endpoint expects 'dimension'
(not 'outputDimensionality') in the parameters field.
This commit is contained in:
Chesars 2026-03-23 11:06:52 -03:00
parent 7805f38cce
commit 265e096a53
2 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
litellm/llms/vertex_ai/multimodal_embeddings/transformation.py
View file

@ -203,7 +203,7 @@ class VertexAIMultimodalEmbeddingConfig(BaseEmbeddingConfig):
if "outputDimensionality" in optional_params:
request_data["parameters"] = {
"outputDimensionality": optional_params["outputDimensionality"]
"dimension": optional_params["outputDimensionality"]
}
return cast(dict, request_data)

2
tests/test_litellm/llms/vertex_ai/multimodal_embeddings/test_vertex_ai_multimodal_embedding_transformation.py
View file

@ -156,7 +156,7 @@ class TestVertexMultimodalEmbedding:
headers={},
)
assert "parameters" in request, "Request should contain 'parameters' field"
assert request["parameters"]["outputDimensionality"] == 128
assert request["parameters"]["dimension"] == 128
def test_no_parameters_field_when_dimensions_not_set(self):
"""Test that no parameters field is added when dimensions is not set."""