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

test(test_router.py): handle azure api instability

Browse source
This commit is contained in:
Krrish Dholakia 2024-07-25 19:54:40 -07:00
parent a2fd8459fc
commit 826bb125e8
1 changed files with 2 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

2
litellm/tests/test_router.py
View file

@ -1117,6 +1117,8 @@ async def test_aimg_gen_on_router():
assert len(response.data) > 0 assert len(response.data) > 0
router.reset() router.reset()
except litellm.InternalServerError as e:
pass
except Exception as e: except Exception as e:
if "Your task failed as a result of our safety system." in str(e): if "Your task failed as a result of our safety system." in str(e):
pass pass