refactor(vertex_ai.py): add more logging for the vertex_ai file

This commit is contained in:
Krrish Dholakia 2024-03-01 08:33:10 -08:00
parent 749bf46d05
commit 2fac06de08

View file

@ -300,7 +300,9 @@ def completion(
f"VERTEX AI: vertex_project={vertex_project}; vertex_location={vertex_location}" f"VERTEX AI: vertex_project={vertex_project}; vertex_location={vertex_location}"
) )
creds, _ = google.auth.default(quota_project_id=vertex_project) creds, _ = google.auth.default(quota_project_id=vertex_project)
print_verbose(f"VERTEX AI: creds={creds}") print_verbose(
f"VERTEX AI: creds={creds}; google application credentials: {os.getenv('GOOGLE_APPLICATION_CREDENTIALS')}"
)
vertexai.init( vertexai.init(
project=vertex_project, location=vertex_location, credentials=creds project=vertex_project, location=vertex_location, credentials=creds
) )