From 5ea27bdea99e0ada8f15f9956e7b46701a30d162 Mon Sep 17 00:00:00 2001
From: Ishaan Jaff <ishaanjaffer0324@gmail.com>
Date: Wed, 28 Aug 2024 12:54:23 -0700
Subject: [PATCH] simpify ssml usage

---
 litellm/tests/test_audio_speech.py | 12 +++++++++---
 1 file changed, 9 insertions(+), 3 deletions(-)

diff --git a/litellm/tests/test_audio_speech.py b/litellm/tests/test_audio_speech.py
index d9ed3fd6ea..2c710d37e7 100644
--- a/litellm/tests/test_audio_speech.py
+++ b/litellm/tests/test_audio_speech.py
@@ -258,6 +258,13 @@ async def test_speech_litellm_vertex_async_with_voice_ssml():
     mock_response.json = return_val
     mock_response.status_code = 200
 
+    ssml = """
+    <speak>
+        <p>Hello, world!</p>
+        <p>This is a test of the <break strength="medium" /> text-to-speech API.</p>
+    </speak>
+    """
+
     # Set up the mock for asynchronous calls
     with patch(
         "litellm.llms.custom_httpx.http_handler.AsyncHTTPHandler.post",
@@ -267,9 +274,8 @@ async def test_speech_litellm_vertex_async_with_voice_ssml():
         model = "vertex_ai/test"
 
         response = await litellm.aspeech(
-            input=None,
+            input=ssml,
             model=model,
-            ssml="async hello what llm guardrail do you have",
             voice={
                 "languageCode": "en-UK",
                 "name": "en-UK-Studio-O",
@@ -291,7 +297,7 @@ async def test_speech_litellm_vertex_async_with_voice_ssml():
         assert kwargs["headers"]["Authorization"] is not None
 
         assert kwargs["json"] == {
-            "input": {"ssml": "async hello what llm guardrail do you have"},
+            "input": {"ssml": ssml},
             "voice": {"languageCode": "en-UK", "name": "en-UK-Studio-O"},
             "audioConfig": {"audioEncoding": "LINEAR22", "speakingRate": "10"},
         }