diff --git a/litellm/tests/test_completion.py b/litellm/tests/test_completion.py index 42cd625ee..13bc92156 100644 --- a/litellm/tests/test_completion.py +++ b/litellm/tests/test_completion.py @@ -935,6 +935,9 @@ def test_completion_base64(model): print(f"\nResponse: {resp}") prompt_tokens = resp.usage.prompt_tokens + except litellm.ServiceUnavailableError as e: + print("got service unavailable error: ", e) + pass except Exception as e: if "500 Internal error encountered.'" in str(e): pass