tiennm99/litellm
1
0
Fork 0
mirror of https://github.com/tiennm99/litellm.git synced 2026-08-13 20:23:56 +00:00
Code Issues Packages Projects Releases Wiki Activity

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
litellm/tests/test_image_generation.py
+3
View File
@@ -173,6 +173,9 @@ async def test_aimage_generation_bedrock_with_optional_params():
@pytest.mark.asyncio
async def test_aimage_generation_vertex_ai():
from test_amazing_vertex_completion import load_vertex_ai_credentials
load_vertex_ai_credentials()
try:
response = await litellm.aimage_generation(
prompt="An olympic size swimming pool",