samples: migrate code from googleapis/java-automl

automl/pom.xml

@@ -0,0 +1,81 @@
+<?xml version='1.0' encoding='UTF-8'?>
+<project xmlns="http://maven.apache.org/POM/4.0.0" xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance" xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd">
+  <modelVersion>4.0.0</modelVersion>
+  <groupId>com.example.automl</groupId>
+  <artifactId>automl-snippets</artifactId>
+  <packaging>jar</packaging>
+  <name>Google Cloud Auto ML Snippets</name>
+  <url>https://github.com/GoogleCloudPlatform/java-docs-samples/tree/main/automl</url>
+
+  <!--
+    The parent pom defines common style checks and testing strategies for our samples.
+    Removing or replacing it should not affect the execution of the samples in anyway.
+  -->
+  <parent>
+    <groupId>com.google.cloud.samples</groupId>
+    <artifactId>shared-configuration</artifactId>
+    <version>1.2.0</version>
+  </parent>
+
+  <properties>
+    <maven.compiler.target>1.8</maven.compiler.target>
+    <maven.compiler.source>1.8</maven.compiler.source>
+    <project.build.sourceEncoding>UTF-8</project.build.sourceEncoding>
+  </properties>
+
+
+  <!-- [START automl_install_with_bom] -->
+  <dependencyManagement>
+    <dependencies>
+      <dependency>
+        <groupId>com.google.cloud</groupId>
+        <artifactId>libraries-bom</artifactId>
+        <version>26.1.4</version>
+        <type>pom</type>
+        <scope>import</scope>
+      </dependency>
+    </dependencies>
+  </dependencyManagement>
+
+  <dependencies>
+    <dependency>
+      <groupId>com.google.cloud</groupId>
+      <artifactId>google-cloud-automl</artifactId>
+    </dependency>
+    <dependency>
+      <groupId>com.google.cloud</groupId>
+      <artifactId>google-cloud-bigquery</artifactId>
+    </dependency>
+    <dependency>
+      <groupId>com.google.cloud</groupId>
+      <artifactId>google-cloud-storage</artifactId>
+    </dependency>
+    <!-- [START_EXCLUDE] -->
+    <dependency>
+      <groupId>net.sourceforge.argparse4j</groupId>
+      <artifactId>argparse4j</artifactId>
+      <version>0.9.0</version>
+    </dependency>
+    <dependency>
+      <groupId>junit</groupId>
+      <artifactId>junit</artifactId>
+      <version>4.13.2</version>
+      <scope>test</scope>
+    </dependency>
+    <dependency>
+      <groupId>com.google.truth</groupId>
+      <artifactId>truth</artifactId>
+      <version>1.1.3</version>
+      <scope>test</scope>
+    </dependency>
+    <dependency>
+      <groupId>com.google.cloud</groupId>
+      <artifactId>google-cloud-core</artifactId>
+      <version>2.8.21</version>
+      <scope>test</scope>
+      <classifier>tests</classifier>
+    </dependency>
+    <!-- [END_EXCLUDE] -->
+  </dependencies>
+  <!-- [END automl_install_with_bom] -->
+</project>

automl/resources/input.txt

@@ -0,0 +1 @@
+Tell me how this ends

automl/src/main/java/beta/automl/BatchPredict.java

@@ -0,0 +1,84 @@
+/*
+ * 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.
+ */
+
+package beta.automl;
+
+// [START automl_tables_batch_predict]
+// [START automl_batch_predict_beta]
+import com.google.api.gax.longrunning.OperationFuture;
+import com.google.cloud.automl.v1beta1.BatchPredictInputConfig;
+import com.google.cloud.automl.v1beta1.BatchPredictOutputConfig;
+import com.google.cloud.automl.v1beta1.BatchPredictRequest;
+import com.google.cloud.automl.v1beta1.BatchPredictResult;
+import com.google.cloud.automl.v1beta1.GcsDestination;
+import com.google.cloud.automl.v1beta1.GcsSource;
+import com.google.cloud.automl.v1beta1.ModelName;
+import com.google.cloud.automl.v1beta1.OperationMetadata;
+import com.google.cloud.automl.v1beta1.PredictionServiceClient;
+import java.io.IOException;
+import java.util.concurrent.ExecutionException;
+
+class BatchPredict {
+
+  static void batchPredict() throws IOException, ExecutionException, InterruptedException {
+    // TODO(developer): Replace these variables before running the sample.
+    String projectId = "YOUR_PROJECT_ID";
+    String modelId = "YOUR_MODEL_ID";
+    String inputUri = "gs://YOUR_BUCKET_ID/path_to_your_input_csv_or_jsonl";
+    String outputUri = "gs://YOUR_BUCKET_ID/path_to_save_results/";
+    batchPredict(projectId, modelId, inputUri, outputUri);
+  }
+
+  static void batchPredict(String projectId, String modelId, String inputUri, String outputUri)
+      throws IOException, ExecutionException, InterruptedException {
+    // Initialize client that will be used to send requests. This client only needs to be created
+    // once, and can be reused for multiple requests. After completing all of your requests, call
+    // the "close" method on the client to safely clean up any remaining background resources.
+    try (PredictionServiceClient client = PredictionServiceClient.create()) {
+      // Get the full path of the model.
+      ModelName name = ModelName.of(projectId, "us-central1", modelId);
+
+      // Configure the source of the file from a GCS bucket
+      GcsSource gcsSource = GcsSource.newBuilder().addInputUris(inputUri).build();
+      BatchPredictInputConfig inputConfig =
+          BatchPredictInputConfig.newBuilder().setGcsSource(gcsSource).build();
+
+      // Configure where to store the output in a GCS bucket
+      GcsDestination gcsDestination =
+          GcsDestination.newBuilder().setOutputUriPrefix(outputUri).build();
+      BatchPredictOutputConfig outputConfig =
+          BatchPredictOutputConfig.newBuilder().setGcsDestination(gcsDestination).build();
+
+      // Build the request that will be sent to the API
+      BatchPredictRequest request =
+          BatchPredictRequest.newBuilder()
+              .setName(name.toString())
+              .setInputConfig(inputConfig)
+              .setOutputConfig(outputConfig)
+              .build();
+
+      // Start an asynchronous request
+      OperationFuture<BatchPredictResult, OperationMetadata> future =
+          client.batchPredictAsync(request);
+
+      System.out.println("Waiting for operation to complete...");
+      BatchPredictResult response = future.get();
+      System.out.println("Batch Prediction results saved to specified Cloud Storage bucket.");
+    }
+  }
+}
+// [END automl_batch_predict_beta]
+// [END automl_tables_batch_predict]

automl/src/main/java/beta/automl/CancelOperation.java

@@ -0,0 +1,48 @@
+/*
+ * 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.
+ */
+
+package beta.automl;
+
+// [START automl_cancel_operation_beta]
+
+import com.google.cloud.automl.v1beta1.AutoMlClient;
+import io.grpc.StatusRuntimeException;
+import java.io.IOException;
+
+class CancelOperation {
+
+  static void cancelOperation() throws IOException, InterruptedException, StatusRuntimeException {
+    // TODO(developer): Replace these variables before running the sample.
+    String projectId = "YOUR_PROJECT_ID";
+    String location = "us-central1";
+    String operationId = "YOUR_OPERATION_ID";
+    String operationFullId =
+        String.format("projects/%s/locations/%s/operations/%s", projectId, location, operationId);
+    cancelOperation(operationFullId);
+  }
+
+  static void cancelOperation(String operationFullId)
+      throws IOException, InterruptedException, StatusRuntimeException {
+    // Initialize client that will be used to send requests. This client only needs to be created
+    // once, and can be reused for multiple requests. After completing all of your requests, call
+    // the "close" method on the client to safely clean up any remaining background resources.
+    try (AutoMlClient client = AutoMlClient.create()) {
+      client.getOperationsClient().cancelOperation(operationFullId);
+      System.out.println("Operation cancelled");
+    }
+  }
+}
+// [END automl_cancel_operation_beta]

automl/src/main/java/beta/automl/DeleteDataset.java

@@ -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.
+ */
+
+package beta.automl;
+
+// [START automl_delete_dataset_beta]
+// [START automl_tables_delete_dataset]
+import com.google.cloud.automl.v1beta1.AutoMlClient;
+import com.google.cloud.automl.v1beta1.DatasetName;
+import com.google.protobuf.Empty;
+import java.io.IOException;
+import java.util.concurrent.ExecutionException;
+
+class DeleteDataset {
+
+  static void deleteDataset() throws IOException, ExecutionException, InterruptedException {
+    // TODO(developer): Replace these variables before running the sample.
+    String projectId = "YOUR_PROJECT_ID";
+    String datasetId = "YOUR_DATASET_ID";
+    deleteDataset(projectId, datasetId);
+  }
+
+  // Delete a dataset
+  static void deleteDataset(String projectId, String datasetId)
+      throws IOException, ExecutionException, InterruptedException {
+    // Initialize client that will be used to send requests. This client only needs to be created
+    // once, and can be reused for multiple requests. After completing all of your requests, call
+    // the "close" method on the client to safely clean up any remaining background resources.
+    try (AutoMlClient client = AutoMlClient.create()) {
+      // Get the full path of the dataset.
+      DatasetName datasetFullId = DatasetName.of(projectId, "us-central1", datasetId);
+      Empty response = client.deleteDatasetAsync(datasetFullId).get();
+      System.out.format("Dataset deleted. %s%n", response);
+    }
+  }
+}
+// [END automl_tables_delete_dataset]
+// [END automl_delete_dataset_beta]

automl/src/main/java/beta/automl/DeleteModel.java

@@ -0,0 +1,56 @@
+/*
+ * 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.
+ */
+
+package beta.automl;
+
+// [START automl_tables_delete_model]
+// [START automl_delete_model_beta]
+import com.google.cloud.automl.v1beta1.AutoMlClient;
+import com.google.cloud.automl.v1beta1.ModelName;
+import com.google.protobuf.Empty;
+import java.io.IOException;
+import java.util.concurrent.ExecutionException;
+
+class DeleteModel {
+
+  public static void main(String[] args)
+      throws IOException, ExecutionException, InterruptedException {
+    // TODO(developer): Replace these variables before running the sample.
+    String projectId = "YOUR_PROJECT_ID";
+    String modelId = "YOUR_MODEL_ID";
+    deleteModel(projectId, modelId);
+  }
+
+  // Delete a model
+  static void deleteModel(String projectId, String modelId)
+      throws IOException, ExecutionException, InterruptedException {
+    // Initialize client that will be used to send requests. This client only needs to be created
+    // once, and can be reused for multiple requests. After completing all of your requests, call
+    // the "close" method on the client to safely clean up any remaining background resources.
+    try (AutoMlClient client = AutoMlClient.create()) {
+      // Get the full path of the model.
+      ModelName modelFullId = ModelName.of(projectId, "us-central1", modelId);
+
+      // Delete a model.
+      Empty response = client.deleteModelAsync(modelFullId).get();
+
+      System.out.println("Model deletion started...");
+      System.out.println(String.format("Model deleted. %s", response));
+    }
+  }
+}
+// [END automl_delete_model_beta]
+// [END automl_tables_delete_model]

automl/src/main/java/beta/automl/DeployModel.java

@@ -0,0 +1,59 @@
+/*
+ * 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.
+ */
+
+package beta.automl;
+
+// [START automl_tables_deploy_model]
+// [START automl_deploy_model_beta]
+import com.google.api.gax.longrunning.OperationFuture;
+import com.google.cloud.automl.v1beta1.AutoMlClient;
+import com.google.cloud.automl.v1beta1.DeployModelRequest;
+import com.google.cloud.automl.v1beta1.ModelName;
+import com.google.cloud.automl.v1beta1.OperationMetadata;
+import com.google.protobuf.Empty;
+import java.io.IOException;
+import java.util.concurrent.ExecutionException;
+
+class DeployModel {
+
+  public static void main(String[] args)
+      throws IOException, ExecutionException, InterruptedException {
+    // TODO(developer): Replace these variables before running the sample.
+    String projectId = "YOUR_PROJECT_ID";
+    String modelId = "YOUR_MODEL_ID";
+    deployModel(projectId, modelId);
+  }
+
+  // Deploy a model for prediction
+  static void deployModel(String projectId, String modelId)
+      throws IOException, ExecutionException, InterruptedException {
+    // Initialize client that will be used to send requests. This client only needs to be created
+    // once, and can be reused for multiple requests. After completing all of your requests, call
+    // the "close" method on the client to safely clean up any remaining background resources.
+    try (AutoMlClient client = AutoMlClient.create()) {
+      // Get the full path of the model.
+      ModelName modelFullId = ModelName.of(projectId, "us-central1", modelId);
+      DeployModelRequest request =
+          DeployModelRequest.newBuilder().setName(modelFullId.toString()).build();
+      OperationFuture<Empty, OperationMetadata> future = client.deployModelAsync(request);
+
+      future.get();
+      System.out.println("Model deployment finished");
+    }
+  }
+}
+// [END automl_deploy_model_beta]
+// [END automl_tables_deploy_model]