feat: Add new embed_content_config to message EmbedContentRequest

Google APIs · copybara-github · commit 7feb58252a4c · 2026-03-17T17:21:42.000-07:00
PiperOrigin-RevId: 885289577
diff --git a/google/cloud/aiplatform/v1beta1/prediction_service.proto b/google/cloud/aiplatform/v1beta1/prediction_service.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.
@@ -964,6 +964,38 @@ message GenerateVideoResponse {
 // Request message for
 // [PredictionService.EmbedContent][google.cloud.aiplatform.v1beta1.PredictionService.EmbedContent].
 message EmbedContentRequest {
+  // Configurations for the EmbedContent API.
+  message EmbedContentConfig {
+    // Optional. The title for the text.
+    //
+    // Only applicable to text-only embedding models.
+    optional string title = 1 [(google.api.field_behavior) = OPTIONAL];
+
+    // Optional. The task type of the embedding.
+    //
+    // Only applicable to text-only embedding models.
+    optional EmbeddingTaskType task_type = 2
+        [(google.api.field_behavior) = OPTIONAL];
+
+    // Optional. Whether to silently truncate the input content if it's longer
+    // than the maximum sequence length.
+    //
+    // Only applicable to text-only embedding models.
+    optional bool auto_truncate = 3 [(google.api.field_behavior) = OPTIONAL];
+
+    // Optional. Reduced dimension for the output embedding. If set, excessive
+    // values in the output embedding are truncated from the end.
+    optional int32 output_dimensionality = 4
+        [(google.api.field_behavior) = OPTIONAL];
+
+    // Optional. Whether to enable OCR for document content.
+    optional bool document_ocr = 5 [(google.api.field_behavior) = OPTIONAL];
+
+    // Optional. Whether to extract audio from video content.
+    optional bool audio_track_extraction = 6
+        [(google.api.field_behavior) = OPTIONAL];
+  }
+
   // Represents a downstream task the embeddings will be used for.
   enum EmbeddingTaskType {
     // Unset value, which will default to one of the other enum values.
@@ -1001,24 +1033,34 @@ message EmbedContentRequest {
     type: "aiplatform.googleapis.com/Endpoint"
   }];
 
-  // Required. Input content to be embedded. Required.
+  // Required. Input content to be embedded.
   optional Content content = 2;
 
-  // Optional. An optional title for the text.
-  optional string title = 4 [(google.api.field_behavior) = OPTIONAL];
+  // Optional. Deprecated: Please use EmbedContentConfig.title instead.
+  // The title for the text.
+  optional string title = 4
+      [deprecated = true, (google.api.field_behavior) = OPTIONAL];
 
-  // Optional. The task type of the embedding.
+  // Optional. Deprecated: Please use EmbedContentConfig.task_type instead.
+  // The task type of the embedding.
   optional EmbeddingTaskType task_type = 5
-      [(google.api.field_behavior) = OPTIONAL];
+      [deprecated = true, (google.api.field_behavior) = OPTIONAL];
 
-  // Optional. Optional reduced dimension for the output embedding. If set,
-  // excessive values in the output embedding are truncated from the end.
+  // Optional. Deprecated: Please use EmbedContentConfig.output_dimensionality
+  // instead. Reduced dimension for the output embedding. If set, excessive
+  // values in the output embedding are truncated from the end.
   optional int32 output_dimensionality = 6
-      [(google.api.field_behavior) = OPTIONAL];
+      [deprecated = true, (google.api.field_behavior) = OPTIONAL];
 
-  // Optional. Whether to silently truncate the input content if it's longer
+  // Optional. Deprecated: Please use EmbedContentConfig.auto_truncate instead.
+  // Whether to silently truncate the input content if it's longer
   // than the maximum sequence length.
-  optional bool auto_truncate = 7 [(google.api.field_behavior) = OPTIONAL];
+  optional bool auto_truncate = 7
+      [deprecated = true, (google.api.field_behavior) = OPTIONAL];
+
+  // Optional. Configuration for the EmbedContent request.
+  optional EmbedContentConfig embed_content_config = 8
+      [(google.api.field_behavior) = OPTIONAL];
 }
 
 // Response message for
@@ -1033,7 +1075,7 @@ message EmbedContentResponse {
   // The embedding generated from the input content.
   Embedding embedding = 1;
 
-  // Metadata about the response(s).
+  // Usage metadata about the response(s).
   UsageMetadata usage_metadata = 2;
 
   // Whether the input content was truncated before generating the embedding.
