litellm map Azure GPT ContentPolicyViolationError

2024-05-04 11:14:47 -07:00 · 2024-05-04 11:14:47 -07:00 · c6beaf53af
commit c6beaf53af
parent 59dac1bc7a
1 changed files with 3 additions and 0 deletions
--- a/litellm/utils.py
+++ b/litellm/utils.py
@ -9077,6 +9077,9 @@ def exception_type(
                elif (
                    "invalid_request_error" in error_str
                    and "content_policy_violation" in error_str
+                ) or (
+                    "The response was filtered due to the prompt triggering Azure OpenAI's content management"
+                    in error_str
                ):
                    exception_mapping_worked = True
                    raise ContentPolicyViolationError(