feat: add FeatureViewDirectWrite API in v1

Google APIs · copybara-github · commit 8c74a4f9ad52 · 2025-08-04T13:44:31.000-07:00
PiperOrigin-RevId: 790876671
diff --git a/google/cloud/aiplatform/v1/aiplatform_v1.yaml b/google/cloud/aiplatform/v1/aiplatform_v1.yaml
@@ -759,11 +759,7 @@ authentication:
     oauth:
       canonical_scopes: |-
         https://www.googleapis.com/auth/cloud-platform
-  - selector: google.cloud.aiplatform.v1.FeatureOnlineStoreService.FetchFeatureValues
-    oauth:
-      canonical_scopes: |-
-        https://www.googleapis.com/auth/cloud-platform
-  - selector: google.cloud.aiplatform.v1.FeatureOnlineStoreService.SearchNearestEntities
+  - selector: 'google.cloud.aiplatform.v1.FeatureOnlineStoreService.*'
     oauth:
       canonical_scopes: |-
         https://www.googleapis.com/auth/cloud-platform
diff --git a/google/cloud/aiplatform/v1/feature_online_store_service.proto b/google/cloud/aiplatform/v1/feature_online_store_service.proto
@@ -22,6 +22,8 @@ import "google/api/field_behavior.proto";
 import "google/api/resource.proto";
 import "google/cloud/aiplatform/v1/featurestore_online_service.proto";
 import "google/protobuf/struct.proto";
+import "google/protobuf/timestamp.proto";
+import "google/rpc/status.proto";
 
 option csharp_namespace = "Google.Cloud.AIPlatform.V1";
 option go_package = "cloud.google.com/go/aiplatform/apiv1/aiplatformpb;aiplatformpb";
@@ -57,6 +59,17 @@ service FeatureOnlineStoreService {
       body: "*"
     };
   }
+
+  // Bidirectional streaming RPC to directly write to feature values in a
+  // feature view. Requests may not have a one-to-one mapping to responses and
+  // responses may be returned out-of-order to reduce latency.
+  rpc FeatureViewDirectWrite(stream FeatureViewDirectWriteRequest)
+      returns (stream FeatureViewDirectWriteResponse) {
+    option (google.api.http) = {
+      post: "/v1/{feature_view=projects/*/locations/*/featureOnlineStores/*/featureViews/*}:directWrite"
+      body: "*"
+    };
+  }
 }
 
 // Format of the data in the Feature View.
@@ -329,3 +342,75 @@ message SearchNearestEntitiesResponse {
   // The nearest neighbors of the query entity.
   NearestNeighbors nearest_neighbors = 1;
 }
+
+// Request message for
+// [FeatureOnlineStoreService.FeatureViewDirectWrite][google.cloud.aiplatform.v1.FeatureOnlineStoreService.FeatureViewDirectWrite].
+message FeatureViewDirectWriteRequest {
+  // A data key and associated feature values to write to the feature view.
+  message DataKeyAndFeatureValues {
+    // Feature name & value pair.
+    message Feature {
+      // Feature value data to write.
+      oneof data_oneof {
+        // Feature value. A user provided timestamp may be set in the
+        // `FeatureValue.metadata.generate_time` field.
+        FeatureValue value = 3;
+      }
+
+      // Feature short name.
+      string name = 1;
+    }
+
+    // The data key.
+    FeatureViewDataKey data_key = 1;
+
+    // List of features to write.
+    repeated Feature features = 2;
+  }
+
+  // FeatureView resource format
+  // `projects/{project}/locations/{location}/featureOnlineStores/{featureOnlineStore}/featureViews/{featureView}`
+  string feature_view = 1 [(google.api.resource_reference) = {
+    type: "aiplatform.googleapis.com/FeatureView"
+  }];
+
+  // Required. The data keys and associated feature values.
+  repeated DataKeyAndFeatureValues data_key_and_feature_values = 2
+      [(google.api.field_behavior) = REQUIRED];
+}
+
+// Response message for
+// [FeatureOnlineStoreService.FeatureViewDirectWrite][google.cloud.aiplatform.v1.FeatureOnlineStoreService.FeatureViewDirectWrite].
+message FeatureViewDirectWriteResponse {
+  // Details about the write for each key.
+  message WriteResponse {
+    // What key is this write response associated with.
+    FeatureViewDataKey data_key = 1;
+
+    // When the feature values were written to the online store.
+    // If
+    // [FeatureViewDirectWriteResponse.status][google.cloud.aiplatform.v1.FeatureViewDirectWriteResponse.status]
+    // is not OK, this field is not populated.
+    google.protobuf.Timestamp online_store_write_time = 2;
+  }
+
+  // Response status for the keys listed in
+  // [FeatureViewDirectWriteResponse.write_responses][google.cloud.aiplatform.v1.FeatureViewDirectWriteResponse.write_responses].
+  //
+  // The error only applies to the
+  // listed data keys - the stream will remain open for further
+  // [FeatureOnlineStoreService.FeatureViewDirectWriteRequest][] requests.
+  //
+  // Partial failures (e.g. if the first 10 keys of a request fail, but the
+  // rest succeed) from a single request may result in multiple responses -
+  // there will be one response for the successful request keys and one response
+  // for the failing request keys.
+  google.rpc.Status status = 1;
+
+  // Details about write for each key. If status is not OK,
+  // [WriteResponse.data_key][google.cloud.aiplatform.v1.FeatureViewDirectWriteResponse.WriteResponse.data_key]
+  // will have the key with error, but
+  // [WriteResponse.online_store_write_time][google.cloud.aiplatform.v1.FeatureViewDirectWriteResponse.WriteResponse.online_store_write_time]
+  // will not be present.
+  repeated WriteResponse write_responses = 2;
+}
diff --git a/google/cloud/aiplatform/v1/featurestore_online_service.proto b/google/cloud/aiplatform/v1/featurestore_online_service.proto
@@ -238,9 +238,11 @@ message FeatureValue {
     // Feature generation timestamp. Typically, it is provided by user at
     // feature ingestion time. If not, feature store
     // will use the system timestamp when the data is ingested into feature
-    // store. For streaming ingestion, the time, aligned by days, must be no
-    // older than five years (1825 days) and no later than one year (366 days)
-    // in the future.
+    // store.
+    //
+    // Legacy Feature Store: For streaming ingestion, the time, aligned by days,
+    // must be no older than five years (1825 days) and no later than one year
+    // (366 days) in the future.
     google.protobuf.Timestamp generate_time = 1;
   }
 
diff --git a/google/cloud/aiplatform/v1/featurestore_service.proto b/google/cloud/aiplatform/v1/featurestore_service.proto
@@ -1176,7 +1176,6 @@ message SearchFeaturesRequest {
   // containing the substring `foo` and description containing the substring
   // `bar`.
   //
-  //
   // Besides field queries, the following exact-match filters are
   // supported. The exact-match filters do not support wildcards. Unlike
   // field-restricted queries, exact-match filters are case-sensitive.
diff --git a/google/cloud/aiplatform/v1beta1/feature_online_store_service.proto b/google/cloud/aiplatform/v1beta1/feature_online_store_service.proto
@@ -442,6 +442,10 @@ message FeatureViewDirectWriteRequest {
 
       // Feature value data to write.
       oneof data_oneof {
+        // Feature value. A user provided timestamp may be set in the
+        // `FeatureValue.metadata.generate_time` field.
+        FeatureValue value = 3;
+
         // Feature value and timestamp.
         FeatureValueAndTimestamp value_and_timestamp = 2;
       }
diff --git a/google/cloud/aiplatform/v1beta1/featurestore_online_service.proto b/google/cloud/aiplatform/v1beta1/featurestore_online_service.proto
@@ -206,7 +206,7 @@ message ReadFeatureValuesResponse {
 }
 
 // Request message for
-// [FeaturestoreOnlineServingService.StreamingFeatureValuesRead][].
+// [FeaturestoreOnlineServingService.StreamingReadFeatureValues][google.cloud.aiplatform.v1beta1.FeaturestoreOnlineServingService.StreamingReadFeatureValues].
 message StreamingReadFeatureValuesRequest {
   // Required. The resource name of the entities' type.
   // Value format:
@@ -238,9 +238,11 @@ message FeatureValue {
     // Feature generation timestamp. Typically, it is provided by user at
     // feature ingestion time. If not, feature store
     // will use the system timestamp when the data is ingested into feature
-    // store. For streaming ingestion, the time, aligned by days, must be no
-    // older than five years (1825 days) and no later than one year (366 days)
-    // in the future.
+    // store.
+    //
+    // Legacy Feature Store: For streaming ingestion, the time, aligned by days,
+    // must be no older than five years (1825 days) and no later than one year
+    // (366 days) in the future.
     google.protobuf.Timestamp generate_time = 1;
   }
 
