phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-24 10:14:26 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

test: update test
All checks were successful
Read Version from pyproject.toml / read-version (push) Successful in 16s
Details
Helm unit test / unit-test (push) Successful in 25s
Details

Browse source
This commit is contained in:
Krrish Dholakia 2025-04-23 10:51:07 -07:00
parent 47420d8d68
commit 174a1aa007
1 changed files with 6 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

6
tests/local_testing/test_amazing_vertex_completion.py
View file

@ -455,6 +455,7 @@ async def test_async_vertexai_response():
or "gemini-2.0-flash-thinking-exp" in model
or "gemini-2.0-pro-exp-02-05" in model
or "gemini-pro" in model
or "gemini-1.0-pro" in model
):
# our account does not have access to this model
continue
@ -465,6 +466,8 @@ async def test_async_vertexai_response():
model=model, messages=messages, temperature=0.7, timeout=5
)
print(f"response: {response}")
except litellm.NotFoundError as e:
pass
except litellm.RateLimitError as e:
pass
except litellm.Timeout as e:
@ -503,6 +506,7 @@ async def test_async_vertexai_streaming_response():
or "gemini-2.0-flash-thinking-exp" in model
or "gemini-2.0-pro-exp-02-05" in model
or "gemini-pro" in model
or "gemini-1.0-pro" in model
):
# our account does not have access to this model
continue
@ -524,6 +528,8 @@ async def test_async_vertexai_streaming_response():
complete_response += chunk.choices[0].delta.content
print(f"complete_response: {complete_response}")
assert len(complete_response) > 0
except litellm.NotFoundError as e:
pass
except litellm.RateLimitError as e:
pass
except litellm.APIConnectionError: