fix: Add endpoint override for Document AI Warehouse samples (#3585)

* fix: Add endpoint override for Document AI Warehouse samples

- Reported in:
  - GoogleCloudPlatform/document-ai-samples#676
  - https://www.googlecloudcommunity.com/gc/AI-ML/Document-AI-Warehouse-API-Sample-Python-Code-location-eu-Issue/td-p/677608
  - https://stackoverflow.com/a/77648118/6216983

## Description

Fixes GoogleCloudPlatform/document-ai-samples#676

* Try changing endpoint to default for us

* Change default endpoint to us

* Fix formatting error

---------

Co-authored-by: Patti Shin <[email protected]>

Author: holtskinner

document-warehouse/create-document-schema.js

@@ -31,31 +31,38 @@ async function main(
   const {DocumentSchemaServiceClient} =
     require('@google-cloud/contentwarehouse').v1;
 
+  const apiEndpoint =
+    location === 'us'
+      ? 'contentwarehouse.googleapis.com'
+      : `${location}-contentwarehouse.googleapis.com`;
+
   // Create service client
-  const serviceClient = new DocumentSchemaServiceClient();
+  const serviceClient = new DocumentSchemaServiceClient({
+    apiEndpoint: apiEndpoint,
+  });
 
   // Create Document Schema
   async function createDocumentSchema() {
-    // Initialize request argument(s)
-    const request = {};
-
-    // Property Definition
-    const propertyDefinition = {};
-    propertyDefinition.name = 'testPropertyDefinitionName'; // Must be unique within a document schema (case insensitive)
-    propertyDefinition.displayName = 'searchable text';
-    propertyDefinition.isSearchable = true;
-    propertyDefinition.textTypeOptions = {};
-
-    // Document Schema
-    const documentSchema = {};
-    documentSchema.displayName = 'My Test Schema';
-    documentSchema.propertyDefinitions = [propertyDefinition];
-
-    request.documentSchema = documentSchema;
-
     // The full resource name of the location, e.g.:
     // projects/{project_number}/locations/{location}
-    request.parent = `projects/${projectNumber}/locations/${location}`;
+    const parent = `projects/${projectNumber}/locations/${location}`;
+    // Initialize request argument(s)
+    const request = {
+      parent: parent,
+      // Document Schema
+      documentSchema: {
+        displayName: 'My Test Schema',
+        // Property Definition
+        propertyDefinitions: [
+          {
+            name: 'testPropertyDefinitionName', // Must be unique within a document schema (case insensitive)
+            displayName: 'searchable text',
+            isSearchable: true,
+            textTypeOptions: {},
+          },
+        ],
+      },
+    };
 
     // Make Request
     const response = serviceClient.createDocumentSchema(request);

document-warehouse/delete-document-schema.js

@@ -34,8 +34,15 @@ async function main(
   const {DocumentSchemaServiceClient} =
     require('@google-cloud/contentwarehouse').v1;
 
+  const apiEndpoint =
+    location === 'us'
+      ? 'contentwarehouse.googleapis.com'
+      : `${location}-contentwarehouse.googleapis.com`;
+
   // Create service client
-  const serviceClient = new DocumentSchemaServiceClient();
+  const serviceClient = new DocumentSchemaServiceClient({
+    apiEndpoint: apiEndpoint,
+  });
 
   // Delete Document Schema
   async function deleteDocumentSchema() {

document-warehouse/fetch-acl.js

@@ -34,8 +34,13 @@ async function main(
   // Import from google cloud
   const {DocumentServiceClient} = require('@google-cloud/contentwarehouse').v1;
 
+  const apiEndpoint =
+    location === 'us'
+      ? 'contentwarehouse.googleapis.com'
+      : `${location}-contentwarehouse.googleapis.com`;
+
   // Create service client
-  const serviceClient = new DocumentServiceClient();
+  const serviceClient = new DocumentServiceClient({apiEndpoint: apiEndpoint});
 
   // Fetches access control policies on project or document level.
   async function fetchACL() {

document-warehouse/get-document-schema.js

@@ -32,8 +32,15 @@ async function main(
   const {DocumentSchemaServiceClient} =
     require('@google-cloud/contentwarehouse').v1;
 
+  const apiEndpoint =
+    location === 'us'
+      ? 'contentwarehouse.googleapis.com'
+      : `${location}-contentwarehouse.googleapis.com`;
+
   // Create service client
-  const serviceClient = new DocumentSchemaServiceClient();
+  const serviceClient = new DocumentSchemaServiceClient({
+    apiEndpoint: apiEndpoint,
+  });
 
   // Get Document Schema
   async function getDocumentSchema() {

document-warehouse/get-document.js

@@ -33,8 +33,15 @@ async function main(
   // Import from google cloud
   const {DocumentServiceClient} = require('@google-cloud/contentwarehouse').v1;
 
+  const apiEndpoint =
+    location === 'us'
+      ? 'contentwarehouse.googleapis.com'
+      : `${location}-contentwarehouse.googleapis.com`;
+
   // Create service client
-  const serviceClient = new DocumentServiceClient();
+  const serviceClient = new DocumentServiceClient({
+    apiEndpoint: apiEndpoint,
+  });
 
   // Get Document Schema
   async function getDocument() {

document-warehouse/quickstart.js

@@ -31,9 +31,16 @@ async function main(
   const {DocumentSchemaServiceClient, DocumentServiceClient} =
     require('@google-cloud/contentwarehouse').v1;
 
+  const apiEndpoint =
+    location === 'us'
+      ? 'contentwarehouse.googleapis.com'
+      : `${location}-contentwarehouse.googleapis.com`;
+
   // Create service client
-  const schemaClient = new DocumentSchemaServiceClient();
-  const serviceClient = new DocumentServiceClient();
+  const schemaClient = new DocumentSchemaServiceClient({
+    apiEndpoint: apiEndpoint,
+  });
+  const serviceClient = new DocumentServiceClient({apiEndpoint: apiEndpoint});
 
   // Get Document Schema
   async function quickstart() {

document-warehouse/search-document.js

@@ -32,8 +32,13 @@ async function main(
   // Import from google cloud
   const {DocumentServiceClient} = require('@google-cloud/contentwarehouse').v1;
 
+  const apiEndpoint =
+    location === 'us'
+      ? 'contentwarehouse.googleapis.com'
+      : `${location}-contentwarehouse.googleapis.com`;
+
   // Create service client
-  const serviceClient = new DocumentServiceClient();
+  const serviceClient = new DocumentServiceClient({apiEndpoint: apiEndpoint});
 
   // Get Document Schema
   async function searchDocuments() {

document-warehouse/set-acl.js

@@ -37,8 +37,13 @@ async function main(
   //Import service client from google cloud
   const {DocumentServiceClient} = require('@google-cloud/contentwarehouse').v1;
 
+  const apiEndpoint =
+    location === 'us'
+      ? 'contentwarehouse.googleapis.com'
+      : `${location}-contentwarehouse.googleapis.com`;
+
   // Create client
-  const serviceClient = new DocumentServiceClient();
+  const serviceClient = new DocumentServiceClient({apiEndpoint: apiEndpoint});
 
   // Set access control policies on project or document level.
   async function setACL() {