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}")