-
Notifications
You must be signed in to change notification settings - Fork 1.1k
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
feat: [aiplatform] add ComputeTokens API (#10053)
- [ ] 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
1 parent
1961c40
commit 5352821
Showing
511 changed files
with
139,884 additions
and
3,475 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
3,039 changes: 3,039 additions & 0 deletions
3,039
...rm/src/main/java/com/google/cloud/aiplatform/v1/FeatureOnlineStoreAdminServiceClient.java
Large diffs are not rendered by default.
Oops, something went wrong.
526 changes: 526 additions & 0 deletions
526
.../src/main/java/com/google/cloud/aiplatform/v1/FeatureOnlineStoreAdminServiceSettings.java
Large diffs are not rendered by default.
Oops, something went wrong.
806 changes: 806 additions & 0 deletions
806
...latform/src/main/java/com/google/cloud/aiplatform/v1/FeatureOnlineStoreServiceClient.java
Large diffs are not rendered by default.
Oops, something went wrong.
258 changes: 258 additions & 0 deletions
258
...tform/src/main/java/com/google/cloud/aiplatform/v1/FeatureOnlineStoreServiceSettings.java
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
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); | ||
} | ||
} | ||
} |
Oops, something went wrong.