feat: add WriteFeatureValues in aiplatform v1beta1 featurestore_online_service.proto

README.md

@@ -45,13 +45,13 @@ implementation 'com.google.cloud:google-cloud-aiplatform'
 If you are using Gradle without BOM, add this to your dependencies:
 
 ```Groovy
-implementation 'com.google.cloud:google-cloud-aiplatform:3.1.0'
+implementation 'com.google.cloud:google-cloud-aiplatform:3.2.0'
 ```
 
 If you are using SBT, add this to your dependencies:
 
 ```Scala
-libraryDependencies += "com.google.cloud" % "google-cloud-aiplatform" % "3.1.0"
+libraryDependencies += "com.google.cloud" % "google-cloud-aiplatform" % "3.2.0"
 ```
 
 ## Authentication

google-cloud-aiplatform/src/main/java/com/google/cloud/aiplatform/v1beta1/FeaturestoreOnlineServingServiceClient.java

@@ -335,6 +335,153 @@ public final ReadFeatureValuesResponse readFeatureValues(ReadFeatureValuesReques
     return stub.streamingReadFeatureValuesCallable();
   }
 
+  // AUTO-GENERATED DOCUMENTATION AND METHOD.
+  /**
+   * Writes Feature values of one or more entities of an EntityType.
+   *
+   * <p>The Feature values are merged into existing entities if any. The Feature values to be
+   * written must have timestamp within the online storage retention.
+   *
+   * <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 (FeaturestoreOnlineServingServiceClient featurestoreOnlineServingServiceClient =
+   *     FeaturestoreOnlineServingServiceClient.create()) {
+   *   EntityTypeName entityType =
+   *       EntityTypeName.of("[PROJECT]", "[LOCATION]", "[FEATURESTORE]", "[ENTITY_TYPE]");
+   *   List<WriteFeatureValuesPayload> payloads = new ArrayList<>();
+   *   WriteFeatureValuesResponse response =
+   *       featurestoreOnlineServingServiceClient.writeFeatureValues(entityType, payloads);
+   * }
+   * }</pre>
+   *
+   * @param entityType Required. The resource name of the EntityType for the entities being written.
+   *     Value format: `projects/{project}/locations/{location}/featurestores/
+   *     {featurestore}/entityTypes/{entityType}`. For example, for a machine learning model
+   *     predicting user clicks on a website, an EntityType ID could be `user`.
+   * @param payloads Required. The entities to be written. Up to 100,000 feature values can be
+   *     written across all `payloads`.
+   * @throws com.google.api.gax.rpc.ApiException if the remote call fails
+   */
+  public final WriteFeatureValuesResponse writeFeatureValues(
+      EntityTypeName entityType, List<WriteFeatureValuesPayload> payloads) {
+    WriteFeatureValuesRequest request =
+        WriteFeatureValuesRequest.newBuilder()
+            .setEntityType(entityType == null ? null : entityType.toString())
+            .addAllPayloads(payloads)
+            .build();
+    return writeFeatureValues(request);
+  }
+
+  // AUTO-GENERATED DOCUMENTATION AND METHOD.
+  /**
+   * Writes Feature values of one or more entities of an EntityType.
+   *
+   * <p>The Feature values are merged into existing entities if any. The Feature values to be
+   * written must have timestamp within the online storage retention.
+   *
+   * <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 (FeaturestoreOnlineServingServiceClient featurestoreOnlineServingServiceClient =
+   *     FeaturestoreOnlineServingServiceClient.create()) {
+   *   String entityType =
+   *       EntityTypeName.of("[PROJECT]", "[LOCATION]", "[FEATURESTORE]", "[ENTITY_TYPE]")
+   *           .toString();
+   *   List<WriteFeatureValuesPayload> payloads = new ArrayList<>();
+   *   WriteFeatureValuesResponse response =
+   *       featurestoreOnlineServingServiceClient.writeFeatureValues(entityType, payloads);
+   * }
+   * }</pre>
+   *
+   * @param entityType Required. The resource name of the EntityType for the entities being written.
+   *     Value format: `projects/{project}/locations/{location}/featurestores/
+   *     {featurestore}/entityTypes/{entityType}`. For example, for a machine learning model
+   *     predicting user clicks on a website, an EntityType ID could be `user`.
+   * @param payloads Required. The entities to be written. Up to 100,000 feature values can be
+   *     written across all `payloads`.
+   * @throws com.google.api.gax.rpc.ApiException if the remote call fails
+   */
+  public final WriteFeatureValuesResponse writeFeatureValues(
+      String entityType, List<WriteFeatureValuesPayload> payloads) {
+    WriteFeatureValuesRequest request =
+        WriteFeatureValuesRequest.newBuilder()
+            .setEntityType(entityType)
+            .addAllPayloads(payloads)
+            .build();
+    return writeFeatureValues(request);
+  }
+
+  // AUTO-GENERATED DOCUMENTATION AND METHOD.
+  /**
+   * Writes Feature values of one or more entities of an EntityType.
+   *
+   * <p>The Feature values are merged into existing entities if any. The Feature values to be
+   * written must have timestamp within the online storage retention.
+   *
+   * <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 (FeaturestoreOnlineServingServiceClient featurestoreOnlineServingServiceClient =
+   *     FeaturestoreOnlineServingServiceClient.create()) {
+   *   WriteFeatureValuesRequest request =
+   *       WriteFeatureValuesRequest.newBuilder()
+   *           .setEntityType(
+   *               EntityTypeName.of("[PROJECT]", "[LOCATION]", "[FEATURESTORE]", "[ENTITY_TYPE]")
+   *                   .toString())
+   *           .addAllPayloads(new ArrayList<WriteFeatureValuesPayload>())
+   *           .build();
+   *   WriteFeatureValuesResponse response =
+   *       featurestoreOnlineServingServiceClient.writeFeatureValues(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 WriteFeatureValuesResponse writeFeatureValues(WriteFeatureValuesRequest request) {
+    return writeFeatureValuesCallable().call(request);
+  }
+
+  // AUTO-GENERATED DOCUMENTATION AND METHOD.
+  /**
+   * Writes Feature values of one or more entities of an EntityType.
+   *
+   * <p>The Feature values are merged into existing entities if any. The Feature values to be
+   * written must have timestamp within the online storage retention.
+   *
+   * <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 (FeaturestoreOnlineServingServiceClient featurestoreOnlineServingServiceClient =
+   *     FeaturestoreOnlineServingServiceClient.create()) {
+   *   WriteFeatureValuesRequest request =
+   *       WriteFeatureValuesRequest.newBuilder()
+   *           .setEntityType(
+   *               EntityTypeName.of("[PROJECT]", "[LOCATION]", "[FEATURESTORE]", "[ENTITY_TYPE]")
+   *                   .toString())
+   *           .addAllPayloads(new ArrayList<WriteFeatureValuesPayload>())
+   *           .build();
+   *   ApiFuture<WriteFeatureValuesResponse> future =
+   *       featurestoreOnlineServingServiceClient.writeFeatureValuesCallable().futureCall(request);
+   *   // Do something.
+   *   WriteFeatureValuesResponse response = future.get();
+   * }
+   * }</pre>
+   */
+  public final UnaryCallable<WriteFeatureValuesRequest, WriteFeatureValuesResponse>
+      writeFeatureValuesCallable() {
+    return stub.writeFeatureValuesCallable();
+  }
+
   // AUTO-GENERATED DOCUMENTATION AND METHOD.
   /**
    * Lists information about the supported locations for this service.

google-cloud-aiplatform/src/main/java/com/google/cloud/aiplatform/v1beta1/FeaturestoreOnlineServingServiceSettings.java

@@ -99,6 +99,13 @@ public class FeaturestoreOnlineServingServiceSettings
         .streamingReadFeatureValuesSettings();
   }
 
+  /** Returns the object with the settings used for calls to writeFeatureValues. */
+  public UnaryCallSettings<WriteFeatureValuesRequest, WriteFeatureValuesResponse>
+      writeFeatureValuesSettings() {
+    return ((FeaturestoreOnlineServingServiceStubSettings) getStubSettings())
+        .writeFeatureValuesSettings();
+  }
+
   /** Returns the object with the settings used for calls to listLocations. */
   public PagedCallSettings<ListLocationsRequest, ListLocationsResponse, ListLocationsPagedResponse>
       listLocationsSettings() {
@@ -241,6 +248,12 @@ public Builder applyToAllUnaryMethods(
       return getStubSettingsBuilder().streamingReadFeatureValuesSettings();
     }
 
+    /** Returns the builder for the settings used for calls to writeFeatureValues. */
+    public UnaryCallSettings.Builder<WriteFeatureValuesRequest, WriteFeatureValuesResponse>
+        writeFeatureValuesSettings() {
+      return getStubSettingsBuilder().writeFeatureValuesSettings();
+    }
+
     /** 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/v1beta1/gapic_metadata.json

@@ -171,6 +171,9 @@
             },
             "TestIamPermissions": {
               "methods": ["testIamPermissions", "testIamPermissionsCallable"]
+            },
+            "WriteFeatureValues": {
+              "methods": ["writeFeatureValues", "writeFeatureValues", "writeFeatureValues", "writeFeatureValuesCallable"]
             }
           }
         }

google-cloud-aiplatform/src/main/java/com/google/cloud/aiplatform/v1beta1/stub/FeaturestoreOnlineServingServiceStub.java

@@ -25,6 +25,8 @@
 import com.google.cloud.aiplatform.v1beta1.ReadFeatureValuesRequest;
 import com.google.cloud.aiplatform.v1beta1.ReadFeatureValuesResponse;
 import com.google.cloud.aiplatform.v1beta1.StreamingReadFeatureValuesRequest;
+import com.google.cloud.aiplatform.v1beta1.WriteFeatureValuesRequest;
+import com.google.cloud.aiplatform.v1beta1.WriteFeatureValuesResponse;
 import com.google.cloud.location.GetLocationRequest;
 import com.google.cloud.location.ListLocationsRequest;
 import com.google.cloud.location.ListLocationsResponse;
@@ -57,6 +59,11 @@ public abstract class FeaturestoreOnlineServingServiceStub implements Background
         "Not implemented: streamingReadFeatureValuesCallable()");
   }
 
+  public UnaryCallable<WriteFeatureValuesRequest, WriteFeatureValuesResponse>
+      writeFeatureValuesCallable() {
+    throw new UnsupportedOperationException("Not implemented: writeFeatureValuesCallable()");
+  }
+
   public UnaryCallable<ListLocationsRequest, ListLocationsPagedResponse>
       listLocationsPagedCallable() {
     throw new UnsupportedOperationException("Not implemented: listLocationsPagedCallable()");

google-cloud-aiplatform/src/main/java/com/google/cloud/aiplatform/v1beta1/stub/FeaturestoreOnlineServingServiceStubSettings.java

@@ -44,6 +44,8 @@
 import com.google.cloud.aiplatform.v1beta1.ReadFeatureValuesRequest;
 import com.google.cloud.aiplatform.v1beta1.ReadFeatureValuesResponse;
 import com.google.cloud.aiplatform.v1beta1.StreamingReadFeatureValuesRequest;
+import com.google.cloud.aiplatform.v1beta1.WriteFeatureValuesRequest;
+import com.google.cloud.aiplatform.v1beta1.WriteFeatureValuesResponse;
 import com.google.cloud.location.GetLocationRequest;
 import com.google.cloud.location.ListLocationsRequest;
 import com.google.cloud.location.ListLocationsResponse;
@@ -111,6 +113,8 @@ public class FeaturestoreOnlineServingServiceStubSettings
   private final ServerStreamingCallSettings<
           StreamingReadFeatureValuesRequest, ReadFeatureValuesResponse>
       streamingReadFeatureValuesSettings;
+  private final UnaryCallSettings<WriteFeatureValuesRequest, WriteFeatureValuesResponse>
+      writeFeatureValuesSettings;
   private final PagedCallSettings<
           ListLocationsRequest, ListLocationsResponse, ListLocationsPagedResponse>
       listLocationsSettings;
@@ -185,6 +189,12 @@ public ApiFuture<ListLocationsPagedResponse> getFuturePagedResponse(
     return streamingReadFeatureValuesSettings;
   }
 
+  /** Returns the object with the settings used for calls to writeFeatureValues. */
+  public UnaryCallSettings<WriteFeatureValuesRequest, WriteFeatureValuesResponse>
+      writeFeatureValuesSettings() {
+    return writeFeatureValuesSettings;
+  }
+
   /** Returns the object with the settings used for calls to listLocations. */
   public PagedCallSettings<ListLocationsRequest, ListLocationsResponse, ListLocationsPagedResponse>
       listLocationsSettings() {
@@ -292,6 +302,7 @@ protected FeaturestoreOnlineServingServiceStubSettings(Builder settingsBuilder)
     readFeatureValuesSettings = settingsBuilder.readFeatureValuesSettings().build();
     streamingReadFeatureValuesSettings =
         settingsBuilder.streamingReadFeatureValuesSettings().build();
+    writeFeatureValuesSettings = settingsBuilder.writeFeatureValuesSettings().build();
     listLocationsSettings = settingsBuilder.listLocationsSettings().build();
     getLocationSettings = settingsBuilder.getLocationSettings().build();
     setIamPolicySettings = settingsBuilder.setIamPolicySettings().build();
@@ -308,6 +319,8 @@ public static class Builder
     private final ServerStreamingCallSettings.Builder<
             StreamingReadFeatureValuesRequest, ReadFeatureValuesResponse>
         streamingReadFeatureValuesSettings;
+    private final UnaryCallSettings.Builder<WriteFeatureValuesRequest, WriteFeatureValuesResponse>
+        writeFeatureValuesSettings;
     private final PagedCallSettings.Builder<
             ListLocationsRequest, ListLocationsResponse, ListLocationsPagedResponse>
         listLocationsSettings;
@@ -355,6 +368,7 @@ protected Builder(ClientContext clientContext) {
 
       readFeatureValuesSettings = UnaryCallSettings.newUnaryCallSettingsBuilder();
       streamingReadFeatureValuesSettings = ServerStreamingCallSettings.newBuilder();
+      writeFeatureValuesSettings = UnaryCallSettings.newUnaryCallSettingsBuilder();
       listLocationsSettings = PagedCallSettings.newBuilder(LIST_LOCATIONS_PAGE_STR_FACT);
       getLocationSettings = UnaryCallSettings.newUnaryCallSettingsBuilder();
       setIamPolicySettings = UnaryCallSettings.newUnaryCallSettingsBuilder();
@@ -364,6 +378,7 @@ protected Builder(ClientContext clientContext) {
       unaryMethodSettingsBuilders =
           ImmutableList.<UnaryCallSettings.Builder<?, ?>>of(
               readFeatureValuesSettings,
+              writeFeatureValuesSettings,
               listLocationsSettings,
               getLocationSettings,
               setIamPolicySettings,
@@ -377,6 +392,7 @@ protected Builder(FeaturestoreOnlineServingServiceStubSettings settings) {
 
       readFeatureValuesSettings = settings.readFeatureValuesSettings.toBuilder();
       streamingReadFeatureValuesSettings = settings.streamingReadFeatureValuesSettings.toBuilder();
+      writeFeatureValuesSettings = settings.writeFeatureValuesSettings.toBuilder();
       listLocationsSettings = settings.listLocationsSettings.toBuilder();
       getLocationSettings = settings.getLocationSettings.toBuilder();
       setIamPolicySettings = settings.setIamPolicySettings.toBuilder();
@@ -386,6 +402,7 @@ protected Builder(FeaturestoreOnlineServingServiceStubSettings settings) {
       unaryMethodSettingsBuilders =
           ImmutableList.<UnaryCallSettings.Builder<?, ?>>of(
               readFeatureValuesSettings,
+              writeFeatureValuesSettings,
               listLocationsSettings,
               getLocationSettings,
               setIamPolicySettings,
@@ -417,6 +434,11 @@ private static Builder initDefaults(Builder builder) {
           .setRetryableCodes(RETRYABLE_CODE_DEFINITIONS.get("no_retry_13_codes"))
           .setRetrySettings(RETRY_PARAM_DEFINITIONS.get("no_retry_13_params"));
 
+      builder
+          .writeFeatureValuesSettings()
+          .setRetryableCodes(RETRYABLE_CODE_DEFINITIONS.get("no_retry_codes"))
+          .setRetrySettings(RETRY_PARAM_DEFINITIONS.get("no_retry_params"));
+
       builder
           .listLocationsSettings()
           .setRetryableCodes(RETRYABLE_CODE_DEFINITIONS.get("no_retry_codes"))
@@ -473,6 +495,12 @@ public Builder applyToAllUnaryMethods(
       return streamingReadFeatureValuesSettings;
     }
 
+    /** Returns the builder for the settings used for calls to writeFeatureValues. */
+    public UnaryCallSettings.Builder<WriteFeatureValuesRequest, WriteFeatureValuesResponse>
+        writeFeatureValuesSettings() {
+      return writeFeatureValuesSettings;
+    }
+
     /** Returns the builder for the settings used for calls to listLocations. */
     public PagedCallSettings.Builder<
             ListLocationsRequest, ListLocationsResponse, ListLocationsPagedResponse>