From 6d408dcce71b755e7bcd84eb699c7017c549394d Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Thu, 28 Mar 2024 12:27:26 -0700 Subject: [PATCH] (fix) test aimg gen on router --- litellm/tests/test_router.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/litellm/tests/test_router.py b/litellm/tests/test_router.py index 208fad983..16a3c2475 100644 --- a/litellm/tests/test_router.py +++ b/litellm/tests/test_router.py @@ -590,6 +590,8 @@ async def test_aimg_gen_on_router(): pass elif "Operation polling timed out" in str(e): pass + elif "Connection error" in str(e): + pass else: traceback.print_exc() pytest.fail(f"Error occurred: {e}") @@ -671,6 +673,8 @@ def test_aembedding_on_router(): pass elif "Operation polling timed out" in str(e): pass + elif "Connection error" in str(e): + pass else: traceback.print_exc() pytest.fail(f"Error occurred: {e}")