feat: add NVIDIA_A100_80GB to AcceleratorType

feat: add blocking_operation_ids to ImportFeatureValuesOperationMetadata

PiperOrigin-RevId: 537124229

google/cloud/aiplatform/v1/accelerator_type.proto

@@ -47,6 +47,9 @@ enum AcceleratorType {
   // Nvidia Tesla A100 GPU.
   NVIDIA_TESLA_A100 = 8;
 
+  // Nvidia A100 80GB GPU.
+  NVIDIA_A100_80GB = 9;
+
   // Nvidia L4 GPU.
   NVIDIA_L4 = 11;
 

google/cloud/aiplatform/v1/aiplatform_v1.yaml

@@ -706,3 +706,9 @@ publishing:
         datalabeling.googleapis.com/Dataset: DataLabelingDataset
         automl.googleapis.com/Dataset: AutoMLDataset
         automl.googleapis.com/Model: AutoMLModel
+  - version: google.cloud.aiplatform.v1beta1
+    dotnet_settings:
+      renamed_resources:
+        datalabeling.googleapis.com/Dataset: DataLabelingDataset
+        automl.googleapis.com/Dataset: AutoMLDataset
+        automl.googleapis.com/Model: AutoMLModel

google/cloud/aiplatform/v1/featurestore_service.proto

@@ -534,7 +534,7 @@ message ImportFeatureValuesRequest {
   ];
 
   // Source column that holds entity IDs. If not provided, entity IDs are
-  // extracted from the column named `entity_id`.
+  // extracted from the column named entity_id.
   string entity_id_field = 5;
 
   // Required. Specifications defining which Feature values to import from the
@@ -1288,6 +1288,10 @@ message ImportFeatureValuesOperationMetadata {
   // The number rows that weren't ingested due to having timestamps outside the
   // retention boundary.
   int64 timestamp_outside_retention_rows_count = 7;
+
+  // List of ImportFeatureValues operations running under a single EntityType
+  // that are blocking this operation.
+  repeated int64 blocking_operation_ids = 8;
 }
 
 // Details of operations that exports Features values.
@@ -1439,6 +1443,6 @@ message EntityIdSelector {
   }
 
   // Source column that holds entity IDs. If not provided, entity IDs are
-  // extracted from the column named `entity_id`.
+  // extracted from the column named entity_id.
   string entity_id_field = 5;
 }

google/cloud/aiplatform/v1/index_endpoint.proto

@@ -159,17 +159,16 @@ message DeployedIndex {
   // Output only. The DeployedIndex may depend on various data on its original
   // Index. Additionally when certain changes to the original Index are being
   // done (e.g. when what the Index contains is being changed) the DeployedIndex
-  // may be asynchronously updated in the background to reflect this changes. If
-  // this timestamp's value is at least the
+  // may be asynchronously updated in the background to reflect these changes.
+  // If this timestamp's value is at least the
   // [Index.update_time][google.cloud.aiplatform.v1.Index.update_time] of the
   // original Index, it means that this DeployedIndex and the original Index are
   // in sync. If this timestamp is older, then to see which updates this
-  // DeployedIndex already contains (and which not), one must
-  // [list][Operations.ListOperations] [Operations][Operation]
-  // [working][Operation.name] on the original Index. Only
-  // the successfully completed Operations with
-  // [Operations.metadata.generic_metadata.update_time]
-  // [google.cloud.aiplatform.v1.GenericOperationMetadata.update_time]
+  // DeployedIndex already contains (and which it does not), one must
+  // [list][google.longrunning.Operations.ListOperations] the operations that
+  // are running on the original Index. Only the successfully completed
+  // Operations with
+  // [update_time][google.cloud.aiplatform.v1.GenericOperationMetadata.update_time]
   // equal or before this sync time are contained in this DeployedIndex.
   google.protobuf.Timestamp index_sync_time = 6
       [(google.api.field_behavior) = OUTPUT_ONLY];

google/cloud/aiplatform/v1/metadata_service.proto

@@ -582,6 +582,9 @@ message ListArtifactsRequest {
   //     To filter on metadata fields use traversal operation as follows:
   //     `metadata.<field_name>.<type_value>`.
   //     For example: `metadata.field_1.number_value = 10.0`
+  //     In case the field name contains special characters (such as colon), one
+  //     can embed it inside double quote.
+  //     For example: `metadata."field:1".number_value = 10.0`
   // *   **Context based filtering**:
   //     To filter Artifacts based on the contexts to which they belong, use the
   //     function operator with the full resource name
@@ -628,7 +631,6 @@ message UpdateArtifactRequest {
   Artifact artifact = 1 [(google.api.field_behavior) = REQUIRED];
 
   // Optional. A FieldMask indicating which fields should be updated.
-  // Functionality of this field is not yet supported.
   google.protobuf.FieldMask update_mask = 2
       [(google.api.field_behavior) = OPTIONAL];
 
@@ -784,6 +786,9 @@ message ListContextsRequest {
   //    To filter on metadata fields use traversal operation as follows:
   //    `metadata.<field_name>.<type_value>`.
   //    For example: `metadata.field_1.number_value = 10.0`.
+  //    In case the field name contains special characters (such as colon), one
+  //    can embed it inside double quote.
+  //    For example: `metadata."field:1".number_value = 10.0`
   // *  **Parent Child filtering**:
   //    To filter Contexts based on parent-child relationship use the HAS
   //    operator as follows:
@@ -834,7 +839,6 @@ message UpdateContextRequest {
   Context context = 1 [(google.api.field_behavior) = REQUIRED];
 
   // Optional. A FieldMask indicating which fields should be updated.
-  // Functionality of this field is not yet supported.
   google.protobuf.FieldMask update_mask = 2
       [(google.api.field_behavior) = OPTIONAL];
 
@@ -1097,6 +1101,9 @@ message ListExecutionsRequest {
   //    To filter on metadata fields use traversal operation as follows:
   //    `metadata.<field_name>.<type_value>`
   //    For example: `metadata.field_1.number_value = 10.0`
+  //    In case the field name contains special characters (such as colon), one
+  //    can embed it inside double quote.
+  //    For example: `metadata."field:1".number_value = 10.0`
   // *  **Context based filtering**:
   //    To filter Executions based on the contexts to which they belong use
   //    the function operator with the full resource name:
@@ -1143,7 +1150,6 @@ message UpdateExecutionRequest {
   Execution execution = 1 [(google.api.field_behavior) = REQUIRED];
 
   // Optional. A FieldMask indicating which fields should be updated.
-  // Functionality of this field is not yet supported.
   google.protobuf.FieldMask update_mask = 2
       [(google.api.field_behavior) = OPTIONAL];
 
@@ -1379,6 +1385,9 @@ message QueryArtifactLineageSubgraphRequest {
   //    To filter on metadata fields use traversal operation as follows:
   //    `metadata.<field_name>.<type_value>`.
   //    For example: `metadata.field_1.number_value = 10.0`
+  //    In case the field name contains special characters (such as colon), one
+  //    can embed it inside double quote.
+  //    For example: `metadata."field:1".number_value = 10.0`
   //
   // Each of the above supported filter types can be combined together using
   // logical operators (`AND` & `OR`). Maximum nested expression depth allowed

google/cloud/aiplatform/v1/model_evaluation.proto

@@ -71,8 +71,9 @@ message ModelEvaluation {
   google.protobuf.Timestamp create_time = 4
       [(google.api.field_behavior) = OUTPUT_ONLY];
 
-  // All possible [dimensions][ModelEvaluationSlice.slice.dimension] of
-  // ModelEvaluationSlices. The dimensions can be used as the filter of the
+  // All possible
+  // [dimensions][google.cloud.aiplatform.v1.ModelEvaluationSlice.Slice.dimension]
+  // of ModelEvaluationSlices. The dimensions can be used as the filter of the
   // [ModelService.ListModelEvaluationSlices][google.cloud.aiplatform.v1.ModelService.ListModelEvaluationSlices]
   // request, in the form of `slice.dimension = <dimension>`.
   repeated string slice_dimensions = 5;

google/cloud/aiplatform/v1/prediction_service.proto

@@ -84,8 +84,7 @@ service PredictionService {
   // [deployed_model_id][google.cloud.aiplatform.v1.ExplainRequest.deployed_model_id]
   // is not specified, all DeployedModels must have
   // [explanation_spec][google.cloud.aiplatform.v1.DeployedModel.explanation_spec]
-  // populated. Only deployed AutoML tabular Models have
-  // explanation_spec.
+  // populated.
   rpc Explain(ExplainRequest) returns (ExplainResponse) {
     option (google.api.http) = {
       post: "/v1/{endpoint=projects/*/locations/*/endpoints/*}:explain"

google/cloud/aiplatform/v1/tensorboard_experiment.proto

@@ -55,7 +55,8 @@ message TensorboardExperiment {
   google.protobuf.Timestamp update_time = 5
       [(google.api.field_behavior) = OUTPUT_ONLY];
 
-  // The labels with user-defined metadata to organize your Datasets.
+  // The labels with user-defined metadata to organize your
+  // TensorboardExperiment.
   //
   // Label keys and values cannot be longer than 64 characters
   // (Unicode codepoints), can only contain lowercase letters, numeric

google/cloud/aiplatform/v1/vizier_service.proto

@@ -92,7 +92,8 @@ service VizierService {
   // suggested by Vertex AI Vizier. Returns a long-running
   // operation associated with the generation of Trial suggestions.
   // When this long-running operation succeeds, it will contain
-  // a [SuggestTrialsResponse][google.cloud.ml.v1.SuggestTrialsResponse].
+  // a
+  // [SuggestTrialsResponse][google.cloud.aiplatform.v1.SuggestTrialsResponse].
   rpc SuggestTrials(SuggestTrialsRequest)
       returns (google.longrunning.Operation) {
     option (google.api.http) = {