feat: [aiplatform] add blocking_operation_ids to ImportFeatureValuesOperationMetadata

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.15.0</version>
+      <version>26.16.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.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

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

@@ -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) {
@@ -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) {
@@ -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) {

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

@@ -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);
+    }
+  }
+}

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

@@ -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:
    *
@@ -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:
    *
@@ -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:
    *
@@ -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:
    *

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

@@ -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:
    *
@@ -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:
    *
@@ -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:
    *

java-aiplatform/google-cloud-aiplatform/src/main/java/com/google/cloud/aiplatform/v1/gapic_metadata.json

@@ -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": {