From f7af18d72a6b4094f85ffdc8daa30a4256deabc5 Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Mon, 26 Feb 2024 12:57:23 -0800 Subject: [PATCH] test(test_amazing_vertex_completion.py): fix test --- litellm/tests/test_amazing_vertex_completion.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/litellm/tests/test_amazing_vertex_completion.py b/litellm/tests/test_amazing_vertex_completion.py index 35beb75fe..34370b057 100644 --- a/litellm/tests/test_amazing_vertex_completion.py +++ b/litellm/tests/test_amazing_vertex_completion.py @@ -159,6 +159,8 @@ def test_vertex_ai_stream(): "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