google-cloud-ai_platform-v1 0.14.0 → 0.16.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: e79a3a2efa15a0d271bc2a109641f3dbef05ff5e891ca4ad601e1121b17d3ec7
-  data.tar.gz: 3793cb6f87e9ff8a1cf0d84f2119cd0220cd667ef01009f38386a02554159345
+  metadata.gz: 804093fd89318fcc105ad9179cf3543f501fd7f084ab3b3e5ea3547925670c6f
+  data.tar.gz: b23fe659b2c2f041c25108eaba4b4c5208829d5934a4deaf49cb3fc4e1583606
 SHA512:
-  metadata.gz: 7dd1efb01ceaae47bab4e89331f7527d0da887333192d6c3b844ed4a07c1b439e85fbfad3875c46a7f65786b1b6f46f4ba0f0bbf8ea10ab676d66c9e0e784888
-  data.tar.gz: 7a2a5404e027609893d2be2d50529df8925fb145ae89f721ecbdf45d1b07d7dcef85dc01bf4e9d85e082aba9b31661a908612fe32e75fbc36b98997646088617
+  metadata.gz: c5bcf56c96f0d879dddff0e2de4cd6525a100d54d9e752933bf114ebd706ea23ce7d788a1a4529234495cab6b6796391ccb16d4aee1d34bb69d1e789b7090e63
+  data.tar.gz: e3f80e129635eda2f455c1a405ae255573edf069ed84cf96109cb5b5cc3b47b2a5ba3daa680e5289789d73015a4f2ffd4d5ed93a4bef5ddf8d998201f8147b49

data/README.md

@@ -46,7 +46,7 @@ for general usage information.
 ## Enabling Logging
 
 To enable logging for this library, set the logger for the underlying [gRPC](https://github.com/grpc/grpc/tree/master/src/ruby) library.
-The logger that you set may be a Ruby stdlib [`Logger`](https://ruby-doc.org/stdlib/libdoc/logger/rdoc/Logger.html) as shown below,
+The logger that you set may be a Ruby stdlib [`Logger`](https://ruby-doc.org/current/stdlibs/logger/Logger.html) as shown below,
 or a [`Google::Cloud::Logging::Logger`](https://googleapis.dev/ruby/google-cloud-logging/latest)
 that will write logs to [Cloud Logging](https://cloud.google.com/logging/). See [grpc/logconfig.rb](https://github.com/grpc/grpc/blob/master/src/ruby/lib/grpc/logconfig.rb)
 and the gRPC [spec_helper.rb](https://github.com/grpc/grpc/blob/master/src/ruby/spec/spec_helper.rb) for additional information.

data/lib/google/cloud/ai_platform/v1/dataset_service/client.rb

@@ -979,6 +979,156 @@ module Google
               raise ::Google::Cloud::Error.from_error(e)
             end
 
+            ##
+            # Searches DataItems in a Dataset.
+            #
+            # @overload search_data_items(request, options = nil)
+            #   Pass arguments to `search_data_items` via a request object, either of type
+            #   {::Google::Cloud::AIPlatform::V1::SearchDataItemsRequest} or an equivalent Hash.
+            #
+            #   @param request [::Google::Cloud::AIPlatform::V1::SearchDataItemsRequest, ::Hash]
+            #     A request object representing the call parameters. Required. To specify no
+            #     parameters, or to keep all the default parameter values, pass an empty Hash.
+            #   @param options [::Gapic::CallOptions, ::Hash]
+            #     Overrides the default settings for this call, e.g, timeout, retries, etc. Optional.
+            #
+            # @overload search_data_items(order_by_data_item: nil, order_by_annotation: nil, dataset: nil, saved_query: nil, data_labeling_job: nil, data_item_filter: nil, annotations_filter: nil, annotation_filters: nil, field_mask: nil, annotations_limit: nil, page_size: nil, order_by: nil, page_token: nil)
+            #   Pass arguments to `search_data_items` via keyword arguments. Note that at
+            #   least one keyword argument is required. To specify no parameters, or to keep all
+            #   the default parameter values, pass an empty Hash as a request object (see above).
+            #
+            #   @param order_by_data_item [::String]
+            #     A comma-separated list of data item fields to order by, sorted in
+            #     ascending order. Use "desc" after a field name for descending.
+            #   @param order_by_annotation [::Google::Cloud::AIPlatform::V1::SearchDataItemsRequest::OrderByAnnotation, ::Hash]
+            #     Expression that allows ranking results based on annotation's property.
+            #   @param dataset [::String]
+            #     Required. The resource name of the Dataset from which to search DataItems.
+            #     Format:
+            #     `projects/{project}/locations/{location}/datasets/{dataset}`
+            #   @param saved_query [::String]
+            #     The resource name of a SavedQuery(annotation set in UI).
+            #     Format:
+            #     `projects/{project}/locations/{location}/datasets/{dataset}/savedQueries/{saved_query}`
+            #     All of the search will be done in the context of this SavedQuery.
+            #   @param data_labeling_job [::String]
+            #     The resource name of a DataLabelingJob.
+            #     Format:
+            #     `projects/{project}/locations/{location}/dataLabelingJobs/{data_labeling_job}`
+            #     If this field is set, all of the search will be done in the context of
+            #     this DataLabelingJob.
+            #   @param data_item_filter [::String]
+            #     An expression for filtering the DataItem that will be returned.
+            #
+            #       * `data_item_id` - for = or !=.
+            #       * `labeled` - for = or !=.
+            #       * `has_annotation(ANNOTATION_SPEC_ID)` - true only for DataItem that
+            #         have at least one annotation with annotation_spec_id =
+            #         `ANNOTATION_SPEC_ID` in the context of SavedQuery or DataLabelingJob.
+            #
+            #     For example:
+            #
+            #     * `data_item=1`
+            #     * `has_annotation(5)`
+            #   @param annotations_filter [::String]
+            #     An expression for filtering the Annotations that will be returned per
+            #     DataItem.
+            #       * `annotation_spec_id` - for = or !=.
+            #   @param annotation_filters [::Array<::String>]
+            #     An expression that specifies what Annotations will be returned per
+            #     DataItem. Annotations satisfied either of the conditions will be returned.
+            #       * `annotation_spec_id` - for = or !=.
+            #     Must specify `saved_query_id=` - saved query id that annotations should
+            #     belong to.
+            #   @param field_mask [::Google::Protobuf::FieldMask, ::Hash]
+            #     Mask specifying which fields of {::Google::Cloud::AIPlatform::V1::DataItemView DataItemView} to read.
+            #   @param annotations_limit [::Integer]
+            #     If set, only up to this many of Annotations will be returned per
+            #     DataItemView. The maximum value is 1000. If not set, the maximum value will
+            #     be used.
+            #   @param page_size [::Integer]
+            #     Requested page size. Server may return fewer results than requested.
+            #     Default and maximum page size is 100.
+            #   @param order_by [::String]
+            #     A comma-separated list of fields to order by, sorted in ascending order.
+            #     Use "desc" after a field name for descending.
+            #   @param page_token [::String]
+            #     A token identifying a page of results for the server to return
+            #     Typically obtained via
+            #     {::Google::Cloud::AIPlatform::V1::SearchDataItemsResponse#next_page_token SearchDataItemsResponse.next_page_token} of the previous
+            #     {::Google::Cloud::AIPlatform::V1::DatasetService::Client#search_data_items DatasetService.SearchDataItems} call.
+            #
+            # @yield [response, operation] Access the result along with the RPC operation
+            # @yieldparam response [::Gapic::PagedEnumerable<::Google::Cloud::AIPlatform::V1::DataItemView>]
+            # @yieldparam operation [::GRPC::ActiveCall::Operation]
+            #
+            # @return [::Gapic::PagedEnumerable<::Google::Cloud::AIPlatform::V1::DataItemView>]
+            #
+            # @raise [::Google::Cloud::Error] if the RPC is aborted.
+            #
+            # @example Basic example
+            #   require "google/cloud/ai_platform/v1"
+            #
+            #   # Create a client object. The client can be reused for multiple calls.
+            #   client = Google::Cloud::AIPlatform::V1::DatasetService::Client.new
+            #
+            #   # Create a request. To set request fields, pass in keyword arguments.
+            #   request = Google::Cloud::AIPlatform::V1::SearchDataItemsRequest.new
+            #
+            #   # Call the search_data_items method.
+            #   result = client.search_data_items request
+            #
+            #   # The returned object is of type Gapic::PagedEnumerable. You can
+            #   # iterate over all elements by calling #each, and the enumerable
+            #   # will lazily make API calls to fetch subsequent pages. Other
+            #   # methods are also available for managing paging directly.
+            #   result.each do |response|
+            #     # Each element is of type ::Google::Cloud::AIPlatform::V1::DataItemView.
+            #     p response
+            #   end
+            #
+            def search_data_items request, options = nil
+              raise ::ArgumentError, "request must be provided" if request.nil?
+
+              request = ::Gapic::Protobuf.coerce request, to: ::Google::Cloud::AIPlatform::V1::SearchDataItemsRequest
+
+              # Converts hash and nil to an options object
+              options = ::Gapic::CallOptions.new(**options.to_h) if options.respond_to? :to_h
+
+              # Customize the options with defaults
+              metadata = @config.rpcs.search_data_items.metadata.to_h
+
+              # Set x-goog-api-client and x-goog-user-project headers
+              metadata[:"x-goog-api-client"] ||= ::Gapic::Headers.x_goog_api_client \
+                lib_name: @config.lib_name, lib_version: @config.lib_version,
+                gapic_version: ::Google::Cloud::AIPlatform::V1::VERSION
+              metadata[:"x-goog-user-project"] = @quota_project_id if @quota_project_id
+
+              header_params = {}
+              if request.dataset
+                header_params["dataset"] = request.dataset
+              end
+
+              request_params_header = header_params.map { |k, v| "#{k}=#{v}" }.join("&")
+              metadata[:"x-goog-request-params"] ||= request_params_header
+
+              options.apply_defaults timeout:      @config.rpcs.search_data_items.timeout,
+                                     metadata:     metadata,
+                                     retry_policy: @config.rpcs.search_data_items.retry_policy
+
+              options.apply_defaults timeout:      @config.timeout,
+                                     metadata:     @config.metadata,
+                                     retry_policy: @config.retry_policy
+
+              @dataset_service_stub.call_rpc :search_data_items, request, options: options do |response, operation|
+                response = ::Gapic::PagedEnumerable.new @dataset_service_stub, :search_data_items, request, response, operation, options
+                yield response, operation if block_given?
+                return response
+              end
+            rescue ::GRPC::BadStatus => e
+              raise ::Google::Cloud::Error.from_error(e)
+            end
+
             ##
             # Lists SavedQueries in a Dataset.
             #
@@ -1454,6 +1604,11 @@ module Google
                 #
                 attr_reader :list_data_items
                 ##
+                # RPC-specific configuration for `search_data_items`
+                # @return [::Gapic::Config::Method]
+                #
+                attr_reader :search_data_items
+                ##
                 # RPC-specific configuration for `list_saved_queries`
                 # @return [::Gapic::Config::Method]
                 #
@@ -1487,6 +1642,8 @@ module Google
                   @export_data = ::Gapic::Config::Method.new export_data_config
                   list_data_items_config = parent_rpcs.list_data_items if parent_rpcs.respond_to? :list_data_items
                   @list_data_items = ::Gapic::Config::Method.new list_data_items_config
+                  search_data_items_config = parent_rpcs.search_data_items if parent_rpcs.respond_to? :search_data_items
+                  @search_data_items = ::Gapic::Config::Method.new search_data_items_config
                   list_saved_queries_config = parent_rpcs.list_saved_queries if parent_rpcs.respond_to? :list_saved_queries
                   @list_saved_queries = ::Gapic::Config::Method.new list_saved_queries_config
                   get_annotation_spec_config = parent_rpcs.get_annotation_spec if parent_rpcs.respond_to? :get_annotation_spec

data/lib/google/cloud/ai_platform/v1/dataset_service/paths.rb

@@ -102,6 +102,27 @@ module Google
               "projects/#{project}/locations/#{location}"
             end
 
+            ##
+            # Create a fully-qualified SavedQuery resource string.
+            #
+            # The resource will be in the following format:
+            #
+            # `projects/{project}/locations/{location}/datasets/{dataset}/savedQueries/{saved_query}`
+            #
+            # @param project [String]
+            # @param location [String]
+            # @param dataset [String]
+            # @param saved_query [String]
+            #
+            # @return [::String]
+            def saved_query_path project:, location:, dataset:, saved_query:
+              raise ::ArgumentError, "project cannot contain /" if project.to_s.include? "/"
+              raise ::ArgumentError, "location cannot contain /" if location.to_s.include? "/"
+              raise ::ArgumentError, "dataset cannot contain /" if dataset.to_s.include? "/"
+
+              "projects/#{project}/locations/#{location}/datasets/#{dataset}/savedQueries/#{saved_query}"
+            end
+
             extend self
           end
         end

data/lib/google/cloud/ai_platform/v1/featurestore_online_serving_service/client.rb

@@ -371,6 +371,102 @@ module Google
               raise ::Google::Cloud::Error.from_error(e)
             end
 
+            ##
+            # Writes Feature values of one or more entities of an EntityType.
+            #
+            # The Feature values are merged into existing entities if any. The Feature
+            # values to be written must have timestamp within the online storage
+            # retention.
+            #
+            # @overload write_feature_values(request, options = nil)
+            #   Pass arguments to `write_feature_values` via a request object, either of type
+            #   {::Google::Cloud::AIPlatform::V1::WriteFeatureValuesRequest} or an equivalent Hash.
+            #
+            #   @param request [::Google::Cloud::AIPlatform::V1::WriteFeatureValuesRequest, ::Hash]
+            #     A request object representing the call parameters. Required. To specify no
+            #     parameters, or to keep all the default parameter values, pass an empty Hash.
+            #   @param options [::Gapic::CallOptions, ::Hash]
+            #     Overrides the default settings for this call, e.g, timeout, retries, etc. Optional.
+            #
+            # @overload write_feature_values(entity_type: nil, payloads: nil)
+            #   Pass arguments to `write_feature_values` via keyword arguments. Note that at
+            #   least one keyword argument is required. To specify no parameters, or to keep all
+            #   the default parameter values, pass an empty Hash as a request object (see above).
+            #
+            #   @param entity_type [::String]
+            #     Required. The resource name of the EntityType for the entities being written.
+            #     Value format: `projects/{project}/locations/{location}/featurestores/
+            #     \\{featurestore}/entityTypes/\\{entityType}`. For example,
+            #     for a machine learning model predicting user clicks on a website, an
+            #     EntityType ID could be `user`.
+            #   @param payloads [::Array<::Google::Cloud::AIPlatform::V1::WriteFeatureValuesPayload, ::Hash>]
+            #     Required. The entities to be written. Up to 100,000 feature values can be written
+            #     across all `payloads`.
+            #
+            # @yield [response, operation] Access the result along with the RPC operation
+            # @yieldparam response [::Google::Cloud::AIPlatform::V1::WriteFeatureValuesResponse]
+            # @yieldparam operation [::GRPC::ActiveCall::Operation]
+            #
+            # @return [::Google::Cloud::AIPlatform::V1::WriteFeatureValuesResponse]
+            #
+            # @raise [::Google::Cloud::Error] if the RPC is aborted.
+            #
+            # @example Basic example
+            #   require "google/cloud/ai_platform/v1"
+            #
+            #   # Create a client object. The client can be reused for multiple calls.
+            #   client = Google::Cloud::AIPlatform::V1::FeaturestoreOnlineServingService::Client.new
+            #
+            #   # Create a request. To set request fields, pass in keyword arguments.
+            #   request = Google::Cloud::AIPlatform::V1::WriteFeatureValuesRequest.new
+            #
+            #   # Call the write_feature_values method.
+            #   result = client.write_feature_values request
+            #
+            #   # The returned object is of type Google::Cloud::AIPlatform::V1::WriteFeatureValuesResponse.
+            #   p result
+            #
+            def write_feature_values request, options = nil
+              raise ::ArgumentError, "request must be provided" if request.nil?
+
+              request = ::Gapic::Protobuf.coerce request, to: ::Google::Cloud::AIPlatform::V1::WriteFeatureValuesRequest
+
+              # Converts hash and nil to an options object
+              options = ::Gapic::CallOptions.new(**options.to_h) if options.respond_to? :to_h
+
+              # Customize the options with defaults
+              metadata = @config.rpcs.write_feature_values.metadata.to_h
+
+              # Set x-goog-api-client and x-goog-user-project headers
+              metadata[:"x-goog-api-client"] ||= ::Gapic::Headers.x_goog_api_client \
+                lib_name: @config.lib_name, lib_version: @config.lib_version,
+                gapic_version: ::Google::Cloud::AIPlatform::V1::VERSION
+              metadata[:"x-goog-user-project"] = @quota_project_id if @quota_project_id
+
+              header_params = {}
+              if request.entity_type
+                header_params["entity_type"] = request.entity_type
+              end
+
+              request_params_header = header_params.map { |k, v| "#{k}=#{v}" }.join("&")
+              metadata[:"x-goog-request-params"] ||= request_params_header
+
+              options.apply_defaults timeout:      @config.rpcs.write_feature_values.timeout,
+                                     metadata:     metadata,
+                                     retry_policy: @config.rpcs.write_feature_values.retry_policy
+
+              options.apply_defaults timeout:      @config.timeout,
+                                     metadata:     @config.metadata,
+                                     retry_policy: @config.retry_policy
+
+              @featurestore_online_serving_service_stub.call_rpc :write_feature_values, request, options: options do |response, operation|
+                yield response, operation if block_given?
+                return response
+              end
+            rescue ::GRPC::BadStatus => e
+              raise ::Google::Cloud::Error.from_error(e)
+            end
+
             ##
             # Configuration class for the FeaturestoreOnlineServingService API.
             #
@@ -516,6 +612,11 @@ module Google
                 # @return [::Gapic::Config::Method]
                 #
                 attr_reader :streaming_read_feature_values
+                ##
+                # RPC-specific configuration for `write_feature_values`
+                # @return [::Gapic::Config::Method]
+                #
+                attr_reader :write_feature_values
 
                 # @private
                 def initialize parent_rpcs = nil
@@ -523,6 +624,8 @@ module Google
                   @read_feature_values = ::Gapic::Config::Method.new read_feature_values_config
                   streaming_read_feature_values_config = parent_rpcs.streaming_read_feature_values if parent_rpcs.respond_to? :streaming_read_feature_values
                   @streaming_read_feature_values = ::Gapic::Config::Method.new streaming_read_feature_values_config
+                  write_feature_values_config = parent_rpcs.write_feature_values if parent_rpcs.respond_to? :write_feature_values
+                  @write_feature_values = ::Gapic::Config::Method.new write_feature_values_config
 
                   yield self if block_given?
                 end

data/lib/google/cloud/ai_platform/v1/model_service/client.rb

@@ -198,7 +198,7 @@ module Google
             #   @param options [::Gapic::CallOptions, ::Hash]
             #     Overrides the default settings for this call, e.g, timeout, retries, etc. Optional.
             #
-            # @overload upload_model(parent: nil, parent_model: nil, model_id: nil, model: nil)
+            # @overload upload_model(parent: nil, parent_model: nil, model_id: nil, model: nil, service_account: nil)
             #   Pass arguments to `upload_model` via keyword arguments. Note that at
             #   least one keyword argument is required. To specify no parameters, or to keep all
             #   the default parameter values, pass an empty Hash as a request object (see above).
@@ -217,6 +217,14 @@ module Google
             #     `[a-z0-9_-]`. The first character cannot be a number or hyphen.
             #   @param model [::Google::Cloud::AIPlatform::V1::Model, ::Hash]
             #     Required. The Model to create.
+            #   @param service_account [::String]
+            #     Optional. The user-provided custom service account to use to do the model
+            #     upload. If empty, [Vertex AI Service
+            #     Agent](https://cloud.google.com/vertex-ai/docs/general/access-control#service-agents)
+            #     will be used. Users uploading the Model must have the
+            #     `iam.serviceAccounts.actAs` permission on this service account. Also, this
+            #     account must belong to the project specified in the `parent` field and have
+            #     all necessary read permissions.
             #
             # @yield [response, operation] Access the result along with the RPC operation
             # @yieldparam response [::Gapic::Operation]

data/lib/google/cloud/ai_platform/v1/version.rb

@@ -21,7 +21,7 @@ module Google
   module Cloud
     module AIPlatform
       module V1
-        VERSION = "0.14.0"
+        VERSION = "0.16.0"
       end
     end
   end

data/lib/google/cloud/aiplatform/v1/batch_prediction_job_pb.rb

@@ -29,6 +29,7 @@ Google::Protobuf::DescriptorPool.generated_pool.build do
       optional :model_parameters, :message, 5, "google.protobuf.Value"
       optional :output_config, :message, 6, "google.cloud.aiplatform.v1.BatchPredictionJob.OutputConfig"
       optional :dedicated_resources, :message, 7, "google.cloud.aiplatform.v1.BatchDedicatedResources"
+      optional :service_account, :string, 29
       optional :manual_batch_tuning_parameters, :message, 8, "google.cloud.aiplatform.v1.ManualBatchTuningParameters"
       optional :generate_explanation, :bool, 23
       optional :explanation_spec, :message, 25, "google.cloud.aiplatform.v1.ExplanationSpec"

data/lib/google/cloud/aiplatform/v1/dataset_pb.rb

@@ -23,6 +23,7 @@ Google::Protobuf::DescriptorPool.generated_pool.build do
       optional :etag, :string, 6
       map :labels, :string, :string, 7
       optional :encryption_spec, :message, 11, "google.cloud.aiplatform.v1.EncryptionSpec"
+      optional :metadata_artifact, :string, 17
     end
     add_message "google.cloud.aiplatform.v1.ImportDataConfig" do
       map :data_item_labels, :string, :string, 2

data/lib/google/cloud/aiplatform/v1/dataset_service_pb.rb

@@ -80,6 +80,36 @@ Google::Protobuf::DescriptorPool.generated_pool.build do
       repeated :data_items, :message, 1, "google.cloud.aiplatform.v1.DataItem"
       optional :next_page_token, :string, 2
     end
+    add_message "google.cloud.aiplatform.v1.SearchDataItemsRequest" do
+      optional :dataset, :string, 1
+      optional :saved_query, :string, 2
+      optional :data_labeling_job, :string, 3
+      optional :data_item_filter, :string, 4
+      optional :annotations_filter, :string, 5
+      repeated :annotation_filters, :string, 11
+      optional :field_mask, :message, 6, "google.protobuf.FieldMask"
+      optional :annotations_limit, :int32, 7
+      optional :page_size, :int32, 8
+      optional :order_by, :string, 9
+      optional :page_token, :string, 10
+      oneof :order do
+        optional :order_by_data_item, :string, 12
+        optional :order_by_annotation, :message, 13, "google.cloud.aiplatform.v1.SearchDataItemsRequest.OrderByAnnotation"
+      end
+    end
+    add_message "google.cloud.aiplatform.v1.SearchDataItemsRequest.OrderByAnnotation" do
+      optional :saved_query, :string, 1
+      optional :order_by, :string, 2
+    end
+    add_message "google.cloud.aiplatform.v1.SearchDataItemsResponse" do
+      repeated :data_item_views, :message, 1, "google.cloud.aiplatform.v1.DataItemView"
+      optional :next_page_token, :string, 2
+    end
+    add_message "google.cloud.aiplatform.v1.DataItemView" do
+      optional :data_item, :message, 1, "google.cloud.aiplatform.v1.DataItem"
+      repeated :annotations, :message, 2, "google.cloud.aiplatform.v1.Annotation"
+      optional :has_truncated_annotations, :bool, 3
+    end
     add_message "google.cloud.aiplatform.v1.ListSavedQueriesRequest" do
       optional :parent, :string, 1
       optional :filter, :string, 2
@@ -130,6 +160,10 @@ module Google
         ExportDataOperationMetadata = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.ExportDataOperationMetadata").msgclass
         ListDataItemsRequest = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.ListDataItemsRequest").msgclass
         ListDataItemsResponse = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.ListDataItemsResponse").msgclass
+        SearchDataItemsRequest = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.SearchDataItemsRequest").msgclass
+        SearchDataItemsRequest::OrderByAnnotation = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.SearchDataItemsRequest.OrderByAnnotation").msgclass
+        SearchDataItemsResponse = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.SearchDataItemsResponse").msgclass
+        DataItemView = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.DataItemView").msgclass
         ListSavedQueriesRequest = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.ListSavedQueriesRequest").msgclass
         ListSavedQueriesResponse = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.ListSavedQueriesResponse").msgclass
         GetAnnotationSpecRequest = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.GetAnnotationSpecRequest").msgclass

data/lib/google/cloud/aiplatform/v1/dataset_service_services_pb.rb

@@ -50,6 +50,8 @@ module Google
             rpc :ExportData, ::Google::Cloud::AIPlatform::V1::ExportDataRequest, ::Google::Longrunning::Operation
             # Lists DataItems in a Dataset.
             rpc :ListDataItems, ::Google::Cloud::AIPlatform::V1::ListDataItemsRequest, ::Google::Cloud::AIPlatform::V1::ListDataItemsResponse
+            # Searches DataItems in a Dataset.
+            rpc :SearchDataItems, ::Google::Cloud::AIPlatform::V1::SearchDataItemsRequest, ::Google::Cloud::AIPlatform::V1::SearchDataItemsResponse
             # Lists SavedQueries in a Dataset.
             rpc :ListSavedQueries, ::Google::Cloud::AIPlatform::V1::ListSavedQueriesRequest, ::Google::Cloud::AIPlatform::V1::ListSavedQueriesResponse
             # Gets an AnnotationSpec.

data/lib/google/cloud/aiplatform/v1/featurestore_online_service_pb.rb

@@ -13,6 +13,16 @@ require 'google/protobuf/timestamp_pb'
 
 Google::Protobuf::DescriptorPool.generated_pool.build do
   add_file("google/cloud/aiplatform/v1/featurestore_online_service.proto", :syntax => :proto3) do
+    add_message "google.cloud.aiplatform.v1.WriteFeatureValuesRequest" do
+      optional :entity_type, :string, 1
+      repeated :payloads, :message, 2, "google.cloud.aiplatform.v1.WriteFeatureValuesPayload"
+    end
+    add_message "google.cloud.aiplatform.v1.WriteFeatureValuesPayload" do
+      optional :entity_id, :string, 1
+      map :feature_values, :string, :message, 2, "google.cloud.aiplatform.v1.FeatureValue"
+    end
+    add_message "google.cloud.aiplatform.v1.WriteFeatureValuesResponse" do
+    end
     add_message "google.cloud.aiplatform.v1.ReadFeatureValuesRequest" do
       optional :entity_type, :string, 1
       optional :entity_id, :string, 2
@@ -71,6 +81,9 @@ module Google
   module Cloud
     module AIPlatform
       module V1
+        WriteFeatureValuesRequest = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.WriteFeatureValuesRequest").msgclass
+        WriteFeatureValuesPayload = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.WriteFeatureValuesPayload").msgclass
+        WriteFeatureValuesResponse = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.WriteFeatureValuesResponse").msgclass
         ReadFeatureValuesRequest = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.ReadFeatureValuesRequest").msgclass
         ReadFeatureValuesResponse = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.ReadFeatureValuesResponse").msgclass
         ReadFeatureValuesResponse::FeatureDescriptor = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.ReadFeatureValuesResponse.FeatureDescriptor").msgclass

data/lib/google/cloud/aiplatform/v1/featurestore_online_service_services_pb.rb

@@ -41,6 +41,12 @@ module Google
             # for different entities may be broken
             # up across multiple responses.
             rpc :StreamingReadFeatureValues, ::Google::Cloud::AIPlatform::V1::StreamingReadFeatureValuesRequest, stream(::Google::Cloud::AIPlatform::V1::ReadFeatureValuesResponse)
+            # Writes Feature values of one or more entities of an EntityType.
+            #
+            # The Feature values are merged into existing entities if any. The Feature
+            # values to be written must have timestamp within the online storage
+            # retention.
+            rpc :WriteFeatureValues, ::Google::Cloud::AIPlatform::V1::WriteFeatureValuesRequest, ::Google::Cloud::AIPlatform::V1::WriteFeatureValuesResponse
           end
 
           Stub = Service.rpc_stub_class

data/lib/google/cloud/aiplatform/v1/featurestore_service_pb.rb

@@ -209,6 +209,7 @@ Google::Protobuf::DescriptorPool.generated_pool.build do
       optional :generic_metadata, :message, 1, "google.cloud.aiplatform.v1.GenericOperationMetadata"
       optional :imported_entity_count, :int64, 2
       optional :imported_feature_value_count, :int64, 3
+      repeated :source_uris, :string, 4
       optional :invalid_row_count, :int64, 6
       optional :timestamp_outside_retention_rows_count, :int64, 7
     end

data/lib/google/cloud/aiplatform/v1/model_service_pb.rb

@@ -22,6 +22,7 @@ Google::Protobuf::DescriptorPool.generated_pool.build do
       optional :parent_model, :string, 4
       optional :model_id, :string, 5
       optional :model, :message, 2, "google.cloud.aiplatform.v1.Model"
+      optional :service_account, :string, 6
     end
     add_message "google.cloud.aiplatform.v1.UploadModelOperationMetadata" do
       optional :generic_metadata, :message, 1, "google.cloud.aiplatform.v1.GenericOperationMetadata"