test: testing fixes

2024-02-26 10:38:49 -08:00 · 2024-02-26 10:38:49 -08:00 · f460da4770
commit f460da4770
parent 144254bd9d
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
@ -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