ai-workspace-services/litellm
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix testing retry audio test 3 times

Browse Source
This commit is contained in:
Ishaan Jaff 2024-12-12 20:09:14 -08:00
parent b889d7c72f
commit 36862d0a98
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tests/local_testing/test_audio_speech.py
View File

@ -42,6 +42,7 @@ import litellm
],
) # ,
@pytest.mark.asyncio
@pytest.mark.flaky(retries=3, delay=1)
async def test_audio_speech_litellm(sync_mode, model, api_base, api_key):
speech_file_path = Path(__file__).parent / "speech.mp3"
@ -89,6 +90,7 @@ async def test_audio_speech_litellm(sync_mode, model, api_base, api_key):
)
@pytest.mark.skip(reason="local only test - we run testing using MockRequests below")
@pytest.mark.asyncio
@pytest.mark.flaky(retries=3, delay=1)
async def test_audio_speech_litellm_vertex(sync_mode):
litellm.set_verbose = True
speech_file_path = Path(__file__).parent / "speech_vertex.mp3"