From 23cccba070a7b9033b9a52abec20007ba9ee5ab2 Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Sat, 13 Jul 2024 09:52:49 -0700 Subject: [PATCH] fix str from BadRequestError --- litellm/utils.py | 10 ++++++++-- 1 file changed, 8 insertions(+), 2 deletions(-) diff --git a/litellm/utils.py b/litellm/utils.py index 03ed52b5c0..adab3c81af 100644 --- a/litellm/utils.py +++ b/litellm/utils.py @@ -5961,6 +5961,12 @@ def exception_type( extra_kwargs={}, ): global user_logger_fn, liteDebuggerClient + + if any( + isinstance(original_exception, exc_type) + for exc_type in litellm.LITELLM_EXCEPTION_TYPES + ): + return original_exception exception_mapping_worked = False if litellm.suppress_debug_info is False: print() # noqa @@ -6131,7 +6137,7 @@ def exception_type( ): exception_mapping_worked = True raise BadRequestError( - message=f"BadRequestError: {exception_provider} - {message}", + message=f"{exception_provider} - {message}", llm_provider=custom_llm_provider, model=model, response=original_exception.response, @@ -6218,7 +6224,7 @@ def exception_type( elif original_exception.status_code == 422: exception_mapping_worked = True raise BadRequestError( - message=f"BadRequestError: {exception_provider} - {message}", + message=f"{exception_provider} - {message}", model=model, llm_provider=custom_llm_provider, response=original_exception.response,