feat: add WriteFeatureValues in aiplatform v1beta1 featurestore_online_service.proto

Google APIs · copybara-github · commit 0dd98150b55b · 2022-08-11T19:50:27.000-07:00
PiperOrigin-RevId: 467107520
diff --git a/google/cloud/aiplatform/v1beta1/aiplatform_v1beta1.yaml b/google/cloud/aiplatform/v1beta1/aiplatform_v1beta1.yaml
@@ -131,9 +131,7 @@ backend:
     deadline: 30.0
   - selector: 'google.cloud.aiplatform.v1beta1.EndpointService.*'
     deadline: 60.0
-  - selector: google.cloud.aiplatform.v1beta1.FeaturestoreOnlineServingService.ReadFeatureValues
-    deadline: 60.0
-  - selector: google.cloud.aiplatform.v1beta1.FeaturestoreOnlineServingService.StreamingReadFeatureValues
+  - selector: 'google.cloud.aiplatform.v1beta1.FeaturestoreOnlineServingService.*'
     deadline: 60.0
   - selector: 'google.cloud.aiplatform.v1beta1.FeaturestoreService.*'
     deadline: 60.0
@@ -530,11 +528,7 @@ authentication:
     oauth:
       canonical_scopes: |-
         https://www.googleapis.com/auth/cloud-platform
-  - selector: google.cloud.aiplatform.v1beta1.FeaturestoreOnlineServingService.ReadFeatureValues
-    oauth:
-      canonical_scopes: |-
-        https://www.googleapis.com/auth/cloud-platform
-  - selector: google.cloud.aiplatform.v1beta1.FeaturestoreOnlineServingService.StreamingReadFeatureValues
+  - selector: 'google.cloud.aiplatform.v1beta1.FeaturestoreOnlineServingService.*'
     oauth:
       canonical_scopes: |-
         https://www.googleapis.com/auth/cloud-platform
diff --git a/google/cloud/aiplatform/v1beta1/featurestore_online_service.proto b/google/cloud/aiplatform/v1beta1/featurestore_online_service.proto
@@ -58,6 +58,55 @@ service FeaturestoreOnlineServingService {
     };
     option (google.api.method_signature) = "entity_type";
   }
+
+  // Writes Feature values of one or more entities of an EntityType.
+  //
+  // The Feature values are merged into existing entities if any. The Feature
+  // values to be written must have timestamp within the online storage
+  // retention.
+  rpc WriteFeatureValues(WriteFeatureValuesRequest) returns (WriteFeatureValuesResponse) {
+    option (google.api.http) = {
+      post: "/v1beta1/{entity_type=projects/*/locations/*/featurestores/*/entityTypes/*}:writeFeatureValues"
+      body: "*"
+    };
+    option (google.api.method_signature) = "entity_type,payloads";
+  }
+}
+
+// Request message for [FeaturestoreOnlineServingService.WriteFeatureValues][google.cloud.aiplatform.v1beta1.FeaturestoreOnlineServingService.WriteFeatureValues].
+message WriteFeatureValuesRequest {
+  // 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`.
+  string entity_type = 1 [
+    (google.api.field_behavior) = REQUIRED,
+    (google.api.resource_reference) = {
+      type: "aiplatform.googleapis.com/EntityType"
+    }
+  ];
+
+  // Required. The entities to be written. Up to 100,000 feature values can be written
+  // across all `payloads`.
+  repeated WriteFeatureValuesPayload payloads = 2 [(google.api.field_behavior) = REQUIRED];
+}
+
+// Contains Feature values to be written for a specific entity.
+message WriteFeatureValuesPayload {
+  // Required. The ID of the entity.
+  string entity_id = 1 [(google.api.field_behavior) = REQUIRED];
+
+  // Required. Feature values to be written, mapping from Feature ID to value. Up to
+  // 100,000 `feature_values` entries may be written across all payloads.  The
+  // feature generation time, aligned by days, must be no older than five
+  // years (1825 days) and no later than one year (366 days) in the future.
+  map<string, FeatureValue> feature_values = 2 [(google.api.field_behavior) = REQUIRED];
+}
+
+// Response message for [FeaturestoreOnlineServingService.WriteFeatureValues][google.cloud.aiplatform.v1beta1.FeaturestoreOnlineServingService.WriteFeatureValues].
+message WriteFeatureValuesResponse {
+
 }
 
 // Request message for [FeaturestoreOnlineServingService.ReadFeatureValues][google.cloud.aiplatform.v1beta1.FeaturestoreOnlineServingService.ReadFeatureValues].
