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

fix vertex test

Browse source
This commit is contained in:
Ishaan Jaff 2024-05-20 10:51:25 -07:00
parent 2519879e67
commit 2da89a0c8e
1 changed files with 3 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

3
litellm/tests/test_image_generation.py
View file

@ -173,6 +173,9 @@ async def test_aimage_generation_bedrock_with_optional_params():
@pytest.mark.asyncio @pytest.mark.asyncio
async def test_aimage_generation_vertex_ai(): async def test_aimage_generation_vertex_ai():
from test_amazing_vertex_completion import load_vertex_ai_credentials
load_vertex_ai_credentials()
try: try:
response = await litellm.aimage_generation( response = await litellm.aimage_generation(
prompt="An olympic size swimming pool", prompt="An olympic size swimming pool",