phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-25 18:54:30 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix(utils.py): raise correct error for azure content blocked error

Browse source
This commit is contained in:
Krrish Dholakia 2024-01-10 23:31:32 +05:30
parent b8de5636d4
commit 3080f27b54
3 changed files with 13 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

11
litellm/utils.py
View file

@ -7147,6 +7147,16 @@ class CustomStreamWrapper:
else:
logprobs = None
if (
hasattr(str_line.choices[0], "content_filter_result")
and str_line.choices[0].content_filter_result is not None
):
error_message = json.dumps(
str_line.choices[0].content_filter_result
)
raise litellm.AzureOpenAIError(
status_code=400, message=error_message
)
return {
"text": text,
"is_finished": is_finished,
@ -7695,7 +7705,6 @@ class CustomStreamWrapper:
chunk = self.completion_stream
else:
chunk = next(self.completion_stream)
print_verbose(f"value of chunk: {chunk} ")
if chunk is not None and chunk != b"":
print_verbose(f"PROCESSED CHUNK PRE CHUNK CREATOR: {chunk}")
response = self.chunk_creator(chunk=chunk)