migrate code from googleapis/nodejs-automl

.github/workflows/automl.yaml

@@ -0,0 +1,74 @@
+name: automl
+on:
+  push:
+    branches:
+    - main
+    paths:
+    - 'automl/**'
+  pull_request:
+    paths:
+    - 'automl/**'
+  pull_request_target:
+    types: [labeled]
+  schedule:
+  - cron:  '0 0 * * 0'
+jobs:
+  test:
+    if: ${{ github.event.action != 'labeled' || github.event.label.name == 'actions:force-run' }}
+    runs-on: ubuntu-latest
+    timeout-minutes: 60
+    permissions:
+      contents: 'write'
+      pull-requests: 'write'
+      id-token: 'write'
+    steps:
+    - uses: actions/checkout@v3
+      with:
+        ref: ${{github.event.pull_request.head.ref}}
+        repository: ${{github.event.pull_request.head.repo.full_name}}
+    - uses: google-github-actions/auth@v1.0.0
+      with:
+        workload_identity_provider: 'projects/1046198160504/locations/global/workloadIdentityPools/github-actions-pool/providers/github-actions-provider'
+        service_account: 'kokoro-system-test@long-door-651.iam.gserviceaccount.com'
+        create_credentials_file: 'true'
+        credentials_json: '${{ secrets.GOOGLE_CREDENTIALS }}'
+        access_token_lifetime: 600s
+    - id: secrets
+      uses: 'google-github-actions/get-secretmanager-secrets@v0'
+      with:
+        secrets: |-
+          table_model_id:nodejs-docs-samples-tests/nodejs-docs-samples-table-model-id
+    - uses: actions/setup-node@v3
+      with:
+        node-version: 14
+    - run: npm install
+      working-directory: automl
+    - run: npm test
+      working-directory: automl
+      env:
+        MOCHA_REPORTER_SUITENAME: automl
+        MOCHA_REPORTER_OUTPUT: automl_sponge_log.xml
+        MOCHA_REPORTER: xunit
+        TABLE_MODEL_ID: ${{ steps.secrets.outputs.table_model_id }}
+    - if: ${{ github.event.action == 'labeled' && github.event.label.name == 'actions:force-run' }}
+      uses: actions/github-script@v6
+      with:
+        github-token: ${{ secrets.GITHUB_TOKEN }}
+        script: |
+          try {
+            await github.rest.issues.removeLabel({
+              name: 'actions:force-run',
+              owner: 'GoogleCloudPlatform',
+              repo: 'nodejs-docs-samples',
+              issue_number: context.payload.pull_request.number
+            });
+          } catch (e) {
+            if (!e.message.includes('Label does not exist')) {
+              throw e;
+            }
+          }
+    - if: ${{ github.event_name == 'schedule' && always() }}
+      run: |
+        curl https://github.com/googleapis/repo-automation-bots/releases/download/flakybot-1.1.0/flakybot -o flakybot -s -L
+        chmod +x ./flakybot
+        ./flakybot --repo GoogleCloudPlatform/nodejs-docs-samples --commit_hash ${{github.sha}} --build_url https://github.com/${{github.repository}}/actions/runs/${{github.run_id}}

.github/workflows/workflows.json

@@ -14,6 +14,7 @@
   "appengine/storage/flexible",
   "appengine/storage/standard",
   "auth",
+  "automl",
   "appengine/typescript",
   "appengine/websockets",
   "appengine/twilio",

automl/.eslintrc.yml

@@ -0,0 +1,3 @@
+---
+rules:
+  no-console: off

automl/batch_predict.js

@@ -0,0 +1,74 @@
+// Copyright 2020 Google LLC
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      https://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+'use strict';
+
+function main(
+  projectId = 'YOUR_PROJECT_ID',
+  location = 'us-central1',
+  modelId = 'YOUR_MODEL_ID',
+  inputUri = 'gs://YOUR_BUCKET_ID/path_to_your_input_csv_or_jsonl',
+  outputUri = 'gs://YOUR_BUCKET_ID/path_to_save_results/'
+) {
+  // [START automl_batch_predict]
+  /**
+   * TODO(developer): Uncomment these variables before running the sample.
+   */
+  // const projectId = 'YOUR_PROJECT_ID';
+  // const location = 'us-central1';
+  // const modelId = 'YOUR_MODEL_ID';
+  // const inputUri = 'gs://YOUR_BUCKET_ID/path_to_your_input_csv_or_jsonl';
+  // const outputUri = 'gs://YOUR_BUCKET_ID/path_to_save_results/';
+
+  // Imports the Google Cloud AutoML library
+  const {PredictionServiceClient} = require('@google-cloud/automl').v1;
+
+  // Instantiates a client
+  const client = new PredictionServiceClient();
+
+  async function batchPredict() {
+    // Construct request
+    const request = {
+      name: client.modelPath(projectId, location, modelId),
+      inputConfig: {
+        gcsSource: {
+          inputUris: [inputUri],
+        },
+      },
+      outputConfig: {
+        gcsDestination: {
+          outputUriPrefix: outputUri,
+        },
+      },
+    };
+
+    const [operation] = await client.batchPredict(request);
+
+    console.log('Waiting for operation to complete...');
+    // Wait for operation to complete.
+    const [response] = await operation.promise();
+    console.log(
+      `Batch Prediction results saved to Cloud Storage bucket. ${response}`
+    );
+  }
+
+  batchPredict();
+  // [END automl_batch_predict]
+}
+
+process.on('unhandledRejection', err => {
+  console.error(err.message);
+  process.exitCode = 1;
+});
+main(...process.argv.slice(2));

automl/beta/batch_predict.js

@@ -0,0 +1,74 @@
+// Copyright 2020 Google LLC
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      https://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+'use strict';
+
+function main(
+  projectId = 'YOUR_PROJECT_ID',
+  location = 'us-central1',
+  modelId = 'YOUR_MODEL_ID',
+  inputUri = 'gs://YOUR_BUCKET_ID/path_to_your_input_csv_or_jsonl',
+  outputUri = 'gs://YOUR_BUCKET_ID/path_to_save_results/'
+) {
+  // [START automl_batch_predict_beta]
+  /**
+   * TODO(developer): Uncomment these variables before running the sample.
+   */
+  // const projectId = 'YOUR_PROJECT_ID';
+  // const location = 'us-central1';
+  // const modelId = 'YOUR_MODEL_ID';
+  // const inputUri = 'gs://YOUR_BUCKET_ID/path_to_your_input_csv_or_jsonl';
+  // const outputUri = 'gs://YOUR_BUCKET_ID/path_to_save_results/';
+
+  // Imports the Google Cloud AutoML library
+  const {PredictionServiceClient} = require('@google-cloud/automl').v1beta1;
+
+  // Instantiates a client
+  const client = new PredictionServiceClient();
+
+  async function batchPredict() {
+    // Construct request
+    const request = {
+      name: client.modelPath(projectId, location, modelId),
+      inputConfig: {
+        gcsSource: {
+          inputUris: [inputUri],
+        },
+      },
+      outputConfig: {
+        gcsDestination: {
+          outputUriPrefix: outputUri,
+        },
+      },
+    };
+
+    const [operation] = await client.batchPredict(request);
+
+    console.log('Waiting for operation to complete...');
+    // Wait for operation to complete.
+    const [response] = await operation.promise();
+    console.log(
+      `Batch Prediction results saved to Cloud Storage bucket. ${response}`
+    );
+  }
+
+  batchPredict();
+  // [END automl_batch_predict_beta]
+}
+
+process.on('unhandledRejection', err => {
+  console.error(err.message);
+  process.exitCode = 1;
+});
+main(...process.argv.slice(2));

automl/beta/cancel_operation.js

@@ -0,0 +1,51 @@
+// Copyright 2020 Google LLC
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+'use strict';
+
+async function main(operationFullId) {
+  // [START automl_cancel_operation_beta]
+
+  /**
+   * TODO(developer): Uncomment these variables before running the sample.
+   */
+  // const operationFullId = 'projects/YOUR_PROJECT_ID/locations/YOUR_LOCATIOIN/operations/OPERATION_ID';
+
+  // Imports the Google Cloud AutoML library
+  const {AutoMlClient} = require('@google-cloud/automl').v1beta1;
+
+  // Instantiates a client
+  const client = new AutoMlClient();
+
+  async function cancelOperation() {
+    client.operationsClient.cancelOperation({
+      name: operationFullId,
+    });
+
+    // Wait for operation to complete.
+    console.log('Cancelled operation');
+  }
+
+  cancelOperation();
+  // [END automl_cancel_operation_beta]
+}
+
+main(...process.argv.slice(2)).catch(err => {
+  console.error(err.message);
+  process.exitCode = 1;
+});
+process.on('unhandledRejection', err => {
+  console.error(err.message);
+  process.exitCode = 1;
+});

automl/beta/delete-dataset.js

@@ -0,0 +1,57 @@
+// Copyright 2020 Google LLC
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+'use strict';
+
+function main(
+  projectId = 'YOUR_PROJECT_ID',
+  location = 'us-central1',
+  datasetId = 'YOUR_DATASET_ID'
+) {
+  // [START automl_delete_dataset_beta]
+  /**
+   * TODO(developer): Uncomment these variables before running the sample.
+   */
+  // const projectId = 'YOUR_PROJECT_ID';
+  // const location = 'us-central1';
+  // const datasetId = 'YOUR_DATASET_ID';
+
+  // Imports the Google Cloud AutoML library
+  const {AutoMlClient} = require('@google-cloud/automl').v1beta1;
+
+  // Instantiates a client
+  const client = new AutoMlClient();
+
+  async function deleteDataset() {
+    // Construct request
+    const request = {
+      name: client.datasetPath(projectId, location, datasetId),
+    };
+
+    const [operation] = await client.deleteDataset(request);
+
+    // Wait for operation to complete.
+    const [response] = await operation.promise();
+    console.log(`Dataset deleted: ${response}`);
+  }
+
+  deleteDataset();
+  // [END automl_delete_dataset_beta]
+}
+
+process.on('unhandledRejection', err => {
+  console.error(err.message);
+  process.exitCode = 1;
+});
+main(...process.argv.slice(2));