fix import

This commit is contained in:
Ishaan Jaff 2024-09-05 12:40:45 -07:00
parent 219fa492dc
commit 936e486c95

View file

@ -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)