test(test_amazing_vertex_completion.py): fix test

2024-02-26 12:57:23 -08:00 · 2024-02-26 12:57:23 -08:00 · f7af18d72a
commit f7af18d72a
parent 069704c85b
1 changed files with 2 additions and 0 deletions
--- 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