tiennm99/litellm
1
0
Fork 0
mirror of https://github.com/tiennm99/litellm.git synced 2026-07-19 18:19:21 +00:00
Code Issues Packages Projects Releases Wiki Activity

handle Azure rate limit errors for test_transcription

Browse Source
This commit is contained in:
Ishaan Jaff
2024-12-02 23:32:28 -08:00
parent ed31476934
commit 6e6cd05f69
1 changed files with 8 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/local_testing/test_whisper.py
+8 -20
View File
@@ -55,30 +55,18 @@ from litellm import Router
"response_format, timestamp_granularities",
[("json", None), ("vtt", None), ("verbose_json", ["word"])],
)
@pytest.mark.parametrize("sync_mode", [True, False])
@pytest.mark.asyncio
async def test_transcription(
model, api_key, api_base, response_format, sync_mode, timestamp_granularities
):
if sync_mode:
transcript = litellm.transcription(
model=model,
file=audio_file,
api_key=api_key,
api_base=api_base,
response_format=response_format,
timestamp_granularities=timestamp_granularities,
drop_params=True,
)
else:
transcript = await litellm.atranscription(
model=model,
file=audio_file,
api_key=api_key,
api_base=api_base,
response_format=response_format,
drop_params=True,
)
transcript = await litellm.atranscription(
model=model,
file=audio_file,
api_key=api_key,
api_base=api_base,
response_format=response_format,
drop_params=True,
)
print(f"transcript: {transcript.model_dump()}")
print(f"transcript: {transcript._hidden_params}")