adk-code-mode 1.1.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.1.0 → adk_code_mode-1.2.0}/CHANGELOG.md

@@ -1,5 +1,12 @@
 # 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)
 
 

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: adk-code-mode
-Version: 1.1.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.1.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.1.0"  # x-release-please-version
+__version__ = "1.2.0"  # x-release-please-version

{adk_code_mode-1.1.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.1.0"  # x-release-please-version
+__version__ = "1.2.0"  # x-release-please-version

{adk_code_mode-1.1.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.1.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.1.0/.release-please-manifest.json

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