From 8f9638f2c1fc7a9ec063d9e875c5e3e27b36dc8e Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Mon, 22 Jul 2024 16:03:15 -0700 Subject: [PATCH] fix raise correct provider on content policy violation --- litellm/utils.py | 11 +++++++---- 1 file changed, 7 insertions(+), 4 deletions(-) diff --git a/litellm/utils.py b/litellm/utils.py index 42422fc445a..9d798f1196a 100644 --- a/litellm/utils.py +++ b/litellm/utils.py @@ -8808,11 +8808,14 @@ class CustomStreamWrapper: str_line.choices[0].content_filter_result ) else: - error_message = "Azure Response={}".format( - str(dict(str_line)) + error_message = "{} Response={}".format( + self.custom_llm_provider, str(dict(str_line)) ) - raise litellm.AzureOpenAIError( - status_code=400, message=error_message + + raise litellm.ContentPolicyViolationError( + message=error_message, + llm_provider=self.custom_llm_provider, + model=self.model, ) # checking for logprobs