phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

fix(utils.py): fix azure streaming content filter error chunk

Browse source
This commit is contained in:
Krrish Dholakia 2024-05-01 13:45:21 -07:00
parent 56a75ee7fe
commit 37eb7910d2
1 changed files with 8 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

11
litellm/utils.py
View file

@ -9603,9 +9603,14 @@ class CustomStreamWrapper:
is_finished = True is_finished = True
finish_reason = str_line.choices[0].finish_reason finish_reason = str_line.choices[0].finish_reason
if finish_reason == "content_filter": if finish_reason == "content_filter":
error_message = json.dumps( if hasattr(str_line.choices[0], "content_filter_result"):
str_line.choices[0].content_filter_result error_message = json.dumps(
) str_line.choices[0].content_filter_result
)
else:
error_message = "Azure Response={}".format(
str(dict(str_line))
)
raise litellm.AzureOpenAIError( raise litellm.AzureOpenAIError(
status_code=400, message=error_message status_code=400, message=error_message
) )