Skip to content

Commit

Permalink
feat: enable FeatureView Service Agents
Browse files Browse the repository at this point in the history
PiperOrigin-RevId: 609447516
  • Loading branch information
Google APIs authored and Copybara-Service committed Feb 22, 2024
1 parent 2f14a03 commit 694219f
Showing 1 changed file with 31 additions and 0 deletions.
31 changes: 31 additions & 0 deletions google/cloud/aiplatform/v1beta1/feature_view.proto
Original file line number Diff line number Diff line change
Expand Up @@ -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.
Expand Down Expand Up @@ -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];
}

0 comments on commit 694219f

Please sign in to comment.