From 43e69a7aa87411efcfd77e26c171d253fbd21b27 Mon Sep 17 00:00:00 2001
From: ishaan-jaff <ishaanjaffer0324@gmail.com>
Date: Fri, 3 Nov 2023 12:54:35 -0700
Subject: [PATCH] (test) vertex ai

---
 litellm/tests/test_completion.py | 11 ++++++-----
 1 file changed, 6 insertions(+), 5 deletions(-)

diff --git a/litellm/tests/test_completion.py b/litellm/tests/test_completion.py
index ed0229278a1..35f408e7bb2 100644
--- a/litellm/tests/test_completion.py
+++ b/litellm/tests/test_completion.py
@@ -846,7 +846,7 @@ def test_customprompt_together_ai():
     except Exception as e:
         pytest.fail(f"Error occurred: {e}")
 
-test_customprompt_together_ai()
+# test_customprompt_together_ai()
 
 def test_completion_sagemaker():
     try:
@@ -1113,14 +1113,15 @@ def test_completion_bedrock_claude_completion_auth():
 # test_vertex_ai()
 
 # def test_vertex_ai_stream():
-#     litellm.vertex_project = "hardy-device-386718"
-#     litellm.vertex_location = "us-central1"
+#     litellm.set_verbose=False
 #     test_models = litellm.vertex_chat_models + litellm.vertex_code_chat_models + litellm.vertex_text_models + litellm.vertex_code_text_models
 #     for model in test_models:
 #         try:
+#             if model in ["code-gecko@001", "code-gecko@latest"]:
+#                 # our account does not have access to this model
+#                 continue
 #             print("making request", model)
-#             response = completion(model=model, messages=[{"role": "user", "content": "write code for saying hi"}], stream=True)
-#             print(response)
+#             response = completion(model=model, messages=[{"role": "user", "content": "write 100 line code code for saying hi"}], stream=True)
 #             for chunk in response:
 #                 print(chunk)
 #                 # pass