axio 0.12.0__tar.gz → 0.12.2__tar.gz

{axio-0.12.0 → axio-0.12.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.5
 Name: axio
-Version: 0.12.0
+Version: 0.12.2
 Summary: Minimal, streaming-first, protocol-driven foundation for LLM-powered agents
 Project-URL: Documentation, https://docs.axio-agent.com
 Project-URL: Homepage, https://github.com/mosquito/axio-agent

{axio-0.12.0 → axio-0.12.2}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "axio"
-version = "0.12.0"  # bumped to the git tag by CI on publish
+version = "0.12.2"  # bumped to the git tag by CI on publish
 description = "Minimal, streaming-first, protocol-driven foundation for LLM-powered agents"
 readme = "README.md"
 requires-python = ">=3.12"

{axio-0.12.0 → axio-0.12.2}/src/axio/credentials/codec.py

@@ -6,7 +6,7 @@ import hashlib
 import json
 import math
 import types
-from dataclasses import asdict, fields, is_dataclass
+from dataclasses import MISSING, asdict, fields, is_dataclass
 from enum import Enum
 from typing import Annotated, Any, Literal, Union, cast, get_args, get_origin, get_type_hints
 
@@ -98,11 +98,20 @@ def restore(value: Any, annotation: Any) -> Any:
             raise ValueError
         return {key: restore(item, args[1]) for key, item in value.items()}
     if isinstance(annotation, type) and is_dataclass(annotation) and isinstance(value, dict):
-        declared = {item.name for item in fields(annotation) if item.init}
-        if value.keys() != declared:
+        declared = {item.name: item for item in fields(annotation) if item.init}
+        # Schema additions must explicitly opt into restoring their default from
+        # older records. Ordinary defaults (especially recovery markers) remain
+        # required so a damaged record cannot silently reset security state.
+        optional = {
+            name
+            for name, item in declared.items()
+            if item.metadata.get("credential_optional") is True
+            and (item.default is not MISSING or item.default_factory is not MISSING)
+        }
+        if not value.keys() <= declared.keys() or not declared.keys() - optional <= value.keys():
             raise ValueError
         hints = get_type_hints(annotation, include_extras=True)
-        return annotation(**{name: restore(value[name], hints[name]) for name in declared})
+        return annotation(**{name: restore(item, hints[name]) for name, item in value.items()})
     raise ValueError
 
 

{axio-0.12.0 → axio-0.12.2}/src/axio/models.py

@@ -15,6 +15,7 @@ class Capability(StrEnum):
     audio = "audio"
     # Output modalities
     image_generation = "image_generation"
+    image_editing = "image_editing"
     video_generation = "video_generation"
     # Processing capabilities
     reasoning = "reasoning"

{axio-0.12.0 → axio-0.12.2}/src/axio/transport.py

@@ -11,7 +11,7 @@ from dataclasses import dataclass, field
 from pathlib import Path
 from typing import Any, Protocol, runtime_checkable
 
-from .blocks import ContentBlock
+from .blocks import ContentBlock, ImageBlock
 from .events import StreamEvent
 from .messages import Message
 from .models import ModelSpec
@@ -68,6 +68,25 @@ class ImageGenTransport(Protocol):
     async def generate_images(self, prompt: str, *, model: str | None = None, n: int = 1) -> list[bytes]: ...
 
 
+@runtime_checkable
+class ImageEditTransport(Protocol):
+    """Edit supplied images according to a prompt, returning complete image bytes.
+
+    Sources are required. Implementations must reject unsupported masks instead
+    of ignoring them; supported formats and counts are provider-specific.
+    """
+
+    async def edit_images(
+        self,
+        prompt: str,
+        *,
+        images: list[ImageBlock],
+        mask: ImageBlock | None = None,
+        model: str | None = None,
+        n: int = 1,
+    ) -> list[bytes]: ...
+
+
 @runtime_checkable
 class VideoGenTransport(Protocol):
     """Generate ``n`` video samples for a text prompt.  Returns raw video bytes
@@ -205,6 +224,20 @@ class DummyImageGenTransport(DummyTransport, ImageGenTransport):
         raise AssertionError("unreachable")
 
 
+class DummyImageEditTransport(DummyTransport, ImageEditTransport):
+    async def edit_images(
+        self,
+        prompt: str,
+        *,
+        images: list[ImageBlock],
+        mask: ImageBlock | None = None,
+        model: str | None = None,
+        n: int = 1,
+    ) -> list[bytes]:
+        self._do_fail()
+        raise AssertionError("unreachable")
+
+
 class DummyVideoGenTransport(DummyTransport, VideoGenTransport):
     async def generate_videos(
         self,

axio-0.12.2/tests/test_credential_schema_evolution.py

@@ -0,0 +1,58 @@
+"""Credential schema additions preserve old records without hiding missing state."""
+
+from __future__ import annotations
+
+from dataclasses import dataclass, field
+from typing import Any
+
+import pytest
+
+from axio.credentials import CredentialCodec, CredentialError
+
+
+@dataclass(frozen=True, slots=True)
+class Account:
+    key: str = field(metadata={"credential_key": True})
+    native: bool = field(default=False, metadata={"credential_optional": True})
+    tags: list[str] = field(default_factory=list, metadata={"credential_optional": True})
+
+
+@dataclass(frozen=True, slots=True)
+class Record:
+    account: Account
+    operation: str | None = None
+
+
+def test_explicit_optional_fields_restore_defaults_and_round_trip() -> None:
+    codec = CredentialCodec(Record)
+    old = {"version": 1, "account": {"key": "account"}, "operation": "refresh"}
+    restored = codec.from_dict(old)
+    assert restored == Record(Account("account"), operation="refresh")
+    assert codec.decode(codec.encode(restored)) == restored
+    assert codec.from_dict(old).account.tags is not restored.account.tags
+    assert old["account"] == {"key": "account"}
+
+
+@pytest.mark.parametrize(
+    "data",
+    [
+        {"version": 1, "account": {"key": "account"}},
+        {"version": 1, "account": {"key": "account", "unknown": False}, "operation": None},
+        {"version": 1, "account": {"key": "account", "native": "false"}, "operation": None},
+        {"version": 1, "account": {"key": "account", "native": None}, "operation": None},
+        {"version": 1, "account": {"key": "account", "tags": [1]}, "operation": None},
+        {"version": 1, "account": {}, "operation": None},
+    ],
+)
+def test_optional_fields_do_not_weaken_record_validation(data: dict[str, Any]) -> None:
+    with pytest.raises(CredentialError):
+        CredentialCodec(Record).from_dict(data)
+
+
+def test_optional_metadata_without_default_does_not_remove_requirement() -> None:
+    @dataclass(frozen=True, slots=True)
+    class Required:
+        key: str = field(metadata={"credential_key": True, "credential_optional": True})
+
+    with pytest.raises(CredentialError):
+        CredentialCodec(Required).from_dict({"version": 1})