diff --git a/packages/google-cloud-documentai/google/cloud/documentai/gapic_version.py b/packages/google-cloud-documentai/google/cloud/documentai/gapic_version.py index b3b122d9344e..558c8aab67c5 100644 --- a/packages/google-cloud-documentai/google/cloud/documentai/gapic_version.py +++ b/packages/google-cloud-documentai/google/cloud/documentai/gapic_version.py @@ -13,4 +13,4 @@ # See the License for the specific language governing permissions and # limitations under the License. # -__version__ = "2.29.3" # {x-release-please-version} +__version__ = "0.0.0" # {x-release-please-version} diff --git a/packages/google-cloud-documentai/google/cloud/documentai_v1/gapic_version.py b/packages/google-cloud-documentai/google/cloud/documentai_v1/gapic_version.py index b3b122d9344e..558c8aab67c5 100644 --- a/packages/google-cloud-documentai/google/cloud/documentai_v1/gapic_version.py +++ b/packages/google-cloud-documentai/google/cloud/documentai_v1/gapic_version.py @@ -13,4 +13,4 @@ # See the License for the specific language governing permissions and # limitations under the License. # -__version__ = "2.29.3" # {x-release-please-version} +__version__ = "0.0.0" # {x-release-please-version} diff --git a/packages/google-cloud-documentai/google/cloud/documentai_v1beta2/gapic_version.py b/packages/google-cloud-documentai/google/cloud/documentai_v1beta2/gapic_version.py index b3b122d9344e..558c8aab67c5 100644 --- a/packages/google-cloud-documentai/google/cloud/documentai_v1beta2/gapic_version.py +++ b/packages/google-cloud-documentai/google/cloud/documentai_v1beta2/gapic_version.py @@ -13,4 +13,4 @@ # See the License for the specific language governing permissions and # limitations under the License. # -__version__ = "2.29.3" # {x-release-please-version} +__version__ = "0.0.0" # {x-release-please-version} diff --git a/packages/google-cloud-documentai/google/cloud/documentai_v1beta3/gapic_version.py b/packages/google-cloud-documentai/google/cloud/documentai_v1beta3/gapic_version.py index b3b122d9344e..558c8aab67c5 100644 --- a/packages/google-cloud-documentai/google/cloud/documentai_v1beta3/gapic_version.py +++ b/packages/google-cloud-documentai/google/cloud/documentai_v1beta3/gapic_version.py @@ -13,4 +13,4 @@ # See the License for the specific language governing permissions and # limitations under the License. # -__version__ = "2.29.3" # {x-release-please-version} +__version__ = "0.0.0" # {x-release-please-version} diff --git a/packages/google-cloud-documentai/google/cloud/documentai_v1beta3/services/document_processor_service/async_client.py b/packages/google-cloud-documentai/google/cloud/documentai_v1beta3/services/document_processor_service/async_client.py index 99df297dc44e..55817c6f8fc8 100644 --- a/packages/google-cloud-documentai/google/cloud/documentai_v1beta3/services/document_processor_service/async_client.py +++ b/packages/google-cloud-documentai/google/cloud/documentai_v1beta3/services/document_processor_service/async_client.py @@ -1942,7 +1942,11 @@ async def create_processor( r"""Creates a processor from the [ProcessorType][google.cloud.documentai.v1beta3.ProcessorType] provided. The processor will be at ``ENABLED`` state by default - after its creation. + after its creation. Note that this method requires the + ``documentai.processors.create`` permission on the project, + which is highly privileged. A user or service account with this + permission can create new processors that can interact with any + gcs bucket in your project. .. code-block:: python diff --git a/packages/google-cloud-documentai/google/cloud/documentai_v1beta3/services/document_processor_service/client.py b/packages/google-cloud-documentai/google/cloud/documentai_v1beta3/services/document_processor_service/client.py index bbd72c6ec38f..372b0e860aa5 100644 --- a/packages/google-cloud-documentai/google/cloud/documentai_v1beta3/services/document_processor_service/client.py +++ b/packages/google-cloud-documentai/google/cloud/documentai_v1beta3/services/document_processor_service/client.py @@ -2393,7 +2393,11 @@ def create_processor( r"""Creates a processor from the [ProcessorType][google.cloud.documentai.v1beta3.ProcessorType] provided. The processor will be at ``ENABLED`` state by default - after its creation. + after its creation. Note that this method requires the + ``documentai.processors.create`` permission on the project, + which is highly privileged. A user or service account with this + permission can create new processors that can interact with any + gcs bucket in your project. .. code-block:: python diff --git a/packages/google-cloud-documentai/google/cloud/documentai_v1beta3/services/document_processor_service/transports/grpc.py b/packages/google-cloud-documentai/google/cloud/documentai_v1beta3/services/document_processor_service/transports/grpc.py index 331f4e2efbf9..04ea1fa9d195 100644 --- a/packages/google-cloud-documentai/google/cloud/documentai_v1beta3/services/document_processor_service/transports/grpc.py +++ b/packages/google-cloud-documentai/google/cloud/documentai_v1beta3/services/document_processor_service/transports/grpc.py @@ -650,7 +650,11 @@ def create_processor( Creates a processor from the [ProcessorType][google.cloud.documentai.v1beta3.ProcessorType] provided. The processor will be at ``ENABLED`` state by default - after its creation. + after its creation. Note that this method requires the + ``documentai.processors.create`` permission on the project, + which is highly privileged. A user or service account with this + permission can create new processors that can interact with any + gcs bucket in your project. Returns: Callable[[~.CreateProcessorRequest], diff --git a/packages/google-cloud-documentai/google/cloud/documentai_v1beta3/services/document_processor_service/transports/grpc_asyncio.py b/packages/google-cloud-documentai/google/cloud/documentai_v1beta3/services/document_processor_service/transports/grpc_asyncio.py index 704a433e8dfd..5055928b8c89 100644 --- a/packages/google-cloud-documentai/google/cloud/documentai_v1beta3/services/document_processor_service/transports/grpc_asyncio.py +++ b/packages/google-cloud-documentai/google/cloud/documentai_v1beta3/services/document_processor_service/transports/grpc_asyncio.py @@ -658,7 +658,11 @@ def create_processor( Creates a processor from the [ProcessorType][google.cloud.documentai.v1beta3.ProcessorType] provided. The processor will be at ``ENABLED`` state by default - after its creation. + after its creation. Note that this method requires the + ``documentai.processors.create`` permission on the project, + which is highly privileged. A user or service account with this + permission can create new processors that can interact with any + gcs bucket in your project. Returns: Callable[[~.CreateProcessorRequest], diff --git a/packages/google-cloud-documentai/google/cloud/documentai_v1beta3/services/document_service/async_client.py b/packages/google-cloud-documentai/google/cloud/documentai_v1beta3/services/document_service/async_client.py index c3ff68912450..f38e07332b70 100644 --- a/packages/google-cloud-documentai/google/cloud/documentai_v1beta3/services/document_service/async_client.py +++ b/packages/google-cloud-documentai/google/cloud/documentai_v1beta3/services/document_service/async_client.py @@ -289,7 +289,12 @@ async def update_dataset( timeout: Union[float, object] = gapic_v1.method.DEFAULT, metadata: Sequence[Tuple[str, str]] = (), ) -> operation_async.AsyncOperation: - r"""Updates metadata associated with a dataset. + r"""Updates metadata associated with a dataset. Note that this + method requires the + ``documentai.googleapis.com/datasets.update`` permission on the + project, which is highly privileged. A user or service account + with this permission can create new processors that can interact + with any gcs bucket in your project. .. code-block:: python diff --git a/packages/google-cloud-documentai/google/cloud/documentai_v1beta3/services/document_service/client.py b/packages/google-cloud-documentai/google/cloud/documentai_v1beta3/services/document_service/client.py index 785064d14e67..45ed0f99af74 100644 --- a/packages/google-cloud-documentai/google/cloud/documentai_v1beta3/services/document_service/client.py +++ b/packages/google-cloud-documentai/google/cloud/documentai_v1beta3/services/document_service/client.py @@ -747,7 +747,12 @@ def update_dataset( timeout: Union[float, object] = gapic_v1.method.DEFAULT, metadata: Sequence[Tuple[str, str]] = (), ) -> operation.Operation: - r"""Updates metadata associated with a dataset. + r"""Updates metadata associated with a dataset. Note that this + method requires the + ``documentai.googleapis.com/datasets.update`` permission on the + project, which is highly privileged. A user or service account + with this permission can create new processors that can interact + with any gcs bucket in your project. .. code-block:: python diff --git a/packages/google-cloud-documentai/google/cloud/documentai_v1beta3/services/document_service/transports/grpc.py b/packages/google-cloud-documentai/google/cloud/documentai_v1beta3/services/document_service/transports/grpc.py index ea1e84748968..5bb62c9211e7 100644 --- a/packages/google-cloud-documentai/google/cloud/documentai_v1beta3/services/document_service/transports/grpc.py +++ b/packages/google-cloud-documentai/google/cloud/documentai_v1beta3/services/document_service/transports/grpc.py @@ -258,7 +258,12 @@ def update_dataset( ) -> Callable[[document_service.UpdateDatasetRequest], operations_pb2.Operation]: r"""Return a callable for the update dataset method over gRPC. - Updates metadata associated with a dataset. + Updates metadata associated with a dataset. Note that this + method requires the + ``documentai.googleapis.com/datasets.update`` permission on the + project, which is highly privileged. A user or service account + with this permission can create new processors that can interact + with any gcs bucket in your project. Returns: Callable[[~.UpdateDatasetRequest], diff --git a/packages/google-cloud-documentai/google/cloud/documentai_v1beta3/services/document_service/transports/grpc_asyncio.py b/packages/google-cloud-documentai/google/cloud/documentai_v1beta3/services/document_service/transports/grpc_asyncio.py index 259f7f1b2259..d5f6af7cd85e 100644 --- a/packages/google-cloud-documentai/google/cloud/documentai_v1beta3/services/document_service/transports/grpc_asyncio.py +++ b/packages/google-cloud-documentai/google/cloud/documentai_v1beta3/services/document_service/transports/grpc_asyncio.py @@ -266,7 +266,12 @@ def update_dataset( ]: r"""Return a callable for the update dataset method over gRPC. - Updates metadata associated with a dataset. + Updates metadata associated with a dataset. Note that this + method requires the + ``documentai.googleapis.com/datasets.update`` permission on the + project, which is highly privileged. A user or service account + with this permission can create new processors that can interact + with any gcs bucket in your project. Returns: Callable[[~.UpdateDatasetRequest], diff --git a/packages/google-cloud-documentai/google/cloud/documentai_v1beta3/types/dataset.py b/packages/google-cloud-documentai/google/cloud/documentai_v1beta3/types/dataset.py index 1bed44a52489..2c63a0f3915e 100644 --- a/packages/google-cloud-documentai/google/cloud/documentai_v1beta3/types/dataset.py +++ b/packages/google-cloud-documentai/google/cloud/documentai_v1beta3/types/dataset.py @@ -78,6 +78,10 @@ class Dataset(proto.Message): state (google.cloud.documentai_v1beta3.types.Dataset.State): Required. State of the dataset. Ignored when updating dataset. + satisfies_pzs (bool): + Output only. Reserved for future use. + satisfies_pzi (bool): + Output only. Reserved for future use. """ class State(proto.Enum): @@ -176,6 +180,14 @@ class SpannerIndexingConfig(proto.Message): number=2, enum=State, ) + satisfies_pzs: bool = proto.Field( + proto.BOOL, + number=8, + ) + satisfies_pzi: bool = proto.Field( + proto.BOOL, + number=9, + ) class DocumentId(proto.Message): diff --git a/packages/google-cloud-documentai/google/cloud/documentai_v1beta3/types/document.py b/packages/google-cloud-documentai/google/cloud/documentai_v1beta3/types/document.py index 3941b500ec64..e1bad6707a2b 100644 --- a/packages/google-cloud-documentai/google/cloud/documentai_v1beta3/types/document.py +++ b/packages/google-cloud-documentai/google/cloud/documentai_v1beta3/types/document.py @@ -2079,9 +2079,7 @@ class Chunk(proto.Message): chunk_id (str): ID of the chunk. source_block_ids (MutableSequence[str]): - DO NOT USE. - List of all parsed documents layout source - blocks used to generate the chunk. + Unused. content (str): Text content of the chunk. page_span (google.cloud.documentai_v1beta3.types.Document.ChunkedDocument.Chunk.ChunkPageSpan): diff --git a/packages/google-cloud-documentai/google/cloud/documentai_v1beta3/types/document_processor_service.py b/packages/google-cloud-documentai/google/cloud/documentai_v1beta3/types/document_processor_service.py index d46fe1cca39e..6739bf59742c 100644 --- a/packages/google-cloud-documentai/google/cloud/documentai_v1beta3/types/document_processor_service.py +++ b/packages/google-cloud-documentai/google/cloud/documentai_v1beta3/types/document_processor_service.py @@ -293,6 +293,9 @@ class ProcessRequest(proto.Message): underscores, and dashes. International characters are allowed. Label values are optional. Label keys must start with a letter. + imageless_mode (bool): + Optional. Option to remove images from the + document. """ inline_document: gcd_document.Document = proto.Field( @@ -341,6 +344,10 @@ class ProcessRequest(proto.Message): proto.STRING, number=10, ) + imageless_mode: bool = proto.Field( + proto.BOOL, + number=11, + ) class HumanReviewStatus(proto.Message): diff --git a/packages/google-cloud-documentai/google/cloud/documentai_v1beta3/types/document_service.py b/packages/google-cloud-documentai/google/cloud/documentai_v1beta3/types/document_service.py index 86af095d60ab..5fd1139ddcfb 100644 --- a/packages/google-cloud-documentai/google/cloud/documentai_v1beta3/types/document_service.py +++ b/packages/google-cloud-documentai/google/cloud/documentai_v1beta3/types/document_service.py @@ -396,8 +396,10 @@ class ListDocumentsRequest(proto.Message): https://google.aip.dev/160. Currently support query strings are: + ------------------------------------ + + ``SplitType=DATASET_SPLIT_TEST|DATASET_SPLIT_TRAIN|DATASET_SPLIT_UNASSIGNED`` - - ``SplitType=DATASET_SPLIT_TEST|DATASET_SPLIT_TRAIN|DATASET_SPLIT_UNASSIGNED`` - ``LabelingState=DOCUMENT_LABELED|DOCUMENT_UNLABELED|DOCUMENT_AUTO_LABELED`` - ``DisplayName=\"file_name.pdf\"`` - ``EntityType=abc/def`` diff --git a/packages/google-cloud-documentai/google/cloud/documentai_v1beta3/types/processor.py b/packages/google-cloud-documentai/google/cloud/documentai_v1beta3/types/processor.py index 6801da0b543e..a4c136956e6d 100644 --- a/packages/google-cloud-documentai/google/cloud/documentai_v1beta3/types/processor.py +++ b/packages/google-cloud-documentai/google/cloud/documentai_v1beta3/types/processor.py @@ -72,6 +72,13 @@ class ProcessorVersion(proto.Message): model_type (google.cloud.documentai_v1beta3.types.ProcessorVersion.ModelType): Output only. The model type of this processor version. + satisfies_pzs (bool): + Output only. Reserved for future use. + satisfies_pzi (bool): + Output only. Reserved for future use. + gen_ai_model_info (google.cloud.documentai_v1beta3.types.ProcessorVersion.GenAiModelInfo): + Output only. Information about Generative AI + model-based processor versions. """ class State(proto.Enum): @@ -151,6 +158,102 @@ class DeprecationInfo(proto.Message): number=2, ) + class GenAiModelInfo(proto.Message): + r"""Information about Generative AI model-based processor + versions. + + This message has `oneof`_ fields (mutually exclusive fields). + For each oneof, at most one member field can be set at the same time. + Setting any member of the oneof automatically clears all other + members. + + .. _oneof: https://proto-plus-python.readthedocs.io/en/stable/fields.html#oneofs-mutually-exclusive-fields + + Attributes: + foundation_gen_ai_model_info (google.cloud.documentai_v1beta3.types.ProcessorVersion.GenAiModelInfo.FoundationGenAiModelInfo): + Information for a pretrained Google-managed + foundation model. + + This field is a member of `oneof`_ ``model_info``. + custom_gen_ai_model_info (google.cloud.documentai_v1beta3.types.ProcessorVersion.GenAiModelInfo.CustomGenAiModelInfo): + Information for a custom Generative AI model + created by the user. + + This field is a member of `oneof`_ ``model_info``. + """ + + class FoundationGenAiModelInfo(proto.Message): + r"""Information for a pretrained Google-managed foundation model. + + Attributes: + finetuning_allowed (bool): + Whether finetuning is allowed for this base + processor version. + min_train_labeled_documents (int): + The minimum number of labeled documents in + the training dataset required for finetuning. + """ + + finetuning_allowed: bool = proto.Field( + proto.BOOL, + number=1, + ) + min_train_labeled_documents: int = proto.Field( + proto.INT32, + number=2, + ) + + class CustomGenAiModelInfo(proto.Message): + r"""Information for a custom Generative AI model created by the user. + These are created with ``Create New Version`` in either the + ``Call foundation model`` or ``Fine tuning`` tabs. + + Attributes: + custom_model_type (google.cloud.documentai_v1beta3.types.ProcessorVersion.GenAiModelInfo.CustomGenAiModelInfo.CustomModelType): + The type of custom model created by the user. + base_processor_version_id (str): + The base processor version ID for the custom + model. + """ + + class CustomModelType(proto.Enum): + r"""The type of custom model created by the user. + + Values: + CUSTOM_MODEL_TYPE_UNSPECIFIED (0): + The model type is unspecified. + VERSIONED_FOUNDATION (1): + The model is a versioned foundation model. + FINE_TUNED (2): + The model is a finetuned foundation model. + """ + CUSTOM_MODEL_TYPE_UNSPECIFIED = 0 + VERSIONED_FOUNDATION = 1 + FINE_TUNED = 2 + + custom_model_type: "ProcessorVersion.GenAiModelInfo.CustomGenAiModelInfo.CustomModelType" = proto.Field( + proto.ENUM, + number=1, + enum="ProcessorVersion.GenAiModelInfo.CustomGenAiModelInfo.CustomModelType", + ) + base_processor_version_id: str = proto.Field( + proto.STRING, + number=2, + ) + + foundation_gen_ai_model_info: "ProcessorVersion.GenAiModelInfo.FoundationGenAiModelInfo" = proto.Field( + proto.MESSAGE, + number=1, + oneof="model_info", + message="ProcessorVersion.GenAiModelInfo.FoundationGenAiModelInfo", + ) + custom_gen_ai_model_info: "ProcessorVersion.GenAiModelInfo.CustomGenAiModelInfo" = proto.Field( + proto.MESSAGE, + number=2, + oneof="model_info", + message="ProcessorVersion.GenAiModelInfo.CustomGenAiModelInfo", + ) + name: str = proto.Field( proto.STRING, number=1, @@ -201,6 +304,19 @@ class DeprecationInfo(proto.Message): number=15, enum=ModelType, ) + satisfies_pzs: bool = proto.Field( + proto.BOOL, + number=16, + ) + satisfies_pzi: bool = proto.Field( + proto.BOOL, + number=17, + ) + gen_ai_model_info: GenAiModelInfo = proto.Field( + proto.MESSAGE, + number=18, + message=GenAiModelInfo, + ) class ProcessorVersionAlias(proto.Message): @@ -256,6 +372,10 @@ class Processor(proto.Message): The `KMS key `__ used for encryption and decryption in CMEK scenarios. + satisfies_pzs (bool): + Output only. Reserved for future use. + satisfies_pzi (bool): + Output only. Reserved for future use. """ class State(proto.Enum): @@ -344,6 +464,14 @@ class State(proto.Enum): proto.STRING, number=8, ) + satisfies_pzs: bool = proto.Field( + proto.BOOL, + number=12, + ) + satisfies_pzi: bool = proto.Field( + proto.BOOL, + number=13, + ) __all__ = tuple(sorted(__protobuf__.manifest)) diff --git a/packages/google-cloud-documentai/samples/generated_samples/snippet_metadata_google.cloud.documentai.v1.json b/packages/google-cloud-documentai/samples/generated_samples/snippet_metadata_google.cloud.documentai.v1.json index 1d4924cfdca5..96d60af285a2 100644 --- a/packages/google-cloud-documentai/samples/generated_samples/snippet_metadata_google.cloud.documentai.v1.json +++ b/packages/google-cloud-documentai/samples/generated_samples/snippet_metadata_google.cloud.documentai.v1.json @@ -8,7 +8,7 @@ ], "language": "PYTHON", "name": "google-cloud-documentai", - "version": "2.29.3" + "version": "0.1.0" }, "snippets": [ { diff --git a/packages/google-cloud-documentai/samples/generated_samples/snippet_metadata_google.cloud.documentai.v1beta2.json b/packages/google-cloud-documentai/samples/generated_samples/snippet_metadata_google.cloud.documentai.v1beta2.json index 6c6dba55881a..ef56bd7e3eb3 100644 --- a/packages/google-cloud-documentai/samples/generated_samples/snippet_metadata_google.cloud.documentai.v1beta2.json +++ b/packages/google-cloud-documentai/samples/generated_samples/snippet_metadata_google.cloud.documentai.v1beta2.json @@ -8,7 +8,7 @@ ], "language": "PYTHON", "name": "google-cloud-documentai", - "version": "2.29.3" + "version": "0.1.0" }, "snippets": [ { diff --git a/packages/google-cloud-documentai/samples/generated_samples/snippet_metadata_google.cloud.documentai.v1beta3.json b/packages/google-cloud-documentai/samples/generated_samples/snippet_metadata_google.cloud.documentai.v1beta3.json index 3be9d4d7d653..f47545a8ed3e 100644 --- a/packages/google-cloud-documentai/samples/generated_samples/snippet_metadata_google.cloud.documentai.v1beta3.json +++ b/packages/google-cloud-documentai/samples/generated_samples/snippet_metadata_google.cloud.documentai.v1beta3.json @@ -8,7 +8,7 @@ ], "language": "PYTHON", "name": "google-cloud-documentai", - "version": "2.29.3" + "version": "0.1.0" }, "snippets": [ { diff --git a/packages/google-cloud-documentai/scripts/fixup_documentai_v1beta3_keywords.py b/packages/google-cloud-documentai/scripts/fixup_documentai_v1beta3_keywords.py index f0b457f6b1c0..15d34f8ecda8 100644 --- a/packages/google-cloud-documentai/scripts/fixup_documentai_v1beta3_keywords.py +++ b/packages/google-cloud-documentai/scripts/fixup_documentai_v1beta3_keywords.py @@ -62,7 +62,7 @@ class documentaiCallTransformer(cst.CSTTransformer): 'list_processors': ('parent', 'page_size', 'page_token', ), 'list_processor_types': ('parent', 'page_size', 'page_token', ), 'list_processor_versions': ('parent', 'page_size', 'page_token', ), - 'process_document': ('name', 'inline_document', 'raw_document', 'gcs_document', 'document', 'skip_human_review', 'field_mask', 'process_options', 'labels', ), + 'process_document': ('name', 'inline_document', 'raw_document', 'gcs_document', 'document', 'skip_human_review', 'field_mask', 'process_options', 'labels', 'imageless_mode', ), 'review_document': ('human_review_config', 'inline_document', 'document', 'enable_schema_validation', 'priority', 'document_schema', ), 'set_default_processor_version': ('processor', 'default_processor_version', ), 'train_processor_version': ('parent', 'processor_version', 'custom_document_extraction_options', 'foundation_model_tuning_options', 'document_schema', 'input_data', 'base_processor_version', ), diff --git a/packages/google-cloud-documentai/tests/unit/gapic/documentai_v1beta3/test_document_processor_service.py b/packages/google-cloud-documentai/tests/unit/gapic/documentai_v1beta3/test_document_processor_service.py index be58c586d2d1..19e12fae57c4 100644 --- a/packages/google-cloud-documentai/tests/unit/gapic/documentai_v1beta3/test_document_processor_service.py +++ b/packages/google-cloud-documentai/tests/unit/gapic/documentai_v1beta3/test_document_processor_service.py @@ -3957,6 +3957,8 @@ def test_get_processor(request_type, transport: str = "grpc"): default_processor_version="default_processor_version_value", process_endpoint="process_endpoint_value", kms_key_name="kms_key_name_value", + satisfies_pzs=True, + satisfies_pzi=True, ) response = client.get_processor(request) @@ -3975,6 +3977,8 @@ def test_get_processor(request_type, transport: str = "grpc"): assert response.default_processor_version == "default_processor_version_value" assert response.process_endpoint == "process_endpoint_value" assert response.kms_key_name == "kms_key_name_value" + assert response.satisfies_pzs is True + assert response.satisfies_pzi is True def test_get_processor_empty_call(): @@ -4080,6 +4084,8 @@ async def test_get_processor_empty_call_async(): default_processor_version="default_processor_version_value", process_endpoint="process_endpoint_value", kms_key_name="kms_key_name_value", + satisfies_pzs=True, + satisfies_pzi=True, ) ) response = await client.get_processor() @@ -4155,6 +4161,8 @@ async def test_get_processor_async( default_processor_version="default_processor_version_value", process_endpoint="process_endpoint_value", kms_key_name="kms_key_name_value", + satisfies_pzs=True, + satisfies_pzi=True, ) ) response = await client.get_processor(request) @@ -4174,6 +4182,8 @@ async def test_get_processor_async( assert response.default_processor_version == "default_processor_version_value" assert response.process_endpoint == "process_endpoint_value" assert response.kms_key_name == "kms_key_name_value" + assert response.satisfies_pzs is True + assert response.satisfies_pzi is True @pytest.mark.asyncio @@ -4750,6 +4760,8 @@ def test_get_processor_version(request_type, transport: str = "grpc"): kms_key_version_name="kms_key_version_name_value", google_managed=True, model_type=processor.ProcessorVersion.ModelType.MODEL_TYPE_GENERATIVE, + satisfies_pzs=True, + satisfies_pzi=True, ) response = client.get_processor_version(request) @@ -4771,6 +4783,8 @@ def test_get_processor_version(request_type, transport: str = "grpc"): response.model_type == processor.ProcessorVersion.ModelType.MODEL_TYPE_GENERATIVE ) + assert response.satisfies_pzs is True + assert response.satisfies_pzi is True def test_get_processor_version_empty_call(): @@ -4887,6 +4901,8 @@ async def test_get_processor_version_empty_call_async(): kms_key_version_name="kms_key_version_name_value", google_managed=True, model_type=processor.ProcessorVersion.ModelType.MODEL_TYPE_GENERATIVE, + satisfies_pzs=True, + satisfies_pzi=True, ) ) response = await client.get_processor_version() @@ -4964,6 +4980,8 @@ async def test_get_processor_version_async( kms_key_version_name="kms_key_version_name_value", google_managed=True, model_type=processor.ProcessorVersion.ModelType.MODEL_TYPE_GENERATIVE, + satisfies_pzs=True, + satisfies_pzi=True, ) ) response = await client.get_processor_version(request) @@ -4986,6 +5004,8 @@ async def test_get_processor_version_async( response.model_type == processor.ProcessorVersion.ModelType.MODEL_TYPE_GENERATIVE ) + assert response.satisfies_pzs is True + assert response.satisfies_pzi is True @pytest.mark.asyncio @@ -6928,6 +6948,8 @@ def test_create_processor(request_type, transport: str = "grpc"): default_processor_version="default_processor_version_value", process_endpoint="process_endpoint_value", kms_key_name="kms_key_name_value", + satisfies_pzs=True, + satisfies_pzi=True, ) response = client.create_processor(request) @@ -6946,6 +6968,8 @@ def test_create_processor(request_type, transport: str = "grpc"): assert response.default_processor_version == "default_processor_version_value" assert response.process_endpoint == "process_endpoint_value" assert response.kms_key_name == "kms_key_name_value" + assert response.satisfies_pzs is True + assert response.satisfies_pzi is True def test_create_processor_empty_call(): @@ -7053,6 +7077,8 @@ async def test_create_processor_empty_call_async(): default_processor_version="default_processor_version_value", process_endpoint="process_endpoint_value", kms_key_name="kms_key_name_value", + satisfies_pzs=True, + satisfies_pzi=True, ) ) response = await client.create_processor() @@ -7128,6 +7154,8 @@ async def test_create_processor_async( default_processor_version="default_processor_version_value", process_endpoint="process_endpoint_value", kms_key_name="kms_key_name_value", + satisfies_pzs=True, + satisfies_pzi=True, ) ) response = await client.create_processor(request) @@ -7147,6 +7175,8 @@ async def test_create_processor_async( assert response.default_processor_version == "default_processor_version_value" assert response.process_endpoint == "process_endpoint_value" assert response.kms_key_name == "kms_key_name_value" + assert response.satisfies_pzs is True + assert response.satisfies_pzi is True @pytest.mark.asyncio @@ -12696,6 +12726,8 @@ def test_get_processor_rest(request_type): default_processor_version="default_processor_version_value", process_endpoint="process_endpoint_value", kms_key_name="kms_key_name_value", + satisfies_pzs=True, + satisfies_pzi=True, ) # Wrap the value into a proper Response obj @@ -12718,6 +12750,8 @@ def test_get_processor_rest(request_type): assert response.default_processor_version == "default_processor_version_value" assert response.process_endpoint == "process_endpoint_value" assert response.kms_key_name == "kms_key_name_value" + assert response.satisfies_pzs is True + assert response.satisfies_pzi is True def test_get_processor_rest_use_cached_wrapped_rpc(): @@ -13333,6 +13367,8 @@ def test_get_processor_version_rest(request_type): kms_key_version_name="kms_key_version_name_value", google_managed=True, model_type=processor.ProcessorVersion.ModelType.MODEL_TYPE_GENERATIVE, + satisfies_pzs=True, + satisfies_pzi=True, ) # Wrap the value into a proper Response obj @@ -13358,6 +13394,8 @@ def test_get_processor_version_rest(request_type): response.model_type == processor.ProcessorVersion.ModelType.MODEL_TYPE_GENERATIVE ) + assert response.satisfies_pzs is True + assert response.satisfies_pzi is True def test_get_processor_version_rest_use_cached_wrapped_rpc(): @@ -14997,6 +15035,8 @@ def test_create_processor_rest(request_type): "process_endpoint": "process_endpoint_value", "create_time": {"seconds": 751, "nanos": 543}, "kms_key_name": "kms_key_name_value", + "satisfies_pzs": True, + "satisfies_pzi": True, } # The version of a generated dependency at test runtime may differ from the version used during generation. # Delete any fields which are not present in the current runtime dependency @@ -15080,6 +15120,8 @@ def get_message_fields(field): default_processor_version="default_processor_version_value", process_endpoint="process_endpoint_value", kms_key_name="kms_key_name_value", + satisfies_pzs=True, + satisfies_pzi=True, ) # Wrap the value into a proper Response obj @@ -15102,6 +15144,8 @@ def get_message_fields(field): assert response.default_processor_version == "default_processor_version_value" assert response.process_endpoint == "process_endpoint_value" assert response.kms_key_name == "kms_key_name_value" + assert response.satisfies_pzs is True + assert response.satisfies_pzi is True def test_create_processor_rest_use_cached_wrapped_rpc(): diff --git a/packages/google-cloud-documentai/tests/unit/gapic/documentai_v1beta3/test_document_service.py b/packages/google-cloud-documentai/tests/unit/gapic/documentai_v1beta3/test_document_service.py index 3a4be228dd30..69a71a4df7d9 100644 --- a/packages/google-cloud-documentai/tests/unit/gapic/documentai_v1beta3/test_document_service.py +++ b/packages/google-cloud-documentai/tests/unit/gapic/documentai_v1beta3/test_document_service.py @@ -4052,6 +4052,8 @@ def test_update_dataset_rest(request_type): "spanner_indexing_config": {}, "name": "projects/sample1/locations/sample2/processors/sample3/dataset", "state": 1, + "satisfies_pzs": True, + "satisfies_pzi": True, } # The version of a generated dependency at test runtime may differ from the version used during generation. # Delete any fields which are not present in the current runtime dependency