From 2f3d40014856ce5b85353c40adfdd761463a337c Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Wed, 28 Aug 2024 15:30:39 -0700 Subject: [PATCH] fix vertex ai test --- litellm/tests/test_amazing_vertex_completion.py | 1 + 1 file changed, 1 insertion(+) diff --git a/litellm/tests/test_amazing_vertex_completion.py b/litellm/tests/test_amazing_vertex_completion.py index c6d6cd26bd1..8cba313e436 100644 --- a/litellm/tests/test_amazing_vertex_completion.py +++ b/litellm/tests/test_amazing_vertex_completion.py @@ -52,6 +52,7 @@ VERTEX_MODELS_TO_NOT_TEST = [ "gemini-1.5-pro-preview-0215", "gemini-pro-experimental", "gemini-flash-experimental", + "gemini-pro-flash", ]