From 85668ff41553aa8c13d4c9fa198f84441c99fad9 Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Thu, 7 Dec 2023 10:44:45 -0800 Subject: [PATCH] (ci/cd) vertex ai --- litellm/tests/test_amazing_vertex_completion.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/litellm/tests/test_amazing_vertex_completion.py b/litellm/tests/test_amazing_vertex_completion.py index a028250ce15..ddc83487682 100644 --- a/litellm/tests/test_amazing_vertex_completion.py +++ b/litellm/tests/test_amazing_vertex_completion.py @@ -25,12 +25,15 @@ messages = [{"content": user_message, "role": "user"}] def load_vertex_ai_credentials(): # Define the path to the vertex_key.json file - vertex_key_path = os.getcwd() + '/vertex_key.json' + print("loading vertex ai credentials") + filepath = os.path.dirname(os.path.abspath(__file__)) + vertex_key_path = filepath + '/vertex_key.json' # Read the existing content of the file or create an empty dictionary try: with open(vertex_key_path, 'r') as file: # Read the file content + print("Read vertexai file path") content = file.read() # If the file is empty or not valid JSON, create an empty dictionary