fix(vertexai): use opts in NewClient

Fixes googleapis#9160
eliben · Dec 13, 2023 · e5eb975 · e5eb975
1 parent f41f85e
commit e5eb975
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/vertexai/genai/client.go b/vertexai/genai/client.go
@@ -47,7 +47,8 @@ type Client struct {
 // package.
 func NewClient(ctx context.Context, projectID, location string, opts ...option.ClientOption) (*Client, error) {
 	apiEndpoint := fmt.Sprintf("%s-aiplatform.googleapis.com:443", location)
-	c, err := aiplatform.NewPredictionClient(ctx, option.WithEndpoint(apiEndpoint))
+	opts = append(opts, option.WithEndpoint(apiEndpoint))
+	c, err := aiplatform.NewPredictionClient(ctx, opts...)
 	if err != nil {
 		return nil, err
 	}