scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-06 08:16:43 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 78

test vertex ai text to speech

Browse source
This commit is contained in:
Ishaan Jaff 2024-08-23 16:06:39 -07:00
parent 755a0514f6
commit 225ff8432d
1 changed files with 3 additions and 9 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
litellm/tests/test_audio_speech.py
View file

@ -121,7 +121,7 @@ async def test_audio_speech_router(mode):
@pytest.mark.parametrize(
"sync_mode",
[False],
[False, True],
)
@pytest.mark.asyncio
async def test_audio_speech_litellm_vertex(sync_mode):
@ -130,21 +130,15 @@ async def test_audio_speech_litellm_vertex(sync_mode):
model = "vertex_ai/test"
if sync_mode:
response = litellm.speech(
model=model,
voice="alloy",
model="vertex_ai/test",
input="hello what llm guardrail do you have",
)
from types import SimpleNamespace
from litellm.llms.openai import HttpxBinaryResponseContent
response.stream_to_file(speech_file_path)
else:
response = await litellm.aspeech(
model=model,
voice="alloy",
model="vertex_ai/",
input="async hello what llm guardrail do you have",
)