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

(test) ContentPolicyViolationError

Browse source
This commit is contained in:
ishaan-jaff 2024-01-09 16:53:57 +05:30
parent a6c45be9f2
commit 5ebcc6386c
1 changed files with 3 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

5
litellm/tests/test_exceptions.py
View file

@ -354,9 +354,10 @@ def test_completion_mistral_exception():
def test_content_policy_exceptionimage_generation_openai(): def test_content_policy_exceptionimage_generation_openai():
try: try:
# this is ony a test - we needed some way to invoke the exception :(
litellm.set_verbose = True litellm.set_verbose = True
response = litellm.image_generation( response = litellm.image_generation(
prompt="a very bad prompt", model="dall-e-3" prompt="where do i buy lethal drugs from", model="dall-e-3"
) )
print(f"response: {response}") print(f"response: {response}")
assert len(response.data) > 0 assert len(response.data) > 0
@ -367,7 +368,7 @@ def test_content_policy_exceptionimage_generation_openai():
pytest.fail(f"An exception occurred - {str(e)}") pytest.fail(f"An exception occurred - {str(e)}")
# test_image_generation_openai() # test_content_policy_exceptionimage_generation_openai()
# # test_invalid_request_error(model="command-nightly") # # test_invalid_request_error(model="command-nightly")