feat: add gemini_request_read_config field to AssembleDataRequest and AssessDataRequest in aiplatform v1beta

Google APIs · copybara-github · commit 3cd5070263af · 2025-04-29T02:32:12.000-07:00
docs: fix comment for field_mapping field in GeminiTemplateConfig

PiperOrigin-RevId: 752648609
diff --git a/google/cloud/aiplatform/v1beta1/dataset_service.proto b/google/cloud/aiplatform/v1beta1/dataset_service.proto
@@ -1004,6 +1004,10 @@ message AssessDataRequest {
       type: "aiplatform.googleapis.com/Dataset"
     }
   ];
+
+  // Optional. The Gemini request read config for the dataset.
+  GeminiRequestReadConfig gemini_request_read_config = 8
+      [(google.api.field_behavior) = OPTIONAL];
 }
 
 // Response message for
@@ -1072,11 +1076,25 @@ message GeminiTemplateConfig {
   // for downstream applications.
   GeminiExample gemini_example = 1 [(google.api.field_behavior) = REQUIRED];
 
-  // Required. Map of template params to the columns in the dataset table.
+  // Required. Map of template parameters to the columns in the dataset table.
   map<string, string> field_mapping = 2
       [(google.api.field_behavior) = REQUIRED];
 }
 
+// Configuration for how to read Gemini requests from a multimodal dataset.
+message GeminiRequestReadConfig {
+  // The read config for the dataset.
+  oneof read_config {
+    // Gemini request template with placeholders.
+    GeminiTemplateConfig template_config = 1;
+
+    // Optional. Column name in the dataset table that contains already fully
+    // assembled Gemini requests.
+    string assembled_request_column_name = 4
+        [(google.api.field_behavior) = OPTIONAL];
+  }
+}
+
 // Format for Gemini examples used for Vertex Multimodal datasets.
 message GeminiExample {
   // Optional. The fully qualified name of the publisher model or tuned model
@@ -1175,6 +1193,10 @@ message AssembleDataRequest {
       type: "aiplatform.googleapis.com/Dataset"
     }
   ];
+
+  // Optional. The read config for the dataset.
+  GeminiRequestReadConfig gemini_request_read_config = 6
+      [(google.api.field_behavior) = OPTIONAL];
 }
 
 // Response message for
