Skip to content
This repository was archived by the owner on Sep 9, 2023. It is now read-only.
Closed
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view

Large diffs are not rendered by default.

Original file line number Diff line number Diff line change
Expand Up @@ -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;
Expand Down Expand Up @@ -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();
Expand Down Expand Up @@ -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);
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -37,9 +37,9 @@
* <pre>{@code
* try (PredictionServiceClient predictionServiceClient = PredictionServiceClient.create()) {
* EndpointName endpoint = EndpointName.of("[PROJECT]", "[LOCATION]", "[ENDPOINT]");
* List<Value> instances = new ArrayList<>();
* Value parameters = Value.newBuilder().build();
* PredictResponse response = predictionServiceClient.predict(endpoint, instances, parameters);
* List<Value> instances = new ArrayList<>();
* PredictResponse response = predictionServiceClient.predict(endpoint, parameters, instances);
* }
* }</pre>
*
Expand Down Expand Up @@ -154,14 +154,19 @@ public PredictionServiceStub getStub() {
* <pre>{@code
* try (PredictionServiceClient predictionServiceClient = PredictionServiceClient.create()) {
* EndpointName endpoint = EndpointName.of("[PROJECT]", "[LOCATION]", "[ENDPOINT]");
* List<Value> instances = new ArrayList<>();
* Value parameters = Value.newBuilder().build();
* PredictResponse response = predictionServiceClient.predict(endpoint, instances, parameters);
* List<Value> instances = new ArrayList<>();
* PredictResponse response = predictionServiceClient.predict(endpoint, parameters, instances);
* }
* }</pre>
*
* @param endpoint Required. The name of the Endpoint requested to serve the prediction. Format:
* `projects/{project}/locations/{location}/endpoints/{endpoint}`
* @param parameters The parameters that govern the prediction. The schema of the parameters may
* be specified via Endpoint's DeployedModels' [Model's
* ][google.cloud.aiplatform.v1.DeployedModel.model]
* [PredictSchemata's][google.cloud.aiplatform.v1.Model.predict_schemata]
* [parameters_schema_uri][google.cloud.aiplatform.v1.PredictSchemata.parameters_schema_uri].
* @param instances Required. The instances that are the input to the prediction call. A
* DeployedModel may have an upper limit on the number of instances it supports per request,
* and when it is exceeded the prediction call errors in case of AutoML Models, or, in case of
Expand All @@ -170,20 +175,15 @@ public PredictionServiceStub getStub() {
* [Model's][google.cloud.aiplatform.v1.DeployedModel.model]
* [PredictSchemata's][google.cloud.aiplatform.v1.Model.predict_schemata]
* [instance_schema_uri][google.cloud.aiplatform.v1.PredictSchemata.instance_schema_uri].
* @param parameters The parameters that govern the prediction. The schema of the parameters may
* be specified via Endpoint's DeployedModels' [Model's
* ][google.cloud.aiplatform.v1.DeployedModel.model]
* [PredictSchemata's][google.cloud.aiplatform.v1.Model.predict_schemata]
* [parameters_schema_uri][google.cloud.aiplatform.v1.PredictSchemata.parameters_schema_uri].
* @throws com.google.api.gax.rpc.ApiException if the remote call fails
*/
public final PredictResponse predict(
EndpointName endpoint, List<Value> instances, Value parameters) {
EndpointName endpoint, Value parameters, List<Value> instances) {
PredictRequest request =
PredictRequest.newBuilder()
.setEndpoint(endpoint == null ? null : endpoint.toString())
.addAllInstances(instances)
.setParameters(parameters)
.addAllInstances(instances)
.build();
return predict(request);
}
Expand All @@ -197,14 +197,19 @@ public final PredictResponse predict(
* <pre>{@code
* try (PredictionServiceClient predictionServiceClient = PredictionServiceClient.create()) {
* String endpoint = EndpointName.of("[PROJECT]", "[LOCATION]", "[ENDPOINT]").toString();
* List<Value> instances = new ArrayList<>();
* Value parameters = Value.newBuilder().build();
* PredictResponse response = predictionServiceClient.predict(endpoint, instances, parameters);
* List<Value> instances = new ArrayList<>();
* PredictResponse response = predictionServiceClient.predict(endpoint, parameters, instances);
* }
* }</pre>
*
* @param endpoint Required. The name of the Endpoint requested to serve the prediction. Format:
* `projects/{project}/locations/{location}/endpoints/{endpoint}`
* @param parameters The parameters that govern the prediction. The schema of the parameters may
* be specified via Endpoint's DeployedModels' [Model's
* ][google.cloud.aiplatform.v1.DeployedModel.model]
* [PredictSchemata's][google.cloud.aiplatform.v1.Model.predict_schemata]
* [parameters_schema_uri][google.cloud.aiplatform.v1.PredictSchemata.parameters_schema_uri].
* @param instances Required. The instances that are the input to the prediction call. A
* DeployedModel may have an upper limit on the number of instances it supports per request,
* and when it is exceeded the prediction call errors in case of AutoML Models, or, in case of
Expand All @@ -213,19 +218,14 @@ public final PredictResponse predict(
* [Model's][google.cloud.aiplatform.v1.DeployedModel.model]
* [PredictSchemata's][google.cloud.aiplatform.v1.Model.predict_schemata]
* [instance_schema_uri][google.cloud.aiplatform.v1.PredictSchemata.instance_schema_uri].
* @param parameters The parameters that govern the prediction. The schema of the parameters may
* be specified via Endpoint's DeployedModels' [Model's
* ][google.cloud.aiplatform.v1.DeployedModel.model]
* [PredictSchemata's][google.cloud.aiplatform.v1.Model.predict_schemata]
* [parameters_schema_uri][google.cloud.aiplatform.v1.PredictSchemata.parameters_schema_uri].
* @throws com.google.api.gax.rpc.ApiException if the remote call fails
*/
public final PredictResponse predict(String endpoint, List<Value> instances, Value parameters) {
public final PredictResponse predict(String endpoint, Value parameters, List<Value> instances) {
PredictRequest request =
PredictRequest.newBuilder()
.setEndpoint(endpoint)
.addAllInstances(instances)
.setParameters(parameters)
.addAllInstances(instances)
.build();
return predict(request);
}
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -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"]
}
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -109,9 +109,9 @@
* <pre>{@code
* try (PredictionServiceClient predictionServiceClient = PredictionServiceClient.create()) {
* EndpointName endpoint = EndpointName.of("[PROJECT]", "[LOCATION]", "[ENDPOINT]");
* List<Value> instances = new ArrayList<>();
* Value parameters = Value.newBuilder().build();
* PredictResponse response = predictionServiceClient.predict(endpoint, instances, parameters);
* List<Value> instances = new ArrayList<>();
* PredictResponse response = predictionServiceClient.predict(endpoint, parameters, instances);
* }
* }</pre>
*
Expand Down
Loading