axio-context-sqlite 0.11.3__tar.gz → 0.12.1__tar.gz

{axio_context_sqlite-0.11.3 → axio_context_sqlite-0.12.1}/.gitignore

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

{axio_context_sqlite-0.11.3 → axio_context_sqlite-0.12.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.5
 Name: axio-context-sqlite
-Version: 0.11.3
+Version: 0.12.1
 Summary: SQLite-backed context store for Axio
 Project-URL: Documentation, https://docs.axio-agent.com
 Project-URL: Homepage, https://github.com/mosquito/axio-agent

{axio_context_sqlite-0.11.3 → axio_context_sqlite-0.12.1}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "axio-context-sqlite"
-version = "0.11.3"
+version = "0.12.1"
 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.1/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