phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

test: fix test

Browse source
This commit is contained in:
Krrish Dholakia 2024-08-03 20:28:11 -07:00
parent ac6c39c283
commit 23cad65256
1 changed files with 4 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

5
litellm/tests/test_image_generation.py
View file

@ -194,6 +194,9 @@ async def test_aimage_generation_bedrock_with_optional_params():
pytest.fail(f"An exception occurred - {str(e)}") pytest.fail(f"An exception occurred - {str(e)}")
from openai.types.image import Image
@pytest.mark.parametrize("sync_mode", [True, False]) @pytest.mark.parametrize("sync_mode", [True, False])
@pytest.mark.asyncio @pytest.mark.asyncio
async def test_aimage_generation_vertex_ai(sync_mode): async def test_aimage_generation_vertex_ai(sync_mode):
@ -218,7 +221,7 @@ async def test_aimage_generation_vertex_ai(sync_mode):
assert len(response.data) > 0 assert len(response.data) > 0
for d in response.data: for d in response.data:
assert isinstance(d, litellm.ImageObject) assert isinstance(d, Image)
print("data in response.data", d) print("data in response.data", d)
assert d.b64_json is not None assert d.b64_json is not None
except litellm.ServiceUnavailableError as e: except litellm.ServiceUnavailableError as e: