google-cloud-discovery_engine-v1beta 0.10.0 → 0.11.0

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

@@ -530,7 +530,7 @@ module Google
             #   @param options [::Gapic::CallOptions, ::Hash]
             #     Overrides the default settings for this call, e.g, timeout, retries, etc. Optional.
             #
-            # @overload update_document(document: nil, allow_missing: nil)
+            # @overload update_document(document: nil, allow_missing: nil, update_mask: nil)
             #   Pass arguments to `update_document` 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).
@@ -551,6 +551,9 @@ module Google
             #     {::Google::Cloud::DiscoveryEngine::V1beta::Document Document} is not found, a
             #     new {::Google::Cloud::DiscoveryEngine::V1beta::Document Document} will be
             #     created.
+            #   @param update_mask [::Google::Protobuf::FieldMask, ::Hash]
+            #     Indicates which fields in the provided imported 'document' to update. If
+            #     not set, will by default update all fields.
             #
             # @yield [response, operation] Access the result along with the RPC operation
             # @yieldparam response [::Google::Cloud::DiscoveryEngine::V1beta::Document]
@@ -729,7 +732,7 @@ module Google
             #   @param options [::Gapic::CallOptions, ::Hash]
             #     Overrides the default settings for this call, e.g, timeout, retries, etc. Optional.
             #
-            # @overload import_documents(inline_source: nil, gcs_source: nil, bigquery_source: nil, parent: nil, error_config: nil, reconciliation_mode: nil, auto_generate_ids: nil, id_field: nil)
+            # @overload import_documents(inline_source: nil, gcs_source: nil, bigquery_source: nil, fhir_store_source: nil, spanner_source: nil, cloud_sql_source: nil, firestore_source: nil, bigtable_source: nil, parent: nil, error_config: nil, reconciliation_mode: nil, update_mask: nil, auto_generate_ids: nil, id_field: nil)
             #   Pass arguments to `import_documents` 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).
@@ -740,6 +743,16 @@ module Google
             #     Cloud Storage location for the input content.
             #   @param bigquery_source [::Google::Cloud::DiscoveryEngine::V1beta::BigQuerySource, ::Hash]
             #     BigQuery input source.
+            #   @param fhir_store_source [::Google::Cloud::DiscoveryEngine::V1beta::FhirStoreSource, ::Hash]
+            #     FhirStore input source.
+            #   @param spanner_source [::Google::Cloud::DiscoveryEngine::V1beta::SpannerSource, ::Hash]
+            #     Spanner input source.
+            #   @param cloud_sql_source [::Google::Cloud::DiscoveryEngine::V1beta::CloudSqlSource, ::Hash]
+            #     Cloud SQL input source.
+            #   @param firestore_source [::Google::Cloud::DiscoveryEngine::V1beta::FirestoreSource, ::Hash]
+            #     Firestore input source.
+            #   @param bigtable_source [::Google::Cloud::DiscoveryEngine::V1beta::BigtableSource, ::Hash]
+            #     Cloud Bigtable input source.
             #   @param parent [::String]
             #     Required. The parent branch resource name, such as
             #     `projects/{project}/locations/{location}/collections/{collection}/dataStores/{data_store}/branches/{branch}`.
@@ -750,6 +763,9 @@ module Google
             #     The mode of reconciliation between existing documents and the documents to
             #     be imported. Defaults to
             #     {::Google::Cloud::DiscoveryEngine::V1beta::ImportDocumentsRequest::ReconciliationMode::INCREMENTAL ReconciliationMode.INCREMENTAL}.
+            #   @param update_mask [::Google::Protobuf::FieldMask, ::Hash]
+            #     Indicates which fields in the provided imported documents to update. If
+            #     not set, the default is to update all fields.
             #   @param auto_generate_ids [::Boolean]
             #     Whether to automatically generate IDs for the documents if absent.
             #
@@ -764,45 +780,54 @@ module Google
             #     {::Google::Cloud::DiscoveryEngine::V1beta::ImportDocumentsRequest#id_field id_field},
             #     otherwise, documents without IDs fail to be imported.
             #
-            #     Only set this field when using
-            #     {::Google::Cloud::DiscoveryEngine::V1beta::GcsSource GcsSource} or
-            #     {::Google::Cloud::DiscoveryEngine::V1beta::BigQuerySource BigQuerySource}, and
-            #     when
+            #     Supported data sources:
+            #
+            #     * {::Google::Cloud::DiscoveryEngine::V1beta::GcsSource GcsSource}.
             #     {::Google::Cloud::DiscoveryEngine::V1beta::GcsSource#data_schema GcsSource.data_schema}
-            #     or
+            #     must be `custom` or `csv`. Otherwise, an INVALID_ARGUMENT error is thrown.
+            #     * {::Google::Cloud::DiscoveryEngine::V1beta::BigQuerySource BigQuerySource}.
             #     {::Google::Cloud::DiscoveryEngine::V1beta::BigQuerySource#data_schema BigQuerySource.data_schema}
-            #     is `custom` or `csv`. Otherwise, an INVALID_ARGUMENT error is thrown.
+            #     must be `custom` or `csv`. Otherwise, an INVALID_ARGUMENT error is thrown.
+            #     * {::Google::Cloud::DiscoveryEngine::V1beta::SpannerSource SpannerSource}.
+            #     * {::Google::Cloud::DiscoveryEngine::V1beta::CloudSqlSource CloudSqlSource}.
+            #     * {::Google::Cloud::DiscoveryEngine::V1beta::FirestoreSource FirestoreSource}.
+            #     * {::Google::Cloud::DiscoveryEngine::V1beta::BigtableSource BigtableSource}.
             #   @param id_field [::String]
-            #     The field in the Cloud Storage and BigQuery sources that indicates the
-            #     unique IDs of the documents.
+            #     The field indicates the ID field or column to be used as unique IDs of
+            #     the documents.
             #
             #     For {::Google::Cloud::DiscoveryEngine::V1beta::GcsSource GcsSource} it is the
             #     key of the JSON field. For instance, `my_id` for JSON `{"my_id":
-            #     "some_uuid"}`. For
-            #     {::Google::Cloud::DiscoveryEngine::V1beta::BigQuerySource BigQuerySource} it is
-            #     the column name of the BigQuery table where the unique ids are stored.
+            #     "some_uuid"}`. For others, it may be the column name of the table where the
+            #     unique ids are stored.
             #
-            #     The values of the JSON field or the BigQuery column are used as the
+            #     The values of the JSON field or the table column are used as the
             #     {::Google::Cloud::DiscoveryEngine::V1beta::Document#id Document.id}s. The JSON
-            #     field or the BigQuery column must be of string type, and the values must be
+            #     field or the table column must be of string type, and the values must be
             #     set as valid strings conform to
             #     [RFC-1034](https://tools.ietf.org/html/rfc1034) with 1-63 characters.
             #     Otherwise, documents without valid IDs fail to be imported.
             #
-            #     Only set this field when using
-            #     {::Google::Cloud::DiscoveryEngine::V1beta::GcsSource GcsSource} or
-            #     {::Google::Cloud::DiscoveryEngine::V1beta::BigQuerySource BigQuerySource}, and
-            #     when
-            #     {::Google::Cloud::DiscoveryEngine::V1beta::GcsSource#data_schema GcsSource.data_schema}
-            #     or
-            #     {::Google::Cloud::DiscoveryEngine::V1beta::BigQuerySource#data_schema BigQuerySource.data_schema}
-            #     is `custom`. And only set this field when
+            #     Only set this field when
             #     {::Google::Cloud::DiscoveryEngine::V1beta::ImportDocumentsRequest#auto_generate_ids auto_generate_ids}
             #     is unset or set as `false`. Otherwise, an INVALID_ARGUMENT error is thrown.
             #
             #     If it is unset, a default value `_id` is used when importing from the
             #     allowed data sources.
             #
+            #     Supported data sources:
+            #
+            #     * {::Google::Cloud::DiscoveryEngine::V1beta::GcsSource GcsSource}.
+            #     {::Google::Cloud::DiscoveryEngine::V1beta::GcsSource#data_schema GcsSource.data_schema}
+            #     must be `custom` or `csv`. Otherwise, an INVALID_ARGUMENT error is thrown.
+            #     * {::Google::Cloud::DiscoveryEngine::V1beta::BigQuerySource BigQuerySource}.
+            #     {::Google::Cloud::DiscoveryEngine::V1beta::BigQuerySource#data_schema BigQuerySource.data_schema}
+            #     must be `custom` or `csv`. Otherwise, an INVALID_ARGUMENT error is thrown.
+            #     * {::Google::Cloud::DiscoveryEngine::V1beta::SpannerSource SpannerSource}.
+            #     * {::Google::Cloud::DiscoveryEngine::V1beta::CloudSqlSource CloudSqlSource}.
+            #     * {::Google::Cloud::DiscoveryEngine::V1beta::FirestoreSource FirestoreSource}.
+            #     * {::Google::Cloud::DiscoveryEngine::V1beta::BigtableSource BigtableSource}.
+            #
             # @yield [response, operation] Access the result along with the RPC operation
             # @yieldparam response [::Gapic::Operation]
             # @yieldparam operation [::GRPC::ActiveCall::Operation]

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

@@ -126,6 +126,27 @@ module Google
               resource.call(**args)
             end
 
+            ##
+            # Create a fully-qualified FhirStore resource string.
+            #
+            # The resource will be in the following format:
+            #
+            # `projects/{project}/locations/{location}/datasets/{dataset}/fhirStores/{fhir_store}`
+            #
+            # @param project [String]
+            # @param location [String]
+            # @param dataset [String]
+            # @param fhir_store [String]
+            #
+            # @return [::String]
+            def fhir_store_path project:, location:, dataset:, fhir_store:
+              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}/fhirStores/#{fhir_store}"
+            end
+
             extend self
           end
         end

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

@@ -502,7 +502,7 @@ module Google
               #   @param options [::Gapic::CallOptions, ::Hash]
               #     Overrides the default settings for this call, e.g, timeout, retries etc. Optional.
               #
-              # @overload update_document(document: nil, allow_missing: nil)
+              # @overload update_document(document: nil, allow_missing: nil, update_mask: nil)
               #   Pass arguments to `update_document` 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).
@@ -523,6 +523,9 @@ module Google
               #     {::Google::Cloud::DiscoveryEngine::V1beta::Document Document} is not found, a
               #     new {::Google::Cloud::DiscoveryEngine::V1beta::Document Document} will be
               #     created.
+              #   @param update_mask [::Google::Protobuf::FieldMask, ::Hash]
+              #     Indicates which fields in the provided imported 'document' to update. If
+              #     not set, will by default update all fields.
               # @yield [result, operation] Access the result along with the TransportOperation object
               # @yieldparam result [::Google::Cloud::DiscoveryEngine::V1beta::Document]
               # @yieldparam operation [::Gapic::Rest::TransportOperation]
@@ -687,7 +690,7 @@ module Google
               #   @param options [::Gapic::CallOptions, ::Hash]
               #     Overrides the default settings for this call, e.g, timeout, retries etc. Optional.
               #
-              # @overload import_documents(inline_source: nil, gcs_source: nil, bigquery_source: nil, parent: nil, error_config: nil, reconciliation_mode: nil, auto_generate_ids: nil, id_field: nil)
+              # @overload import_documents(inline_source: nil, gcs_source: nil, bigquery_source: nil, fhir_store_source: nil, spanner_source: nil, cloud_sql_source: nil, firestore_source: nil, bigtable_source: nil, parent: nil, error_config: nil, reconciliation_mode: nil, update_mask: nil, auto_generate_ids: nil, id_field: nil)
               #   Pass arguments to `import_documents` 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).
@@ -698,6 +701,16 @@ module Google
               #     Cloud Storage location for the input content.
               #   @param bigquery_source [::Google::Cloud::DiscoveryEngine::V1beta::BigQuerySource, ::Hash]
               #     BigQuery input source.
+              #   @param fhir_store_source [::Google::Cloud::DiscoveryEngine::V1beta::FhirStoreSource, ::Hash]
+              #     FhirStore input source.
+              #   @param spanner_source [::Google::Cloud::DiscoveryEngine::V1beta::SpannerSource, ::Hash]
+              #     Spanner input source.
+              #   @param cloud_sql_source [::Google::Cloud::DiscoveryEngine::V1beta::CloudSqlSource, ::Hash]
+              #     Cloud SQL input source.
+              #   @param firestore_source [::Google::Cloud::DiscoveryEngine::V1beta::FirestoreSource, ::Hash]
+              #     Firestore input source.
+              #   @param bigtable_source [::Google::Cloud::DiscoveryEngine::V1beta::BigtableSource, ::Hash]
+              #     Cloud Bigtable input source.
               #   @param parent [::String]
               #     Required. The parent branch resource name, such as
               #     `projects/{project}/locations/{location}/collections/{collection}/dataStores/{data_store}/branches/{branch}`.
@@ -708,6 +721,9 @@ module Google
               #     The mode of reconciliation between existing documents and the documents to
               #     be imported. Defaults to
               #     {::Google::Cloud::DiscoveryEngine::V1beta::ImportDocumentsRequest::ReconciliationMode::INCREMENTAL ReconciliationMode.INCREMENTAL}.
+              #   @param update_mask [::Google::Protobuf::FieldMask, ::Hash]
+              #     Indicates which fields in the provided imported documents to update. If
+              #     not set, the default is to update all fields.
               #   @param auto_generate_ids [::Boolean]
               #     Whether to automatically generate IDs for the documents if absent.
               #
@@ -722,44 +738,53 @@ module Google
               #     {::Google::Cloud::DiscoveryEngine::V1beta::ImportDocumentsRequest#id_field id_field},
               #     otherwise, documents without IDs fail to be imported.
               #
-              #     Only set this field when using
-              #     {::Google::Cloud::DiscoveryEngine::V1beta::GcsSource GcsSource} or
-              #     {::Google::Cloud::DiscoveryEngine::V1beta::BigQuerySource BigQuerySource}, and
-              #     when
+              #     Supported data sources:
+              #
+              #     * {::Google::Cloud::DiscoveryEngine::V1beta::GcsSource GcsSource}.
               #     {::Google::Cloud::DiscoveryEngine::V1beta::GcsSource#data_schema GcsSource.data_schema}
-              #     or
+              #     must be `custom` or `csv`. Otherwise, an INVALID_ARGUMENT error is thrown.
+              #     * {::Google::Cloud::DiscoveryEngine::V1beta::BigQuerySource BigQuerySource}.
               #     {::Google::Cloud::DiscoveryEngine::V1beta::BigQuerySource#data_schema BigQuerySource.data_schema}
-              #     is `custom` or `csv`. Otherwise, an INVALID_ARGUMENT error is thrown.
+              #     must be `custom` or `csv`. Otherwise, an INVALID_ARGUMENT error is thrown.
+              #     * {::Google::Cloud::DiscoveryEngine::V1beta::SpannerSource SpannerSource}.
+              #     * {::Google::Cloud::DiscoveryEngine::V1beta::CloudSqlSource CloudSqlSource}.
+              #     * {::Google::Cloud::DiscoveryEngine::V1beta::FirestoreSource FirestoreSource}.
+              #     * {::Google::Cloud::DiscoveryEngine::V1beta::BigtableSource BigtableSource}.
               #   @param id_field [::String]
-              #     The field in the Cloud Storage and BigQuery sources that indicates the
-              #     unique IDs of the documents.
+              #     The field indicates the ID field or column to be used as unique IDs of
+              #     the documents.
               #
               #     For {::Google::Cloud::DiscoveryEngine::V1beta::GcsSource GcsSource} it is the
               #     key of the JSON field. For instance, `my_id` for JSON `{"my_id":
-              #     "some_uuid"}`. For
-              #     {::Google::Cloud::DiscoveryEngine::V1beta::BigQuerySource BigQuerySource} it is
-              #     the column name of the BigQuery table where the unique ids are stored.
+              #     "some_uuid"}`. For others, it may be the column name of the table where the
+              #     unique ids are stored.
               #
-              #     The values of the JSON field or the BigQuery column are used as the
+              #     The values of the JSON field or the table column are used as the
               #     {::Google::Cloud::DiscoveryEngine::V1beta::Document#id Document.id}s. The JSON
-              #     field or the BigQuery column must be of string type, and the values must be
+              #     field or the table column must be of string type, and the values must be
               #     set as valid strings conform to
               #     [RFC-1034](https://tools.ietf.org/html/rfc1034) with 1-63 characters.
               #     Otherwise, documents without valid IDs fail to be imported.
               #
-              #     Only set this field when using
-              #     {::Google::Cloud::DiscoveryEngine::V1beta::GcsSource GcsSource} or
-              #     {::Google::Cloud::DiscoveryEngine::V1beta::BigQuerySource BigQuerySource}, and
-              #     when
-              #     {::Google::Cloud::DiscoveryEngine::V1beta::GcsSource#data_schema GcsSource.data_schema}
-              #     or
-              #     {::Google::Cloud::DiscoveryEngine::V1beta::BigQuerySource#data_schema BigQuerySource.data_schema}
-              #     is `custom`. And only set this field when
+              #     Only set this field when
               #     {::Google::Cloud::DiscoveryEngine::V1beta::ImportDocumentsRequest#auto_generate_ids auto_generate_ids}
               #     is unset or set as `false`. Otherwise, an INVALID_ARGUMENT error is thrown.
               #
               #     If it is unset, a default value `_id` is used when importing from the
               #     allowed data sources.
+              #
+              #     Supported data sources:
+              #
+              #     * {::Google::Cloud::DiscoveryEngine::V1beta::GcsSource GcsSource}.
+              #     {::Google::Cloud::DiscoveryEngine::V1beta::GcsSource#data_schema GcsSource.data_schema}
+              #     must be `custom` or `csv`. Otherwise, an INVALID_ARGUMENT error is thrown.
+              #     * {::Google::Cloud::DiscoveryEngine::V1beta::BigQuerySource BigQuerySource}.
+              #     {::Google::Cloud::DiscoveryEngine::V1beta::BigQuerySource#data_schema BigQuerySource.data_schema}
+              #     must be `custom` or `csv`. Otherwise, an INVALID_ARGUMENT error is thrown.
+              #     * {::Google::Cloud::DiscoveryEngine::V1beta::SpannerSource SpannerSource}.
+              #     * {::Google::Cloud::DiscoveryEngine::V1beta::CloudSqlSource CloudSqlSource}.
+              #     * {::Google::Cloud::DiscoveryEngine::V1beta::FirestoreSource FirestoreSource}.
+              #     * {::Google::Cloud::DiscoveryEngine::V1beta::BigtableSource BigtableSource}.
               # @yield [result, operation] Access the result along with the TransportOperation object
               # @yieldparam result [::Gapic::Operation]
               # @yieldparam operation [::Gapic::Rest::TransportOperation]

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

@@ -683,6 +683,281 @@ module Google
               raise ::Google::Cloud::Error.from_error(e)
             end
 
+            ##
+            # Pauses the training of an existing engine. Only applicable if
+            # {::Google::Cloud::DiscoveryEngine::V1beta::SolutionType SolutionType} is
+            # {::Google::Cloud::DiscoveryEngine::V1beta::SolutionType::SOLUTION_TYPE_RECOMMENDATION SOLUTION_TYPE_RECOMMENDATION}.
+            #
+            # @overload pause_engine(request, options = nil)
+            #   Pass arguments to `pause_engine` via a request object, either of type
+            #   {::Google::Cloud::DiscoveryEngine::V1beta::PauseEngineRequest} or an equivalent Hash.
+            #
+            #   @param request [::Google::Cloud::DiscoveryEngine::V1beta::PauseEngineRequest, ::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 pause_engine(name: nil)
+            #   Pass arguments to `pause_engine` 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 name [::String]
+            #     Required. The name of the engine to pause.
+            #     Format:
+            #     `projects/{project_number}/locations/{location_id}/collections/{collection_id}/engines/{engine_id}`
+            #
+            # @yield [response, operation] Access the result along with the RPC operation
+            # @yieldparam response [::Google::Cloud::DiscoveryEngine::V1beta::Engine]
+            # @yieldparam operation [::GRPC::ActiveCall::Operation]
+            #
+            # @return [::Google::Cloud::DiscoveryEngine::V1beta::Engine]
+            #
+            # @raise [::Google::Cloud::Error] if the RPC 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::EngineService::Client.new
+            #
+            #   # Create a request. To set request fields, pass in keyword arguments.
+            #   request = Google::Cloud::DiscoveryEngine::V1beta::PauseEngineRequest.new
+            #
+            #   # Call the pause_engine method.
+            #   result = client.pause_engine request
+            #
+            #   # The returned object is of type Google::Cloud::DiscoveryEngine::V1beta::Engine.
+            #   p result
+            #
+            def pause_engine request, options = nil
+              raise ::ArgumentError, "request must be provided" if request.nil?
+
+              request = ::Gapic::Protobuf.coerce request, to: ::Google::Cloud::DiscoveryEngine::V1beta::PauseEngineRequest
+
+              # 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.pause_engine.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::DiscoveryEngine::V1beta::VERSION
+              metadata[:"x-goog-user-project"] = @quota_project_id if @quota_project_id
+
+              header_params = {}
+              if request.name
+                header_params["name"] = request.name
+              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.pause_engine.timeout,
+                                     metadata:     metadata,
+                                     retry_policy: @config.rpcs.pause_engine.retry_policy
+
+              options.apply_defaults timeout:      @config.timeout,
+                                     metadata:     @config.metadata,
+                                     retry_policy: @config.retry_policy
+
+              @engine_service_stub.call_rpc :pause_engine, 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
+
+            ##
+            # Resumes the training of an existing engine. Only applicable if
+            # {::Google::Cloud::DiscoveryEngine::V1beta::SolutionType SolutionType} is
+            # {::Google::Cloud::DiscoveryEngine::V1beta::SolutionType::SOLUTION_TYPE_RECOMMENDATION SOLUTION_TYPE_RECOMMENDATION}.
+            #
+            # @overload resume_engine(request, options = nil)
+            #   Pass arguments to `resume_engine` via a request object, either of type
+            #   {::Google::Cloud::DiscoveryEngine::V1beta::ResumeEngineRequest} or an equivalent Hash.
+            #
+            #   @param request [::Google::Cloud::DiscoveryEngine::V1beta::ResumeEngineRequest, ::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 resume_engine(name: nil)
+            #   Pass arguments to `resume_engine` 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 name [::String]
+            #     Required. The name of the engine to resume.
+            #     Format:
+            #     `projects/{project_number}/locations/{location_id}/collections/{collection_id}/engines/{engine_id}`
+            #
+            # @yield [response, operation] Access the result along with the RPC operation
+            # @yieldparam response [::Google::Cloud::DiscoveryEngine::V1beta::Engine]
+            # @yieldparam operation [::GRPC::ActiveCall::Operation]
+            #
+            # @return [::Google::Cloud::DiscoveryEngine::V1beta::Engine]
+            #
+            # @raise [::Google::Cloud::Error] if the RPC 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::EngineService::Client.new
+            #
+            #   # Create a request. To set request fields, pass in keyword arguments.
+            #   request = Google::Cloud::DiscoveryEngine::V1beta::ResumeEngineRequest.new
+            #
+            #   # Call the resume_engine method.
+            #   result = client.resume_engine request
+            #
+            #   # The returned object is of type Google::Cloud::DiscoveryEngine::V1beta::Engine.
+            #   p result
+            #
+            def resume_engine request, options = nil
+              raise ::ArgumentError, "request must be provided" if request.nil?
+
+              request = ::Gapic::Protobuf.coerce request, to: ::Google::Cloud::DiscoveryEngine::V1beta::ResumeEngineRequest
+
+              # 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.resume_engine.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::DiscoveryEngine::V1beta::VERSION
+              metadata[:"x-goog-user-project"] = @quota_project_id if @quota_project_id
+
+              header_params = {}
+              if request.name
+                header_params["name"] = request.name
+              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.resume_engine.timeout,
+                                     metadata:     metadata,
+                                     retry_policy: @config.rpcs.resume_engine.retry_policy
+
+              options.apply_defaults timeout:      @config.timeout,
+                                     metadata:     @config.metadata,
+                                     retry_policy: @config.retry_policy
+
+              @engine_service_stub.call_rpc :resume_engine, 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
+
+            ##
+            # Tunes an existing engine. Only applicable if
+            # {::Google::Cloud::DiscoveryEngine::V1beta::SolutionType SolutionType} is
+            # {::Google::Cloud::DiscoveryEngine::V1beta::SolutionType::SOLUTION_TYPE_RECOMMENDATION SOLUTION_TYPE_RECOMMENDATION}.
+            #
+            # @overload tune_engine(request, options = nil)
+            #   Pass arguments to `tune_engine` via a request object, either of type
+            #   {::Google::Cloud::DiscoveryEngine::V1beta::TuneEngineRequest} or an equivalent Hash.
+            #
+            #   @param request [::Google::Cloud::DiscoveryEngine::V1beta::TuneEngineRequest, ::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 tune_engine(name: nil)
+            #   Pass arguments to `tune_engine` 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 name [::String]
+            #     Required. The resource name of the engine to tune.
+            #     Format:
+            #     `projects/{project_number}/locations/{location_id}/collections/{collection_id}/engines/{engine_id}`
+            #
+            # @yield [response, operation] Access the result along with the RPC operation
+            # @yieldparam response [::Gapic::Operation]
+            # @yieldparam operation [::GRPC::ActiveCall::Operation]
+            #
+            # @return [::Gapic::Operation]
+            #
+            # @raise [::Google::Cloud::Error] if the RPC 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::EngineService::Client.new
+            #
+            #   # Create a request. To set request fields, pass in keyword arguments.
+            #   request = Google::Cloud::DiscoveryEngine::V1beta::TuneEngineRequest.new
+            #
+            #   # Call the tune_engine method.
+            #   result = client.tune_engine request
+            #
+            #   # The returned object is of type Gapic::Operation. You can use it to
+            #   # check the status of an operation, cancel it, or wait for results.
+            #   # Here is how to wait for a response.
+            #   result.wait_until_done! timeout: 60
+            #   if result.response?
+            #     p result.response
+            #   else
+            #     puts "No response received."
+            #   end
+            #
+            def tune_engine request, options = nil
+              raise ::ArgumentError, "request must be provided" if request.nil?
+
+              request = ::Gapic::Protobuf.coerce request, to: ::Google::Cloud::DiscoveryEngine::V1beta::TuneEngineRequest
+
+              # 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.tune_engine.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::DiscoveryEngine::V1beta::VERSION
+              metadata[:"x-goog-user-project"] = @quota_project_id if @quota_project_id
+
+              header_params = {}
+              if request.name
+                header_params["name"] = request.name
+              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.tune_engine.timeout,
+                                     metadata:     metadata,
+                                     retry_policy: @config.rpcs.tune_engine.retry_policy
+
+              options.apply_defaults timeout:      @config.timeout,
+                                     metadata:     @config.metadata,
+                                     retry_policy: @config.retry_policy
+
+              @engine_service_stub.call_rpc :tune_engine, request, options: options do |response, operation|
+                response = ::Gapic::Operation.new response, @operations_client, options: options
+                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 EngineService API.
             #
@@ -861,6 +1136,21 @@ module Google
                 # @return [::Gapic::Config::Method]
                 #
                 attr_reader :list_engines
+                ##
+                # RPC-specific configuration for `pause_engine`
+                # @return [::Gapic::Config::Method]
+                #
+                attr_reader :pause_engine
+                ##
+                # RPC-specific configuration for `resume_engine`
+                # @return [::Gapic::Config::Method]
+                #
+                attr_reader :resume_engine
+                ##
+                # RPC-specific configuration for `tune_engine`
+                # @return [::Gapic::Config::Method]
+                #
+                attr_reader :tune_engine
 
                 # @private
                 def initialize parent_rpcs = nil
@@ -874,6 +1164,12 @@ module Google
                   @get_engine = ::Gapic::Config::Method.new get_engine_config
                   list_engines_config = parent_rpcs.list_engines if parent_rpcs.respond_to? :list_engines
                   @list_engines = ::Gapic::Config::Method.new list_engines_config
+                  pause_engine_config = parent_rpcs.pause_engine if parent_rpcs.respond_to? :pause_engine
+                  @pause_engine = ::Gapic::Config::Method.new pause_engine_config
+                  resume_engine_config = parent_rpcs.resume_engine if parent_rpcs.respond_to? :resume_engine
+                  @resume_engine = ::Gapic::Config::Method.new resume_engine_config
+                  tune_engine_config = parent_rpcs.tune_engine if parent_rpcs.respond_to? :tune_engine
+                  @tune_engine = ::Gapic::Config::Method.new tune_engine_config
 
                   yield self if block_given?
                 end