From 3f4fe668a8db85c9f2642ae07bc69d2d35cd99b8 Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Thu, 7 Dec 2023 18:20:58 -0800 Subject: [PATCH] (ci/cd) vertex ai test --- litellm/tests/test_amazing_vertex_completion.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/litellm/tests/test_amazing_vertex_completion.py b/litellm/tests/test_amazing_vertex_completion.py index 6a3f5cab75..8b6821caae 100644 --- a/litellm/tests/test_amazing_vertex_completion.py +++ b/litellm/tests/test_amazing_vertex_completion.py @@ -75,7 +75,7 @@ def test_vertex_ai(): test_models = random.sample(test_models, 4) for model in test_models: try: - if model in ["code-gecko@001", "code-gecko@latest", "code-bison@001"]: + if model in ["code-gecko@001", "code-gecko@latest", "code-bison@001", "text-bison@001"]: # our account does not have access to this model continue print("making request", model) @@ -98,7 +98,7 @@ def test_vertex_ai_stream(): test_models = random.sample(test_models, 4) for model in test_models: try: - if model in ["code-gecko@001", "code-gecko@latest", "code-bison@001"]: + if model in ["code-gecko@001", "code-gecko@latest", "code-bison@001", "text-bison@001"]: # our account does not have access to this model continue print("making request", model)