axio 0.3.5__tar.gz → 0.4.1__tar.gz

{axio-0.3.5 → axio-0.4.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: axio
-Version: 0.3.5
+Version: 0.4.1
 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,8 +202,8 @@ my_guard = "my_package:MyGuard"
 
 | Package | Purpose |
 |---|---|
-| [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-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 (OpenAI, Nebius, OpenRouter, custom) |
 | [axio-transport-codex](https://github.com/axio-agent/axio-transport-codex) | ChatGPT OAuth transport |
 | [axio-tools-local](https://github.com/axio-agent/axio-tools-local) | Shell, file, Python tools |
 | [axio-tools-mcp](https://github.com/axio-agent/axio-tools-mcp) | MCP server bridge |

{axio-0.3.5 → axio-0.4.1}/README.md

@@ -64,7 +64,7 @@ asyncio.run(main())
        ▼
   ┌─────────┐   stream()    ┌─────────────────────┐
   │  Agent  │ ────────────▶ │ CompletionTransport  │
-  │  loop   │ ◀──────────── │ (OpenAI, Nebius, …)  │
+  │  loop   │ ◀──────────── │ (Anthropic, OpenAI, …) │
   └─────────┘  StreamEvent  └─────────────────────┘
        │
        │ tool_use?
@@ -189,8 +189,8 @@ my_guard = "my_package:MyGuard"
 
 | Package | Purpose |
 |---|---|
-| [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-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 (OpenAI, Nebius, OpenRouter, custom) |
 | [axio-transport-codex](https://github.com/axio-agent/axio-transport-codex) | ChatGPT OAuth transport |
 | [axio-tools-local](https://github.com/axio-agent/axio-tools-local) | Shell, file, Python tools |
 | [axio-tools-mcp](https://github.com/axio-agent/axio-tools-mcp) | MCP server bridge |

{axio-0.3.5 → axio-0.4.1}/pyproject.toml

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

{axio-0.3.5 → axio-0.4.1}/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.5 → axio-0.4.1}/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.1/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.3.5/src/axio/selector.py

@@ -1,121 +0,0 @@
-"""Tool selector: choose relevant tools per query via embedding similarity."""
-
-from __future__ import annotations
-
-import math
-from collections.abc import AsyncIterator
-from typing import Protocol, runtime_checkable
-
-from axio.blocks import TextBlock, ToolResultBlock
-from axio.events import StreamEvent
-from axio.messages import Message
-from axio.tool import Tool
-from axio.transport import CompletionTransport, EmbeddingTransport
-
-
-@runtime_checkable
-class ToolSelector(Protocol):
-    async def select(self, messages: list[Message], tools: list[Tool]) -> list[Tool]: ...
-
-
-def _cosine_similarity(a: list[float], b: list[float]) -> float:
-    dot = sum(x * y for x, y in zip(a, b))
-    norm_a = math.sqrt(sum(x * x for x in a))
-    norm_b = math.sqrt(sum(x * x for x in b))
-    if norm_a == 0.0 or norm_b == 0.0:
-        return 0.0
-    return dot / (norm_a * norm_b)
-
-
-def _extract_last_user_text(messages: list[Message]) -> str | None:
-    """Return joined text from the last user message, or None if it's a tool-result iteration."""
-    for msg in reversed(messages):
-        if msg.role != "user":
-            continue
-        texts = [b.text for b in msg.content if isinstance(b, TextBlock)]
-        if texts:
-            return " ".join(texts)
-        # User message with only ToolResultBlocks → tool-result iteration
-        if any(isinstance(b, ToolResultBlock) for b in msg.content):
-            return None
-        return None
-    return None
-
-
-class EmbeddingToolSelector:
-    def __init__(
-        self,
-        transport: EmbeddingTransport,
-        *,
-        top_k: int = 5,
-        pinned: frozenset[str] = frozenset(),
-    ) -> None:
-        self._transport = transport
-        self._top_k = top_k
-        self._pinned = pinned
-        self._tool_embeddings: dict[str, list[float]] = {}
-        self._tool_descriptions: dict[str, str] = {}
-
-    async def _ensure_embeddings(self, tools: list[Tool]) -> None:
-        to_embed: list[tuple[str, str]] = []
-        for tool in tools:
-            cached_desc = self._tool_descriptions.get(tool.name)
-            if cached_desc is None or cached_desc != tool.description:
-                to_embed.append((tool.name, tool.description))
-
-        if not to_embed:
-            return
-
-        texts = [desc for _, desc in to_embed]
-        vectors = await self._transport.embed(texts)
-        for (name, desc), vec in zip(to_embed, vectors):
-            self._tool_embeddings[name] = vec
-            self._tool_descriptions[name] = desc
-
-    async def select(self, messages: list[Message], tools: list[Tool]) -> list[Tool]:
-        if len(tools) <= self._top_k:
-            return tools
-
-        query = _extract_last_user_text(messages)
-        if query is None:
-            return tools
-
-        await self._ensure_embeddings(tools)
-
-        query_vec = (await self._transport.embed([query]))[0]
-
-        pinned_tools: list[Tool] = []
-        scorable: list[tuple[float, Tool]] = []
-
-        for tool in tools:
-            if tool.name in self._pinned:
-                pinned_tools.append(tool)
-            elif tool.name in self._tool_embeddings:
-                score = _cosine_similarity(query_vec, self._tool_embeddings[tool.name])
-                scorable.append((score, tool))
-
-        scorable.sort(key=lambda x: x[0], reverse=True)
-        remaining_slots = max(0, self._top_k - len(pinned_tools))
-        selected = pinned_tools + [t for _, t in scorable[:remaining_slots]]
-        return selected
-
-
-class ToolFilteringTransport:
-    """CompletionTransport decorator that filters tools via a ToolSelector."""
-
-    def __init__(self, transport: CompletionTransport, selector: ToolSelector) -> None:
-        self._transport = transport
-        self._selector = selector
-
-    def stream(self, messages: list[Message], tools: list[Tool], system: str) -> AsyncIterator[StreamEvent]:
-        return self._filtered_stream(messages, tools, system)
-
-    async def _filtered_stream(
-        self,
-        messages: list[Message],
-        tools: list[Tool],
-        system: str,
-    ) -> AsyncIterator[StreamEvent]:
-        selected = await self._selector.select(messages, tools)
-        async for event in self._transport.stream(messages, selected, system):
-            yield event

axio-0.3.5/tests/test_selector.py

@@ -1,237 +0,0 @@
-"""Tests for axon.selector — ToolSelector protocol, EmbeddingToolSelector, ToolFilteringTransport."""
-
-from __future__ import annotations
-
-import math
-from collections.abc import AsyncGenerator
-
-import pytest
-
-from axio.blocks import TextBlock, ToolResultBlock
-from axio.events import StreamEvent, TextDelta
-from axio.messages import Message
-from axio.selector import (
-    EmbeddingToolSelector,
-    ToolFilteringTransport,
-    ToolSelector,
-    _cosine_similarity,
-    _extract_last_user_text,
-)
-from axio.testing import StubTransport, make_text_response
-from axio.tool import Tool, ToolHandler
-from axio.transport import CompletionTransport
-
-
-class DummyHandler(ToolHandler):
-    async def __call__(self) -> str:
-        return "ok"
-
-
-def _make_tool(name: str, description: str = "") -> Tool:
-    return Tool(name=name, description=description or f"Tool {name}", handler=DummyHandler)
-
-
-class StubEmbeddingTransport:
-    """Returns deterministic vectors: hash-based unit vectors for reproducibility."""
-
-    def __init__(self) -> None:
-        self.call_count = 0
-
-    async def embed(self, texts: list[str]) -> list[list[float]]:
-        self.call_count += 1
-        return [self._vector_for(t) for t in texts]
-
-    @staticmethod
-    def _vector_for(text: str) -> list[float]:
-        """Generate a deterministic 8-dim vector from text hash."""
-        h = hash(text) & 0xFFFFFFFF
-        raw = [float((h >> (i * 4)) & 0xF) - 7.5 for i in range(8)]
-        norm = math.sqrt(sum(x * x for x in raw))
-        if norm == 0:
-            return [0.0] * 8
-        return [x / norm for x in raw]
-
-
-def test_protocol_conformance() -> None:
-    transport = StubEmbeddingTransport()
-    selector = EmbeddingToolSelector(transport=transport)
-    assert isinstance(selector, ToolSelector)
-
-
-async def test_returns_all_when_few_tools() -> None:
-    transport = StubEmbeddingTransport()
-    selector = EmbeddingToolSelector(transport=transport, top_k=5)
-    tools = [_make_tool(f"t{i}") for i in range(3)]
-    messages = [Message(role="user", content=[TextBlock(text="hello")])]
-
-    result = await selector.select(messages, tools)
-    assert result == tools
-
-
-async def test_selects_top_k() -> None:
-    transport = StubEmbeddingTransport()
-    selector = EmbeddingToolSelector(transport=transport, top_k=3)
-    tools = [_make_tool(f"tool_{i}", f"description {i}") for i in range(6)]
-    messages = [Message(role="user", content=[TextBlock(text="description 0")])]
-
-    result = await selector.select(messages, tools)
-    assert len(result) == 3
-    # All returned tools should be from the original list
-    for t in result:
-        assert t in tools
-
-
-async def test_pinned_always_included() -> None:
-    transport = StubEmbeddingTransport()
-    selector = EmbeddingToolSelector(
-        transport=transport,
-        top_k=2,
-        pinned=frozenset({"pinned_tool"}),
-    )
-    tools = [
-        _make_tool("pinned_tool", "totally unrelated description xyz"),
-        _make_tool("tool_a", "alpha"),
-        _make_tool("tool_b", "beta"),
-        _make_tool("tool_c", "gamma"),
-    ]
-    messages = [Message(role="user", content=[TextBlock(text="alpha beta")])]
-
-    result = await selector.select(messages, tools)
-    assert len(result) == 2
-    names = {t.name for t in result}
-    assert "pinned_tool" in names
-
-
-async def test_returns_all_on_tool_result_iteration() -> None:
-    transport = StubEmbeddingTransport()
-    selector = EmbeddingToolSelector(transport=transport, top_k=2)
-    tools = [_make_tool(f"t{i}") for i in range(6)]
-    messages = [
-        Message(role="user", content=[TextBlock(text="hello")]),
-        Message(role="assistant", content=[TextBlock(text="running tool")]),
-        Message(role="user", content=[ToolResultBlock(tool_use_id="call_1", content="result")]),
-    ]
-
-    result = await selector.select(messages, tools)
-    assert result == tools
-
-
-def test_cosine_similarity_identical() -> None:
-    a = [1.0, 0.0, 0.0]
-    assert _cosine_similarity(a, a) == pytest.approx(1.0)
-
-
-def test_cosine_similarity_orthogonal() -> None:
-    a = [1.0, 0.0]
-    b = [0.0, 1.0]
-    assert _cosine_similarity(a, b) == pytest.approx(0.0)
-
-
-def test_cosine_similarity_opposite() -> None:
-    a = [1.0, 0.0]
-    b = [-1.0, 0.0]
-    assert _cosine_similarity(a, b) == pytest.approx(-1.0)
-
-
-def test_cosine_similarity_zero_vector() -> None:
-    a = [0.0, 0.0]
-    b = [1.0, 0.0]
-    assert _cosine_similarity(a, b) == 0.0
-
-
-async def test_caches_embeddings() -> None:
-    transport = StubEmbeddingTransport()
-    selector = EmbeddingToolSelector(transport=transport, top_k=2)
-    tools = [_make_tool(f"t{i}", f"desc {i}") for i in range(4)]
-    messages = [Message(role="user", content=[TextBlock(text="query")])]
-
-    await selector.select(messages, tools)
-    first_count = transport.call_count
-
-    await selector.select(messages, tools)
-    # Only the query embedding should be called again, not tool descriptions
-    assert transport.call_count == first_count + 1
-
-
-def test_extract_last_user_text_with_text() -> None:
-    messages = [Message(role="user", content=[TextBlock(text="hello world")])]
-    assert _extract_last_user_text(messages) == "hello world"
-
-
-def test_extract_last_user_text_tool_result() -> None:
-    messages = [
-        Message(role="user", content=[TextBlock(text="initial")]),
-        Message(role="assistant", content=[TextBlock(text="response")]),
-        Message(role="user", content=[ToolResultBlock(tool_use_id="call_1", content="ok")]),
-    ]
-    assert _extract_last_user_text(messages) is None
-
-
-def test_extract_last_user_text_empty() -> None:
-    assert _extract_last_user_text([]) is None
-
-
-def test_extract_last_user_text_multiple_text_blocks() -> None:
-    messages = [Message(role="user", content=[TextBlock(text="hello"), TextBlock(text="world")])]
-    assert _extract_last_user_text(messages) == "hello world"
-
-
-# --- ToolFilteringTransport tests ---
-
-
-class StubSelector:
-    """ToolSelector that keeps only tools whose names are in the allowed set."""
-
-    def __init__(self, allowed: set[str]) -> None:
-        self._allowed = allowed
-
-    async def select(self, messages: list[Message], tools: list[Tool]) -> list[Tool]:
-        return [t for t in tools if t.name in self._allowed]
-
-
-def test_filtering_transport_protocol() -> None:
-    inner = StubTransport([make_text_response("ok")])
-    selector = StubSelector(allowed=set())
-    ft = ToolFilteringTransport(inner, selector)
-    assert isinstance(ft, CompletionTransport)
-
-
-async def test_filtering_transport_filters_tools() -> None:
-    captured_tools: list[list[Tool]] = []
-
-    class CapturingTransport:
-        async def _gen(self, events: list[StreamEvent]) -> AsyncGenerator[StreamEvent, None]:
-            for e in events:
-                yield e
-
-        def stream(self, messages: list[Message], tools: list[Tool], system: str) -> AsyncGenerator[StreamEvent, None]:
-            captured_tools.append(tools)
-            return self._gen(make_text_response("ok"))
-
-    tools = [_make_tool("a"), _make_tool("b"), _make_tool("c")]
-    selector = StubSelector(allowed={"a", "c"})
-    ft = ToolFilteringTransport(CapturingTransport(), selector)
-
-    messages = [Message(role="user", content=[TextBlock(text="hello")])]
-    events: list[StreamEvent] = []
-    async for e in ft.stream(messages, tools, "sys"):
-        events.append(e)
-
-    assert len(captured_tools) == 1
-    names = [t.name for t in captured_tools[0]]
-    assert names == ["a", "c"]
-
-
-async def test_filtering_transport_passes_events() -> None:
-    inner = StubTransport([make_text_response("hello")])
-    selector = StubSelector(allowed=set())  # allows nothing, but events still pass
-    ft = ToolFilteringTransport(inner, selector)
-
-    messages = [Message(role="user", content=[TextBlock(text="hi")])]
-    events: list[StreamEvent] = []
-    async for e in ft.stream(messages, [], "sys"):
-        events.append(e)
-
-    assert len(events) == 2
-    assert isinstance(events[0], TextDelta)
-    assert events[0].delta == "hello"