diff --git a/litellm/llms/vertex_ai/multimodal_embeddings/transformation.py b/litellm/llms/vertex_ai/multimodal_embeddings/transformation.py index d5cfca96812..f1d121099f9 100644 --- a/litellm/llms/vertex_ai/multimodal_embeddings/transformation.py +++ b/litellm/llms/vertex_ai/multimodal_embeddings/transformation.py @@ -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) diff --git a/tests/test_litellm/llms/vertex_ai/multimodal_embeddings/test_vertex_ai_multimodal_embedding_transformation.py b/tests/test_litellm/llms/vertex_ai/multimodal_embeddings/test_vertex_ai_multimodal_embedding_transformation.py index e060e2a7e1a..19024104fe7 100644 --- a/tests/test_litellm/llms/vertex_ai/multimodal_embeddings/test_vertex_ai_multimodal_embedding_transformation.py +++ b/tests/test_litellm/llms/vertex_ai/multimodal_embeddings/test_vertex_ai_multimodal_embedding_transformation.py @@ -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."""