Skip to content

Commit

Permalink
feat: [aiplatform] add blocking_operation_ids to ImportFeatureValuesO…
Browse files Browse the repository at this point in the history
…perationMetadata (#9481)

* feat: support for Model Garden -- A single place to search, discover, and interact with a wide variety of foundation models from Google and Google partners, available on Vertex AI

PiperOrigin-RevId: 536794833

Source-Link: googleapis/googleapis@df07060

Source-Link: googleapis/googleapis-gen@d9492ce
Copy-Tag: eyJwIjoiamF2YS1haXBsYXRmb3JtLy5Pd2xCb3QueWFtbCIsImgiOiJkOTQ5MmNlOTAxZmM4YmMyOTg3MjgwZjE2YTQ2NmRhMWI0NDFiYzc1In0=

* 🦉 Updates from OwlBot post-processor

See https://github.com/googleapis/repo-automation-bots/blob/main/packages/owl-bot/README.md

* 🦉 Updates from OwlBot post-processor

See https://github.com/googleapis/repo-automation-bots/blob/main/packages/owl-bot/README.md

* chore: upgrading protobuf to 23.2 for code generation

PiperOrigin-RevId: 537035741

Source-Link: googleapis/googleapis@640f743

Source-Link: googleapis/googleapis-gen@5ec02ed
Copy-Tag: eyJwIjoiamF2YS1haXBsYXRmb3JtLy5Pd2xCb3QueWFtbCIsImgiOiI1ZWMwMmVkMjAyMjg3OGY5OWM5NGM3NjMwNjBmZjZhMDE4ODk2YjZlIn0=

* 🦉 Updates from OwlBot post-processor

See https://github.com/googleapis/repo-automation-bots/blob/main/packages/owl-bot/README.md

* 🦉 Updates from OwlBot post-processor

See https://github.com/googleapis/repo-automation-bots/blob/main/packages/owl-bot/README.md

* 🦉 Updates from OwlBot post-processor

See https://github.com/googleapis/repo-automation-bots/blob/main/packages/owl-bot/README.md

* feat: add NVIDIA_A100_80GB to AcceleratorType
feat: add blocking_operation_ids to ImportFeatureValuesOperationMetadata

PiperOrigin-RevId: 537124229

Source-Link: googleapis/googleapis@064f6cd

Source-Link: googleapis/googleapis-gen@b5a4a98
Copy-Tag: eyJwIjoiamF2YS1haXBsYXRmb3JtLy5Pd2xCb3QueWFtbCIsImgiOiJiNWE0YTk4MTI1NmI4NjNmNjk2MGFhYTZmMWEwM2E2NTUyMWQwMDJjIn0=

* 🦉 Updates from OwlBot post-processor

See https://github.com/googleapis/repo-automation-bots/blob/main/packages/owl-bot/README.md

* 🦉 Updates from OwlBot post-processor

See https://github.com/googleapis/repo-automation-bots/blob/main/packages/owl-bot/README.md

* feat: add blocking_operation_ids to ImportFeatureValuesOperationMetadata
feat: add launch_stage to PublisherModel

PiperOrigin-RevId: 537154438

Source-Link: googleapis/googleapis@d2d95e5

Source-Link: googleapis/googleapis-gen@2688bd0
Copy-Tag: eyJwIjoiamF2YS1haXBsYXRmb3JtLy5Pd2xCb3QueWFtbCIsImgiOiIyNjg4YmQwMWZjODIxZWYwMjI1NDM5NTUzMDllOWMwNTZjZWM4ZDU2In0=

* 🦉 Updates from OwlBot post-processor

See https://github.com/googleapis/repo-automation-bots/blob/main/packages/owl-bot/README.md

* 🦉 Updates from OwlBot post-processor

See https://github.com/googleapis/repo-automation-bots/blob/main/packages/owl-bot/README.md

---------

Co-authored-by: Owl Bot <gcf-owl-bot[bot]@users.noreply.github.com>
  • Loading branch information
gcf-owl-bot[bot] and gcf-owl-bot[bot] committed Jun 6, 2023
1 parent d06bf1f commit ba2c590
Show file tree
Hide file tree
Showing 1,607 changed files with 47,394 additions and 19,463 deletions.
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.15.0</version>
<version>26.16.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.18.0
[maven-version-link]: https://central.sonatype.com/artifact/com.google.cloud/google-cloud-aiplatform/3.19.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
Expand Up @@ -1272,7 +1272,6 @@ public final UnaryCallable<ListArtifactsRequest, ListArtifactsResponse> listArti
* Artifact to be updated. Format:
* `projects/{project}/locations/{location}/metadataStores/{metadatastore}/artifacts/{artifact}`
* @param updateMask Optional. A FieldMask indicating which fields should be updated.
* Functionality of this field is not yet supported.
* @throws com.google.api.gax.rpc.ApiException if the remote call fails
*/
public final Artifact updateArtifact(Artifact artifact, FieldMask updateMask) {
Expand Down Expand Up @@ -2110,7 +2109,6 @@ public final UnaryCallable<ListContextsRequest, ListContextsResponse> listContex
* Context to be updated. Format:
* `projects/{project}/locations/{location}/metadataStores/{metadatastore}/contexts/{context}`
* @param updateMask Optional. A FieldMask indicating which fields should be updated.
* Functionality of this field is not yet supported.
* @throws com.google.api.gax.rpc.ApiException if the remote call fails
*/
public final Context updateContext(Context context, FieldMask updateMask) {
Expand Down Expand Up @@ -3537,7 +3535,6 @@ public final ListExecutionsPagedResponse listExecutions(ListExecutionsRequest re
* Execution to be updated. Format:
* `projects/{project}/locations/{location}/metadataStores/{metadatastore}/executions/{execution}`
* @param updateMask Optional. A FieldMask indicating which fields should be updated.
* Functionality of this field is not yet supported.
* @throws com.google.api.gax.rpc.ApiException if the remote call fails
*/
public final Execution updateExecution(Execution execution, FieldMask updateMask) {
Expand Down

Large diffs are not rendered by default.

@@ -0,0 +1,255 @@
/*
* Copyright 2022 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.ModelGardenServiceClient.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.ModelGardenServiceStubSettings;
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 ModelGardenServiceClient}.
*
* <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 getPublisherModel 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
* ModelGardenServiceSettings.Builder modelGardenServiceSettingsBuilder =
* ModelGardenServiceSettings.newBuilder();
* modelGardenServiceSettingsBuilder
* .getPublisherModelSettings()
* .setRetrySettings(
* modelGardenServiceSettingsBuilder
* .getPublisherModelSettings()
* .getRetrySettings()
* .toBuilder()
* .setTotalTimeout(Duration.ofSeconds(30))
* .build());
* ModelGardenServiceSettings modelGardenServiceSettings =
* modelGardenServiceSettingsBuilder.build();
* }</pre>
*/
@Generated("by gapic-generator-java")
public class ModelGardenServiceSettings extends ClientSettings<ModelGardenServiceSettings> {

/** Returns the object with the settings used for calls to getPublisherModel. */
public UnaryCallSettings<GetPublisherModelRequest, PublisherModel> getPublisherModelSettings() {
return ((ModelGardenServiceStubSettings) getStubSettings()).getPublisherModelSettings();
}

/** Returns the object with the settings used for calls to listLocations. */
public PagedCallSettings<ListLocationsRequest, ListLocationsResponse, ListLocationsPagedResponse>
listLocationsSettings() {
return ((ModelGardenServiceStubSettings) getStubSettings()).listLocationsSettings();
}

/** Returns the object with the settings used for calls to getLocation. */
public UnaryCallSettings<GetLocationRequest, Location> getLocationSettings() {
return ((ModelGardenServiceStubSettings) getStubSettings()).getLocationSettings();
}

/** Returns the object with the settings used for calls to setIamPolicy. */
public UnaryCallSettings<SetIamPolicyRequest, Policy> setIamPolicySettings() {
return ((ModelGardenServiceStubSettings) getStubSettings()).setIamPolicySettings();
}

/** Returns the object with the settings used for calls to getIamPolicy. */
public UnaryCallSettings<GetIamPolicyRequest, Policy> getIamPolicySettings() {
return ((ModelGardenServiceStubSettings) getStubSettings()).getIamPolicySettings();
}

/** Returns the object with the settings used for calls to testIamPermissions. */
public UnaryCallSettings<TestIamPermissionsRequest, TestIamPermissionsResponse>
testIamPermissionsSettings() {
return ((ModelGardenServiceStubSettings) getStubSettings()).testIamPermissionsSettings();
}

public static final ModelGardenServiceSettings create(ModelGardenServiceStubSettings stub)
throws IOException {
return new ModelGardenServiceSettings.Builder(stub.toBuilder()).build();
}

/** Returns a builder for the default ExecutorProvider for this service. */
public static InstantiatingExecutorProvider.Builder defaultExecutorProviderBuilder() {
return ModelGardenServiceStubSettings.defaultExecutorProviderBuilder();
}

/** Returns the default service endpoint. */
public static String getDefaultEndpoint() {
return ModelGardenServiceStubSettings.getDefaultEndpoint();
}

/** Returns the default service scopes. */
public static List<String> getDefaultServiceScopes() {
return ModelGardenServiceStubSettings.getDefaultServiceScopes();
}

/** Returns a builder for the default credentials for this service. */
public static GoogleCredentialsProvider.Builder defaultCredentialsProviderBuilder() {
return ModelGardenServiceStubSettings.defaultCredentialsProviderBuilder();
}

/** Returns a builder for the default ChannelProvider for this service. */
public static InstantiatingGrpcChannelProvider.Builder defaultGrpcTransportProviderBuilder() {
return ModelGardenServiceStubSettings.defaultGrpcTransportProviderBuilder();
}

public static TransportChannelProvider defaultTransportChannelProvider() {
return ModelGardenServiceStubSettings.defaultTransportChannelProvider();
}

@BetaApi("The surface for customizing headers is not stable yet and may change in the future.")
public static ApiClientHeaderProvider.Builder defaultApiClientHeaderProviderBuilder() {
return ModelGardenServiceStubSettings.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 ModelGardenServiceSettings(Builder settingsBuilder) throws IOException {
super(settingsBuilder);
}

/** Builder for ModelGardenServiceSettings. */
public static class Builder extends ClientSettings.Builder<ModelGardenServiceSettings, Builder> {

protected Builder() throws IOException {
this(((ClientContext) null));
}

protected Builder(ClientContext clientContext) {
super(ModelGardenServiceStubSettings.newBuilder(clientContext));
}

protected Builder(ModelGardenServiceSettings settings) {
super(settings.getStubSettings().toBuilder());
}

protected Builder(ModelGardenServiceStubSettings.Builder stubSettings) {
super(stubSettings);
}

private static Builder createDefault() {
return new Builder(ModelGardenServiceStubSettings.newBuilder());
}

public ModelGardenServiceStubSettings.Builder getStubSettingsBuilder() {
return ((ModelGardenServiceStubSettings.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 getPublisherModel. */
public UnaryCallSettings.Builder<GetPublisherModelRequest, PublisherModel>
getPublisherModelSettings() {
return getStubSettingsBuilder().getPublisherModelSettings();
}

/** 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 ModelGardenServiceSettings build() throws IOException {
return new ModelGardenServiceSettings(this);
}
}
}
Expand Up @@ -544,8 +544,7 @@ public final UnaryCallable<RawPredictRequest, HttpBody> rawPredictCallable() {
* [explanation_spec][google.cloud.aiplatform.v1.DeployedModel.explanation_spec] populated. If
* [deployed_model_id][google.cloud.aiplatform.v1.ExplainRequest.deployed_model_id] is not
* specified, all DeployedModels must have
* [explanation_spec][google.cloud.aiplatform.v1.DeployedModel.explanation_spec] populated. Only
* deployed AutoML tabular Models have explanation_spec.
* [explanation_spec][google.cloud.aiplatform.v1.DeployedModel.explanation_spec] populated.
*
* <p>Sample code:
*
Expand Down Expand Up @@ -607,8 +606,7 @@ public final ExplainResponse explain(
* [explanation_spec][google.cloud.aiplatform.v1.DeployedModel.explanation_spec] populated. If
* [deployed_model_id][google.cloud.aiplatform.v1.ExplainRequest.deployed_model_id] is not
* specified, all DeployedModels must have
* [explanation_spec][google.cloud.aiplatform.v1.DeployedModel.explanation_spec] populated. Only
* deployed AutoML tabular Models have explanation_spec.
* [explanation_spec][google.cloud.aiplatform.v1.DeployedModel.explanation_spec] populated.
*
* <p>Sample code:
*
Expand Down Expand Up @@ -671,8 +669,7 @@ public final ExplainResponse explain(
* [explanation_spec][google.cloud.aiplatform.v1.DeployedModel.explanation_spec] populated. If
* [deployed_model_id][google.cloud.aiplatform.v1.ExplainRequest.deployed_model_id] is not
* specified, all DeployedModels must have
* [explanation_spec][google.cloud.aiplatform.v1.DeployedModel.explanation_spec] populated. Only
* deployed AutoML tabular Models have explanation_spec.
* [explanation_spec][google.cloud.aiplatform.v1.DeployedModel.explanation_spec] populated.
*
* <p>Sample code:
*
Expand Down Expand Up @@ -714,8 +711,7 @@ public final ExplainResponse explain(ExplainRequest request) {
* [explanation_spec][google.cloud.aiplatform.v1.DeployedModel.explanation_spec] populated. If
* [deployed_model_id][google.cloud.aiplatform.v1.ExplainRequest.deployed_model_id] is not
* specified, all DeployedModels must have
* [explanation_spec][google.cloud.aiplatform.v1.DeployedModel.explanation_spec] populated. Only
* deployed AutoML tabular Models have explanation_spec.
* [explanation_spec][google.cloud.aiplatform.v1.DeployedModel.explanation_spec] populated.
*
* <p>Sample code:
*
Expand Down
Expand Up @@ -812,7 +812,7 @@ public final UnaryCallable<LookupStudyRequest, Study> lookupStudyCallable() {
* Adds one or more Trials to a Study, with parameter values suggested by Vertex AI Vizier.
* Returns a long-running operation associated with the generation of Trial suggestions. When this
* long-running operation succeeds, it will contain a
* [SuggestTrialsResponse][google.cloud.ml.v1.SuggestTrialsResponse].
* [SuggestTrialsResponse][google.cloud.aiplatform.v1.SuggestTrialsResponse].
*
* <p>Sample code:
*
Expand Down Expand Up @@ -846,7 +846,7 @@ public final OperationFuture<SuggestTrialsResponse, SuggestTrialsMetadata> sugge
* Adds one or more Trials to a Study, with parameter values suggested by Vertex AI Vizier.
* Returns a long-running operation associated with the generation of Trial suggestions. When this
* long-running operation succeeds, it will contain a
* [SuggestTrialsResponse][google.cloud.ml.v1.SuggestTrialsResponse].
* [SuggestTrialsResponse][google.cloud.aiplatform.v1.SuggestTrialsResponse].
*
* <p>Sample code:
*
Expand Down Expand Up @@ -880,7 +880,7 @@ public final OperationFuture<SuggestTrialsResponse, SuggestTrialsMetadata> sugge
* Adds one or more Trials to a Study, with parameter values suggested by Vertex AI Vizier.
* Returns a long-running operation associated with the generation of Trial suggestions. When this
* long-running operation succeeds, it will contain a
* [SuggestTrialsResponse][google.cloud.ml.v1.SuggestTrialsResponse].
* [SuggestTrialsResponse][google.cloud.aiplatform.v1.SuggestTrialsResponse].
*
* <p>Sample code:
*
Expand Down
Expand Up @@ -710,6 +710,33 @@
}
}
},
"ModelGardenService": {
"clients": {
"grpc": {
"libraryClient": "ModelGardenServiceClient",
"rpcs": {
"GetIamPolicy": {
"methods": ["getIamPolicy", "getIamPolicyCallable"]
},
"GetLocation": {
"methods": ["getLocation", "getLocationCallable"]
},
"GetPublisherModel": {
"methods": ["getPublisherModel", "getPublisherModel", "getPublisherModel", "getPublisherModelCallable"]
},
"ListLocations": {
"methods": ["listLocations", "listLocationsPagedCallable", "listLocationsCallable"]
},
"SetIamPolicy": {
"methods": ["setIamPolicy", "setIamPolicyCallable"]
},
"TestIamPermissions": {
"methods": ["testIamPermissions", "testIamPermissionsCallable"]
}
}
}
}
},
"PipelineService": {
"clients": {
"grpc": {
Expand Down

0 comments on commit ba2c590

Please sign in to comment.