feat: add model_monitoring_config to BatchPredictionJob in aiplatform v1beta1 batch_prediction_job.proto

Google APIs · copybara-github · commit 3b9c451c749a · 2022-06-21T13:37:24.000-07:00
PiperOrigin-RevId: 456339444
diff --git a/google/cloud/aiplatform/v1beta1/batch_prediction_job.proto b/google/cloud/aiplatform/v1beta1/batch_prediction_job.proto
@@ -25,6 +25,7 @@ import "google/cloud/aiplatform/v1beta1/io.proto";
 import "google/cloud/aiplatform/v1beta1/job_state.proto";
 import "google/cloud/aiplatform/v1beta1/machine_resources.proto";
 import "google/cloud/aiplatform/v1beta1/manual_batch_tuning_parameters.proto";
+import "google/cloud/aiplatform/v1beta1/model_monitoring.proto";
 import "google/cloud/aiplatform/v1beta1/unmanaged_container_model.proto";
 import "google/protobuf/struct.proto";
 import "google/protobuf/timestamp.proto";
@@ -298,4 +299,9 @@ message BatchPredictionJob {
   // is set, then all resources created by the BatchPredictionJob will be
   // encrypted with the provided encryption key.
   EncryptionSpec encryption_spec = 24;
+
+  // Model monitoring config will be used for analysis model behaviors, based on
+  // the input and output to the batch prediction job, as well as the provided
+  // training dataset.
+  ModelMonitoringConfig model_monitoring_config = 26;
 }
diff --git a/google/cloud/aiplatform/v1beta1/model_monitoring.proto b/google/cloud/aiplatform/v1beta1/model_monitoring.proto
@@ -27,6 +27,24 @@ option java_package = "com.google.cloud.aiplatform.v1beta1";
 option php_namespace = "Google\\Cloud\\AIPlatform\\V1beta1";
 option ruby_package = "Google::Cloud::AIPlatform::V1beta1";
 
+// Next ID: 5
+message ModelMonitoringConfig {
+  // Model monitoring objective config.
+  repeated ModelMonitoringObjectiveConfig objective_configs = 3;
+
+  // Model monitoring alert config.
+  ModelMonitoringAlertConfig alert_config = 2;
+
+  // YAML schema file uri in Cloud Storage describing the format of a single
+  // instance that you want Tensorflow Data Validation (TFDV) to analyze.
+  //
+  // If there are any data type differences between predict instance and TFDV
+  // instance, this field can be used to override the schema.
+  // For models trained with Vertex AI, this field must be set as all the
+  // fields in predict instance formatted as string.
+  string analysis_instance_schema_uri = 4;
+}
+
 // Next ID: 8
 message ModelMonitoringObjectiveConfig {
   // Training Dataset information.
