mirror of
https://github.com/BerriAI/litellm.git
synced 2025-04-25 10:44:24 +00:00
Litellm fix router testing (#5748)
* test: fix testing - azure changed content policy error logic * test: fix tests to use mock responses * test(test_image_generation.py): handle api instability * test(test_image_generation.py): handle azure api instability * fix(utils.py): fix unbounded variable error * fix(utils.py): fix unbounded variable error * test: refactor test to use mock response * test: mark flaky azure tests
This commit is contained in:
parent
8d4339c702
commit
dd602753c0
8 changed files with 36 additions and 11 deletions
|
@ -440,6 +440,7 @@ async def test_embeddings():
|
|||
await embeddings(session=session, key=key, model="mistral-embed")
|
||||
|
||||
|
||||
@pytest.mark.flaky(retries=5, delay=1)
|
||||
@pytest.mark.asyncio
|
||||
async def test_image_generation():
|
||||
"""
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue