diff --git a/packages/google-cloud-contentwarehouse/protos/google/cloud/documentai/v1/document_io.proto b/packages/google-cloud-contentwarehouse/protos/google/cloud/documentai/v1/document_io.proto
index 3240d494e8f..d56086336d2 100644
--- a/packages/google-cloud-contentwarehouse/protos/google/cloud/documentai/v1/document_io.proto
+++ b/packages/google-cloud-contentwarehouse/protos/google/cloud/documentai/v1/document_io.proto
@@ -74,6 +74,15 @@ message BatchDocumentsInputConfig {
 message DocumentOutputConfig {
   // The configuration used when outputting documents.
   message GcsOutputConfig {
+    // The sharding config for the output document.
+    message ShardingConfig {
+      // The number of pages per shard.
+      int32 pages_per_shard = 1;
+
+      // The number of overlapping pages between consecutive shards.
+      int32 pages_overlap = 2;
+    }
+
     // The Cloud Storage uri (a directory) of the output.
     string gcs_uri = 1;
 
@@ -81,6 +90,9 @@ message DocumentOutputConfig {
     // Only supports top level document and pages field so it must be in the
     // form of `{document_field_name}` or `pages.{page_field_name}`.
     google.protobuf.FieldMask field_mask = 2;
+
+    // Specifies the sharding config for the output document.
+    ShardingConfig sharding_config = 3;
   }
 
   // The destination of the results.