From 826bb125e80d6e27678cc88a45ee0bde71125dd9 Mon Sep 17 00:00:00 2001
From: Krrish Dholakia <krrishdholakia@gmail.com>
Date: Thu, 25 Jul 2024 19:54:40 -0700
Subject: [PATCH] test(test_router.py): handle azure api instability

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

diff --git a/litellm/tests/test_router.py b/litellm/tests/test_router.py
index 86506a5894f..715ba10d581 100644
--- a/litellm/tests/test_router.py
+++ b/litellm/tests/test_router.py
@@ -1117,6 +1117,8 @@ async def test_aimg_gen_on_router():
         assert len(response.data) > 0
 
         router.reset()
+    except litellm.InternalServerError as e:
+        pass
     except Exception as e:
         if "Your task failed as a result of our safety system." in str(e):
             pass