feat: add Vertex Multimodal Datasets as input and output types in google.cloud.aiplatform.v1beta1.BatchPredictionJob

Google APIs · copybara-github · commit 58b1634863a6 · 2026-02-06T03:24:09.000-08:00
PiperOrigin-RevId: 866379511
diff --git a/google/cloud/aiplatform/v1beta1/batch_prediction_job.proto b/google/cloud/aiplatform/v1beta1/batch_prediction_job.proto
@@ -1,4 +1,4 @@
-// Copyright 2025 Google LLC
+// Copyright 2026 Google LLC
 //
 // Licensed under the Apache License, Version 2.0 (the "License");
 // you may not use this file except in compliance with the License.
@@ -70,6 +70,10 @@ message BatchPredictionJob {
       // additional columns that are not described by the schema, and they will
       // be ignored.
       BigQuerySource bigquery_source = 3;
+
+      // A Vertex Managed Dataset. Currently, only datasets of type Multimodal
+      // are supported.
+      VertexMultimodalDatasetSource vertex_multimodal_dataset_source = 4;
     }
 
     // Required. The format in which instances are given, must be one of the
@@ -230,6 +234,11 @@ message BatchPredictionJob {
       // has [google.rpc.Status][google.rpc.Status]
       // represented as a STRUCT, and containing only `code` and `message`.
       BigQueryDestination bigquery_destination = 3;
+
+      // The details for a Vertex Multimodal Dataset that will be created for
+      // the output.
+      VertexMultimodalDatasetDestination vertex_multimodal_dataset_destination =
+          6;
     }
 
     // Required. The format in which Vertex AI gives the predictions, must be
@@ -255,6 +264,16 @@ message BatchPredictionJob {
       // format, into which the prediction output is written.
       string bigquery_output_dataset = 2
           [(google.api.field_behavior) = OUTPUT_ONLY];
+
+      // Output only. The resource name of the Vertex Managed Dataset created,
+      // into which the prediction output is written. Format:
+      // `projects/{project}/locations/{location}/datasets/{dataset}`
+      string vertex_multimodal_dataset_name = 5 [
+        (google.api.field_behavior) = OUTPUT_ONLY,
+        (google.api.resource_reference) = {
+          type: "aiplatform.googleapis.com/Dataset"
+        }
+      ];
     }
 
     // Output only. The name of the BigQuery table created, in
diff --git a/google/cloud/aiplatform/v1beta1/io.proto b/google/cloud/aiplatform/v1beta1/io.proto
@@ -1,4 +1,4 @@
-// Copyright 2025 Google LLC
+// Copyright 2026 Google LLC
 //
 // Licensed under the Apache License, Version 2.0 (the "License");
 // you may not use this file except in compliance with the License.
@@ -17,6 +17,7 @@ syntax = "proto3";
 package google.cloud.aiplatform.v1beta1;
 
 import "google/api/field_behavior.proto";
+import "google/api/resource.proto";
 import "google/cloud/aiplatform/v1beta1/api_auth.proto";
 import "google/protobuf/timestamp.proto";
 
@@ -82,6 +83,30 @@ message BigQueryDestination {
   string output_uri = 1 [(google.api.field_behavior) = REQUIRED];
 }
 
+// The Vertex Multimodal Dataset for the input content.
+message VertexMultimodalDatasetSource {
+  // Required. The resource name of the Vertex Dataset.
+  // Format: `projects/{project}/locations/{location}/datasets/{dataset}`
+  string dataset_name = 1 [
+    (google.api.field_behavior) = REQUIRED,
+    (google.api.resource_reference) = {
+      type: "aiplatform.googleapis.com/Dataset"
+    }
+  ];
+}
+
+// The details for a Vertex Multimodal Dataset output.
+message VertexMultimodalDatasetDestination {
+  // Optional. The destination of the underlying BigQuery table that will be
+  // created for the output Multimodal Dataset. If not specified, the BigQuery
+  // table will be created in a default BigQuery dataset.
+  BigQueryDestination bigquery_destination = 1
+      [(google.api.field_behavior) = OPTIONAL];
+
+  // Optional. Display name of the output dataset.
+  string display_name = 2 [(google.api.field_behavior) = OPTIONAL];
+}
+
 // The storage details for CSV output content.
 message CsvDestination {
   // Required. Google Cloud Storage location.
