axio 0.3.4__tar.gz → 0.4.0__tar.gz

{axio-0.3.4 → axio-0.4.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: axio
-Version: 0.3.4
+Version: 0.4.0
 Summary: Minimal, streaming-first, protocol-driven foundation for LLM-powered agents
 Project-URL: Homepage, https://github.com/axio-agent/axio
 Project-URL: Repository, https://github.com/axio-agent/axio
@@ -77,7 +77,7 @@ asyncio.run(main())
        ▼
   ┌─────────┐   stream()    ┌─────────────────────┐
   │  Agent  │ ────────────▶ │ CompletionTransport  │
-  │  loop   │ ◀──────────── │ (OpenAI, Nebius, …)  │
+  │  loop   │ ◀──────────── │ (Anthropic, OpenAI, …) │
   └─────────┘  StreamEvent  └─────────────────────┘
        │
        │ tool_use?
@@ -202,6 +202,7 @@ my_guard = "my_package:MyGuard"
 
 | Package | Purpose |
 |---|---|
+| [axio-transport-anthropic](https://github.com/axio-agent/axio-transport-anthropic) | Anthropic Claude transport |
 | [axio-transport-openai](https://github.com/axio-agent/axio-transport-openai) | OpenAI-compatible transport |
 | [axio-transport-nebius](https://github.com/axio-agent/axio-transport-nebius) | Nebius AI Studio transport |
 | [axio-transport-codex](https://github.com/axio-agent/axio-transport-codex) | ChatGPT OAuth transport |

{axio-0.3.4 → axio-0.4.0}/README.md

@@ -64,7 +64,7 @@ asyncio.run(main())
        ▼
   ┌─────────┐   stream()    ┌─────────────────────┐
   │  Agent  │ ────────────▶ │ CompletionTransport  │
-  │  loop   │ ◀──────────── │ (OpenAI, Nebius, …)  │
+  │  loop   │ ◀──────────── │ (Anthropic, OpenAI, …) │
   └─────────┘  StreamEvent  └─────────────────────┘
        │
        │ tool_use?
@@ -189,6 +189,7 @@ my_guard = "my_package:MyGuard"
 
 | Package | Purpose |
 |---|---|
+| [axio-transport-anthropic](https://github.com/axio-agent/axio-transport-anthropic) | Anthropic Claude transport |
 | [axio-transport-openai](https://github.com/axio-agent/axio-transport-openai) | OpenAI-compatible transport |
 | [axio-transport-nebius](https://github.com/axio-agent/axio-transport-nebius) | Nebius AI Studio transport |
 | [axio-transport-codex](https://github.com/axio-agent/axio-transport-codex) | ChatGPT OAuth transport |

{axio-0.3.4 → axio-0.4.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "axio"
-version = "0.3.4"
+version = "0.4.0"
 description = "Minimal, streaming-first, protocol-driven foundation for LLM-powered agents"
 readme = "README.md"
 requires-python = ">=3.12"

{axio-0.3.4 → axio-0.4.0}/src/axio/agent.py

@@ -5,7 +5,7 @@ from __future__ import annotations
 import asyncio
 import json
 import logging
-from collections.abc import AsyncGenerator
+from collections.abc import AsyncGenerator, Iterable
 from dataclasses import dataclass, field
 from typing import Any
 
@@ -22,6 +22,7 @@ from axio.events import (
     ToolUseStart,
 )
 from axio.messages import Message
+from axio.selector import ToolSelector
 from axio.stream import AgentStream
 from axio.tool import Tool
 from axio.transport import CompletionTransport
@@ -35,6 +36,7 @@ class Agent:
     system: str
     tools: list[Tool]
     transport: CompletionTransport
+    selector: ToolSelector | None = field(default=None)
     max_iterations: int = field(default=50)
 
     def run_stream(self, user_message: str, context: ContextStore) -> AgentStream:
@@ -120,6 +122,13 @@ class Agent:
             blocks.append(ToolUseBlock(id=tid, name=info["name"], input=inp))
         return blocks, malformed
 
+    async def _select_tools(self, history: list[Message], tools: list[Tool]) -> Iterable[Tool]:
+        if not tools:
+            return []
+        if not self.selector:
+            return tools
+        return await self.selector.select(history, tools)
+
     async def _run_loop(self, user_message: str, context: ContextStore) -> AsyncGenerator[StreamEvent, None]:
         total_usage = Usage(0, 0)
         session_end_emitted = False
@@ -129,8 +138,7 @@ class Agent:
             for iteration in range(1, self.max_iterations + 1):
                 history = await context.get_history()
                 logger.info("Iteration %d, history length=%d", iteration, len(history))
-                active_tools = self.tools
-
+                active_tools = list(await self._select_tools(history, self.tools))
                 content: list[TextBlock | ToolUseBlock] = []
                 pending: dict[str, dict[str, Any]] = {}
                 stop_reason = StopReason.end_turn

{axio-0.3.4 → axio-0.4.0}/src/axio/events.py

@@ -34,6 +34,28 @@ class ToolInputDelta:
     partial_json: str
 
 
+@dataclass(frozen=True, slots=True)
+class ToolFieldStart:
+    index: int
+    tool_use_id: ToolCallID
+    key: str
+
+
+@dataclass(frozen=True, slots=True)
+class ToolFieldDelta:
+    index: int
+    tool_use_id: ToolCallID
+    key: str
+    text: str
+
+
+@dataclass(frozen=True, slots=True)
+class ToolFieldEnd:
+    index: int
+    tool_use_id: ToolCallID
+    key: str
+
+
 @dataclass(frozen=True, slots=True)
 class ToolResult:
     tool_use_id: ToolCallID
@@ -62,5 +84,15 @@ class SessionEndEvent:
 
 
 type StreamEvent = (
-    ReasoningDelta | TextDelta | ToolUseStart | ToolInputDelta | ToolResult | IterationEnd | Error | SessionEndEvent
+    ReasoningDelta
+    | TextDelta
+    | ToolUseStart
+    | ToolInputDelta
+    | ToolFieldStart
+    | ToolFieldDelta
+    | ToolFieldEnd
+    | ToolResult
+    | IterationEnd
+    | Error
+    | SessionEndEvent
 )

{axio-0.3.4 → axio-0.4.0}/src/axio/models.py

@@ -7,15 +7,6 @@ from dataclasses import dataclass
 from enum import StrEnum
 
 
-@dataclass(frozen=True, slots=True)
-class TransportMeta:
-    """Metadata a transport plugin declares about itself."""
-
-    label: str
-    api_key_env: str
-    role_defaults: dict[str, str]
-
-
 class Capability(StrEnum):
     text = "text"
     vision = "vision"

axio-0.4.0/src/axio/selector.py

@@ -0,0 +1,10 @@
+from collections.abc import Iterable
+from typing import Protocol, runtime_checkable
+
+from axio.messages import Message
+from axio.tool import Tool
+
+
+@runtime_checkable
+class ToolSelector(Protocol):
+    async def select(self, messages: Iterable[Message], tools: Iterable[Tool]) -> Iterable[Tool]: ...

axio-0.4.0/src/axio/tool_args.py

@@ -0,0 +1,243 @@
+"""Incremental streaming parser for tool call JSON arguments.
+
+Feeds partial JSON chunks (from ``ToolInputDelta.partial_json``) and emits
+structured ``ToolField*`` events as top-level object fields are discovered.
+
+Top-level *string* values are decoded (escape sequences resolved, quotes
+stripped).  All other top-level values are emitted as raw JSON fragments.
+"""
+
+from __future__ import annotations
+
+from collections.abc import Mapping
+from enum import IntEnum
+from types import MappingProxyType
+
+from axio.events import ToolFieldDelta, ToolFieldEnd, ToolFieldStart
+
+type ToolFieldEvent = ToolFieldStart | ToolFieldDelta | ToolFieldEnd
+
+
+class State(IntEnum):
+    INIT = 0
+    OBJ = 1
+    KEY = 2
+    COLON = 3
+    VAL = 4
+    STR = 5
+    RAW = 6
+    AFTER = 7
+    ESC = 8
+    UESC = 9
+
+
+ESCAPES: Mapping[str, str] = MappingProxyType(
+    {
+        "n": "\n",
+        "t": "\t",
+        "r": "\r",
+        "b": "\b",
+        "f": "\f",
+        '"': '"',
+        "\\": "\\",
+        "/": "/",
+    }
+)
+
+
+class ToolArgStream:
+    """O(1)-per-character streaming parser for tool argument JSON.
+
+    Usage::
+
+        stream = ToolArgStream("call_1")
+        events = stream.feed('{"path":"/tmp/f')
+        # [ToolFieldStart(0, "call_1", "path"), ToolFieldDelta(0, "call_1", "path", "/tmp/f")]
+        events = stream.feed('oo.py"}')
+        # [ToolFieldDelta(0, "call_1", "path", "oo.py"), ToolFieldEnd(0, "call_1", "path")]
+    """
+
+    __slots__ = (
+        "_id",
+        "_idx",
+        "_st",
+        "_key_chars",
+        "_key",
+        "_buf",
+        "_u",
+        "_high",
+        "_depth",
+        "_raw_str",
+        "_raw_esc",
+        "_esc_key",
+        "_esc_ret",
+        "_events",
+    )
+
+    def __init__(self, tool_use_id: str, index: int = 0) -> None:
+        self._id = tool_use_id
+        self._idx = index
+        self._st = State.INIT
+        self._key_chars: list[str] = []
+        self._key = ""
+        self._buf: list[str] = []
+        self._u: list[str] = []
+        self._high = 0
+        self._depth = 0
+        self._raw_str = False
+        self._raw_esc = False
+        self._esc_key = False
+        self._esc_ret = State.KEY
+        self._events: list[ToolFieldEvent] = []
+
+    @property
+    def current_key(self) -> str:
+        """The field currently being streamed, or ``""``."""
+        return self._key
+
+    def feed(self, chunk: str) -> list[ToolFieldEvent]:
+        """Process a partial JSON chunk and return any field events produced."""
+        self._events = []
+        for ch in chunk:
+            self._step(ch)
+        self._flush()
+        return self._events
+
+    def _flush(self) -> None:
+        if self._buf:
+            self._events.append(ToolFieldDelta(self._idx, self._id, self._key, "".join(self._buf)))
+            self._buf.clear()
+
+    def _start(self) -> None:
+        self._flush()
+        self._events.append(ToolFieldStart(self._idx, self._id, self._key))
+
+    def _end(self) -> None:
+        self._flush()
+        self._events.append(ToolFieldEnd(self._idx, self._id, self._key))
+
+    def _step(self, ch: str) -> None:  # noqa: PLR0912
+        match self._st:
+            case State.INIT:
+                if ch == "{":
+                    self._st = State.OBJ
+
+            case State.OBJ:
+                if ch == '"':
+                    self._key_chars.clear()
+                    self._st = State.KEY
+                elif ch == "}":
+                    self._st = State.INIT
+
+            case State.KEY:
+                if ch == "\\":
+                    self._esc_key = True
+                    self._esc_ret = State.KEY
+                    self._st = State.ESC
+                elif ch == '"':
+                    self._key = "".join(self._key_chars)
+                    self._st = State.COLON
+                else:
+                    self._key_chars.append(ch)
+
+            case State.COLON:
+                if ch == ":":
+                    self._start()
+                    self._st = State.VAL
+
+            case State.VAL:
+                if ch in " \t\r\n":
+                    pass
+                elif ch == '"':
+                    self._st = State.STR
+                else:
+                    self._buf.append(ch)
+                    self._depth = 1 if ch in "{[" else 0
+                    self._raw_str = False
+                    self._raw_esc = False
+                    self._st = State.RAW
+
+            case State.STR:
+                if ch == "\\":
+                    self._esc_key = False
+                    self._esc_ret = State.STR
+                    self._st = State.ESC
+                elif ch == '"':
+                    if self._high:
+                        self._buf.append("\ufffd")
+                        self._high = 0
+                    self._end()
+                    self._st = State.AFTER
+                else:
+                    if self._high:
+                        self._buf.append("\ufffd")
+                        self._high = 0
+                    self._buf.append(ch)
+
+            case State.RAW:
+                if self._raw_str:
+                    self._buf.append(ch)
+                    if self._raw_esc:
+                        self._raw_esc = False
+                    elif ch == "\\":
+                        self._raw_esc = True
+                    elif ch == '"':
+                        self._raw_str = False
+                elif self._depth == 0 and ch in " \t\r\n,}":
+                    # simple value (number/bool/null) ends on whitespace or delimiter
+                    self._end()
+                    self._st = State.AFTER
+                    if ch in ",}":
+                        self._step(ch)  # reprocess delimiter
+                elif ch == '"':
+                    self._buf.append(ch)
+                    self._raw_str = True
+                elif ch in "{[":
+                    self._buf.append(ch)
+                    self._depth += 1
+                elif ch in "}]":
+                    self._buf.append(ch)
+                    self._depth -= 1
+                    if self._depth == 0:
+                        self._end()
+                        self._st = State.AFTER
+                else:
+                    self._buf.append(ch)
+
+            case State.AFTER:
+                if ch == ",":
+                    self._st = State.OBJ
+                elif ch == "}":
+                    self._st = State.INIT
+
+            case State.ESC:
+                if ch == "u":
+                    self._u.clear()
+                    self._st = State.UESC
+                else:
+                    dec = ESCAPES.get(ch, ch)
+                    if self._esc_key:
+                        self._key_chars.append(dec)
+                    else:
+                        self._buf.append(dec)
+                    self._st = self._esc_ret
+
+            case State.UESC:
+                self._u.append(ch)
+                if len(self._u) == 4:
+                    code = int("".join(self._u), 16)
+                    if self._esc_key:
+                        self._key_chars.append(chr(code))
+                    elif self._high:
+                        if 0xDC00 <= code <= 0xDFFF:
+                            full = 0x10000 + (self._high - 0xD800) * 0x400 + (code - 0xDC00)
+                            self._buf.append(chr(full))
+                        else:
+                            self._buf.append("\ufffd")
+                            self._buf.append(chr(code))
+                        self._high = 0
+                    elif 0xD800 <= code <= 0xDBFF:
+                        self._high = code
+                    else:
+                        self._buf.append(chr(code))
+                    self._st = self._esc_ret