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

{adk_code_mode-1.6.1 → adk_code_mode-2.0.1}/.github/workflows/release-please.yml

@@ -112,5 +112,5 @@ jobs:
           file: docker/Dockerfile
           push: true
           tags: |
-            ghcr.io/a2anet/adk-code-mode:${{ needs.release-please.outputs.version }}
-            ghcr.io/a2anet/adk-code-mode:latest
+            ghcr.io/buildagenta/adk-code-mode:${{ needs.release-please.outputs.version }}
+            ghcr.io/buildagenta/adk-code-mode:latest

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

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

adk_code_mode-2.0.1/CHANGELOG.md

@@ -0,0 +1,169 @@
+# Changelog
+
+## [2.0.1](https://github.com/buildagenta/adk-code-mode/compare/adk-code-mode-v2.0.0...adk-code-mode-v2.0.1) (2026-11-02)
+
+
+### Miscellaneous Chores
+
+* credit Agenta as the author ([#25](https://github.com/buildagenta/adk-code-mode/issues/25)) ([bfe577e](https://github.com/buildagenta/adk-code-mode/commit/bfe577e9aaba31ed9d8f6423485b061deb77b963))
+
+## [2.0.0](https://github.com/buildagenta/adk-code-mode/compare/adk-code-mode-v1.6.1...adk-code-mode-v2.0.0) (2026-09-25)
+
+
+### ⚠ BREAKING CHANGES
+
+* sandbox code that checked a tool result for an `error` key now gets an exception instead, and the sandbox image must be rebuilt for protocol version 3.
+
+### Features
+
+* raise when a tool call fails instead of returning the error ([#22](https://github.com/buildagenta/adk-code-mode/issues/22)) ([489083d](https://github.com/buildagenta/adk-code-mode/commit/489083d988cd7e958e94f3ef6acba6443284abd8))
+* stop a timed-out block without losing the turn's state ([#23](https://github.com/buildagenta/adk-code-mode/issues/23)) ([4720156](https://github.com/buildagenta/adk-code-mode/commit/47201568d0bb06527e32f850b7abb1b465e36920))
+
+## [1.6.1](https://github.com/buildagenta/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/buildagenta/adk-code-mode/commit/bd3fb0fb0d8a0cab9370f2ad6db56dfacf566cb4))
+
+## [1.6.0](https://github.com/buildagenta/adk-code-mode/compare/adk-code-mode-v1.5.0...adk-code-mode-v1.6.0) (2026-08-29)
+
+
+### Features
+
+* bound an execute_code block by default ([1609c0d](https://github.com/buildagenta/adk-code-mode/commit/1609c0d24b8563cba51434cd1214b4299aeefc86))
+* expand object oneOf/anyOf branches in tool stub docstrings ([#20](https://github.com/buildagenta/adk-code-mode/issues/20)) ([7d42cb1](https://github.com/buildagenta/adk-code-mode/commit/7d42cb1ee9d9e217d008776dc74a84034f4b67a9))
+
+## [1.5.0](https://github.com/buildagenta/adk-code-mode/compare/adk-code-mode-v1.4.0...adk-code-mode-v1.5.0) (2026-08-10)
+
+
+### Features
+
+* scope tool-name uniqueness to each namespace ([#17](https://github.com/buildagenta/adk-code-mode/issues/17)) ([437bc0d](https://github.com/buildagenta/adk-code-mode/commit/437bc0dc930521a933e5c25b4bd6103e9434fc27))
+
+## [1.4.0](https://github.com/buildagenta/adk-code-mode/compare/adk-code-mode-v1.3.0...adk-code-mode-v1.4.0) (2026-07-28)
+
+
+### Features
+
+* surface nested schema properties and validation keywords in tool stubs ([608d14b](https://github.com/buildagenta/adk-code-mode/commit/608d14bb115159b2991c7d8d4e3c641977fd0d66))
+
+
+### Documentation
+
+* document stub schema fidelity and the `JSON_SCHEMA_FOR_FUNC_DECL` flag in `README.md` ([219f02b](https://github.com/buildagenta/adk-code-mode/commit/219f02bd493c2e2d7275a0034ef2a40db933859f))
+
+## [1.3.0](https://github.com/buildagenta/adk-code-mode/compare/adk-code-mode-v1.2.0...adk-code-mode-v1.3.0) (2026-07-23)
+
+
+### Features
+
+* describe the sandbox to the model in a tagged `<code-mode>` block ([#14](https://github.com/buildagenta/adk-code-mode/issues/14)) ([662a281](https://github.com/buildagenta/adk-code-mode/commit/662a28138e8632b434e5f1ecd52793f7d83ee097))
+
+## [1.2.0](https://github.com/buildagenta/adk-code-mode/compare/adk-code-mode-v1.1.0...adk-code-mode-v1.2.0) (2026-07-23)
+
+
+### Features
+
+* add `custom_metadata` to the `save_artifact` tool ([dfd9a59](https://github.com/buildagenta/adk-code-mode/commit/dfd9a59bf1718084a1d5c02cc1ea453269223f16))
+
+## [1.1.0](https://github.com/buildagenta/adk-code-mode/compare/adk-code-mode-v1.0.0...adk-code-mode-v1.1.0) (2026-07-15)
+
+
+### Features
+
+* expose `tools` and `backend` as public properties on `ExecuteCodeTool` ([acfb704](https://github.com/buildagenta/adk-code-mode/commit/acfb704d388da5d66e318775e03fe055467679c2))
+
+## [1.0.0](https://github.com/buildagenta/adk-code-mode/compare/adk-code-mode-v0.4.1...adk-code-mode-v1.0.0) (2026-07-15)
+
+
+### ⚠ BREAKING CHANGES
+
+* `CodeModeCodeExecutor`, `code_mode_before_model_callback`, and `CODE_MODE_SYSTEM_INSTRUCTION` no longer exist. Add `ExecuteCodeTool` to the agent's `tools=[...]` instead of wiring `code_executor=`; drop `before_model_callback=`, `generate_content_config=`, and `CODE_MODE_SYSTEM_INSTRUCTION` entirely; `release_invocation` is now `async`. See the README's migration section.
+
+### Features
+
+* replace `CodeModeCodeExecutor` with `ExecuteCodeTool` ([#10](https://github.com/buildagenta/adk-code-mode/issues/10)) ([b78572e](https://github.com/buildagenta/adk-code-mode/commit/b78572e72446930548760756255390d1a3cc388a))
+
+## [0.4.1](https://github.com/buildagenta/adk-code-mode/compare/adk-code-mode-v0.4.0...adk-code-mode-v0.4.1) (2026-07-10)
+
+
+### Documentation
+
+* note the sandbox writes to `/tools` in remote deployment guide ([96bb45e](https://github.com/buildagenta/adk-code-mode/commit/96bb45e4e18d9d54709f4c10b6b09cf3567ee385))
+
+## [0.4.0](https://github.com/buildagenta/adk-code-mode/compare/adk-code-mode-v0.3.0...adk-code-mode-v0.4.0) (2026-07-09)
+
+
+### Features
+
+* save tool results as artifacts with optional model-supplied naming ([76a3671](https://github.com/buildagenta/adk-code-mode/commit/76a36713b2c52bd5b8dc51d3669226a47e7108e5))
+
+
+### Bug Fixes
+
+* align code-mode example with the `<code-mode>` catalog and fix fences ([4a0cc6d](https://github.com/buildagenta/adk-code-mode/commit/4a0cc6d0fc99d5425f6682cc489e76c9ddef7d46))
+* clarify code-mode tools are a Python library, not function tools ([48d1018](https://github.com/buildagenta/adk-code-mode/commit/48d1018152a2ed2a8e7361de6ce0f43b31b429ce))
+* nudge the model to call tools via Python in the code-mode block ([141baf3](https://github.com/buildagenta/adk-code-mode/commit/141baf3a44976e82cf437b2064666be893d6d409))
+* update `CODE_MODE_SYSTEM_INSTRUCTION` and `README.md` ([933d42c](https://github.com/buildagenta/adk-code-mode/commit/933d42cb237b9d6deeb6611511cba28a0d5e9261))
+
+
+### Documentation
+
+* streamline `README.md` and fix broken code fences ([c237514](https://github.com/buildagenta/adk-code-mode/commit/c2375149df48e21318967090b6c85d0b1f411ea2))
+
+## [0.3.0](https://github.com/buildagenta/adk-code-mode/compare/adk-code-mode-v0.2.2...adk-code-mode-v0.3.0) (2026-07-06)
+
+
+### Features
+
+* hold one sandbox container open per turn (protocol v2) ([#6](https://github.com/buildagenta/adk-code-mode/issues/6)) ([aebe17d](https://github.com/buildagenta/adk-code-mode/commit/aebe17d2bbbf21cc0f05852efaf910ff6adc3154))
+
+
+### Documentation
+
+* lower startup probe period in `README.md` for faster cold starts ([e251af2](https://github.com/buildagenta/adk-code-mode/commit/e251af2cf7307c9944361e1a0e37c7191822a9e1))
+* note HTTP `/health` startup probe in `README.md` Cloud Run section ([d58fb10](https://github.com/buildagenta/adk-code-mode/commit/d58fb10f7aeda95e5c8cf29263a13e1333f661dd))
+
+## [0.2.2](https://github.com/buildagenta/adk-code-mode/compare/adk-code-mode-v0.2.1...adk-code-mode-v0.2.2) (2026-05-08)
+
+
+### Bug Fixes
+
+* use `python` fences instead of `tool_code` in `CodeModeCodeExecutor` ([efcc11e](https://github.com/buildagenta/adk-code-mode/commit/efcc11e1020c7544223074e00a8649f9c85d4ad6))
+
+## [0.2.1](https://github.com/buildagenta/adk-code-mode/compare/adk-code-mode-v0.2.0...adk-code-mode-v0.2.1) (2026-05-05)
+
+
+### Bug Fixes
+
+* rename `<tools>` tag to `<code-mode>` and update `CODE_MODE_SYSTEM_INSTRUCTION` ([0a06bfb](https://github.com/buildagenta/adk-code-mode/commit/0a06bfbdf22db6e0226e477d0ed074a6cc4d1515))
+
+
+### Documentation
+
+* update `README.md` with deployment instructions ([1dd7a95](https://github.com/buildagenta/adk-code-mode/commit/1dd7a95197fd56231ba97182526a825baf18508a))
+
+## [0.2.0](https://github.com/buildagenta/adk-code-mode/compare/adk-code-mode-v0.1.0...adk-code-mode-v0.2.0) (2026-05-03)
+
+
+### Features
+
+* add `RemoteBackend`, sandbox hardening, and make `README.md` more concise ([a1a70ee](https://github.com/buildagenta/adk-code-mode/commit/a1a70eec0686f164ce3796ba1d82bb82b7fbda59))
+
+
+### Bug Fixes
+
+* improve `CODE_MODE_SYSTEM_INSTRUCTION` ([0d00863](https://github.com/buildagenta/adk-code-mode/commit/0d008636e2e273b438192f737e20e207c165a610))
+
+## 0.1.0 (2026-04-29)
+
+
+### Features
+
+* ADK Code Mode: `CodeModeExecutor`, `DockerRuntime`, and more ([6ab1ac2](https://github.com/buildagenta/adk-code-mode/commit/6ab1ac28236d21ecb740d8e920e08e4d9bd969c5))
+
+
+### Documentation
+
+* shorten ADK code executor comparison table headers ([28116cf](https://github.com/buildagenta/adk-code-mode/commit/28116cf9f40b3623c31782a55885fa5c4f8c6bbf))

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

@@ -1,11 +1,11 @@
 Metadata-Version: 2.5
 Name: adk-code-mode
-Version: 1.6.1
+Version: 2.0.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
-Project-URL: Source, https://github.com/A2ANet/adk-code-mode
-Author-email: A2A Net <hello@a2anet.com>
+Project-URL: Documentation, https://github.com/buildagenta/adk-code-mode#readme
+Project-URL: Issues, https://github.com/buildagenta/adk-code-mode/issues
+Project-URL: Source, https://github.com/buildagenta/adk-code-mode
+Author-email: Agenta <hello@buildagenta.com>
 License-Expression: Apache-2.0
 License-File: LICENSE
 Keywords: a2a,adk,agents,code-mode,execute-code
@@ -39,6 +39,7 @@ Inspired by Cloudflare's [Code Mode](https://blog.cloudflare.com/code-mode/) and
 ## ✨ Features
 
 - **Call ADK tools from sandbox code** — imports against the `tools` package proxy back to the host and run through ADK's `before_tool` / `after_tool` / `on_error` callbacks and the plugin manager exactly as direct tool calls would.
+- **Failed tool calls raise** — a REST tool whose API answers with an HTTP error raises `HTTPStatusError` (with `status_code` and `body`), and any other failure, including an MCP tool that reports one, raises `ToolError`. ADK returns these as values for a model to read, but sandbox code reads return values, so it would otherwise carry on as if the call had worked.
 - **Bake any Python package into the image** — extend the published base image with anything the model's code needs to `import`, no runtime `pip install` required. The sandbox reports what it has installed, so the model is told about it automatically.
 - **Cross-turn persistence via ADK Artifacts** — `save_artifact` / `load_artifact` / `list_artifacts` are auto-injected and route through your configured `ArtifactService`. Files the code creates or changes are saved as artifacts automatically too.
 - **Tool results saved as artifacts** — on by default; every tool's result is persisted as a `code_mode.tool_result` artifact (with optional model-supplied name/description) so hosts can forward outputs and large results stay out of the prompt. Opt out with `save_tool_results_as_artifacts=False`.
@@ -124,7 +125,7 @@ override_feature_enabled(FeatureName.JSON_SCHEMA_FOR_FUNC_DECL, True)
 
 tool = ExecuteCodeTool(
     tools=[my_fn_tool, McpToolset(...), OpenAPIToolset(...)],
-    backend=UnsafeLocalDockerBackend(image="ghcr.io/a2anet/adk-code-mode:latest"),
+    backend=UnsafeLocalDockerBackend(image="ghcr.io/buildagenta/adk-code-mode:latest"),
 )
 ```
 
@@ -144,8 +145,8 @@ print(send_message(channel="C123", text="hi"))
 ```bash
 # Push the sandbox image to Artifact Registry
 gcloud auth configure-docker <region>-docker.pkg.dev
-docker pull --platform linux/amd64 ghcr.io/a2anet/adk-code-mode:latest
-docker tag  ghcr.io/a2anet/adk-code-mode:latest \
+docker pull --platform linux/amd64 ghcr.io/buildagenta/adk-code-mode:latest
+docker tag  ghcr.io/buildagenta/adk-code-mode:latest \
     <region>-docker.pkg.dev/<project>/<repository>/adk-code-mode-sandbox:latest
 docker push <region>-docker.pkg.dev/<project>/<repository>/adk-code-mode-sandbox:latest
 
@@ -279,10 +280,10 @@ ExecuteCodeTool(tools=..., backend=..., save_tool_results_as_artifacts=True)
 
 ## 🐳 Sandbox Image
 
-The published base image (`ghcr.io/a2anet/adk-code-mode`) works as-is for tools whose execution is fully host-side. To bake in extra Python packages:
+The published base image (`ghcr.io/buildagenta/adk-code-mode`) works as-is for tools whose execution is fully host-side. To bake in extra Python packages:
 
 ```dockerfile
-FROM ghcr.io/a2anet/adk-code-mode:latest
+FROM ghcr.io/buildagenta/adk-code-mode:latest
 RUN pip install --no-cache-dir pandas==2.2.*
 ```
 
@@ -300,7 +301,7 @@ All settings are `ExecuteCodeTool` constructor arguments:
 | `max_code_mode_metadata_chars` | `50_000` | Budget for the whole rendered `<code-mode>` block, tags included. Oversized blocks degrade in tiers rather than disappearing — see [What the model sees](#what-the-model-sees). |
 | `max_output_chars` | `50_000` | Caps stdout/stderr handed back to the model. Overflow is saved as a session artifact at `code_mode/stdout/<call-id>.txt` and the model sees a head-and-tail view pointing to it. |
 | `max_code_chars` | `1_000_000` | Rejects oversized code payloads before starting a container. |
-| `timeout_seconds` | `60` | Caps overall execution time of one `execute_code` call, and with it every tool the block calls. `None` lifts the cap, leaving the platform request timeout (e.g. Cloud Run `--timeout`) as the only bound. |
+| `timeout_seconds` | `60` | Caps overall execution time of one `execute_code` call, and with it every tool the block calls. A block past the cap is stopped with its output so far, keeping the turn's variables and working directory. If it won't stop within a few seconds or leaves a worker thread or child process running, its sandbox is restarted and its output and state are lost. The model is told the cap. `None` lifts it, leaving the platform request timeout (e.g. Cloud Run `--timeout`) as the only bound. |
 | `per_tool_timeout_seconds` | `None` | Caps each individual tool call made from within the sandbox. |
 | `session_idle_timeout_seconds` | `600` | Idle reaper: closes a turn's container once it goes untouched this long. Backstop for turns that never call `release_invocation`. |
 
@@ -469,9 +470,9 @@ uv run pytest -m docker
 
 `adk-code-mode` is distributed under the terms of the [Apache-2.0](https://spdx.org/licenses/Apache-2.0.html) license.
 
-## 🤝 Join the A2A Net Community
+## 🤝 Join the Agenta Community
 
-A2A Net is a site to find and share AI agents and open-source community. Join to share your A2A agents, ask questions, stay up-to-date with the latest A2A news, be the first to hear about open-source releases, tutorials, and more!
+Agenta is a site to find and share AI agents and open-source community. Join to share your A2A agents, ask questions, stay up-to-date with the latest A2A news, be the first to hear about open-source releases, tutorials, and more!
 
-- 🌍 Site: [A2A Net](https://a2anet.com)
+- 🌍 Site: [Agenta](https://buildagenta.com)
 - 🤖 Discord: [Join the Discord](https://discord.gg/674NGXpAjU)

{adk_code_mode-1.6.1 → adk_code_mode-2.0.1}/README.md

@@ -12,6 +12,7 @@ Inspired by Cloudflare's [Code Mode](https://blog.cloudflare.com/code-mode/) and
 ## ✨ Features
 
 - **Call ADK tools from sandbox code** — imports against the `tools` package proxy back to the host and run through ADK's `before_tool` / `after_tool` / `on_error` callbacks and the plugin manager exactly as direct tool calls would.
+- **Failed tool calls raise** — a REST tool whose API answers with an HTTP error raises `HTTPStatusError` (with `status_code` and `body`), and any other failure, including an MCP tool that reports one, raises `ToolError`. ADK returns these as values for a model to read, but sandbox code reads return values, so it would otherwise carry on as if the call had worked.
 - **Bake any Python package into the image** — extend the published base image with anything the model's code needs to `import`, no runtime `pip install` required. The sandbox reports what it has installed, so the model is told about it automatically.
 - **Cross-turn persistence via ADK Artifacts** — `save_artifact` / `load_artifact` / `list_artifacts` are auto-injected and route through your configured `ArtifactService`. Files the code creates or changes are saved as artifacts automatically too.
 - **Tool results saved as artifacts** — on by default; every tool's result is persisted as a `code_mode.tool_result` artifact (with optional model-supplied name/description) so hosts can forward outputs and large results stay out of the prompt. Opt out with `save_tool_results_as_artifacts=False`.
@@ -97,7 +98,7 @@ override_feature_enabled(FeatureName.JSON_SCHEMA_FOR_FUNC_DECL, True)
 
 tool = ExecuteCodeTool(
     tools=[my_fn_tool, McpToolset(...), OpenAPIToolset(...)],
-    backend=UnsafeLocalDockerBackend(image="ghcr.io/a2anet/adk-code-mode:latest"),
+    backend=UnsafeLocalDockerBackend(image="ghcr.io/buildagenta/adk-code-mode:latest"),
 )
 ```
 
@@ -117,8 +118,8 @@ print(send_message(channel="C123", text="hi"))
 ```bash
 # Push the sandbox image to Artifact Registry
 gcloud auth configure-docker <region>-docker.pkg.dev
-docker pull --platform linux/amd64 ghcr.io/a2anet/adk-code-mode:latest
-docker tag  ghcr.io/a2anet/adk-code-mode:latest \
+docker pull --platform linux/amd64 ghcr.io/buildagenta/adk-code-mode:latest
+docker tag  ghcr.io/buildagenta/adk-code-mode:latest \
     <region>-docker.pkg.dev/<project>/<repository>/adk-code-mode-sandbox:latest
 docker push <region>-docker.pkg.dev/<project>/<repository>/adk-code-mode-sandbox:latest
 
@@ -252,10 +253,10 @@ ExecuteCodeTool(tools=..., backend=..., save_tool_results_as_artifacts=True)
 
 ## 🐳 Sandbox Image
 
-The published base image (`ghcr.io/a2anet/adk-code-mode`) works as-is for tools whose execution is fully host-side. To bake in extra Python packages:
+The published base image (`ghcr.io/buildagenta/adk-code-mode`) works as-is for tools whose execution is fully host-side. To bake in extra Python packages:
 
 ```dockerfile
-FROM ghcr.io/a2anet/adk-code-mode:latest
+FROM ghcr.io/buildagenta/adk-code-mode:latest
 RUN pip install --no-cache-dir pandas==2.2.*
 ```
 
@@ -273,7 +274,7 @@ All settings are `ExecuteCodeTool` constructor arguments:
 | `max_code_mode_metadata_chars` | `50_000` | Budget for the whole rendered `<code-mode>` block, tags included. Oversized blocks degrade in tiers rather than disappearing — see [What the model sees](#what-the-model-sees). |
 | `max_output_chars` | `50_000` | Caps stdout/stderr handed back to the model. Overflow is saved as a session artifact at `code_mode/stdout/<call-id>.txt` and the model sees a head-and-tail view pointing to it. |
 | `max_code_chars` | `1_000_000` | Rejects oversized code payloads before starting a container. |
-| `timeout_seconds` | `60` | Caps overall execution time of one `execute_code` call, and with it every tool the block calls. `None` lifts the cap, leaving the platform request timeout (e.g. Cloud Run `--timeout`) as the only bound. |
+| `timeout_seconds` | `60` | Caps overall execution time of one `execute_code` call, and with it every tool the block calls. A block past the cap is stopped with its output so far, keeping the turn's variables and working directory. If it won't stop within a few seconds or leaves a worker thread or child process running, its sandbox is restarted and its output and state are lost. The model is told the cap. `None` lifts it, leaving the platform request timeout (e.g. Cloud Run `--timeout`) as the only bound. |
 | `per_tool_timeout_seconds` | `None` | Caps each individual tool call made from within the sandbox. |
 | `session_idle_timeout_seconds` | `600` | Idle reaper: closes a turn's container once it goes untouched this long. Backstop for turns that never call `release_invocation`. |
 
@@ -442,9 +443,9 @@ uv run pytest -m docker
 
 `adk-code-mode` is distributed under the terms of the [Apache-2.0](https://spdx.org/licenses/Apache-2.0.html) license.
 
-## 🤝 Join the A2A Net Community
+## 🤝 Join the Agenta Community
 
-A2A Net is a site to find and share AI agents and open-source community. Join to share your A2A agents, ask questions, stay up-to-date with the latest A2A news, be the first to hear about open-source releases, tutorials, and more!
+Agenta is a site to find and share AI agents and open-source community. Join to share your A2A agents, ask questions, stay up-to-date with the latest A2A news, be the first to hear about open-source releases, tutorials, and more!
 
-- 🌍 Site: [A2A Net](https://a2anet.com)
+- 🌍 Site: [Agenta](https://buildagenta.com)
 - 🤖 Discord: [Join the Discord](https://discord.gg/674NGXpAjU)

{adk_code_mode-1.6.1 → adk_code_mode-2.0.1}/docker/Dockerfile

@@ -1,6 +1,6 @@
 FROM python:3.13-slim
 
-LABEL org.opencontainers.image.source="https://github.com/a2anet/adk-code-mode"
+LABEL org.opencontainers.image.source="https://github.com/buildagenta/adk-code-mode"
 LABEL org.opencontainers.image.description="Sandbox base image for ADK Code Mode"
 LABEL org.opencontainers.image.licenses="Apache-2.0"
 

{adk_code_mode-1.6.1 → adk_code_mode-2.0.1}/pyproject.toml

@@ -11,7 +11,7 @@ requires-python = ">=3.10"
 license = "Apache-2.0"
 keywords = ["adk", "code-mode", "execute-code", "agents", "a2a"]
 authors = [
-  { name = "A2A Net", email = "hello@a2anet.com" },
+  { name = "Agenta", email = "hello@buildagenta.com" },
 ]
 classifiers = [
   "Development Status :: 4 - Beta",
@@ -47,9 +47,9 @@ dev = [
 ]
 
 [project.urls]
-Documentation = "https://github.com/A2ANet/adk-code-mode#readme"
-Issues = "https://github.com/A2ANet/adk-code-mode/issues"
-Source = "https://github.com/A2ANet/adk-code-mode"
+Documentation = "https://github.com/buildagenta/adk-code-mode#readme"
+Issues = "https://github.com/buildagenta/adk-code-mode/issues"
+Source = "https://github.com/buildagenta/adk-code-mode"
 
 [tool.hatch.version]
 path = "src/adk_code_mode/__about__.py"

adk_code_mode-2.0.1/sandbox-wheel/README.md

@@ -0,0 +1,14 @@
+# ADK Code Mode Sandbox
+
+In-sandbox shim for [`adk-code-mode`](https://github.com/buildagenta/adk-code-mode). The host package launches a container that has this wheel installed, then talks to it over a TCP control connection so model code can call ADK tools that run on the host.
+
+## 📄 License
+
+`adk-code-mode-sandbox` is distributed under the terms of the [Apache-2.0](https://spdx.org/licenses/Apache-2.0.html) license.
+
+## 🤝 Join the Agenta Community
+
+Agenta is a site to find and share AI agents and open-source community. Join to share your A2A agents, ask questions, stay up-to-date with the latest A2A news, be the first to hear about open-source releases, tutorials, and more!
+
+- 🌍 Site: [Agenta](https://buildagenta.com)
+- 🤖 Discord: [Join the Discord](https://discord.gg/674NGXpAjU)

{adk_code_mode-1.6.1 → adk_code_mode-2.0.1}/sandbox-wheel/pyproject.toml

@@ -11,7 +11,7 @@ requires-python = ">=3.10"
 license = "Apache-2.0"
 keywords = ["adk", "code-mode", "sandbox"]
 authors = [
-  { name = "A2A Net", email = "hello@a2anet.com" },
+  { name = "Agenta", email = "hello@buildagenta.com" },
 ]
 classifiers = [
   "Development Status :: 4 - Beta",
@@ -27,9 +27,9 @@ dependencies = [
 ]
 
 [project.urls]
-Documentation = "https://github.com/A2ANet/adk-code-mode#readme"
-Issues = "https://github.com/A2ANet/adk-code-mode/issues"
-Source = "https://github.com/A2ANet/adk-code-mode"
+Documentation = "https://github.com/buildagenta/adk-code-mode#readme"
+Issues = "https://github.com/buildagenta/adk-code-mode/issues"
+Source = "https://github.com/buildagenta/adk-code-mode"
 
 [tool.hatch.version]
 path = "src/adk_code_mode_sandbox/__about__.py"

adk_code_mode-2.0.1/sandbox-wheel/src/adk_code_mode_sandbox/__about__.py

@@ -0,0 +1,4 @@
+# SPDX-FileCopyrightText: 2025-present Agenta <hello@buildagenta.com>
+#
+# SPDX-License-Identifier: Apache-2.0
+__version__ = "2.0.1"  # x-release-please-version

{adk_code_mode-1.6.1 → adk_code_mode-2.0.1}/sandbox-wheel/src/adk_code_mode_sandbox/__init__.py

@@ -1,4 +1,4 @@
-# SPDX-FileCopyrightText: 2025-present A2A Net <hello@a2anet.com>
+# SPDX-FileCopyrightText: 2025-present Agenta <hello@buildagenta.com>
 #
 # SPDX-License-Identifier: Apache-2.0
 from adk_code_mode_sandbox.__about__ import __version__

{adk_code_mode-1.6.1 → adk_code_mode-2.0.1}/sandbox-wheel/src/adk_code_mode_sandbox/__main__.py

@@ -1,4 +1,4 @@
-# SPDX-FileCopyrightText: 2025-present A2A Net <hello@a2anet.com>
+# SPDX-FileCopyrightText: 2025-present Agenta <hello@buildagenta.com>
 #
 # SPDX-License-Identifier: Apache-2.0
 from adk_code_mode_sandbox._entry import main

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

@@ -1,4 +1,4 @@
-# SPDX-FileCopyrightText: 2025-present A2A Net <hello@a2anet.com>
+# SPDX-FileCopyrightText: 2025-present Agenta <hello@buildagenta.com>
 #
 # SPDX-License-Identifier: Apache-2.0
 """Sandbox entry point.
@@ -28,11 +28,13 @@ import io
 import os
 import platform
 import socket
+import subprocess
 import sys
+import threading
 import traceback
 from typing import Any
 
-from adk_code_mode_sandbox import _rpc_client
+from adk_code_mode_sandbox import _interrupt, _rpc_client
 from adk_code_mode_sandbox._rpc_client import RpcClient
 from adk_code_mode_sandbox.protocol import (
     DoneFrame,
@@ -184,7 +186,8 @@ def _exec_into(code: str, globs: dict[str, Any]) -> int:
         traceback.print_exc()
         return 1
     try:
-        exec(compiled, globs)
+        with _interrupt.running():
+            exec(compiled, globs)
     except SystemExit as exc:
         code_val = exc.code
         if code_val is None:
@@ -199,7 +202,54 @@ def _exec_into(code: str, globs: dict[str, Any]) -> int:
     return 0
 
 
-def run_block(code: str, globs: dict[str, Any]) -> tuple[str, str, int]:
+def _has_linux_child_processes(task_dir: str = "/proc/self/task") -> bool:
+    """Check each Linux thread for child processes still running."""
+    checked = False
+    try:
+        with os.scandir(task_dir) as tasks:
+            for task in tasks:
+                try:
+                    with open(f"{task.path}/children") as children:
+                        checked = True
+                        if children.read().strip():
+                            return True
+                except FileNotFoundError:
+                    continue
+    except OSError:
+        return True
+    return not checked
+
+
+def _has_child_processes() -> bool:
+    """Check whether user code left a child process running."""
+    if sys.platform == "linux":
+        return _has_linux_child_processes()
+    try:
+        probe = subprocess.Popen(["ps", "-axo", "pid=,ppid="], stdout=subprocess.PIPE, text=True)
+        try:
+            output, _ = probe.communicate(timeout=2)
+        except subprocess.TimeoutExpired:
+            probe.kill()
+            probe.wait()
+            return True
+    except OSError:
+        return True
+    if probe.returncode != 0:
+        return True
+    return any(
+        int(pid) != probe.pid and int(ppid) == os.getpid()
+        for pid, ppid in (line.split() for line in output.splitlines())
+    )
+
+
+def _has_background_work() -> bool:
+    """Check whether code remains active outside the interrupted execution thread."""
+    reader_thread = _rpc_client.get()._reader_thread
+    trusted = {threading.current_thread(), threading.main_thread(), reader_thread}
+    return any(thread not in trusted for thread in threading.enumerate()) or _has_child_processes()
+
+
+def run_block(code: str, globs: dict[str, Any]) -> tuple[str, str, int, bool]:
     """Run one code block with its stdout/stderr captured.
 
     Redirects ``sys.stdout`` / ``sys.stderr`` to in-memory buffers for the
@@ -216,7 +266,7 @@ def run_block(code: str, globs: dict[str, Any]) -> tuple[str, str, int]:
     finally:
         sys.stdout = old_stdout
         sys.stderr = old_stderr
-    return capture_out.getvalue(), capture_err.getvalue(), exit_code
+    return capture_out.getvalue(), capture_err.getvalue(), exit_code, _has_background_work()
 
 
 def main() -> int:
@@ -246,8 +296,8 @@ def main() -> int:
         if isinstance(frame, ShutdownFrame):
             return 0
         if isinstance(frame, RunFrame):
-            stdout_text, stderr_text, exit_code = run_block(frame.code, globs)
-            client.send(DoneFrame(exit_code=exit_code))
+            stdout_text, stderr_text, exit_code, background_work = run_block(frame.code, globs)
+            client.send(DoneFrame(exit_code=exit_code, background_work=background_work))
             client.send(OutputFrame(stdout=stdout_text, stderr=stderr_text, exit_code=exit_code))
             continue
         # Any other frame kind is a host protocol error; keep going but log.

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

@@ -1,4 +1,4 @@
-# SPDX-FileCopyrightText: 2025-present A2A Net <hello@a2anet.com>
+# SPDX-FileCopyrightText: 2025-present Agenta <hello@buildagenta.com>
 #
 # SPDX-License-Identifier: Apache-2.0
 """HTTP/WebSocket server mode for the sandbox (one connection per turn).
@@ -258,7 +258,7 @@ async def _run_one_block(
     _rpc_client.install(client)
     pump_task = asyncio.create_task(_pump_ws_to_reader(ws, bridge_reader))
     try:
-        stdout_text, stderr_text, exit_code = await loop.run_in_executor(
+        stdout_text, stderr_text, exit_code, background_work = await loop.run_in_executor(
             None, run_block, run_frame.code, globs
         )
     finally:
@@ -269,7 +269,9 @@ async def _run_one_block(
             pass
 
     # d-f. DoneFrame, OutputFrame, then the updated workspace tar (binary).
-    await ws.send(encode(DoneFrame(exit_code=exit_code)).decode("utf-8"))
+    await ws.send(
+        encode(DoneFrame(exit_code=exit_code, background_work=background_work)).decode("utf-8")
+    )
     await ws.send(
         encode(OutputFrame(stdout=stdout_text, stderr=stderr_text, exit_code=exit_code)).decode(
             "utf-8"

adk_code_mode-2.0.1/sandbox-wheel/src/adk_code_mode_sandbox/_interrupt.py

@@ -0,0 +1,67 @@
+# SPDX-FileCopyrightText: 2025-present Agenta <hello@buildagenta.com>
+#
+# SPDX-License-Identifier: Apache-2.0
+"""Stop the running code block when the host asks, keeping the sandbox's state.
+
+The host sends an ``InterruptFrame`` when a block runs past its timeout. Raising
+``CodeTimeout`` in the thread running the block ends it like any other exception,
+so its output so far, the globals and the working directory all survive for the
+turn's next block. Code stuck outside the interpreter (a C call, a long
+``time.sleep``) only sees the exception once it returns, which is why the host
+still falls back to restarting the sandbox after a grace period.
+"""
+
+from __future__ import annotations
+
+import ctypes
+import threading
+from collections.abc import Iterator
+from contextlib import contextmanager
+
+
+class CodeTimeout(BaseException):
+    """Raised in running code when the host stops it at its timeout.
+
+    A ``BaseException`` so the code's own ``except Exception`` can't swallow it.
+    """
+
+    message = ""
+
+    def __init__(self, *args: object) -> None:
+        # The interpreter raises an asynchronous exception without arguments.
+        super().__init__(*(args or (type(self).message,)))
+
+
+_lock = threading.Lock()
+_running_thread: int | None = None
+
+
+def _set_async_exc(thread_id: int, exc: type[BaseException] | None) -> None:
+    ctypes.pythonapi.PyThreadState_SetAsyncExc(
+        ctypes.c_ulong(thread_id), ctypes.py_object(exc) if exc else None
+    )
+
+
+@contextmanager
+def running() -> Iterator[None]:
+    """Mark the current thread as the one running a code block."""
+    global _running_thread
+    with _lock:
+        _running_thread = threading.get_ident()
+    try:
+        yield
+    finally:
+        with _lock:
+            _running_thread = None
+            # A stop that arrived as the block finished must not fire in the
+            # sandbox's own code afterwards.
+            _set_async_exc(threading.get_ident(), None)
+
+
+def interrupt(message: str) -> None:
+    """Raise ``CodeTimeout`` with ``message`` in the running block, if there is one."""
+    with _lock:
+        if _running_thread is None:
+            return
+        CodeTimeout.message = message
+        _set_async_exc(_running_thread, CodeTimeout)