tiennm99/litellm
1
0
Fork 0
mirror of https://github.com/tiennm99/litellm.git synced 2026-08-17 10:26:07 +00:00
Code Issues Packages Projects Releases Wiki Activity

test(test_router.py): handle image gen timeouts

Browse Source
This commit is contained in:
Krrish Dholakia
2024-01-10 20:56:52 +05:30
parent fe632c08a4
commit 990c32a5d6
1 changed files with 9 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
litellm/tests/test_router.py
+9 -2
View File
@@ -458,6 +458,8 @@ async def test_aimg_gen_on_router():
except Exception as e:
if "Your task failed as a result of our safety system." in str(e):
pass
elif "Operation polling timed out" in str(e):
pass
else:
traceback.print_exc()
pytest.fail(f"Error occurred: {e}")
@@ -533,8 +535,13 @@ def test_aembedding_on_router():
)
router.reset()
except Exception as e:
traceback.print_exc()
pytest.fail(f"Error occurred: {e}")
if "Your task failed as a result of our safety system." in str(e):
pass
elif "Operation polling timed out" in str(e):
pass
else:
traceback.print_exc()
pytest.fail(f"Error occurred: {e}")
# test_aembedding_on_router()