From 077f6b1298101079b72396bdf04f8ca0cf737720 Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Thu, 7 Dec 2023 11:19:29 -0800 Subject: [PATCH] (ci/cd) vertex ai --- litellm/tests/test_amazing_vertex_completion.py | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/litellm/tests/test_amazing_vertex_completion.py b/litellm/tests/test_amazing_vertex_completion.py index b1e4e6b5e..c87438489 100644 --- a/litellm/tests/test_amazing_vertex_completion.py +++ b/litellm/tests/test_amazing_vertex_completion.py @@ -50,13 +50,16 @@ def load_vertex_ai_credentials(): # Update the service_account_key_data with environment variables private_key_id = os.environ.get("VERTEX_AI_PRIVATE_KEY_ID", "") private_key = os.environ.get("VERTEX_AI_PRIVATE_KEY", "") + private_key = private_key.replace("\\n", "\n") + + # from infisical import InfisicalClient + # client = InfisicalClient(token=os.getenv("INFISICAL_TOKEN")) + # private_key = (client.get_secret(secret_name="VERTEX_AI_PRIVATE_KEY", environment="prod").secret_value) + # private_key = private_key.replace("\\n", "\n") service_account_key_data["private_key_id"] = private_key_id service_account_key_data["private_key"] = private_key - for key in service_account_key_data: - if key not in ["private_key_id", "private_key"]: - print(f"Key: {key}, Value: {service_account_key_data[key][:10]}...{service_account_key_data[key][:-10]}") # Create a temporary file with tempfile.NamedTemporaryFile(mode='w+', delete=False) as temp_file: