diff --git a/litellm/tests/test_amazing_vertex_completion.py b/litellm/tests/test_amazing_vertex_completion.py
index c7f3aa03fbe..5291f269160 100644
--- a/litellm/tests/test_amazing_vertex_completion.py
+++ b/litellm/tests/test_amazing_vertex_completion.py
@@ -816,10 +816,14 @@ def test_vertexai_embedding_embedding_latest():
         litellm.set_verbose = True
         response = embedding(
             model="vertex_ai/text-embedding-004",
-            input=["hi"],
+            input=[
+                {"text": "good morning from litellm", "task_type": "RETRIEVAL_DOCUMENT"}
+            ],
             dimensions=1,
+            auto_truncate=True,
         )
         assert len(response.data[0]["embedding"]) == 1
+        assert response.usage.prompt_tokens > 0
         print(f"response:", response)
     except litellm.RateLimitError as e:
         pass