adk-code-mode 1.6.0__tar.gz → 1.6.1__tar.gz

adk_code_mode-1.6.1/.release-please-manifest.json

@@ -0,0 +1,3 @@
+{
+    ".": "1.6.1"
+}

{adk_code_mode-1.6.0 → adk_code_mode-1.6.1}/CHANGELOG.md

@@ -1,5 +1,12 @@
 # Changelog
 
+## [1.6.1](https://github.com/a2anet/adk-code-mode/compare/adk-code-mode-v1.6.0...adk-code-mode-v1.6.1) (2026-09-22)
+
+
+### Bug Fixes
+
+* exit the sandbox when the host disconnects mid-block ([bd3fb0f](https://github.com/a2anet/adk-code-mode/commit/bd3fb0fb0d8a0cab9370f2ad6db56dfacf566cb4))
+
 ## [1.6.0](https://github.com/a2anet/adk-code-mode/compare/adk-code-mode-v1.5.0...adk-code-mode-v1.6.0) (2026-08-29)
 
 

{adk_code_mode-1.6.0 → adk_code_mode-1.6.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.5
 Name: adk-code-mode
-Version: 1.6.0
+Version: 1.6.1
 Summary: A "Code Mode" sandboxed code-execution tool for ADK agents to interact with tools, files, and custom packages with Python
 Project-URL: Documentation, https://github.com/A2ANet/adk-code-mode#readme
 Project-URL: Issues, https://github.com/A2ANet/adk-code-mode/issues

{adk_code_mode-1.6.0 → adk_code_mode-1.6.1}/sandbox-wheel/src/adk_code_mode_sandbox/__about__.py

@@ -1,4 +1,4 @@
 # SPDX-FileCopyrightText: 2025-present A2A Net <hello@a2anet.com>
 #
 # SPDX-License-Identifier: Apache-2.0
-__version__ = "1.6.0"  # x-release-please-version
+__version__ = "1.6.1"  # x-release-please-version

{adk_code_mode-1.6.0 → adk_code_mode-1.6.1}/sandbox-wheel/src/adk_code_mode_sandbox/_entry.py

@@ -228,7 +228,6 @@ def main() -> int:
         port = int(os.environ.get("PORT", "8080"))
         token = os.environ.get(AUTH_TOKEN_ENV)
         asyncio.run(serve(port, token))
-        return 0
 
     reader, writer = _open_control_streams()
     client = RpcClient(reader=reader, writer=writer)

{adk_code_mode-1.6.0 → adk_code_mode-1.6.1}/sandbox-wheel/src/adk_code_mode_sandbox/_http_server.py

@@ -38,7 +38,7 @@ import shutil
 import tarfile
 import tempfile
 from http import HTTPStatus
-from typing import Any
+from typing import Any, NoReturn
 
 import websockets
 import websockets.asyncio.server
@@ -112,10 +112,11 @@ class _WsBridgeWriter:
         pass
 
 
-async def serve(port: int, token: str | None) -> None:
+async def serve(port: int, token: str | None) -> NoReturn:
     """Start the WebSocket server on *port* with optional bearer *token* auth.
 
-    Accepts exactly one WebSocket connection, runs user code, then returns.
+    Accepts exactly one WebSocket connection, runs user code, then exits the
+    process.
     """
     stop = asyncio.Event()
     accepted = False
@@ -136,9 +137,15 @@ async def serve(port: int, token: str | None) -> None:
         return None
 
     async def handler(ws: Any) -> None:
+        turn = asyncio.create_task(_handle_connection(ws))
+        # End the turn when the socket closes, even mid-block: user code that
+        # ignores the disconnect would otherwise keep this single-use container
+        # alive, answering every later connection with "Busy".
+        ws_closed = asyncio.create_task(ws.wait_closed())
+        ws_closed.add_done_callback(lambda _: turn.cancel())
         try:
-            await _handle_connection(ws)
-        except websockets.ConnectionClosed:
+            await turn
+        except (websockets.ConnectionClosed, asyncio.CancelledError):
             logger.debug("client disconnected")
         except Exception:
             logger.exception("error handling WebSocket connection")
@@ -154,6 +161,9 @@ async def serve(port: int, token: str | None) -> None:
     ):
         logger.info("sandbox HTTP server listening on port %d", port)
         await stop.wait()
+    # Don't let interpreter shutdown join a worker thread still running user code
+    # the host has given up on.
+    os._exit(0)
 
 
 async def _handle_connection(ws: Any) -> None:

{adk_code_mode-1.6.0 → adk_code_mode-1.6.1}/src/adk_code_mode/__about__.py

@@ -1,4 +1,4 @@
 # SPDX-FileCopyrightText: 2025-present A2A Net <hello@a2anet.com>
 #
 # SPDX-License-Identifier: Apache-2.0
-__version__ = "1.6.0"  # x-release-please-version
+__version__ = "1.6.1"  # x-release-please-version

{adk_code_mode-1.6.0 → adk_code_mode-1.6.1}/tests/test_remote_integration.py

@@ -340,3 +340,38 @@ async def test_remote_close_shuts_the_container_down(
             break
         await asyncio.sleep(0.02)
     assert http_server.poll() is not None  # container exited after ShutdownFrame
+
+
+@pytest.mark.asyncio
+async def test_remote_timed_out_block_shuts_the_container_down(
+    http_server: subprocess.Popen[bytes],
+) -> None:
+    """A block still running when the host gives up doesn't keep the server alive."""
+    artifact_service = InMemoryArtifactService()
+    session = Session(
+        id="remote-timeout",
+        app_name="test-app",
+        user_id="u1",
+        state={},
+        events=[],
+        last_update_time=0.0,
+    )
+    ctx = _make_ctx(artifact_service, session)
+
+    tool = ExecuteCodeTool(
+        tools=[],
+        backend=RemoteBackend(url=_server_url(http_server)),
+        max_output_chars=10_000,
+        timeout_seconds=1,
+    )
+    result = await tool.run_async(
+        args={"code": "import time\ntime.sleep(3600)\n"},
+        tool_context=_tool_context(ctx, call_id="remote-timeout-1"),
+    )
+    assert "exceeded timeout" in result["stderr"]
+
+    for _ in range(200):
+        if http_server.poll() is not None:
+            break
+        await asyncio.sleep(0.02)
+    assert http_server.poll() is not None  # container exited despite the running block

adk_code_mode-1.6.0/.release-please-manifest.json

@@ -1,3 +0,0 @@
-{
-    ".": "1.6.0"
-}