google-cloud-discovery_engine-v1beta 0.26.0 → 0.28.0

data/lib/google/cloud/discovery_engine/v1beta/conversational_search_service/rest/client.rb

@@ -791,7 +791,7 @@ module Google
               #   @param options [::Gapic::CallOptions, ::Hash]
               #     Overrides the default settings for this call, e.g, timeout, retries etc. Optional.
               #
-              # @overload answer_query(serving_config: nil, query: nil, session: nil, safety_spec: nil, related_questions_spec: nil, grounding_spec: nil, answer_generation_spec: nil, search_spec: nil, query_understanding_spec: nil, asynchronous_mode: nil, user_pseudo_id: nil, user_labels: nil)
+              # @overload answer_query(serving_config: nil, query: nil, session: nil, safety_spec: nil, related_questions_spec: nil, grounding_spec: nil, answer_generation_spec: nil, search_spec: nil, query_understanding_spec: nil, asynchronous_mode: nil, user_pseudo_id: nil, user_labels: nil, end_user_spec: nil)
               #   Pass arguments to `answer_query` 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).
@@ -801,6 +801,12 @@ module Google
               #     `projects/*/locations/global/collections/default_collection/engines/*/servingConfigs/default_serving_config`,
               #     or
               #     `projects/*/locations/global/collections/default_collection/dataStores/*/servingConfigs/default_serving_config`.
+              #
+              #     Or the resource name of the agent engine serving config, such as:
+              #     `projects/*/locations/global/collections/default_collection/engines/*/servingConfigs/default_agent_answer`.
+              #     (use when `enable_agent_invocation` set to true, and you have custom
+              #     `AI_MODE` agent engine configured)
+              #
               #     This field is used to identify the serving configuration name, set
               #     of models used to make the search.
               #   @param query [::Google::Cloud::DiscoveryEngine::V1beta::Query, ::Hash]
@@ -865,6 +871,8 @@ module Google
               #     See [Google Cloud
               #     Document](https://cloud.google.com/resource-manager/docs/creating-managing-labels#requirements)
               #     for more details.
+              #   @param end_user_spec [::Google::Cloud::DiscoveryEngine::V1beta::AnswerQueryRequest::EndUserSpec, ::Hash]
+              #     Optional. End user specification.
               # @yield [result, operation] Access the result along with the TransportOperation object
               # @yieldparam result [::Google::Cloud::DiscoveryEngine::V1beta::AnswerQueryResponse]
               # @yieldparam operation [::Gapic::Rest::TransportOperation]
@@ -923,6 +931,170 @@ module Google
                 raise ::Google::Cloud::Error.from_error(e)
               end
 
+              ##
+              # Answer query method (streaming).
+              #
+              # It takes one
+              # {::Google::Cloud::DiscoveryEngine::V1beta::AnswerQueryRequest AnswerQueryRequest}
+              # and returns multiple
+              # {::Google::Cloud::DiscoveryEngine::V1beta::AnswerQueryResponse AnswerQueryResponse}
+              # messages in a stream.
+              #
+              # @overload stream_answer_query(request, options = nil)
+              #   Pass arguments to `stream_answer_query` via a request object, either of type
+              #   {::Google::Cloud::DiscoveryEngine::V1beta::AnswerQueryRequest} or an equivalent Hash.
+              #
+              #   @param request [::Google::Cloud::DiscoveryEngine::V1beta::AnswerQueryRequest, ::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 stream_answer_query(serving_config: nil, query: nil, session: nil, safety_spec: nil, related_questions_spec: nil, grounding_spec: nil, answer_generation_spec: nil, search_spec: nil, query_understanding_spec: nil, asynchronous_mode: nil, user_pseudo_id: nil, user_labels: nil, end_user_spec: nil)
+              #   Pass arguments to `stream_answer_query` 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 serving_config [::String]
+              #     Required. The resource name of the Search serving config, such as
+              #     `projects/*/locations/global/collections/default_collection/engines/*/servingConfigs/default_serving_config`,
+              #     or
+              #     `projects/*/locations/global/collections/default_collection/dataStores/*/servingConfigs/default_serving_config`.
+              #
+              #     Or the resource name of the agent engine serving config, such as:
+              #     `projects/*/locations/global/collections/default_collection/engines/*/servingConfigs/default_agent_answer`.
+              #     (use when `enable_agent_invocation` set to true, and you have custom
+              #     `AI_MODE` agent engine configured)
+              #
+              #     This field is used to identify the serving configuration name, set
+              #     of models used to make the search.
+              #   @param query [::Google::Cloud::DiscoveryEngine::V1beta::Query, ::Hash]
+              #     Required. Current user query.
+              #   @param session [::String]
+              #     The session resource name. Not required.
+              #
+              #     When session field is not set, the API is in sessionless mode.
+              #
+              #     We support auto session mode: users can use the wildcard symbol `-` as
+              #     session ID.  A new ID will be automatically generated and assigned.
+              #   @param safety_spec [::Google::Cloud::DiscoveryEngine::V1beta::AnswerQueryRequest::SafetySpec, ::Hash]
+              #     Model specification.
+              #   @param related_questions_spec [::Google::Cloud::DiscoveryEngine::V1beta::AnswerQueryRequest::RelatedQuestionsSpec, ::Hash]
+              #     Related questions specification.
+              #   @param grounding_spec [::Google::Cloud::DiscoveryEngine::V1beta::AnswerQueryRequest::GroundingSpec, ::Hash]
+              #     Optional. Grounding specification.
+              #   @param answer_generation_spec [::Google::Cloud::DiscoveryEngine::V1beta::AnswerQueryRequest::AnswerGenerationSpec, ::Hash]
+              #     Answer generation specification.
+              #   @param search_spec [::Google::Cloud::DiscoveryEngine::V1beta::AnswerQueryRequest::SearchSpec, ::Hash]
+              #     Search specification.
+              #   @param query_understanding_spec [::Google::Cloud::DiscoveryEngine::V1beta::AnswerQueryRequest::QueryUnderstandingSpec, ::Hash]
+              #     Query understanding specification.
+              #   @param asynchronous_mode [::Boolean]
+              #     Deprecated: This field is deprecated. Streaming Answer API will be
+              #     supported.
+              #
+              #     Asynchronous mode control.
+              #
+              #     If enabled, the response will be returned with answer/session resource
+              #     name without final answer. The API users need to do the polling to get
+              #     the latest status of answer/session by calling
+              #     {::Google::Cloud::DiscoveryEngine::V1beta::ConversationalSearchService::Rest::Client#get_answer ConversationalSearchService.GetAnswer}
+              #     or
+              #     {::Google::Cloud::DiscoveryEngine::V1beta::ConversationalSearchService::Rest::Client#get_session ConversationalSearchService.GetSession}
+              #     method.
+              #   @param user_pseudo_id [::String]
+              #     A unique identifier for tracking visitors. For example, this could be
+              #     implemented with an HTTP cookie, which should be able to uniquely identify
+              #     a visitor on a single device. This unique identifier should not change if
+              #     the visitor logs in or out of the website.
+              #
+              #     This field should NOT have a fixed value such as `unknown_visitor`.
+              #
+              #     The field must be a UTF-8 encoded string with a length limit of 128
+              #     characters. Otherwise, an  `INVALID_ARGUMENT`  error is returned.
+              #   @param user_labels [::Hash{::String => ::String}]
+              #     The user labels applied to a resource must meet the following requirements:
+              #
+              #     * Each resource can have multiple labels, up to a maximum of 64.
+              #     * Each label must be a key-value pair.
+              #     * Keys have a minimum length of 1 character and a maximum length of 63
+              #       characters and cannot be empty. Values can be empty and have a maximum
+              #       length of 63 characters.
+              #     * Keys and values can contain only lowercase letters, numeric characters,
+              #       underscores, and dashes. All characters must use UTF-8 encoding, and
+              #       international characters are allowed.
+              #     * The key portion of a label must be unique. However, you can use the same
+              #       key with multiple resources.
+              #     * Keys must start with a lowercase letter or international character.
+              #
+              #     See [Google Cloud
+              #     Document](https://cloud.google.com/resource-manager/docs/creating-managing-labels#requirements)
+              #     for more details.
+              #   @param end_user_spec [::Google::Cloud::DiscoveryEngine::V1beta::AnswerQueryRequest::EndUserSpec, ::Hash]
+              #     Optional. End user specification.
+              # @return [::Enumerable<::Google::Cloud::DiscoveryEngine::V1beta::AnswerQueryResponse>]
+              #
+              # @raise [::Google::Cloud::Error] if the REST call is aborted.
+              #
+              # @example Basic example
+              #   require "google/cloud/discovery_engine/v1beta"
+              #
+              #   # Create a client object. The client can be reused for multiple calls.
+              #   client = Google::Cloud::DiscoveryEngine::V1beta::ConversationalSearchService::Rest::Client.new
+              #
+              #   # Create a request. To set request fields, pass in keyword arguments.
+              #   request = Google::Cloud::DiscoveryEngine::V1beta::AnswerQueryRequest.new
+              #
+              #   # Call the stream_answer_query method to start streaming.
+              #   output = client.stream_answer_query request
+              #
+              #   # The returned object is a streamed enumerable yielding elements of type
+              #   # ::Google::Cloud::DiscoveryEngine::V1beta::AnswerQueryResponse
+              #   output.each do |current_response|
+              #     p current_response
+              #   end
+              #
+              def stream_answer_query request, options = nil
+                raise ::ArgumentError, "request must be provided" if request.nil?
+
+                request = ::Gapic::Protobuf.coerce request, to: ::Google::Cloud::DiscoveryEngine::V1beta::AnswerQueryRequest
+
+                # 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
+                call_metadata = @config.rpcs.stream_answer_query.metadata.to_h
+
+                # Set x-goog-api-client, x-goog-user-project and x-goog-api-version headers
+                call_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::DiscoveryEngine::V1beta::VERSION,
+                  transports_version_send: [:rest]
+
+                call_metadata[:"x-goog-api-version"] = API_VERSION unless API_VERSION.empty?
+                call_metadata[:"x-goog-user-project"] = @quota_project_id if @quota_project_id
+
+                options.apply_defaults timeout:      @config.rpcs.stream_answer_query.timeout,
+                                       metadata:     call_metadata,
+                                       retry_policy: @config.rpcs.stream_answer_query.retry_policy
+
+                options.apply_defaults timeout:      @config.timeout,
+                                       metadata:     @config.metadata,
+                                       retry_policy: @config.retry_policy
+
+                ::Gapic::Rest::ServerStream.new(
+                  ::Google::Cloud::DiscoveryEngine::V1beta::AnswerQueryResponse,
+                  ::Gapic::Rest::ThreadedEnumerator.new do |in_q, out_q|
+                    @conversational_search_service_stub.stream_answer_query request, options do |chunk|
+                      in_q.deq
+                      out_q.enq chunk
+                    end
+                  end
+                )
+              rescue ::Gapic::Rest::Error => e
+                raise ::Google::Cloud::Error.from_error(e)
+              end
+
               ##
               # Gets a Answer.
               #
@@ -1018,7 +1190,7 @@ module Google
               #   @param options [::Gapic::CallOptions, ::Hash]
               #     Overrides the default settings for this call, e.g, timeout, retries etc. Optional.
               #
-              # @overload create_session(parent: nil, session: nil)
+              # @overload create_session(parent: nil, session: nil, session_id: nil)
               #   Pass arguments to `create_session` 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).
@@ -1028,6 +1200,13 @@ module Google
               #     `projects/{project}/locations/{location}/collections/{collection}/dataStores/{data_store_id}`
               #   @param session [::Google::Cloud::DiscoveryEngine::V1beta::Session, ::Hash]
               #     Required. The session to create.
+              #   @param session_id [::String]
+              #     Optional. The ID to use for the session, which will become the final
+              #     component of the session's resource name.
+              #
+              #     This value should be 1-63 characters, and valid characters
+              #     are /[a-z0-9][a-z0-9-]\\{0,61}[a-z0-9]/. If not specified, a unique ID will
+              #     be generated.
               # @yield [result, operation] Access the result along with the TransportOperation object
               # @yieldparam result [::Google::Cloud::DiscoveryEngine::V1beta::Session]
               # @yieldparam operation [::Gapic::Rest::TransportOperation]
@@ -1370,7 +1549,9 @@ module Google
               #     Provide this to retrieve the subsequent page.
               #   @param filter [::String]
               #     A comma-separated list of fields to filter by, in EBNF grammar.
+              #
               #     The supported fields are:
+              #
               #     * `user_pseudo_id`
               #     * `state`
               #     * `display_name`
@@ -1379,28 +1560,35 @@ module Google
               #     * `labels`
               #     * `create_time`
               #     * `update_time`
+              #     * `collaborative_project`
               #
               #     Examples:
-              #     "user_pseudo_id = some_id"
-              #     "display_name = \"some_name\""
-              #     "starred = true"
-              #     "is_pinned=true AND (NOT labels:hidden)"
-              #     "create_time > \"1970-01-01T12:00:00Z\""
+              #
+              #     * `user_pseudo_id = some_id`
+              #     * `display_name = "some_name"`
+              #     * `starred = true`
+              #     * `is_pinned=true AND (NOT labels:hidden)`
+              #     * `create_time > "1970-01-01T12:00:00Z"`
+              #     * `collaborative_project =
+              #          "projects/123/locations/global/collections/default_collection/engines/"
+              #          "default_engine/collaborative_projects/cp1"`
               #   @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.
+              #
               #     Supported fields:
               #
               #       * `update_time`
               #       * `create_time`
               #       * `session_name`
               #       * `is_pinned`
+              #       * `display_name`
               #
               #     Example:
               #
-              #     * "update_time desc"
-              #     * "create_time"
-              #     * "is_pinned desc,update_time desc": list sessions by is_pinned first, then
+              #     * `update_time desc`
+              #     * `create_time`
+              #     * `is_pinned desc,update_time desc`: list sessions by is_pinned first, then
               #        by update_time.
               # @yield [result, operation] Access the result along with the TransportOperation object
               # @yieldparam result [::Gapic::Rest::PagedEnumerable<::Google::Cloud::DiscoveryEngine::V1beta::Session>]
@@ -1536,6 +1724,7 @@ module Google
               #    *  `:initial_delay` (*type:* `Numeric`) - The initial delay in seconds.
               #    *  `:max_delay` (*type:* `Numeric`) - The max delay in seconds.
               #    *  `:multiplier` (*type:* `Numeric`) - The incremental backoff multiplier.
+              #    *  `:jitter` (*type:* `Numeric`) - The jitter in seconds. Default: 1.0.
               #    *  `:retry_codes` (*type:* `Array<String>`) - The error codes that should
               #       trigger a retry.
               #   @return [::Hash]
@@ -1608,6 +1797,7 @@ module Google
                 #      *  `:initial_delay` (*type:* `Numeric`) - The initial delay in seconds.
                 #      *  `:max_delay` (*type:* `Numeric`) - The max delay in seconds.
                 #      *  `:multiplier` (*type:* `Numeric`) - The incremental backoff multiplier.
+                #      *  `:jitter` (*type:* `Numeric`) - The jitter in seconds. Default: 1.0.
                 #      *  `:retry_codes` (*type:* `Array<String>`) - The error codes that should
                 #         trigger a retry.
                 #
@@ -1648,6 +1838,11 @@ module Google
                   #
                   attr_reader :answer_query
                   ##
+                  # RPC-specific configuration for `stream_answer_query`
+                  # @return [::Gapic::Config::Method]
+                  #
+                  attr_reader :stream_answer_query
+                  ##
                   # RPC-specific configuration for `get_answer`
                   # @return [::Gapic::Config::Method]
                   #
@@ -1694,6 +1889,8 @@ module Google
                     @list_conversations = ::Gapic::Config::Method.new list_conversations_config
                     answer_query_config = parent_rpcs.answer_query if parent_rpcs.respond_to? :answer_query
                     @answer_query = ::Gapic::Config::Method.new answer_query_config
+                    stream_answer_query_config = parent_rpcs.stream_answer_query if parent_rpcs.respond_to? :stream_answer_query
+                    @stream_answer_query = ::Gapic::Config::Method.new stream_answer_query_config
                     get_answer_config = parent_rpcs.get_answer if parent_rpcs.respond_to? :get_answer
                     @get_answer = ::Gapic::Config::Method.new get_answer_config
                     create_session_config = parent_rpcs.create_session if parent_rpcs.respond_to? :create_session

data/lib/google/cloud/discovery_engine/v1beta/conversational_search_service/rest/service_stub.rb

@@ -353,6 +353,40 @@ module Google
                 end
               end
 
+              ##
+              # Baseline implementation for the stream_answer_query REST call
+              #
+              # @param request_pb [::Google::Cloud::DiscoveryEngine::V1beta::AnswerQueryRequest]
+              #   A request object representing the call parameters. Required.
+              # @param options [::Gapic::CallOptions]
+              #   Overrides the default settings for this call, e.g, timeout, retries etc. Optional.
+              #
+              # @yieldparam chunk [::String] The chunk of data received during server streaming.
+              #
+              # @return [::Gapic::Rest::TransportOperation]
+              def stream_answer_query(request_pb, options = nil, &)
+                raise ::ArgumentError, "request must be provided" if request_pb.nil?
+
+                verb, uri, query_string_params, body = ServiceStub.transcode_stream_answer_query_request request_pb
+                query_string_params = if query_string_params.any?
+                                        query_string_params.to_h { |p| p.split "=", 2 }
+                                      else
+                                        {}
+                                      end
+
+                response = @client_stub.make_http_request(
+                  verb,
+                  uri: uri,
+                  body: body || "",
+                  params: query_string_params,
+                  method_name: "stream_answer_query",
+                  options: options,
+                  is_server_streaming: true,
+                  &
+                )
+                ::Gapic::Rest::TransportOperation.new response
+              end
+
               ##
               # Baseline implementation for the get_answer REST call
               #
@@ -850,6 +884,44 @@ module Google
                 transcoder.transcode request_pb
               end
 
+              ##
+              # @private
+              #
+              # GRPC transcoding helper method for the stream_answer_query REST call
+              #
+              # @param request_pb [::Google::Cloud::DiscoveryEngine::V1beta::AnswerQueryRequest]
+              #   A request object representing the call parameters. Required.
+              # @return [Array(String, [String, nil], Hash{String => String})]
+              #   Uri, Body, Query string parameters
+              def self.transcode_stream_answer_query_request request_pb
+                transcoder = Gapic::Rest::GrpcTranscoder.new
+                                                        .with_bindings(
+                                                          uri_method: :post,
+                                                          uri_template: "/v1beta/{serving_config}:streamAnswer",
+                                                          body: "*",
+                                                          matches: [
+                                                            ["serving_config", %r{^projects/[^/]+/locations/[^/]+/dataStores/[^/]+/servingConfigs/[^/]+/?$}, false]
+                                                          ]
+                                                        )
+                                                        .with_bindings(
+                                                          uri_method: :post,
+                                                          uri_template: "/v1beta/{serving_config}:streamAnswer",
+                                                          body: "*",
+                                                          matches: [
+                                                            ["serving_config", %r{^projects/[^/]+/locations/[^/]+/collections/[^/]+/dataStores/[^/]+/servingConfigs/[^/]+/?$}, false]
+                                                          ]
+                                                        )
+                                                        .with_bindings(
+                                                          uri_method: :post,
+                                                          uri_template: "/v1beta/{serving_config}:streamAnswer",
+                                                          body: "*",
+                                                          matches: [
+                                                            ["serving_config", %r{^projects/[^/]+/locations/[^/]+/collections/[^/]+/engines/[^/]+/servingConfigs/[^/]+/?$}, false]
+                                                          ]
+                                                        )
+                transcoder.transcode request_pb
+              end
+
               ##
               # @private
               #

data/lib/google/cloud/discovery_engine/v1beta/data_store_service/client.rb

@@ -233,11 +233,21 @@ module Google
             #   @param options [::Gapic::CallOptions, ::Hash]
             #     Overrides the default settings for this call, e.g, timeout, retries, etc. Optional.
             #
-            # @overload create_data_store(parent: nil, data_store: nil, data_store_id: nil, create_advanced_site_search: nil, skip_default_schema_creation: nil)
+            # @overload create_data_store(cmek_config_name: nil, disable_cmek: nil, parent: nil, data_store: nil, data_store_id: nil, create_advanced_site_search: nil, skip_default_schema_creation: nil)
             #   Pass arguments to `create_data_store` 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 cmek_config_name [::String]
+            #     Resource name of the CmekConfig to use for protecting this DataStore.
+            #
+            #     Note: The following parameters are mutually exclusive: `cmek_config_name`, `disable_cmek`. At most one of these parameters can be set. If more than one is set, only one will be used, and it is not defined which one.
+            #   @param disable_cmek [::Boolean]
+            #     DataStore without CMEK protections. If a default CmekConfig is set for
+            #     the project, setting this field will override the default CmekConfig as
+            #     well.
+            #
+            #     Note: The following parameters are mutually exclusive: `disable_cmek`, `cmek_config_name`. At most one of these parameters can be set. If more than one is set, only one will be used, and it is not defined which one.
             #   @param parent [::String]
             #     Required. The parent resource name, such as
             #     `projects/{project}/locations/{location}/collections/{collection}`.
@@ -853,6 +863,7 @@ module Google
             #    *  `:initial_delay` (*type:* `Numeric`) - The initial delay in seconds.
             #    *  `:max_delay` (*type:* `Numeric`) - The max delay in seconds.
             #    *  `:multiplier` (*type:* `Numeric`) - The incremental backoff multiplier.
+            #    *  `:jitter` (*type:* `Numeric`) - The jitter in seconds. Default: 1.0.
             #    *  `:retry_codes` (*type:* `Array<String>`) - The error codes that should
             #       trigger a retry.
             #   @return [::Hash]
@@ -936,6 +947,7 @@ module Google
               #      *  `:initial_delay` (*type:* `Numeric`) - The initial delay in seconds.
               #      *  `:max_delay` (*type:* `Numeric`) - The max delay in seconds.
               #      *  `:multiplier` (*type:* `Numeric`) - The incremental backoff multiplier.
+              #      *  `:jitter` (*type:* `Numeric`) - The jitter in seconds. Default: 1.0.
               #      *  `:retry_codes` (*type:* `Array<String>`) - The error codes that should
               #         trigger a retry.
               #

data/lib/google/cloud/discovery_engine/v1beta/data_store_service/credentials.rb

@@ -26,7 +26,9 @@ module Google
           # Credentials for the DataStoreService API.
           class Credentials < ::Google::Auth::Credentials
             self.scope = [
-              "https://www.googleapis.com/auth/cloud-platform"
+              "https://www.googleapis.com/auth/cloud-platform",
+              "https://www.googleapis.com/auth/discoveryengine.readwrite",
+              "https://www.googleapis.com/auth/discoveryengine.serving.readwrite"
             ]
             self.env_vars = [
               "GOOGLE_CLOUD_CREDENTIALS",

data/lib/google/cloud/discovery_engine/v1beta/data_store_service/operations.rb

@@ -703,6 +703,7 @@ module Google
             #    *  `:initial_delay` (*type:* `Numeric`) - The initial delay in seconds.
             #    *  `:max_delay` (*type:* `Numeric`) - The max delay in seconds.
             #    *  `:multiplier` (*type:* `Numeric`) - The incremental backoff multiplier.
+            #    *  `:jitter` (*type:* `Numeric`) - The jitter in seconds. Default: 1.0.
             #    *  `:retry_codes` (*type:* `Array<String>`) - The error codes that should
             #       trigger a retry.
             #   @return [::Hash]
@@ -786,6 +787,7 @@ module Google
               #      *  `:initial_delay` (*type:* `Numeric`) - The initial delay in seconds.
               #      *  `:max_delay` (*type:* `Numeric`) - The max delay in seconds.
               #      *  `:multiplier` (*type:* `Numeric`) - The incremental backoff multiplier.
+              #      *  `:jitter` (*type:* `Numeric`) - The jitter in seconds. Default: 1.0.
               #      *  `:retry_codes` (*type:* `Array<String>`) - The error codes that should
               #         trigger a retry.
               #

data/lib/google/cloud/discovery_engine/v1beta/data_store_service/paths.rb

@@ -24,6 +24,47 @@ module Google
         module DataStoreService
           # Path helper methods for the DataStoreService API.
           module Paths
+            ##
+            # Create a fully-qualified CmekConfig resource string.
+            #
+            # @overload cmek_config_path(project:, location:)
+            #   The resource will be in the following format:
+            #
+            #   `projects/{project}/locations/{location}/cmekConfig`
+            #
+            #   @param project [String]
+            #   @param location [String]
+            #
+            # @overload cmek_config_path(project:, location:, cmek_config:)
+            #   The resource will be in the following format:
+            #
+            #   `projects/{project}/locations/{location}/cmekConfigs/{cmek_config}`
+            #
+            #   @param project [String]
+            #   @param location [String]
+            #   @param cmek_config [String]
+            #
+            # @return [::String]
+            def cmek_config_path **args
+              resources = {
+                "location:project" => (proc do |project:, location:|
+                  raise ::ArgumentError, "project cannot contain /" if project.to_s.include? "/"
+
+                  "projects/#{project}/locations/#{location}/cmekConfig"
+                end),
+                "cmek_config:location:project" => (proc do |project:, location:, cmek_config:|
+                  raise ::ArgumentError, "project cannot contain /" if project.to_s.include? "/"
+                  raise ::ArgumentError, "location cannot contain /" if location.to_s.include? "/"
+
+                  "projects/#{project}/locations/#{location}/cmekConfigs/#{cmek_config}"
+                end)
+              }
+
+              resource = resources[args.keys.sort.join(":")]
+              raise ::ArgumentError, "no resource found for values #{args.keys}" if resource.nil?
+              resource.call(**args)
+            end
+
             ##
             # Create a fully-qualified Collection resource string.
             #
@@ -43,6 +84,50 @@ module Google
               "projects/#{project}/locations/#{location}/collections/#{collection}"
             end
 
+            ##
+            # Create a fully-qualified CryptoKeyVersions resource string.
+            #
+            # The resource will be in the following format:
+            #
+            # `projects/{project}/locations/{location}/keyRings/{key_ring}/cryptoKeys/{crypto_key}/cryptoKeyVersions/{crypto_key_version}`
+            #
+            # @param project [String]
+            # @param location [String]
+            # @param key_ring [String]
+            # @param crypto_key [String]
+            # @param crypto_key_version [String]
+            #
+            # @return [::String]
+            def crypto_key_versions_path project:, location:, key_ring:, crypto_key:, crypto_key_version:
+              raise ::ArgumentError, "project cannot contain /" if project.to_s.include? "/"
+              raise ::ArgumentError, "location cannot contain /" if location.to_s.include? "/"
+              raise ::ArgumentError, "key_ring cannot contain /" if key_ring.to_s.include? "/"
+              raise ::ArgumentError, "crypto_key cannot contain /" if crypto_key.to_s.include? "/"
+
+              "projects/#{project}/locations/#{location}/keyRings/#{key_ring}/cryptoKeys/#{crypto_key}/cryptoKeyVersions/#{crypto_key_version}"
+            end
+
+            ##
+            # Create a fully-qualified CryptoKeys resource string.
+            #
+            # The resource will be in the following format:
+            #
+            # `projects/{project}/locations/{location}/keyRings/{key_ring}/cryptoKeys/{crypto_key}`
+            #
+            # @param project [String]
+            # @param location [String]
+            # @param key_ring [String]
+            # @param crypto_key [String]
+            #
+            # @return [::String]
+            def crypto_keys_path project:, location:, key_ring:, crypto_key:
+              raise ::ArgumentError, "project cannot contain /" if project.to_s.include? "/"
+              raise ::ArgumentError, "location cannot contain /" if location.to_s.include? "/"
+              raise ::ArgumentError, "key_ring cannot contain /" if key_ring.to_s.include? "/"
+
+              "projects/#{project}/locations/#{location}/keyRings/#{key_ring}/cryptoKeys/#{crypto_key}"
+            end
+
             ##
             # Create a fully-qualified DataStore resource string.
             #
@@ -133,6 +218,25 @@ module Google
               resource.call(**args)
             end
 
+            ##
+            # Create a fully-qualified IdentityMappingStore resource string.
+            #
+            # The resource will be in the following format:
+            #
+            # `projects/{project}/locations/{location}/identityMappingStores/{identity_mapping_store}`
+            #
+            # @param project [String]
+            # @param location [String]
+            # @param identity_mapping_store [String]
+            #
+            # @return [::String]
+            def identity_mapping_store_path project:, location:, identity_mapping_store:
+              raise ::ArgumentError, "project cannot contain /" if project.to_s.include? "/"
+              raise ::ArgumentError, "location cannot contain /" if location.to_s.include? "/"
+
+              "projects/#{project}/locations/#{location}/identityMappingStores/#{identity_mapping_store}"
+            end
+
             ##
             # Create a fully-qualified Schema resource string.
             #

data/lib/google/cloud/discovery_engine/v1beta/data_store_service/rest/client.rb

@@ -226,11 +226,21 @@ module Google
               #   @param options [::Gapic::CallOptions, ::Hash]
               #     Overrides the default settings for this call, e.g, timeout, retries etc. Optional.
               #
-              # @overload create_data_store(parent: nil, data_store: nil, data_store_id: nil, create_advanced_site_search: nil, skip_default_schema_creation: nil)
+              # @overload create_data_store(cmek_config_name: nil, disable_cmek: nil, parent: nil, data_store: nil, data_store_id: nil, create_advanced_site_search: nil, skip_default_schema_creation: nil)
               #   Pass arguments to `create_data_store` 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 cmek_config_name [::String]
+              #     Resource name of the CmekConfig to use for protecting this DataStore.
+              #
+              #     Note: The following parameters are mutually exclusive: `cmek_config_name`, `disable_cmek`. At most one of these parameters can be set. If more than one is set, only one will be used, and it is not defined which one.
+              #   @param disable_cmek [::Boolean]
+              #     DataStore without CMEK protections. If a default CmekConfig is set for
+              #     the project, setting this field will override the default CmekConfig as
+              #     well.
+              #
+              #     Note: The following parameters are mutually exclusive: `disable_cmek`, `cmek_config_name`. At most one of these parameters can be set. If more than one is set, only one will be used, and it is not defined which one.
               #   @param parent [::String]
               #     Required. The parent resource name, such as
               #     `projects/{project}/locations/{location}/collections/{collection}`.
@@ -785,6 +795,7 @@ module Google
               #    *  `:initial_delay` (*type:* `Numeric`) - The initial delay in seconds.
               #    *  `:max_delay` (*type:* `Numeric`) - The max delay in seconds.
               #    *  `:multiplier` (*type:* `Numeric`) - The incremental backoff multiplier.
+              #    *  `:jitter` (*type:* `Numeric`) - The jitter in seconds. Default: 1.0.
               #    *  `:retry_codes` (*type:* `Array<String>`) - The error codes that should
               #       trigger a retry.
               #   @return [::Hash]
@@ -857,6 +868,7 @@ module Google
                 #      *  `:initial_delay` (*type:* `Numeric`) - The initial delay in seconds.
                 #      *  `:max_delay` (*type:* `Numeric`) - The max delay in seconds.
                 #      *  `:multiplier` (*type:* `Numeric`) - The incremental backoff multiplier.
+                #      *  `:jitter` (*type:* `Numeric`) - The jitter in seconds. Default: 1.0.
                 #      *  `:retry_codes` (*type:* `Array<String>`) - The error codes that should
                 #         trigger a retry.
                 #