phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

test(test_whisper.py): hardcode api base

Browse source
This commit is contained in:
Krrish Dholakia 2024-03-08 22:51:17 -08:00
parent c0c76707a1
commit 7ff8fa09d6
1 changed files with 4 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

9
tests/test_whisper.py
View file

@ -36,13 +36,12 @@ def test_transcription():
def test_transcription_azure(): def test_transcription_azure():
litellm.set_verbose = True litellm.set_verbose = True
print(f"AZURE EUROPE API BASE: {os.getenv('AZURE_EUROPE_API_BASE', None)}")
transcript = litellm.transcription( transcript = litellm.transcription(
model="azure/azure-whisper", model="azure/azure-whisper",
file=audio_file, file=audio_file,
api_base=os.getenv("AZURE_EUROPE_API_BASE"), api_base="https://my-endpoint-europe-berri-992.openai.azure.com/",
api_key=os.getenv("AZURE_EUROPE_API_KEY"), api_key=os.getenv("AZURE_EUROPE_API_KEY"),
api_version=os.getenv("2024-02-15-preview"), api_version="2024-02-15-preview",
) )
assert transcript.text is not None assert transcript.text is not None
@ -57,9 +56,9 @@ async def test_transcription_async_azure():
transcript = await litellm.atranscription( transcript = await litellm.atranscription(
model="azure/azure-whisper", model="azure/azure-whisper",
file=audio_file, file=audio_file,
api_base=os.getenv("AZURE_EUROPE_API_BASE"), api_base="https://my-endpoint-europe-berri-992.openai.azure.com/",
api_key=os.getenv("AZURE_EUROPE_API_KEY"), api_key=os.getenv("AZURE_EUROPE_API_KEY"),
api_version=os.getenv("2024-02-15-preview"), api_version="2024-02-15-preview",
) )
assert transcript.text is not None assert transcript.text is not None