feat: add UpdateExplanationDataset to aiplatform

PiperOrigin-RevId: 541213716

google/cloud/aiplatform/v1/model_service.proto

@@ -22,6 +22,7 @@ import "google/api/field_behavior.proto";
 import "google/api/resource.proto";
 import "google/cloud/aiplatform/v1/encryption_spec.proto";
 import "google/cloud/aiplatform/v1/evaluated_annotation.proto";
+import "google/cloud/aiplatform/v1/explanation.proto";
 import "google/cloud/aiplatform/v1/io.proto";
 import "google/cloud/aiplatform/v1/model.proto";
 import "google/cloud/aiplatform/v1/model_evaluation.proto";
@@ -92,6 +93,20 @@ service ModelService {
     option (google.api.method_signature) = "model,update_mask";
   }
 
+  // Incrementally update the dataset used for an examples model.
+  rpc UpdateExplanationDataset(UpdateExplanationDatasetRequest)
+      returns (google.longrunning.Operation) {
+    option (google.api.http) = {
+      post: "/v1/{model=projects/*/locations/*/models/*}:updateExplanationDataset"
+      body: "*"
+    };
+    option (google.api.method_signature) = "model";
+    option (google.longrunning.operation_info) = {
+      response_type: "UpdateExplanationDatasetResponse"
+      metadata_type: "UpdateExplanationDatasetOperationMetadata"
+    };
+  }
+
   // Deletes a Model.
   //
   // A model cannot be deleted if any
@@ -484,6 +499,29 @@ message UpdateModelRequest {
       [(google.api.field_behavior) = REQUIRED];
 }
 
+// Request message for
+// [ModelService.UpdateExplanationDataset][google.cloud.aiplatform.v1.ModelService.UpdateExplanationDataset].
+message UpdateExplanationDatasetRequest {
+  // Required. The resource name of the Model to update.
+  // Format: `projects/{project}/locations/{location}/models/{model}`
+  string model = 1 [
+    (google.api.field_behavior) = REQUIRED,
+    (google.api.resource_reference) = {
+      type: "aiplatform.googleapis.com/Model"
+    }
+  ];
+
+  // The example config containing the location of the dataset.
+  Examples examples = 2;
+}
+
+// Runtime operation information for
+// [ModelService.UpdateExplanationDataset][google.cloud.aiplatform.v1.ModelService.UpdateExplanationDataset].
+message UpdateExplanationDatasetOperationMetadata {
+  // The common part of the operation metadata.
+  GenericOperationMetadata generic_metadata = 1;
+}
+
 // Request message for
 // [ModelService.DeleteModel][google.cloud.aiplatform.v1.ModelService.DeleteModel].
 message DeleteModelRequest {
@@ -584,6 +622,11 @@ message ExportModelRequest {
   OutputConfig output_config = 2 [(google.api.field_behavior) = REQUIRED];
 }
 
+// Response message of
+// [ModelService.UpdateExplanationDataset][google.cloud.aiplatform.v1.ModelService.UpdateExplanationDataset]
+// operation.
+message UpdateExplanationDatasetResponse {}
+
 // Details of
 // [ModelService.ExportModel][google.cloud.aiplatform.v1.ModelService.ExportModel]
 // operation.