From 28f4b5809c60f171ce3639e9e47bffec72c9603e Mon Sep 17 00:00:00 2001
From: Krrish Dholakia <krrishdholakia@gmail.com>
Date: Mon, 26 Feb 2024 10:42:05 -0800
Subject: [PATCH] test(test_amazing_vertex_completion.py): fix test

---
 litellm/main.py                                 | 1 -
 litellm/tests/test_amazing_vertex_completion.py | 2 ++
 2 files changed, 2 insertions(+), 1 deletion(-)

diff --git a/litellm/main.py b/litellm/main.py
index ed8dddf05..33fad52cc 100644
--- a/litellm/main.py
+++ b/litellm/main.py
@@ -10,7 +10,6 @@
 import os, openai, sys, json, inspect, uuid, datetime, threading
 from typing import Any, Literal, Union
 from functools import partial
-
 import dotenv, traceback, random, asyncio, time, contextvars
 from copy import deepcopy
 import httpx
diff --git a/litellm/tests/test_amazing_vertex_completion.py b/litellm/tests/test_amazing_vertex_completion.py
index bca241cd7..35beb75fe 100644
--- a/litellm/tests/test_amazing_vertex_completion.py
+++ b/litellm/tests/test_amazing_vertex_completion.py
@@ -250,6 +250,8 @@ async def test_async_vertexai_streaming_response():
             "code-gecko@latest",
             "code-bison@001",
             "text-bison@001",
+            "gemini-1.5-pro",
+            "gemini-1.5-pro-vision",
         ]:
             # our account does not have access to this model
             continue