google-cloud-ai_platform-v1 0.55.0 → 0.56.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 292afb58f3cdf4ff9b22dbbc1ca24184910ed35aff12f3b2788d131d594c4f72
-  data.tar.gz: 7a0fd8ad2d6c4cc1708b03a0ad6cb8463ce464a69c8ff3f26c0c9c215a66d0e4
+  metadata.gz: c216c1d77f12c3de06e07ae0561e027d1e13944c6a6188ba452b9ce63ee1f75d
+  data.tar.gz: 46ef4d8af087c9e28b01825dbc28381e2d0a639049af957411a1682e08573260
 SHA512:
-  metadata.gz: e29510b89343e267a882cdf9d2f83ffd901e5317c223472d598065a328291db04c9df721aba7461c7b4a6d687a7fb2481c2e812ad5484bb51602ebe71c40f832
-  data.tar.gz: 8fd987400b2a37917694c9dbafebca57f2cde883519e0c02d9de26fd60c826495c3ff1bfd398cbc8bd830bc81e60d9de7d5cfe4727584fb38588f97ab045a1d7
+  metadata.gz: 1c05f3acbbc6ec0639effe7751b2ddead1fedf02f3ccad6d09c3b71d1683f70824166c30e3a056c27b9f7361d90d015abf3bd65eefe90d13e17a985dfa096ab8
+  data.tar.gz: afcfcbad50fc84c6a2304c6f9d5f84b3847c756be5adbcbe44a9de6226849c3a204c5c23d2b769e9f352ae025ec2f6ed0316113565aa4396b14c739ef3eb1a8d

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

@@ -236,7 +236,7 @@ module Google
             #     Required. The ID to use for this FeatureGroup, which will become the final
             #     component of the FeatureGroup's resource name.
             #
-            #     This value may be up to 60 characters, and valid characters are
+            #     This value may be up to 128 characters, and valid characters are
             #     `[a-z0-9_]`. The first character cannot be a number.
             #
             #     The value must be unique within the project and location.
@@ -850,6 +850,109 @@ module Google
               raise ::Google::Cloud::Error.from_error(e)
             end
 
+            ##
+            # Creates a batch of Features in a given FeatureGroup.
+            #
+            # @overload batch_create_features(request, options = nil)
+            #   Pass arguments to `batch_create_features` via a request object, either of type
+            #   {::Google::Cloud::AIPlatform::V1::BatchCreateFeaturesRequest} or an equivalent Hash.
+            #
+            #   @param request [::Google::Cloud::AIPlatform::V1::BatchCreateFeaturesRequest, ::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 batch_create_features(parent: nil, requests: nil)
+            #   Pass arguments to `batch_create_features` 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 parent [::String]
+            #     Required. The resource name of the EntityType/FeatureGroup to create the
+            #     batch of Features under. Format:
+            #     `projects/{project}/locations/{location}/featurestores/{featurestore}/entityTypes/{entity_type}`
+            #     `projects/{project}/locations/{location}/featureGroups/{feature_group}`
+            #   @param requests [::Array<::Google::Cloud::AIPlatform::V1::CreateFeatureRequest, ::Hash>]
+            #     Required. The request message specifying the Features to create. All
+            #     Features must be created under the same parent EntityType / FeatureGroup.
+            #     The `parent` field in each child request message can be omitted. If
+            #     `parent` is set in a child request, then the value must match the `parent`
+            #     value in this request message.
+            #
+            # @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/ai_platform/v1"
+            #
+            #   # Create a client object. The client can be reused for multiple calls.
+            #   client = Google::Cloud::AIPlatform::V1::FeatureRegistryService::Client.new
+            #
+            #   # Create a request. To set request fields, pass in keyword arguments.
+            #   request = Google::Cloud::AIPlatform::V1::BatchCreateFeaturesRequest.new
+            #
+            #   # Call the batch_create_features method.
+            #   result = client.batch_create_features 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 batch_create_features request, options = nil
+              raise ::ArgumentError, "request must be provided" if request.nil?
+
+              request = ::Gapic::Protobuf.coerce request, to: ::Google::Cloud::AIPlatform::V1::BatchCreateFeaturesRequest
+
+              # 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.batch_create_features.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::AIPlatform::V1::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.parent
+                header_params["parent"] = request.parent
+              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.batch_create_features.timeout,
+                                     metadata:     metadata,
+                                     retry_policy: @config.rpcs.batch_create_features.retry_policy
+
+              options.apply_defaults timeout:      @config.timeout,
+                                     metadata:     @config.metadata,
+                                     retry_policy: @config.retry_policy
+
+              @feature_registry_service_stub.call_rpc :batch_create_features, 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
+
             ##
             # Gets details of a single Feature.
             #
@@ -1486,6 +1589,11 @@ module Google
                 #
                 attr_reader :create_feature
                 ##
+                # RPC-specific configuration for `batch_create_features`
+                # @return [::Gapic::Config::Method]
+                #
+                attr_reader :batch_create_features
+                ##
                 # RPC-specific configuration for `get_feature`
                 # @return [::Gapic::Config::Method]
                 #
@@ -1520,6 +1628,8 @@ module Google
                   @delete_feature_group = ::Gapic::Config::Method.new delete_feature_group_config
                   create_feature_config = parent_rpcs.create_feature if parent_rpcs.respond_to? :create_feature
                   @create_feature = ::Gapic::Config::Method.new create_feature_config
+                  batch_create_features_config = parent_rpcs.batch_create_features if parent_rpcs.respond_to? :batch_create_features
+                  @batch_create_features = ::Gapic::Config::Method.new batch_create_features_config
                   get_feature_config = parent_rpcs.get_feature if parent_rpcs.respond_to? :get_feature
                   @get_feature = ::Gapic::Config::Method.new get_feature_config
                   list_features_config = parent_rpcs.list_features if parent_rpcs.respond_to? :list_features

data/lib/google/cloud/ai_platform/v1/feature_registry_service/rest/client.rb

@@ -231,7 +231,7 @@ module Google
               #     Required. The ID to use for this FeatureGroup, which will become the final
               #     component of the FeatureGroup's resource name.
               #
-              #     This value may be up to 60 characters, and valid characters are
+              #     This value may be up to 128 characters, and valid characters are
               #     `[a-z0-9_]`. The first character cannot be a number.
               #
               #     The value must be unique within the project and location.
@@ -803,6 +803,102 @@ module Google
                 raise ::Google::Cloud::Error.from_error(e)
               end
 
+              ##
+              # Creates a batch of Features in a given FeatureGroup.
+              #
+              # @overload batch_create_features(request, options = nil)
+              #   Pass arguments to `batch_create_features` via a request object, either of type
+              #   {::Google::Cloud::AIPlatform::V1::BatchCreateFeaturesRequest} or an equivalent Hash.
+              #
+              #   @param request [::Google::Cloud::AIPlatform::V1::BatchCreateFeaturesRequest, ::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 batch_create_features(parent: nil, requests: nil)
+              #   Pass arguments to `batch_create_features` 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 parent [::String]
+              #     Required. The resource name of the EntityType/FeatureGroup to create the
+              #     batch of Features under. Format:
+              #     `projects/{project}/locations/{location}/featurestores/{featurestore}/entityTypes/{entity_type}`
+              #     `projects/{project}/locations/{location}/featureGroups/{feature_group}`
+              #   @param requests [::Array<::Google::Cloud::AIPlatform::V1::CreateFeatureRequest, ::Hash>]
+              #     Required. The request message specifying the Features to create. All
+              #     Features must be created under the same parent EntityType / FeatureGroup.
+              #     The `parent` field in each child request message can be omitted. If
+              #     `parent` is set in a child request, then the value must match the `parent`
+              #     value in this request message.
+              # @yield [result, operation] Access the result along with the TransportOperation object
+              # @yieldparam result [::Gapic::Operation]
+              # @yieldparam operation [::Gapic::Rest::TransportOperation]
+              #
+              # @return [::Gapic::Operation]
+              #
+              # @raise [::Google::Cloud::Error] if the REST call is aborted.
+              #
+              # @example Basic example
+              #   require "google/cloud/ai_platform/v1"
+              #
+              #   # Create a client object. The client can be reused for multiple calls.
+              #   client = Google::Cloud::AIPlatform::V1::FeatureRegistryService::Rest::Client.new
+              #
+              #   # Create a request. To set request fields, pass in keyword arguments.
+              #   request = Google::Cloud::AIPlatform::V1::BatchCreateFeaturesRequest.new
+              #
+              #   # Call the batch_create_features method.
+              #   result = client.batch_create_features 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 batch_create_features request, options = nil
+                raise ::ArgumentError, "request must be provided" if request.nil?
+
+                request = ::Gapic::Protobuf.coerce request, to: ::Google::Cloud::AIPlatform::V1::BatchCreateFeaturesRequest
+
+                # 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.batch_create_features.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::AIPlatform::V1::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.batch_create_features.timeout,
+                                       metadata:     call_metadata,
+                                       retry_policy: @config.rpcs.batch_create_features.retry_policy
+
+                options.apply_defaults timeout:      @config.timeout,
+                                       metadata:     @config.metadata,
+                                       retry_policy: @config.retry_policy
+
+                @feature_registry_service_stub.batch_create_features request, options do |result, operation|
+                  result = ::Gapic::Operation.new result, @operations_client, options: options
+                  yield result, operation if block_given?
+                  return result
+                end
+              rescue ::Gapic::Rest::Error => e
+                raise ::Google::Cloud::Error.from_error(e)
+              end
+
               ##
               # Gets details of a single Feature.
               #
@@ -1398,6 +1494,11 @@ module Google
                   #
                   attr_reader :create_feature
                   ##
+                  # RPC-specific configuration for `batch_create_features`
+                  # @return [::Gapic::Config::Method]
+                  #
+                  attr_reader :batch_create_features
+                  ##
                   # RPC-specific configuration for `get_feature`
                   # @return [::Gapic::Config::Method]
                   #
@@ -1432,6 +1533,8 @@ module Google
                     @delete_feature_group = ::Gapic::Config::Method.new delete_feature_group_config
                     create_feature_config = parent_rpcs.create_feature if parent_rpcs.respond_to? :create_feature
                     @create_feature = ::Gapic::Config::Method.new create_feature_config
+                    batch_create_features_config = parent_rpcs.batch_create_features if parent_rpcs.respond_to? :batch_create_features
+                    @batch_create_features = ::Gapic::Config::Method.new batch_create_features_config
                     get_feature_config = parent_rpcs.get_feature if parent_rpcs.respond_to? :get_feature
                     @get_feature = ::Gapic::Config::Method.new get_feature_config
                     list_features_config = parent_rpcs.list_features if parent_rpcs.respond_to? :list_features

data/lib/google/cloud/ai_platform/v1/feature_registry_service/rest/service_stub.rb

@@ -289,6 +289,44 @@ module Google
                 result
               end
 
+              ##
+              # Baseline implementation for the batch_create_features REST call
+              #
+              # @param request_pb [::Google::Cloud::AIPlatform::V1::BatchCreateFeaturesRequest]
+              #   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.
+              #
+              # @yield [result, operation] Access the result along with the TransportOperation object
+              # @yieldparam result [::Google::Longrunning::Operation]
+              # @yieldparam operation [::Gapic::Rest::TransportOperation]
+              #
+              # @return [::Google::Longrunning::Operation]
+              #   A result object deserialized from the server's reply
+              def batch_create_features request_pb, options = nil
+                raise ::ArgumentError, "request must be provided" if request_pb.nil?
+
+                verb, uri, query_string_params, body = ServiceStub.transcode_batch_create_features_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,
+                  options: options
+                )
+                operation = ::Gapic::Rest::TransportOperation.new response
+                result = ::Google::Longrunning::Operation.decode_json response.body, ignore_unknown_fields: true
+
+                yield result, operation if block_given?
+                result
+              end
+
               ##
               # Baseline implementation for the get_feature REST call
               #
@@ -570,6 +608,28 @@ module Google
                 transcoder.transcode request_pb
               end
 
+              ##
+              # @private
+              #
+              # GRPC transcoding helper method for the batch_create_features REST call
+              #
+              # @param request_pb [::Google::Cloud::AIPlatform::V1::BatchCreateFeaturesRequest]
+              #   A request object representing the call parameters. Required.
+              # @return [Array(String, [String, nil], Hash{String => String})]
+              #   Uri, Body, Query string parameters
+              def self.transcode_batch_create_features_request request_pb
+                transcoder = Gapic::Rest::GrpcTranscoder.new
+                                                        .with_bindings(
+                                                          uri_method: :post,
+                                                          uri_template: "/v1/{parent}/features:batchCreate",
+                                                          body: "*",
+                                                          matches: [
+                                                            ["parent", %r{^projects/[^/]+/locations/[^/]+/featureGroups/[^/]+/?$}, false]
+                                                          ]
+                                                        )
+                transcoder.transcode request_pb
+              end
+
               ##
               # @private
               #

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

@@ -1413,15 +1413,16 @@ module Google
             #   the default parameter values, pass an empty Hash as a request object (see above).
             #
             #   @param parent [::String]
-            #     Required. The resource name of the EntityType to create the batch of
-            #     Features under. Format:
+            #     Required. The resource name of the EntityType/FeatureGroup to create the
+            #     batch of Features under. Format:
             #     `projects/{project}/locations/{location}/featurestores/{featurestore}/entityTypes/{entity_type}`
+            #     `projects/{project}/locations/{location}/featureGroups/{feature_group}`
             #   @param requests [::Array<::Google::Cloud::AIPlatform::V1::CreateFeatureRequest, ::Hash>]
             #     Required. The request message specifying the Features to create. All
-            #     Features must be created under the same parent EntityType. The `parent`
-            #     field in each child request message can be omitted. If `parent` is set in a
-            #     child request, then the value must match the `parent` value in this request
-            #     message.
+            #     Features must be created under the same parent EntityType / FeatureGroup.
+            #     The `parent` field in each child request message can be omitted. If
+            #     `parent` is set in a child request, then the value must match the `parent`
+            #     value in this request message.
             #
             # @yield [response, operation] Access the result along with the RPC operation
             # @yieldparam response [::Gapic::Operation]

data/lib/google/cloud/ai_platform/v1/featurestore_service/rest/client.rb

@@ -1331,15 +1331,16 @@ module Google
               #   the default parameter values, pass an empty Hash as a request object (see above).
               #
               #   @param parent [::String]
-              #     Required. The resource name of the EntityType to create the batch of
-              #     Features under. Format:
+              #     Required. The resource name of the EntityType/FeatureGroup to create the
+              #     batch of Features under. Format:
               #     `projects/{project}/locations/{location}/featurestores/{featurestore}/entityTypes/{entity_type}`
+              #     `projects/{project}/locations/{location}/featureGroups/{feature_group}`
               #   @param requests [::Array<::Google::Cloud::AIPlatform::V1::CreateFeatureRequest, ::Hash>]
               #     Required. The request message specifying the Features to create. All
-              #     Features must be created under the same parent EntityType. The `parent`
-              #     field in each child request message can be omitted. If `parent` is set in a
-              #     child request, then the value must match the `parent` value in this request
-              #     message.
+              #     Features must be created under the same parent EntityType / FeatureGroup.
+              #     The `parent` field in each child request message can be omitted. If
+              #     `parent` is set in a child request, then the value must match the `parent`
+              #     value in this request message.
               # @yield [result, operation] Access the result along with the TransportOperation object
               # @yieldparam result [::Gapic::Operation]
               # @yieldparam operation [::Gapic::Rest::TransportOperation]

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

@@ -1351,6 +1351,103 @@ module Google
               raise ::Google::Cloud::Error.from_error(e)
             end
 
+            ##
+            # Stops a NotebookRuntime.
+            #
+            # @overload stop_notebook_runtime(request, options = nil)
+            #   Pass arguments to `stop_notebook_runtime` via a request object, either of type
+            #   {::Google::Cloud::AIPlatform::V1::StopNotebookRuntimeRequest} or an equivalent Hash.
+            #
+            #   @param request [::Google::Cloud::AIPlatform::V1::StopNotebookRuntimeRequest, ::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 stop_notebook_runtime(name: nil)
+            #   Pass arguments to `stop_notebook_runtime` 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 NotebookRuntime resource to be stopped.
+            #     Instead of checking whether the name is in valid NotebookRuntime resource
+            #     name format, directly throw NotFound exception if there is no such
+            #     NotebookRuntime in spanner.
+            #
+            # @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/ai_platform/v1"
+            #
+            #   # Create a client object. The client can be reused for multiple calls.
+            #   client = Google::Cloud::AIPlatform::V1::NotebookService::Client.new
+            #
+            #   # Create a request. To set request fields, pass in keyword arguments.
+            #   request = Google::Cloud::AIPlatform::V1::StopNotebookRuntimeRequest.new
+            #
+            #   # Call the stop_notebook_runtime method.
+            #   result = client.stop_notebook_runtime 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 stop_notebook_runtime request, options = nil
+              raise ::ArgumentError, "request must be provided" if request.nil?
+
+              request = ::Gapic::Protobuf.coerce request, to: ::Google::Cloud::AIPlatform::V1::StopNotebookRuntimeRequest
+
+              # 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.stop_notebook_runtime.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::AIPlatform::V1::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.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.stop_notebook_runtime.timeout,
+                                     metadata:     metadata,
+                                     retry_policy: @config.rpcs.stop_notebook_runtime.retry_policy
+
+              options.apply_defaults timeout:      @config.timeout,
+                                     metadata:     @config.metadata,
+                                     retry_policy: @config.retry_policy
+
+              @notebook_service_stub.call_rpc :stop_notebook_runtime, 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
+
             ##
             # Creates a NotebookExecutionJob.
             #
@@ -1966,6 +2063,11 @@ module Google
                 #
                 attr_reader :start_notebook_runtime
                 ##
+                # RPC-specific configuration for `stop_notebook_runtime`
+                # @return [::Gapic::Config::Method]
+                #
+                attr_reader :stop_notebook_runtime
+                ##
                 # RPC-specific configuration for `create_notebook_execution_job`
                 # @return [::Gapic::Config::Method]
                 #
@@ -2010,6 +2112,8 @@ module Google
                   @upgrade_notebook_runtime = ::Gapic::Config::Method.new upgrade_notebook_runtime_config
                   start_notebook_runtime_config = parent_rpcs.start_notebook_runtime if parent_rpcs.respond_to? :start_notebook_runtime
                   @start_notebook_runtime = ::Gapic::Config::Method.new start_notebook_runtime_config
+                  stop_notebook_runtime_config = parent_rpcs.stop_notebook_runtime if parent_rpcs.respond_to? :stop_notebook_runtime
+                  @stop_notebook_runtime = ::Gapic::Config::Method.new stop_notebook_runtime_config
                   create_notebook_execution_job_config = parent_rpcs.create_notebook_execution_job if parent_rpcs.respond_to? :create_notebook_execution_job
                   @create_notebook_execution_job = ::Gapic::Config::Method.new create_notebook_execution_job_config
                   get_notebook_execution_job_config = parent_rpcs.get_notebook_execution_job if parent_rpcs.respond_to? :get_notebook_execution_job

data/lib/google/cloud/ai_platform/v1/notebook_service/rest/client.rb

@@ -1269,6 +1269,96 @@ module Google
                 raise ::Google::Cloud::Error.from_error(e)
               end
 
+              ##
+              # Stops a NotebookRuntime.
+              #
+              # @overload stop_notebook_runtime(request, options = nil)
+              #   Pass arguments to `stop_notebook_runtime` via a request object, either of type
+              #   {::Google::Cloud::AIPlatform::V1::StopNotebookRuntimeRequest} or an equivalent Hash.
+              #
+              #   @param request [::Google::Cloud::AIPlatform::V1::StopNotebookRuntimeRequest, ::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 stop_notebook_runtime(name: nil)
+              #   Pass arguments to `stop_notebook_runtime` 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 NotebookRuntime resource to be stopped.
+              #     Instead of checking whether the name is in valid NotebookRuntime resource
+              #     name format, directly throw NotFound exception if there is no such
+              #     NotebookRuntime in spanner.
+              # @yield [result, operation] Access the result along with the TransportOperation object
+              # @yieldparam result [::Gapic::Operation]
+              # @yieldparam operation [::Gapic::Rest::TransportOperation]
+              #
+              # @return [::Gapic::Operation]
+              #
+              # @raise [::Google::Cloud::Error] if the REST call is aborted.
+              #
+              # @example Basic example
+              #   require "google/cloud/ai_platform/v1"
+              #
+              #   # Create a client object. The client can be reused for multiple calls.
+              #   client = Google::Cloud::AIPlatform::V1::NotebookService::Rest::Client.new
+              #
+              #   # Create a request. To set request fields, pass in keyword arguments.
+              #   request = Google::Cloud::AIPlatform::V1::StopNotebookRuntimeRequest.new
+              #
+              #   # Call the stop_notebook_runtime method.
+              #   result = client.stop_notebook_runtime 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 stop_notebook_runtime request, options = nil
+                raise ::ArgumentError, "request must be provided" if request.nil?
+
+                request = ::Gapic::Protobuf.coerce request, to: ::Google::Cloud::AIPlatform::V1::StopNotebookRuntimeRequest
+
+                # 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.stop_notebook_runtime.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::AIPlatform::V1::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.stop_notebook_runtime.timeout,
+                                       metadata:     call_metadata,
+                                       retry_policy: @config.rpcs.stop_notebook_runtime.retry_policy
+
+                options.apply_defaults timeout:      @config.timeout,
+                                       metadata:     @config.metadata,
+                                       retry_policy: @config.retry_policy
+
+                @notebook_service_stub.stop_notebook_runtime request, options do |result, operation|
+                  result = ::Gapic::Operation.new result, @operations_client, options: options
+                  yield result, operation if block_given?
+                  return result
+                end
+              rescue ::Gapic::Rest::Error => e
+                raise ::Google::Cloud::Error.from_error(e)
+              end
+
               ##
               # Creates a NotebookExecutionJob.
               #
@@ -1843,6 +1933,11 @@ module Google
                   #
                   attr_reader :start_notebook_runtime
                   ##
+                  # RPC-specific configuration for `stop_notebook_runtime`
+                  # @return [::Gapic::Config::Method]
+                  #
+                  attr_reader :stop_notebook_runtime
+                  ##
                   # RPC-specific configuration for `create_notebook_execution_job`
                   # @return [::Gapic::Config::Method]
                   #
@@ -1887,6 +1982,8 @@ module Google
                     @upgrade_notebook_runtime = ::Gapic::Config::Method.new upgrade_notebook_runtime_config
                     start_notebook_runtime_config = parent_rpcs.start_notebook_runtime if parent_rpcs.respond_to? :start_notebook_runtime
                     @start_notebook_runtime = ::Gapic::Config::Method.new start_notebook_runtime_config
+                    stop_notebook_runtime_config = parent_rpcs.stop_notebook_runtime if parent_rpcs.respond_to? :stop_notebook_runtime
+                    @stop_notebook_runtime = ::Gapic::Config::Method.new stop_notebook_runtime_config
                     create_notebook_execution_job_config = parent_rpcs.create_notebook_execution_job if parent_rpcs.respond_to? :create_notebook_execution_job
                     @create_notebook_execution_job = ::Gapic::Config::Method.new create_notebook_execution_job_config
                     get_notebook_execution_job_config = parent_rpcs.get_notebook_execution_job if parent_rpcs.respond_to? :get_notebook_execution_job