diff --git a/litellm/tests/gettysburg.wav b/litellm/tests/gettysburg.wav new file mode 100644 index 0000000000..9690f521e8 Binary files /dev/null and b/litellm/tests/gettysburg.wav differ diff --git a/tests/test_whisper.py b/litellm/tests/test_whisper.py similarity index 92% rename from tests/test_whisper.py rename to litellm/tests/test_whisper.py index 09819f796c..d0f1ed9eb0 100644 --- a/tests/test_whisper.py +++ b/litellm/tests/test_whisper.py @@ -1,16 +1,22 @@ # What is this? ## Tests `litellm.transcription` endpoint. Outside litellm module b/c of audio file used in testing (it's ~700kb). -import pytest -import asyncio, time -import aiohttp, traceback -from openai import AsyncOpenAI -import sys, os, dotenv -from typing import Optional -from dotenv import load_dotenv -from litellm.integrations.custom_logger import CustomLogger -import litellm +import asyncio import logging +import os +import sys +import time +import traceback +from typing import Optional + +import aiohttp +import dotenv +import pytest +from dotenv import load_dotenv +from openai import AsyncOpenAI + +import litellm +from litellm.integrations.custom_logger import CustomLogger # Get the current directory of the file being run pwd = os.path.dirname(os.path.realpath(__file__)) @@ -41,6 +47,19 @@ def test_transcription(): # test_transcription() +def test_transcription_groq(): + litellm.set_verbose = True + transcript = litellm.transcription( + model="groq/whisper-large-v3", + file=audio_file, + ) + print(f"response=: {transcript.model_dump()}") + print(f"hidden_params: {transcript._hidden_params}") + + +# test_transcription() + + def test_transcription_azure(): litellm.set_verbose = True transcript = litellm.transcription(