test: update test case for VERTEX_CREDENTIALS

tmc · Jun 15, 2024 · 65fd3ee · 65fd3ee
1 parent 57693f7
commit 65fd3ee
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/llms/googleai/shared_test/shared_test.go b/llms/googleai/shared_test/shared_test.go
@@ -50,13 +50,18 @@ func newVertexClient(t *testing.T, opts ...googleai.Option) *vertex.Vertex {
 	if location == "" {
 		location = "us-central1"
 	}
-	credentials := os.Getenv("VERTEX_CREDENTIALS")
 
 	opts = append(opts,
 		googleai.WithCloudProject(project),
 		googleai.WithCloudLocation(location),
-		googleai.WithCredentialsFile(credentials),
 	)
+
+	// If credentials are set, use them.
+	credentials := os.Getenv("VERTEX_CREDENTIALS")
+	if credentials != "" {
+		opts = append(opts, googleai.WithCredentialsFile(credentials))
+	}
+
 	llm, err := vertex.New(context.Background(), opts...)
 	require.NoError(t, err)
 	return llm