diff --git a/tests/local_testing/test_completion.py b/tests/local_testing/test_completion.py index a0e961070..5b9a534fa 100644 --- a/tests/local_testing/test_completion.py +++ b/tests/local_testing/test_completion.py @@ -917,6 +917,9 @@ def test_completion_base64(model): except litellm.ServiceUnavailableError as e: print("got service unavailable error: ", e) pass + except litellm.InternalServerError as e: + print("got internal server error: ", e) + pass except Exception as e: if "500 Internal error encountered.'" in str(e): pass