axio-context-sqlite 0.11.2__tar.gz → 0.12.0__tar.gz

{axio_context_sqlite-0.11.2 → axio_context_sqlite-0.12.0}/.gitignore

@@ -12,3 +12,4 @@ _build/
 .DS_Store
 *.sqlite
 *.db
+.axio/mcp.local.json

{axio_context_sqlite-0.11.2 → axio_context_sqlite-0.12.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.5
 Name: axio-context-sqlite
-Version: 0.11.2
+Version: 0.12.0
 Summary: SQLite-backed context store for Axio
 Project-URL: Documentation, https://docs.axio-agent.com
 Project-URL: Homepage, https://github.com/mosquito/axio-agent
@@ -70,6 +70,7 @@ from axio_context_sqlite import connect, SQLiteContextStore
 from axio.messages import Message
 from axio.blocks import TextBlock
 
+
 async def main() -> None:
     conn = await connect(pathlib.Path(tempfile.mkdtemp()) / "chat.db")
     try:
@@ -80,6 +81,7 @@ async def main() -> None:
     finally:
         await conn.close()
 
+
 asyncio.run(main())
 ```
 
@@ -109,6 +111,7 @@ Every connection opened by `connect()` is configured with:
 name: test_readme_agent
 ```python
 from axio.testing import StubTransport, make_text_response
+
 transport = StubTransport([make_text_response("Hi!")])
 ```
 -->
@@ -120,6 +123,7 @@ import pathlib
 from axio.agent import Agent
 from axio_context_sqlite import connect, SQLiteContextStore
 
+
 async def main() -> None:
     conn = await connect(pathlib.Path(tempfile.mkdtemp()) / "chat.db")
     try:
@@ -130,6 +134,7 @@ async def main() -> None:
     finally:
         await conn.close()
 
+
 asyncio.run(main())
 ```
 
@@ -144,6 +149,7 @@ from axio_context_sqlite import connect, SQLiteContextStore
 from axio.messages import Message
 from axio.blocks import TextBlock
 
+
 async def main() -> None:
     conn = await connect(pathlib.Path(tempfile.mkdtemp()) / "chat.db")
     try:
@@ -156,6 +162,7 @@ async def main() -> None:
     finally:
         await conn.close()
 
+
 asyncio.run(main())
 ```
 
@@ -183,6 +190,7 @@ from axio_context_sqlite import connect, SQLiteContextStore
 from axio.messages import Message
 from axio.blocks import TextBlock
 
+
 async def main() -> None:
     conn = await connect(pathlib.Path(tempfile.mkdtemp()) / "chat.db")
     try:
@@ -194,6 +202,7 @@ async def main() -> None:
     finally:
         await conn.close()
 
+
 asyncio.run(main())
 ```
 

{axio_context_sqlite-0.11.2 → axio_context_sqlite-0.12.0}/README.md

@@ -52,6 +52,7 @@ from axio_context_sqlite import connect, SQLiteContextStore
 from axio.messages import Message
 from axio.blocks import TextBlock
 
+
 async def main() -> None:
     conn = await connect(pathlib.Path(tempfile.mkdtemp()) / "chat.db")
     try:
@@ -62,6 +63,7 @@ async def main() -> None:
     finally:
         await conn.close()
 
+
 asyncio.run(main())
 ```
 
@@ -91,6 +93,7 @@ Every connection opened by `connect()` is configured with:
 name: test_readme_agent
 ```python
 from axio.testing import StubTransport, make_text_response
+
 transport = StubTransport([make_text_response("Hi!")])
 ```
 -->
@@ -102,6 +105,7 @@ import pathlib
 from axio.agent import Agent
 from axio_context_sqlite import connect, SQLiteContextStore
 
+
 async def main() -> None:
     conn = await connect(pathlib.Path(tempfile.mkdtemp()) / "chat.db")
     try:
@@ -112,6 +116,7 @@ async def main() -> None:
     finally:
         await conn.close()
 
+
 asyncio.run(main())
 ```
 
@@ -126,6 +131,7 @@ from axio_context_sqlite import connect, SQLiteContextStore
 from axio.messages import Message
 from axio.blocks import TextBlock
 
+
 async def main() -> None:
     conn = await connect(pathlib.Path(tempfile.mkdtemp()) / "chat.db")
     try:
@@ -138,6 +144,7 @@ async def main() -> None:
     finally:
         await conn.close()
 
+
 asyncio.run(main())
 ```
 
@@ -165,6 +172,7 @@ from axio_context_sqlite import connect, SQLiteContextStore
 from axio.messages import Message
 from axio.blocks import TextBlock
 
+
 async def main() -> None:
     conn = await connect(pathlib.Path(tempfile.mkdtemp()) / "chat.db")
     try:
@@ -176,6 +184,7 @@ async def main() -> None:
     finally:
         await conn.close()
 
+
 asyncio.run(main())
 ```
 

{axio_context_sqlite-0.11.2 → axio_context_sqlite-0.12.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "axio-context-sqlite"
-version = "0.11.2"
+version = "0.12.0"
 description = "SQLite-backed context store for Axio"
 readme = "README.md"
 requires-python = ">=3.12"
@@ -14,6 +14,9 @@ classifiers = [
 keywords = ["llm", "agent", "ai", "sqlite", "context", "storage"]
 dependencies = ["axio", "aiosqlite>=0.20"]
 
+[project.entry-points."axio.context"]
+sqlite = "axio_context_sqlite.plugin:SQLiteContextBackend"
+
 [project.urls]
 Documentation = "https://docs.axio-agent.com"
 Homepage = "https://github.com/mosquito/axio-agent"

axio_context_sqlite-0.12.0/src/axio_context_sqlite/plugin.py

@@ -0,0 +1,28 @@
+"""Application context backend with one shared SQLite connection."""
+
+from __future__ import annotations
+
+from pathlib import Path
+from uuid import uuid4
+
+import aiosqlite
+
+from .store import SQLiteContextStore, connect
+
+
+class SQLiteContextBackend:
+    def __init__(self) -> None:
+        self.connection: aiosqlite.Connection | None = None
+
+    async def init(self, path: Path) -> None:
+        self.connection = await connect(path)
+
+    def create(self, session_id: str | None = None) -> SQLiteContextStore:
+        if self.connection is None:
+            raise RuntimeError("SQLite context backend is not initialized")
+        return SQLiteContextStore(self.connection, session_id or uuid4().hex)
+
+    async def close(self) -> None:
+        if self.connection is not None:
+            await self.connection.close()
+            self.connection = None