From ce46d11a37ca71c46ade02c9bd0f33a81fbb5288 Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Tue, 27 Feb 2024 20:37:08 -0800 Subject: [PATCH] test: testing fixes --- litellm/tests/test_amazing_vertex_completion.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/litellm/tests/test_amazing_vertex_completion.py b/litellm/tests/test_amazing_vertex_completion.py index fb2ebc0d65..1de26d3b96 100644 --- a/litellm/tests/test_amazing_vertex_completion.py +++ b/litellm/tests/test_amazing_vertex_completion.py @@ -110,6 +110,7 @@ def test_vertex_ai(): "code-bison@001", "text-bison@001", "gemini-1.5-pro", + "gemini-1.5-pro-preview-0215", "gemini-1.5-pro-vision", ]: # our account does not have access to this model @@ -160,6 +161,7 @@ def test_vertex_ai_stream(): "code-bison@001", "text-bison@001", "gemini-1.5-pro", + "gemini-1.5-pro-preview-0215", "gemini-1.5-pro-vision", ]: # our account does not have access to this model @@ -211,6 +213,7 @@ async def test_async_vertexai_response(): "code-bison@001", "text-bison@001", "gemini-1.5-pro", + "gemini-1.5-pro-preview-0215", "gemini-1.5-pro-vision", ]: # our account does not have access to this model @@ -255,6 +258,7 @@ async def test_async_vertexai_streaming_response(): "code-bison@001", "text-bison@001", "gemini-1.5-pro", + "gemini-1.5-pro-preview-0215", "gemini-1.5-pro-vision", ]: # our account does not have access to this model