feat: add UpsertDatapoints and RemoveDatapoints rpcs to IndexService in aiplatform v1 index_service.proto

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

@@ -800,6 +800,112 @@ public final UnaryCallable<DeleteIndexRequest, Operation> deleteIndexCallable()
     return stub.deleteIndexCallable();
   }
 
+  // AUTO-GENERATED DOCUMENTATION AND METHOD.
+  /**
+   * Add/update Datapoints into an Index.
+   *
+   * <p>Sample code:
+   *
+   * <pre>{@code
+   * // This snippet has been automatically generated for illustrative purposes only.
+   * // It may require modifications to work in your environment.
+   * try (IndexServiceClient indexServiceClient = IndexServiceClient.create()) {
+   *   UpsertDatapointsRequest request =
+   *       UpsertDatapointsRequest.newBuilder()
+   *           .setIndex(IndexName.of("[PROJECT]", "[LOCATION]", "[INDEX]").toString())
+   *           .addAllDatapoints(new ArrayList<IndexDatapoint>())
+   *           .build();
+   *   UpsertDatapointsResponse response = indexServiceClient.upsertDatapoints(request);
+   * }
+   * }</pre>
+   *
+   * @param request The request object containing all of the parameters for the API call.
+   * @throws com.google.api.gax.rpc.ApiException if the remote call fails
+   */
+  public final UpsertDatapointsResponse upsertDatapoints(UpsertDatapointsRequest request) {
+    return upsertDatapointsCallable().call(request);
+  }
+
+  // AUTO-GENERATED DOCUMENTATION AND METHOD.
+  /**
+   * Add/update Datapoints into an Index.
+   *
+   * <p>Sample code:
+   *
+   * <pre>{@code
+   * // This snippet has been automatically generated for illustrative purposes only.
+   * // It may require modifications to work in your environment.
+   * try (IndexServiceClient indexServiceClient = IndexServiceClient.create()) {
+   *   UpsertDatapointsRequest request =
+   *       UpsertDatapointsRequest.newBuilder()
+   *           .setIndex(IndexName.of("[PROJECT]", "[LOCATION]", "[INDEX]").toString())
+   *           .addAllDatapoints(new ArrayList<IndexDatapoint>())
+   *           .build();
+   *   ApiFuture<UpsertDatapointsResponse> future =
+   *       indexServiceClient.upsertDatapointsCallable().futureCall(request);
+   *   // Do something.
+   *   UpsertDatapointsResponse response = future.get();
+   * }
+   * }</pre>
+   */
+  public final UnaryCallable<UpsertDatapointsRequest, UpsertDatapointsResponse>
+      upsertDatapointsCallable() {
+    return stub.upsertDatapointsCallable();
+  }
+
+  // AUTO-GENERATED DOCUMENTATION AND METHOD.
+  /**
+   * Remove Datapoints from an Index.
+   *
+   * <p>Sample code:
+   *
+   * <pre>{@code
+   * // This snippet has been automatically generated for illustrative purposes only.
+   * // It may require modifications to work in your environment.
+   * try (IndexServiceClient indexServiceClient = IndexServiceClient.create()) {
+   *   RemoveDatapointsRequest request =
+   *       RemoveDatapointsRequest.newBuilder()
+   *           .setIndex(IndexName.of("[PROJECT]", "[LOCATION]", "[INDEX]").toString())
+   *           .addAllDatapointIds(new ArrayList<String>())
+   *           .build();
+   *   RemoveDatapointsResponse response = indexServiceClient.removeDatapoints(request);
+   * }
+   * }</pre>
+   *
+   * @param request The request object containing all of the parameters for the API call.
+   * @throws com.google.api.gax.rpc.ApiException if the remote call fails
+   */
+  public final RemoveDatapointsResponse removeDatapoints(RemoveDatapointsRequest request) {
+    return removeDatapointsCallable().call(request);
+  }
+
+  // AUTO-GENERATED DOCUMENTATION AND METHOD.
+  /**
+   * Remove Datapoints from an Index.
+   *
+   * <p>Sample code:
+   *
+   * <pre>{@code
+   * // This snippet has been automatically generated for illustrative purposes only.
+   * // It may require modifications to work in your environment.
+   * try (IndexServiceClient indexServiceClient = IndexServiceClient.create()) {
+   *   RemoveDatapointsRequest request =
+   *       RemoveDatapointsRequest.newBuilder()
+   *           .setIndex(IndexName.of("[PROJECT]", "[LOCATION]", "[INDEX]").toString())
+   *           .addAllDatapointIds(new ArrayList<String>())
+   *           .build();
+   *   ApiFuture<RemoveDatapointsResponse> future =
+   *       indexServiceClient.removeDatapointsCallable().futureCall(request);
+   *   // Do something.
+   *   RemoveDatapointsResponse response = future.get();
+   * }
+   * }</pre>
+   */
+  public final UnaryCallable<RemoveDatapointsRequest, RemoveDatapointsResponse>
+      removeDatapointsCallable() {
+    return stub.removeDatapointsCallable();
+  }
+
   // AUTO-GENERATED DOCUMENTATION AND METHOD.
   /**
    * Lists information about the supported locations for this service.

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

@@ -127,6 +127,18 @@ public UnaryCallSettings<DeleteIndexRequest, Operation> deleteIndexSettings() {
     return ((IndexServiceStubSettings) getStubSettings()).deleteIndexOperationSettings();
   }
 
+  /** Returns the object with the settings used for calls to upsertDatapoints. */
+  public UnaryCallSettings<UpsertDatapointsRequest, UpsertDatapointsResponse>
+      upsertDatapointsSettings() {
+    return ((IndexServiceStubSettings) getStubSettings()).upsertDatapointsSettings();
+  }
+
+  /** Returns the object with the settings used for calls to removeDatapoints. */
+  public UnaryCallSettings<RemoveDatapointsRequest, RemoveDatapointsResponse>
+      removeDatapointsSettings() {
+    return ((IndexServiceStubSettings) getStubSettings()).removeDatapointsSettings();
+  }
+
   /** Returns the object with the settings used for calls to listLocations. */
   public PagedCallSettings<ListLocationsRequest, ListLocationsResponse, ListLocationsPagedResponse>
       listLocationsSettings() {
@@ -296,6 +308,18 @@ public UnaryCallSettings.Builder<DeleteIndexRequest, Operation> deleteIndexSetti
       return getStubSettingsBuilder().deleteIndexOperationSettings();
     }
 
+    /** Returns the builder for the settings used for calls to upsertDatapoints. */
+    public UnaryCallSettings.Builder<UpsertDatapointsRequest, UpsertDatapointsResponse>
+        upsertDatapointsSettings() {
+      return getStubSettingsBuilder().upsertDatapointsSettings();
+    }
+
+    /** Returns the builder for the settings used for calls to removeDatapoints. */
+    public UnaryCallSettings.Builder<RemoveDatapointsRequest, RemoveDatapointsResponse>
+        removeDatapointsSettings() {
+      return getStubSettingsBuilder().removeDatapointsSettings();
+    }
+
     /** Returns the builder for the settings used for calls to listLocations. */
     public PagedCallSettings.Builder<
             ListLocationsRequest, ListLocationsResponse, ListLocationsPagedResponse>

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

@@ -1151,9 +1151,9 @@ public final UnaryCallable<DeleteModelVersionRequest, Operation> deleteModelVers
    *     explicitly included.
    *     <p>Example: `projects/{project}/locations/{location}/models/{model}{@literal @}1234`
    * @param versionAliases Required. The set of version aliases to merge. The alias should be at
-   *     most 128 characters, and match `[a-z][a-z0-9-]{0,126}[a-z-0-9]`. Add the `-` prefix to an
-   *     alias means removing that alias from the version. `-` is NOT counted in the 128 characters.
-   *     Example: `-golden` means removing the `golden` alias from the version.
+   *     most 128 characters, and match `[a-z][a-zA-Z0-9-]{0,126}[a-z-0-9]`. Add the `-` prefix to
+   *     an alias means removing that alias from the version. `-` is NOT counted in the 128
+   *     characters. Example: `-golden` means removing the `golden` alias from the version.
    *     <p>There is NO ordering in aliases, which means 1) The aliases returned from GetModel API
    *     might not have the exactly same order from this MergeVersionAliases API. 2) Adding and
    *     deleting the same alias in the request is not recommended, and the 2 operations will be
@@ -1189,9 +1189,9 @@ public final Model mergeVersionAliases(ModelName name, List<String> versionAlias
    *     explicitly included.
    *     <p>Example: `projects/{project}/locations/{location}/models/{model}{@literal @}1234`
    * @param versionAliases Required. The set of version aliases to merge. The alias should be at
-   *     most 128 characters, and match `[a-z][a-z0-9-]{0,126}[a-z-0-9]`. Add the `-` prefix to an
-   *     alias means removing that alias from the version. `-` is NOT counted in the 128 characters.
-   *     Example: `-golden` means removing the `golden` alias from the version.
+   *     most 128 characters, and match `[a-z][a-zA-Z0-9-]{0,126}[a-z-0-9]`. Add the `-` prefix to
+   *     an alias means removing that alias from the version. `-` is NOT counted in the 128
+   *     characters. Example: `-golden` means removing the `golden` alias from the version.
    *     <p>There is NO ordering in aliases, which means 1) The aliases returned from GetModel API
    *     might not have the exactly same order from this MergeVersionAliases API. 2) Adding and
    *     deleting the same alias in the request is not recommended, and the 2 operations will be

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

@@ -295,6 +295,9 @@
             "ListLocations": {
               "methods": ["listLocations", "listLocationsPagedCallable", "listLocationsCallable"]
             },
+            "RemoveDatapoints": {
+              "methods": ["removeDatapoints", "removeDatapointsCallable"]
+            },
             "SetIamPolicy": {
               "methods": ["setIamPolicy", "setIamPolicyCallable"]
             },
@@ -303,6 +306,9 @@
             },
             "UpdateIndex": {
               "methods": ["updateIndexAsync", "updateIndexAsync", "updateIndexOperationCallable", "updateIndexCallable"]
+            },
+            "UpsertDatapoints": {
+              "methods": ["upsertDatapoints", "upsertDatapointsCallable"]
             }
           }
         }

google-cloud-aiplatform/src/main/java/com/google/cloud/aiplatform/v1/stub/GrpcIndexServiceStub.java

@@ -34,8 +34,12 @@
 import com.google.cloud.aiplatform.v1.Index;
 import com.google.cloud.aiplatform.v1.ListIndexesRequest;
 import com.google.cloud.aiplatform.v1.ListIndexesResponse;
+import com.google.cloud.aiplatform.v1.RemoveDatapointsRequest;
+import com.google.cloud.aiplatform.v1.RemoveDatapointsResponse;
 import com.google.cloud.aiplatform.v1.UpdateIndexOperationMetadata;
 import com.google.cloud.aiplatform.v1.UpdateIndexRequest;
+import com.google.cloud.aiplatform.v1.UpsertDatapointsRequest;
+import com.google.cloud.aiplatform.v1.UpsertDatapointsResponse;
 import com.google.cloud.location.GetLocationRequest;
 import com.google.cloud.location.ListLocationsRequest;
 import com.google.cloud.location.ListLocationsResponse;
@@ -105,6 +109,28 @@ public class GrpcIndexServiceStub extends IndexServiceStub {
           .setResponseMarshaller(ProtoUtils.marshaller(Operation.getDefaultInstance()))
           .build();
 
+  private static final MethodDescriptor<UpsertDatapointsRequest, UpsertDatapointsResponse>
+      upsertDatapointsMethodDescriptor =
+          MethodDescriptor.<UpsertDatapointsRequest, UpsertDatapointsResponse>newBuilder()
+              .setType(MethodDescriptor.MethodType.UNARY)
+              .setFullMethodName("google.cloud.aiplatform.v1.IndexService/UpsertDatapoints")
+              .setRequestMarshaller(
+                  ProtoUtils.marshaller(UpsertDatapointsRequest.getDefaultInstance()))
+              .setResponseMarshaller(
+                  ProtoUtils.marshaller(UpsertDatapointsResponse.getDefaultInstance()))
+              .build();
+
+  private static final MethodDescriptor<RemoveDatapointsRequest, RemoveDatapointsResponse>
+      removeDatapointsMethodDescriptor =
+          MethodDescriptor.<RemoveDatapointsRequest, RemoveDatapointsResponse>newBuilder()
+              .setType(MethodDescriptor.MethodType.UNARY)
+              .setFullMethodName("google.cloud.aiplatform.v1.IndexService/RemoveDatapoints")
+              .setRequestMarshaller(
+                  ProtoUtils.marshaller(RemoveDatapointsRequest.getDefaultInstance()))
+              .setResponseMarshaller(
+                  ProtoUtils.marshaller(RemoveDatapointsResponse.getDefaultInstance()))
+              .build();
+
   private static final MethodDescriptor<ListLocationsRequest, ListLocationsResponse>
       listLocationsMethodDescriptor =
           MethodDescriptor.<ListLocationsRequest, ListLocationsResponse>newBuilder()
@@ -164,6 +190,10 @@ public class GrpcIndexServiceStub extends IndexServiceStub {
   private final UnaryCallable<DeleteIndexRequest, Operation> deleteIndexCallable;
   private final OperationCallable<DeleteIndexRequest, Empty, DeleteOperationMetadata>
       deleteIndexOperationCallable;
+  private final UnaryCallable<UpsertDatapointsRequest, UpsertDatapointsResponse>
+      upsertDatapointsCallable;
+  private final UnaryCallable<RemoveDatapointsRequest, RemoveDatapointsResponse>
+      removeDatapointsCallable;
   private final UnaryCallable<ListLocationsRequest, ListLocationsResponse> listLocationsCallable;
   private final UnaryCallable<ListLocationsRequest, ListLocationsPagedResponse>
       listLocationsPagedCallable;
@@ -265,6 +295,28 @@ protected GrpcIndexServiceStub(
                   return params.build();
                 })
             .build();
+    GrpcCallSettings<UpsertDatapointsRequest, UpsertDatapointsResponse>
+        upsertDatapointsTransportSettings =
+            GrpcCallSettings.<UpsertDatapointsRequest, UpsertDatapointsResponse>newBuilder()
+                .setMethodDescriptor(upsertDatapointsMethodDescriptor)
+                .setParamsExtractor(
+                    request -> {
+                      ImmutableMap.Builder<String, String> params = ImmutableMap.builder();
+                      params.put("index", String.valueOf(request.getIndex()));
+                      return params.build();
+                    })
+                .build();
+    GrpcCallSettings<RemoveDatapointsRequest, RemoveDatapointsResponse>
+        removeDatapointsTransportSettings =
+            GrpcCallSettings.<RemoveDatapointsRequest, RemoveDatapointsResponse>newBuilder()
+                .setMethodDescriptor(removeDatapointsMethodDescriptor)
+                .setParamsExtractor(
+                    request -> {
+                      ImmutableMap.Builder<String, String> params = ImmutableMap.builder();
+                      params.put("index", String.valueOf(request.getIndex()));
+                      return params.build();
+                    })
+                .build();
     GrpcCallSettings<ListLocationsRequest, ListLocationsResponse> listLocationsTransportSettings =
         GrpcCallSettings.<ListLocationsRequest, ListLocationsResponse>newBuilder()
             .setMethodDescriptor(listLocationsMethodDescriptor)
@@ -353,6 +405,12 @@ protected GrpcIndexServiceStub(
             settings.deleteIndexOperationSettings(),
             clientContext,
             operationsStub);
+    this.upsertDatapointsCallable =
+        callableFactory.createUnaryCallable(
+            upsertDatapointsTransportSettings, settings.upsertDatapointsSettings(), clientContext);
+    this.removeDatapointsCallable =
+        callableFactory.createUnaryCallable(
+            removeDatapointsTransportSettings, settings.removeDatapointsSettings(), clientContext);
     this.listLocationsCallable =
         callableFactory.createUnaryCallable(
             listLocationsTransportSettings, settings.listLocationsSettings(), clientContext);
@@ -430,6 +488,18 @@ public UnaryCallable<DeleteIndexRequest, Operation> deleteIndexCallable() {
     return deleteIndexOperationCallable;
   }
 
+  @Override
+  public UnaryCallable<UpsertDatapointsRequest, UpsertDatapointsResponse>
+      upsertDatapointsCallable() {
+    return upsertDatapointsCallable;
+  }
+
+  @Override
+  public UnaryCallable<RemoveDatapointsRequest, RemoveDatapointsResponse>
+      removeDatapointsCallable() {
+    return removeDatapointsCallable;
+  }
+
   @Override
   public UnaryCallable<ListLocationsRequest, ListLocationsResponse> listLocationsCallable() {
     return listLocationsCallable;

google-cloud-aiplatform/src/main/java/com/google/cloud/aiplatform/v1/stub/IndexServiceStub.java

@@ -30,8 +30,12 @@
 import com.google.cloud.aiplatform.v1.Index;
 import com.google.cloud.aiplatform.v1.ListIndexesRequest;
 import com.google.cloud.aiplatform.v1.ListIndexesResponse;
+import com.google.cloud.aiplatform.v1.RemoveDatapointsRequest;
+import com.google.cloud.aiplatform.v1.RemoveDatapointsResponse;
 import com.google.cloud.aiplatform.v1.UpdateIndexOperationMetadata;
 import com.google.cloud.aiplatform.v1.UpdateIndexRequest;
+import com.google.cloud.aiplatform.v1.UpsertDatapointsRequest;
+import com.google.cloud.aiplatform.v1.UpsertDatapointsResponse;
 import com.google.cloud.location.GetLocationRequest;
 import com.google.cloud.location.ListLocationsRequest;
 import com.google.cloud.location.ListLocationsResponse;
@@ -98,6 +102,16 @@ public UnaryCallable<DeleteIndexRequest, Operation> deleteIndexCallable() {
     throw new UnsupportedOperationException("Not implemented: deleteIndexCallable()");
   }
 
+  public UnaryCallable<UpsertDatapointsRequest, UpsertDatapointsResponse>
+      upsertDatapointsCallable() {
+    throw new UnsupportedOperationException("Not implemented: upsertDatapointsCallable()");
+  }
+
+  public UnaryCallable<RemoveDatapointsRequest, RemoveDatapointsResponse>
+      removeDatapointsCallable() {
+    throw new UnsupportedOperationException("Not implemented: removeDatapointsCallable()");
+  }
+
   public UnaryCallable<ListLocationsRequest, ListLocationsPagedResponse>
       listLocationsPagedCallable() {
     throw new UnsupportedOperationException("Not implemented: listLocationsPagedCallable()");