scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-09 03:18:44 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 109

test test_vertexai_embedding_embedding_latest

Browse source
This commit is contained in:
Ishaan Jaff 2024-06-12 10:13:48 -07:00
parent aaa4a32d65
commit 8dffc8620c
1 changed files with 5 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
litellm/tests/test_amazing_vertex_completion.py
View file

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