tiennm99/litellm
1
0
Fork 0
mirror of https://github.com/tiennm99/litellm.git synced 2026-08-06 22:22:23 +00:00
Code Issues Packages Projects Releases Wiki Activity

(fix) OpenAI img gen endpoints unstable

Browse Source
This commit is contained in:
Ishaan Jaff
2024-03-28 12:42:04 -07:00
parent 2495587e7c
commit 365497e860
1 changed files with 10 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
litellm/tests/test_image_generation.py
+10
View File
@@ -31,6 +31,8 @@ def test_image_generation_openai():
except litellm.ContentPolicyViolationError:
pass # OpenAI randomly raises these errors - skip when they occur
except Exception as e:
if "Connection error" in str(e):
pass
pytest.fail(f"An exception occurred - {str(e)}")
@@ -53,6 +55,8 @@ def test_image_generation_azure():
except Exception as e:
if "Your task failed as a result of our safety system." in str(e):
pass
if "Connection error" in str(e):
pass
else:
pytest.fail(f"An exception occurred - {str(e)}")
@@ -79,6 +83,8 @@ def test_image_generation_azure_dall_e_3():
except Exception as e:
if "Your task failed as a result of our safety system." in str(e):
pass
if "Connection error" in str(e):
pass
else:
pytest.fail(f"An exception occurred - {str(e)}")
@@ -97,6 +103,8 @@ async def test_async_image_generation_openai():
except litellm.ContentPolicyViolationError:
pass # openai randomly raises these errors - skip when they occur
except Exception as e:
if "Connection error" in str(e):
pass
pytest.fail(f"An exception occurred - {str(e)}")
@@ -117,6 +125,8 @@ async def test_async_image_generation_azure():
except Exception as e:
if "Your task failed as a result of our safety system." in str(e):
pass
if "Connection error" in str(e):
pass
else:
pytest.fail(f"An exception occurred - {str(e)}")