Merge pull request #9366 from JamesGuthrie/jg/vertex-output-dimensionality

fix: VertexAI outputDimensionality configuration

litellm/llms/vertex_ai/vertex_embeddings/transformation.py

@@ -79,7 +79,7 @@ class VertexAITextEmbeddingConfig(BaseModel):
     ):
         for param, value in non_default_params.items():
             if param == "dimensions":
-                optional_params["output_dimensionality"] = value
+                optional_params["outputDimensionality"] = value
 
         if "input_type" in kwargs:
             optional_params["task_type"] = kwargs.pop("input_type")

tests/local_testing/test_amazing_vertex_completion.py

@@ -2174,9 +2174,6 @@ async def test_vertexai_multimodal_embedding_base64image_in_input():
         print("Response:", response)
 
 
-@pytest.mark.skip(
-    reason="new test - works locally running into vertex version issues on ci/cd"
-)
 def test_vertexai_embedding_embedding_latest():
     try:
         load_vertex_ai_credentials()