diff --git a/litellm/main.py b/litellm/main.py index a50c908c673..1826fcb9ba9 100644 --- a/litellm/main.py +++ b/litellm/main.py @@ -524,6 +524,15 @@ def mock_completion( llm_provider=getattr(mock_response, "llm_provider", custom_llm_provider or "openai"), # type: ignore model=model, ) + elif ( + isinstance(mock_response, str) + and mock_response == "litellm.InternalServerError" + ): + raise litellm.InternalServerError( + message="this is a mock internal server error", + llm_provider=getattr(mock_response, "llm_provider", custom_llm_provider or "openai"), # type: ignore + model=model, + ) elif isinstance(mock_response, str) and mock_response.startswith( "Exception: content_filter_policy" ):