(feat) add ContentPolicyViolationError for azure

2026-08-11 00:25:15 +00:00 · 2024-01-09 16:58:09 +05:30
parent 170ae74118
commit 186fc4614d
1 changed files with 11 additions and 0 deletions
@@ -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(