From e1365fdc2d52cd01b1e69f3892c62b5f48753cc1 Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Sat, 14 Sep 2024 15:47:46 -0700 Subject: [PATCH] add mock InternalServerError --- litellm/main.py | 9 +++++++++ 1 file changed, 9 insertions(+) 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" ):