feat: add network_attachment to PscInterfaceConfig

Google APIs · copybara-github · commit 89d75d55e24f · 2025-05-30T11:18:21.000-07:00
feat: add psc_interface_config to CustomJobSpec
feat: add psc_interface_config to PersistentResource
feat: add psc_interface_config to PipelineJob

PiperOrigin-RevId: 765264778
diff --git a/google/cloud/aiplatform/v1/custom_job.proto b/google/cloud/aiplatform/v1/custom_job.proto
@@ -23,6 +23,7 @@ import "google/cloud/aiplatform/v1/env_var.proto";
 import "google/cloud/aiplatform/v1/io.proto";
 import "google/cloud/aiplatform/v1/job_state.proto";
 import "google/cloud/aiplatform/v1/machine_resources.proto";
+import "google/cloud/aiplatform/v1/service_networking.proto";
 import "google/protobuf/duration.proto";
 import "google/protobuf/timestamp.proto";
 import "google/rpc/status.proto";
@@ -178,6 +179,10 @@ message CustomJobSpec {
   repeated string reserved_ip_ranges = 13
       [(google.api.field_behavior) = OPTIONAL];
 
+  // Optional. Configuration for PSC-I for CustomJob.
+  PscInterfaceConfig psc_interface_config = 21
+      [(google.api.field_behavior) = OPTIONAL];
+
   // The Cloud Storage location to store the output of this CustomJob or
   // HyperparameterTuningJob. For HyperparameterTuningJob,
   // the baseOutputDirectory of
diff --git a/google/cloud/aiplatform/v1/persistent_resource.proto b/google/cloud/aiplatform/v1/persistent_resource.proto
@@ -20,6 +20,7 @@ import "google/api/field_behavior.proto";
 import "google/api/resource.proto";
 import "google/cloud/aiplatform/v1/encryption_spec.proto";
 import "google/cloud/aiplatform/v1/machine_resources.proto";
+import "google/cloud/aiplatform/v1/service_networking.proto";
 import "google/protobuf/timestamp.proto";
 import "google/rpc/status.proto";
 
@@ -131,6 +132,10 @@ message PersistentResource {
     (google.api.resource_reference) = { type: "compute.googleapis.com/Network" }
   ];
 
+  // Optional. Configuration for PSC-I for PersistentResource.
+  PscInterfaceConfig psc_interface_config = 17
+      [(google.api.field_behavior) = OPTIONAL];
+
   // Optional. Customer-managed encryption key spec for a PersistentResource.
   // If set, this PersistentResource and all sub-resources of this
   // PersistentResource will be secured by this key.
diff --git a/google/cloud/aiplatform/v1/pipeline_job.proto b/google/cloud/aiplatform/v1/pipeline_job.proto
@@ -24,6 +24,7 @@ import "google/cloud/aiplatform/v1/encryption_spec.proto";
 import "google/cloud/aiplatform/v1/execution.proto";
 import "google/cloud/aiplatform/v1/pipeline_failure_policy.proto";
 import "google/cloud/aiplatform/v1/pipeline_state.proto";
+import "google/cloud/aiplatform/v1/service_networking.proto";
 import "google/cloud/aiplatform/v1/value.proto";
 import "google/protobuf/struct.proto";
 import "google/protobuf/timestamp.proto";
@@ -197,6 +198,10 @@ message PipelineJob {
   // Example: ['vertex-ai-ip-range'].
   repeated string reserved_ip_ranges = 25;
 
+  // Optional. Configuration for PSC-I for PipelineJob.
+  PscInterfaceConfig psc_interface_config = 31
+      [(google.api.field_behavior) = OPTIONAL];
+
   // A template uri from where the
   // [PipelineJob.pipeline_spec][google.cloud.aiplatform.v1.PipelineJob.pipeline_spec],
   // if empty, will be downloaded. Currently, only uri from Vertex Template
diff --git a/google/cloud/aiplatform/v1/service_networking.proto b/google/cloud/aiplatform/v1/service_networking.proto
@@ -74,3 +74,20 @@ message PscAutomatedEndpoints {
   // Ip Address created by the automated forwarding rule.
   string match_address = 3;
 }
+
+// Configuration for PSC-I.
+message PscInterfaceConfig {
+  // Optional. The name of the Compute Engine
+  // [network
+  // attachment](https://cloud.google.com/vpc/docs/about-network-attachments) to
+  // attach to the resource within the region and user project.
+  // To specify this field, you must have already [created a network attachment]
+  // (https://cloud.google.com/vpc/docs/create-manage-network-attachments#create-network-attachments).
+  // This field is only used for resources using PSC-I.
+  string network_attachment = 1 [
+    (google.api.field_behavior) = OPTIONAL,
+    (google.api.resource_reference) = {
+      type: "compute.googleapis.com/NetworkAttachment"
+    }
+  ];
+}
