Skip to content

Commit 267827e

Browse files
gcf-owl-bot[bot]gcf-owl-bot[bot]
authored
feat: add WriteFeatureValues in aiplatform v1beta1 featurestore_online_service.proto (#1022)
- [ ] Regenerate this pull request now. PiperOrigin-RevId: 467107520 Source-Link: googleapis/googleapis@0dd9815 Source-Link: https://github.com/googleapis/googleapis-gen/commit/03d96fd2965824e0fe50b136c1babe0e5c3c5f57 Copy-Tag: eyJwIjoiLmdpdGh1Yi8uT3dsQm90LnlhbWwiLCJoIjoiMDNkOTZmZDI5NjU4MjRlMGZlNTBiMTM2YzFiYWJlMGU1YzNjNWY1NyJ9
1 parent 527ca47 commit 267827e

File tree

18 files changed

+3752
-76
lines changed

18 files changed

+3752
-76
lines changed

‎java-aiplatform/README.md‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -45,13 +45,13 @@ implementation 'com.google.cloud:google-cloud-aiplatform'
4545
If you are using Gradle without BOM, add this to your dependencies:
4646

4747
```Groovy
48-
implementation 'com.google.cloud:google-cloud-aiplatform:3.1.0'
48+
implementation 'com.google.cloud:google-cloud-aiplatform:3.2.0'
4949
```
5050

5151
If you are using SBT, add this to your dependencies:
5252

5353
```Scala
54-
libraryDependencies += "com.google.cloud" % "google-cloud-aiplatform" % "3.1.0"
54+
libraryDependencies += "com.google.cloud" % "google-cloud-aiplatform" % "3.2.0"
5555
```
5656

5757
## Authentication

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

Lines changed: 147 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -335,6 +335,153 @@ public final ReadFeatureValuesResponse readFeatureValues(ReadFeatureValuesReques
335335
return stub.streamingReadFeatureValuesCallable();
336336
}
337337

338+
// AUTO-GENERATED DOCUMENTATION AND METHOD.
339+
/**
340+
* Writes Feature values of one or more entities of an EntityType.
341+
*
342+
* <p>The Feature values are merged into existing entities if any. The Feature values to be
343+
* written must have timestamp within the online storage retention.
344+
*
345+
* <p>Sample code:
346+
*
347+
* <pre>{@code
348+
* // This snippet has been automatically generated for illustrative purposes only.
349+
* // It may require modifications to work in your environment.
350+
* try (FeaturestoreOnlineServingServiceClient featurestoreOnlineServingServiceClient =
351+
* FeaturestoreOnlineServingServiceClient.create()) {
352+
* EntityTypeName entityType =
353+
* EntityTypeName.of("[PROJECT]", "[LOCATION]", "[FEATURESTORE]", "[ENTITY_TYPE]");
354+
* List<WriteFeatureValuesPayload> payloads = new ArrayList<>();
355+
* WriteFeatureValuesResponse response =
356+
* featurestoreOnlineServingServiceClient.writeFeatureValues(entityType, payloads);
357+
* }
358+
* }</pre>
359+
*
360+
* @param entityType Required. The resource name of the EntityType for the entities being written.
361+
* Value format: `projects/{project}/locations/{location}/featurestores/
362+
* {featurestore}/entityTypes/{entityType}`. For example, for a machine learning model
363+
* predicting user clicks on a website, an EntityType ID could be `user`.
364+
* @param payloads Required. The entities to be written. Up to 100,000 feature values can be
365+
* written across all `payloads`.
366+
* @throws com.google.api.gax.rpc.ApiException if the remote call fails
367+
*/
368+
public final WriteFeatureValuesResponse writeFeatureValues(
369+
EntityTypeName entityType, List<WriteFeatureValuesPayload> payloads) {
370+
WriteFeatureValuesRequest request =
371+
WriteFeatureValuesRequest.newBuilder()
372+
.setEntityType(entityType == null ? null : entityType.toString())
373+
.addAllPayloads(payloads)
374+
.build();
375+
return writeFeatureValues(request);
376+
}
377+
378+
// AUTO-GENERATED DOCUMENTATION AND METHOD.
379+
/**
380+
* Writes Feature values of one or more entities of an EntityType.
381+
*
382+
* <p>The Feature values are merged into existing entities if any. The Feature values to be
383+
* written must have timestamp within the online storage retention.
384+
*
385+
* <p>Sample code:
386+
*
387+
* <pre>{@code
388+
* // This snippet has been automatically generated for illustrative purposes only.
389+
* // It may require modifications to work in your environment.
390+
* try (FeaturestoreOnlineServingServiceClient featurestoreOnlineServingServiceClient =
391+
* FeaturestoreOnlineServingServiceClient.create()) {
392+
* String entityType =
393+
* EntityTypeName.of("[PROJECT]", "[LOCATION]", "[FEATURESTORE]", "[ENTITY_TYPE]")
394+
* .toString();
395+
* List<WriteFeatureValuesPayload> payloads = new ArrayList<>();
396+
* WriteFeatureValuesResponse response =
397+
* featurestoreOnlineServingServiceClient.writeFeatureValues(entityType, payloads);
398+
* }
399+
* }</pre>
400+
*
401+
* @param entityType Required. The resource name of the EntityType for the entities being written.
402+
* Value format: `projects/{project}/locations/{location}/featurestores/
403+
* {featurestore}/entityTypes/{entityType}`. For example, for a machine learning model
404+
* predicting user clicks on a website, an EntityType ID could be `user`.
405+
* @param payloads Required. The entities to be written. Up to 100,000 feature values can be
406+
* written across all `payloads`.
407+
* @throws com.google.api.gax.rpc.ApiException if the remote call fails
408+
*/
409+
public final WriteFeatureValuesResponse writeFeatureValues(
410+
String entityType, List<WriteFeatureValuesPayload> payloads) {
411+
WriteFeatureValuesRequest request =
412+
WriteFeatureValuesRequest.newBuilder()
413+
.setEntityType(entityType)
414+
.addAllPayloads(payloads)
415+
.build();
416+
return writeFeatureValues(request);
417+
}
418+
419+
// AUTO-GENERATED DOCUMENTATION AND METHOD.
420+
/**
421+
* Writes Feature values of one or more entities of an EntityType.
422+
*
423+
* <p>The Feature values are merged into existing entities if any. The Feature values to be
424+
* written must have timestamp within the online storage retention.
425+
*
426+
* <p>Sample code:
427+
*
428+
* <pre>{@code
429+
* // This snippet has been automatically generated for illustrative purposes only.
430+
* // It may require modifications to work in your environment.
431+
* try (FeaturestoreOnlineServingServiceClient featurestoreOnlineServingServiceClient =
432+
* FeaturestoreOnlineServingServiceClient.create()) {
433+
* WriteFeatureValuesRequest request =
434+
* WriteFeatureValuesRequest.newBuilder()
435+
* .setEntityType(
436+
* EntityTypeName.of("[PROJECT]", "[LOCATION]", "[FEATURESTORE]", "[ENTITY_TYPE]")
437+
* .toString())
438+
* .addAllPayloads(new ArrayList<WriteFeatureValuesPayload>())
439+
* .build();
440+
* WriteFeatureValuesResponse response =
441+
* featurestoreOnlineServingServiceClient.writeFeatureValues(request);
442+
* }
443+
* }</pre>
444+
*
445+
* @param request The request object containing all of the parameters for the API call.
446+
* @throws com.google.api.gax.rpc.ApiException if the remote call fails
447+
*/
448+
public final WriteFeatureValuesResponse writeFeatureValues(WriteFeatureValuesRequest request) {
449+
return writeFeatureValuesCallable().call(request);
450+
}
451+
452+
// AUTO-GENERATED DOCUMENTATION AND METHOD.
453+
/**
454+
* Writes Feature values of one or more entities of an EntityType.
455+
*
456+
* <p>The Feature values are merged into existing entities if any. The Feature values to be
457+
* written must have timestamp within the online storage retention.
458+
*
459+
* <p>Sample code:
460+
*
461+
* <pre>{@code
462+
* // This snippet has been automatically generated for illustrative purposes only.
463+
* // It may require modifications to work in your environment.
464+
* try (FeaturestoreOnlineServingServiceClient featurestoreOnlineServingServiceClient =
465+
* FeaturestoreOnlineServingServiceClient.create()) {
466+
* WriteFeatureValuesRequest request =
467+
* WriteFeatureValuesRequest.newBuilder()
468+
* .setEntityType(
469+
* EntityTypeName.of("[PROJECT]", "[LOCATION]", "[FEATURESTORE]", "[ENTITY_TYPE]")
470+
* .toString())
471+
* .addAllPayloads(new ArrayList<WriteFeatureValuesPayload>())
472+
* .build();
473+
* ApiFuture<WriteFeatureValuesResponse> future =
474+
* featurestoreOnlineServingServiceClient.writeFeatureValuesCallable().futureCall(request);
475+
* // Do something.
476+
* WriteFeatureValuesResponse response = future.get();
477+
* }
478+
* }</pre>
479+
*/
480+
public final UnaryCallable<WriteFeatureValuesRequest, WriteFeatureValuesResponse>
481+
writeFeatureValuesCallable() {
482+
return stub.writeFeatureValuesCallable();
483+
}
484+
338485
// AUTO-GENERATED DOCUMENTATION AND METHOD.
339486
/**
340487
* Lists information about the supported locations for this service.

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

Lines changed: 13 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -99,6 +99,13 @@ public class FeaturestoreOnlineServingServiceSettings
9999
.streamingReadFeatureValuesSettings();
100100
}
101101

102+
/** Returns the object with the settings used for calls to writeFeatureValues. */
103+
public UnaryCallSettings<WriteFeatureValuesRequest, WriteFeatureValuesResponse>
104+
writeFeatureValuesSettings() {
105+
return ((FeaturestoreOnlineServingServiceStubSettings) getStubSettings())
106+
.writeFeatureValuesSettings();
107+
}
108+
102109
/** Returns the object with the settings used for calls to listLocations. */
103110
public PagedCallSettings<ListLocationsRequest, ListLocationsResponse, ListLocationsPagedResponse>
104111
listLocationsSettings() {
@@ -241,6 +248,12 @@ public Builder applyToAllUnaryMethods(
241248
return getStubSettingsBuilder().streamingReadFeatureValuesSettings();
242249
}
243250

251+
/** Returns the builder for the settings used for calls to writeFeatureValues. */
252+
public UnaryCallSettings.Builder<WriteFeatureValuesRequest, WriteFeatureValuesResponse>
253+
writeFeatureValuesSettings() {
254+
return getStubSettingsBuilder().writeFeatureValuesSettings();
255+
}
256+
244257
/** Returns the builder for the settings used for calls to listLocations. */
245258
public PagedCallSettings.Builder<
246259
ListLocationsRequest, ListLocationsResponse, ListLocationsPagedResponse>

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

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -171,6 +171,9 @@
171171
},
172172
"TestIamPermissions": {
173173
"methods": ["testIamPermissions", "testIamPermissionsCallable"]
174+
},
175+
"WriteFeatureValues": {
176+
"methods": ["writeFeatureValues", "writeFeatureValues", "writeFeatureValues", "writeFeatureValuesCallable"]
174177
}
175178
}
176179
}

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

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -25,6 +25,8 @@
2525
import com.google.cloud.aiplatform.v1beta1.ReadFeatureValuesRequest;
2626
import com.google.cloud.aiplatform.v1beta1.ReadFeatureValuesResponse;
2727
import com.google.cloud.aiplatform.v1beta1.StreamingReadFeatureValuesRequest;
28+
import com.google.cloud.aiplatform.v1beta1.WriteFeatureValuesRequest;
29+
import com.google.cloud.aiplatform.v1beta1.WriteFeatureValuesResponse;
2830
import com.google.cloud.location.GetLocationRequest;
2931
import com.google.cloud.location.ListLocationsRequest;
3032
import com.google.cloud.location.ListLocationsResponse;
@@ -57,6 +59,11 @@ public abstract class FeaturestoreOnlineServingServiceStub implements Background
5759
"Not implemented: streamingReadFeatureValuesCallable()");
5860
}
5961

62+
public UnaryCallable<WriteFeatureValuesRequest, WriteFeatureValuesResponse>
63+
writeFeatureValuesCallable() {
64+
throw new UnsupportedOperationException("Not implemented: writeFeatureValuesCallable()");
65+
}
66+
6067
public UnaryCallable<ListLocationsRequest, ListLocationsPagedResponse>
6168
listLocationsPagedCallable() {
6269
throw new UnsupportedOperationException("Not implemented: listLocationsPagedCallable()");

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

Lines changed: 28 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -44,6 +44,8 @@
4444
import com.google.cloud.aiplatform.v1beta1.ReadFeatureValuesRequest;
4545
import com.google.cloud.aiplatform.v1beta1.ReadFeatureValuesResponse;
4646
import com.google.cloud.aiplatform.v1beta1.StreamingReadFeatureValuesRequest;
47+
import com.google.cloud.aiplatform.v1beta1.WriteFeatureValuesRequest;
48+
import com.google.cloud.aiplatform.v1beta1.WriteFeatureValuesResponse;
4749
import com.google.cloud.location.GetLocationRequest;
4850
import com.google.cloud.location.ListLocationsRequest;
4951
import com.google.cloud.location.ListLocationsResponse;
@@ -111,6 +113,8 @@ public class FeaturestoreOnlineServingServiceStubSettings
111113
private final ServerStreamingCallSettings<
112114
StreamingReadFeatureValuesRequest, ReadFeatureValuesResponse>
113115
streamingReadFeatureValuesSettings;
116+
private final UnaryCallSettings<WriteFeatureValuesRequest, WriteFeatureValuesResponse>
117+
writeFeatureValuesSettings;
114118
private final PagedCallSettings<
115119
ListLocationsRequest, ListLocationsResponse, ListLocationsPagedResponse>
116120
listLocationsSettings;
@@ -185,6 +189,12 @@ public ApiFuture<ListLocationsPagedResponse> getFuturePagedResponse(
185189
return streamingReadFeatureValuesSettings;
186190
}
187191

192+
/** Returns the object with the settings used for calls to writeFeatureValues. */
193+
public UnaryCallSettings<WriteFeatureValuesRequest, WriteFeatureValuesResponse>
194+
writeFeatureValuesSettings() {
195+
return writeFeatureValuesSettings;
196+
}
197+
188198
/** Returns the object with the settings used for calls to listLocations. */
189199
public PagedCallSettings<ListLocationsRequest, ListLocationsResponse, ListLocationsPagedResponse>
190200
listLocationsSettings() {
@@ -292,6 +302,7 @@ protected FeaturestoreOnlineServingServiceStubSettings(Builder settingsBuilder)
292302
readFeatureValuesSettings = settingsBuilder.readFeatureValuesSettings().build();
293303
streamingReadFeatureValuesSettings =
294304
settingsBuilder.streamingReadFeatureValuesSettings().build();
305+
writeFeatureValuesSettings = settingsBuilder.writeFeatureValuesSettings().build();
295306
listLocationsSettings = settingsBuilder.listLocationsSettings().build();
296307
getLocationSettings = settingsBuilder.getLocationSettings().build();
297308
setIamPolicySettings = settingsBuilder.setIamPolicySettings().build();
@@ -308,6 +319,8 @@ public static class Builder
308319
private final ServerStreamingCallSettings.Builder<
309320
StreamingReadFeatureValuesRequest, ReadFeatureValuesResponse>
310321
streamingReadFeatureValuesSettings;
322+
private final UnaryCallSettings.Builder<WriteFeatureValuesRequest, WriteFeatureValuesResponse>
323+
writeFeatureValuesSettings;
311324
private final PagedCallSettings.Builder<
312325
ListLocationsRequest, ListLocationsResponse, ListLocationsPagedResponse>
313326
listLocationsSettings;
@@ -355,6 +368,7 @@ protected Builder(ClientContext clientContext) {
355368

356369
readFeatureValuesSettings = UnaryCallSettings.newUnaryCallSettingsBuilder();
357370
streamingReadFeatureValuesSettings = ServerStreamingCallSettings.newBuilder();
371+
writeFeatureValuesSettings = UnaryCallSettings.newUnaryCallSettingsBuilder();
358372
listLocationsSettings = PagedCallSettings.newBuilder(LIST_LOCATIONS_PAGE_STR_FACT);
359373
getLocationSettings = UnaryCallSettings.newUnaryCallSettingsBuilder();
360374
setIamPolicySettings = UnaryCallSettings.newUnaryCallSettingsBuilder();
@@ -364,6 +378,7 @@ protected Builder(ClientContext clientContext) {
364378
unaryMethodSettingsBuilders =
365379
ImmutableList.<UnaryCallSettings.Builder<?, ?>>of(
366380
readFeatureValuesSettings,
381+
writeFeatureValuesSettings,
367382
listLocationsSettings,
368383
getLocationSettings,
369384
setIamPolicySettings,
@@ -377,6 +392,7 @@ protected Builder(FeaturestoreOnlineServingServiceStubSettings settings) {
377392

378393
readFeatureValuesSettings = settings.readFeatureValuesSettings.toBuilder();
379394
streamingReadFeatureValuesSettings = settings.streamingReadFeatureValuesSettings.toBuilder();
395+
writeFeatureValuesSettings = settings.writeFeatureValuesSettings.toBuilder();
380396
listLocationsSettings = settings.listLocationsSettings.toBuilder();
381397
getLocationSettings = settings.getLocationSettings.toBuilder();
382398
setIamPolicySettings = settings.setIamPolicySettings.toBuilder();
@@ -386,6 +402,7 @@ protected Builder(FeaturestoreOnlineServingServiceStubSettings settings) {
386402
unaryMethodSettingsBuilders =
387403
ImmutableList.<UnaryCallSettings.Builder<?, ?>>of(
388404
readFeatureValuesSettings,
405+
writeFeatureValuesSettings,
389406
listLocationsSettings,
390407
getLocationSettings,
391408
setIamPolicySettings,
@@ -417,6 +434,11 @@ private static Builder initDefaults(Builder builder) {
417434
.setRetryableCodes(RETRYABLE_CODE_DEFINITIONS.get("no_retry_13_codes"))
418435
.setRetrySettings(RETRY_PARAM_DEFINITIONS.get("no_retry_13_params"));
419436

437+
builder
438+
.writeFeatureValuesSettings()
439+
.setRetryableCodes(RETRYABLE_CODE_DEFINITIONS.get("no_retry_codes"))
440+
.setRetrySettings(RETRY_PARAM_DEFINITIONS.get("no_retry_params"));
441+
420442
builder
421443
.listLocationsSettings()
422444
.setRetryableCodes(RETRYABLE_CODE_DEFINITIONS.get("no_retry_codes"))
@@ -473,6 +495,12 @@ public Builder applyToAllUnaryMethods(
473495
return streamingReadFeatureValuesSettings;
474496
}
475497

498+
/** Returns the builder for the settings used for calls to writeFeatureValues. */
499+
public UnaryCallSettings.Builder<WriteFeatureValuesRequest, WriteFeatureValuesResponse>
500+
writeFeatureValuesSettings() {
501+
return writeFeatureValuesSettings;
502+
}
503+
476504
/** Returns the builder for the settings used for calls to listLocations. */
477505
public PagedCallSettings.Builder<
478506
ListLocationsRequest, ListLocationsResponse, ListLocationsPagedResponse>

0 commit comments

Comments
 (0)