fix: adds comment with explicit hostname change (#94)

* fix: adds switching code for client_options based upon location
GoogleCloudPlatform · Mar 11, 2021 · 5b15d2d · 5b15d2d
1 parent a8959c7
commit 5b15d2d
Show file tree

Hide file tree

Showing 3 changed files with 19 additions and 4 deletions.
diff --git a/document_ai/snippets/batch_process_documents_sample_v1beta3.py b/document_ai/snippets/batch_process_documents_sample_v1beta3.py
@@ -38,7 +38,12 @@ def batch_process_documents(
     timeout: int = 300,
 ):
 
-    client = documentai.DocumentProcessorServiceClient()
+    # You must set the api_endpoint if you use a location other than 'us', e.g.:
+    opts = {}
+    if location == "eu":
+        opts = {"api_endpoint": "eu-documentai.googleapis.com"}
+
+    client = documentai.DocumentProcessorServiceClient(client_options=opts)
 
     destination_uri = f"{gcs_output_uri}/{gcs_output_uri_prefix}/"
 

diff --git a/document_ai/snippets/process_document_sample_v1beta3.py b/document_ai/snippets/process_document_sample_v1beta3.py
@@ -27,8 +27,12 @@ def process_document_sample(
 ):
     from google.cloud import documentai_v1beta3 as documentai
 
-    # Instantiates a client
-    client = documentai.DocumentProcessorServiceClient()
+    # You must set the api_endpoint if you use a location other than 'us', e.g.:
+    opts = {}
+    if location == "eu":
+        opts = {"api_endpoint": "eu-documentai.googleapis.com"}
+
+    client = documentai.DocumentProcessorServiceClient(client_options=opts)
 
     # The full resource name of the processor, e.g.:
     # projects/project-id/locations/location/processor/processor-id

diff --git a/document_ai/snippets/quickstart_sample_v1beta3.py b/document_ai/snippets/quickstart_sample_v1beta3.py
@@ -25,7 +25,13 @@
 
 
 def quickstart(project_id: str, location: str, processor_id: str, file_path: str):
-    client = documentai.DocumentProcessorServiceClient()
+
+    # You must set the api_endpoint if you use a location other than 'us', e.g.:
+    opts = {}
+    if location == "eu":
+        opts = {"api_endpoint": "eu-documentai.googleapis.com"}
+
+    client = documentai.DocumentProcessorServiceClient(client_options=opts)
 
     # The full resource name of the processor, e.g.:
     # projects/project-id/locations/location/processor/processor-id