scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-09 22:31:41 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 86

fix test_vertexai_embedding_embedding_latest_input_type

Browse source
This commit is contained in:
Ishaan Jaff 2024-08-30 15:57:44 -07:00
parent 69dea613ca
commit 9f18aa4588
1 changed files with 1 additions and 5 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

@ -2014,9 +2014,7 @@ def test_vertexai_embedding_embedding_latest():
pytest.fail(f"Error occurred: {e}")
# @pytest.mark.skip(
# reason="new test - works locally running into vertex version issues on ci/cd"
# )
@pytest.mark.flaky(retries=3, delay=1)
def test_vertexai_embedding_embedding_latest_input_type():
try:
load_vertex_ai_credentials()
@ -2027,8 +2025,6 @@ def test_vertexai_embedding_embedding_latest_input_type():
input=["hi"],
input_type="RETRIEVAL_QUERY",
)
assert len(response.data[0]["embedding"]) == 1
assert response.usage.prompt_tokens > 0
print(f"response:", response)
except litellm.RateLimitError as e: