Fix singleton window bug

Author: aws-ajangg

packages/core/src/awsService/sagemaker/commands.ts

@@ -132,8 +132,10 @@ export async function deeplinkConnect(
 
     try {
         let connectionType = 'sm_dl'
-        if (domain === '') {
+        if (domain === '' && eksClusterArn) {
+            const { accountId, region, clusterName } = parseEKSClusterArn(eksClusterArn)
             connectionType = 'sm_hp'
+            session = `${workspaceName}_${namespace}_${clusterName}_${region}_${accountId}`
         }
         const remoteEnv = await prepareDevEnvConnection(
             connectionIdentifier,
@@ -179,6 +181,17 @@ export async function deeplinkConnect(
     }
 }
 
+function parseEKSClusterArn(eksClusterArn: string): { accountId: string; region: string; clusterName: string } {
+    const parts = eksClusterArn.split(':')
+    if (parts.length !== 6) {
+        throw new Error(`Invalid EKS cluster ARN: ${eksClusterArn}`)
+    }
+    const accountId = parts[4]
+    const region = parts[3]
+    const clusterName = parts[5].split('/')[1]
+    return { accountId, region, clusterName }
+}
+
 export async function stopSpace(
     node: SagemakerSpaceNode | SagemakerUnifiedStudioSpaceNode,
     ctx: vscode.ExtensionContext,

packages/core/src/awsService/sagemaker/model.ts

@@ -186,7 +186,7 @@ export async function prepareDevEnvConnection(
 
                   const hyperPodEnv: NodeJS.ProcessEnv = {
                       AWS_REGION: region,
-                      SESSION_ID: session || '',
+                      SESSION_ID: hostname || '',
                       STREAM_URL: decodedWsUrl,
                       TOKEN: decodedToken,
                       AWS_SSM_CLI: ssm,