feat: add FeatureViewDirectWrite API

Google APIs · copybara-github · commit e53d98d874c3 · 2025-04-03T16:36:16.000-07:00
PiperOrigin-RevId: 743740095
diff --git a/google/cloud/aiplatform/v1beta1/feature_online_store_service.proto b/google/cloud/aiplatform/v1beta1/feature_online_store_service.proto
@@ -22,6 +22,7 @@ import "google/api/field_behavior.proto";
 import "google/api/resource.proto";
 import "google/cloud/aiplatform/v1beta1/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.V1Beta1";
@@ -70,6 +71,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: "/v1beta1/{feature_view=projects/*/locations/*/featureOnlineStores/*/featureViews/*}:directWrite"
+      body: "*"
+    };
+  }
 }
 
 // Format of the data in the Feature View.
@@ -409,3 +421,85 @@ message SearchNearestEntitiesResponse {
   // The nearest neighbors of the query entity.
   NearestNeighbors nearest_neighbors = 1;
 }
+
+// Request message for
+// [FeatureOnlineStoreService.FeatureViewDirectWrite][google.cloud.aiplatform.v1beta1.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 and timestamp.
+      message FeatureValueAndTimestamp {
+        // The feature value.
+        FeatureValue value = 1;
+
+        // The feature timestamp to store with this value.
+        // If not set, then the Feature Store server will generate a timestamp
+        // when it receives the write request.
+        google.protobuf.Timestamp timestamp = 2;
+      }
+
+      // Feature value data to write.
+      oneof data_oneof {
+        // Feature value and timestamp.
+        FeatureValueAndTimestamp value_and_timestamp = 2;
+      }
+
+      // 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.v1beta1.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.v1beta1.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.v1beta1.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.v1beta1.FeatureViewDirectWriteResponse.WriteResponse.data_key]
+  // will have the key with error, but
+  // [WriteResponse.online_store_write_time][google.cloud.aiplatform.v1beta1.FeatureViewDirectWriteResponse.WriteResponse.online_store_write_time]
+  // will not be present.
+  repeated WriteResponse write_responses = 2;
+}
