diff --git a/litellm/llms/vertex_ai/vertex_embeddings/transformation.py b/litellm/llms/vertex_ai/vertex_embeddings/transformation.py index 0e9c073f8dd..d9e84fca038 100644 --- a/litellm/llms/vertex_ai/vertex_embeddings/transformation.py +++ b/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") diff --git a/tests/local_testing/test_amazing_vertex_completion.py b/tests/local_testing/test_amazing_vertex_completion.py index 8595b54f709..25993d6d5bb 100644 --- a/tests/local_testing/test_amazing_vertex_completion.py +++ b/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()