feat: enable FeatureView Service Agents

Google APIs · copybara-github · commit 694219fe5d75 · 2024-02-22T11:48:35.000-08:00
PiperOrigin-RevId: 609447516
diff --git a/google/cloud/aiplatform/v1beta1/feature_view.proto b/google/cloud/aiplatform/v1beta1/feature_view.proto
@@ -147,6 +147,22 @@ message FeatureView {
         [(google.api.field_behavior) = REQUIRED];
   }
 
+  // Service agent type used during data sync.
+  enum ServiceAgentType {
+    // By default, the project-level Vertex AI Service Agent is enabled.
+    SERVICE_AGENT_TYPE_UNSPECIFIED = 0;
+
+    // Indicates the project-level Vertex AI Service Agent
+    // (https://cloud.google.com/vertex-ai/docs/general/access-control#service-agents)
+    // will be used during sync jobs.
+    SERVICE_AGENT_TYPE_PROJECT = 1;
+
+    // Enable a FeatureView service account to be created by Vertex AI and
+    // output in the field `service_account_email`. This service account will
+    // be used to read from the source BigQuery table during sync.
+    SERVICE_AGENT_TYPE_FEATURE_VIEW = 2;
+  }
+
   oneof source {
     // Optional. Configures how data is supposed to be extracted from a BigQuery
     // source to be loaded onto the FeatureOnlineStore.
@@ -199,4 +215,19 @@ message FeatureView {
   // online serving.
   VectorSearchConfig vector_search_config = 8
       [(google.api.field_behavior) = OPTIONAL];
+
+  // Optional. Service agent type used during data sync. By default, the Vertex
+  // AI Service Agent is used. When using an IAM Policy to isolate this
+  // FeatureView within a project
+  // (https://cloud.google.com/vertex-ai/docs/featurestore/latest/resource-policy)
+  // a separate service account should be provisioned by
+  // setting this field to `SERVICE_AGENT_TYPE_FEATURE_VIEW`. This will generate
+  // a separate service account to access the BigQuery source table.
+  ServiceAgentType service_agent_type = 14
+      [(google.api.field_behavior) = OPTIONAL];
+
+  // Output only. A Service Account unique to this FeatureView. The role
+  // bigquery.dataViewer should be granted to this service account to allow
+  // Vertex AI Feature Store to sync data to the online store.
+  string service_account_email = 13 [(google.api.field_behavior) = OUTPUT_ONLY];
 }
