Skip to content

Commit 1ce4fa6

Browse files
gcf-owl-bot[bot]gcf-owl-bot[bot]
authored
feat: [aiplatform] add encryption_spec to index.proto and index_endpoint.proto (#9805)
* feat: add encryption_spec to index.proto and index_endpoint.proto feat: add TrialContext to study.proto feat: add contexts to SuggestTrialsRequest in vizier_service.proto PiperOrigin-RevId: 563249077 Source-Link: googleapis/googleapis@0e828f8 Source-Link: https://github.com/googleapis/googleapis-gen/commit/e3d57604e959d4d65c1ce0c6196a6b6693162c96 Copy-Tag: eyJwIjoiamF2YS1haXBsYXRmb3JtLy5Pd2xCb3QueWFtbCIsImgiOiJlM2Q1NzYwNGU5NTlkNGQ2NWMxY2UwYzYxOTZhNmI2NjkzMTYyYzk2In0= * 🦉 Updates from OwlBot post-processor See https://github.com/googleapis/repo-automation-bots/blob/main/packages/owl-bot/README.md * 🦉 Updates from OwlBot post-processor See https://github.com/googleapis/repo-automation-bots/blob/main/packages/owl-bot/README.md * feat: add encryption_spec to index.proto and index_endpoint.proto feat: add TrialContext to study.proto feat: add contexts to SuggestTrialsRequest in vizier_service.proto PiperOrigin-RevId: 563257278 Source-Link: googleapis/googleapis@7534629 Source-Link: https://github.com/googleapis/googleapis-gen/commit/63658af6c1664acfabdb4f557c7497101895492a Copy-Tag: eyJwIjoiamF2YS1haXBsYXRmb3JtLy5Pd2xCb3QueWFtbCIsImgiOiI2MzY1OGFmNmMxNjY0YWNmYWJkYjRmNTU3Yzc0OTcxMDE4OTU0OTJhIn0= * 🦉 Updates from OwlBot post-processor See https://github.com/googleapis/repo-automation-bots/blob/main/packages/owl-bot/README.md * 🦉 Updates from OwlBot post-processor See https://github.com/googleapis/repo-automation-bots/blob/main/packages/owl-bot/README.md --------- Co-authored-by: Owl Bot <gcf-owl-bot[bot]@users.noreply.github.com>
1 parent ab08654 commit 1ce4fa6

File tree

53 files changed

+8487
-688
lines changed

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

53 files changed

+8487
-688
lines changed

‎java-aiplatform/google-cloud-aiplatform/src/main/java/com/google/cloud/aiplatform/v1/VizierServiceClient.java‎

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -828,6 +828,7 @@ public final UnaryCallable<LookupStudyRequest, Study> lookupStudyCallable() {
828828
* .setParent(StudyName.of("[PROJECT]", "[LOCATION]", "[STUDY]").toString())
829829
* .setSuggestionCount(390130452)
830830
* .setClientId("clientId908408390")
831+
* .addAllContexts(new ArrayList<TrialContext>())
831832
* .build();
832833
* SuggestTrialsResponse response = vizierServiceClient.suggestTrialsAsync(request).get();
833834
* }
@@ -862,6 +863,7 @@ public final OperationFuture<SuggestTrialsResponse, SuggestTrialsMetadata> sugge
862863
* .setParent(StudyName.of("[PROJECT]", "[LOCATION]", "[STUDY]").toString())
863864
* .setSuggestionCount(390130452)
864865
* .setClientId("clientId908408390")
866+
* .addAllContexts(new ArrayList<TrialContext>())
865867
* .build();
866868
* OperationFuture<SuggestTrialsResponse, SuggestTrialsMetadata> future =
867869
* vizierServiceClient.suggestTrialsOperationCallable().futureCall(request);
@@ -896,6 +898,7 @@ public final OperationFuture<SuggestTrialsResponse, SuggestTrialsMetadata> sugge
896898
* .setParent(StudyName.of("[PROJECT]", "[LOCATION]", "[STUDY]").toString())
897899
* .setSuggestionCount(390130452)
898900
* .setClientId("clientId908408390")
901+
* .addAllContexts(new ArrayList<TrialContext>())
899902
* .build();
900903
* ApiFuture<Operation> future = vizierServiceClient.suggestTrialsCallable().futureCall(request);
901904
* // Do something.

‎java-aiplatform/google-cloud-aiplatform/src/main/java/com/google/cloud/aiplatform/v1beta1/VizierServiceClient.java‎

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -830,6 +830,7 @@ public final UnaryCallable<LookupStudyRequest, Study> lookupStudyCallable() {
830830
* .setParent(StudyName.of("[PROJECT]", "[LOCATION]", "[STUDY]").toString())
831831
* .setSuggestionCount(390130452)
832832
* .setClientId("clientId908408390")
833+
* .addAllContexts(new ArrayList<TrialContext>())
833834
* .build();
834835
* SuggestTrialsResponse response = vizierServiceClient.suggestTrialsAsync(request).get();
835836
* }
@@ -864,6 +865,7 @@ public final OperationFuture<SuggestTrialsResponse, SuggestTrialsMetadata> sugge
864865
* .setParent(StudyName.of("[PROJECT]", "[LOCATION]", "[STUDY]").toString())
865866
* .setSuggestionCount(390130452)
866867
* .setClientId("clientId908408390")
868+
* .addAllContexts(new ArrayList<TrialContext>())
867869
* .build();
868870
* OperationFuture<SuggestTrialsResponse, SuggestTrialsMetadata> future =
869871
* vizierServiceClient.suggestTrialsOperationCallable().futureCall(request);
@@ -898,6 +900,7 @@ public final OperationFuture<SuggestTrialsResponse, SuggestTrialsMetadata> sugge
898900
* .setParent(StudyName.of("[PROJECT]", "[LOCATION]", "[STUDY]").toString())
899901
* .setSuggestionCount(390130452)
900902
* .setClientId("clientId908408390")
903+
* .addAllContexts(new ArrayList<TrialContext>())
901904
* .build();
902905
* ApiFuture<Operation> future = vizierServiceClient.suggestTrialsCallable().futureCall(request);
903906
* // Do something.

‎java-aiplatform/google-cloud-aiplatform/src/main/resources/META-INF/native-image/com.google.cloud.aiplatform.v1/reflect-config.json‎

Lines changed: 18 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -11213,6 +11213,24 @@
1121311213
"allDeclaredClasses": true,
1121411214
"allPublicClasses": true
1121511215
},
11216+
{
11217+
"name": "com.google.cloud.aiplatform.v1.TrialContext",
11218+
"queryAllDeclaredConstructors": true,
11219+
"queryAllPublicConstructors": true,
11220+
"queryAllDeclaredMethods": true,
11221+
"allPublicMethods": true,
11222+
"allDeclaredClasses": true,
11223+
"allPublicClasses": true
11224+
},
11225+
{
11226+
"name": "com.google.cloud.aiplatform.v1.TrialContext$Builder",
11227+
"queryAllDeclaredConstructors": true,
11228+
"queryAllPublicConstructors": true,
11229+
"queryAllDeclaredMethods": true,
11230+
"allPublicMethods": true,
11231+
"allDeclaredClasses": true,
11232+
"allPublicClasses": true
11233+
},
1121611234
{
1121711235
"name": "com.google.cloud.aiplatform.v1.UndeployIndexOperationMetadata",
1121811236
"queryAllDeclaredConstructors": true,

‎java-aiplatform/google-cloud-aiplatform/src/main/resources/META-INF/native-image/com.google.cloud.aiplatform.v1beta1/reflect-config.json‎

Lines changed: 18 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -11762,6 +11762,24 @@
1176211762
"allDeclaredClasses": true,
1176311763
"allPublicClasses": true
1176411764
},
11765+
{
11766+
"name": "com.google.cloud.aiplatform.v1beta1.TrialContext",
11767+
"queryAllDeclaredConstructors": true,
11768+
"queryAllPublicConstructors": true,
11769+
"queryAllDeclaredMethods": true,
11770+
"allPublicMethods": true,
11771+
"allDeclaredClasses": true,
11772+
"allPublicClasses": true
11773+
},
11774+
{
11775+
"name": "com.google.cloud.aiplatform.v1beta1.TrialContext$Builder",
11776+
"queryAllDeclaredConstructors": true,
11777+
"queryAllPublicConstructors": true,
11778+
"queryAllDeclaredMethods": true,
11779+
"allPublicMethods": true,
11780+
"allDeclaredClasses": true,
11781+
"allPublicClasses": true
11782+
},
1176511783
{
1176611784
"name": "com.google.cloud.aiplatform.v1beta1.UndeployIndexOperationMetadata",
1176711785
"queryAllDeclaredConstructors": true,

‎java-aiplatform/google-cloud-aiplatform/src/test/java/com/google/cloud/aiplatform/v1/IndexEndpointServiceClientTest.java‎

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -123,6 +123,7 @@ public void createIndexEndpointTest() throws Exception {
123123
.setPrivateServiceConnectConfig(PrivateServiceConnectConfig.newBuilder().build())
124124
.setPublicEndpointEnabled(true)
125125
.setPublicEndpointDomainName("publicEndpointDomainName1244171053")
126+
.setEncryptionSpec(EncryptionSpec.newBuilder().build())
126127
.build();
127128
Operation resultOperation =
128129
Operation.newBuilder()
@@ -184,6 +185,7 @@ public void createIndexEndpointTest2() throws Exception {
184185
.setPrivateServiceConnectConfig(PrivateServiceConnectConfig.newBuilder().build())
185186
.setPublicEndpointEnabled(true)
186187
.setPublicEndpointDomainName("publicEndpointDomainName1244171053")
188+
.setEncryptionSpec(EncryptionSpec.newBuilder().build())
187189
.build();
188190
Operation resultOperation =
189191
Operation.newBuilder()
@@ -245,6 +247,7 @@ public void getIndexEndpointTest() throws Exception {
245247
.setPrivateServiceConnectConfig(PrivateServiceConnectConfig.newBuilder().build())
246248
.setPublicEndpointEnabled(true)
247249
.setPublicEndpointDomainName("publicEndpointDomainName1244171053")
250+
.setEncryptionSpec(EncryptionSpec.newBuilder().build())
248251
.build();
249252
mockIndexEndpointService.addResponse(expectedResponse);
250253

@@ -295,6 +298,7 @@ public void getIndexEndpointTest2() throws Exception {
295298
.setPrivateServiceConnectConfig(PrivateServiceConnectConfig.newBuilder().build())
296299
.setPublicEndpointEnabled(true)
297300
.setPublicEndpointDomainName("publicEndpointDomainName1244171053")
301+
.setEncryptionSpec(EncryptionSpec.newBuilder().build())
298302
.build();
299303
mockIndexEndpointService.addResponse(expectedResponse);
300304

@@ -433,6 +437,7 @@ public void updateIndexEndpointTest() throws Exception {
433437
.setPrivateServiceConnectConfig(PrivateServiceConnectConfig.newBuilder().build())
434438
.setPublicEndpointEnabled(true)
435439
.setPublicEndpointDomainName("publicEndpointDomainName1244171053")
440+
.setEncryptionSpec(EncryptionSpec.newBuilder().build())
436441
.build();
437442
mockIndexEndpointService.addResponse(expectedResponse);
438443

‎java-aiplatform/google-cloud-aiplatform/src/test/java/com/google/cloud/aiplatform/v1/IndexServiceClientTest.java‎

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -122,6 +122,7 @@ public void createIndexTest() throws Exception {
122122
.setCreateTime(Timestamp.newBuilder().build())
123123
.setUpdateTime(Timestamp.newBuilder().build())
124124
.setIndexStats(IndexStats.newBuilder().build())
125+
.setEncryptionSpec(EncryptionSpec.newBuilder().build())
125126
.build();
126127
Operation resultOperation =
127128
Operation.newBuilder()
@@ -181,6 +182,7 @@ public void createIndexTest2() throws Exception {
181182
.setCreateTime(Timestamp.newBuilder().build())
182183
.setUpdateTime(Timestamp.newBuilder().build())
183184
.setIndexStats(IndexStats.newBuilder().build())
185+
.setEncryptionSpec(EncryptionSpec.newBuilder().build())
184186
.build();
185187
Operation resultOperation =
186188
Operation.newBuilder()
@@ -240,6 +242,7 @@ public void getIndexTest() throws Exception {
240242
.setCreateTime(Timestamp.newBuilder().build())
241243
.setUpdateTime(Timestamp.newBuilder().build())
242244
.setIndexStats(IndexStats.newBuilder().build())
245+
.setEncryptionSpec(EncryptionSpec.newBuilder().build())
243246
.build();
244247
mockIndexService.addResponse(expectedResponse);
245248

@@ -288,6 +291,7 @@ public void getIndexTest2() throws Exception {
288291
.setCreateTime(Timestamp.newBuilder().build())
289292
.setUpdateTime(Timestamp.newBuilder().build())
290293
.setIndexStats(IndexStats.newBuilder().build())
294+
.setEncryptionSpec(EncryptionSpec.newBuilder().build())
291295
.build();
292296
mockIndexService.addResponse(expectedResponse);
293297

@@ -424,6 +428,7 @@ public void updateIndexTest() throws Exception {
424428
.setCreateTime(Timestamp.newBuilder().build())
425429
.setUpdateTime(Timestamp.newBuilder().build())
426430
.setIndexStats(IndexStats.newBuilder().build())
431+
.setEncryptionSpec(EncryptionSpec.newBuilder().build())
427432
.build();
428433
Operation resultOperation =
429434
Operation.newBuilder()

‎java-aiplatform/google-cloud-aiplatform/src/test/java/com/google/cloud/aiplatform/v1/VizierServiceClientTest.java‎

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -542,6 +542,7 @@ public void suggestTrialsTest() throws Exception {
542542
.setParent(StudyName.of("[PROJECT]", "[LOCATION]", "[STUDY]").toString())
543543
.setSuggestionCount(390130452)
544544
.setClientId("clientId908408390")
545+
.addAllContexts(new ArrayList<TrialContext>())
545546
.build();
546547

547548
SuggestTrialsResponse actualResponse = client.suggestTrialsAsync(request).get();
@@ -554,6 +555,7 @@ public void suggestTrialsTest() throws Exception {
554555
Assert.assertEquals(request.getParent(), actualRequest.getParent());
555556
Assert.assertEquals(request.getSuggestionCount(), actualRequest.getSuggestionCount());
556557
Assert.assertEquals(request.getClientId(), actualRequest.getClientId());
558+
Assert.assertEquals(request.getContextsList(), actualRequest.getContextsList());
557559
Assert.assertTrue(
558560
channelProvider.isHeaderSent(
559561
ApiClientHeaderProvider.getDefaultApiClientHeaderKey(),
@@ -571,6 +573,7 @@ public void suggestTrialsExceptionTest() throws Exception {
571573
.setParent(StudyName.of("[PROJECT]", "[LOCATION]", "[STUDY]").toString())
572574
.setSuggestionCount(390130452)
573575
.setClientId("clientId908408390")
576+
.addAllContexts(new ArrayList<TrialContext>())
574577
.build();
575578
client.suggestTrialsAsync(request).get();
576579
Assert.fail("No exception raised");

‎java-aiplatform/google-cloud-aiplatform/src/test/java/com/google/cloud/aiplatform/v1beta1/IndexEndpointServiceClientTest.java‎

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -123,6 +123,7 @@ public void createIndexEndpointTest() throws Exception {
123123
.setPrivateServiceConnectConfig(PrivateServiceConnectConfig.newBuilder().build())
124124
.setPublicEndpointEnabled(true)
125125
.setPublicEndpointDomainName("publicEndpointDomainName1244171053")
126+
.setEncryptionSpec(EncryptionSpec.newBuilder().build())
126127
.build();
127128
Operation resultOperation =
128129
Operation.newBuilder()
@@ -184,6 +185,7 @@ public void createIndexEndpointTest2() throws Exception {
184185
.setPrivateServiceConnectConfig(PrivateServiceConnectConfig.newBuilder().build())
185186
.setPublicEndpointEnabled(true)
186187
.setPublicEndpointDomainName("publicEndpointDomainName1244171053")
188+
.setEncryptionSpec(EncryptionSpec.newBuilder().build())
187189
.build();
188190
Operation resultOperation =
189191
Operation.newBuilder()
@@ -245,6 +247,7 @@ public void getIndexEndpointTest() throws Exception {
245247
.setPrivateServiceConnectConfig(PrivateServiceConnectConfig.newBuilder().build())
246248
.setPublicEndpointEnabled(true)
247249
.setPublicEndpointDomainName("publicEndpointDomainName1244171053")
250+
.setEncryptionSpec(EncryptionSpec.newBuilder().build())
248251
.build();
249252
mockIndexEndpointService.addResponse(expectedResponse);
250253

@@ -295,6 +298,7 @@ public void getIndexEndpointTest2() throws Exception {
295298
.setPrivateServiceConnectConfig(PrivateServiceConnectConfig.newBuilder().build())
296299
.setPublicEndpointEnabled(true)
297300
.setPublicEndpointDomainName("publicEndpointDomainName1244171053")
301+
.setEncryptionSpec(EncryptionSpec.newBuilder().build())
298302
.build();
299303
mockIndexEndpointService.addResponse(expectedResponse);
300304

@@ -433,6 +437,7 @@ public void updateIndexEndpointTest() throws Exception {
433437
.setPrivateServiceConnectConfig(PrivateServiceConnectConfig.newBuilder().build())
434438
.setPublicEndpointEnabled(true)
435439
.setPublicEndpointDomainName("publicEndpointDomainName1244171053")
440+
.setEncryptionSpec(EncryptionSpec.newBuilder().build())
436441
.build();
437442
mockIndexEndpointService.addResponse(expectedResponse);
438443

‎java-aiplatform/google-cloud-aiplatform/src/test/java/com/google/cloud/aiplatform/v1beta1/IndexServiceClientTest.java‎

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -122,6 +122,7 @@ public void createIndexTest() throws Exception {
122122
.setCreateTime(Timestamp.newBuilder().build())
123123
.setUpdateTime(Timestamp.newBuilder().build())
124124
.setIndexStats(IndexStats.newBuilder().build())
125+
.setEncryptionSpec(EncryptionSpec.newBuilder().build())
125126
.build();
126127
Operation resultOperation =
127128
Operation.newBuilder()
@@ -181,6 +182,7 @@ public void createIndexTest2() throws Exception {
181182
.setCreateTime(Timestamp.newBuilder().build())
182183
.setUpdateTime(Timestamp.newBuilder().build())
183184
.setIndexStats(IndexStats.newBuilder().build())
185+
.setEncryptionSpec(EncryptionSpec.newBuilder().build())
184186
.build();
185187
Operation resultOperation =
186188
Operation.newBuilder()
@@ -240,6 +242,7 @@ public void getIndexTest() throws Exception {
240242
.setCreateTime(Timestamp.newBuilder().build())
241243
.setUpdateTime(Timestamp.newBuilder().build())
242244
.setIndexStats(IndexStats.newBuilder().build())
245+
.setEncryptionSpec(EncryptionSpec.newBuilder().build())
243246
.build();
244247
mockIndexService.addResponse(expectedResponse);
245248

@@ -288,6 +291,7 @@ public void getIndexTest2() throws Exception {
288291
.setCreateTime(Timestamp.newBuilder().build())
289292
.setUpdateTime(Timestamp.newBuilder().build())
290293
.setIndexStats(IndexStats.newBuilder().build())
294+
.setEncryptionSpec(EncryptionSpec.newBuilder().build())
291295
.build();
292296
mockIndexService.addResponse(expectedResponse);
293297

@@ -424,6 +428,7 @@ public void updateIndexTest() throws Exception {
424428
.setCreateTime(Timestamp.newBuilder().build())
425429
.setUpdateTime(Timestamp.newBuilder().build())
426430
.setIndexStats(IndexStats.newBuilder().build())
431+
.setEncryptionSpec(EncryptionSpec.newBuilder().build())
427432
.build();
428433
Operation resultOperation =
429434
Operation.newBuilder()

‎java-aiplatform/google-cloud-aiplatform/src/test/java/com/google/cloud/aiplatform/v1beta1/VizierServiceClientTest.java‎

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -542,6 +542,7 @@ public void suggestTrialsTest() throws Exception {
542542
.setParent(StudyName.of("[PROJECT]", "[LOCATION]", "[STUDY]").toString())
543543
.setSuggestionCount(390130452)
544544
.setClientId("clientId908408390")
545+
.addAllContexts(new ArrayList<TrialContext>())
545546
.build();
546547

547548
SuggestTrialsResponse actualResponse = client.suggestTrialsAsync(request).get();
@@ -554,6 +555,7 @@ public void suggestTrialsTest() throws Exception {
554555
Assert.assertEquals(request.getParent(), actualRequest.getParent());
555556
Assert.assertEquals(request.getSuggestionCount(), actualRequest.getSuggestionCount());
556557
Assert.assertEquals(request.getClientId(), actualRequest.getClientId());
558+
Assert.assertEquals(request.getContextsList(), actualRequest.getContextsList());
557559
Assert.assertTrue(
558560
channelProvider.isHeaderSent(
559561
ApiClientHeaderProvider.getDefaultApiClientHeaderKey(),
@@ -571,6 +573,7 @@ public void suggestTrialsExceptionTest() throws Exception {
571573
.setParent(StudyName.of("[PROJECT]", "[LOCATION]", "[STUDY]").toString())
572574
.setSuggestionCount(390130452)
573575
.setClientId("clientId908408390")
576+
.addAllContexts(new ArrayList<TrialContext>())
574577
.build();
575578
client.suggestTrialsAsync(request).get();
576579
Assert.fail("No exception raised");

0 commit comments

Comments
 (0)