Skip to content

Commit

Permalink
feat: [aiplatform] add ComputeTokens API (#10053)
Browse files Browse the repository at this point in the history
- [ ] Regenerate this pull request now.

PiperOrigin-RevId: 582473450

Source-Link: https://github.com/googleapis/googleapis/commit/a9e4e72e32b550a5451030feae99d1602ee40e0d

Source-Link: https://github.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://github.com/googleapis/googleapis/commit/49c44ef1f0959f7d5c44a88357b18a9a057c580b

Source-Link: https://github.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://github.com/googleapis/googleapis/commit/92d962d3814b061a905437e05ea1e249f7a30c4d

Source-Link: https://github.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://github.com/googleapis/googleapis/commit/fdea53f759cb7a8e6167af8740ca9cfc850bfae9

Source-Link: https://github.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://github.com/googleapis/googleapis/commit/01af66acc8111ec1e4d6831e31952dabe530c694

Source-Link: https://github.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://github.com/googleapis/googleapis/commit/7603960c7cf66ab44c4c3c75156f25dbceffc3f1

Source-Link: https://github.com/googleapis/googleapis-gen/commit/9d03dc661bc2b672029b6cc640a4d7f3561466e7
Copy-Tag: eyJwIjoiamF2YS1haXBsYXRmb3JtLy5Pd2xCb3QueWFtbCIsImgiOiI5ZDAzZGM2NjFiYzJiNjcyMDI5YjZjYzY0MGE0ZDdmMzU2MTQ2NmU3In0=
END_NESTED_COMMIT
  • Loading branch information
gcf-owl-bot[bot] authored Nov 16, 2023
1 parent 1961c40 commit 5352821
Show file tree
Hide file tree
Showing 511 changed files with 139,884 additions and 3,475 deletions.
4 changes: 2 additions & 2 deletions java-aiplatform/README.md
Original file line number Diff line number Diff line change
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.26.0</version>
<version>26.27.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.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
Expand Down

Large diffs are not rendered by default.

Large diffs are not rendered by default.

Large diffs are not rendered by default.

Original file line number Diff line number Diff line change
@@ -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);
}
}
}
Loading

0 comments on commit 5352821

Please sign in to comment.