google-cloud-notebooks-v1 0.3.0 → 0.4.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: ed3c67d3e3f49fad38a0556bc58986751bc16c7d25a30729aa8418575603b3d0
-  data.tar.gz: 0c4be1a911656ebcaec3fe41803dd0aee1a5858c82cdb1eb412de48d2fb406dd
+  metadata.gz: 3d4ee3e61545feb3914c66453e7f9b75d7125ff5ba193db4031783d4c835d380
+  data.tar.gz: 18337a4c34add0c1ed20e18dd1be34f447df5cbc7cf35c4a4476043175dec668
 SHA512:
-  metadata.gz: d48a9a060f6e610e0a013369dba8e3fe292ea552048e3687f98d87111aff82d6ffd5d4f08a586f3a049dadb26dbc8c23f684dd4bc9309f5bd4bfe623f37abf08
-  data.tar.gz: 6c403dc29cdd7f3bcd16f7a9638d55695fff2c03851b3788fb2ea3f93aec1c9f01d8a53f2c623b9cbda5a9832a47ae440c23f14c356678a97c990bc219683407
+  metadata.gz: 60ae359568c21c2b4503442d31762b43d3a9ecd51d371a705da5c8ead245c1ba015735c458b83514c88ef477a40d729f269a5e2efbbe91096b88317b4dca9e72
+  data.tar.gz: 836407b41b59b805fd5a0ed0cde569283a0e5e882a1b050569d03139da18282bb7de479ebd9822872df0ce8bd1355f371fae5eb519591620e90f489d1d39f4c1

data/AUTHENTICATION.md

@@ -114,7 +114,7 @@ credentials are discovered.
 To configure your system for this, simply:
 
 1. [Download and install the Cloud SDK](https://cloud.google.com/sdk)
-2. Authenticate using OAuth 2.0 `$ gcloud auth login`
+2. Authenticate using OAuth 2.0 `$ gcloud auth application-default login`
 3. Write code as if already authenticated.
 
 **NOTE:** This is _not_ recommended for running in production. The Cloud SDK

data/README.md

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

data/lib/google/cloud/notebooks/v1/diagnostic_config_pb.rb

@@ -0,0 +1,28 @@
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: google/cloud/notebooks/v1/diagnostic_config.proto
+
+require 'google/protobuf'
+
+require 'google/api/field_behavior_pb'
+
+Google::Protobuf::DescriptorPool.generated_pool.build do
+  add_file("google/cloud/notebooks/v1/diagnostic_config.proto", :syntax => :proto3) do
+    add_message "google.cloud.notebooks.v1.DiagnosticConfig" do
+      optional :gcs_bucket, :string, 1
+      optional :relative_path, :string, 2
+      optional :repair_flag_enabled, :bool, 3
+      optional :packet_capture_flag_enabled, :bool, 4
+      optional :copy_home_files_flag_enabled, :bool, 5
+    end
+  end
+end
+
+module Google
+  module Cloud
+    module Notebooks
+      module V1
+        DiagnosticConfig = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.notebooks.v1.DiagnosticConfig").msgclass
+      end
+    end
+  end
+end

data/lib/google/cloud/notebooks/v1/managed_notebook_service/client.rb

@@ -18,6 +18,8 @@
 
 require "google/cloud/errors"
 require "google/cloud/notebooks/v1/managed_service_pb"
+require "google/cloud/location"
+require "google/iam/v1"
 
 module Google
   module Cloud
@@ -75,6 +77,8 @@ module Google
 
                 default_config.rpcs.create_runtime.timeout = 60.0
 
+                default_config.rpcs.update_runtime.timeout = 60.0
+
                 default_config.rpcs.delete_runtime.timeout = 60.0
 
                 default_config.rpcs.start_runtime.timeout = 60.0
@@ -85,6 +89,8 @@ module Google
 
                 default_config.rpcs.report_runtime_event.timeout = 60.0
 
+                default_config.rpcs.diagnose_runtime.timeout = 60.0
+
                 default_config
               end
               yield @configure if block_given?
@@ -160,6 +166,18 @@ module Google
                 config.endpoint = @config.endpoint
               end
 
+              @location_client = Google::Cloud::Location::Locations::Client.new do |config|
+                config.credentials = credentials
+                config.quota_project = @quota_project_id
+                config.endpoint = @config.endpoint
+              end
+
+              @iam_policy_client = Google::Iam::V1::IAMPolicy::Client.new do |config|
+                config.credentials = credentials
+                config.quota_project = @quota_project_id
+                config.endpoint = @config.endpoint
+              end
+
               @managed_notebook_service_stub = ::Gapic::ServiceStub.new(
                 ::Google::Cloud::Notebooks::V1::ManagedNotebookService::Stub,
                 credentials:  credentials,
@@ -176,6 +194,20 @@ module Google
             #
             attr_reader :operations_client
 
+            ##
+            # Get the associated client for mix-in of the Locations.
+            #
+            # @return [Google::Cloud::Location::Locations::Client]
+            #
+            attr_reader :location_client
+
+            ##
+            # Get the associated client for mix-in of the IAMPolicy.
+            #
+            # @return [Google::Iam::V1::IAMPolicy::Client]
+            #
+            attr_reader :iam_policy_client
+
             # Service calls
 
             ##
@@ -463,6 +495,125 @@ module Google
               raise ::Google::Cloud::Error.from_error(e)
             end
 
+            ##
+            # Update Notebook Runtime configuration.
+            #
+            # @overload update_runtime(request, options = nil)
+            #   Pass arguments to `update_runtime` via a request object, either of type
+            #   {::Google::Cloud::Notebooks::V1::UpdateRuntimeRequest} or an equivalent Hash.
+            #
+            #   @param request [::Google::Cloud::Notebooks::V1::UpdateRuntimeRequest, ::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 update_runtime(runtime: nil, update_mask: nil, request_id: nil)
+            #   Pass arguments to `update_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 runtime [::Google::Cloud::Notebooks::V1::Runtime, ::Hash]
+            #     Required. The Runtime to be updated.
+            #   @param update_mask [::Google::Protobuf::FieldMask, ::Hash]
+            #     Required. Specifies the path, relative to `Runtime`, of
+            #     the field to update. For example, to change the software configuration
+            #     kernels, the `update_mask` parameter would be
+            #     specified as `software_config.kernels`,
+            #     and the `PATCH` request body would specify the new value, as follows:
+            #
+            #         {
+            #           "software_config":{
+            #             "kernels": [{
+            #                'repository':
+            #                'gcr.io/deeplearning-platform-release/pytorch-gpu', 'tag':
+            #                'latest' }],
+            #             }
+            #         }
+            #
+            #
+            #     Currently, only the following fields can be updated:
+            #     - software_config.kernels
+            #     - software_config.post_startup_script
+            #     - software_config.custom_gpu_driver_path
+            #     - software_config.idle_shutdown
+            #     - software_config.idle_shutdown_timeout
+            #     - software_config.disable_terminal
+            #   @param request_id [::String]
+            #     Idempotent request UUID.
+            #
+            # @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/notebooks/v1"
+            #
+            #   # Create a client object. The client can be reused for multiple calls.
+            #   client = Google::Cloud::Notebooks::V1::ManagedNotebookService::Client.new
+            #
+            #   # Create a request. To set request fields, pass in keyword arguments.
+            #   request = Google::Cloud::Notebooks::V1::UpdateRuntimeRequest.new
+            #
+            #   # Call the update_runtime method.
+            #   result = client.update_runtime request
+            #
+            #   # The returned object is of type Gapic::Operation. You can use this
+            #   # object to check the status of an operation, cancel it, or wait
+            #   # for results. Here is how to block until completion:
+            #   result.wait_until_done! timeout: 60
+            #   if result.response?
+            #     p result.response
+            #   else
+            #     puts "Error!"
+            #   end
+            #
+            def update_runtime request, options = nil
+              raise ::ArgumentError, "request must be provided" if request.nil?
+
+              request = ::Gapic::Protobuf.coerce request, to: ::Google::Cloud::Notebooks::V1::UpdateRuntimeRequest
+
+              # 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.update_runtime.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::Notebooks::V1::VERSION
+              metadata[:"x-goog-user-project"] = @quota_project_id if @quota_project_id
+
+              header_params = {}
+              if request.runtime&.name
+                header_params["runtime.name"] = request.runtime.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.update_runtime.timeout,
+                                     metadata:     metadata,
+                                     retry_policy: @config.rpcs.update_runtime.retry_policy
+
+              options.apply_defaults timeout:      @config.timeout,
+                                     metadata:     @config.metadata,
+                                     retry_policy: @config.retry_policy
+
+              @managed_notebook_service_stub.call_rpc :update_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
+
             ##
             # Deletes a single Runtime.
             #
@@ -955,6 +1106,102 @@ module Google
               raise ::Google::Cloud::Error.from_error(e)
             end
 
+            ##
+            # Upgrades a Managed Notebook Runtime to the latest version.
+            #
+            # @overload upgrade_runtime(request, options = nil)
+            #   Pass arguments to `upgrade_runtime` via a request object, either of type
+            #   {::Google::Cloud::Notebooks::V1::UpgradeRuntimeRequest} or an equivalent Hash.
+            #
+            #   @param request [::Google::Cloud::Notebooks::V1::UpgradeRuntimeRequest, ::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 upgrade_runtime(name: nil, request_id: nil)
+            #   Pass arguments to `upgrade_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. Format:
+            #     `projects/{project_id}/locations/{location}/runtimes/{runtime_id}`
+            #   @param request_id [::String]
+            #     Idempotent request UUID.
+            #
+            # @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/notebooks/v1"
+            #
+            #   # Create a client object. The client can be reused for multiple calls.
+            #   client = Google::Cloud::Notebooks::V1::ManagedNotebookService::Client.new
+            #
+            #   # Create a request. To set request fields, pass in keyword arguments.
+            #   request = Google::Cloud::Notebooks::V1::UpgradeRuntimeRequest.new
+            #
+            #   # Call the upgrade_runtime method.
+            #   result = client.upgrade_runtime request
+            #
+            #   # The returned object is of type Gapic::Operation. You can use this
+            #   # object to check the status of an operation, cancel it, or wait
+            #   # for results. Here is how to block until completion:
+            #   result.wait_until_done! timeout: 60
+            #   if result.response?
+            #     p result.response
+            #   else
+            #     puts "Error!"
+            #   end
+            #
+            def upgrade_runtime request, options = nil
+              raise ::ArgumentError, "request must be provided" if request.nil?
+
+              request = ::Gapic::Protobuf.coerce request, to: ::Google::Cloud::Notebooks::V1::UpgradeRuntimeRequest
+
+              # 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.upgrade_runtime.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::Notebooks::V1::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.upgrade_runtime.timeout,
+                                     metadata:     metadata,
+                                     retry_policy: @config.rpcs.upgrade_runtime.retry_policy
+
+              options.apply_defaults timeout:      @config.timeout,
+                                     metadata:     @config.metadata,
+                                     retry_policy: @config.retry_policy
+
+              @managed_notebook_service_stub.call_rpc :upgrade_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
+
             ##
             # Report and process a runtime event.
             #
@@ -1144,6 +1391,102 @@ module Google
               raise ::Google::Cloud::Error.from_error(e)
             end
 
+            ##
+            # Creates a Diagnostic File and runs Diagnostic Tool given a Runtime.
+            #
+            # @overload diagnose_runtime(request, options = nil)
+            #   Pass arguments to `diagnose_runtime` via a request object, either of type
+            #   {::Google::Cloud::Notebooks::V1::DiagnoseRuntimeRequest} or an equivalent Hash.
+            #
+            #   @param request [::Google::Cloud::Notebooks::V1::DiagnoseRuntimeRequest, ::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 diagnose_runtime(name: nil, diagnostic_config: nil)
+            #   Pass arguments to `diagnose_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. Format:
+            #     `projects/{project_id}/locations/{location}/runtimes/{runtimes_id}`
+            #   @param diagnostic_config [::Google::Cloud::Notebooks::V1::DiagnosticConfig, ::Hash]
+            #     Required. Defines flags that are used to run the diagnostic tool
+            #
+            # @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/notebooks/v1"
+            #
+            #   # Create a client object. The client can be reused for multiple calls.
+            #   client = Google::Cloud::Notebooks::V1::ManagedNotebookService::Client.new
+            #
+            #   # Create a request. To set request fields, pass in keyword arguments.
+            #   request = Google::Cloud::Notebooks::V1::DiagnoseRuntimeRequest.new
+            #
+            #   # Call the diagnose_runtime method.
+            #   result = client.diagnose_runtime request
+            #
+            #   # The returned object is of type Gapic::Operation. You can use this
+            #   # object to check the status of an operation, cancel it, or wait
+            #   # for results. Here is how to block until completion:
+            #   result.wait_until_done! timeout: 60
+            #   if result.response?
+            #     p result.response
+            #   else
+            #     puts "Error!"
+            #   end
+            #
+            def diagnose_runtime request, options = nil
+              raise ::ArgumentError, "request must be provided" if request.nil?
+
+              request = ::Gapic::Protobuf.coerce request, to: ::Google::Cloud::Notebooks::V1::DiagnoseRuntimeRequest
+
+              # 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.diagnose_runtime.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::Notebooks::V1::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.diagnose_runtime.timeout,
+                                     metadata:     metadata,
+                                     retry_policy: @config.rpcs.diagnose_runtime.retry_policy
+
+              options.apply_defaults timeout:      @config.timeout,
+                                     metadata:     @config.metadata,
+                                     retry_policy: @config.retry_policy
+
+              @managed_notebook_service_stub.call_rpc :diagnose_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
+
             ##
             # Configuration class for the ManagedNotebookService API.
             #
@@ -1295,6 +1638,11 @@ module Google
                 #
                 attr_reader :create_runtime
                 ##
+                # RPC-specific configuration for `update_runtime`
+                # @return [::Gapic::Config::Method]
+                #
+                attr_reader :update_runtime
+                ##
                 # RPC-specific configuration for `delete_runtime`
                 # @return [::Gapic::Config::Method]
                 #
@@ -1320,6 +1668,11 @@ module Google
                 #
                 attr_reader :reset_runtime
                 ##
+                # RPC-specific configuration for `upgrade_runtime`
+                # @return [::Gapic::Config::Method]
+                #
+                attr_reader :upgrade_runtime
+                ##
                 # RPC-specific configuration for `report_runtime_event`
                 # @return [::Gapic::Config::Method]
                 #
@@ -1329,6 +1682,11 @@ module Google
                 # @return [::Gapic::Config::Method]
                 #
                 attr_reader :refresh_runtime_token_internal
+                ##
+                # RPC-specific configuration for `diagnose_runtime`
+                # @return [::Gapic::Config::Method]
+                #
+                attr_reader :diagnose_runtime
 
                 # @private
                 def initialize parent_rpcs = nil
@@ -1338,6 +1696,8 @@ module Google
                   @get_runtime = ::Gapic::Config::Method.new get_runtime_config
                   create_runtime_config = parent_rpcs.create_runtime if parent_rpcs.respond_to? :create_runtime
                   @create_runtime = ::Gapic::Config::Method.new create_runtime_config
+                  update_runtime_config = parent_rpcs.update_runtime if parent_rpcs.respond_to? :update_runtime
+                  @update_runtime = ::Gapic::Config::Method.new update_runtime_config
                   delete_runtime_config = parent_rpcs.delete_runtime if parent_rpcs.respond_to? :delete_runtime
                   @delete_runtime = ::Gapic::Config::Method.new delete_runtime_config
                   start_runtime_config = parent_rpcs.start_runtime if parent_rpcs.respond_to? :start_runtime
@@ -1348,10 +1708,14 @@ module Google
                   @switch_runtime = ::Gapic::Config::Method.new switch_runtime_config
                   reset_runtime_config = parent_rpcs.reset_runtime if parent_rpcs.respond_to? :reset_runtime
                   @reset_runtime = ::Gapic::Config::Method.new reset_runtime_config
+                  upgrade_runtime_config = parent_rpcs.upgrade_runtime if parent_rpcs.respond_to? :upgrade_runtime
+                  @upgrade_runtime = ::Gapic::Config::Method.new upgrade_runtime_config
                   report_runtime_event_config = parent_rpcs.report_runtime_event if parent_rpcs.respond_to? :report_runtime_event
                   @report_runtime_event = ::Gapic::Config::Method.new report_runtime_event_config
                   refresh_runtime_token_internal_config = parent_rpcs.refresh_runtime_token_internal if parent_rpcs.respond_to? :refresh_runtime_token_internal
                   @refresh_runtime_token_internal = ::Gapic::Config::Method.new refresh_runtime_token_internal_config
+                  diagnose_runtime_config = parent_rpcs.diagnose_runtime if parent_rpcs.respond_to? :diagnose_runtime
+                  @diagnose_runtime = ::Gapic::Config::Method.new diagnose_runtime_config
 
                   yield self if block_given?
                 end

data/lib/google/cloud/notebooks/v1/managed_notebook_service/paths.rb

@@ -24,6 +24,23 @@ module Google
         module ManagedNotebookService
           # Path helper methods for the ManagedNotebookService API.
           module Paths
+            ##
+            # Create a fully-qualified Location resource string.
+            #
+            # The resource will be in the following format:
+            #
+            # `projects/{project}/locations/{location}`
+            #
+            # @param project [String]
+            # @param location [String]
+            #
+            # @return [::String]
+            def location_path project:, location:
+              raise ::ArgumentError, "project cannot contain /" if project.to_s.include? "/"
+
+              "projects/#{project}/locations/#{location}"
+            end
+
             ##
             # Create a fully-qualified Runtime resource string.
             #

data/lib/google/cloud/notebooks/v1/managed_service_pb.rb

@@ -7,9 +7,11 @@ require 'google/api/annotations_pb'
 require 'google/api/client_pb'
 require 'google/api/field_behavior_pb'
 require 'google/api/resource_pb'
+require 'google/cloud/notebooks/v1/diagnostic_config_pb'
 require 'google/cloud/notebooks/v1/event_pb'
 require 'google/cloud/notebooks/v1/runtime_pb'
 require 'google/longrunning/operations_pb'
+require 'google/protobuf/field_mask_pb'
 require 'google/protobuf/timestamp_pb'
 
 Google::Protobuf::DescriptorPool.generated_pool.build do
@@ -55,11 +57,20 @@ Google::Protobuf::DescriptorPool.generated_pool.build do
       optional :name, :string, 1
       optional :request_id, :string, 2
     end
+    add_message "google.cloud.notebooks.v1.UpgradeRuntimeRequest" do
+      optional :name, :string, 1
+      optional :request_id, :string, 2
+    end
     add_message "google.cloud.notebooks.v1.ReportRuntimeEventRequest" do
       optional :name, :string, 1
       optional :vm_id, :string, 2
       optional :event, :message, 3, "google.cloud.notebooks.v1.Event"
     end
+    add_message "google.cloud.notebooks.v1.UpdateRuntimeRequest" do
+      optional :runtime, :message, 1, "google.cloud.notebooks.v1.Runtime"
+      optional :update_mask, :message, 2, "google.protobuf.FieldMask"
+      optional :request_id, :string, 3
+    end
     add_message "google.cloud.notebooks.v1.RefreshRuntimeTokenInternalRequest" do
       optional :name, :string, 1
       optional :vm_id, :string, 2
@@ -68,6 +79,10 @@ Google::Protobuf::DescriptorPool.generated_pool.build do
       optional :access_token, :string, 1
       optional :expire_time, :message, 2, "google.protobuf.Timestamp"
     end
+    add_message "google.cloud.notebooks.v1.DiagnoseRuntimeRequest" do
+      optional :name, :string, 1
+      optional :diagnostic_config, :message, 2, "google.cloud.notebooks.v1.DiagnosticConfig"
+    end
   end
 end
 
@@ -84,9 +99,12 @@ module Google
         StopRuntimeRequest = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.notebooks.v1.StopRuntimeRequest").msgclass
         SwitchRuntimeRequest = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.notebooks.v1.SwitchRuntimeRequest").msgclass
         ResetRuntimeRequest = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.notebooks.v1.ResetRuntimeRequest").msgclass
+        UpgradeRuntimeRequest = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.notebooks.v1.UpgradeRuntimeRequest").msgclass
         ReportRuntimeEventRequest = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.notebooks.v1.ReportRuntimeEventRequest").msgclass
+        UpdateRuntimeRequest = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.notebooks.v1.UpdateRuntimeRequest").msgclass
         RefreshRuntimeTokenInternalRequest = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.notebooks.v1.RefreshRuntimeTokenInternalRequest").msgclass
         RefreshRuntimeTokenInternalResponse = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.notebooks.v1.RefreshRuntimeTokenInternalResponse").msgclass
+        DiagnoseRuntimeRequest = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.notebooks.v1.DiagnoseRuntimeRequest").msgclass
       end
     end
   end

data/lib/google/cloud/notebooks/v1/managed_service_services_pb.rb

@@ -40,6 +40,8 @@ module Google
             rpc :GetRuntime, ::Google::Cloud::Notebooks::V1::GetRuntimeRequest, ::Google::Cloud::Notebooks::V1::Runtime
             # Creates a new Runtime in a given project and location.
             rpc :CreateRuntime, ::Google::Cloud::Notebooks::V1::CreateRuntimeRequest, ::Google::Longrunning::Operation
+            # Update Notebook Runtime configuration.
+            rpc :UpdateRuntime, ::Google::Cloud::Notebooks::V1::UpdateRuntimeRequest, ::Google::Longrunning::Operation
             # Deletes a single Runtime.
             rpc :DeleteRuntime, ::Google::Cloud::Notebooks::V1::DeleteRuntimeRequest, ::Google::Longrunning::Operation
             # Starts a Managed Notebook Runtime.
@@ -58,11 +60,15 @@ module Google
             rpc :SwitchRuntime, ::Google::Cloud::Notebooks::V1::SwitchRuntimeRequest, ::Google::Longrunning::Operation
             # Resets a Managed Notebook Runtime.
             rpc :ResetRuntime, ::Google::Cloud::Notebooks::V1::ResetRuntimeRequest, ::Google::Longrunning::Operation
+            # Upgrades a Managed Notebook Runtime to the latest version.
+            rpc :UpgradeRuntime, ::Google::Cloud::Notebooks::V1::UpgradeRuntimeRequest, ::Google::Longrunning::Operation
             # Report and process a runtime event.
             rpc :ReportRuntimeEvent, ::Google::Cloud::Notebooks::V1::ReportRuntimeEventRequest, ::Google::Longrunning::Operation
             # Gets an access token for the consumer service account that the customer
             # attached to the runtime. Only accessible from the tenant instance.
             rpc :RefreshRuntimeTokenInternal, ::Google::Cloud::Notebooks::V1::RefreshRuntimeTokenInternalRequest, ::Google::Cloud::Notebooks::V1::RefreshRuntimeTokenInternalResponse
+            # Creates a Diagnostic File and runs Diagnostic Tool given a Runtime.
+            rpc :DiagnoseRuntime, ::Google::Cloud::Notebooks::V1::DiagnoseRuntimeRequest, ::Google::Longrunning::Operation
           end
 
           Stub = Service.rpc_stub_class