feat: [vertexai] Update Gapic client. (#10574)

PiperOrigin-RevId: 617258114 Co-authored-by: Zhenyi Qi <zhenyiqi@google.com>
googleapis · Mar 21, 2024 · bedcddf · bedcddf
1 parent 99427af
commit bedcddf
Show file tree

Hide file tree

Showing 40 changed files with 1,355 additions and 518 deletions.
diff --git a/...e-cloud-vertexai/src/main/java/com/google/cloud/vertexai/api/EndpointServiceSettings.java b/...e-cloud-vertexai/src/main/java/com/google/cloud/vertexai/api/EndpointServiceSettings.java
@@ -242,7 +242,6 @@ public static Builder newBuilder() {
   }
 
   /** Returns a new REST builder for this class. */
-  @BetaApi
   public static Builder newHttpJsonBuilder() {
     return Builder.createHttpJsonDefault();
   }
@@ -284,7 +283,6 @@ private static Builder createDefault() {
       return new Builder(EndpointServiceStubSettings.newBuilder());
     }
 
-    @BetaApi
     private static Builder createHttpJsonDefault() {
       return new Builder(EndpointServiceStubSettings.newHttpJsonBuilder());
     }

diff --git a/...cloud-vertexai/src/main/java/com/google/cloud/vertexai/api/LlmUtilityServiceSettings.java b/...cloud-vertexai/src/main/java/com/google/cloud/vertexai/api/LlmUtilityServiceSettings.java
@@ -172,7 +172,6 @@ public static Builder newBuilder() {
   }
 
   /** Returns a new REST builder for this class. */
-  @BetaApi
   public static Builder newHttpJsonBuilder() {
     return Builder.createHttpJsonDefault();
   }
@@ -214,7 +213,6 @@ private static Builder createDefault() {
       return new Builder(LlmUtilityServiceStubSettings.newBuilder());
     }
 
-    @BetaApi
     private static Builder createHttpJsonDefault() {
       return new Builder(LlmUtilityServiceStubSettings.newHttpJsonBuilder());
     }

diff --git a/...cloud-vertexai/src/main/java/com/google/cloud/vertexai/api/PredictionServiceSettings.java b/...cloud-vertexai/src/main/java/com/google/cloud/vertexai/api/PredictionServiceSettings.java
@@ -238,7 +238,6 @@ public static Builder newBuilder() {
   }
 
   /** Returns a new REST builder for this class. */
-  @BetaApi
   public static Builder newHttpJsonBuilder() {
     return Builder.createHttpJsonDefault();
   }
@@ -280,7 +279,6 @@ private static Builder createDefault() {
       return new Builder(PredictionServiceStubSettings.newBuilder());
     }
 
-    @BetaApi
     private static Builder createHttpJsonDefault() {
       return new Builder(PredictionServiceStubSettings.newHttpJsonBuilder());
     }

diff --git a/...ertexai/src/main/java/com/google/cloud/vertexai/api/stub/EndpointServiceStubSettings.java b/...ertexai/src/main/java/com/google/cloud/vertexai/api/stub/EndpointServiceStubSettings.java
@@ -897,8 +897,6 @@ public UnaryCallSettings.Builder<CreateEndpointRequest, Operation> createEndpoin
     }
 
     /** Returns the builder for the settings used for calls to createEndpoint. */
-    @BetaApi(
-        "The surface for use by generated code is not stable yet and may change in the future.")
     public OperationCallSettings.Builder<
             CreateEndpointRequest, Endpoint, CreateEndpointOperationMetadata>
         createEndpointOperationSettings() {
@@ -928,8 +926,6 @@ public UnaryCallSettings.Builder<DeleteEndpointRequest, Operation> deleteEndpoin
     }
 
     /** Returns the builder for the settings used for calls to deleteEndpoint. */
-    @BetaApi(
-        "The surface for use by generated code is not stable yet and may change in the future.")
     public OperationCallSettings.Builder<DeleteEndpointRequest, Empty, DeleteOperationMetadata>
         deleteEndpointOperationSettings() {
       return deleteEndpointOperationSettings;
@@ -941,8 +937,6 @@ public UnaryCallSettings.Builder<DeployModelRequest, Operation> deployModelSetti
     }
 
     /** Returns the builder for the settings used for calls to deployModel. */
-    @BetaApi(
-        "The surface for use by generated code is not stable yet and may change in the future.")
     public OperationCallSettings.Builder<
             DeployModelRequest, DeployModelResponse, DeployModelOperationMetadata>
         deployModelOperationSettings() {
@@ -955,8 +949,6 @@ public UnaryCallSettings.Builder<UndeployModelRequest, Operation> undeployModelS
     }
 
     /** Returns the builder for the settings used for calls to undeployModel. */
-    @BetaApi(
-        "The surface for use by generated code is not stable yet and may change in the future.")
     public OperationCallSettings.Builder<
             UndeployModelRequest, UndeployModelResponse, UndeployModelOperationMetadata>
         undeployModelOperationSettings() {
@@ -970,8 +962,6 @@ public UnaryCallSettings.Builder<UndeployModelRequest, Operation> undeployModelS
     }
 
     /** Returns the builder for the settings used for calls to mutateDeployedModel. */
-    @BetaApi(
-        "The surface for use by generated code is not stable yet and may change in the future.")
     public OperationCallSettings.Builder<
             MutateDeployedModelRequest,
             MutateDeployedModelResponse,

diff --git a/.../main/java/com/google/cloud/vertexai/api/stub/HttpJsonEndpointServiceCallableFactory.java b/.../main/java/com/google/cloud/vertexai/api/stub/HttpJsonEndpointServiceCallableFactory.java
@@ -16,7 +16,6 @@
 
 package com.google.cloud.vertexai.api.stub;
 
-import com.google.api.core.BetaApi;
 import com.google.api.gax.httpjson.HttpJsonCallSettings;
 import com.google.api.gax.httpjson.HttpJsonCallableFactory;
 import com.google.api.gax.httpjson.HttpJsonOperationSnapshotCallable;
@@ -41,7 +40,6 @@
  * <p>This class is for advanced usage.
  */
 @Generated("by gapic-generator-java")
-@BetaApi
 public class HttpJsonEndpointServiceCallableFactory
     implements HttpJsonStubCallableFactory<Operation, OperationsStub> {
 
@@ -73,8 +71,6 @@ public <RequestT, ResponseT> UnaryCallable<RequestT, ResponseT> createBatchingCa
         httpJsonCallSettings, callSettings, clientContext);
   }
 
-  @BetaApi(
-      "The surface for long-running operations is not stable yet and may change in the future.")
   @Override
   public <RequestT, ResponseT, MetadataT>
       OperationCallable<RequestT, ResponseT, MetadataT> createOperationCallable(

diff --git a/...ertexai/src/main/java/com/google/cloud/vertexai/api/stub/HttpJsonEndpointServiceStub.java b/...ertexai/src/main/java/com/google/cloud/vertexai/api/stub/HttpJsonEndpointServiceStub.java
@@ -20,7 +20,6 @@
 import static com.google.cloud.vertexai.api.EndpointServiceClient.ListLocationsPagedResponse;
 
 import com.google.api.HttpRule;
-import com.google.api.core.BetaApi;
 import com.google.api.core.InternalApi;
 import com.google.api.gax.core.BackgroundResource;
 import com.google.api.gax.core.BackgroundResourceAggregation;
@@ -82,7 +81,6 @@
  * <p>This class is for advanced usage and reflects the underlying API directly.
  */
 @Generated("by gapic-generator-java")
-@BetaApi
 public class HttpJsonEndpointServiceStub extends EndpointServiceStub {
   private static final TypeRegistry typeRegistry =
       TypeRegistry.newBuilder()
@@ -493,7 +491,9 @@ public class HttpJsonEndpointServiceStub extends EndpointServiceStub {
                           "/ui/{resource=projects/*/locations/*/featurestores/*/entityTypes/*}:setIamPolicy",
                           "/ui/{resource=projects/*/locations/*/models/*}:setIamPolicy",
                           "/ui/{resource=projects/*/locations/*/endpoints/*}:setIamPolicy",
-                          "/ui/{resource=projects/*/locations/*/notebookRuntimeTemplates/*}:setIamPolicy")
+                          "/ui/{resource=projects/*/locations/*/notebookRuntimeTemplates/*}:setIamPolicy",
+                          "/ui/{resource=projects/*/locations/*/featureOnlineStores/*}:setIamPolicy",
+                          "/ui/{resource=projects/*/locations/*/featureOnlineStores/*/featureViews/*}:setIamPolicy")
                       .setQueryParamsExtractor(
                           request -> {
                             Map<String, List<String>> fields = new HashMap<>();
@@ -539,7 +539,9 @@ public class HttpJsonEndpointServiceStub extends EndpointServiceStub {
                           "/ui/{resource=projects/*/locations/*/models/*}:getIamPolicy",
                           "/ui/{resource=projects/*/locations/*/endpoints/*}:getIamPolicy",
                           "/ui/{resource=projects/*/locations/*/notebookRuntimeTemplates/*}:getIamPolicy",
-                          "/ui/{resource=projects/*/locations/*/publishers/*/models/*}:getIamPolicy")
+                          "/ui/{resource=projects/*/locations/*/publishers/*/models/*}:getIamPolicy",
+                          "/ui/{resource=projects/*/locations/*/featureOnlineStores/*}:getIamPolicy",
+                          "/ui/{resource=projects/*/locations/*/featureOnlineStores/*/featureViews/*}:getIamPolicy")
                       .setQueryParamsExtractor(
                           request -> {
                             Map<String, List<String>> fields = new HashMap<>();
@@ -581,7 +583,9 @@ public class HttpJsonEndpointServiceStub extends EndpointServiceStub {
                           "/ui/{resource=projects/*/locations/*/featurestores/*/entityTypes/*}:testIamPermissions",
                           "/ui/{resource=projects/*/locations/*/models/*}:testIamPermissions",
                           "/ui/{resource=projects/*/locations/*/endpoints/*}:testIamPermissions",
-                          "/ui/{resource=projects/*/locations/*/notebookRuntimeTemplates/*}:testIamPermissions")
+                          "/ui/{resource=projects/*/locations/*/notebookRuntimeTemplates/*}:testIamPermissions",
+                          "/ui/{resource=projects/*/locations/*/featureOnlineStores/*}:testIamPermissions",
+                          "/ui/{resource=projects/*/locations/*/featureOnlineStores/*/featureViews/*}:testIamPermissions")
                       .setQueryParamsExtractor(
                           request -> {
                             Map<String, List<String>> fields = new HashMap<>();
@@ -764,6 +768,11 @@ protected HttpJsonEndpointServiceStub(
                                 .setPost(
                                     "/ui/{name=projects/*/locations/*/hyperparameterTuningJobs/*/operations/*}:cancel")
                                 .build())
+                        .addAdditionalBindings(
+                            HttpRule.newBuilder()
+                                .setPost(
+                                    "/ui/{name=projects/*/locations/*/tuningJobs/*/operations/*}:cancel")
+                                .build())
                         .addAdditionalBindings(
                             HttpRule.newBuilder()
                                 .setPost(
@@ -938,6 +947,11 @@ protected HttpJsonEndpointServiceStub(
                                 .setPost(
                                     "/v1/{name=projects/*/locations/*/hyperparameterTuningJobs/*/operations/*}:cancel")
                                 .build())
+                        .addAdditionalBindings(
+                            HttpRule.newBuilder()
+                                .setPost(
+                                    "/v1/{name=projects/*/locations/*/tuningJobs/*/operations/*}:cancel")
+                                .build())
                         .addAdditionalBindings(
                             HttpRule.newBuilder()
                                 .setPost(
@@ -1526,6 +1540,11 @@ protected HttpJsonEndpointServiceStub(
                                 .setGet(
                                     "/ui/{name=projects/*/locations/*/hyperparameterTuningJobs/*/operations/*}")
                                 .build())
+                        .addAdditionalBindings(
+                            HttpRule.newBuilder()
+                                .setGet(
+                                    "/ui/{name=projects/*/locations/*/tuningJobs/*/operations/*}")
+                                .build())
                         .addAdditionalBindings(
                             HttpRule.newBuilder()
                                 .setGet("/ui/{name=projects/*/locations/*/indexes/*/operations/*}")
@@ -1716,6 +1735,11 @@ protected HttpJsonEndpointServiceStub(
                                 .setGet(
                                     "/v1/{name=projects/*/locations/*/hyperparameterTuningJobs/*/operations/*}")
                                 .build())
+                        .addAdditionalBindings(
+                            HttpRule.newBuilder()
+                                .setGet(
+                                    "/v1/{name=projects/*/locations/*/tuningJobs/*/operations/*}")
+                                .build())
                         .addAdditionalBindings(
                             HttpRule.newBuilder()
                                 .setGet("/v1/{name=projects/*/locations/*/indexes/*/operations/*}")
@@ -1914,6 +1938,10 @@ protected HttpJsonEndpointServiceStub(
                                 .setGet(
                                     "/ui/{name=projects/*/locations/*/hyperparameterTuningJobs/*}/operations")
                                 .build())
+                        .addAdditionalBindings(
+                            HttpRule.newBuilder()
+                                .setGet("/ui/{name=projects/*/locations/*/tuningJobs/*}/operations")
+                                .build())
                         .addAdditionalBindings(
                             HttpRule.newBuilder()
                                 .setGet("/ui/{name=projects/*/locations/*/indexes/*}/operations")
@@ -2101,6 +2129,10 @@ protected HttpJsonEndpointServiceStub(
                                 .setGet(
                                     "/v1/{name=projects/*/locations/*/hyperparameterTuningJobs/*}/operations")
                                 .build())
+                        .addAdditionalBindings(
+                            HttpRule.newBuilder()
+                                .setGet("/ui/{name=projects/*/locations/*/tuningJobs/*}/operations")
+                                .build())
                         .addAdditionalBindings(
                             HttpRule.newBuilder()
                                 .setGet("/v1/{name=projects/*/locations/*/indexes/*}/operations")
@@ -2302,6 +2334,11 @@ protected HttpJsonEndpointServiceStub(
                                 .setPost(
                                     "/ui/{name=projects/*/locations/*/hyperparameterTuningJobs/*/operations/*}:wait")
                                 .build())
+                        .addAdditionalBindings(
+                            HttpRule.newBuilder()
+                                .setPost(
+                                    "/ui/{name=projects/*/locations/*/tuningJobs/*/operations/*}:wait")
+                                .build())
                         .addAdditionalBindings(
                             HttpRule.newBuilder()
                                 .setPost(

diff --git a/...ain/java/com/google/cloud/vertexai/api/stub/HttpJsonLlmUtilityServiceCallableFactory.java b/...ain/java/com/google/cloud/vertexai/api/stub/HttpJsonLlmUtilityServiceCallableFactory.java
@@ -16,7 +16,6 @@
 
 package com.google.cloud.vertexai.api.stub;
 
-import com.google.api.core.BetaApi;
 import com.google.api.gax.httpjson.HttpJsonCallSettings;
 import com.google.api.gax.httpjson.HttpJsonCallableFactory;
 import com.google.api.gax.httpjson.HttpJsonOperationSnapshotCallable;
@@ -41,7 +40,6 @@
  * <p>This class is for advanced usage.
  */
 @Generated("by gapic-generator-java")
-@BetaApi
 public class HttpJsonLlmUtilityServiceCallableFactory
     implements HttpJsonStubCallableFactory<Operation, OperationsStub> {
 
@@ -73,8 +71,6 @@ public <RequestT, ResponseT> UnaryCallable<RequestT, ResponseT> createBatchingCa
         httpJsonCallSettings, callSettings, clientContext);
   }
 
-  @BetaApi(
-      "The surface for long-running operations is not stable yet and may change in the future.")
   @Override
   public <RequestT, ResponseT, MetadataT>
       OperationCallable<RequestT, ResponseT, MetadataT> createOperationCallable(

diff --git a/...texai/src/main/java/com/google/cloud/vertexai/api/stub/HttpJsonLlmUtilityServiceStub.java b/...texai/src/main/java/com/google/cloud/vertexai/api/stub/HttpJsonLlmUtilityServiceStub.java
@@ -18,7 +18,6 @@
 
 import static com.google.cloud.vertexai.api.LlmUtilityServiceClient.ListLocationsPagedResponse;
 
-import com.google.api.core.BetaApi;
 import com.google.api.core.InternalApi;
 import com.google.api.gax.core.BackgroundResource;
 import com.google.api.gax.core.BackgroundResourceAggregation;
@@ -60,7 +59,6 @@
  * <p>This class is for advanced usage and reflects the underlying API directly.
  */
 @Generated("by gapic-generator-java")
-@BetaApi
 public class HttpJsonLlmUtilityServiceStub extends LlmUtilityServiceStub {
   private static final TypeRegistry typeRegistry = TypeRegistry.newBuilder().build();
 
@@ -234,7 +232,9 @@ public class HttpJsonLlmUtilityServiceStub extends LlmUtilityServiceStub {
                           "/ui/{resource=projects/*/locations/*/featurestores/*/entityTypes/*}:setIamPolicy",
                           "/ui/{resource=projects/*/locations/*/models/*}:setIamPolicy",
                           "/ui/{resource=projects/*/locations/*/endpoints/*}:setIamPolicy",
-                          "/ui/{resource=projects/*/locations/*/notebookRuntimeTemplates/*}:setIamPolicy")
+                          "/ui/{resource=projects/*/locations/*/notebookRuntimeTemplates/*}:setIamPolicy",
+                          "/ui/{resource=projects/*/locations/*/featureOnlineStores/*}:setIamPolicy",
+                          "/ui/{resource=projects/*/locations/*/featureOnlineStores/*/featureViews/*}:setIamPolicy")
                       .setQueryParamsExtractor(
                           request -> {
                             Map<String, List<String>> fields = new HashMap<>();
@@ -280,7 +280,9 @@ public class HttpJsonLlmUtilityServiceStub extends LlmUtilityServiceStub {
                           "/ui/{resource=projects/*/locations/*/models/*}:getIamPolicy",
                           "/ui/{resource=projects/*/locations/*/endpoints/*}:getIamPolicy",
                           "/ui/{resource=projects/*/locations/*/notebookRuntimeTemplates/*}:getIamPolicy",
-                          "/ui/{resource=projects/*/locations/*/publishers/*/models/*}:getIamPolicy")
+                          "/ui/{resource=projects/*/locations/*/publishers/*/models/*}:getIamPolicy",
+                          "/ui/{resource=projects/*/locations/*/featureOnlineStores/*}:getIamPolicy",
+                          "/ui/{resource=projects/*/locations/*/featureOnlineStores/*/featureViews/*}:getIamPolicy")
                       .setQueryParamsExtractor(
                           request -> {
                             Map<String, List<String>> fields = new HashMap<>();
@@ -322,7 +324,9 @@ public class HttpJsonLlmUtilityServiceStub extends LlmUtilityServiceStub {
                           "/ui/{resource=projects/*/locations/*/featurestores/*/entityTypes/*}:testIamPermissions",
                           "/ui/{resource=projects/*/locations/*/models/*}:testIamPermissions",
                           "/ui/{resource=projects/*/locations/*/endpoints/*}:testIamPermissions",
-                          "/ui/{resource=projects/*/locations/*/notebookRuntimeTemplates/*}:testIamPermissions")
+                          "/ui/{resource=projects/*/locations/*/notebookRuntimeTemplates/*}:testIamPermissions",
+                          "/ui/{resource=projects/*/locations/*/featureOnlineStores/*}:testIamPermissions",
+                          "/ui/{resource=projects/*/locations/*/featureOnlineStores/*/featureViews/*}:testIamPermissions")
                       .setQueryParamsExtractor(
                           request -> {
                             Map<String, List<String>> fields = new HashMap<>();

diff --git a/...ain/java/com/google/cloud/vertexai/api/stub/HttpJsonPredictionServiceCallableFactory.java b/...ain/java/com/google/cloud/vertexai/api/stub/HttpJsonPredictionServiceCallableFactory.java
@@ -16,7 +16,6 @@
 
 package com.google.cloud.vertexai.api.stub;
 
-import com.google.api.core.BetaApi;
 import com.google.api.gax.httpjson.HttpJsonCallSettings;
 import com.google.api.gax.httpjson.HttpJsonCallableFactory;
 import com.google.api.gax.httpjson.HttpJsonOperationSnapshotCallable;
@@ -41,7 +40,6 @@
  * <p>This class is for advanced usage.
  */
 @Generated("by gapic-generator-java")
-@BetaApi
 public class HttpJsonPredictionServiceCallableFactory
     implements HttpJsonStubCallableFactory<Operation, OperationsStub> {
 
@@ -73,8 +71,6 @@ public <RequestT, ResponseT> UnaryCallable<RequestT, ResponseT> createBatchingCa
         httpJsonCallSettings, callSettings, clientContext);
   }
 
-  @BetaApi(
-      "The surface for long-running operations is not stable yet and may change in the future.")
   @Override
   public <RequestT, ResponseT, MetadataT>
       OperationCallable<RequestT, ResponseT, MetadataT> createOperationCallable(