Skip to content

Commit

Permalink
samples: migrate code from googleapis/java-aiplatform (#7451)
Browse files Browse the repository at this point in the history
### Migrating samples from [googleapis/java-aiplatform](https://github.com/googleapis/java-aiplatform/tree/main/samples) into [java-docs-samples/aiplatform](https://github.com/GoogleCloudPlatform/java-docs-samples)
---

- feat: initial code generation
- chore(deps): update dependency com.google.cloud.samples:shared-configuration to v1.0.18 (#2)
- chore(deps): update dependency com.google.cloud.samples:shared-configuration to v1.0.19 (#11)
- samples: update pom.xml (s) in the samples dir (#24)
- chore(deps): update dependency com.google.cloud.samples:shared-configuration to v1.0.21 (#14)
- test(deps): update dependency junit:junit to v4.13.1
- test(deps): update dependency com.google.truth:truth to v1.1 (#33)
- deps: update dependency com.google.cloud:google-cloud-storage to v1.113.2 (#38)
- samples: ucaip samples batch 3 of 6 (#18)
- samples: ucaip batch samples 4 of 6 (#19)
- samples: ucaip batch samples 5 of 6 (#20)
- samples: ucaip samples batch 6 of  6 (#17)
- samples: ucaip sample batch 2 of 6 (#15)
- chore: delete ununsed format script (#48)
- samples: initial commit  and it is missing env vars for testing (#7)
- chore(deps): update dependency com.google.cloud:google-cloud-aiplatform to v0.1.0 (#59)
- samples(deps): update dependency com.google.cloud:google-cloud-storage to v1.113.3 (#55)
- samples(deps): update dependency com.google.protobuf:protobuf-java-util to v4.0.0-rc-2 (#27)
- samples: adds custom model, action recognition samples and tests (#111)
- feat: adds ValueConverter utility and demo samples (#108)
- samples(deps): update dependency com.google.cloud:google-cloud-storage to v1.113.8 (#72)
- chore(deps): update dependency com.google.cloud:google-cloud-aiplatform to v0.2.0 (#133)
- test(deps): update dependency com.google.truth:truth to v1.1.2 (#138)
- samples: adds enhanced library samples (#134)
- test(deps): update dependency junit:junit to v4.13.2 (#161)
- deps: update dependency com.google.cloud:google-cloud-storage to v1.113.9 (#132)
- fix: reduces image object detection test resource file size (#154)
- deps: update dependency com.google.cloud:google-cloud-storage to v1.113.10 (#164)
- feat: updates library for v1 service (#157)
- deps: update dependency com.google.cloud:google-cloud-storage to v1.113.11 (#175)
- docs: updates prediction to new hostname (#173)
- chore: replaced hard-coded values with env vars (#185)
- deps: update dependency com.google.cloud:google-cloud-storage to v1.113.12 (#186)
- chore(deps): update dependency com.google.cloud:google-cloud-aiplatform to v0.3.0 (#188)
- deps: update dependency com.google.api.grpc:proto-google-cloud-aiplatform-v1beta1 to v0.3.0 (#189)
- deps: update dependency com.google.cloud:google-cloud-storage to v1.113.14 (#202)
- chore(deps): update dependency com.google.cloud:google-cloud-aiplatform to v0.4.0 (#209)
- chore(deps): update dependency com.google.cloud.samples:shared-configuration to v1.0.22 (#223)
- deps: update dependency com.google.api.grpc:proto-google-cloud-aiplatform-v1beta1 to v0.4.0 (#210)
- deps: update dependency com.google.cloud:google-cloud-storage to v1.113.15 (#232)
- samples: updates samples to v1 (1 of 8) (#206)
- samples: updates samples to v1 (2 of 8) (#212)
- test(teardown): refactored teardown section of test (#254)
- test(env var): added missing requireEnv val (#255)
- samples:  updates samples to v1 (3 of 8) (#215)
- samples: updates samples to v1 (5 of 10) (#235)
- samples: updates samples to v1 (4 of 10) (#233)
- deps: update dependency com.google.cloud:google-cloud-storage to v1.113.16 (#334)
- chore: Increasing timeout createdataset (#337)
- samples: updates samples to v1 (7 of 10) (#238)
- samples: updates samples to v1 (6 of 10) (#236)
- samples: updates samples to v1 (8 of 8) (#239)
- deps: update dependency com.google.cloud:google-cloud-storage to v1.114.0 (#349)
- deps: update dependency com.google.code.gson:gson to v2.8.7 (#382)
- test(deps): update dependency com.google.truth:truth to v1.1.3 (#383)
- chore: changes product name to 'Vertex AI' (#373)
- chore(deps): update dependency com.google.cloud.samples:shared-configuration to v1.0.23 (#399)
- deps: update dependency com.google.cloud:google-cloud-storage to v1.115.0 (#396)
- chore(deps): update dependency com.google.cloud:google-cloud-aiplatform to v1 (#406)
- chore: disabling ITs to reduce cost & resolve quota issue (#421)
- deps: update dependency com.google.cloud:google-cloud-storage to v1.117.1 (#422)
- test: Refactor teardown (#424)
- deps: update dependency com.google.api.grpc:proto-google-cloud-aiplatform-v1beta1 to v0.5.1 (#428)
- chore(deps): update dependency com.google.cloud:google-cloud-aiplatform to v1.0.2 (#427)
- deps: update dependency com.google.cloud:google-cloud-storage to v1.118.0 (#440)
- deps: update dependency com.google.api.grpc:proto-google-cloud-aiplatform-v1beta1 to v0.5.2 (#434)
- deps: update dependency com.google.cloud:google-cloud-storage to v2 (#610)
- chore(deps): update dependency com.google.cloud:google-cloud-aiplatform to v2 (#609)
- deps: update dependency com.google.cloud:google-cloud-storage to v2.0.1 (#612)
- deps: update dependency com.google.api.grpc:proto-google-cloud-aiplatform-v1beta1 to v0.6.0 (#608)
- chore(deps): update dependency com.google.cloud:google-cloud-aiplatform to v2.0.1 (#616)
- deps: update dependency com.google.code.gson:gson to v2.8.8 (#622)
- deps: update dependency com.google.cloud:google-cloud-storage to v2.0.2 (#621)
- deps: update dependency com.google.api.grpc:proto-google-cloud-aiplatform-v1beta1 to v0.6.1 (#617)
- deps: update dependency com.google.cloud:google-cloud-storage to v2.1.0 (#630)
- deps: update dependency com.google.api.grpc:proto-google-cloud-aiplatform-v1beta1 to v0.6.2 (#629)
- chore(deps): update dependency com.google.cloud:google-cloud-aiplatform to v2.0.2 (#628)
- chore(deps): update dependency com.google.cloud:google-cloud-aiplatform to v2.1.0 (#640)
- deps: update dependency com.google.cloud:google-cloud-storage to v2.1.1 (#644)
- chore(deps): update dependency com.google.cloud:google-cloud-aiplatform to v2.2.0 (#650)
- deps: update dependency com.google.api.grpc:proto-google-cloud-aiplatform-v1beta1 to v0.8.0 (#641)
- chore(deps): update dependency com.google.cloud:google-cloud-aiplatform to v2.2.1 (#660)
- deps: update dependency com.google.cloud:google-cloud-storage to v2.1.2 (#662)
- deps: update dependency com.google.api.grpc:proto-google-cloud-aiplatform-v1beta1 to v0.8.1 (#661)
- deps: update dependency com.google.api.grpc:proto-google-cloud-aiplatform-v1beta1 to v0.8.2 (#666)
- deps: update dependency com.google.cloud:google-cloud-storage to v2.1.3 (#667)
- deps: update dependency com.google.cloud:google-cloud-storage to v2.1.4 (#669)
- deps: update dependency com.google.cloud:google-cloud-storage to v2.1.5 (#673)
- chore(deps): update dependency com.google.cloud:google-cloud-aiplatform to v2.2.2 (#665)
- deps: update dependency com.google.api.grpc:proto-google-cloud-aiplatform-v1beta1 to v0.8.3 (#678)
- chore(deps): update dependency com.google.cloud:google-cloud-aiplatform to v2.2.3 (#677)
- chore(deps): update dependency com.google.cloud:google-cloud-aiplatform to v2.2.4 (#681)
- deps: update dependency com.google.cloud:google-cloud-storage to v2.1.6 (#683)
- deps: update dependency com.google.api.grpc:proto-google-cloud-aiplatform-v1beta1 to v0.8.4 (#682)
- deps: update dependency com.google.cloud:google-cloud-storage to v2.1.7 (#689)
- deps: update dependency com.google.cloud:google-cloud-storage to v2.1.8 (#696)
- deps: update dependency com.google.cloud:google-cloud-storage to v2.1.9 (#698)
- chore(deps): update dependency com.google.cloud:google-cloud-aiplatform to v2.3.0 (#702)
- deps: update dependency com.google.api.grpc:proto-google-cloud-aiplatform-v1beta1 to v0.9.0 (#703)
- deps: update dependency com.google.code.gson:gson to v2.8.9 (#707)
- deps: update dependency com.google.cloud:google-cloud-storage to v2.2.0 (#710)
- deps: update dependency com.google.cloud:google-cloud-storage to v2.2.1 (#722)
- docs: Added evalId comment to samples (#727)
- deps: update dependency com.google.api.grpc:proto-google-cloud-aiplatform-v1beta1 to v0.10.0 (#733)
- chore(deps): update dependency com.google.cloud:google-cloud-aiplatform to v2.4.0 (#732)
- deps: update dependency com.google.api.grpc:proto-google-cloud-aiplatform-v1beta1 to v0.11.0 (#740)
- chore(deps): update dependency com.google.cloud:google-cloud-aiplatform to v2.5.0 (#739)
- deps: update dependency com.google.cloud:google-cloud-storage to v2.2.2 (#747)
- deps: update dependency com.google.api.grpc:proto-google-cloud-aiplatform-v1beta1 to v0.11.2 (#744)
- chore(deps): update dependency com.google.cloud:google-cloud-aiplatform to v2.5.2 (#748)
- chore(deps): update dependency com.google.cloud.samples:shared-configuration to v1.0.24 (#734)
- chore(deps): update dependency com.google.cloud.samples:shared-configuration to v1.2.0 (#749)
- test: Add null check in teardown (#752)
- deps: update dependency com.google.cloud:google-cloud-storage to v2.2.3 (#764)
- deps: update dependency com.google.cloud:google-cloud-storage to v2.3.0 (#773)
- deps: update dependency com.google.api.grpc:proto-google-cloud-aiplatform-v1beta1 to v0.11.3 (#762)
- chore(deps): update dependency com.google.cloud:google-cloud-aiplatform to v2.5.3 (#761)
- chore(deps): update dependency com.google.cloud:google-cloud-aiplatform to v2.6.0 (#784)
- deps: update dependency com.google.api.grpc:proto-google-cloud-aiplatform-v1beta1 to v0.12.0 (#782)
- deps: update dependency com.google.code.gson:gson to v2.9.0 (#792)
- deps: update dependency com.google.cloud:google-cloud-storage to v2.4.0 (#783)
- deps: update dependency com.google.cloud:google-cloud-storage to v2.4.4 (#801)
- chore(deps): update dependency com.google.cloud:google-cloud-aiplatform to v2.7.0 (#807)
- deps: update dependency com.google.api.grpc:proto-google-cloud-aiplatform-v1beta1 to v0.13.0 (#808)
- chore(deps): update dependency com.google.cloud:google-cloud-aiplatform to v2.7.1 (#813)
- deps: update dependency com.google.api.grpc:proto-google-cloud-aiplatform-v1beta1 to v0.13.1 (#812)
- chore(deps): update dependency com.google.cloud:google-cloud-aiplatform to v2.7.2 (#819)
- deps: update dependency com.google.api.grpc:proto-google-cloud-aiplatform-v1beta1 to v0.13.2 (#817)
- deps: update dependency com.google.cloud:google-cloud-storage to v2.4.5 (#818)
- chore(deps): update dependency com.google.cloud:google-cloud-aiplatform to v2.7.3 (#822)
- deps: update dependency com.google.api.grpc:proto-google-cloud-aiplatform-v1beta1 to v0.13.3 (#823)
- chore(deps): update dependency com.google.cloud:google-cloud-aiplatform to v2.7.4 (#828)
- deps: update dependency com.google.api.grpc:proto-google-cloud-aiplatform-v1beta1 to v0.13.4 (#829)
- deps: update dependency com.google.cloud:google-cloud-storage to v2.5.0 (#834)
- deps: update dependency com.google.cloud:google-cloud-storage to v2.5.1 (#835)
- chore(deps): update dependency com.google.cloud:google-cloud-aiplatform to v2.8.0 (#838)
- deps: update dependency com.google.cloud:google-cloud-storage to v2.6.0 (#840)
- deps: update dependency com.google.api.grpc:proto-google-cloud-aiplatform-v1beta1 to v0.14.0 (#839)
- chore(deps): update dependency com.google.cloud:google-cloud-aiplatform to v2.8.1 (#846)
- deps: update dependency com.google.api.grpc:proto-google-cloud-aiplatform-v1beta1 to v0.14.1 (#847)
- deps: update dependency com.google.api.grpc:proto-google-cloud-aiplatform-v1beta1 to v0.14.2 (#852)
- chore(deps): update dependency com.google.cloud:google-cloud-aiplatform to v2.8.2 (#851)
- chore(deps): update dependency com.google.cloud:google-cloud-aiplatform to v2.8.3 (#855)
- deps: update dependency com.google.api.grpc:proto-google-cloud-aiplatform-v1beta1 to v0.14.3 (#856)
- deps: update dependency com.google.api.grpc:proto-google-cloud-aiplatform-v1beta1 to v0.14.4 (#860)
- chore(deps): update dependency com.google.cloud:google-cloud-aiplatform to v2.8.4 (#859)
- deps: update dependency com.google.cloud:google-cloud-storage to v2.6.1 (#863)
- chore(deps): update dependency com.google.cloud:google-cloud-aiplatform to v2.8.5 (#868)
- deps: update dependency com.google.api.grpc:proto-google-cloud-aiplatform-v1beta1 to v0.14.5 (#869)
- deps: update dependency com.google.api.grpc:proto-google-cloud-aiplatform-v1beta1 to v0.14.6 (#875)
- chore(deps): update dependency com.google.cloud:google-cloud-aiplatform to v2.8.6 (#874)
- chore(deps): update dependency com.google.cloud:google-cloud-aiplatform to v2.8.7 (#878)
- deps: update dependency com.google.api.grpc:proto-google-cloud-aiplatform-v1beta1 to v0.14.7 (#879)
- chore(deps): update dependency com.google.cloud:google-cloud-aiplatform to v2.8.8 (#882)
- deps: update dependency com.google.api.grpc:proto-google-cloud-aiplatform-v1beta1 to v0.14.8 (#883)
- deps: update dependency com.google.api.grpc:proto-google-cloud-aiplatform-v1beta1 to v0.14.9 (#890)
- chore(deps): update dependency com.google.cloud:google-cloud-aiplatform to v2.8.9 (#889)
- deps: update dependency com.google.api.grpc:proto-google-cloud-aiplatform-v1beta1 to v0.15.0 (#915)
- deps: update dependency com.google.cloud:google-cloud-storage to v2.7.0 (#916)
- chore(deps): update dependency com.google.cloud:google-cloud-aiplatform to v2.9.0 (#914)
- chore(deps): update dependency com.google.cloud:google-cloud-aiplatform to v2.9.1 (#919)
- deps: update dependency com.google.api.grpc:proto-google-cloud-aiplatform-v1beta1 to v0.15.1 (#920)
- deps: update dependency com.google.api.grpc:proto-google-cloud-aiplatform-v1beta1 to v0.15.2 (#924)
- chore(deps): update dependency com.google.cloud:google-cloud-aiplatform to v2.9.2 (#923)
- chore(deps): update dependency com.google.cloud:google-cloud-aiplatform to v2.9.3 (#927)
- deps: update dependency com.google.api.grpc:proto-google-cloud-aiplatform-v1beta1 to v0.15.3 (#928)
- deps: update dependency com.google.api.grpc:proto-google-cloud-aiplatform-v1beta1 to v0.15.4 (#932)
- chore(deps): update dependency com.google.cloud:google-cloud-aiplatform to v2.9.4 (#931)
- deps: update dependency com.google.api.grpc:proto-google-cloud-aiplatform-v1beta1 to v0.15.5 (#936)
- chore(deps): update dependency com.google.cloud:google-cloud-aiplatform to v2.9.5 (#935)
- chore(deps): update dependency com.google.cloud:google-cloud-aiplatform to v2.9.6 (#941)
- deps: update dependency com.google.api.grpc:proto-google-cloud-aiplatform-v1beta1 to v0.15.6 (#942)
- chore(deps): update dependency com.google.cloud:google-cloud-aiplatform to v2.9.7 (#953)
- deps: update dependency com.google.api.grpc:proto-google-cloud-aiplatform-v1beta1 to v0.15.7 (#954)
- feat(samples): add create-featurestore sample (#948)
- feat(samples): added all entity type samples (#976)
- feat(samples): add remaining featurestore api samples (#974)
- chore(deps): update dependency com.google.cloud:google-cloud-aiplatform to v3 (#986)
- deps: update dependency com.google.api.grpc:proto-google-cloud-aiplatform-v1beta1 to v0.17.0 (#1003)
- deps: update dependency com.google.code.gson:gson to v2.9.1 (#1001)
- deps: update dependency com.google.cloud:google-cloud-storage to v2.10.0 (#1004)
- feat(samples): add all feature samples (#980)
- deps: update dependency com.google.cloud:google-cloud-storage to v2.11.0 (#1005)
- feat(samples): add all feature values samples (#981)
- deps: update dependency com.google.cloud:google-cloud-bigquery to v2.14.1 (#1006)
- deps: update dependency com.google.cloud:google-cloud-storage to v2.11.1 (#1008)
- deps: update dependency com.google.cloud:google-cloud-storage to v2.11.2 (#1010)
- deps: update dependency com.google.cloud:google-cloud-bigquery to v2.14.3 (#1009)
- chore(deps): update dependency com.google.cloud:google-cloud-aiplatform to v3.2.0 (#1020)
- deps: update dependency com.google.api.grpc:proto-google-cloud-aiplatform-v1beta1 to v0.18.0 (#1021)
- deps: update dependency com.google.cloud:google-cloud-bigquery to v2.14.5 (#1016)
- deps: update dependency com.google.cloud:google-cloud-bigquery to v2.14.6 (#1026)
- deps: update dependency com.google.cloud:google-cloud-storage to v2.11.3 (#1015)
- deps: update dependency com.google.cloud:google-cloud-bigquery to v2.14.7 (#1031)
- deps: update dependency com.google.cloud:google-cloud-bigquery to v2.15.0 (#1040)
- deps: update dependency com.google.cloud:google-cloud-bigquery to v2.16.0 (#1044)
- deps: update dependency com.google.api.grpc:proto-google-cloud-aiplatform-v1beta1 to v0.19.0 (#1051)
- chore(deps): update dependency com.google.cloud:google-cloud-aiplatform to v3.3.0 (#1050)
- deps: update dependency com.google.cloud:google-cloud-bigquery to v2.16.1 (#1046)
- deps: update dependency com.google.cloud:google-cloud-storage to v2.12.0 (#1047)
- feat(sample): featurestore node updates (#1028)
- chore(deps): update dependency com.google.cloud:google-cloud-aiplatform to v3.4.0 (#1085)
- deps: update dependency com.google.cloud:google-cloud-bigquery to v2.17.0 (#1082)
- deps: update dependency com.google.api.grpc:proto-google-cloud-aiplatform-v1beta1 to v0.20.0 (#1086)
- deps: update dependency com.google.cloud:google-cloud-storage to v2.13.0 (#1087)
- deps: update dependency com.google.cloud:google-cloud-bigquery to v2.17.1 (#1092)
- chore(deps): update dependency com.google.cloud:google-cloud-aiplatform to v3.4.1 (#1090)
- deps: update dependency com.google.api.grpc:proto-google-cloud-aiplatform-v1beta1 to v0.20.1 (#1091)
- deps: update dependency com.google.cloud:google-cloud-bigquery to v2.18.0 (#1113)

Fixes #issue

> It's a good idea to open an issue first for discussion.

- [ ] I have followed [Sample Format Guide](https://github.com/GoogleCloudPlatform/java-docs-samples/blob/main/SAMPLE_FORMAT.md)
- [ ] `pom.xml` parent set to latest `shared-configuration`
- [ ] Appropriate changes to README are included in PR
- [ ] API's need to be enabled to test (tell us)
- [ ] Environment Variables need to be set (ask us to set them)
- [ ] **Tests** pass:   `mvn clean verify` **required**
- [ ] **Lint**  passes: `mvn -P lint checkstyle:check` **required**
- [ ] **Static Analysis**:  `mvn -P lint clean compile pmd:cpd-check spotbugs:check` **advisory only**
- [ ] Please **merge** this PR for me once it is approved.
  • Loading branch information
Shabirmean committed Dec 20, 2022
1 parent 34fd127 commit 24b693d
Show file tree
Hide file tree
Showing 193 changed files with 19,563 additions and 0 deletions.
71 changes: 71 additions & 0 deletions aiplatform/pom.xml
Original file line number Diff line number Diff line change
@@ -0,0 +1,71 @@
<?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.aiplatform</groupId>
<artifactId>aiplatform-snippets</artifactId>
<packaging>jar</packaging>
<name>Google Cloud Vertex AI Snippets</name>
<url>https://github.com/GoogleCloudPlatform/java-docs-samples/tree/main/aiplatform</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 aiplatform_install_with_bom] -->
<dependencies>
<dependency>
<groupId>com.google.cloud</groupId>
<artifactId>google-cloud-aiplatform</artifactId>
<version>3.4.1</version>
</dependency>
<!-- [END aiplatform_install_with_bom] -->
<dependency>
<groupId>com.google.cloud</groupId>
<artifactId>google-cloud-storage</artifactId>
<version>2.13.0</version>
</dependency>
<dependency>
<groupId>com.google.protobuf</groupId>
<artifactId>protobuf-java-util</artifactId>
<version>4.0.0-rc-2</version>
</dependency>
<dependency>
<groupId>com.google.code.gson</groupId>
<artifactId>gson</artifactId>
<version>2.9.1</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.api.grpc</groupId>
<artifactId>proto-google-cloud-aiplatform-v1beta1</artifactId>
<version>0.20.1</version>
</dependency>
<dependency>
<groupId>com.google.cloud</groupId>
<artifactId>google-cloud-bigquery</artifactId>
<version>2.18.0</version>
</dependency>
</dependencies>
</project>
Binary file added aiplatform/resources/daisy.jpg
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
Binary file added aiplatform/resources/image_flower_daisy.jpg
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
Binary file added aiplatform/resources/iod_caprese_salad.jpg
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
128 changes: 128 additions & 0 deletions aiplatform/src/main/java/aiplatform/BatchCreateFeaturesSample.java
Original file line number Diff line number Diff line change
@@ -0,0 +1,128 @@
/*
* Copyright 2022 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.
*
*
* Create features in bulk for an existing entity type. See
* https://cloud.google.com/vertex-ai/docs/featurestore/setup
* before running the code snippet
*/

package aiplatform;

// [START aiplatform_batch_create_features_sample]

import com.google.api.gax.longrunning.OperationFuture;
import com.google.cloud.aiplatform.v1.BatchCreateFeaturesOperationMetadata;
import com.google.cloud.aiplatform.v1.BatchCreateFeaturesRequest;
import com.google.cloud.aiplatform.v1.BatchCreateFeaturesResponse;
import com.google.cloud.aiplatform.v1.CreateFeatureRequest;
import com.google.cloud.aiplatform.v1.EntityTypeName;
import com.google.cloud.aiplatform.v1.Feature;
import com.google.cloud.aiplatform.v1.Feature.ValueType;
import com.google.cloud.aiplatform.v1.FeaturestoreServiceClient;
import com.google.cloud.aiplatform.v1.FeaturestoreServiceSettings;
import java.io.IOException;
import java.util.ArrayList;
import java.util.List;
import java.util.concurrent.ExecutionException;
import java.util.concurrent.TimeUnit;
import java.util.concurrent.TimeoutException;

public class BatchCreateFeaturesSample {

public static void main(String[] args)
throws IOException, InterruptedException, ExecutionException, TimeoutException {
// TODO(developer): Replace these variables before running the sample.
String project = "YOUR_PROJECT_ID";
String featurestoreId = "YOUR_FEATURESTORE_ID";
String entityTypeId = "YOUR_ENTITY_TYPE_ID";
String location = "us-central1";
String endpoint = "us-central1-aiplatform.googleapis.com:443";
int timeout = 300;
batchCreateFeaturesSample(project, featurestoreId, entityTypeId, location, endpoint, timeout);
}

static void batchCreateFeaturesSample(
String project,
String featurestoreId,
String entityTypeId,
String location,
String endpoint,
int timeout)
throws IOException, InterruptedException, ExecutionException, TimeoutException {
FeaturestoreServiceSettings featurestoreServiceSettings =
FeaturestoreServiceSettings.newBuilder().setEndpoint(endpoint).build();

// 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 (FeaturestoreServiceClient featurestoreServiceClient =
FeaturestoreServiceClient.create(featurestoreServiceSettings)) {

List<CreateFeatureRequest> createFeatureRequests = new ArrayList<>();

Feature titleFeature =
Feature.newBuilder()
.setDescription("The title of the movie")
.setValueType(ValueType.STRING)
.build();
Feature genresFeature =
Feature.newBuilder()
.setDescription("The genres of the movie")
.setValueType(ValueType.STRING)
.build();
Feature averageRatingFeature =
Feature.newBuilder()
.setDescription("The average rating for the movie, range is [1.0-5.0]")
.setValueType(ValueType.DOUBLE)
.build();

createFeatureRequests.add(
CreateFeatureRequest.newBuilder().setFeature(titleFeature).setFeatureId("title").build());

createFeatureRequests.add(
CreateFeatureRequest.newBuilder()
.setFeature(genresFeature)
.setFeatureId("genres")
.build());

createFeatureRequests.add(
CreateFeatureRequest.newBuilder()
.setFeature(averageRatingFeature)
.setFeatureId("average_rating")
.build());

BatchCreateFeaturesRequest batchCreateFeaturesRequest =
BatchCreateFeaturesRequest.newBuilder()
.setParent(
EntityTypeName.of(project, location, featurestoreId, entityTypeId).toString())
.addAllRequests(createFeatureRequests)
.build();

OperationFuture<BatchCreateFeaturesResponse, BatchCreateFeaturesOperationMetadata>
batchCreateFeaturesFuture =
featurestoreServiceClient.batchCreateFeaturesAsync(batchCreateFeaturesRequest);
System.out.format(
"Operation name: %s%n", batchCreateFeaturesFuture.getInitialFuture().get().getName());
System.out.println("Waiting for operation to finish...");
BatchCreateFeaturesResponse batchCreateFeaturesResponse =
batchCreateFeaturesFuture.get(timeout, TimeUnit.SECONDS);
System.out.println("Batch Create Features Response");
System.out.println(batchCreateFeaturesResponse);
featurestoreServiceClient.close();
}
}
}
// [END aiplatform_batch_create_features_sample]
135 changes: 135 additions & 0 deletions aiplatform/src/main/java/aiplatform/BatchReadFeatureValuesSample.java
Original file line number Diff line number Diff line change
@@ -0,0 +1,135 @@
/*
* Copyright 2022 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.
*
*
* Batch read feature values from a featurestore, as determined by your
* read instances list file, to export data. See
* https://cloud.google.com/vertex-ai/docs/featurestore/setup before running
* the code snippet
*/

package aiplatform;

// [START aiplatform_batch_read_feature_values_sample]

import com.google.api.gax.longrunning.OperationFuture;
import com.google.cloud.aiplatform.v1.BatchReadFeatureValuesOperationMetadata;
import com.google.cloud.aiplatform.v1.BatchReadFeatureValuesRequest;
import com.google.cloud.aiplatform.v1.BatchReadFeatureValuesRequest.EntityTypeSpec;
import com.google.cloud.aiplatform.v1.BatchReadFeatureValuesResponse;
import com.google.cloud.aiplatform.v1.BigQueryDestination;
import com.google.cloud.aiplatform.v1.CsvSource;
import com.google.cloud.aiplatform.v1.FeatureSelector;
import com.google.cloud.aiplatform.v1.FeatureValueDestination;
import com.google.cloud.aiplatform.v1.FeaturestoreName;
import com.google.cloud.aiplatform.v1.FeaturestoreServiceClient;
import com.google.cloud.aiplatform.v1.FeaturestoreServiceSettings;
import com.google.cloud.aiplatform.v1.GcsSource;
import com.google.cloud.aiplatform.v1.IdMatcher;
import java.io.IOException;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
import java.util.concurrent.ExecutionException;
import java.util.concurrent.TimeUnit;
import java.util.concurrent.TimeoutException;

public class BatchReadFeatureValuesSample {

public static void main(String[] args)
throws IOException, InterruptedException, ExecutionException, TimeoutException {
// TODO(developer): Replace these variables before running the sample.
String project = "YOUR_PROJECT_ID";
String featurestoreId = "YOUR_FEATURESTORE_ID";
String entityTypeId = "YOUR_ENTITY_TYPE_ID";
String inputCsvFile = "YOU_INPUT_CSV_FILE";
String destinationTableUri = "YOUR_DESTINATION_TABLE_URI";
List<String> featureSelectorIds = Arrays.asList("title", "genres", "average_rating");
String location = "us-central1";
String endpoint = "us-central1-aiplatform.googleapis.com:443";
int timeout = 300;
batchReadFeatureValuesSample(
project,
featurestoreId,
entityTypeId,
inputCsvFile,
destinationTableUri,
featureSelectorIds,
location,
endpoint,
timeout);
}

static void batchReadFeatureValuesSample(
String project,
String featurestoreId,
String entityTypeId,
String inputCsvFile,
String destinationTableUri,
List<String> featureSelectorIds,
String location,
String endpoint,
int timeout)
throws IOException, InterruptedException, ExecutionException, TimeoutException {
FeaturestoreServiceSettings featurestoreServiceSettings =
FeaturestoreServiceSettings.newBuilder().setEndpoint(endpoint).build();

// 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 (FeaturestoreServiceClient featurestoreServiceClient =
FeaturestoreServiceClient.create(featurestoreServiceSettings)) {

List<EntityTypeSpec> entityTypeSpecs = new ArrayList<>();

FeatureSelector featureSelector =
FeatureSelector.newBuilder()
.setIdMatcher(IdMatcher.newBuilder().addAllIds(featureSelectorIds).build())
.build();
EntityTypeSpec entityTypeSpec =
EntityTypeSpec.newBuilder()
.setEntityTypeId(entityTypeId)
.setFeatureSelector(featureSelector)
.build();

entityTypeSpecs.add(entityTypeSpec);

BigQueryDestination bigQueryDestination =
BigQueryDestination.newBuilder().setOutputUri(destinationTableUri).build();
GcsSource gcsSource = GcsSource.newBuilder().addUris(inputCsvFile).build();
BatchReadFeatureValuesRequest batchReadFeatureValuesRequest =
BatchReadFeatureValuesRequest.newBuilder()
.setFeaturestore(FeaturestoreName.of(project, location, featurestoreId).toString())
.setCsvReadInstances(CsvSource.newBuilder().setGcsSource(gcsSource))
.setDestination(
FeatureValueDestination.newBuilder().setBigqueryDestination(bigQueryDestination))
.addAllEntityTypeSpecs(entityTypeSpecs)
.build();

OperationFuture<BatchReadFeatureValuesResponse, BatchReadFeatureValuesOperationMetadata>
batchReadFeatureValuesFuture =
featurestoreServiceClient.batchReadFeatureValuesAsync(batchReadFeatureValuesRequest);
System.out.format(
"Operation name: %s%n", batchReadFeatureValuesFuture.getInitialFuture().get().getName());
System.out.println("Waiting for operation to finish...");
BatchReadFeatureValuesResponse batchReadFeatureValuesResponse =
batchReadFeatureValuesFuture.get(timeout, TimeUnit.SECONDS);
System.out.println("Batch Read Feature Values Response");
System.out.println(batchReadFeatureValuesResponse);
featurestoreServiceClient.close();
}
}
}
// [END aiplatform_batch_read_feature_values_sample]
Original file line number Diff line number Diff line change
@@ -0,0 +1,56 @@
/*
* Copyright 2021 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 aiplatform;

// [START aiplatform_cancel_batch_prediction_job_sample]

import com.google.cloud.aiplatform.v1.BatchPredictionJobName;
import com.google.cloud.aiplatform.v1.JobServiceClient;
import com.google.cloud.aiplatform.v1.JobServiceSettings;
import java.io.IOException;

public class CancelBatchPredictionJobSample {

public static void main(String[] args) throws IOException {
// TODO(developer): Replace these variables before running the sample.
String project = "YOUR_PROJECT_ID";
String batchPredictionJobId = "YOUR_BATCH_PREDICTION_JOB_ID";
cancelBatchPredictionJobSample(project, batchPredictionJobId);
}

static void cancelBatchPredictionJobSample(String project, String batchPredictionJobId)
throws IOException {
JobServiceSettings jobServiceSettings =
JobServiceSettings.newBuilder()
.setEndpoint("us-central1-aiplatform.googleapis.com:443")
.build();

// 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 (JobServiceClient jobServiceClient = JobServiceClient.create(jobServiceSettings)) {
String location = "us-central1";
BatchPredictionJobName batchPredictionJobName =
BatchPredictionJobName.of(project, location, batchPredictionJobId);

jobServiceClient.cancelBatchPredictionJob(batchPredictionJobName);

System.out.println("Cancelled the Batch Prediction Job");
}
}
}
// [END aiplatform_cancel_batch_prediction_job_sample]
Loading

0 comments on commit 24b693d

Please sign in to comment.