Fix creating new kernels

Author: jakubno

js/src/code-interpreter.ts

@@ -145,10 +145,10 @@ export class JupyterExtension {
     cwd: string = '/home/user',
     kernelName?: string
   ): Promise<string> {
-    const data = { path: cwd, kernel: {name: "python3"}, type: "notebook", name: id(16) }
-    if (kernelName) {
-      data.kernel.name = kernelName
-    }
+    kernelName = kernelName || 'python3'
+
+
+    const data = { path: id(16), kernel: {name: kernelName}, type: "notebook", name: id(16) }
 
     const response = await fetch(
       `${this.sandbox.getProtocol()}://${this.sandbox.getHostname(
@@ -164,9 +164,27 @@ export class JupyterExtension {
       throw new Error(`Failed to create kernel: ${response.statusText}`)
     }
 
+
     const sessionInfo = await response.json()
     const kernelID = sessionInfo.kernel.id
-    await this.connectToKernelWS(kernelID, sessionInfo.id)
+    const sessionID = sessionInfo.id
+
+    const patchResponse = await fetch(
+      `${this.sandbox.getProtocol()}://${this.sandbox.getHostname(
+        8888
+      )}/api/sessions/${sessionID}`,
+      {
+        method: 'PATCH',
+        body: JSON.stringify({path: cwd})
+      }
+    )
+
+    if (!patchResponse.ok) {
+      throw new Error(`Failed to create kernel: ${response.statusText}`)
+    }
+
+
+    await this.connectToKernelWS(kernelID, sessionID)
 
     return kernelID
   }

python/e2b_code_interpreter/main.py

@@ -142,17 +142,8 @@ def create_kernel(
         :return: Kernel id of the created kernel
         """
         kernel_name = kernel_name or "python3"
-        response = requests.post(
-            f"{self._sandbox.get_protocol()}://{self._sandbox.get_hostname(8888)}/api/kernels",
-            json={"name": kernel_name, "path": cwd},
-            timeout=timeout,
-        )
-        if not response.ok:
-            raise KernelException(f"Failed to create kernel: {response.text}")
 
-        kernel_data = response.json()
-        kernel_id = kernel_data["id"]
-        data = {"path": str(uuid.uuid4()), "kernel": {"name": kernel_name, "id": kernel_id}, "type": "notebook", "name": str(uuid.uuid4())}
+        data = {"path": str(uuid.uuid4()), "kernel": {"name": kernel_name}, "type": "notebook", "name": str(uuid.uuid4())}
         logger.debug(f"Creating kernel with data: {data}")
 
         response = requests.post(