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

---
 litellm/tests/test_amazing_vertex_completion.py | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/litellm/tests/test_amazing_vertex_completion.py b/litellm/tests/test_amazing_vertex_completion.py
index 76ebde7aefd..35beb75fe6f 100644
--- a/litellm/tests/test_amazing_vertex_completion.py
+++ b/litellm/tests/test_amazing_vertex_completion.py
@@ -109,6 +109,8 @@ def test_vertex_ai():
                 "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
@@ -206,6 +208,8 @@ async def test_async_vertexai_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
@@ -246,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