phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-27 19:54:13 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

simpify ssml usage

Browse source
This commit is contained in:
Ishaan Jaff 2024-08-28 12:54:23 -07:00
parent 58506dbade
commit 5ea27bdea9
1 changed files with 9 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

12
litellm/tests/test_audio_speech.py
View file

@ -258,6 +258,13 @@ async def test_speech_litellm_vertex_async_with_voice_ssml():
mock_response.json = return_val mock_response.json = return_val
mock_response.status_code = 200 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 # Set up the mock for asynchronous calls
with patch( with patch(
"litellm.llms.custom_httpx.http_handler.AsyncHTTPHandler.post", "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" model = "vertex_ai/test"
response = await litellm.aspeech( response = await litellm.aspeech(
input=None, input=ssml,
model=model, model=model,
ssml="async hello what llm guardrail do you have",
voice={ voice={
"languageCode": "en-UK", "languageCode": "en-UK",
"name": "en-UK-Studio-O", "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["headers"]["Authorization"] is not None
assert kwargs["json"] == { 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"}, "voice": {"languageCode": "en-UK", "name": "en-UK-Studio-O"},
"audioConfig": {"audioEncoding": "LINEAR22", "speakingRate": "10"}, "audioConfig": {"audioEncoding": "LINEAR22", "speakingRate": "10"},
} }