Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

feat: [aiplatform] add DatasetVersion and dataset version RPCs to DatasetService #9922

Merged
merged 14 commits into from Oct 9, 2023
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Jump to
Jump to file
Failed to load files.
Diff view
Diff view
4 changes: 2 additions & 2 deletions java-aiplatform/README.md
Expand Up @@ -20,7 +20,7 @@ If you are using Maven with [BOM][libraries-bom], add this to your pom.xml file:
<dependency>
<groupId>com.google.cloud</groupId>
<artifactId>libraries-bom</artifactId>
<version>26.23.0</version>
<version>26.24.0</version>
<type>pom</type>
<scope>import</scope>
</dependency>
Expand Down Expand Up @@ -195,7 +195,7 @@ Java is a registered trademark of Oracle and/or its affiliates.
[kokoro-badge-link-5]: http://storage.googleapis.com/cloud-devrel-public/java/badges/google-cloud-java/java11.html
[stability-image]: https://img.shields.io/badge/stability-stable-green
[maven-version-image]: https://img.shields.io/maven-central/v/com.google.cloud/google-cloud-aiplatform.svg
[maven-version-link]: https://central.sonatype.com/artifact/com.google.cloud/google-cloud-aiplatform/3.25.0
[maven-version-link]: https://central.sonatype.com/artifact/com.google.cloud/google-cloud-aiplatform/3.28.0
[authentication]: https://github.com/googleapis/google-cloud-java#authentication
[auth-scopes]: https://developers.google.com/identity/protocols/oauth2/scopes
[predefined-iam-roles]: https://cloud.google.com/iam/docs/understanding-roles#predefined_roles
Expand Down

Large diffs are not rendered by default.

Expand Up @@ -18,6 +18,7 @@

import static com.google.cloud.aiplatform.v1.DatasetServiceClient.ListAnnotationsPagedResponse;
import static com.google.cloud.aiplatform.v1.DatasetServiceClient.ListDataItemsPagedResponse;
import static com.google.cloud.aiplatform.v1.DatasetServiceClient.ListDatasetVersionsPagedResponse;
import static com.google.cloud.aiplatform.v1.DatasetServiceClient.ListDatasetsPagedResponse;
import static com.google.cloud.aiplatform.v1.DatasetServiceClient.ListLocationsPagedResponse;
import static com.google.cloud.aiplatform.v1.DatasetServiceClient.ListSavedQueriesPagedResponse;
Expand Down Expand Up @@ -151,6 +152,55 @@ public UnaryCallSettings<ExportDataRequest, Operation> exportDataSettings() {
return ((DatasetServiceStubSettings) getStubSettings()).exportDataOperationSettings();
}

/** Returns the object with the settings used for calls to createDatasetVersion. */
public UnaryCallSettings<CreateDatasetVersionRequest, Operation> createDatasetVersionSettings() {
return ((DatasetServiceStubSettings) getStubSettings()).createDatasetVersionSettings();
}

/** Returns the object with the settings used for calls to createDatasetVersion. */
public OperationCallSettings<
CreateDatasetVersionRequest, DatasetVersion, CreateDatasetVersionOperationMetadata>
createDatasetVersionOperationSettings() {
return ((DatasetServiceStubSettings) getStubSettings()).createDatasetVersionOperationSettings();
}

/** Returns the object with the settings used for calls to deleteDatasetVersion. */
public UnaryCallSettings<DeleteDatasetVersionRequest, Operation> deleteDatasetVersionSettings() {
return ((DatasetServiceStubSettings) getStubSettings()).deleteDatasetVersionSettings();
}

/** Returns the object with the settings used for calls to deleteDatasetVersion. */
public OperationCallSettings<DeleteDatasetVersionRequest, Empty, DeleteOperationMetadata>
deleteDatasetVersionOperationSettings() {
return ((DatasetServiceStubSettings) getStubSettings()).deleteDatasetVersionOperationSettings();
}

/** Returns the object with the settings used for calls to getDatasetVersion. */
public UnaryCallSettings<GetDatasetVersionRequest, DatasetVersion> getDatasetVersionSettings() {
return ((DatasetServiceStubSettings) getStubSettings()).getDatasetVersionSettings();
}

/** Returns the object with the settings used for calls to listDatasetVersions. */
public PagedCallSettings<
ListDatasetVersionsRequest, ListDatasetVersionsResponse, ListDatasetVersionsPagedResponse>
listDatasetVersionsSettings() {
return ((DatasetServiceStubSettings) getStubSettings()).listDatasetVersionsSettings();
}

/** Returns the object with the settings used for calls to restoreDatasetVersion. */
public UnaryCallSettings<RestoreDatasetVersionRequest, Operation>
restoreDatasetVersionSettings() {
return ((DatasetServiceStubSettings) getStubSettings()).restoreDatasetVersionSettings();
}

/** Returns the object with the settings used for calls to restoreDatasetVersion. */
public OperationCallSettings<
RestoreDatasetVersionRequest, DatasetVersion, RestoreDatasetVersionOperationMetadata>
restoreDatasetVersionOperationSettings() {
return ((DatasetServiceStubSettings) getStubSettings())
.restoreDatasetVersionOperationSettings();
}

/** Returns the object with the settings used for calls to listDataItems. */
public PagedCallSettings<ListDataItemsRequest, ListDataItemsResponse, ListDataItemsPagedResponse>
listDataItemsSettings() {
Expand Down Expand Up @@ -382,6 +432,60 @@ public UnaryCallSettings.Builder<ExportDataRequest, Operation> exportDataSetting
return getStubSettingsBuilder().exportDataOperationSettings();
}

/** Returns the builder for the settings used for calls to createDatasetVersion. */
public UnaryCallSettings.Builder<CreateDatasetVersionRequest, Operation>
createDatasetVersionSettings() {
return getStubSettingsBuilder().createDatasetVersionSettings();
}

/** Returns the builder for the settings used for calls to createDatasetVersion. */
public OperationCallSettings.Builder<
CreateDatasetVersionRequest, DatasetVersion, CreateDatasetVersionOperationMetadata>
createDatasetVersionOperationSettings() {
return getStubSettingsBuilder().createDatasetVersionOperationSettings();
}

/** Returns the builder for the settings used for calls to deleteDatasetVersion. */
public UnaryCallSettings.Builder<DeleteDatasetVersionRequest, Operation>
deleteDatasetVersionSettings() {
return getStubSettingsBuilder().deleteDatasetVersionSettings();
}

/** Returns the builder for the settings used for calls to deleteDatasetVersion. */
public OperationCallSettings.Builder<
DeleteDatasetVersionRequest, Empty, DeleteOperationMetadata>
deleteDatasetVersionOperationSettings() {
return getStubSettingsBuilder().deleteDatasetVersionOperationSettings();
}

/** Returns the builder for the settings used for calls to getDatasetVersion. */
public UnaryCallSettings.Builder<GetDatasetVersionRequest, DatasetVersion>
getDatasetVersionSettings() {
return getStubSettingsBuilder().getDatasetVersionSettings();
}

/** Returns the builder for the settings used for calls to listDatasetVersions. */
public PagedCallSettings.Builder<
ListDatasetVersionsRequest,
ListDatasetVersionsResponse,
ListDatasetVersionsPagedResponse>
listDatasetVersionsSettings() {
return getStubSettingsBuilder().listDatasetVersionsSettings();
}

/** Returns the builder for the settings used for calls to restoreDatasetVersion. */
public UnaryCallSettings.Builder<RestoreDatasetVersionRequest, Operation>
restoreDatasetVersionSettings() {
return getStubSettingsBuilder().restoreDatasetVersionSettings();
}

/** Returns the builder for the settings used for calls to restoreDatasetVersion. */
public OperationCallSettings.Builder<
RestoreDatasetVersionRequest, DatasetVersion, RestoreDatasetVersionOperationMetadata>
restoreDatasetVersionOperationSettings() {
return getStubSettingsBuilder().restoreDatasetVersionOperationSettings();
}

/** Returns the builder for the settings used for calls to listDataItems. */
public PagedCallSettings.Builder<
ListDataItemsRequest, ListDataItemsResponse, ListDataItemsPagedResponse>
Expand Down
Expand Up @@ -958,7 +958,7 @@ public final void cancelTrainingPipeline(CancelTrainingPipelineRequest request)
* @param pipelineJob Required. The PipelineJob to create.
* @param pipelineJobId The ID to use for the PipelineJob, which will become the final component
* of the PipelineJob name. If not provided, an ID will be automatically generated.
* <p>This value should be less than 128 characters, and valid characters are /[a-z][0-9]-/.
* <p>This value should be less than 128 characters, and valid characters are `/[a-z][0-9]-/`.
* @throws com.google.api.gax.rpc.ApiException if the remote call fails
*/
public final PipelineJob createPipelineJob(
Expand Down Expand Up @@ -998,7 +998,7 @@ public final PipelineJob createPipelineJob(
* @param pipelineJob Required. The PipelineJob to create.
* @param pipelineJobId The ID to use for the PipelineJob, which will become the final component
* of the PipelineJob name. If not provided, an ID will be automatically generated.
* <p>This value should be less than 128 characters, and valid characters are /[a-z][0-9]-/.
* <p>This value should be less than 128 characters, and valid characters are `/[a-z][0-9]-/`.
* @throws com.google.api.gax.rpc.ApiException if the remote call fails
*/
public final PipelineJob createPipelineJob(
Expand Down
Expand Up @@ -1174,7 +1174,7 @@ public final ReadTensorboardSizeResponse readTensorboardSize(ReadTensorboardSize
* @param tensorboardExperiment The TensorboardExperiment to create.
* @param tensorboardExperimentId Required. The ID to use for the Tensorboard experiment, which
* becomes the final component of the Tensorboard experiment's resource name.
* <p>This value should be 1-128 characters, and valid characters are /[a-z][0-9]-/.
* <p>This value should be 1-128 characters, and valid characters are `/[a-z][0-9]-/`.
* @throws com.google.api.gax.rpc.ApiException if the remote call fails
*/
public final TensorboardExperiment createTensorboardExperiment(
Expand Down Expand Up @@ -1220,7 +1220,7 @@ public final TensorboardExperiment createTensorboardExperiment(
* @param tensorboardExperiment The TensorboardExperiment to create.
* @param tensorboardExperimentId Required. The ID to use for the Tensorboard experiment, which
* becomes the final component of the Tensorboard experiment's resource name.
* <p>This value should be 1-128 characters, and valid characters are /[a-z][0-9]-/.
* <p>This value should be 1-128 characters, and valid characters are `/[a-z][0-9]-/`.
* @throws com.google.api.gax.rpc.ApiException if the remote call fails
*/
public final TensorboardExperiment createTensorboardExperiment(
Expand Down Expand Up @@ -1901,7 +1901,7 @@ public final OperationFuture<Empty, DeleteOperationMetadata> deleteTensorboardEx
* @param tensorboardRun Required. The TensorboardRun to create.
* @param tensorboardRunId Required. The ID to use for the Tensorboard run, which becomes the
* final component of the Tensorboard run's resource name.
* <p>This value should be 1-128 characters, and valid characters are /[a-z][0-9]-/.
* <p>This value should be 1-128 characters, and valid characters are `/[a-z][0-9]-/`.
* @throws com.google.api.gax.rpc.ApiException if the remote call fails
*/
public final TensorboardRun createTensorboardRun(
Expand Down Expand Up @@ -1944,7 +1944,7 @@ public final TensorboardRun createTensorboardRun(
* @param tensorboardRun Required. The TensorboardRun to create.
* @param tensorboardRunId Required. The ID to use for the Tensorboard run, which becomes the
* final component of the Tensorboard run's resource name.
* <p>This value should be 1-128 characters, and valid characters are /[a-z][0-9]-/.
* <p>This value should be 1-128 characters, and valid characters are `/[a-z][0-9]-/`.
* @throws com.google.api.gax.rpc.ApiException if the remote call fails
*/
public final TensorboardRun createTensorboardRun(
Expand Down
Expand Up @@ -13,9 +13,15 @@
"CreateDataset": {
"methods": ["createDatasetAsync", "createDatasetAsync", "createDatasetAsync", "createDatasetOperationCallable", "createDatasetCallable"]
},
"CreateDatasetVersion": {
"methods": ["createDatasetVersionAsync", "createDatasetVersionAsync", "createDatasetVersionAsync", "createDatasetVersionOperationCallable", "createDatasetVersionCallable"]
},
"DeleteDataset": {
"methods": ["deleteDatasetAsync", "deleteDatasetAsync", "deleteDatasetAsync", "deleteDatasetOperationCallable", "deleteDatasetCallable"]
},
"DeleteDatasetVersion": {
"methods": ["deleteDatasetVersionAsync", "deleteDatasetVersionAsync", "deleteDatasetVersionAsync", "deleteDatasetVersionOperationCallable", "deleteDatasetVersionCallable"]
},
"DeleteSavedQuery": {
"methods": ["deleteSavedQueryAsync", "deleteSavedQueryAsync", "deleteSavedQueryAsync", "deleteSavedQueryOperationCallable", "deleteSavedQueryCallable"]
},
Expand All @@ -28,6 +34,9 @@
"GetDataset": {
"methods": ["getDataset", "getDataset", "getDataset", "getDatasetCallable"]
},
"GetDatasetVersion": {
"methods": ["getDatasetVersion", "getDatasetVersion", "getDatasetVersion", "getDatasetVersionCallable"]
},
"GetIamPolicy": {
"methods": ["getIamPolicy", "getIamPolicyCallable"]
},
Expand All @@ -43,6 +52,9 @@
"ListDataItems": {
"methods": ["listDataItems", "listDataItems", "listDataItems", "listDataItemsPagedCallable", "listDataItemsCallable"]
},
"ListDatasetVersions": {
"methods": ["listDatasetVersions", "listDatasetVersions", "listDatasetVersions", "listDatasetVersionsPagedCallable", "listDatasetVersionsCallable"]
},
"ListDatasets": {
"methods": ["listDatasets", "listDatasets", "listDatasets", "listDatasetsPagedCallable", "listDatasetsCallable"]
},
Expand All @@ -52,6 +64,9 @@
"ListSavedQueries": {
"methods": ["listSavedQueries", "listSavedQueries", "listSavedQueries", "listSavedQueriesPagedCallable", "listSavedQueriesCallable"]
},
"RestoreDatasetVersion": {
"methods": ["restoreDatasetVersionAsync", "restoreDatasetVersionAsync", "restoreDatasetVersionAsync", "restoreDatasetVersionOperationCallable", "restoreDatasetVersionCallable"]
},
"SearchDataItems": {
"methods": ["searchDataItems", "searchDataItemsPagedCallable", "searchDataItemsCallable"]
},
Expand Down
Expand Up @@ -18,6 +18,7 @@

import static com.google.cloud.aiplatform.v1.DatasetServiceClient.ListAnnotationsPagedResponse;
import static com.google.cloud.aiplatform.v1.DatasetServiceClient.ListDataItemsPagedResponse;
import static com.google.cloud.aiplatform.v1.DatasetServiceClient.ListDatasetVersionsPagedResponse;
import static com.google.cloud.aiplatform.v1.DatasetServiceClient.ListDatasetsPagedResponse;
import static com.google.cloud.aiplatform.v1.DatasetServiceClient.ListLocationsPagedResponse;
import static com.google.cloud.aiplatform.v1.DatasetServiceClient.ListSavedQueriesPagedResponse;
Expand All @@ -29,26 +30,35 @@
import com.google.cloud.aiplatform.v1.AnnotationSpec;
import com.google.cloud.aiplatform.v1.CreateDatasetOperationMetadata;
import com.google.cloud.aiplatform.v1.CreateDatasetRequest;
import com.google.cloud.aiplatform.v1.CreateDatasetVersionOperationMetadata;
import com.google.cloud.aiplatform.v1.CreateDatasetVersionRequest;
import com.google.cloud.aiplatform.v1.Dataset;
import com.google.cloud.aiplatform.v1.DatasetVersion;
import com.google.cloud.aiplatform.v1.DeleteDatasetRequest;
import com.google.cloud.aiplatform.v1.DeleteDatasetVersionRequest;
import com.google.cloud.aiplatform.v1.DeleteOperationMetadata;
import com.google.cloud.aiplatform.v1.DeleteSavedQueryRequest;
import com.google.cloud.aiplatform.v1.ExportDataOperationMetadata;
import com.google.cloud.aiplatform.v1.ExportDataRequest;
import com.google.cloud.aiplatform.v1.ExportDataResponse;
import com.google.cloud.aiplatform.v1.GetAnnotationSpecRequest;
import com.google.cloud.aiplatform.v1.GetDatasetRequest;
import com.google.cloud.aiplatform.v1.GetDatasetVersionRequest;
import com.google.cloud.aiplatform.v1.ImportDataOperationMetadata;
import com.google.cloud.aiplatform.v1.ImportDataRequest;
import com.google.cloud.aiplatform.v1.ImportDataResponse;
import com.google.cloud.aiplatform.v1.ListAnnotationsRequest;
import com.google.cloud.aiplatform.v1.ListAnnotationsResponse;
import com.google.cloud.aiplatform.v1.ListDataItemsRequest;
import com.google.cloud.aiplatform.v1.ListDataItemsResponse;
import com.google.cloud.aiplatform.v1.ListDatasetVersionsRequest;
import com.google.cloud.aiplatform.v1.ListDatasetVersionsResponse;
import com.google.cloud.aiplatform.v1.ListDatasetsRequest;
import com.google.cloud.aiplatform.v1.ListDatasetsResponse;
import com.google.cloud.aiplatform.v1.ListSavedQueriesRequest;
import com.google.cloud.aiplatform.v1.ListSavedQueriesResponse;
import com.google.cloud.aiplatform.v1.RestoreDatasetVersionOperationMetadata;
import com.google.cloud.aiplatform.v1.RestoreDatasetVersionRequest;
import com.google.cloud.aiplatform.v1.SearchDataItemsRequest;
import com.google.cloud.aiplatform.v1.SearchDataItemsResponse;
import com.google.cloud.aiplatform.v1.UpdateDatasetRequest;
Expand Down Expand Up @@ -131,6 +141,52 @@ public UnaryCallable<ExportDataRequest, Operation> exportDataCallable() {
throw new UnsupportedOperationException("Not implemented: exportDataCallable()");
}

public OperationCallable<
CreateDatasetVersionRequest, DatasetVersion, CreateDatasetVersionOperationMetadata>
createDatasetVersionOperationCallable() {
throw new UnsupportedOperationException(
"Not implemented: createDatasetVersionOperationCallable()");
}

public UnaryCallable<CreateDatasetVersionRequest, Operation> createDatasetVersionCallable() {
throw new UnsupportedOperationException("Not implemented: createDatasetVersionCallable()");
}

public OperationCallable<DeleteDatasetVersionRequest, Empty, DeleteOperationMetadata>
deleteDatasetVersionOperationCallable() {
throw new UnsupportedOperationException(
"Not implemented: deleteDatasetVersionOperationCallable()");
}

public UnaryCallable<DeleteDatasetVersionRequest, Operation> deleteDatasetVersionCallable() {
throw new UnsupportedOperationException("Not implemented: deleteDatasetVersionCallable()");
}

public UnaryCallable<GetDatasetVersionRequest, DatasetVersion> getDatasetVersionCallable() {
throw new UnsupportedOperationException("Not implemented: getDatasetVersionCallable()");
}

public UnaryCallable<ListDatasetVersionsRequest, ListDatasetVersionsPagedResponse>
listDatasetVersionsPagedCallable() {
throw new UnsupportedOperationException("Not implemented: listDatasetVersionsPagedCallable()");
}

public UnaryCallable<ListDatasetVersionsRequest, ListDatasetVersionsResponse>
listDatasetVersionsCallable() {
throw new UnsupportedOperationException("Not implemented: listDatasetVersionsCallable()");
}

public OperationCallable<
RestoreDatasetVersionRequest, DatasetVersion, RestoreDatasetVersionOperationMetadata>
restoreDatasetVersionOperationCallable() {
throw new UnsupportedOperationException(
"Not implemented: restoreDatasetVersionOperationCallable()");
}

public UnaryCallable<RestoreDatasetVersionRequest, Operation> restoreDatasetVersionCallable() {
throw new UnsupportedOperationException("Not implemented: restoreDatasetVersionCallable()");
}

public UnaryCallable<ListDataItemsRequest, ListDataItemsPagedResponse>
listDataItemsPagedCallable() {
throw new UnsupportedOperationException("Not implemented: listDataItemsPagedCallable()");
Expand Down