From 0354d6d48f6daf0dd87bf840c66db7f5e5300f04 Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Thu, 7 Dec 2023 12:41:39 -0800 Subject: [PATCH] (ci/cd) vertex test --- litellm/tests/test_amazing_vertex_completion.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/litellm/tests/test_amazing_vertex_completion.py b/litellm/tests/test_amazing_vertex_completion.py index 6bf19e1e13a..6a3f5cab75b 100644 --- a/litellm/tests/test_amazing_vertex_completion.py +++ b/litellm/tests/test_amazing_vertex_completion.py @@ -68,8 +68,9 @@ def test_vertex_ai(): import random load_vertex_ai_credentials() - test_models = ["codechat-bison"] + litellm.vertex_chat_models + litellm.vertex_code_chat_models + litellm.vertex_text_models + litellm.vertex_code_text_models + test_models = litellm.vertex_chat_models + litellm.vertex_code_chat_models + litellm.vertex_text_models + litellm.vertex_code_text_models litellm.set_verbose=False + litellm.vertex_project = "hardy-device-386718" test_models = random.sample(test_models, 4) for model in test_models: @@ -90,9 +91,10 @@ def test_vertex_ai(): def test_vertex_ai_stream(): load_vertex_ai_credentials() litellm.set_verbose=False + litellm.vertex_project = "hardy-device-386718" import random - test_models = ["codechat-bison"] + litellm.vertex_chat_models + litellm.vertex_code_chat_models + litellm.vertex_text_models + litellm.vertex_code_text_models + test_models = litellm.vertex_chat_models + litellm.vertex_code_chat_models + litellm.vertex_text_models + litellm.vertex_code_text_models test_models = random.sample(test_models, 4) for model in test_models: try: