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

test(test_router.py): handle image gen timeouts

Browse source
This commit is contained in:
Krrish Dholakia 2024-01-10 15:49:32 +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

11
litellm/tests/test_router.py
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()