chore: added increased timeout on flaky batch request (#84)

Author: munkhuushmgl

document_ai/snippets/batch_parse_form_v1beta2.py

@@ -24,6 +24,7 @@ def batch_parse_form(
     project_id="YOUR_PROJECT_ID",
     input_uri="gs://cloud-samples-data/documentai/form.pdf",
     destination_uri="gs://your-bucket-id/path/to/save/results/",
+    timeout=90
 ):
     """Parse a form"""
 
@@ -80,7 +81,7 @@ def batch_parse_form(
     operation = client.batch_process_documents(batch_request)
 
     # Wait for the operation to finish
-    operation.result()
+    operation.result(timeout)
 
     # Results are written to GCS. Use a regex to find
     # output files

document_ai/snippets/batch_parse_form_v1beta2_test.py

@@ -41,6 +41,6 @@ def setup_teardown():
 
 
 def test_batch_parse_form(capsys):
-    batch_parse_form_v1beta2.batch_parse_form(PROJECT_ID, INPUT_URI, BATCH_OUTPUT_URI)
+    batch_parse_form_v1beta2.batch_parse_form(PROJECT_ID, INPUT_URI, BATCH_OUTPUT_URI, 120)
     out, _ = capsys.readouterr()
     assert "Output files" in out

document_ai/snippets/batch_parse_table_v1beta2.py

@@ -24,6 +24,7 @@ def batch_parse_table(
     project_id="YOUR_PROJECT_ID",
     input_uri="gs://cloud-samples-data/documentai/form.pdf",
     destination_uri="gs://your-bucket-id/path/to/save/results/",
+    timeout=90
 ):
     """Parse a form"""
 
@@ -88,7 +89,7 @@ def batch_parse_table(
     operation = client.batch_process_documents(batch_request)
 
     # Wait for the operation to finish
-    operation.result()
+    operation.result(timeout)
 
     # Results are written to GCS. Use a regex to find
     # output files

document_ai/snippets/batch_parse_table_v1beta2_test.py

@@ -41,6 +41,6 @@ def setup_teardown():
 
 
 def test_batch_parse_table(capsys):
-    batch_parse_table_v1beta2.batch_parse_table(PROJECT_ID, INPUT_URI, BATCH_OUTPUT_URI)
+    batch_parse_table_v1beta2.batch_parse_table(PROJECT_ID, INPUT_URI, BATCH_OUTPUT_URI, 120)
     out, _ = capsys.readouterr()
     assert "Output files:" in out