diff --git a/litellm/tests/test_amazing_vertex_completion.py b/litellm/tests/test_amazing_vertex_completion.py index fe7d1a8c853..e30a9ed362b 100644 --- a/litellm/tests/test_amazing_vertex_completion.py +++ b/litellm/tests/test_amazing_vertex_completion.py @@ -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: