From b9f50d83c40de52e7d205ecf17b386da5a5d8ce0 Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Wed, 12 Jun 2024 09:32:32 -0700 Subject: [PATCH] test - support vertex ai ertexai_embedding_embedding --- litellm/tests/test_amazing_vertex_completion.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/litellm/tests/test_amazing_vertex_completion.py b/litellm/tests/test_amazing_vertex_completion.py index 84d3a2bfc65..c7f3aa03fbe 100644 --- a/litellm/tests/test_amazing_vertex_completion.py +++ b/litellm/tests/test_amazing_vertex_completion.py @@ -816,8 +816,10 @@ def test_vertexai_embedding_embedding_latest(): litellm.set_verbose = True response = embedding( model="vertex_ai/text-embedding-004", - input=["good morning from litellm", "this is another item"], + input=["hi"], + dimensions=1, ) + assert len(response.data[0]["embedding"]) == 1 print(f"response:", response) except litellm.RateLimitError as e: pass