tiennm99/litellm
1
0
Fork 0
mirror of https://github.com/tiennm99/litellm.git synced 2026-08-10 22:24:51 +00:00
Code Issues 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
Download Patch File Download Diff File Expand all files Collapse all files
tests/local_testing/test_audio_speech.py
+2
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"