diff --git a/litellm/tests/test_audio_speech.py b/litellm/tests/test_audio_speech.py index 2c710d37e7e..4e45b995383 100644 --- a/litellm/tests/test_audio_speech.py +++ b/litellm/tests/test_audio_speech.py @@ -60,7 +60,7 @@ async def test_audio_speech_litellm(sync_mode, model, api_base, api_key): optional_params={}, ) - from litellm.llms.openai import HttpxBinaryResponseContent + from litellm.llms.OpenAI.openai import HttpxBinaryResponseContent assert isinstance(response, HttpxBinaryResponseContent) else: @@ -78,7 +78,7 @@ async def test_audio_speech_litellm(sync_mode, model, api_base, api_key): optional_params={}, ) - from litellm.llms.openai import HttpxBinaryResponseContent + from litellm.llms.OpenAI.openai import HttpxBinaryResponseContent assert isinstance(response, HttpxBinaryResponseContent) @@ -115,7 +115,7 @@ async def test_audio_speech_router(mode): optional_params={}, ) - from litellm.llms.openai import HttpxBinaryResponseContent + from litellm.llms.OpenAI.openai import HttpxBinaryResponseContent assert isinstance(response, HttpxBinaryResponseContent) @@ -146,7 +146,7 @@ async def test_audio_speech_litellm_vertex(sync_mode): from types import SimpleNamespace - from litellm.llms.openai import HttpxBinaryResponseContent + from litellm.llms.OpenAI.openai import HttpxBinaryResponseContent response.stream_to_file(speech_file_path)