feat(sagemakerunifiedstudio): Fetch notebook training jobs (aws#2184)

## Problem
Need backend logic to fetch notebook training jobs for rendering pages

## Solution
- Add SageMaker client for fetching notebook training jobs
- Call SageMaker client in extension host, where it has access to
project role credentials for making API calls
- Add backend logic to transfer jobs between frontend and backend 
- Use Vue composable to store fetched data for reactive state rendering
- Update JobList page to consume fetched data from composable to render
page
- Update JobDetail page to consume fetched data from composable to
render job details

#### TODO
- Based on comments, move time formatting related utils to shared usage

#### List jobs page consuming fetched data for rendering jobs list
<img width="1372" height="1152" alt="Screenshot 2025-08-05 at 10 25
29 AM"
src="https://github.com/user-attachments/assets/c94277ac-024b-41e7-868d-a025a5f20b31"
/>

#### Job detail page using fetch jobs data for rendering job details
<img width="1372" height="1267" alt="Screenshot 2025-08-05 at 10 25
47 AM"
src="https://github.com/user-attachments/assets/d6c1f54c-7e31-4339-9794-de307cb3c7d1"
/>

---

- Treat all work as PUBLIC. Private `feature/x` branches will not be
squash-merged at release time.
- Your code changes must meet the guidelines in
[CONTRIBUTING.md](https://github.com/aws/aws-toolkit-vscode/blob/master/CONTRIBUTING.md#guidelines).
- License: I confirm that my contribution is made under the terms of the
Apache 2.0 license.

---------

Co-authored-by: chungjac <[email protected]>

Author: spkamboj

.gitignore

@@ -32,6 +32,8 @@ src.gen/*
 **/src/codewhisperer/client/codewhispererclient.d.ts
 **/src/codewhisperer/client/codewhispereruserclient.d.ts
 **/src/auth/sso/oidcclientpkce.d.ts
+**/src/sagemakerunifiedstudio/shared/client/gluecatalogapi.d.ts
+**/src/sagemakerunifiedstudio/shared/client/sqlworkbench.d.ts
 
 # Generated by tests
 **/src/testFixtures/**/bin

package-lock.json

@@ -67,6 +67,8 @@ async function createWebview(context: vscode.ExtensionContext, page: Page): Prom
         viewColumn: vscode.ViewColumn.Active,
     })
 
+    // This is temporary setup for now.
+    activePanel.server.initSdkClient('us-east-2', 'd95hwmylut2ai1')
     activePanel.server.setWebviewPanel(webviewPanel)
     activePanel.server.setCurrentPage(page)
 

packages/core/src/sagemakerunifiedstudio/notebookScheduling/activation.ts

@@ -6,6 +6,7 @@
 import * as vscode from 'vscode'
 import { VueWebview } from '../../../webviews/main'
 import { createJobPage, Page } from '../utils/constants'
+import { SageMakerClient, ListJobsResponse } from '../client/sageMakerClient'
 
 /**
  * Webview class for managing SageMaker notebook job scheduling UI.
@@ -27,13 +28,19 @@ export class NotebookJobWebview extends VueWebview {
     /** Tracks the currently displayed page */
     private currentPage: Page = { name: createJobPage, metadata: {} }
 
+    private sageMakerClient?: SageMakerClient
+
     /**
      * Creates a new NotebookJobWebview instance
      */
     public constructor() {
         super(NotebookJobWebview.sourcePath)
     }
 
+    public initSdkClient(regionCode: string, projectId: string): void {
+        this.sageMakerClient = new SageMakerClient(regionCode, projectId)
+    }
+
     public setWebviewPanel(newWebviewPanel: vscode.WebviewPanel): void {
         this.webviewPanel = newWebviewPanel
     }
@@ -54,4 +61,12 @@ export class NotebookJobWebview extends VueWebview {
         this.currentPage = newPage
         this.onShowPage.fire({ page: this.currentPage })
     }
+
+    public async listJobs(): Promise<ListJobsResponse> {
+        if (!this.sageMakerClient) {
+            throw 'SageMakerClient is not initialized'
+        }
+
+        return this.sageMakerClient.listJobs()
+    }
 }

packages/core/src/sagemakerunifiedstudio/notebookScheduling/backend/notebookJobWebview.ts

@@ -0,0 +1,88 @@
+/*!
+ * Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+ * SPDX-License-Identifier: Apache-2.0
+ */
+
+import {
+    SageMakerClient as SageMakerClientSDK,
+    SearchCommandInput,
+    paginateSearch,
+    TrainingJob,
+} from '@aws-sdk/client-sagemaker'
+import { ClientWrapper } from '../../../shared/clients/clientWrapper'
+import { getLogger } from '../../../shared/logger/logger'
+import { SageMakerSearchSortOrder, JobTag } from '../utils/constants'
+
+export interface ListJobsResponse {
+    jobs?: TrainingJob[]
+    error?: string
+}
+
+export class SageMakerClient extends ClientWrapper<SageMakerClientSDK> {
+    private projectId: string
+
+    public constructor(regionCode: string, projectId: string) {
+        super(regionCode, SageMakerClientSDK)
+
+        this.projectId = projectId
+    }
+
+    public async listJobs(): Promise<ListJobsResponse> {
+        const searchRequest: SearchCommandInput = {
+            MaxResults: 100,
+            Resource: 'TrainingJob',
+            SortBy: 'CreationTime',
+            SortOrder: SageMakerSearchSortOrder.DESCENDING,
+            SearchExpression: {
+                Filters: [
+                    {
+                        Name: `Tags.${JobTag.IS_STUDIO_ARCHIVED}`,
+                        Operator: 'Equals',
+                        Value: 'false',
+                    },
+                ],
+                SubExpressions: [
+                    {
+                        Filters: [
+                            {
+                                Name: `Tags.${JobTag.IS_SCHEDULING_NOTEBOOK_JOB}`,
+                                Operator: 'Equals',
+                                Value: 'true',
+                            },
+                            {
+                                Name: `Tags.${JobTag.NOTEBOOK_JOB_ORIGIN}`,
+                                Operator: 'Equals',
+                                Value: 'PIPELINE_STEP',
+                            },
+                        ],
+                        Operator: 'Or',
+                    },
+                ],
+            },
+            VisibilityConditions: [
+                {
+                    Key: `Tags.${JobTag.AmazonDataZoneProject}`,
+                    Value: this.projectId,
+                },
+            ],
+        }
+
+        try {
+            const jobs: TrainingJob[] = []
+            const paginator = paginateSearch({ client: this.getClient() }, searchRequest)
+
+            for await (const page of paginator) {
+                for (const result of page.Results ?? []) {
+                    if (result.TrainingJob) {
+                        jobs.push(result.TrainingJob)
+                    }
+                }
+            }
+
+            return { jobs }
+        } catch (error: any) {
+            getLogger().error('SageMakerClient.listJobs: %s', error)
+            return { error: `${error}` }
+        }
+    }
+}

packages/core/src/sagemakerunifiedstudio/notebookScheduling/client/sageMakerClient.ts

@@ -42,3 +42,68 @@ export const jobDetailPage: string = 'jobDetailPage'
 export const jobDefinitionDetailPage: string = 'jobDefinitionDetailPage'
 
 export const editJobDefinitionPage: string = 'editJobDefinitionPage'
+
+export enum SageMakerSearchSortOrder {
+    ASCENDING = 'Ascending',
+    DESCENDING = 'Descending',
+}
+
+export enum SearchSortOrder {
+    ASCENDING = 'Ascending',
+    DESCENDING = 'Descending',
+    NONE = 'None',
+}
+
+export enum JobTag {
+    IS_SCHEDULING_NOTEBOOK_JOB = 'sagemaker:is-scheduling-notebook-job',
+    IS_STUDIO_ARCHIVED = 'sagemaker:is-studio-archived',
+    JOB_DEFINITION_ID = 'sagemaker:job-definition-id',
+    NAME = 'sagemaker:name',
+    NOTEBOOK_NAME = 'sagemaker:notebook-name',
+    USER_PROFILE_NAME = 'sagemaker:user-profile-name',
+    NOTEBOOK_JOB_ORIGIN = 'sagemaker:notebook-job-origin',
+    AmazonDataZoneProject = 'AmazonDataZoneProject',
+    ONE_TIME_SCHEDULE = 'sagemaker-studio:one-time',
+    SMUS_USER_ID = 'sagemaker-studio:user-id',
+}
+
+export enum RuntimeEnvironmentParameterName {
+    SM_IMAGE = 'sm_image',
+    SM_KERNEL = 'sm_kernel',
+    SM_INIT_SCRIPT = 'sm_init_script',
+    SM_LCC_INIT_SCRIPT_ARN = 'sm_lcc_init_script_arn',
+    S3_INPUT = 's3_input',
+    S3_INPUT_ACCOUNT_ID = 's3_input_account_id',
+    S3_OUTPUT = 's3_output',
+    S3_OUTPUT_ACCOUNT_ID = 's3_output_account_id',
+    ROLE_ARN = 'role_arn',
+    VPC_SECURITY_GROUP_IDS = 'vpc_security_group_ids',
+    VPC_SUBNETS = 'vpc_subnets',
+    SM_OUTPUT_KMS_KEY = 'sm_output_kms_key',
+    SM_VOLUME_KMS_KEY = 'sm_volume_kms_key',
+    MAX_RETRY_ATTEMPTS = 'max_retry_attempts',
+    MAX_RUN_TIME_IN_SECONDS = 'max_run_time_in_seconds',
+    SM_SKIP_EFS_SIMULATION = 'sm_skip_efs_simulation',
+    ENABLE_NETWORK_ISOLATION = 'enable_network_isolation',
+}
+
+export enum JobEnvironmentVariableName {
+    SM_JOB_DEF_VERSION = 'SM_JOB_DEF_VERSION',
+    SM_FIRST_PARTY_IMAGEOWNER = 'SM_FIRST_PARTY_IMAGEOWNER',
+    SM_FIRST_PARTY_IMAGE_ARN = 'SM_FIRST_PARTY_IMAGE_ARN',
+    SM_KERNEL_NAME = 'SM_KERNEL_NAME',
+    SM_SKIP_EFS_SIMULATION = 'SM_SKIP_EFS_SIMULATION',
+    SM_EFS_MOUNT_PATH = 'SM_EFS_MOUNT_PATH',
+    SM_EFS_MOUNT_UID = 'SM_EFS_MOUNT_UID',
+    SM_EFS_MOUNT_GID = 'SM_EFS_MOUNT_GID',
+    SM_INPUT_NOTEBOOK_NAME = 'SM_INPUT_NOTEBOOK_NAME',
+    SM_OUTPUT_NOTEBOOK_NAME = 'SM_OUTPUT_NOTEBOOK_NAME',
+    AWS_DEFAULT_REGION = 'AWS_DEFAULT_REGION',
+    SM_ENV_NAME = 'SM_ENV_NAME',
+    SM_INIT_SCRIPT = 'SM_INIT_SCRIPT',
+    SM_LCC_INIT_SCRIPT = 'SM_LCC_INIT_SCRIPT',
+    SM_LCC_INIT_SCRIPT_ARN = 'SM_LCC_INIT_SCRIPT_ARN',
+    SM_OUTPUT_FORMATS = 'SM_OUTPUT_FORMATS',
+    SM_EXECUTION_INPUT_PATH = 'SM_EXECUTION_INPUT_PATH',
+    SM_PACKAGE_INPUT_FOLDER = 'SM_PACKAGE_INPUT_FOLDER',
+}

packages/core/src/sagemakerunifiedstudio/notebookScheduling/utils/constants.ts

@@ -0,0 +1,123 @@
+/*!
+ * Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+ * SPDX-License-Identifier: Apache-2.0
+ */
+
+import { TrainingJob, TrainingJobStatus } from '@aws-sdk/client-sagemaker'
+import { RuntimeEnvironmentParameterName, JobEnvironmentVariableName, JobTag } from './constants'
+
+export function getJobName(job: TrainingJob): string | undefined {
+    if (job.Tags) {
+        for (const { Key, Value } of job.Tags) {
+            if (Key === JobTag.NAME) {
+                return Value
+            }
+        }
+    }
+}
+
+export function getJobStatus(job: TrainingJob) {
+    if (job.TrainingJobStatus === TrainingJobStatus.IN_PROGRESS) {
+        return 'In progress'
+    }
+
+    return job.TrainingJobStatus
+}
+
+export function getJobInputNotebookName(job: TrainingJob): string | undefined {
+    if (job.Environment) {
+        return job.Environment[JobEnvironmentVariableName.SM_INPUT_NOTEBOOK_NAME]
+    }
+}
+
+export function getJobEnvironmentName(job: TrainingJob): string | undefined {
+    if (job.Environment) {
+        return job.Environment[JobEnvironmentVariableName.SM_ENV_NAME]
+    }
+}
+
+export function getJobKernelName(job: TrainingJob): string | undefined {
+    if (job.Environment) {
+        return job.Environment[JobEnvironmentVariableName.SM_KERNEL_NAME]
+    }
+}
+
+export function getJobParameters(job: TrainingJob): { key: string; value: string }[] | undefined {
+    if (job.HyperParameters) {
+        const result = []
+
+        for (const [key, value] of Object.entries(job.HyperParameters)) {
+            result.push({ key, value })
+        }
+
+        return result
+    }
+}
+
+export function getJobEnvironmentParameters(job: TrainingJob) {
+    if (job.Environment) {
+        const runtimeEnvironmentParameterSet: Set<string> = new Set(Object.values(RuntimeEnvironmentParameterName))
+        const jobEnvironmentVariableSet: Set<string> = new Set(Object.values(JobEnvironmentVariableName))
+
+        const result = []
+
+        for (const [key, value] of Object.entries(job.Environment)) {
+            if (!(runtimeEnvironmentParameterSet.has(key) || jobEnvironmentVariableSet.has(key))) {
+                result.push({ key, value })
+            }
+        }
+
+        return result
+    }
+}
+
+export function getJobRanWithInputFolder(job: TrainingJob): boolean {
+    if (job.Environment && JobEnvironmentVariableName.SM_PACKAGE_INPUT_FOLDER in job.Environment) {
+        return job.Environment[JobEnvironmentVariableName.SM_PACKAGE_INPUT_FOLDER].toLowerCase() === 'true'
+    }
+
+    return false
+}
+
+export function getFormattedDateTime(utcString: string): string {
+    const date = new Date(utcString)
+
+    const pad = (n: number) => n.toString().padStart(2, '0')
+
+    const year = date.getFullYear()
+    const month = pad(date.getMonth() + 1)
+    const day = pad(date.getDate())
+
+    let hours = date.getHours()
+    const minutes = pad(date.getMinutes())
+
+    // Determine AM or PM
+    const ampm = hours >= 12 ? 'PM' : 'AM'
+
+    // Convert to 12-hour format
+    hours = hours % 12
+    hours = hours ? hours : 12 // 0 should be 12
+
+    const formatted = `${year}-${month}-${day}, ${pad(hours)}:${minutes} ${ampm}`
+    return formatted
+}
+
+export function getFormattedCurrentTime(): string {
+    const date = new Date()
+
+    const pad = (n: number) => n.toString().padStart(2, '0')
+
+    let hours = date.getHours()
+    const minutes = pad(date.getMinutes())
+    const seconds = pad(date.getSeconds())
+
+    // Determine AM or PM
+    const ampm = hours >= 12 ? 'PM' : 'AM'
+
+    // Convert to 12-hour format
+    hours = hours % 12
+    hours = hours ? hours : 12 // 0 should be 12
+
+    const formatted = `${pad(hours)}:${minutes}:${seconds} ${ampm}`
+    return formatted
+}