docs(samples): Document AI - Update Train/Evaluate Samples to use v1 (2.13.0) (#9163)

Author: holtskinner

documentai/snippets/evaluate_processor_version_sample.py

@@ -16,7 +16,7 @@
 # [START documentai_evaluate_processor_version]
 
 from google.api_core.client_options import ClientOptions
-from google.cloud import documentai_v1beta3 as documentai
+from google.cloud import documentai
 
 # TODO(developer): Uncomment these variables before running the sample.
 # project_id = 'YOUR_PROJECT_ID'

documentai/snippets/evaluate_processor_version_sample_test.py

@@ -28,9 +28,9 @@
 
 # Mocking request as evaluation can take a long time
 @mock.patch(
-    "google.cloud.documentai_v1beta3.DocumentProcessorServiceClient.evaluate_processor_version"
+    "google.cloud.documentai.DocumentProcessorServiceClient.evaluate_processor_version"
 )
-@mock.patch("google.cloud.documentai_v1beta3.EvaluateProcessorVersionResponse")
+@mock.patch("google.cloud.documentai.EvaluateProcessorVersionResponse")
 @mock.patch("google.api_core.operation.Operation")
 def test_evaluate_processor_version(
     operation_mock,

documentai/snippets/get_evaluation_sample.py

@@ -16,7 +16,7 @@
 # [START documentai_get_evaluation]
 
 from google.api_core.client_options import ClientOptions
-from google.cloud import documentai_v1beta3 as documentai
+from google.cloud import documentai
 
 # TODO(developer): Uncomment these variables before running the sample.
 # project_id = 'YOUR_PROJECT_ID'

documentai/snippets/get_evaluation_sample_test.py

@@ -27,12 +27,9 @@
 
 
 # Mocking request
-@mock.patch(
-    "google.cloud.documentai_v1beta3.DocumentProcessorServiceClient.get_evaluation"
-)
-@mock.patch("google.cloud.documentai_v1beta3.Evaluation")
+@mock.patch("google.cloud.documentai.DocumentProcessorServiceClient.get_evaluation")
+@mock.patch("google.cloud.documentai.Evaluation")
 def test_get_evaluation(evaluation_mock, get_evaluation_mock, capsys):
-
     get_evaluation_mock.return_value = evaluation_mock
 
     get_evaluation_sample.get_evaluation_sample(

documentai/snippets/list_evaluations_sample.py

@@ -16,7 +16,7 @@
 # [START documentai_list_evaluations]
 
 from google.api_core.client_options import ClientOptions
-from google.cloud import documentai_v1beta3 as documentai
+from google.cloud import documentai
 
 # TODO(developer): Uncomment these variables before running the sample.
 # project_id = 'YOUR_PROJECT_ID'

documentai/snippets/requirements.txt

@@ -1,2 +1,2 @@
-google-cloud-documentai==2.10.0
+google-cloud-documentai==2.13.0
 google-cloud-storage==2.7.0

documentai/snippets/train_processor_version_sample.py

@@ -16,7 +16,7 @@
 # [START documentai_train_processor_version]
 
 from google.api_core.client_options import ClientOptions
-from google.cloud import documentai_v1beta3 as documentai
+from google.cloud import documentai
 
 # TODO(developer): Uncomment these variables before running the sample.
 # project_id = 'YOUR_PROJECT_ID'

documentai/snippets/train_processor_version_sample_test.py

@@ -29,10 +29,10 @@
 
 # Mocking request as training can take a long time
 @mock.patch(
-    "google.cloud.documentai_v1beta3.DocumentProcessorServiceClient.train_processor_version"
+    "google.cloud.documentai.DocumentProcessorServiceClient.train_processor_version"
 )
-@mock.patch("google.cloud.documentai_v1beta3.TrainProcessorVersionResponse")
-@mock.patch("google.cloud.documentai_v1beta3.TrainProcessorVersionMetadata")
+@mock.patch("google.cloud.documentai.TrainProcessorVersionResponse")
+@mock.patch("google.cloud.documentai.TrainProcessorVersionMetadata")
 @mock.patch("google.api_core.operation.Operation")
 def test_train_processor_version(
     operation_mock,