diff --git a/litellm/tests/test_completion.py b/litellm/tests/test_completion.py index f38c58b6640..d6d0d93ae30 100644 --- a/litellm/tests/test_completion.py +++ b/litellm/tests/test_completion.py @@ -273,6 +273,8 @@ async def test_completion_predibase(): print(response) except litellm.Timeout as e: pass + except litellm.ServiceUnavailableError as e: + pass except Exception as e: pytest.fail(f"Error occurred: {e}")