feat: add read_mask to ListPipelineJobsRequest in aiplatform v1 pipeline_service

Google APIs · copybara-github · commit eb382eda1d3d · 2022-08-24T15:49:30.000-07:00
feat: add input_artifacts to PipelineJob.runtime_config in aiplatform v1 pipeline_job

PiperOrigin-RevId: 469843667
diff --git a/google/cloud/aiplatform/v1/pipeline_job.proto b/google/cloud/aiplatform/v1/pipeline_job.proto
@@ -50,6 +50,18 @@ message PipelineJob {
 
   // The runtime config of a PipelineJob.
   message RuntimeConfig {
+    // The type of an input artifact.
+    message InputArtifact {
+      oneof kind {
+        // Artifact resource id from MLMD. Which is the last portion of an
+        // artifact resource
+        // name(projects/{project}/locations/{location}/metadataStores/default/artifacts/{artifact_id}).
+        // The artifact must stay within the same project, location and default
+        // metadatastore as the pipeline.
+        string artifact_id = 1;
+      }
+    }
+
     // Deprecated. Use [RuntimeConfig.parameter_values][google.cloud.aiplatform.v1.PipelineJob.RuntimeConfig.parameter_values] instead. The runtime
     // parameters of the PipelineJob. The parameters will be passed into
     // [PipelineJob.pipeline_spec][google.cloud.aiplatform.v1.PipelineJob.pipeline_spec] to replace the placeholders at runtime.
@@ -81,6 +93,10 @@ message PipelineJob {
     // will stop scheduling any new tasks when a task has failed. Any scheduled
     // tasks will continue to completion.
     PipelineFailurePolicy failure_policy = 4;
+
+    // The runtime artifacts of the PipelineJob. The key will be the input
+    // artifact name and the value would be one of the InputArtifact.
+    map<string, InputArtifact> input_artifacts = 5;
   }
 
   // Output only. The resource name of the PipelineJob.
diff --git a/google/cloud/aiplatform/v1/pipeline_service.proto b/google/cloud/aiplatform/v1/pipeline_service.proto
@@ -366,6 +366,9 @@ message ListPipelineJobsRequest {
   //   * `end_time`
   //   * `start_time`
   string order_by = 6;
+
+  // Mask specifying which fields to read.
+  google.protobuf.FieldMask read_mask = 7;
 }
 
 // Response message for [PipelineService.ListPipelineJobs][google.cloud.aiplatform.v1.PipelineService.ListPipelineJobs]

Original file line number	Diff line number	Diff line change
`@@ -366,6 +366,9 @@ message ListPipelineJobsRequest {`
`366`	`366`	// * `end_time`
`367`	`367`	// * `start_time`
`368`	`368`	`string order_by = 6;`
	`369`	`+`
	`370`	`+ // Mask specifying which fields to read.`
	`371`	`+ google.protobuf.FieldMask read_mask = 7;`
`369`	`372`	`}`
`370`	`373`
`371`	`374`	`// Response message for [PipelineService.ListPipelineJobs][google.cloud.aiplatform.v1.PipelineService.ListPipelineJobs]`