Skip to content

Commit c2a9b99

Browse files
Google APIscopybara-github
authored andcommitted
feat: add ValueType.STRUCT to Feature
feat: add encryption_spec to FeatureOnlineStore feat: add struct_value to FeatureValue feat: add encryption_spec, service_account, disable_container_logging to DeploymentResourcePool feat: add model_reference to Dataset feat: add tool_config to GenerateContentRequest feat: add dataplex_config to MetadataStore feat: add model_reference to DatasetVersion feat: add INVALID_SPARSE_DIMENSIONS, INVALID_SPARSE_EMBEDDING, INVALID_EMBEDDING to NearestNeighborSearchOperationMetadata.RecordError feat: add valid_sparse_record_count, invalid_sparse_record_count to NearestNeighborSearchOperationMetadata.ContentValidationStats feat: add sparse_embedding to IndexDatapoint feat: add sparse_vectors_count to IndexStats feat: add sparse_distance to FindNeighborsResponse.Neighbor feat: add more fields in FindNeighborsRequest.Query feat: add UpdateNotebookRuntimeTemplate to NotebookService feat: add direct_notebook_source to NotebookExecutionJob feat: add encryption_spec to NotebookRuntimeTemplate feat: add idle_shutdown_config, encryption_spec, satisfies_pzs, satisfies_pzi to NotebookRuntime feat: add RaySpec to PersistentResource fix!: An existing message `Segment` is removed fix!: An existing message `GroundingAttribution` is removed fix!: An existing field `grounding_attributions` is removed from message `.google.cloud.aiplatform.v1beta1.GroundingMetadata` fix!: An existing field `disable_attribution` is removed from message `.google.cloud.aiplatform.v1beta1.GoogleSearchRetrieval` docs: A comment for field `feature_vector` in message `.google.cloud.aiplatform.v1beta1.IndexDatapoint` is changed docs: A comment for field `vectors_count` in message `.google.cloud.aiplatform.v1beta1.IndexStats` is changed docs: A comment for enum value `EMBEDDING_SIZE_MISMATCH` in enum `RecordErrorType` is changed docs: A comment for field `exec` in message `.google.cloud.aiplatform.v1beta1.Probe` is changed PiperOrigin-RevId: 638338486
1 parent 1e41378 commit c2a9b99

20 files changed

Lines changed: 339 additions & 66 deletions

google/cloud/aiplatform/v1/aiplatform_v1.yaml

Lines changed: 12 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -229,6 +229,8 @@ http:
229229
- selector: google.longrunning.Operations.CancelOperation
230230
post: '/ui/{name=projects/*/locations/*/operations/*}:cancel'
231231
additional_bindings:
232+
- post: '/ui/{name=projects/*/locations/*/agents/*/operations/*}:cancel'
233+
- post: '/ui/{name=projects/*/locations/*/apps/*/operations/*}:cancel'
232234
- post: '/ui/{name=projects/*/locations/*/datasets/*/operations/*}:cancel'
233235
- post: '/ui/{name=projects/*/locations/*/datasets/*/dataItems/*/operations/*}:cancel'
234236
- post: '/ui/{name=projects/*/locations/*/datasets/*/savedQueries/*/operations/*}:cancel'
@@ -307,6 +309,8 @@ http:
307309
- selector: google.longrunning.Operations.DeleteOperation
308310
delete: '/ui/{name=projects/*/locations/*/operations/*}'
309311
additional_bindings:
312+
- delete: '/ui/{name=projects/*/locations/*/agents/*/operations/*}'
313+
- delete: '/ui/{name=projects/*/locations/*/apps/*/operations/*}'
310314
- delete: '/ui/{name=projects/*/locations/*/datasets/*/operations/*}'
311315
- delete: '/ui/{name=projects/*/locations/*/datasets/*/dataItems/*/operations/*}'
312316
- delete: '/ui/{name=projects/*/locations/*/datasets/*/savedQueries/*/operations/*}'
@@ -391,6 +395,8 @@ http:
391395
- selector: google.longrunning.Operations.GetOperation
392396
get: '/ui/{name=projects/*/locations/*/operations/*}'
393397
additional_bindings:
398+
- get: '/ui/{name=projects/*/locations/*/agents/*/operations/*}'
399+
- get: '/ui/{name=projects/*/locations/*/apps/*/operations/*}'
394400
- get: '/ui/{name=projects/*/locations/*/datasets/*/operations/*}'
395401
- get: '/ui/{name=projects/*/locations/*/datasets/*/dataItems/*/operations/*}'
396402
- get: '/ui/{name=projects/*/locations/*/datasets/*/savedQueries/*/operations/*}'
@@ -478,6 +484,8 @@ http:
478484
- selector: google.longrunning.Operations.ListOperations
479485
get: '/ui/{name=projects/*/locations/*}/operations'
480486
additional_bindings:
487+
- get: '/ui/{name=projects/*/locations/*/agents/*}/operations'
488+
- get: '/ui/{name=projects/*/locations/*/apps/*}/operations'
481489
- get: '/ui/{name=projects/*/locations/*/datasets/*}/operations'
482490
- get: '/ui/{name=projects/*/locations/*/datasets/*/dataItems/*}/operations'
483491
- get: '/ui/{name=projects/*/locations/*/datasets/*/savedQueries/*}/operations'
@@ -564,6 +572,8 @@ http:
564572
- selector: google.longrunning.Operations.WaitOperation
565573
post: '/ui/{name=projects/*/locations/*/operations/*}:wait'
566574
additional_bindings:
575+
- post: '/ui/{name=projects/*/locations/*/agents/*/operations/*}:wait'
576+
- post: '/ui/{name=projects/*/locations/*/apps/*/operations/*}:wait'
567577
- post: '/ui/{name=projects/*/locations/*/datasets/*/operations/*}:wait'
568578
- post: '/ui/{name=projects/*/locations/*/datasets/*/dataItems/*/operations/*}:wait'
569579
- post: '/ui/{name=projects/*/locations/*/datasets/*/savedQueries/*/operations/*}:wait'
@@ -757,7 +767,8 @@ authentication:
757767
- selector: 'google.cloud.aiplatform.v1.PredictionService.*'
758768
oauth:
759769
canonical_scopes: |-
760-
https://www.googleapis.com/auth/cloud-platform
770+
https://www.googleapis.com/auth/cloud-platform,
771+
https://www.googleapis.com/auth/cloud-platform.read-only
761772
- selector: 'google.cloud.aiplatform.v1.ScheduleService.*'
762773
oauth:
763774
canonical_scopes: |-

google/cloud/aiplatform/v1/content.proto

Lines changed: 10 additions & 45 deletions
Original file line numberDiff line numberDiff line change
@@ -17,6 +17,7 @@ syntax = "proto3";
1717
package google.cloud.aiplatform.v1;
1818

1919
import "google/api/field_behavior.proto";
20+
import "google/cloud/aiplatform/v1/openapi.proto";
2021
import "google/cloud/aiplatform/v1/tool.proto";
2122
import "google/protobuf/duration.proto";
2223
import "google/type/date.proto";
@@ -168,6 +169,15 @@ message GenerationConfig {
168169
// otherwise the behavior is undefined.
169170
// This is a preview feature.
170171
string response_mime_type = 13 [(google.api.field_behavior) = OPTIONAL];
172+
173+
// Optional. The `Schema` object allows the definition of input and output
174+
// data types. These types can be objects, but also primitives and arrays.
175+
// Represents a select subset of an [OpenAPI 3.0 schema
176+
// object](https://spec.openapis.org/oas/v3.0.3#schema).
177+
// If set, a compatible response_mime_type must also be set.
178+
// Compatible mimetypes:
179+
// `application/json`: Schema for JSON response.
180+
optional Schema response_schema = 16 [(google.api.field_behavior) = OPTIONAL];
171181
}
172182

173183
// Safety settings.
@@ -368,57 +378,12 @@ message Candidate {
368378
[(google.api.field_behavior) = OUTPUT_ONLY];
369379
}
370380

371-
// Segment of the content.
372-
message Segment {
373-
// Output only. The index of a Part object within its parent Content object.
374-
int32 part_index = 1 [(google.api.field_behavior) = OUTPUT_ONLY];
375-
376-
// Output only. Start index in the given Part, measured in bytes. Offset from
377-
// the start of the Part, inclusive, starting at zero.
378-
int32 start_index = 2 [(google.api.field_behavior) = OUTPUT_ONLY];
379-
380-
// Output only. End index in the given Part, measured in bytes. Offset from
381-
// the start of the Part, exclusive, starting at zero.
382-
int32 end_index = 3 [(google.api.field_behavior) = OUTPUT_ONLY];
383-
}
384-
385-
// Grounding attribution.
386-
message GroundingAttribution {
387-
// Attribution from the web.
388-
message Web {
389-
// Output only. URI reference of the attribution.
390-
string uri = 1 [(google.api.field_behavior) = OUTPUT_ONLY];
391-
392-
// Output only. Title of the attribution.
393-
string title = 2 [(google.api.field_behavior) = OUTPUT_ONLY];
394-
}
395-
396-
oneof reference {
397-
// Optional. Attribution from the web.
398-
Web web = 3 [(google.api.field_behavior) = OPTIONAL];
399-
}
400-
401-
// Output only. Segment of the content this attribution belongs to.
402-
Segment segment = 1 [(google.api.field_behavior) = OUTPUT_ONLY];
403-
404-
// Optional. Output only. Confidence score of the attribution. Ranges from 0
405-
// to 1. 1 is the most confident.
406-
optional float confidence_score = 2 [
407-
(google.api.field_behavior) = OPTIONAL,
408-
(google.api.field_behavior) = OUTPUT_ONLY
409-
];
410-
}
411-
412381
// Metadata returned to client when grounding is enabled.
413382
message GroundingMetadata {
414383
// Optional. Web search queries for the following-up web search.
415384
repeated string web_search_queries = 1
416385
[(google.api.field_behavior) = OPTIONAL];
417386

418-
// Optional. List of grounding attributions.
419-
repeated GroundingAttribution grounding_attributions = 2
420-
[(google.api.field_behavior) = OPTIONAL];
421-
422387
// Optional. Google search entry for the following-up web searches.
423388
optional SearchEntryPoint search_entry_point = 4
424389
[(google.api.field_behavior) = OPTIONAL];

google/cloud/aiplatform/v1/dataset.proto

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -111,6 +111,10 @@ message Dataset {
111111
// is
112112
// `projects/{project}/locations/{location}/metadataStores/{metadata_store}/artifacts/{artifact}`.
113113
string metadata_artifact = 17 [(google.api.field_behavior) = OUTPUT_ONLY];
114+
115+
// Optional. Reference to the public base model last used by the dataset. Only
116+
// set for prompt datasets.
117+
string model_reference = 18 [(google.api.field_behavior) = OPTIONAL];
114118
}
115119

116120
// Describes the location from where we import data into a Dataset, together

google/cloud/aiplatform/v1/dataset_service.proto

Lines changed: 25 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -138,6 +138,16 @@ service DatasetService {
138138
};
139139
}
140140

141+
// Updates a DatasetVersion.
142+
rpc UpdateDatasetVersion(UpdateDatasetVersionRequest)
143+
returns (DatasetVersion) {
144+
option (google.api.http) = {
145+
patch: "/v1/{dataset_version.name=projects/*/locations/*/datasets/*/datasetVersions/*}"
146+
body: "dataset_version"
147+
};
148+
option (google.api.method_signature) = "dataset_version,update_mask";
149+
}
150+
141151
// Deletes a Dataset version.
142152
rpc DeleteDatasetVersion(DeleteDatasetVersionRequest)
143153
returns (google.longrunning.Operation) {
@@ -292,6 +302,21 @@ message UpdateDatasetRequest {
292302
[(google.api.field_behavior) = REQUIRED];
293303
}
294304

305+
// Request message for
306+
// [DatasetService.UpdateDatasetVersion][google.cloud.aiplatform.v1.DatasetService.UpdateDatasetVersion].
307+
message UpdateDatasetVersionRequest {
308+
// Required. The DatasetVersion which replaces the resource on the server.
309+
DatasetVersion dataset_version = 1 [(google.api.field_behavior) = REQUIRED];
310+
311+
// Required. The update mask applies to the resource.
312+
// For the `FieldMask` definition, see
313+
// [google.protobuf.FieldMask][google.protobuf.FieldMask]. Updatable fields:
314+
//
315+
// * `display_name`
316+
google.protobuf.FieldMask update_mask = 2
317+
[(google.api.field_behavior) = REQUIRED];
318+
}
319+
295320
// Request message for
296321
// [DatasetService.ListDatasets][google.cloud.aiplatform.v1.DatasetService.ListDatasets].
297322
message ListDatasetsRequest {

google/cloud/aiplatform/v1/dataset_version.proto

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -64,4 +64,8 @@ message DatasetVersion {
6464
(google.api.field_behavior) = OUTPUT_ONLY,
6565
(google.api.field_behavior) = REQUIRED
6666
];
67+
68+
// Output only. Reference to the public base model last used by the dataset
69+
// version. Only set for prompt dataset versions.
70+
string model_reference = 9 [(google.api.field_behavior) = OUTPUT_ONLY];
6771
}

google/cloud/aiplatform/v1/deployment_resource_pool.proto

Lines changed: 24 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -18,6 +18,7 @@ package google.cloud.aiplatform.v1;
1818

1919
import "google/api/field_behavior.proto";
2020
import "google/api/resource.proto";
21+
import "google/cloud/aiplatform/v1/encryption_spec.proto";
2122
import "google/cloud/aiplatform/v1/machine_resources.proto";
2223
import "google/protobuf/timestamp.proto";
2324

@@ -47,6 +48,29 @@ message DeploymentResourcePool {
4748
DedicatedResources dedicated_resources = 2
4849
[(google.api.field_behavior) = REQUIRED];
4950

51+
// Customer-managed encryption key spec for a DeploymentResourcePool. If set,
52+
// this DeploymentResourcePool will be secured by this key. Endpoints and the
53+
// DeploymentResourcePool they deploy in need to have the same EncryptionSpec.
54+
EncryptionSpec encryption_spec = 5;
55+
56+
// The service account that the DeploymentResourcePool's container(s) run as.
57+
// Specify the email address of the service account. If this service account
58+
// is not specified, the container(s) run as a service account that doesn't
59+
// have access to the resource project.
60+
//
61+
// Users deploying the Models to this DeploymentResourcePool must have the
62+
// `iam.serviceAccounts.actAs` permission on this service account.
63+
string service_account = 6;
64+
65+
// If the DeploymentResourcePool is deployed with custom-trained Models or
66+
// AutoML Tabular Models, the container(s) of the DeploymentResourcePool will
67+
// send `stderr` and `stdout` streams to Cloud Logging by default.
68+
// Please note that the logs incur cost, which are subject to [Cloud Logging
69+
// pricing](https://cloud.google.com/logging/pricing).
70+
//
71+
// User can disable container logging by setting this flag to true.
72+
bool disable_container_logging = 7;
73+
5074
// Output only. Timestamp when this DeploymentResourcePool was created.
5175
google.protobuf.Timestamp create_time = 4
5276
[(google.api.field_behavior) = OUTPUT_ONLY];

google/cloud/aiplatform/v1/feature.proto

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -103,6 +103,9 @@ message Feature {
103103

104104
// Used for Feature that is bytes.
105105
BYTES = 13;
106+
107+
// Used for Feature that is struct.
108+
STRUCT = 14;
106109
}
107110

108111
// Immutable. Name of the Feature.

google/cloud/aiplatform/v1/feature_group.proto

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -55,9 +55,9 @@ message FeatureGroup {
5555

5656
oneof source {
5757
// Indicates that features for this group come from BigQuery Table/View.
58-
// By default treats the source as a sparse time series source, which is
59-
// required to have an entity_id and a feature_timestamp column in the
60-
// source.
58+
// By default treats the source as a sparse time series source. The BigQuery
59+
// source table or view must have at least one entity ID column and a column
60+
// named `feature_timestamp`.
6161
BigQuery big_query = 7;
6262
}
6363

google/cloud/aiplatform/v1/feature_online_store.proto

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -18,6 +18,7 @@ package google.cloud.aiplatform.v1;
1818

1919
import "google/api/field_behavior.proto";
2020
import "google/api/resource.proto";
21+
import "google/cloud/aiplatform/v1/encryption_spec.proto";
2122
import "google/protobuf/timestamp.proto";
2223

2324
option csharp_namespace = "Google.Cloud.AIPlatform.V1";
@@ -140,4 +141,8 @@ message FeatureOnlineStore {
140141
// is different from common Vertex service endpoint.
141142
DedicatedServingEndpoint dedicated_serving_endpoint = 10
142143
[(google.api.field_behavior) = OPTIONAL];
144+
145+
// Optional. Customer-managed encryption key spec for data storage. If set,
146+
// online store will be secured by this key.
147+
EncryptionSpec encryption_spec = 13 [(google.api.field_behavior) = OPTIONAL];
143148
}

google/cloud/aiplatform/v1/featurestore_online_service.proto

Lines changed: 18 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -272,12 +272,30 @@ message FeatureValue {
272272

273273
// Bytes feature value.
274274
bytes bytes_value = 13;
275+
276+
// A struct type feature value.
277+
StructValue struct_value = 15;
275278
}
276279

277280
// Metadata of feature value.
278281
Metadata metadata = 14;
279282
}
280283

284+
// Struct (or object) type feature value.
285+
message StructValue {
286+
// A list of field values.
287+
repeated StructFieldValue values = 1;
288+
}
289+
290+
// One field of a Struct (or object) type feature value.
291+
message StructFieldValue {
292+
// Name of the field in the struct feature.
293+
string name = 1;
294+
295+
// The value for this field.
296+
FeatureValue value = 2;
297+
}
298+
281299
// Container for list of values.
282300
message FeatureValueList {
283301
// A list of feature values. All of them should be the same data type.

0 commit comments

Comments
 (0)