feat!: Removes breaking change from v1 version of AI Platform protos

google-cloud-aiplatform/src/main/java/com/google/cloud/aiplatform/v1/PipelineServiceSettings.java

@@ -16,6 +16,7 @@
 
 package com.google.cloud.aiplatform.v1;
 
+import static com.google.cloud.aiplatform.v1.PipelineServiceClient.ListPipelineJobsPagedResponse;
 import static com.google.cloud.aiplatform.v1.PipelineServiceClient.ListTrainingPipelinesPagedResponse;
 
 import com.google.api.core.ApiFunction;
@@ -111,6 +112,39 @@ public UnaryCallSettings<CancelTrainingPipelineRequest, Empty> cancelTrainingPip
  return ((PipelineServiceStubSettings) getStubSettings()).cancelTrainingPipelineSettings();
  }
 
+ /** Returns the object with the settings used for calls to createPipelineJob. */
+ public UnaryCallSettings<CreatePipelineJobRequest, PipelineJob> createPipelineJobSettings() {
+ return ((PipelineServiceStubSettings) getStubSettings()).createPipelineJobSettings();
+ }
+
+ /** Returns the object with the settings used for calls to getPipelineJob. */
+ public UnaryCallSettings<GetPipelineJobRequest, PipelineJob> getPipelineJobSettings() {
+ return ((PipelineServiceStubSettings) getStubSettings()).getPipelineJobSettings();
+ }
+
+ /** Returns the object with the settings used for calls to listPipelineJobs. */
+ public PagedCallSettings<
+ ListPipelineJobsRequest, ListPipelineJobsResponse, ListPipelineJobsPagedResponse>
+ listPipelineJobsSettings() {
+ return ((PipelineServiceStubSettings) getStubSettings()).listPipelineJobsSettings();
+ }
+
+ /** Returns the object with the settings used for calls to deletePipelineJob. */
+ public UnaryCallSettings<DeletePipelineJobRequest, Operation> deletePipelineJobSettings() {
+ return ((PipelineServiceStubSettings) getStubSettings()).deletePipelineJobSettings();
+ }
+
+ /** Returns the object with the settings used for calls to deletePipelineJob. */
+ public OperationCallSettings<DeletePipelineJobRequest, Empty, DeleteOperationMetadata>
+ deletePipelineJobOperationSettings() {
+ return ((PipelineServiceStubSettings) getStubSettings()).deletePipelineJobOperationSettings();
+ }
+
+ /** Returns the object with the settings used for calls to cancelPipelineJob. */
+ public UnaryCallSettings<CancelPipelineJobRequest, Empty> cancelPipelineJobSettings() {
+ return ((PipelineServiceStubSettings) getStubSettings()).cancelPipelineJobSettings();
+ }
+
  public static final PipelineServiceSettings create(PipelineServiceStubSettings stub)
  throws IOException {
  return new PipelineServiceSettings.Builder(stub.toBuilder()).build();
@@ -249,6 +283,41 @@ public Builder applyToAllUnaryMethods(
  return getStubSettingsBuilder().cancelTrainingPipelineSettings();
  }
 
+ /** Returns the builder for the settings used for calls to createPipelineJob. */
+ public UnaryCallSettings.Builder<CreatePipelineJobRequest, PipelineJob>
+ createPipelineJobSettings() {
+ return getStubSettingsBuilder().createPipelineJobSettings();
+ }
+
+ /** Returns the builder for the settings used for calls to getPipelineJob. */
+ public UnaryCallSettings.Builder<GetPipelineJobRequest, PipelineJob> getPipelineJobSettings() {
+ return getStubSettingsBuilder().getPipelineJobSettings();
+ }
+
+ /** Returns the builder for the settings used for calls to listPipelineJobs. */
+ public PagedCallSettings.Builder<
+ ListPipelineJobsRequest, ListPipelineJobsResponse, ListPipelineJobsPagedResponse>
+ listPipelineJobsSettings() {
+ return getStubSettingsBuilder().listPipelineJobsSettings();
+ }
+
+ /** Returns the builder for the settings used for calls to deletePipelineJob. */
+ public UnaryCallSettings.Builder<DeletePipelineJobRequest, Operation>
+ deletePipelineJobSettings() {
+ return getStubSettingsBuilder().deletePipelineJobSettings();
+ }
+
+ /** Returns the builder for the settings used for calls to deletePipelineJob. */
+ public OperationCallSettings.Builder<DeletePipelineJobRequest, Empty, DeleteOperationMetadata>
+ deletePipelineJobOperationSettings() {
+ return getStubSettingsBuilder().deletePipelineJobOperationSettings();
+ }
+
+ /** Returns the builder for the settings used for calls to cancelPipelineJob. */
+ public UnaryCallSettings.Builder<CancelPipelineJobRequest, Empty> cancelPipelineJobSettings() {
+ return getStubSettingsBuilder().cancelPipelineJobSettings();
+ }
+
  @Override
  public PipelineServiceSettings build() throws IOException {
  return new PipelineServiceSettings(this);

google-cloud-aiplatform/src/main/java/com/google/cloud/aiplatform/v1/gapic_metadata.json

@@ -202,18 +202,33 @@
  "grpc": {
  "libraryClient": "PipelineServiceClient",
  "rpcs": {
+ "CancelPipelineJob": {
+ "methods": ["cancelPipelineJob", "cancelPipelineJob", "cancelPipelineJob", "cancelPipelineJobCallable"]
+ },
  "CancelTrainingPipeline": {
  "methods": ["cancelTrainingPipeline", "cancelTrainingPipeline", "cancelTrainingPipeline", "cancelTrainingPipelineCallable"]
  },
+ "CreatePipelineJob": {
+ "methods": ["createPipelineJob", "createPipelineJob", "createPipelineJob", "createPipelineJobCallable"]
+ },
  "CreateTrainingPipeline": {
  "methods": ["createTrainingPipeline", "createTrainingPipeline", "createTrainingPipeline", "createTrainingPipelineCallable"]
  },
+ "DeletePipelineJob": {
+ "methods": ["deletePipelineJobAsync", "deletePipelineJobAsync", "deletePipelineJobAsync", "deletePipelineJobOperationCallable", "deletePipelineJobCallable"]
+ },
  "DeleteTrainingPipeline": {
  "methods": ["deleteTrainingPipelineAsync", "deleteTrainingPipelineAsync", "deleteTrainingPipelineAsync", "deleteTrainingPipelineOperationCallable", "deleteTrainingPipelineCallable"]
  },
+ "GetPipelineJob": {
+ "methods": ["getPipelineJob", "getPipelineJob", "getPipelineJob", "getPipelineJobCallable"]
+ },
  "GetTrainingPipeline": {
  "methods": ["getTrainingPipeline", "getTrainingPipeline", "getTrainingPipeline", "getTrainingPipelineCallable"]
  },
+ "ListPipelineJobs": {
+ "methods": ["listPipelineJobs", "listPipelineJobs", "listPipelineJobs", "listPipelineJobsPagedCallable", "listPipelineJobsCallable"]
+ },
  "ListTrainingPipelines": {
  "methods": ["listTrainingPipelines", "listTrainingPipelines", "listTrainingPipelines", "listTrainingPipelinesPagedCallable", "listTrainingPipelinesCallable"]
  }

google-cloud-aiplatform/src/main/java/com/google/cloud/aiplatform/v1/stub/GrpcPipelineServiceStub.java

@@ -16,6 +16,7 @@
 
 package com.google.cloud.aiplatform.v1.stub;
 
+import static com.google.cloud.aiplatform.v1.PipelineServiceClient.ListPipelineJobsPagedResponse;
 import static com.google.cloud.aiplatform.v1.PipelineServiceClient.ListTrainingPipelinesPagedResponse;
 
 import com.google.api.gax.core.BackgroundResource;
@@ -26,13 +27,20 @@
 import com.google.api.gax.rpc.OperationCallable;
 import com.google.api.gax.rpc.RequestParamsExtractor;
 import com.google.api.gax.rpc.UnaryCallable;
+import com.google.cloud.aiplatform.v1.CancelPipelineJobRequest;
 import com.google.cloud.aiplatform.v1.CancelTrainingPipelineRequest;
+import com.google.cloud.aiplatform.v1.CreatePipelineJobRequest;
 import com.google.cloud.aiplatform.v1.CreateTrainingPipelineRequest;
 import com.google.cloud.aiplatform.v1.DeleteOperationMetadata;
+import com.google.cloud.aiplatform.v1.DeletePipelineJobRequest;
 import com.google.cloud.aiplatform.v1.DeleteTrainingPipelineRequest;
+import com.google.cloud.aiplatform.v1.GetPipelineJobRequest;
 import com.google.cloud.aiplatform.v1.GetTrainingPipelineRequest;
+import com.google.cloud.aiplatform.v1.ListPipelineJobsRequest;
+import com.google.cloud.aiplatform.v1.ListPipelineJobsResponse;
 import com.google.cloud.aiplatform.v1.ListTrainingPipelinesRequest;
 import com.google.cloud.aiplatform.v1.ListTrainingPipelinesResponse;
+import com.google.cloud.aiplatform.v1.PipelineJob;
 import com.google.cloud.aiplatform.v1.TrainingPipeline;
 import com.google.common.collect.ImmutableMap;
 import com.google.longrunning.Operation;
@@ -107,6 +115,57 @@ public class GrpcPipelineServiceStub extends PipelineServiceStub {
  .setResponseMarshaller(ProtoUtils.marshaller(Empty.getDefaultInstance()))
  .build();
 
+ private static final MethodDescriptor<CreatePipelineJobRequest, PipelineJob>
+ createPipelineJobMethodDescriptor =
+ MethodDescriptor.<CreatePipelineJobRequest, PipelineJob>newBuilder()
+ .setType(MethodDescriptor.MethodType.UNARY)
+ .setFullMethodName("google.cloud.aiplatform.v1.PipelineService/CreatePipelineJob")
+ .setRequestMarshaller(
+ ProtoUtils.marshaller(CreatePipelineJobRequest.getDefaultInstance()))
+ .setResponseMarshaller(ProtoUtils.marshaller(PipelineJob.getDefaultInstance()))
+ .build();
+
+ private static final MethodDescriptor<GetPipelineJobRequest, PipelineJob>
+ getPipelineJobMethodDescriptor =
+ MethodDescriptor.<GetPipelineJobRequest, PipelineJob>newBuilder()
+ .setType(MethodDescriptor.MethodType.UNARY)
+ .setFullMethodName("google.cloud.aiplatform.v1.PipelineService/GetPipelineJob")
+ .setRequestMarshaller(
+ ProtoUtils.marshaller(GetPipelineJobRequest.getDefaultInstance()))
+ .setResponseMarshaller(ProtoUtils.marshaller(PipelineJob.getDefaultInstance()))
+ .build();
+
+ private static final MethodDescriptor<ListPipelineJobsRequest, ListPipelineJobsResponse>
+ listPipelineJobsMethodDescriptor =
+ MethodDescriptor.<ListPipelineJobsRequest, ListPipelineJobsResponse>newBuilder()
+ .setType(MethodDescriptor.MethodType.UNARY)
+ .setFullMethodName("google.cloud.aiplatform.v1.PipelineService/ListPipelineJobs")
+ .setRequestMarshaller(
+ ProtoUtils.marshaller(ListPipelineJobsRequest.getDefaultInstance()))
+ .setResponseMarshaller(
+ ProtoUtils.marshaller(ListPipelineJobsResponse.getDefaultInstance()))
+ .build();
+
+ private static final MethodDescriptor<DeletePipelineJobRequest, Operation>
+ deletePipelineJobMethodDescriptor =
+ MethodDescriptor.<DeletePipelineJobRequest, Operation>newBuilder()
+ .setType(MethodDescriptor.MethodType.UNARY)
+ .setFullMethodName("google.cloud.aiplatform.v1.PipelineService/DeletePipelineJob")
+ .setRequestMarshaller(
+ ProtoUtils.marshaller(DeletePipelineJobRequest.getDefaultInstance()))
+ .setResponseMarshaller(ProtoUtils.marshaller(Operation.getDefaultInstance()))
+ .build();
+
+ private static final MethodDescriptor<CancelPipelineJobRequest, Empty>
+ cancelPipelineJobMethodDescriptor =
+ MethodDescriptor.<CancelPipelineJobRequest, Empty>newBuilder()
+ .setType(MethodDescriptor.MethodType.UNARY)
+ .setFullMethodName("google.cloud.aiplatform.v1.PipelineService/CancelPipelineJob")
+ .setRequestMarshaller(
+ ProtoUtils.marshaller(CancelPipelineJobRequest.getDefaultInstance()))
+ .setResponseMarshaller(ProtoUtils.marshaller(Empty.getDefaultInstance()))
+ .build();
+
  private final UnaryCallable<CreateTrainingPipelineRequest, TrainingPipeline>
  createTrainingPipelineCallable;
  private final UnaryCallable<GetTrainingPipelineRequest, TrainingPipeline>
@@ -120,6 +179,16 @@ public class GrpcPipelineServiceStub extends PipelineServiceStub {
  private final OperationCallable<DeleteTrainingPipelineRequest, Empty, DeleteOperationMetadata>
  deleteTrainingPipelineOperationCallable;
  private final UnaryCallable<CancelTrainingPipelineRequest, Empty> cancelTrainingPipelineCallable;
+ private final UnaryCallable<CreatePipelineJobRequest, PipelineJob> createPipelineJobCallable;
+ private final UnaryCallable<GetPipelineJobRequest, PipelineJob> getPipelineJobCallable;
+ private final UnaryCallable<ListPipelineJobsRequest, ListPipelineJobsResponse>
+ listPipelineJobsCallable;
+ private final UnaryCallable<ListPipelineJobsRequest, ListPipelineJobsPagedResponse>
+ listPipelineJobsPagedCallable;
+ private final UnaryCallable<DeletePipelineJobRequest, Operation> deletePipelineJobCallable;
+ private final OperationCallable<DeletePipelineJobRequest, Empty, DeleteOperationMetadata>
+ deletePipelineJobOperationCallable;
+ private final UnaryCallable<CancelPipelineJobRequest, Empty> cancelPipelineJobCallable;
 
  private final BackgroundResource backgroundResources;
  private final GrpcOperationsStub operationsStub;
@@ -235,6 +304,72 @@ public Map<String, String> extract(CancelTrainingPipelineRequest request) {
  }
  })
  .build();
+ GrpcCallSettings<CreatePipelineJobRequest, PipelineJob> createPipelineJobTransportSettings =
+ GrpcCallSettings.<CreatePipelineJobRequest, PipelineJob>newBuilder()
+ .setMethodDescriptor(createPipelineJobMethodDescriptor)
+ .setParamsExtractor(
+ new RequestParamsExtractor<CreatePipelineJobRequest>() {
+ @Override
+ public Map<String, String> extract(CreatePipelineJobRequest request) {
+ ImmutableMap.Builder<String, String> params = ImmutableMap.builder();
+ params.put("parent", String.valueOf(request.getParent()));
+ return params.build();
+ }
+ })
+ .build();
+ GrpcCallSettings<GetPipelineJobRequest, PipelineJob> getPipelineJobTransportSettings =
+ GrpcCallSettings.<GetPipelineJobRequest, PipelineJob>newBuilder()
+ .setMethodDescriptor(getPipelineJobMethodDescriptor)
+ .setParamsExtractor(
+ new RequestParamsExtractor<GetPipelineJobRequest>() {
+ @Override
+ public Map<String, String> extract(GetPipelineJobRequest request) {
+ ImmutableMap.Builder<String, String> params = ImmutableMap.builder();
+ params.put("name", String.valueOf(request.getName()));
+ return params.build();
+ }
+ })
+ .build();
+ GrpcCallSettings<ListPipelineJobsRequest, ListPipelineJobsResponse>
+ listPipelineJobsTransportSettings =
+ GrpcCallSettings.<ListPipelineJobsRequest, ListPipelineJobsResponse>newBuilder()
+ .setMethodDescriptor(listPipelineJobsMethodDescriptor)
+ .setParamsExtractor(
+ new RequestParamsExtractor<ListPipelineJobsRequest>() {
+ @Override
+ public Map<String, String> extract(ListPipelineJobsRequest request) {
+ ImmutableMap.Builder<String, String> params = ImmutableMap.builder();
+ params.put("parent", String.valueOf(request.getParent()));
+ return params.build();
+ }
+ })
+ .build();
+ GrpcCallSettings<DeletePipelineJobRequest, Operation> deletePipelineJobTransportSettings =
+ GrpcCallSettings.<DeletePipelineJobRequest, Operation>newBuilder()
+ .setMethodDescriptor(deletePipelineJobMethodDescriptor)
+ .setParamsExtractor(
+ new RequestParamsExtractor<DeletePipelineJobRequest>() {
+ @Override
+ public Map<String, String> extract(DeletePipelineJobRequest request) {
+ ImmutableMap.Builder<String, String> params = ImmutableMap.builder();
+ params.put("name", String.valueOf(request.getName()));
+ return params.build();
+ }
+ })
+ .build();
+ GrpcCallSettings<CancelPipelineJobRequest, Empty> cancelPipelineJobTransportSettings =
+ GrpcCallSettings.<CancelPipelineJobRequest, Empty>newBuilder()
+ .setMethodDescriptor(cancelPipelineJobMethodDescriptor)
+ .setParamsExtractor(
+ new RequestParamsExtractor<CancelPipelineJobRequest>() {
+ @Override
+ public Map<String, String> extract(CancelPipelineJobRequest request) {
+ ImmutableMap.Builder<String, String> params = ImmutableMap.builder();
+ params.put("name", String.valueOf(request.getName()));
+ return params.build();
+ }
+ })
+ .build();
 
  this.createTrainingPipelineCallable =
  callableFactory.createUnaryCallable(
@@ -272,6 +407,36 @@ public Map<String, String> extract(CancelTrainingPipelineRequest request) {
  cancelTrainingPipelineTransportSettings,
  settings.cancelTrainingPipelineSettings(),
  clientContext);
+ this.createPipelineJobCallable =
+ callableFactory.createUnaryCallable(
+ createPipelineJobTransportSettings,
+ settings.createPipelineJobSettings(),
+ clientContext);
+ this.getPipelineJobCallable =
+ callableFactory.createUnaryCallable(
+ getPipelineJobTransportSettings, settings.getPipelineJobSettings(), clientContext);
+ this.listPipelineJobsCallable =
+ callableFactory.createUnaryCallable(
+ listPipelineJobsTransportSettings, settings.listPipelineJobsSettings(), clientContext);
+ this.listPipelineJobsPagedCallable =
+ callableFactory.createPagedCallable(
+ listPipelineJobsTransportSettings, settings.listPipelineJobsSettings(), clientContext);
+ this.deletePipelineJobCallable =
+ callableFactory.createUnaryCallable(
+ deletePipelineJobTransportSettings,
+ settings.deletePipelineJobSettings(),
+ clientContext);
+ this.deletePipelineJobOperationCallable =
+ callableFactory.createOperationCallable(
+ deletePipelineJobTransportSettings,
+ settings.deletePipelineJobOperationSettings(),
+ clientContext,
+ operationsStub);
+ this.cancelPipelineJobCallable =
+ callableFactory.createUnaryCallable(
+ cancelPipelineJobTransportSettings,
+ settings.cancelPipelineJobSettings(),
+ clientContext);
 
  this.backgroundResources =
  new BackgroundResourceAggregation(clientContext.getBackgroundResources());
@@ -320,6 +485,44 @@ public UnaryCallable<CancelTrainingPipelineRequest, Empty> cancelTrainingPipelin
  return cancelTrainingPipelineCallable;
  }
 
+ @Override
+ public UnaryCallable<CreatePipelineJobRequest, PipelineJob> createPipelineJobCallable() {
+ return createPipelineJobCallable;
+ }
+
+ @Override
+ public UnaryCallable<GetPipelineJobRequest, PipelineJob> getPipelineJobCallable() {
+ return getPipelineJobCallable;
+ }
+
+ @Override
+ public UnaryCallable<ListPipelineJobsRequest, ListPipelineJobsResponse>
+ listPipelineJobsCallable() {
+ return listPipelineJobsCallable;
+ }
+
+ @Override
+ public UnaryCallable<ListPipelineJobsRequest, ListPipelineJobsPagedResponse>
+ listPipelineJobsPagedCallable() {
+ return listPipelineJobsPagedCallable;
+ }
+
+ @Override
+ public UnaryCallable<DeletePipelineJobRequest, Operation> deletePipelineJobCallable() {
+ return deletePipelineJobCallable;
+ }
+
+ @Override
+ public OperationCallable<DeletePipelineJobRequest, Empty, DeleteOperationMetadata>
+ deletePipelineJobOperationCallable() {
+ return deletePipelineJobOperationCallable;
+ }
+
+ @Override
+ public UnaryCallable<CancelPipelineJobRequest, Empty> cancelPipelineJobCallable() {
+ return cancelPipelineJobCallable;
+ }
+
  @Override
  public final void close() {
  shutdown();