addressed comments

Author: mishushakov

.changeset/clever-bears-accept.md

@@ -2,4 +2,4 @@
 '@e2b/code-interpreter-template': patch
 ---
 
-release lock before stream completion
+interrupt kernel execution on client disconnect

js/tests/interrupt.test.ts

@@ -9,7 +9,7 @@ sandboxTest(
     // This simulates a client disconnect: the SDK aborts the connection,
     // which should trigger the server to interrupt the kernel (#213).
     await expect(
-      sandbox.runCode('import time; time.sleep(30)', { timeoutMs: 3_000 })
+      sandbox.runCode('import time; time.sleep(300)', { timeoutMs: 3_000 })
     ).rejects.toThrow()
 
     // Wait for the server to detect the disconnect (via keepalive write

python/tests/async/test_async_interrupt.py

@@ -13,7 +13,7 @@ async def test_subsequent_execution_works_after_client_timeout(
     # This simulates a client disconnect: the SDK closes the connection,
     # which should trigger the server to interrupt the kernel (#213).
     with pytest.raises(TimeoutException):
-        await async_sandbox.run_code("import time; time.sleep(30)", timeout=3)
+        await async_sandbox.run_code("import time; time.sleep(300)", timeout=3)
 
     # Wait for the server to detect the disconnect (via keepalive write
     # failure) and interrupt the kernel.

python/tests/sync/test_interrupt.py

@@ -11,7 +11,7 @@ def test_subsequent_execution_works_after_client_timeout(sandbox: Sandbox):
     # This simulates a client disconnect: the SDK closes the connection,
     # which should trigger the server to interrupt the kernel (#213).
     with pytest.raises(TimeoutException):
-        sandbox.run_code("import time; time.sleep(30)", timeout=3)
+        sandbox.run_code("import time; time.sleep(300)", timeout=3)
 
     # Wait for the server to detect the disconnect (via keepalive write
     # failure) and interrupt the kernel.

template/server/messaging.py

@@ -258,6 +258,13 @@ async def _cleanup_env_vars(self, env_vars: Dict[StrictStr, str]):
     async def _wait_for_result(self, message_id: str):
         queue = self._executions[message_id].queue
 
+        # Use a timeout on queue.get() to periodically send keepalives.
+        # Without keepalives, the generator blocks indefinitely waiting for
+        # kernel output. If the client silently disappears (e.g. network
+        # failure), uvicorn can only detect the broken connection when it
+        # tries to write — so we force a write every KEEPALIVE_INTERVAL
+        # seconds. This ensures timely disconnect detection and kernel
+        # interrupt for abandoned executions (see #213).
         while True:
             try:
                 output = await asyncio.wait_for(queue.get(), timeout=KEEPALIVE_INTERVAL)