From 6d191b48ba330260c2a224926a550e8d6a4e0a1a Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Thu, 8 Aug 2024 19:09:46 -0700 Subject: [PATCH] testing skip internal server errors --- litellm/tests/test_image_generation.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/litellm/tests/test_image_generation.py b/litellm/tests/test_image_generation.py index 56271e37c24..91bb5846507 100644 --- a/litellm/tests/test_image_generation.py +++ b/litellm/tests/test_image_generation.py @@ -229,6 +229,8 @@ async def test_aimage_generation_vertex_ai(sync_mode): pass except litellm.RateLimitError as e: pass + except litellm.InternalServerError as e: + pass except litellm.ContentPolicyViolationError: pass # Azure randomly raises these errors - skip when they occur except Exception as e: