adk-code-mode 1.0.0__tar.gz → 1.2.0__tar.gz

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

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

{adk_code_mode-1.0.0 → adk_code_mode-1.2.0}/CHANGELOG.md

@@ -1,5 +1,19 @@
 # Changelog
 
+## [1.2.0](https://github.com/a2anet/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/a2anet/adk-code-mode/commit/dfd9a59bf1718084a1d5c02cc1ea453269223f16))
+
+## [1.1.0](https://github.com/a2anet/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/a2anet/adk-code-mode/commit/acfb704d388da5d66e318775e03fe055467679c2))
+
 ## [1.0.0](https://github.com/a2anet/adk-code-mode/compare/adk-code-mode-v0.4.1...adk-code-mode-v1.0.0) (2026-07-15)
 
 

{adk_code_mode-1.0.0 → adk_code_mode-1.2.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: adk-code-mode
-Version: 1.0.0
+Version: 1.2.0
 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.0.0 → adk_code_mode-1.2.0}/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.0.0"  # x-release-please-version
+__version__ = "1.2.0"  # x-release-please-version

{adk_code_mode-1.0.0 → adk_code_mode-1.2.0}/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.0.0"  # x-release-please-version
+__version__ = "1.2.0"  # x-release-please-version

{adk_code_mode-1.0.0 → adk_code_mode-1.2.0}/src/adk_code_mode/_artifact_tools.py

@@ -43,6 +43,7 @@ async def save_artifact(
     filename: str,
     content: str,
     mime_type: str | None = None,
+    custom_metadata: dict[str, Any] | None = None,
     tool_context: ToolContext,
 ) -> int:
     """Save an artifact to the session. Returns the new version number.
@@ -62,6 +63,8 @@ async def save_artifact(
             content=base64.b64encode(image_bytes).decode("ascii"),
             mime_type="image/png",
         )
+
+    ``custom_metadata`` is custom metadata to associate with the artifact.
     """
     if _is_text_mime(mime_type):
         part = genai_types.Part(
@@ -77,7 +80,7 @@ async def save_artifact(
                 mime_type=mime_type or "application/octet-stream",
             )
         )
-    return await tool_context.save_artifact(filename, part)
+    return await tool_context.save_artifact(filename, part, custom_metadata=custom_metadata)
 
 
 async def load_artifact(

{adk_code_mode-1.0.0 → adk_code_mode-1.2.0}/src/adk_code_mode/tool.py

@@ -294,6 +294,20 @@ class ExecuteCodeTool(BaseTool):
         if eager_resolved:
             namespacing.build(eager_resolved)
 
+    @property
+    def tools(self) -> Sequence[BaseTool | BaseToolset]:
+        """Tools and toolsets available inside the sandbox, including built-in artifact tools.
+
+        Host apps that own a toolset's lifecycle (e.g. an ``McpToolset`` needing
+        ``close()`` on shutdown) need this to find it again after construction.
+        """
+        return self._tools
+
+    @property
+    def backend(self) -> SandboxBackend:
+        """The configured sandbox runtime backend."""
+        return self._backend
+
     def _get_declaration(self) -> genai_types.FunctionDeclaration | None:
         return genai_types.FunctionDeclaration(
             name=self.name,

{adk_code_mode-1.0.0 → adk_code_mode-1.2.0}/tests/test_artifact_tools.py

@@ -34,12 +34,17 @@ class _StubToolContext:
     ) -> None:
         self._load_result = load_result
         self._list_result = list_result or []
-        self.saved: list[tuple[str, genai_types.Part]] = []
+        self.saved: list[tuple[str, genai_types.Part, dict[str, Any] | None]] = []
         self.loaded: list[tuple[str, int | None]] = []
         self.listed = 0
 
-    async def save_artifact(self, filename: str, part: genai_types.Part) -> int:
-        self.saved.append((filename, part))
+    async def save_artifact(
+        self,
+        filename: str,
+        part: genai_types.Part,
+        custom_metadata: dict[str, Any] | None = None,
+    ) -> int:
+        self.saved.append((filename, part, custom_metadata))
         return len(self.saved)
 
     async def load_artifact(
@@ -65,7 +70,7 @@ async def test_save_artifact_encodes_text_mime_as_utf8() -> None:
 
     assert version == 1
     assert len(ctx.saved) == 1
-    name, part = ctx.saved[0]
+    name, part, _ = ctx.saved[0]
     assert name == "report.json"
     assert part.inline_data is not None
     assert part.inline_data.mime_type == "application/json"
@@ -82,7 +87,7 @@ async def test_save_artifact_treats_text_plus_subtype_as_text() -> None:
         tool_context=ctx,  # type: ignore[arg-type]
     )
 
-    _, part = ctx.saved[0]
+    _, part, _ = ctx.saved[0]
     assert part.inline_data is not None
     assert part.inline_data.data == b"<svg/>"
 
@@ -98,7 +103,7 @@ async def test_save_artifact_decodes_base64_for_binary_mime() -> None:
         tool_context=ctx,  # type: ignore[arg-type]
     )
 
-    _, part = ctx.saved[0]
+    _, part, _ = ctx.saved[0]
     assert part.inline_data is not None
     assert part.inline_data.data == raw
     assert part.inline_data.mime_type == "application/octet-stream"
@@ -114,12 +119,41 @@ async def test_save_artifact_defaults_mime_when_none() -> None:
         tool_context=ctx,  # type: ignore[arg-type]
     )
 
-    _, part = ctx.saved[0]
+    _, part, _ = ctx.saved[0]
     assert part.inline_data is not None
     assert part.inline_data.mime_type == "application/octet-stream"
     assert part.inline_data.data == raw
 
 
+@pytest.mark.asyncio
+async def test_save_artifact_forwards_custom_metadata() -> None:
+    ctx = _StubToolContext()
+    await save_artifact(
+        filename="report.md",
+        content="# Report",
+        mime_type="text/markdown",
+        custom_metadata={"a2anet_agent_run.send_as_a2a_artifact": "true"},
+        tool_context=ctx,  # type: ignore[arg-type]
+    )
+
+    _, _, custom_metadata = ctx.saved[0]
+    assert custom_metadata == {"a2anet_agent_run.send_as_a2a_artifact": "true"}
+
+
+@pytest.mark.asyncio
+async def test_save_artifact_defaults_custom_metadata_to_none() -> None:
+    ctx = _StubToolContext()
+    await save_artifact(
+        filename="scratch.txt",
+        content="notes",
+        mime_type="text/plain",
+        tool_context=ctx,  # type: ignore[arg-type]
+    )
+
+    _, _, custom_metadata = ctx.saved[0]
+    assert custom_metadata is None
+
+
 @pytest.mark.asyncio
 async def test_load_artifact_returns_text_kind_for_text_mime() -> None:
     part = genai_types.Part(
@@ -196,4 +230,4 @@ def test_artifact_tools_declarations_omit_tool_context_param() -> None:
     assert save_decl is not None
     properties: dict[str, Any] = save_decl.parameters.properties or {}  # type: ignore[union-attr]
     assert "tool_context" not in properties
-    assert {"filename", "content"}.issubset(properties.keys())
+    assert {"filename", "content", "custom_metadata"}.issubset(properties.keys())

{adk_code_mode-1.0.0 → adk_code_mode-1.2.0}/tests/test_tool.py

@@ -69,6 +69,16 @@ def test_declaration_exposes_a_single_code_parameter() -> None:
     assert decl.parameters_json_schema["required"] == ["code"]
 
 
+def test_tools_and_backend_properties_expose_the_configured_surface() -> None:
+    schema_tool = _SchemaTool("ping", "Ping.")
+    backend = FakeRuntime()
+
+    tool = ExecuteCodeTool(tools=[schema_tool], backend=backend)
+
+    assert schema_tool in tool.tools
+    assert tool.backend is backend
+
+
 @pytest.mark.asyncio
 async def test_process_llm_request_declares_the_tool() -> None:
     tool = _make_tool([])

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

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