Skip to content

Commit

Permalink
feat: add read_mask to ListPipelineJobsRequest in aiplatform v1 pipel…
Browse files Browse the repository at this point in the history
…ine_service

feat: add input_artifacts to PipelineJob.runtime_config in aiplatform v1 pipeline_job

PiperOrigin-RevId: 469843667
  • Loading branch information
Google APIs authored and copybara-github committed Aug 24, 2022
1 parent 97fa02d commit eb382ed
Show file tree
Hide file tree
Showing 2 changed files with 19 additions and 0 deletions.
16 changes: 16 additions & 0 deletions google/cloud/aiplatform/v1/pipeline_job.proto
Original file line number Diff line number Diff line change
Expand Up @@ -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.
Expand Down Expand Up @@ -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.
Expand Down
3 changes: 3 additions & 0 deletions google/cloud/aiplatform/v1/pipeline_service.proto
Original file line number Diff line number Diff line change
Expand Up @@ -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]
Expand Down

0 comments on commit eb382ed

Please sign in to comment.