axio-tui 0.6.2__tar.gz → 0.7.0__tar.gz

{axio_tui-0.6.2 → axio_tui-0.7.0}/PKG-INFO

@@ -1,9 +1,9 @@
 Metadata-Version: 2.4
 Name: axio-tui
-Version: 0.6.2
+Version: 0.7.0
 Summary: Textual TUI application for Axio
-Project-URL: Homepage, https://github.com/axio-agent/axio-tui
-Project-URL: Repository, https://github.com/axio-agent/axio-tui
+Project-URL: Homepage, https://github.com/mosquito/axio-agent
+Project-URL: Repository, https://github.com/mosquito/axio-agent
 License: MIT
 License-File: LICENSE
 Keywords: agent,ai,chat,llm,terminal,textual,tui
@@ -40,7 +40,7 @@ Description-Content-Type: text/markdown
 [![Python](https://img.shields.io/pypi/pyversions/axio-tui)](https://pypi.org/project/axio-tui/)
 [![License: MIT](https://img.shields.io/badge/License-MIT-blue.svg)](LICENSE)
 
-Textual TUI application for [axio](https://github.com/axio-agent/axio).
+Textual TUI application for [axio](https://github.com/mosquito/axio-agent).
 
 A full-featured terminal chat interface with session management, a plugin system for transports and tools, and a built-in SQLite context store. Pick your LLM backend, load your tools, and start chatting — all from the terminal.
 
@@ -150,7 +150,7 @@ textual-serve axio_tui.__main__:app
 
 ## Part of the axio ecosystem
 
-[axio](https://github.com/axio-agent/axio) · [axio-transport-anthropic](https://github.com/axio-agent/axio-transport-anthropic) · [axio-tui-guards](https://github.com/axio-agent/axio-tui-guards) · [axio-tools-local](https://github.com/axio-agent/axio-tools-local) · [axio-tools-mcp](https://github.com/axio-agent/axio-tools-mcp) · [axio-tools-docker](https://github.com/axio-agent/axio-tools-docker)
+[axio](https://github.com/mosquito/axio-agent) · [axio-transport-anthropic](https://github.com/mosquito/axio-agent) · [axio-tui-guards](https://github.com/mosquito/axio-agent) · [axio-tools-local](https://github.com/mosquito/axio-agent) · [axio-tools-mcp](https://github.com/mosquito/axio-agent) · [axio-tools-docker](https://github.com/mosquito/axio-agent)
 
 ## License
 

{axio_tui-0.6.2 → axio_tui-0.7.0}/README.md

@@ -4,7 +4,7 @@
 [![Python](https://img.shields.io/pypi/pyversions/axio-tui)](https://pypi.org/project/axio-tui/)
 [![License: MIT](https://img.shields.io/badge/License-MIT-blue.svg)](LICENSE)
 
-Textual TUI application for [axio](https://github.com/axio-agent/axio).
+Textual TUI application for [axio](https://github.com/mosquito/axio-agent).
 
 A full-featured terminal chat interface with session management, a plugin system for transports and tools, and a built-in SQLite context store. Pick your LLM backend, load your tools, and start chatting — all from the terminal.
 
@@ -114,7 +114,7 @@ textual-serve axio_tui.__main__:app
 
 ## Part of the axio ecosystem
 
-[axio](https://github.com/axio-agent/axio) · [axio-transport-anthropic](https://github.com/axio-agent/axio-transport-anthropic) · [axio-tui-guards](https://github.com/axio-agent/axio-tui-guards) · [axio-tools-local](https://github.com/axio-agent/axio-tools-local) · [axio-tools-mcp](https://github.com/axio-agent/axio-tools-mcp) · [axio-tools-docker](https://github.com/axio-agent/axio-tools-docker)
+[axio](https://github.com/mosquito/axio-agent) · [axio-transport-anthropic](https://github.com/mosquito/axio-agent) · [axio-tui-guards](https://github.com/mosquito/axio-agent) · [axio-tools-local](https://github.com/mosquito/axio-agent) · [axio-tools-mcp](https://github.com/mosquito/axio-agent) · [axio-tools-docker](https://github.com/mosquito/axio-agent)
 
 ## License
 

{axio_tui-0.6.2 → axio_tui-0.7.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "axio-tui"
-version = "0.6.2"
+version = "0.7.0"
 description = "Textual TUI application for Axio"
 readme = "README.md"
 requires-python = ">=3.12"
@@ -36,8 +36,8 @@ axio = "axio_tui.__main__:main"
 
 
 [project.urls]
-Homepage = "https://github.com/axio-agent/axio-tui"
-Repository = "https://github.com/axio-agent/axio-tui"
+Homepage = "https://github.com/mosquito/axio-agent"
+Repository = "https://github.com/mosquito/axio-agent"
 [build-system]
 requires = ["hatchling"]
 build-backend = "hatchling.build"

{axio_tui-0.6.2 → axio_tui-0.7.0}/src/axio_tui/app.py

@@ -378,14 +378,14 @@ class AgentApp(App[None]):
         self._last_input_tokens = 0
         self._chat_model: ModelSpec | None = None
         self._model_name = ""
-        self._all_tools: list[Tool] = []
+        self._all_tools: list[Tool[Any]] = []
         self._disabled_plugins: set[str] = set()
         self._disabled_transports: set[str] = set()
         self._disabled_guards: set[str] = set()
         self._guard_tool_map: dict[str, set[str]] = {}
         self._guard_instances: dict[str, PermissionGuard] = {}
         self._guard_classes: dict[str, type[PermissionGuard]] = {}
-        self._active_tools: list[Tool] = []
+        self._active_tools: list[Tool[Any]] = []
         self._chat_transport: Any = None
         self._selector_classes: dict[str, type] = {}
         self._selector_instances: dict[str, Any] = {}
@@ -535,7 +535,7 @@ class AgentApp(App[None]):
             self._chat_transport = chat_transport
 
             # Discover tools and guards via entry points
-            tools: list[Tool] = discover_tools()
+            tools: list[Tool[Any]] = discover_tools()
             self._guard_classes = discover_guards()
 
             # Load disabled plugins from config
@@ -842,7 +842,7 @@ class AgentApp(App[None]):
                     return None
 
     def _rebuild_agent_tools(self) -> None:
-        tools: list[Tool] = []
+        tools: list[Tool[Any]] = []
         for t in self._all_tools:
             if t.name in self._disabled_plugins:
                 continue
@@ -1005,7 +1005,7 @@ class AgentApp(App[None]):
             return False
         return self._last_input_tokens / self._chat_model.context_window > 0.7
 
-    async def _compact_with_progress(self, max_messages: int = 20) -> list[Message] | None:
+    async def _compact_with_progress(self) -> list[Message] | None:
         """Run compaction while showing a live elapsed-time counter in the chat."""
         if self._agent is None:
             return None
@@ -1027,7 +1027,7 @@ class AgentApp(App[None]):
                 transport = self._transports.make_transport(cb.transport, cb.model)
             else:
                 transport = self._agent.transport
-            return await compact_context(self._chat_context, transport, max_messages=max_messages)
+            return await compact_context(self._chat_context, transport)
         finally:
             timer.stop()
             await widget.remove()
@@ -1209,7 +1209,7 @@ class AgentApp(App[None]):
 
     def _on_tools_plugin_dismissed(self) -> None:
         # Rebuild tools list with all plugin tools
-        tools: list[Tool] = discover_tools()
+        tools: list[Tool[Any]] = discover_tools()
         for plugin in self._tools_plugins.values():
             tools.extend(plugin.all_tools)
         self._all_tools = tools
@@ -1297,7 +1297,7 @@ class AgentApp(App[None]):
 
         self._is_running = True
         if self._should_compact():
-            messages = await self._compact_with_progress(max_messages=0)
+            messages = await self._compact_with_progress()
             if messages is not None:
                 await self._replace_context(messages)
                 self._last_input_tokens = 0

{axio_tui-0.6.2 → axio_tui-0.7.0}/src/axio_tui/plugin.py

@@ -27,7 +27,7 @@ class ToolsPlugin(Protocol):
     async def init(self, config: Any = None, global_config: Any = None) -> None: ...
 
     @property
-    def all_tools(self) -> list[Tool]: ...
+    def all_tools(self) -> list[Tool[Any]]: ...
 
     def settings_screen(self) -> Any: ...
 
@@ -42,7 +42,7 @@ TOOLS_SETTINGS_GROUP = "axio.tools.settings"
 SELECTOR_GROUP = "axio.selector"
 
 
-def _handler_description(handler: type[ToolHandler]) -> str:
+def _handler_description(handler: type[ToolHandler[Any]]) -> str:
     """Extract handler docstring as tool description."""
     doc = handler.__doc__
     if doc:
@@ -50,9 +50,9 @@ def _handler_description(handler: type[ToolHandler]) -> str:
     return handler.__name__
 
 
-def discover_tools() -> list[Tool]:
+def discover_tools() -> list[Tool[Any]]:
     """Load ToolHandler classes from 'axio.tools' entry points, build Tool objects."""
-    tools: list[Tool] = []
+    tools: list[Tool[Any]] = []
     for ep in entry_points(group=TOOLS_GROUP):
         try:
             handler = ep.load()
@@ -74,9 +74,9 @@ def discover_tools() -> list[Tool]:
     return tools
 
 
-def discover_tools_by_package() -> dict[str, list[Tool]]:
+def discover_tools_by_package() -> dict[str, list[Tool[Any]]]:
     """Return tools from 'axio.tools' entry points grouped by distribution package name."""
-    groups: dict[str, list[Tool]] = {}
+    groups: dict[str, list[Tool[Any]]] = {}
     for ep in entry_points(group=TOOLS_GROUP):
         try:
             handler = ep.load()

{axio_tui-0.6.2 → axio_tui-0.7.0}/src/axio_tui/screens.py

@@ -247,13 +247,13 @@ class ToolSelectScreen(ModalScreen[set[str] | None]):
     #tool-list { height: 1fr; }
     """
 
-    def __init__(self, groups: dict[str, list[Tool]], disabled: set[str]) -> None:
+    def __init__(self, groups: dict[str, list[Tool[Any]]], disabled: set[str]) -> None:
         super().__init__()
         self._groups = groups
-        self._all_tools: list[Tool] = [t for ts in groups.values() for t in ts]
+        self._all_tools: list[Tool[Any]] = [t for ts in groups.values() for t in ts]
         self._disabled = set(disabled)
         self._filter_query = ""
-        self._items: list[str | Tool] = []
+        self._items: list[str | Tool[Any]] = []
         self._rebuild_items()
 
     def _rebuild_items(self) -> None:
@@ -261,7 +261,7 @@ class ToolSelectScreen(ModalScreen[set[str] | None]):
         if q:
             self._items = [t for t in self._all_tools if q in t.name.lower() or q in t.description.lower()]
             return
-        items: list[str | Tool] = []
+        items: list[str | Tool[Any]] = []
         for pkg, tools in self._groups.items():
             items.append(pkg)
             items.extend(tools)
@@ -274,12 +274,12 @@ class ToolSelectScreen(ModalScreen[set[str] | None]):
         mark = "[*]" if n_on == total else ("[ ]" if n_on == 0 else "[-]")
         return f"{mark} ── {pkg} ({n_on}/{total}) ──"
 
-    def _tool_line(self, tool: Tool, *, indent: bool = False) -> str:
+    def _tool_line(self, tool: Tool[Any], *, indent: bool = False) -> str:
         mark = "[ ]" if tool.name in self._disabled else "[*]"
         prefix = "  " if indent else ""
         return f"{prefix}{mark} {tool.name:<22} {tool.description}"
 
-    def _format(self, item: str | Tool) -> str:
+    def _format(self, item: str | Tool[Any]) -> str:
         if isinstance(item, str):
             return self._group_header(item)
         return self._tool_line(item, indent=not self._filter_query)
@@ -323,7 +323,7 @@ class ToolSelectScreen(ModalScreen[set[str] | None]):
             self._rebuild_items()
             self._refresh_list()
 
-    def _select(self, item: str | Tool) -> None:
+    def _select(self, item: str | Tool[Any]) -> None:
         if isinstance(item, str):
             tools = self._groups[item]
             all_on = all(t.name not in self._disabled for t in tools)
@@ -400,7 +400,7 @@ class PluginHubScreen(ModalScreen[None]):
 
     def __init__(
         self,
-        tool_groups: dict[str, list[Tool]],
+        tool_groups: dict[str, list[Tool[Any]]],
         transport_available: list[str],
         transport_discovered: list[str],
         transport_model_counts: dict[str, int],
@@ -419,7 +419,7 @@ class PluginHubScreen(ModalScreen[None]):
     ) -> None:
         super().__init__()
         self._tool_groups = tool_groups
-        self._all_tools: list[Tool] = [t for ts in tool_groups.values() for t in ts]
+        self._all_tools: list[Tool[Any]] = [t for ts in tool_groups.values() for t in ts]
         self._transport_available = transport_available
         self._transport_discovered = transport_discovered
         self._transport_model_counts = transport_model_counts

{axio_tui-0.6.2 → axio_tui-0.7.0}/src/axio_tui/tools.py

@@ -19,7 +19,7 @@ def _short(value: Any, limit: int = 60) -> str:
     return s if len(s) <= limit else s[:limit] + "..."
 
 
-class StatusLine(ToolHandler):
+class StatusLine(ToolHandler[Any]):
     """Set a short status message shown to the user in the TUI status bar.
     Call this at the start of every turn before other tools to indicate
     what you are about to do (e.g. "Reading project files",
@@ -31,13 +31,13 @@ class StatusLine(ToolHandler):
     def __repr__(self) -> str:
         return f"StatusLine(message={self.message!r})"
 
-    async def __call__(self) -> str:
+    async def __call__(self, context: Any) -> str:
         if StatusLine._callback:
             StatusLine._callback(self.message)
         return "ok"
 
 
-class Confirm(ToolHandler):
+class Confirm(ToolHandler[Any]):
     """Submit a safety verdict for a pending tool call. Used by the
     LLM-based permission guard. Verdict must be SAFE, RISKY, or DENY.
     Provide a clear reason and a category for the action."""
@@ -49,7 +49,7 @@ class Confirm(ToolHandler):
     def __repr__(self) -> str:
         return f"Confirm(verdict={self.verdict!r}, category={self.category!r})"
 
-    async def __call__(self) -> str:
+    async def __call__(self, context: Any) -> str:
         return self.verdict
 
 
@@ -142,7 +142,7 @@ SUBAGENT_SYSTEM_PROMPT = """\
 </subagent_behavior>"""
 
 
-class SubAgent(ToolHandler):
+class SubAgent(ToolHandler[Any]):
     """Delegate a task to an independent sub-agent. The sub-agent receives
     conversation context and has access to the same tools (except subagent).
     Up to 3 sub-agents run concurrently. Use for parallelizing independent
@@ -157,11 +157,11 @@ class SubAgent(ToolHandler):
     def __repr__(self) -> str:
         return f"SubAgent(task={_short(self.task)!r})"
 
-    async def __call__(self) -> str:
+    async def __call__(self, context: Any) -> str:
         if SubAgent._factory is None:
             return "SubAgent is not configured"
-        agent, context = await SubAgent._factory()
-        return await agent.run(self.task, context)
+        agent, store = await SubAgent._factory()
+        return await agent.run(self.task, store)
 
 
 _IMAGE_MEDIA_TYPES: dict[str, str] = {
@@ -173,7 +173,7 @@ _IMAGE_MEDIA_TYPES: dict[str, str] = {
 }
 
 
-class VisionAnalyze(ToolHandler):
+class VisionAnalyze(ToolHandler[Any]):
     """Analyze an image file using a vision-capable model. Supports
     JPEG, PNG, GIF, and WebP formats. Provide a path to the image and
     an optional prompt describing what to look for. Returns the model's
@@ -186,7 +186,7 @@ class VisionAnalyze(ToolHandler):
     def __repr__(self) -> str:
         return f"VisionAnalyze(path={_short(self.path)!r})"
 
-    async def __call__(self) -> str:
+    async def __call__(self, context: Any) -> str:
         import os
 
         if VisionAnalyze._transport is None:

{axio_tui-0.6.2 → axio_tui-0.7.0}/tests/test_guard_screens.py

@@ -2,19 +2,21 @@
 
 from __future__ import annotations
 
+from typing import Any
+
 from axio.tool import Tool, ToolHandler
 
 from axio_tui.screens import GuardSelectScreen, GuardToolsScreen, PluginHubScreen, ToolDetailScreen
 
 
-class _StubHandler(ToolHandler):
+class _StubHandler(ToolHandler[Any]):
     """Stub tool handler."""
 
-    async def __call__(self) -> str:
+    async def __call__(self, context: Any) -> str:
         return ""
 
 
-def _make_tools(*names: str) -> list[Tool]:
+def _make_tools(*names: str) -> list[Tool[Any]]:
     return [Tool(name=n, description=f"Description for {n}", handler=_StubHandler) for n in names]
 
 
@@ -109,7 +111,7 @@ class TestGuardSelectScreen:
 
 
 def _make_hub(
-    tools: list[Tool],
+    tools: list[Tool[Any]],
     disabled_plugins: set[str] = frozenset(),  # type: ignore[assignment]
     disabled_guards: set[str] = frozenset(),  # type: ignore[assignment]
     guard_tool_map: dict[str, set[str]] | None = None,

{axio_tui-0.6.2 → axio_tui-0.7.0}/tests/test_plugin.py

@@ -2,7 +2,7 @@
 
 from __future__ import annotations
 
-from typing import ClassVar
+from typing import Any, ClassVar
 from unittest.mock import MagicMock, patch
 
 from axio.permission import PermissionGuard
@@ -11,29 +11,29 @@ from axio.tool import ToolHandler
 from axio_tui.plugin import discover_guards, discover_tools, discover_transports
 
 
-class _EchoHandler(ToolHandler):
+class _EchoHandler(ToolHandler[Any]):
     """Echo back input text."""
 
     text: str = ""
 
-    async def __call__(self) -> str:
+    async def __call__(self, context: Any) -> str:
         return self.text
 
 
-class _NoDocHandler(ToolHandler):
+class _NoDocHandler(ToolHandler[Any]):
     text: str = ""
 
-    async def __call__(self) -> str:
+    async def __call__(self, context: Any) -> str:
         return self.text
 
 
-class _ConcurrentHandler(ToolHandler):
+class _ConcurrentHandler(ToolHandler[Any]):
     """Handler with concurrency limit."""
 
     _tool_concurrency: ClassVar[int | None] = 2
     text: str = ""
 
-    async def __call__(self) -> str:
+    async def __call__(self, context: Any) -> str:
         return self.text
 
 

{axio_tui-0.6.2 → axio_tui-0.7.0}/tests/test_plugin_toggle.py

@@ -2,24 +2,26 @@
 
 from __future__ import annotations
 
+from typing import Any
+
 import pytest
 from axio.tool import Tool, ToolHandler
 
 from axio_tui.screens import ToolSelectScreen
 
 
-class _StubHandler(ToolHandler):
+class _StubHandler(ToolHandler[Any]):
     """Stub tool handler."""
 
-    async def __call__(self) -> str:
+    async def __call__(self, context: Any) -> str:
         return ""
 
 
-def _make_tools(*names: str) -> list[Tool]:
+def _make_tools(*names: str) -> list[Tool[Any]]:
     return [Tool(name=n, description=f"Description for {n}", handler=_StubHandler) for n in names]
 
 
-def _screen(tools: list[Tool], disabled: set[str]) -> ToolSelectScreen:
+def _screen(tools: list[Tool[Any]], disabled: set[str]) -> ToolSelectScreen:
     return ToolSelectScreen(groups={"pkg": tools}, disabled=disabled)
 
 

{axio_tui-0.6.2 → axio_tui-0.7.0}/tests/test_tools.py

@@ -5,6 +5,7 @@ from __future__ import annotations
 import os
 from collections.abc import AsyncIterator
 from pathlib import Path
+from typing import Any
 
 import pytest
 from axio.agent import Agent
@@ -20,13 +21,13 @@ from axio_tui.tools import Confirm, StatusLine, SubAgent, VisionAnalyze
 class TestStatusLine:
     async def test_returns_ok(self) -> None:
         handler = StatusLine(message="working")
-        assert await handler() == "ok"
+        assert await handler({}) == "ok"
 
 
 class TestConfirm:
     async def test_returns_verdict(self) -> None:
         handler = Confirm(verdict="SAFE", reason="harmless", category="read")
-        assert await handler() == "SAFE"
+        assert await handler({}) == "SAFE"
 
     async def test_repr(self) -> None:
         handler = Confirm(verdict="DENY", reason="bad", category="exec")
@@ -36,7 +37,7 @@ class TestConfirm:
 
 
 class TestSubAgent:
-    def _make_agent(self, transport: StubTransport, tools: list[Tool] | None = None) -> Agent:
+    def _make_agent(self, transport: StubTransport, tools: list[Tool[Any]] | None = None) -> Agent:
         return Agent(system="test", tools=tools or [], transport=transport, max_iterations=5)
 
     async def test_returns_subagent_result(self) -> None:
@@ -49,7 +50,7 @@ class TestSubAgent:
         SubAgent._factory = factory
         try:
             handler = SubAgent(task="do something")
-            result = await handler()
+            result = await handler({})
             assert result == "sub-result"
         finally:
             SubAgent._factory = None
@@ -57,7 +58,7 @@ class TestSubAgent:
     async def test_no_factory_returns_error(self) -> None:
         SubAgent._factory = None
         handler = SubAgent(task="do something")
-        result = await handler()
+        result = await handler({})
         assert result == "SubAgent is not configured"
 
     async def test_context_forking(self) -> None:
@@ -80,7 +81,7 @@ class TestSubAgent:
         SubAgent._factory = factory
         try:
             handler = SubAgent(task="check context")
-            await handler()
+            await handler({})
             assert len(received_context) == 1
             history = await received_context[0].get_history()
             assert history[0].content[0].text == "hello"  # type: ignore[attr-defined]
@@ -97,7 +98,7 @@ class TestSubAgent:
         try:
             handler = SubAgent(task="boom")
             with pytest.raises(RuntimeError, match="factory failed"):
-                await handler()
+                await handler({})
         finally:
             SubAgent._factory = None
 
@@ -152,7 +153,7 @@ class TestVisionAnalyze:
     async def test_no_transport_returns_error(self) -> None:
         VisionAnalyze._transport = None
         handler = VisionAnalyze(path="img.png")
-        result = await handler()
+        result = await handler({})
         assert "not configured" in result
 
     async def test_file_not_found(self, tmp_path: Path) -> None:
@@ -161,7 +162,7 @@ class TestVisionAnalyze:
         try:
             os.chdir(tmp_path)
             handler = VisionAnalyze(path="missing.png")
-            result = await handler()
+            result = await handler({})
             assert "File not found" in result
         finally:
             os.chdir(old_cwd)
@@ -174,7 +175,7 @@ class TestVisionAnalyze:
         try:
             os.chdir(tmp_path)
             handler = VisionAnalyze(path="file.bmp")
-            result = await handler()
+            result = await handler({})
             assert "Unsupported image format" in result
         finally:
             os.chdir(old_cwd)
@@ -188,7 +189,7 @@ class TestVisionAnalyze:
         try:
             os.chdir(tmp_path)
             handler = VisionAnalyze(path="photo.png", prompt="What is this?")
-            result = await handler()
+            result = await handler({})
             assert result == "A red pixel"
         finally:
             os.chdir(old_cwd)
@@ -202,7 +203,9 @@ class TestVisionAnalyze:
         captured: list[list[Message]] = []
 
         class CapturingTransport(StubTransport):
-            def stream(self, messages: list[Message], tools: list[Tool], system: str) -> AsyncIterator[StreamEvent]:
+            def stream(
+                self, messages: list[Message], tools: list[Tool[Any]], system: str
+            ) -> AsyncIterator[StreamEvent]:
                 captured.append(messages)
                 return super().stream(messages, tools, system)
 
@@ -212,7 +215,7 @@ class TestVisionAnalyze:
         try:
             os.chdir(tmp_path)
             handler = VisionAnalyze(path="test.jpg", prompt="Describe it")
-            await handler()
+            await handler({})
             assert len(captured) == 1
             msg = captured[0][0]
             assert len(msg.content) == 2