google-cloud-discovery_engine-v1beta 0.11.0 → 0.13.0

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

@@ -33,6 +33,9 @@ module Google
             # Service for managing {::Google::Cloud::DiscoveryEngine::V1beta::Schema Schema}s.
             #
             class Client
+              # @private
+              API_VERSION = ""
+
               # @private
               DEFAULT_ENDPOINT_TEMPLATE = "discoveryengine.$UNIVERSE_DOMAIN$"
 
@@ -242,12 +245,13 @@ module Google
                 # Customize the options with defaults
                 call_metadata = @config.rpcs.get_schema.metadata.to_h
 
-                # Set x-goog-api-client and x-goog-user-project headers
+                # 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.get_schema.timeout,
@@ -292,9 +296,9 @@ module Google
               #     to return. The service may return fewer than this value.
               #
               #     If unspecified, at most 100
-              #     {::Google::Cloud::DiscoveryEngine::V1beta::Schema Schema}s will be returned.
+              #     {::Google::Cloud::DiscoveryEngine::V1beta::Schema Schema}s are returned.
               #
-              #     The maximum value is 1000; values above 1000 will be coerced to 1000.
+              #     The maximum value is 1000; values above 1000 are set to 1000.
               #   @param page_token [::String]
               #     A page token, received from a previous
               #     {::Google::Cloud::DiscoveryEngine::V1beta::SchemaService::Rest::Client#list_schemas SchemaService.ListSchemas}
@@ -341,12 +345,13 @@ module Google
                 # Customize the options with defaults
                 call_metadata = @config.rpcs.list_schemas.metadata.to_h
 
-                # Set x-goog-api-client and x-goog-user-project headers
+                # 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.list_schemas.timeout,
@@ -392,7 +397,7 @@ module Google
               #     create.
               #   @param schema_id [::String]
               #     Required. The ID to use for the
-              #     {::Google::Cloud::DiscoveryEngine::V1beta::Schema Schema}, which will become the
+              #     {::Google::Cloud::DiscoveryEngine::V1beta::Schema Schema}, which becomes the
               #     final component of the
               #     {::Google::Cloud::DiscoveryEngine::V1beta::Schema#name Schema.name}.
               #
@@ -440,12 +445,13 @@ module Google
                 # Customize the options with defaults
                 call_metadata = @config.rpcs.create_schema.metadata.to_h
 
-                # Set x-goog-api-client and x-goog-user-project headers
+                # 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.create_schema.timeout,
@@ -489,8 +495,8 @@ module Google
               #   @param allow_missing [::Boolean]
               #     If set to true, and the
               #     {::Google::Cloud::DiscoveryEngine::V1beta::Schema Schema} is not found, a new
-              #     {::Google::Cloud::DiscoveryEngine::V1beta::Schema Schema} will be created. In
-              #     this situation, `update_mask` is ignored.
+              #     {::Google::Cloud::DiscoveryEngine::V1beta::Schema Schema} is created. In this
+              #     situation, `update_mask` is ignored.
               # @yield [result, operation] Access the result along with the TransportOperation object
               # @yieldparam result [::Gapic::Operation]
               # @yieldparam operation [::Gapic::Rest::TransportOperation]
@@ -532,12 +538,13 @@ module Google
                 # Customize the options with defaults
                 call_metadata = @config.rpcs.update_schema.metadata.to_h
 
-                # Set x-goog-api-client and x-goog-user-project headers
+                # 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.update_schema.timeout,
@@ -619,12 +626,13 @@ module Google
                 # Customize the options with defaults
                 call_metadata = @config.rpcs.delete_schema.metadata.to_h
 
-                # Set x-goog-api-client and x-goog-user-project headers
+                # 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.delete_schema.timeout,

data/lib/google/cloud/discovery_engine/v1beta/schema_service/rest/operations.rb

@@ -26,6 +26,9 @@ module Google
           module Rest
             # Service that implements Longrunning Operations API.
             class Operations
+              # @private
+              API_VERSION = ""
+
               # @private
               DEFAULT_ENDPOINT_TEMPLATE = "discoveryengine.$UNIVERSE_DOMAIN$"
 
@@ -181,12 +184,13 @@ module Google
                 # Customize the options with defaults
                 call_metadata = @config.rpcs.list_operations.metadata.to_h
 
-                # Set x-goog-api-client and x-goog-user-project headers
+                # 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.list_operations.timeout,
@@ -269,12 +273,13 @@ module Google
                 # Customize the options with defaults
                 call_metadata = @config.rpcs.get_operation.metadata.to_h
 
-                # Set x-goog-api-client and x-goog-user-project headers
+                # 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.get_operation.timeout,
@@ -351,12 +356,13 @@ module Google
                 # Customize the options with defaults
                 call_metadata = @config.rpcs.delete_operation.metadata.to_h
 
-                # Set x-goog-api-client and x-goog-user-project headers
+                # 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.delete_operation.timeout,
@@ -438,12 +444,13 @@ module Google
                 # Customize the options with defaults
                 call_metadata = @config.rpcs.cancel_operation.metadata.to_h
 
-                # Set x-goog-api-client and x-goog-user-project headers
+                # 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.cancel_operation.timeout,
@@ -1009,6 +1016,13 @@ module Google
                                                             ["name", %r{^projects/[^/]+/locations/[^/]+/dataStores/[^/]+/operations/[^/]+/?$}, false]
                                                           ]
                                                         )
+                                                        .with_bindings(
+                                                          uri_method: :get,
+                                                          uri_template: "/v1beta/{name}",
+                                                          matches: [
+                                                            ["name", %r{^projects/[^/]+/locations/[^/]+/evaluations/[^/]+/operations/[^/]+/?$}, false]
+                                                          ]
+                                                        )
                                                         .with_bindings(
                                                           uri_method: :get,
                                                           uri_template: "/v1beta/{name}",
@@ -1016,6 +1030,13 @@ module Google
                                                             ["name", %r{^projects/[^/]+/locations/[^/]+/operations/[^/]+/?$}, false]
                                                           ]
                                                         )
+                                                        .with_bindings(
+                                                          uri_method: :get,
+                                                          uri_template: "/v1beta/{name}",
+                                                          matches: [
+                                                            ["name", %r{^projects/[^/]+/locations/[^/]+/sampleQuerySets/[^/]+/operations/[^/]+/?$}, false]
+                                                          ]
+                                                        )
                                                         .with_bindings(
                                                           uri_method: :get,
                                                           uri_template: "/v1beta/{name}",
@@ -1060,10 +1081,18 @@ module Google
                 transcoder = Gapic::Rest::GrpcTranscoder.new
                                                         .with_bindings(
                                                           uri_method: :post,
-                                                          uri_template: "/v1/{name}:cancel",
+                                                          uri_template: "/v1beta/{name}:cancel",
                                                           body: "*",
                                                           matches: [
-                                                            ["name", %r{^operations(?:/.*)?$}, true]
+                                                            ["name", %r{^projects/[^/]+/locations/[^/]+/collections/[^/]+/dataStores/[^/]+/branches/[^/]+/operations/[^/]+/?$}, false]
+                                                          ]
+                                                        )
+                                                        .with_bindings(
+                                                          uri_method: :post,
+                                                          uri_template: "/v1beta/{name}:cancel",
+                                                          body: "*",
+                                                          matches: [
+                                                            ["name", %r{^projects/[^/]+/locations/[^/]+/dataStores/[^/]+/branches/[^/]+/operations/[^/]+/?$}, false]
                                                           ]
                                                         )
                 transcoder.transcode request_pb

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

@@ -31,6 +31,9 @@ module Google
           # Service for search.
           #
           class Client
+            # @private
+            API_VERSION = ""
+
             # @private
             DEFAULT_ENDPOINT_TEMPLATE = "discoveryengine.$UNIVERSE_DOMAIN$"
 
@@ -244,7 +247,10 @@ module Google
             #
             #     If this field is negative, an  `INVALID_ARGUMENT`  is returned.
             #   @param data_store_specs [::Array<::Google::Cloud::DiscoveryEngine::V1beta::SearchRequest::DataStoreSpec, ::Hash>]
-            #     A list of data store specs to apply on a search call.
+            #     Specs defining dataStores to filter on in a search call and configurations
+            #     for those dataStores. This is only considered for engines with multiple
+            #     dataStores use case. For single dataStore within an engine, they should
+            #     use the specs at the top level.
             #   @param filter [::String]
             #     The filter syntax consists of an expression language for constructing a
             #     predicate from one or more fields of the documents being filtered. Filter
@@ -278,7 +284,9 @@ module Google
             #     The order in which documents are returned. Documents can be ordered by
             #     a field in an {::Google::Cloud::DiscoveryEngine::V1beta::Document Document}
             #     object. Leave it unset if ordered by relevance. `order_by` expression is
-            #     case-sensitive. For more information on ordering, see
+            #     case-sensitive.
+            #
+            #     For more information on ordering for retail search, see
             #     [Ordering](https://cloud.google.com/retail/docs/filter-and-order#order)
             #
             #     If this field is unrecognizable, an `INVALID_ARGUMENT` is returned.
@@ -295,7 +303,7 @@ module Google
             #   @param boost_spec [::Google::Cloud::DiscoveryEngine::V1beta::SearchRequest::BoostSpec, ::Hash]
             #     Boost specification to boost certain documents.
             #     For more information on boosting, see
-            #     [Boosting](https://cloud.google.com/retail/docs/boosting#boost)
+            #     [Boosting](https://cloud.google.com/generative-ai-app-builder/docs/boost-search-results)
             #   @param params [::Hash{::String => ::Google::Protobuf::Value, ::Hash}]
             #     Additional search parameters.
             #
@@ -303,8 +311,7 @@ module Google
             #
             #     * `user_country_code`: string. Default empty. If set to non-empty, results
             #        are restricted or boosted based on the location provided.
-            #        Example:
-            #        user_country_code: "au"
+            #        For example, `user_country_code: "au"`
             #
             #        For available codes see [Country
             #        Codes](https://developers.google.com/custom-search/docs/json_api_reference#countryCodes)
@@ -312,8 +319,7 @@ module Google
             #     * `search_type`: double. Default empty. Enables non-webpage searching
             #        depending on the value. The only valid non-default value is 1,
             #        which enables image searching.
-            #        Example:
-            #        search_type: 1
+            #        For example, `search_type: 1`
             #   @param query_expansion_spec [::Google::Cloud::DiscoveryEngine::V1beta::SearchRequest::QueryExpansionSpec, ::Hash]
             #     The query expansion specification that specifies the conditions under which
             #     query expansion occurs.
@@ -353,20 +359,26 @@ module Google
             #     documents. This overrides
             #     {::Google::Cloud::DiscoveryEngine::V1beta::ServingConfig#ranking_expression ServingConfig.ranking_expression}.
             #     The ranking expression is a single function or multiple functions that are
-            #     joint by "+".
+            #     joined by "+".
+            #
             #       * ranking_expression = function, { " + ", function };
+            #
             #     Supported functions:
+            #
             #       * double * relevance_score
             #       * double * dotProduct(embedding_field_path)
+            #
             #     Function variables:
-            #       `relevance_score`: pre-defined keywords, used for measure relevance
+            #
+            #       * `relevance_score`: pre-defined keywords, used for measure relevance
             #       between query and document.
-            #       `embedding_field_path`: the document embedding field
+            #       * `embedding_field_path`: the document embedding field
             #       used with query embedding vector.
-            #       `dotProduct`: embedding function between embedding_field_path and query
+            #       * `dotProduct`: embedding function between embedding_field_path and query
             #       embedding vector.
             #
             #      Example ranking expression:
+            #
             #        If document has an embedding field doc_embedding, the ranking expression
             #        could be `0.5 * relevance_score + 0.3 * dotProduct(doc_embedding)`.
             #   @param safe_search [::Boolean]
@@ -429,10 +441,11 @@ module Google
               # Customize the options with defaults
               metadata = @config.rpcs.search.metadata.to_h
 
-              # Set x-goog-api-client and x-goog-user-project headers
+              # Set x-goog-api-client, x-goog-user-project and x-goog-api-version 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-api-version"] = API_VERSION unless API_VERSION.empty?
               metadata[:"x-goog-user-project"] = @quota_project_id if @quota_project_id
 
               header_params = {}

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

@@ -33,6 +33,9 @@ module Google
             # Service for search.
             #
             class Client
+              # @private
+              API_VERSION = ""
+
               # @private
               DEFAULT_ENDPOINT_TEMPLATE = "discoveryengine.$UNIVERSE_DOMAIN$"
 
@@ -237,7 +240,10 @@ module Google
               #
               #     If this field is negative, an  `INVALID_ARGUMENT`  is returned.
               #   @param data_store_specs [::Array<::Google::Cloud::DiscoveryEngine::V1beta::SearchRequest::DataStoreSpec, ::Hash>]
-              #     A list of data store specs to apply on a search call.
+              #     Specs defining dataStores to filter on in a search call and configurations
+              #     for those dataStores. This is only considered for engines with multiple
+              #     dataStores use case. For single dataStore within an engine, they should
+              #     use the specs at the top level.
               #   @param filter [::String]
               #     The filter syntax consists of an expression language for constructing a
               #     predicate from one or more fields of the documents being filtered. Filter
@@ -271,7 +277,9 @@ module Google
               #     The order in which documents are returned. Documents can be ordered by
               #     a field in an {::Google::Cloud::DiscoveryEngine::V1beta::Document Document}
               #     object. Leave it unset if ordered by relevance. `order_by` expression is
-              #     case-sensitive. For more information on ordering, see
+              #     case-sensitive.
+              #
+              #     For more information on ordering for retail search, see
               #     [Ordering](https://cloud.google.com/retail/docs/filter-and-order#order)
               #
               #     If this field is unrecognizable, an `INVALID_ARGUMENT` is returned.
@@ -288,7 +296,7 @@ module Google
               #   @param boost_spec [::Google::Cloud::DiscoveryEngine::V1beta::SearchRequest::BoostSpec, ::Hash]
               #     Boost specification to boost certain documents.
               #     For more information on boosting, see
-              #     [Boosting](https://cloud.google.com/retail/docs/boosting#boost)
+              #     [Boosting](https://cloud.google.com/generative-ai-app-builder/docs/boost-search-results)
               #   @param params [::Hash{::String => ::Google::Protobuf::Value, ::Hash}]
               #     Additional search parameters.
               #
@@ -296,8 +304,7 @@ module Google
               #
               #     * `user_country_code`: string. Default empty. If set to non-empty, results
               #        are restricted or boosted based on the location provided.
-              #        Example:
-              #        user_country_code: "au"
+              #        For example, `user_country_code: "au"`
               #
               #        For available codes see [Country
               #        Codes](https://developers.google.com/custom-search/docs/json_api_reference#countryCodes)
@@ -305,8 +312,7 @@ module Google
               #     * `search_type`: double. Default empty. Enables non-webpage searching
               #        depending on the value. The only valid non-default value is 1,
               #        which enables image searching.
-              #        Example:
-              #        search_type: 1
+              #        For example, `search_type: 1`
               #   @param query_expansion_spec [::Google::Cloud::DiscoveryEngine::V1beta::SearchRequest::QueryExpansionSpec, ::Hash]
               #     The query expansion specification that specifies the conditions under which
               #     query expansion occurs.
@@ -346,20 +352,26 @@ module Google
               #     documents. This overrides
               #     {::Google::Cloud::DiscoveryEngine::V1beta::ServingConfig#ranking_expression ServingConfig.ranking_expression}.
               #     The ranking expression is a single function or multiple functions that are
-              #     joint by "+".
+              #     joined by "+".
+              #
               #       * ranking_expression = function, { " + ", function };
+              #
               #     Supported functions:
+              #
               #       * double * relevance_score
               #       * double * dotProduct(embedding_field_path)
+              #
               #     Function variables:
-              #       `relevance_score`: pre-defined keywords, used for measure relevance
+              #
+              #       * `relevance_score`: pre-defined keywords, used for measure relevance
               #       between query and document.
-              #       `embedding_field_path`: the document embedding field
+              #       * `embedding_field_path`: the document embedding field
               #       used with query embedding vector.
-              #       `dotProduct`: embedding function between embedding_field_path and query
+              #       * `dotProduct`: embedding function between embedding_field_path and query
               #       embedding vector.
               #
               #      Example ranking expression:
+              #
               #        If document has an embedding field doc_embedding, the ranking expression
               #        could be `0.5 * relevance_score + 0.3 * dotProduct(doc_embedding)`.
               #   @param safe_search [::Boolean]
@@ -421,12 +433,13 @@ module Google
                 # Customize the options with defaults
                 call_metadata = @config.rpcs.search.metadata.to_h
 
-                # Set x-goog-api-client and x-goog-user-project headers
+                # 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.search.timeout,

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

@@ -31,6 +31,9 @@ module Google
           # Service for search tuning.
           #
           class Client
+            # @private
+            API_VERSION = ""
+
             # @private
             DEFAULT_ENDPOINT_TEMPLATE = "discoveryengine.$UNIVERSE_DOMAIN$"
 
@@ -202,7 +205,7 @@ module Google
             #   @param options [::Gapic::CallOptions, ::Hash]
             #     Overrides the default settings for this call, e.g, timeout, retries, etc. Optional.
             #
-            # @overload train_custom_model(gcs_training_input: nil, data_store: nil, model_type: nil, error_config: nil)
+            # @overload train_custom_model(gcs_training_input: nil, data_store: nil, model_type: nil, error_config: nil, model_id: nil)
             #   Pass arguments to `train_custom_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).
@@ -220,6 +223,8 @@ module Google
             #   @param error_config [::Google::Cloud::DiscoveryEngine::V1beta::ImportErrorConfig, ::Hash]
             #     The desired location of errors incurred during the data ingestion and
             #     training.
+            #   @param model_id [::String]
+            #     If not provided, a UUID will be generated.
             #
             # @yield [response, operation] Access the result along with the RPC operation
             # @yieldparam response [::Gapic::Operation]
@@ -262,10 +267,11 @@ module Google
               # Customize the options with defaults
               metadata = @config.rpcs.train_custom_model.metadata.to_h
 
-              # Set x-goog-api-client and x-goog-user-project headers
+              # Set x-goog-api-client, x-goog-user-project and x-goog-api-version 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-api-version"] = API_VERSION unless API_VERSION.empty?
               metadata[:"x-goog-user-project"] = @quota_project_id if @quota_project_id
 
               header_params = {}
@@ -293,6 +299,95 @@ module Google
               raise ::Google::Cloud::Error.from_error(e)
             end
 
+            ##
+            # Gets a list of all the custom models.
+            #
+            # @overload list_custom_models(request, options = nil)
+            #   Pass arguments to `list_custom_models` via a request object, either of type
+            #   {::Google::Cloud::DiscoveryEngine::V1beta::ListCustomModelsRequest} or an equivalent Hash.
+            #
+            #   @param request [::Google::Cloud::DiscoveryEngine::V1beta::ListCustomModelsRequest, ::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 list_custom_models(data_store: nil)
+            #   Pass arguments to `list_custom_models` 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 data_store [::String]
+            #     Required. The resource name of the parent Data Store, such as
+            #     `projects/*/locations/global/collections/default_collection/dataStores/default_data_store`.
+            #     This field is used to identify the data store where to fetch the models
+            #     from.
+            #
+            # @yield [response, operation] Access the result along with the RPC operation
+            # @yieldparam response [::Google::Cloud::DiscoveryEngine::V1beta::ListCustomModelsResponse]
+            # @yieldparam operation [::GRPC::ActiveCall::Operation]
+            #
+            # @return [::Google::Cloud::DiscoveryEngine::V1beta::ListCustomModelsResponse]
+            #
+            # @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::SearchTuningService::Client.new
+            #
+            #   # Create a request. To set request fields, pass in keyword arguments.
+            #   request = Google::Cloud::DiscoveryEngine::V1beta::ListCustomModelsRequest.new
+            #
+            #   # Call the list_custom_models method.
+            #   result = client.list_custom_models request
+            #
+            #   # The returned object is of type Google::Cloud::DiscoveryEngine::V1beta::ListCustomModelsResponse.
+            #   p result
+            #
+            def list_custom_models request, options = nil
+              raise ::ArgumentError, "request must be provided" if request.nil?
+
+              request = ::Gapic::Protobuf.coerce request, to: ::Google::Cloud::DiscoveryEngine::V1beta::ListCustomModelsRequest
+
+              # 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.list_custom_models.metadata.to_h
+
+              # Set x-goog-api-client, x-goog-user-project and x-goog-api-version 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-api-version"] = API_VERSION unless API_VERSION.empty?
+              metadata[:"x-goog-user-project"] = @quota_project_id if @quota_project_id
+
+              header_params = {}
+              if request.data_store
+                header_params["data_store"] = request.data_store
+              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.list_custom_models.timeout,
+                                     metadata:     metadata,
+                                     retry_policy: @config.rpcs.list_custom_models.retry_policy
+
+              options.apply_defaults timeout:      @config.timeout,
+                                     metadata:     @config.metadata,
+                                     retry_policy: @config.retry_policy
+
+              @search_tuning_service_stub.call_rpc :list_custom_models, 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 SearchTuningService API.
             #
@@ -451,11 +546,18 @@ module Google
                 # @return [::Gapic::Config::Method]
                 #
                 attr_reader :train_custom_model
+                ##
+                # RPC-specific configuration for `list_custom_models`
+                # @return [::Gapic::Config::Method]
+                #
+                attr_reader :list_custom_models
 
                 # @private
                 def initialize parent_rpcs = nil
                   train_custom_model_config = parent_rpcs.train_custom_model if parent_rpcs.respond_to? :train_custom_model
                   @train_custom_model = ::Gapic::Config::Method.new train_custom_model_config
+                  list_custom_models_config = parent_rpcs.list_custom_models if parent_rpcs.respond_to? :list_custom_models
+                  @list_custom_models = ::Gapic::Config::Method.new list_custom_models_config
 
                   yield self if block_given?
                 end