From 186fc4614de3bb4fba7ab8a3b2b0aea0cf3c9210 Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Tue, 9 Jan 2024 16:58:09 +0530 Subject: [PATCH] (feat) add ContentPolicyViolationError for azure --- litellm/utils.py | 11 +++++++++++ 1 file changed, 11 insertions(+) diff --git a/litellm/utils.py b/litellm/utils.py index ecbaa9e56f..1cf1ae9f88 100644 --- a/litellm/utils.py +++ b/litellm/utils.py @@ -6509,6 +6509,17 @@ def exception_type( model=model, response=original_exception.response, ) + elif ( + "invalid_request_error" in error_str + and "content_policy_violation" in error_str + ): + exception_mapping_worked = True + raise ContentPolicyViolationError( + message=f"AzureException - {original_exception.message}", + llm_provider="azure", + model=model, + response=original_exception.response, + ) elif "invalid_request_error" in error_str: exception_mapping_worked = True raise BadRequestError(