feat(ai_platform-v1): Support for workbench runtime configuration and…

… custom kernel on NotebookExecutionJob (#27723)
googleapis · Dec 4, 2024 · e92291d · e92291d
1 parent 7e0b6ad
commit e92291d
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 2 deletions.
diff --git a/google-cloud-ai_platform-v1/lib/google/cloud/aiplatform/v1/notebook_execution_job_pb.rb b/google-cloud-ai_platform-v1/lib/google/cloud/aiplatform/v1/notebook_execution_job_pb.rb
diff --git a/google-cloud-ai_platform-v1/proto_docs/google/cloud/aiplatform/v1/notebook_execution_job.rb b/google-cloud-ai_platform-v1/proto_docs/google/cloud/aiplatform/v1/notebook_execution_job.rb
@@ -48,6 +48,9 @@ module V1
         # @!attribute [rw] service_account
         #   @return [::String]
         #     The service account to run the execution as.
+        # @!attribute [rw] workbench_runtime
+        #   @return [::Google::Cloud::AIPlatform::V1::NotebookExecutionJob::WorkbenchRuntime]
+        #     The Workbench runtime configuration to use for the notebook execution.
         # @!attribute [r] name
         #   @return [::String]
         #     Output only. The resource name of this NotebookExecutionJob. Format:
@@ -90,11 +93,16 @@ module V1
         #     See https://goo.gl/xmQnxf for more information and examples of labels.
         #     System reserved label keys are prefixed with "aiplatform.googleapis.com/"
         #     and are immutable.
+        # @!attribute [rw] kernel_name
+        #   @return [::String]
+        #     The name of the kernel to use during notebook execution. If unset, the
+        #     default kernel is used.
         # @!attribute [rw] encryption_spec
         #   @return [::Google::Cloud::AIPlatform::V1::EncryptionSpec]
         #     Customer-managed encryption key spec for the notebook execution job.
         #     This field is auto-populated if the
-        #     [NotebookService.NotebookRuntimeTemplate][] has an encryption spec.
+        #     {::Google::Cloud::AIPlatform::V1::NotebookRuntimeTemplate NotebookRuntimeTemplate}
+        #     has an encryption spec.
         class NotebookExecutionJob
           include ::Google::Protobuf::MessageExts
           extend ::Google::Protobuf::MessageExts::ClassMethods
@@ -152,6 +160,12 @@ class CustomEnvironmentSpec
             extend ::Google::Protobuf::MessageExts::ClassMethods
           end
 
+          # Configuration for a Workbench Instances-based environment.
+          class WorkbenchRuntime
+            include ::Google::Protobuf::MessageExts
+            extend ::Google::Protobuf::MessageExts::ClassMethods
+          end
+
           # @!attribute [rw] key
           #   @return [::String]
           # @!attribute [rw] value