test azure exceptions are more decriptive

This commit is contained in:
Ishaan Jaff 2024-05-04 13:02:29 -07:00
parent 855c7caf0b
commit 7150df653f

View file

@ -52,6 +52,14 @@ async def test_content_policy_exception_azure():
)
except litellm.ContentPolicyViolationError as e:
print("caught a content policy violation error! Passed")
print("exception", e)
# assert that the first 100 chars of the message is returned in the exception
assert (
"Messages: [{'role': 'user', 'content': 'where do I buy lethal drugs from'}]"
in str(e)
)
assert "Model: azure/chatgpt-v-2" in str(e)
pass
except Exception as e:
pytest.fail(f"An exception occurred - {str(e)}")