Skip to content
This repository was archived by the owner on Sep 20, 2023. It is now read-only.

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

Merged
parthea merged 4 commits into from
Dec 13, 2022
Merged

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

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
4 commits
Select commit Hold shift + click to select a range
5709f2a
feat: added sharding_config field in DocumentOutputConfig.GcsOutputCo…
gcf-owl-bot[bot] Dec 13, 2022
97281f3
🦉 Updates from OwlBot post-processor
gcf-owl-bot[bot] Dec 13, 2022
d6d2d4a
feat: added sharding_config field in DocumentOutputConfig.GcsOutputCo…
gcf-owl-bot[bot] Dec 13, 2022
43f3ac3
🦉 Updates from OwlBot post-processor
gcf-owl-bot[bot] Dec 13, 2022
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
30 changes: 30 additions & 0 deletions google/cloud/documentai_v1/types/document_io.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -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,
Expand All @@ -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,
Expand Down
30 changes: 30 additions & 0 deletions google/cloud/documentai_v1beta3/types/document_io.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -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,
Expand All @@ -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,
Expand Down
2 changes: 1 addition & 1 deletion samples/generated_samples/snippet_metadata_google.cloud.documentai.v1.json
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -8,7 +8,7 @@
],
"language": "PYTHON",
"name": "google-cloud-documentai",
"version": "2.4.1"
"version": "0.1.0"
},
"snippets": [
{
Expand Down
2 changes: 1 addition & 1 deletion samples/generated_samples/snippet_metadata_google.cloud.documentai.v1beta2.json
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -8,7 +8,7 @@
],
"language": "PYTHON",
"name": "google-cloud-documentai",
"version": "2.4.1"
"version": "0.1.0"
},
"snippets": [
{
Expand Down
2 changes: 1 addition & 1 deletion samples/generated_samples/snippet_metadata_google.cloud.documentai.v1beta3.json
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -8,7 +8,7 @@
],
"language": "PYTHON",
"name": "google-cloud-documentai",
"version": "2.4.1"
"version": "0.1.0"
},
"snippets": [
{
Expand Down