feat: add workbench_runtime and kernel_name to NotebookExecutionJob (#…

…7881) docs: A comment for field `encryption_spec` in message `.google.cloud.aiplatform.v1.NotebookExecutionJob` is changed PiperOrigin-RevId: 702775459 Source-Link: googleapis/googleapis@e865ba0 Source-Link: googleapis/googleapis-gen@6e54e78 Copy-Tag: eyJwIjoiQWlQbGF0Zm9ybS8uT3dsQm90LnlhbWwiLCJoIjoiNmU1NGU3ODJmOWNlYjBmMmY5MjA4ZTVjZjAwZWM3Y2JlMzNmY2Y5MyJ9
googleapis · Dec 6, 2024 · 6cd78f9 · 6cd78f9
1 parent 28acbc3
commit 6cd78f9
Show file tree

Hide file tree

Showing 4 changed files with 126 additions and 4 deletions.
diff --git a/AiPlatform/metadata/V1/NotebookExecutionJob.php b/AiPlatform/metadata/V1/NotebookExecutionJob.php
diff --git a/AiPlatform/src/V1/NotebookExecutionJob.php b/AiPlatform/src/V1/NotebookExecutionJob.php
diff --git a/AiPlatform/src/V1/NotebookExecutionJob/WorkbenchRuntime.php b/AiPlatform/src/V1/NotebookExecutionJob/WorkbenchRuntime.php
diff --git a/AiPlatform/tests/Unit/V1/Client/NotebookServiceClientTest.php b/AiPlatform/tests/Unit/V1/Client/NotebookServiceClientTest.php
@@ -299,13 +299,15 @@ public function createNotebookExecutionJobTest()
         $name = 'name3373707';
         $displayName = 'displayName1615086568';
         $scheduleResourceName = 'scheduleResourceName-627484556';
+        $kernelName = 'kernelName-497416883';
         $expectedResponse = new NotebookExecutionJob();
         $expectedResponse->setNotebookRuntimeTemplateResourceName($notebookRuntimeTemplateResourceName);
         $expectedResponse->setGcsOutputUri($gcsOutputUri);
         $expectedResponse->setExecutionUser($executionUser);
         $expectedResponse->setName($name);
         $expectedResponse->setDisplayName($displayName);
         $expectedResponse->setScheduleResourceName($scheduleResourceName);
+        $expectedResponse->setKernelName($kernelName);
         $anyResponse = new Any();
         $anyResponse->setValue($expectedResponse->serializeToString());
         $completeOperation = new Operation();
@@ -955,13 +957,15 @@ public function getNotebookExecutionJobTest()
         $name2 = 'name2-1052831874';
         $displayName = 'displayName1615086568';
         $scheduleResourceName = 'scheduleResourceName-627484556';
+        $kernelName = 'kernelName-497416883';
         $expectedResponse = new NotebookExecutionJob();
         $expectedResponse->setNotebookRuntimeTemplateResourceName($notebookRuntimeTemplateResourceName);
         $expectedResponse->setGcsOutputUri($gcsOutputUri);
         $expectedResponse->setExecutionUser($executionUser);
         $expectedResponse->setName($name2);
         $expectedResponse->setDisplayName($displayName);
         $expectedResponse->setScheduleResourceName($scheduleResourceName);
+        $expectedResponse->setKernelName($kernelName);
         $transport->addResponse($expectedResponse);
         // Mock request
         $formattedName = $gapicClient->notebookExecutionJobName('[PROJECT]', '[LOCATION]', '[NOTEBOOK_EXECUTION_JOB]');