fix: VertexAI outputDimensionality configuration

VertexAI's API documentation [1] is an absolute mess. In it, they
describe the parameter to configure output dimensionality as
`output_dimensionality`. In the API example, they switch to using snake
case `outputDimensionality`, which is the correct variant.

[1]: https://cloud.google.com/vertex-ai/generative-ai/docs/model-reference/text-embeddings-api#generative-ai-get-text-embedding-drest
This commit is contained in:
James Guthrie 2025-03-19 10:47:09 +01:00
parent 01c6cbd270
commit 437dbe7246
2 changed files with 1 additions and 4 deletions

View file

@ -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")

View file

@ -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()