feat: [google-cloud-documentai] Updated the exponential backoff settings for the Document AI ProcessDocument and BatchProcessDocuments methods (#13190)

gcf-owl-bot[bot] · web-flow · commit 91e9f9f262ac · 2024-10-22T10:10:21.000-04:00
- [ ] Regenerate this pull request now. BEGIN_COMMIT_OVERRIDE feat: [google-cloud-documentai] Updated the exponential backoff settings for the Document AI ProcessDocument and BatchProcessDocuments methods feat: Add RESOURCE_EXHAUSTED to retryable status codes for ProcessDocument method END_COMMIT_OVERRIDE PiperOrigin-RevId: 688214664 Source-Link: googleapis/googleapis@76ca663 Source-Link: googleapis/googleapis-gen@6ad9727 Copy-Tag: eyJwIjoicGFja2FnZXMvZ29vZ2xlLWNsb3VkLWRvY3VtZW50YWkvLk93bEJvdC55YW1sIiwiaCI6IjZhZDk3Mjc3NDMwNzQ0ZGQ5YzczZTdmZjM5N2RjZTAwNTQ3ZDhlMGMifQ== --------- Co-authored-by: Owl Bot <gcf-owl-bot[bot]@users.noreply.github.com>
diff --git a/packages/google-cloud-documentai/google/cloud/documentai_v1/services/document_processor_service/transports/base.py b/packages/google-cloud-documentai/google/cloud/documentai_v1/services/document_processor_service/transports/base.py
@@ -136,11 +136,12 @@ def _prep_wrapped_messages(self, client_info):
             self.process_document: gapic_v1.method.wrap_method(
                 self.process_document,
                 default_retry=retries.Retry(
-                    initial=0.1,
-                    maximum=60.0,
-                    multiplier=1.3,
+                    initial=1.0,
+                    maximum=90.0,
+                    multiplier=9.0,
                     predicate=retries.if_exception_type(
                         core_exceptions.DeadlineExceeded,
+                        core_exceptions.ResourceExhausted,
                         core_exceptions.ServiceUnavailable,
                     ),
                     deadline=300.0,
@@ -151,9 +152,9 @@ def _prep_wrapped_messages(self, client_info):
             self.batch_process_documents: gapic_v1.method.wrap_method(
                 self.batch_process_documents,
                 default_retry=retries.Retry(
-                    initial=0.1,
+                    initial=1.0,
                     maximum=60.0,
-                    multiplier=1.3,
+                    multiplier=1.5,
                     predicate=retries.if_exception_type(
                         core_exceptions.DeadlineExceeded,
                         core_exceptions.ServiceUnavailable,
@@ -246,9 +247,9 @@ def _prep_wrapped_messages(self, client_info):
             self.review_document: gapic_v1.method.wrap_method(
                 self.review_document,
                 default_retry=retries.Retry(
-                    initial=0.1,
+                    initial=1.0,
                     maximum=60.0,
-                    multiplier=1.3,
+                    multiplier=1.5,
                     predicate=retries.if_exception_type(
                         core_exceptions.DeadlineExceeded,
                         core_exceptions.ServiceUnavailable,
diff --git a/packages/google-cloud-documentai/google/cloud/documentai_v1/services/document_processor_service/transports/grpc_asyncio.py b/packages/google-cloud-documentai/google/cloud/documentai_v1/services/document_processor_service/transports/grpc_asyncio.py
@@ -936,11 +936,12 @@ def _prep_wrapped_messages(self, client_info):
             self.process_document: self._wrap_method(
                 self.process_document,
                 default_retry=retries.AsyncRetry(
-                    initial=0.1,
-                    maximum=60.0,
-                    multiplier=1.3,
+                    initial=1.0,
+                    maximum=90.0,
+                    multiplier=9.0,
                     predicate=retries.if_exception_type(
                         core_exceptions.DeadlineExceeded,
+                        core_exceptions.ResourceExhausted,
                         core_exceptions.ServiceUnavailable,
                     ),
                     deadline=300.0,
@@ -951,9 +952,9 @@ def _prep_wrapped_messages(self, client_info):
             self.batch_process_documents: self._wrap_method(
                 self.batch_process_documents,
                 default_retry=retries.AsyncRetry(
-                    initial=0.1,
+                    initial=1.0,
                     maximum=60.0,
-                    multiplier=1.3,
+                    multiplier=1.5,
                     predicate=retries.if_exception_type(
                         core_exceptions.DeadlineExceeded,
                         core_exceptions.ServiceUnavailable,
@@ -1046,9 +1047,9 @@ def _prep_wrapped_messages(self, client_info):
             self.review_document: self._wrap_method(
                 self.review_document,
                 default_retry=retries.AsyncRetry(
-                    initial=0.1,
+                    initial=1.0,
                     maximum=60.0,
-                    multiplier=1.3,
+                    multiplier=1.5,
                     predicate=retries.if_exception_type(
                         core_exceptions.DeadlineExceeded,
                         core_exceptions.ServiceUnavailable,
