From 225ff8432deda80c498aeb183f5fbc807da23c57 Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Fri, 23 Aug 2024 16:06:39 -0700 Subject: [PATCH] test vertex ai text to speech --- litellm/tests/test_audio_speech.py | 12 +++--------- 1 file changed, 3 insertions(+), 9 deletions(-) diff --git a/litellm/tests/test_audio_speech.py b/litellm/tests/test_audio_speech.py index 5e0cf60779..129f4ea76b 100644 --- a/litellm/tests/test_audio_speech.py +++ b/litellm/tests/test_audio_speech.py @@ -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", )