feat: [aiplatform] add ComputeTokens API (#10053)

- [ ] Regenerate this pull request now. PiperOrigin-RevId: 582473450 Source-Link: https://togithub.com/googleapis/googleapis/commit/a9e4e72e32b550a5451030feae99d1602ee40e0d Source-Link: https://togithub.com/googleapis/googleapis-gen/commit/56ea7e1c067eed56256ba6310f44d6b94c67748a Copy-Tag: eyJwIjoiamF2YS1haXBsYXRmb3JtLy5Pd2xCb3QueWFtbCIsImgiOiI1NmVhN2UxYzA2N2VlZDU2MjU2YmE2MzEwZjQ0ZDZiOTRjNjc3NDhhIn0= BEGIN_NESTED_COMMIT feat: [aiplatform] add deployment_timeout to UploadModel ModelContainerSpec feat: add shared_memory_size_mb to UploadModel ModelContainerSpec feat: add startup_probe, health_probe to UploadModel ModelContainerSpec PiperOrigin-RevId: 582390619 Source-Link: https://togithub.com/googleapis/googleapis/commit/49c44ef1f0959f7d5c44a88357b18a9a057c580b Source-Link: https://togithub.com/googleapis/googleapis-gen/commit/7a1c0d15b82e062678a516d18551c67c1b0176aa Copy-Tag: eyJwIjoiamF2YS1haXBsYXRmb3JtLy5Pd2xCb3QueWFtbCIsImgiOiI3YTFjMGQxNWI4MmUwNjI2NzhhNTE2ZDE4NTUxYzY3YzFiMDE3NmFhIn0= END_NESTED_COMMIT BEGIN_NESTED_COMMIT feat: [aiplatform] add deployment_timeout to UploadModel ModelContainerSpec feat: add shared_memory_size_mb to UploadModel ModelContainerSpec feat: add startup_probe, health_probe to UploadModel ModelContainerSpec PiperOrigin-RevId: 582381401 Source-Link: https://togithub.com/googleapis/googleapis/commit/92d962d3814b061a905437e05ea1e249f7a30c4d Source-Link: https://togithub.com/googleapis/googleapis-gen/commit/c19c7a7fa3eac85508a2fe0afe177d76f510fcaf Copy-Tag: eyJwIjoiamF2YS1haXBsYXRmb3JtLy5Pd2xCb3QueWFtbCIsImgiOiJjMTljN2E3ZmEzZWFjODU1MDhhMmZlMGFmZTE3N2Q3NmY1MTBmY2FmIn0= END_NESTED_COMMIT BEGIN_NESTED_COMMIT feat: [aiplatform] add ComputeTokens and CountTokens API PiperOrigin-RevId: 582364082 Source-Link: https://togithub.com/googleapis/googleapis/commit/fdea53f759cb7a8e6167af8740ca9cfc850bfae9 Source-Link: https://togithub.com/googleapis/googleapis-gen/commit/295f8eab2a87743dccc65b88e73d71f480323023 Copy-Tag: eyJwIjoiamF2YS1haXBsYXRmb3JtLy5Pd2xCb3QueWFtbCIsImgiOiIyOTVmOGVhYjJhODc3NDNkY2NjNjViODhlNzNkNzFmNDgwMzIzMDIzIn0= END_NESTED_COMMIT BEGIN_NESTED_COMMIT feat: [aiplatform] add protected_artifact_location_id to CustomJob feat: add value_type, version_column_name to Feature feat: add FeatureGroup, FeatureOnlineStore, FeatureOnlineStoreAdminService, FeatureOnlineStoreService, FeatureRegistryService, FeatureView, FeatureViewSync feat: add NumericRestriction to Index feat: add tpu_topology to MachineSpec chore: change DeleteOperationMetadata to DeleteModelVersionOperationMetadata in ModelService feat: add CountTokensRequest to PredictionService feat: add StudyTimeConstraint to Study PiperOrigin-RevId: 581306827 Source-Link: https://togithub.com/googleapis/googleapis/commit/01af66acc8111ec1e4d6831e31952dabe530c694 Source-Link: https://togithub.com/googleapis/googleapis-gen/commit/842299dec7a7f5d7622b2a918aaa1888812d80dc Copy-Tag: eyJwIjoiamF2YS1haXBsYXRmb3JtLy5Pd2xCb3QueWFtbCIsImgiOiI4NDIyOTlkZWM3YTdmNWQ3NjIyYjJhOTE4YWFhMTg4ODgxMmQ4MGRjIn0= END_NESTED_COMMIT BEGIN_NESTED_COMMIT feat: [aiplatform] add protected_artifact_location_id to CustomJob chore: add comment fixes to Feature and FeaturestoreService feat!: add FeatureViewDataFormat, FeatureViewDataKey to FeatureOnlineStoreService feat: add NumericRestriction to Index feat: add tpu_topology to MachineResources feat: add BatchDeletePipelineJobs, BatchDeletePipelineJobsRequest and BatchDeletePipelineJobsResponse to PipelineService feat: add StudyTimeConstraint to Study PiperOrigin-RevId: 581306075 Source-Link: https://togithub.com/googleapis/googleapis/commit/7603960c7cf66ab44c4c3c75156f25dbceffc3f1 Source-Link: https://togithub.com/googleapis/googleapis-gen/commit/9d03dc661bc2b672029b6cc640a4d7f3561466e7 Copy-Tag: eyJwIjoiamF2YS1haXBsYXRmb3JtLy5Pd2xCb3QueWFtbCIsImgiOiI5ZDAzZGM2NjFiYzJiNjcyMDI5YjZjYzY0MGE0ZDdmMzU2MTQ2NmU3In0= END_NESTED_COMMIT

Author: gcf-owl-bot[bot]

java-aiplatform/README.md

@@ -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.26.0</version>
+ <version>26.27.0</version>
  <type>pom</type>
  <scope>import</scope>
  </dependency>
@@ -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.30.0
+[maven-version-link]: https://central.sonatype.com/artifact/com.google.cloud/google-cloud-aiplatform/3.31.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

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

@@ -0,0 +1,258 @@
+/*
+ * Copyright 2023 Google LLC
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * https://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package com.google.cloud.aiplatform.v1;
+
+import static com.google.cloud.aiplatform.v1.FeatureOnlineStoreServiceClient.ListLocationsPagedResponse;
+
+import com.google.api.core.ApiFunction;
+import com.google.api.core.BetaApi;
+import com.google.api.gax.core.GoogleCredentialsProvider;
+import com.google.api.gax.core.InstantiatingExecutorProvider;
+import com.google.api.gax.grpc.InstantiatingGrpcChannelProvider;
+import com.google.api.gax.rpc.ApiClientHeaderProvider;
+import com.google.api.gax.rpc.ClientContext;
+import com.google.api.gax.rpc.ClientSettings;
+import com.google.api.gax.rpc.PagedCallSettings;
+import com.google.api.gax.rpc.TransportChannelProvider;
+import com.google.api.gax.rpc.UnaryCallSettings;
+import com.google.cloud.aiplatform.v1.stub.FeatureOnlineStoreServiceStubSettings;
+import com.google.cloud.location.GetLocationRequest;
+import com.google.cloud.location.ListLocationsRequest;
+import com.google.cloud.location.ListLocationsResponse;
+import com.google.cloud.location.Location;
+import com.google.iam.v1.GetIamPolicyRequest;
+import com.google.iam.v1.Policy;
+import com.google.iam.v1.SetIamPolicyRequest;
+import com.google.iam.v1.TestIamPermissionsRequest;
+import com.google.iam.v1.TestIamPermissionsResponse;
+import java.io.IOException;
+import java.util.List;
+import javax.annotation.Generated;
+
+// AUTO-GENERATED DOCUMENTATION AND CLASS.
+/**
+ * Settings class to configure an instance of {@link FeatureOnlineStoreServiceClient}.
+ *
+ * <p>The default instance has everything set to sensible defaults:
+ *
+ * <ul>
+ * <li>The default service address (aiplatform.googleapis.com) and default port (443) are used.
+ * <li>Credentials are acquired automatically through Application Default Credentials.
+ * <li>Retries are configured for idempotent methods but not for non-idempotent methods.
+ * </ul>
+ *
+ * <p>The builder of this class is recursive, so contained classes are themselves builders. When
+ * build() is called, the tree of builders is called to create the complete settings object.
+ *
+ * <p>For example, to set the total timeout of fetchFeatureValues to 30 seconds:
+ *
+ * <pre>{@code
+ * // This snippet has been automatically generated and should be regarded as a code template only.
+ * // It will require modifications to work:
+ * // - It may require correct/in-range values for request initialization.
+ * // - It may require specifying regional endpoints when creating the service client as shown in
+ * // https://cloud.google.com/java/docs/setup#configure_endpoints_for_the_client_library
+ * FeatureOnlineStoreServiceSettings.Builder featureOnlineStoreServiceSettingsBuilder =
+ * FeatureOnlineStoreServiceSettings.newBuilder();
+ * featureOnlineStoreServiceSettingsBuilder
+ * .fetchFeatureValuesSettings()
+ * .setRetrySettings(
+ * featureOnlineStoreServiceSettingsBuilder
+ * .fetchFeatureValuesSettings()
+ * .getRetrySettings()
+ * .toBuilder()
+ * .setTotalTimeout(Duration.ofSeconds(30))
+ * .build());
+ * FeatureOnlineStoreServiceSettings featureOnlineStoreServiceSettings =
+ * featureOnlineStoreServiceSettingsBuilder.build();
+ * }</pre>
+ */
+@Generated("by gapic-generator-java")
+public class FeatureOnlineStoreServiceSettings
+ extends ClientSettings<FeatureOnlineStoreServiceSettings> {
+
+ /** Returns the object with the settings used for calls to fetchFeatureValues. */
+ public UnaryCallSettings<FetchFeatureValuesRequest, FetchFeatureValuesResponse>
+ fetchFeatureValuesSettings() {
+ return ((FeatureOnlineStoreServiceStubSettings) getStubSettings()).fetchFeatureValuesSettings();
+ }
+
+ /** Returns the object with the settings used for calls to listLocations. */
+ public PagedCallSettings<ListLocationsRequest, ListLocationsResponse, ListLocationsPagedResponse>
+ listLocationsSettings() {
+ return ((FeatureOnlineStoreServiceStubSettings) getStubSettings()).listLocationsSettings();
+ }
+
+ /** Returns the object with the settings used for calls to getLocation. */
+ public UnaryCallSettings<GetLocationRequest, Location> getLocationSettings() {
+ return ((FeatureOnlineStoreServiceStubSettings) getStubSettings()).getLocationSettings();
+ }
+
+ /** Returns the object with the settings used for calls to setIamPolicy. */
+ public UnaryCallSettings<SetIamPolicyRequest, Policy> setIamPolicySettings() {
+ return ((FeatureOnlineStoreServiceStubSettings) getStubSettings()).setIamPolicySettings();
+ }
+
+ /** Returns the object with the settings used for calls to getIamPolicy. */
+ public UnaryCallSettings<GetIamPolicyRequest, Policy> getIamPolicySettings() {
+ return ((FeatureOnlineStoreServiceStubSettings) getStubSettings()).getIamPolicySettings();
+ }
+
+ /** Returns the object with the settings used for calls to testIamPermissions. */
+ public UnaryCallSettings<TestIamPermissionsRequest, TestIamPermissionsResponse>
+ testIamPermissionsSettings() {
+ return ((FeatureOnlineStoreServiceStubSettings) getStubSettings()).testIamPermissionsSettings();
+ }
+
+ public static final FeatureOnlineStoreServiceSettings create(
+ FeatureOnlineStoreServiceStubSettings stub) throws IOException {
+ return new FeatureOnlineStoreServiceSettings.Builder(stub.toBuilder()).build();
+ }
+
+ /** Returns a builder for the default ExecutorProvider for this service. */
+ public static InstantiatingExecutorProvider.Builder defaultExecutorProviderBuilder() {
+ return FeatureOnlineStoreServiceStubSettings.defaultExecutorProviderBuilder();
+ }
+
+ /** Returns the default service endpoint. */
+ public static String getDefaultEndpoint() {
+ return FeatureOnlineStoreServiceStubSettings.getDefaultEndpoint();
+ }
+
+ /** Returns the default service scopes. */
+ public static List<String> getDefaultServiceScopes() {
+ return FeatureOnlineStoreServiceStubSettings.getDefaultServiceScopes();
+ }
+
+ /** Returns a builder for the default credentials for this service. */
+ public static GoogleCredentialsProvider.Builder defaultCredentialsProviderBuilder() {
+ return FeatureOnlineStoreServiceStubSettings.defaultCredentialsProviderBuilder();
+ }
+
+ /** Returns a builder for the default ChannelProvider for this service. */
+ public static InstantiatingGrpcChannelProvider.Builder defaultGrpcTransportProviderBuilder() {
+ return FeatureOnlineStoreServiceStubSettings.defaultGrpcTransportProviderBuilder();
+ }
+
+ public static TransportChannelProvider defaultTransportChannelProvider() {
+ return FeatureOnlineStoreServiceStubSettings.defaultTransportChannelProvider();
+ }
+
+ @BetaApi("The surface for customizing headers is not stable yet and may change in the future.")
+ public static ApiClientHeaderProvider.Builder defaultApiClientHeaderProviderBuilder() {
+ return FeatureOnlineStoreServiceStubSettings.defaultApiClientHeaderProviderBuilder();
+ }
+
+ /** Returns a new builder for this class. */
+ public static Builder newBuilder() {
+ return Builder.createDefault();
+ }
+
+ /** Returns a new builder for this class. */
+ public static Builder newBuilder(ClientContext clientContext) {
+ return new Builder(clientContext);
+ }
+
+ /** Returns a builder containing all the values of this settings class. */
+ public Builder toBuilder() {
+ return new Builder(this);
+ }
+
+ protected FeatureOnlineStoreServiceSettings(Builder settingsBuilder) throws IOException {
+ super(settingsBuilder);
+ }
+
+ /** Builder for FeatureOnlineStoreServiceSettings. */
+ public static class Builder
+ extends ClientSettings.Builder<FeatureOnlineStoreServiceSettings, Builder> {
+
+ protected Builder() throws IOException {
+ this(((ClientContext) null));
+ }
+
+ protected Builder(ClientContext clientContext) {
+ super(FeatureOnlineStoreServiceStubSettings.newBuilder(clientContext));
+ }
+
+ protected Builder(FeatureOnlineStoreServiceSettings settings) {
+ super(settings.getStubSettings().toBuilder());
+ }
+
+ protected Builder(FeatureOnlineStoreServiceStubSettings.Builder stubSettings) {
+ super(stubSettings);
+ }
+
+ private static Builder createDefault() {
+ return new Builder(FeatureOnlineStoreServiceStubSettings.newBuilder());
+ }
+
+ public FeatureOnlineStoreServiceStubSettings.Builder getStubSettingsBuilder() {
+ return ((FeatureOnlineStoreServiceStubSettings.Builder) getStubSettings());
+ }
+
+ /**
+ * Applies the given settings updater function to all of the unary API methods in this service.
+ *
+ * <p>Note: This method does not support applying settings to streaming methods.
+ */
+ public Builder applyToAllUnaryMethods(
+ ApiFunction<UnaryCallSettings.Builder<?, ?>, Void> settingsUpdater) {
+ super.applyToAllUnaryMethods(
+ getStubSettingsBuilder().unaryMethodSettingsBuilders(), settingsUpdater);
+ return this;
+ }
+
+ /** Returns the builder for the settings used for calls to fetchFeatureValues. */
+ public UnaryCallSettings.Builder<FetchFeatureValuesRequest, FetchFeatureValuesResponse>
+ fetchFeatureValuesSettings() {
+ return getStubSettingsBuilder().fetchFeatureValuesSettings();
+ }
+
+ /** Returns the builder for the settings used for calls to listLocations. */
+ public PagedCallSettings.Builder<
+ ListLocationsRequest, ListLocationsResponse, ListLocationsPagedResponse>
+ listLocationsSettings() {
+ return getStubSettingsBuilder().listLocationsSettings();
+ }
+
+ /** Returns the builder for the settings used for calls to getLocation. */
+ public UnaryCallSettings.Builder<GetLocationRequest, Location> getLocationSettings() {
+ return getStubSettingsBuilder().getLocationSettings();
+ }
+
+ /** Returns the builder for the settings used for calls to setIamPolicy. */
+ public UnaryCallSettings.Builder<SetIamPolicyRequest, Policy> setIamPolicySettings() {
+ return getStubSettingsBuilder().setIamPolicySettings();
+ }
+
+ /** Returns the builder for the settings used for calls to getIamPolicy. */
+ public UnaryCallSettings.Builder<GetIamPolicyRequest, Policy> getIamPolicySettings() {
+ return getStubSettingsBuilder().getIamPolicySettings();
+ }
+
+ /** Returns the builder for the settings used for calls to testIamPermissions. */
+ public UnaryCallSettings.Builder<TestIamPermissionsRequest, TestIamPermissionsResponse>
+ testIamPermissionsSettings() {
+ return getStubSettingsBuilder().testIamPermissionsSettings();
+ }
+
+ @Override
+ public FeatureOnlineStoreServiceSettings build() throws IOException {
+ return new FeatureOnlineStoreServiceSettings(this);
+ }
+ }
+}