From 5e707091ee7840b411a911594cb4a2d63cfcc998 Mon Sep 17 00:00:00 2001
From: Krrish Dholakia <krrishdholakia@gmail.com>
Date: Wed, 21 Aug 2024 17:08:58 -0700
Subject: [PATCH] test(test_amazing_vertex_completion.py): handle vertex api
 instability

---
 litellm/tests/test_amazing_vertex_completion.py | 2 ++
 1 file changed, 2 insertions(+)

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: