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

fix test test_aimage_generation_vertex_ai

Browse source
This commit is contained in:
Ishaan Jaff 2024-08-03 18:32:52 -07:00
parent 87721023e2
commit 5f23f51a68
1 changed files with 2 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

3
litellm/tests/test_image_generation.py
View file

@ -7,6 +7,7 @@ import sys
import traceback
from dotenv import load_dotenv
from openai.types.image import Image
logging.basicConfig(level=logging.DEBUG)
load_dotenv()
@ -218,7 +219,7 @@ async def test_aimage_generation_vertex_ai(sync_mode):
assert len(response.data) > 0
for d in response.data:
assert isinstance(d, litellm.ImageObject)
assert isinstance(d, Image)
print("data in response.data", d)
assert d.b64_json is not None
except litellm.ServiceUnavailableError as e: