test(test_amazing_vertex_completion.py): handle vertex api instability

This commit is contained in:
Krrish Dholakia 2024-08-21 17:08:58 -07:00
parent dd524a4f50
commit 5e707091ee

View file

@ -501,6 +501,8 @@ async def test_async_vertexai_streaming_response():
assert len(complete_response) > 0
except litellm.RateLimitError as e:
pass
except litellm.APIConnectionError:
pass
except litellm.Timeout as e:
pass
except litellm.InternalServerError as e: