feat: added sharding_config field in DocumentOutputConfig.GcsOutputConfig in document_io.proto (#430)

gcf-owl-bot[bot] · web-flow · commit 80df6cb24b2e · 2022-12-13T17:07:10.000-05:00
* feat: added sharding_config field in DocumentOutputConfig.GcsOutputConfig in document_io.proto PiperOrigin-RevId: 495050408 Source-Link: googleapis/googleapis@46a1452 Source-Link: https://github.com/googleapis/googleapis-gen/commit/865eeff1ce9e70e0989d6b0213ae6a5c987b10d7 Copy-Tag: eyJwIjoiLmdpdGh1Yi8uT3dsQm90LnlhbWwiLCJoIjoiODY1ZWVmZjFjZTllNzBlMDk4OWQ2YjAyMTNhZTZhNWM5ODdiMTBkNyJ9 * 🦉 Updates from OwlBot post-processor See https://github.com/googleapis/repo-automation-bots/blob/main/packages/owl-bot/README.md * feat: added sharding_config field in DocumentOutputConfig.GcsOutputConfig in document_io.proto PiperOrigin-RevId: 495051185 Source-Link: googleapis/googleapis@0ff0b24 Source-Link: https://github.com/googleapis/googleapis-gen/commit/ffa8b5f51bd4258481b8fc73ca7370f78847d5d7 Copy-Tag: eyJwIjoiLmdpdGh1Yi8uT3dsQm90LnlhbWwiLCJoIjoiZmZhOGI1ZjUxYmQ0MjU4NDgxYjhmYzczY2E3MzcwZjc4ODQ3ZDVkNyJ9 * 🦉 Updates from OwlBot post-processor See https://github.com/googleapis/repo-automation-bots/blob/main/packages/owl-bot/README.md Co-authored-by: Owl Bot <gcf-owl-bot[bot]@users.noreply.github.com>
diff --git a/google/cloud/documentai_v1/types/document_io.py b/google/cloud/documentai_v1/types/document_io.py
@@ -168,8 +168,31 @@ class GcsOutputConfig(proto.Message):
                 Only supports top level document and pages field so it must
                 be in the form of ``{document_field_name}`` or
                 ``pages.{page_field_name}``.
+            sharding_config (google.cloud.documentai_v1.types.DocumentOutputConfig.GcsOutputConfig.ShardingConfig):
+                Specifies the sharding config for the output
+                document.
         """
 
+        class ShardingConfig(proto.Message):
+            r"""The sharding config for the output document.
+
+            Attributes:
+                pages_per_shard (int):
+                    The number of pages per shard.
+                pages_overlap (int):
+                    The number of overlapping pages between
+                    consecutive shards.
+            """
+
+            pages_per_shard: int = proto.Field(
+                proto.INT32,
+                number=1,
+            )
+            pages_overlap: int = proto.Field(
+                proto.INT32,
+                number=2,
+            )
+
         gcs_uri: str = proto.Field(
             proto.STRING,
             number=1,
@@ -179,6 +202,13 @@ class GcsOutputConfig(proto.Message):
             number=2,
             message=field_mask_pb2.FieldMask,
         )
+        sharding_config: "DocumentOutputConfig.GcsOutputConfig.ShardingConfig" = (
+            proto.Field(
+                proto.MESSAGE,
+                number=3,
+                message="DocumentOutputConfig.GcsOutputConfig.ShardingConfig",
+            )
+        )
 
     gcs_output_config: GcsOutputConfig = proto.Field(
         proto.MESSAGE,
diff --git a/google/cloud/documentai_v1beta3/types/document_io.py b/google/cloud/documentai_v1beta3/types/document_io.py
@@ -168,8 +168,31 @@ class GcsOutputConfig(proto.Message):
                 Only supports top level document and pages field so it must
                 be in the form of ``{document_field_name}`` or
                 ``pages.{page_field_name}``.
+            sharding_config (google.cloud.documentai_v1beta3.types.DocumentOutputConfig.GcsOutputConfig.ShardingConfig):
+                Specifies the sharding config for the output
+                document.
         """
 
+        class ShardingConfig(proto.Message):
+            r"""The sharding config for the output document.
+
+            Attributes:
+                pages_per_shard (int):
+                    The number of pages per shard.
+                pages_overlap (int):
+                    The number of overlapping pages between
+                    consecutive shards.
+            """
+
+            pages_per_shard: int = proto.Field(
+                proto.INT32,
+                number=1,
+            )
+            pages_overlap: int = proto.Field(
+                proto.INT32,
+                number=2,
+            )
+
         gcs_uri: str = proto.Field(
             proto.STRING,
             number=1,
@@ -179,6 +202,13 @@ class GcsOutputConfig(proto.Message):
             number=2,
             message=field_mask_pb2.FieldMask,
         )
+        sharding_config: "DocumentOutputConfig.GcsOutputConfig.ShardingConfig" = (
+            proto.Field(
+                proto.MESSAGE,
+                number=3,
+                message="DocumentOutputConfig.GcsOutputConfig.ShardingConfig",
+            )
+        )
 
     gcs_output_config: GcsOutputConfig = proto.Field(
         proto.MESSAGE,
diff --git a/samples/generated_samples/snippet_metadata_google.cloud.documentai.v1.json b/samples/generated_samples/snippet_metadata_google.cloud.documentai.v1.json
@@ -8,7 +8,7 @@
     ],
     "language": "PYTHON",
     "name": "google-cloud-documentai",
-    "version": "2.4.1"
+    "version": "0.1.0"
   },
   "snippets": [
     {
diff --git a/samples/generated_samples/snippet_metadata_google.cloud.documentai.v1beta2.json b/samples/generated_samples/snippet_metadata_google.cloud.documentai.v1beta2.json
@@ -8,7 +8,7 @@
     ],
     "language": "PYTHON",
     "name": "google-cloud-documentai",
-    "version": "2.4.1"
+    "version": "0.1.0"
   },
   "snippets": [
     {
diff --git a/samples/generated_samples/snippet_metadata_google.cloud.documentai.v1beta3.json b/samples/generated_samples/snippet_metadata_google.cloud.documentai.v1beta3.json
@@ -8,7 +8,7 @@
     ],
     "language": "PYTHON",
     "name": "google-cloud-documentai",
-    "version": "2.4.1"
+    "version": "0.1.0"
   },
   "snippets": [
     {

Original file line number	Diff line number	Diff line change
`@@ -8,7 +8,7 @@`
`8`	`8`	`],`
`9`	`9`	`"language": "PYTHON",`
`10`	`10`	`"name": "google-cloud-documentai",`
`11`		`- "version": "2.4.1"`
	`11`	`+ "version": "0.1.0"`
`12`	`12`	`},`
`13`	`13`	`"snippets": [`
`14`	`14`	`{`