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

This commit is contained in:
Krrish Dholakia 2024-01-10 23:31:32 +05:30
parent 43ccfbba66
commit 3ed296e2dd
3 changed files with 13 additions and 3 deletions

View file

@ -93,7 +93,7 @@ def test_stream_chunk_builder_litellm_function_call():
def test_stream_chunk_builder_litellm_tool_call():
try:
litellm.set_verbose = False
litellm.set_verbose = True
response = litellm.completion(
model="azure/gpt-4-nov-release",
messages=messages,
@ -101,6 +101,7 @@ def test_stream_chunk_builder_litellm_tool_call():
stream=True,
api_key="os.environ/AZURE_FRANCE_API_KEY",
api_base="https://openai-france-1234.openai.azure.com",
api_version="2023-12-01-preview",
complete_response=True,
)