sqlspec 0.43.0__tar.gz → 0.45.0__tar.gz

{sqlspec-0.43.0 → sqlspec-0.45.0}/.gitignore

@@ -69,3 +69,8 @@ uv.toml
 .beads/
 tools/scripts/profiles/*.prof
 .agents/
+
+# Beads / Dolt files (added by bd init)
+.dolt/
+.beads-credential-key
+.codex

{sqlspec-0.43.0 → sqlspec-0.45.0}/.pre-commit-config.yaml

@@ -17,7 +17,7 @@ repos:
       - id: mixed-line-ending
       - id: trailing-whitespace
   - repo: https://github.com/charliermarsh/ruff-pre-commit
-    rev: "v0.15.9"
+    rev: "v0.15.12"
     hooks:
       - id: ruff
         args: ["--fix"]

{sqlspec-0.43.0 → sqlspec-0.45.0}/Makefile

@@ -221,7 +221,7 @@ pre-commit:                                        ## Run pre-commit hooks
 .PHONY: slotscheck
 slotscheck:                                        ## Run slotscheck
 	@echo "${INFO} Running slotscheck... 🔍"
-	@uv run slotscheck sqlspec/
+	@PYTHONWARNINGS="ignore:::google.adk.features._feature_decorator" uv run slotscheck sqlspec/
 	@echo "${OK} Slotscheck complete ✨"
 
 .PHONY: fix

{sqlspec-0.43.0 → sqlspec-0.45.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: sqlspec
-Version: 0.43.0
+Version: 0.45.0
 Summary: SQL Experiments in Python
 Project-URL: Discord, https://discord.gg/litestar
 Project-URL: Issue, https://github.com/litestar-org/sqlspec/issues/
@@ -29,6 +29,8 @@ Requires-Dist: adbc-driver-manager; extra == 'adbc'
 Requires-Dist: pyarrow; extra == 'adbc'
 Provides-Extra: adk
 Requires-Dist: google-adk; extra == 'adk'
+Provides-Extra: aiomysql
+Requires-Dist: aiomysql; extra == 'aiomysql'
 Provides-Extra: aioodbc
 Requires-Dist: aioodbc; extra == 'aioodbc'
 Provides-Extra: aiosqlite
@@ -66,7 +68,8 @@ Requires-Dist: msgspec; extra == 'msgspec'
 Provides-Extra: mypyc
 Requires-Dist: sqlglot[c]>=30.0.0; extra == 'mypyc'
 Provides-Extra: mysql-connector
-Requires-Dist: mysql-connector-python; extra == 'mysql-connector'
+Requires-Dist: mysql-connector-python; (python_version < '3.12') and extra == 'mysql-connector'
+Requires-Dist: mysql-connector-python<9.7.0; (python_version >= '3.12') and extra == 'mysql-connector'
 Provides-Extra: nanoid
 Requires-Dist: fastnanoid>=0.4.1; extra == 'nanoid'
 Provides-Extra: obstore
@@ -98,8 +101,12 @@ Provides-Extra: pymssql
 Requires-Dist: pymssql; extra == 'pymssql'
 Provides-Extra: pymysql
 Requires-Dist: pymysql; extra == 'pymysql'
+Provides-Extra: sanic
+Requires-Dist: sanic; extra == 'sanic'
 Provides-Extra: spanner
 Requires-Dist: google-cloud-spanner; extra == 'spanner'
+Provides-Extra: starlette
+Requires-Dist: starlette; extra == 'starlette'
 Provides-Extra: uuid
 Requires-Dist: uuid-utils; extra == 'uuid'
 Description-Content-Type: text/markdown
@@ -113,7 +120,7 @@ Description-Content-Type: text/markdown
 
 SQLSpec is a SQL execution layer for Python. You write the SQL -- as strings, through a builder API, or loaded from files -- and SQLSpec handles connections, parameter binding, SQL injection prevention, dialect translation, and mapping results back to typed Python objects. It uses [sqlglot](https://github.com/tobymao/sqlglot) under the hood to parse, validate, and optimize your queries before they hit the database.
 
-It works with PostgreSQL (asyncpg, psycopg, psqlpy), SQLite (sqlite3, aiosqlite), DuckDB, MySQL (asyncmy, mysql-connector, pymysql), Oracle (oracledb), CockroachDB, BigQuery, Spanner, and anything ADBC-compatible. Sync or async, same API. It also includes a built-in storage layer, native and bridged Arrow support for all drivers, and integrations for Litestar, FastAPI, Flask, and Starlette.
+It works with PostgreSQL (asyncpg, psycopg, psqlpy), SQLite (sqlite3, aiosqlite), DuckDB, MySQL (asyncmy, mysql-connector, pymysql), Oracle (oracledb), CockroachDB, BigQuery, Spanner, and anything ADBC-compatible. Sync or async, same API. It also includes a built-in storage layer, native and bridged Arrow support for all drivers, and integrations for Litestar, FastAPI, Flask, Sanic, and Starlette.
 
 ## Quick Start
 
@@ -163,7 +170,7 @@ users = session.select(
 - **Parameter binding and dialect translation** -- powered by sqlglot, with a fluent query builder and `.sql` file loader
 - **Result mapping** -- map rows to Pydantic, msgspec, attrs, or dataclass models, or export to Arrow tables for pandas and Polars
 - **Storage layer** -- read and write Arrow tables to local files, fsspec, or object stores
-- **Framework integrations** -- Litestar plugin with DI, Starlette/FastAPI middleware, Flask extension
+- **Framework integrations** -- Litestar plugin with DI, Starlette/FastAPI/Sanic middleware, Flask extension
 - **Observability** -- OpenTelemetry and Prometheus instrumentation, structured logging with correlation IDs
 - **Event channels** -- LISTEN/NOTIFY, Oracle AQ, and a portable polling fallback
 - **Migrations** -- schema versioning CLI built on Alembic

{sqlspec-0.43.0 → sqlspec-0.45.0}/README.md

@@ -7,7 +7,7 @@
 
 SQLSpec is a SQL execution layer for Python. You write the SQL -- as strings, through a builder API, or loaded from files -- and SQLSpec handles connections, parameter binding, SQL injection prevention, dialect translation, and mapping results back to typed Python objects. It uses [sqlglot](https://github.com/tobymao/sqlglot) under the hood to parse, validate, and optimize your queries before they hit the database.
 
-It works with PostgreSQL (asyncpg, psycopg, psqlpy), SQLite (sqlite3, aiosqlite), DuckDB, MySQL (asyncmy, mysql-connector, pymysql), Oracle (oracledb), CockroachDB, BigQuery, Spanner, and anything ADBC-compatible. Sync or async, same API. It also includes a built-in storage layer, native and bridged Arrow support for all drivers, and integrations for Litestar, FastAPI, Flask, and Starlette.
+It works with PostgreSQL (asyncpg, psycopg, psqlpy), SQLite (sqlite3, aiosqlite), DuckDB, MySQL (asyncmy, mysql-connector, pymysql), Oracle (oracledb), CockroachDB, BigQuery, Spanner, and anything ADBC-compatible. Sync or async, same API. It also includes a built-in storage layer, native and bridged Arrow support for all drivers, and integrations for Litestar, FastAPI, Flask, Sanic, and Starlette.
 
 ## Quick Start
 
@@ -57,7 +57,7 @@ users = session.select(
 - **Parameter binding and dialect translation** -- powered by sqlglot, with a fluent query builder and `.sql` file loader
 - **Result mapping** -- map rows to Pydantic, msgspec, attrs, or dataclass models, or export to Arrow tables for pandas and Polars
 - **Storage layer** -- read and write Arrow tables to local files, fsspec, or object stores
-- **Framework integrations** -- Litestar plugin with DI, Starlette/FastAPI middleware, Flask extension
+- **Framework integrations** -- Litestar plugin with DI, Starlette/FastAPI/Sanic middleware, Flask extension
 - **Observability** -- OpenTelemetry and Prometheus instrumentation, structured logging with correlation IDs
 - **Event channels** -- LISTEN/NOTIFY, Oracle AQ, and a portable polling fallback
 - **Migrations** -- schema versioning CLI built on Alembic

{sqlspec-0.43.0 → sqlspec-0.45.0}/pyproject.toml

@@ -24,7 +24,7 @@ maintainers = [{ name = "Litestar Developers", email = "hello@litestar.dev" }]
 name = "sqlspec"
 readme = "README.md"
 requires-python = ">=3.10, <4.0"
-version = "0.43.0"
+version = "0.45.0"
 
 [project.urls]
 Discord = "https://discord.gg/litestar"
@@ -34,6 +34,7 @@ Source = "https://github.com/litestar-org/sqlspec"
 [project.optional-dependencies]
 adbc = ["adbc_driver_manager", "pyarrow"]
 adk = ["google-adk"]
+aiomysql = ["aiomysql"]
 aioodbc = ["aioodbc"]
 aiosqlite = ["aiosqlite"]
 alloydb = ["google-cloud-alloydb-connector"]
@@ -50,7 +51,10 @@ fsspec = ["fsspec"]
 litestar = ["litestar"]
 msgspec = ["msgspec"]
 mypyc = ["sqlglot[c]>=30.0.0"]
-mysql-connector = ["mysql-connector-python"]
+mysql-connector = [
+  "mysql-connector-python; python_version < '3.12'",
+  "mysql-connector-python<9.7.0; python_version >= '3.12'",
+]
 nanoid = ["fastnanoid>=0.4.1"]
 obstore = ["obstore"]
 opentelemetry = ["opentelemetry-instrumentation"]
@@ -65,7 +69,9 @@ psycopg = ["psycopg[binary,pool]"]
 pydantic = ["pydantic", "pydantic-extra-types"]
 pymssql = ["pymssql"]
 pymysql = ["pymysql"]
+sanic = ["sanic"]
 spanner = ["google-cloud-spanner"]
+starlette = ["starlette"]
 uuid = ["uuid-utils"]
 
 [dependency-groups]
@@ -144,6 +150,7 @@ test = [
   "pytest-sugar>=1.0.0",
   "pytest-xdist>=3.6.1",
   "pytest-timeout>=2.3.1",
+  "sanic-testing",
 ]
 
 [project.scripts]
@@ -153,6 +160,16 @@ sqlspec = "sqlspec.__main__:run_cli"
 sqlspec-dark = "tools.sphinx_ext.pygments_styles:SQLSpecDarkStyle"
 sqlspec-light = "tools.sphinx_ext.pygments_styles:SQLSpecLightStyle"
 
+[tool.uv]
+# mysql-connector-python 9.7.0 dropped cp312/cp313/cp314 wheels (regression vs 9.6.0).
+# Override dependency metadata for every source that requests mysql-connector-python.
+# Keep the uncapped dependency on Python <3.12, and force the cap on Python >=3.12
+# so transitive pulls (e.g., pytest-databases[mysql]) resolve to an existing wheel.
+override-dependencies = [
+  "mysql-connector-python; python_version < '3.12'",
+  "mysql-connector-python<9.7.0; python_version >= '3.12'",
+]
+
 [build-system]
 build-backend = "hatchling.build"
 requires = ["hatchling", "hatch-mypyc"]
@@ -183,7 +200,6 @@ exclude = [
   "sqlspec/dialects/**/*.py",                 # Keep SQLGlot dialect subclasses interpreted
   "sqlspec/**/__init__.py",                   # Init files (usually just imports)
   "sqlspec/protocols.py",                     # Protocol definitions
-  "sqlspec/adapters/mock/**",                 # Mock adapter (testing only)
   "sqlspec/migrations/commands.py",           # Migration command CLI (dynamic imports)
   "sqlspec/data_dictionary/_loader.py",       # Loader relies on __file__ which fails in compiled modules
   "sqlspec/extensions/fastapi/providers.py",  # Uses SingletonMeta metaclass
@@ -191,6 +207,7 @@ exclude = [
   "sqlspec/adapters/**/data_dictionary.py",   # Cross-module inheritance causes mypyc segfaults
   "sqlspec/observability/_formatting.py",     # Inherits from non-compiled logging.Formatter
   "sqlspec/utils/arrow_helpers.py",           # Arrow operations cause segfaults when compiled
+  "sqlspec/core/_pagination.py",              # @dataclass mutates class at def time; annotations must survive mypyc for Litestar OpenAPI (#419)
 ]
 include = [
   "sqlspec/core/**/*.py",                  # Core module
@@ -205,6 +222,10 @@ include = [
   "sqlspec/data_dictionary/**/*.py",       # Data dictionary mixin (required for adapter inheritance)
   "sqlspec/adapters/**/core.py",           # Adapter compiled helpers
   "sqlspec/adapters/**/type_converter.py", # All adapters type converters
+  "sqlspec/adapters/oracledb/_param_types.py",     # Slot-based LOB/JSON parameter wrappers
+  "sqlspec/adapters/oracledb/_json_handlers.py",   # Native JSON inputtypehandler / outputtypehandler chain
+  "sqlspec/adapters/oracledb/_uuid_handlers.py",   # UUID ↔ RAW(16) inputtypehandler / outputtypehandler chain
+  "sqlspec/adapters/oracledb/_vector_handlers.py", # DB_TYPE_VECTOR inputtypehandler / outputtypehandler dispatch
   "sqlspec/utils/text.py",                 # Text utilities
   "sqlspec/utils/sync_tools.py",           # Synchronous utility functions
   "sqlspec/utils/type_guards.py",          # Type guard utilities
@@ -243,7 +264,7 @@ opt_level = "3"   # Maximum optimization (0-3)
 allow_dirty = true
 commit = false
 commit_args = "--no-verify"
-current_version = "0.43.0"
+current_version = "0.45.0"
 ignore_missing_files = false
 ignore_missing_version = false
 message = "chore(release): bump to v{new_version}"
@@ -358,6 +379,7 @@ markers = [
   "mssql: marks tests specific to Microsoft SQL Server",
   # Driver markers
   "adbc: marks tests using ADBC drivers",
+  "aiomysql: marks tests using aiomysql",
   "aioodbc: marks tests using aioodbc",
   "aiosqlite: marks tests using aiosqlite",
   "asyncmy: marks tests using asyncmy",
@@ -400,6 +422,8 @@ module = [
   "orjson",
   "uvicorn.*",
   "uvloop.*",
+  "aiomysql",
+  "aiomysql.*",
   "asyncmy",
   "asyncmy.*",
   "pyarrow",

{sqlspec-0.43.0 → sqlspec-0.45.0}/sqlspec/adapters/adbc/_typing.py

@@ -45,7 +45,7 @@ class AdbcCursor:
     def __exit__(self, *_: Any) -> None:
         if self.cursor is not None:
             with contextlib.suppress(Exception):
-                self.cursor.close()  # type: ignore[no-untyped-call]
+                self.cursor.close()
 
 
 class AdbcSessionContext:

{sqlspec-0.43.0 → sqlspec-0.45.0}/sqlspec/adapters/adbc/adk/store.py

@@ -445,7 +445,7 @@ class AdbcADKStore(BaseAsyncADKStore["AdbcConfig"]):
                 cursor.execute(events_idx)
                 conn.commit()
             finally:
-                cursor.close()  # type: ignore[no-untyped-call]
+                cursor.close()
 
     async def create_tables(self) -> None:
         """Create tables if they don't exist."""
@@ -506,7 +506,7 @@ class AdbcADKStore(BaseAsyncADKStore["AdbcConfig"]):
                 cursor.execute(sql, params)
                 conn.commit()
             finally:
-                cursor.close()  # type: ignore[no-untyped-call]
+                cursor.close()
 
         result = self._get_session(session_id)
         if result is None:
@@ -557,7 +557,7 @@ class AdbcADKStore(BaseAsyncADKStore["AdbcConfig"]):
                         update_time=row[5],
                     )
                 finally:
-                    cursor.close()  # type: ignore[no-untyped-call]
+                    cursor.close()
         except Exception as e:
             error_msg = str(e).lower()
             if any(pattern in error_msg for pattern in ADBC_TABLE_NOT_FOUND_PATTERNS):
@@ -592,7 +592,7 @@ class AdbcADKStore(BaseAsyncADKStore["AdbcConfig"]):
                 cursor.execute(sql, (state_json, session_id))
                 conn.commit()
             finally:
-                cursor.close()  # type: ignore[no-untyped-call]
+                cursor.close()
 
     async def update_session_state(self, session_id: str, state: "dict[str, Any]") -> None:
         """Update session state."""
@@ -616,7 +616,7 @@ class AdbcADKStore(BaseAsyncADKStore["AdbcConfig"]):
                 cursor.execute(sql, (session_id,))
                 conn.commit()
             finally:
-                cursor.close()  # type: ignore[no-untyped-call]
+                cursor.close()
 
     async def delete_session(self, session_id: str) -> None:
         """Delete session and associated events."""
@@ -671,7 +671,7 @@ class AdbcADKStore(BaseAsyncADKStore["AdbcConfig"]):
                         for row in rows
                     ]
                 finally:
-                    cursor.close()  # type: ignore[no-untyped-call]
+                    cursor.close()
         except Exception as e:
             error_msg = str(e).lower()
             if any(pattern in error_msg for pattern in ADBC_TABLE_NOT_FOUND_PATTERNS):
@@ -710,7 +710,7 @@ class AdbcADKStore(BaseAsyncADKStore["AdbcConfig"]):
                 )
                 conn.commit()
             finally:
-                cursor.close()  # type: ignore[no-untyped-call]
+                cursor.close()
 
     def _append_event_and_update_state(
         self, event_record: "EventRecord", session_id: str, state: "dict[str, Any]"
@@ -760,7 +760,7 @@ class AdbcADKStore(BaseAsyncADKStore["AdbcConfig"]):
                     conn.rollback()
                 raise
             finally:
-                cursor.close()  # type: ignore[no-untyped-call]
+                cursor.close()
 
     async def append_event_and_update_state(
         self, event_record: EventRecord, session_id: str, state: "dict[str, Any]"
@@ -820,7 +820,7 @@ class AdbcADKStore(BaseAsyncADKStore["AdbcConfig"]):
                         for row in rows
                     ]
                 finally:
-                    cursor.close()  # type: ignore[no-untyped-call]
+                    cursor.close()
         except Exception as e:
             error_msg = str(e).lower()
             if any(pattern in error_msg for pattern in ADBC_TABLE_NOT_FOUND_PATTERNS):
@@ -1014,7 +1014,7 @@ class AdbcADKMemoryStore(BaseAsyncADKMemoryStore["AdbcConfig"]):
                 cursor.execute(idx_session)
                 conn.commit()
             finally:
-                cursor.close()  # type: ignore[no-untyped-call]
+                cursor.close()
 
     async def create_tables(self) -> None:
         """Create tables if they don't exist."""
@@ -1122,7 +1122,7 @@ class AdbcADKMemoryStore(BaseAsyncADKMemoryStore["AdbcConfig"]):
                             raise
                 conn.commit()
             finally:
-                cursor.close()  # type: ignore[no-untyped-call]
+                cursor.close()
 
         return inserted_count
 
@@ -1161,7 +1161,7 @@ class AdbcADKMemoryStore(BaseAsyncADKMemoryStore["AdbcConfig"]):
                     cursor.execute(sql, (app_name, user_id, pattern, effective_limit))
                     rows = cursor.fetchall()
                 finally:
-                    cursor.close()  # type: ignore[no-untyped-call]
+                    cursor.close()
         except Exception as exc:
             error_msg = str(exc).lower()
             if any(pattern in error_msg for pattern in ADBC_TABLE_NOT_FOUND_PATTERNS):
@@ -1193,7 +1193,7 @@ class AdbcADKMemoryStore(BaseAsyncADKMemoryStore["AdbcConfig"]):
                 conn.commit()
                 return cursor.rowcount if cursor.rowcount and cursor.rowcount > 0 else 0
             finally:
-                cursor.close()  # type: ignore[no-untyped-call]
+                cursor.close()
 
     async def delete_entries_by_session(self, session_id: str) -> int:
         """Delete all memory entries for a specific session."""
@@ -1217,7 +1217,7 @@ class AdbcADKMemoryStore(BaseAsyncADKMemoryStore["AdbcConfig"]):
                 conn.commit()
                 return cursor.rowcount if cursor.rowcount and cursor.rowcount > 0 else 0
             finally:
-                cursor.close()  # type: ignore[no-untyped-call]
+                cursor.close()
 
     async def delete_entries_older_than(self, days: int) -> int:
         """Delete memory entries older than specified days."""

sqlspec-0.45.0/sqlspec/adapters/aiomysql/__init__.py

@@ -0,0 +1,21 @@
+from sqlspec.adapters.aiomysql._typing import AiomysqlConnection, AiomysqlCursor
+from sqlspec.adapters.aiomysql.config import (
+    AiomysqlConfig,
+    AiomysqlConnectionParams,
+    AiomysqlDriverFeatures,
+    AiomysqlPoolParams,
+)
+from sqlspec.adapters.aiomysql.core import default_statement_config
+from sqlspec.adapters.aiomysql.driver import AiomysqlDriver, AiomysqlExceptionHandler
+
+__all__ = (
+    "AiomysqlConfig",
+    "AiomysqlConnection",
+    "AiomysqlConnectionParams",
+    "AiomysqlCursor",
+    "AiomysqlDriver",
+    "AiomysqlDriverFeatures",
+    "AiomysqlExceptionHandler",
+    "AiomysqlPoolParams",
+    "default_statement_config",
+)

sqlspec-0.45.0/sqlspec/adapters/aiomysql/_typing.py

@@ -0,0 +1,137 @@
+"""aiomysql adapter type definitions.
+
+This module contains type aliases and classes that are excluded from mypyc
+compilation to avoid ABI boundary issues.
+"""
+
+from typing import TYPE_CHECKING, Any
+
+from aiomysql import Connection  # pyright: ignore
+from aiomysql import Pool as _AiomysqlPool  # pyright: ignore
+from aiomysql.cursors import Cursor as _AiomysqlCursor  # pyright: ignore
+from aiomysql.cursors import DictCursor as _AiomysqlDictCursor  # pyright: ignore
+
+if TYPE_CHECKING:
+    from collections.abc import Callable
+    from types import TracebackType
+    from typing import Protocol, TypeAlias
+
+    from sqlspec.adapters.aiomysql.driver import AiomysqlDriver
+    from sqlspec.core import StatementConfig
+
+    class AiomysqlConnectionProtocol(Protocol):
+        async def cursor(self, cursor: "type[Any] | None" = None) -> "AiomysqlRawCursor": ...
+
+        async def commit(self) -> Any: ...
+
+        async def rollback(self) -> Any: ...
+
+        def close(self) -> Any: ...
+
+    AiomysqlConnection: TypeAlias = AiomysqlConnectionProtocol
+    AiomysqlRawCursor: TypeAlias = _AiomysqlCursor
+    AiomysqlDictCursor: TypeAlias = _AiomysqlDictCursor
+    AiomysqlPool: TypeAlias = _AiomysqlPool
+
+if not TYPE_CHECKING:
+    AiomysqlConnection = Connection
+    AiomysqlRawCursor = _AiomysqlCursor
+    AiomysqlDictCursor = _AiomysqlDictCursor
+    AiomysqlPool = _AiomysqlPool
+
+
+class AiomysqlCursor:
+    """Context manager for aiomysql cursor operations.
+
+    Provides automatic cursor acquisition and cleanup for database operations.
+
+    The optional ``cursor_class`` argument forces a specific cursor type (e.g.,
+    the tuple-returning ``AiomysqlRawCursor``) regardless of the user's
+    ``cursor_class`` setting in ``AiomysqlConnectionParams``. This is used by
+    first-party store code (ADK, Litestar, Events) that relies on positional
+    row access and must not be broken when a user configures ``DictCursor`` at
+    the connection level.
+    """
+
+    __slots__ = ("connection", "cursor", "cursor_class")
+
+    def __init__(self, connection: "AiomysqlConnection", cursor_class: "type[Any] | None" = None) -> None:
+        self.connection = connection
+        self.cursor_class = cursor_class
+        self.cursor: AiomysqlRawCursor | None = None
+
+    async def __aenter__(self) -> "AiomysqlRawCursor":
+        if self.cursor_class is None:
+            self.cursor = await self.connection.cursor()
+        else:
+            self.cursor = await self.connection.cursor(self.cursor_class)
+        return self.cursor
+
+    async def __aexit__(self, *_: Any) -> None:
+        if self.cursor is not None:
+            await self.cursor.close()
+
+
+class AiomysqlSessionContext:
+    """Async context manager for aiomysql sessions.
+
+    This class is intentionally excluded from mypyc compilation to avoid ABI
+    boundary issues. It receives callables from uncompiled config classes and
+    instantiates compiled Driver objects, acting as a bridge between compiled
+    and uncompiled code.
+
+    Uses callable-based connection management to decouple from config implementation.
+    """
+
+    __slots__ = (
+        "_acquire_connection",
+        "_connection",
+        "_driver",
+        "_driver_features",
+        "_prepare_driver",
+        "_release_connection",
+        "_statement_config",
+    )
+
+    def __init__(
+        self,
+        acquire_connection: "Callable[[], Any]",
+        release_connection: "Callable[[Any], Any]",
+        statement_config: "StatementConfig",
+        driver_features: "dict[str, Any]",
+        prepare_driver: "Callable[[AiomysqlDriver], AiomysqlDriver]",
+    ) -> None:
+        self._acquire_connection = acquire_connection
+        self._release_connection = release_connection
+        self._statement_config = statement_config
+        self._driver_features = driver_features
+        self._prepare_driver = prepare_driver
+        self._connection: Any = None
+        self._driver: AiomysqlDriver | None = None
+
+    async def __aenter__(self) -> "AiomysqlDriver":
+        from sqlspec.adapters.aiomysql.driver import AiomysqlDriver
+
+        self._connection = await self._acquire_connection()
+        self._driver = AiomysqlDriver(
+            connection=self._connection, statement_config=self._statement_config, driver_features=self._driver_features
+        )
+        return self._prepare_driver(self._driver)
+
+    async def __aexit__(
+        self, exc_type: "type[BaseException] | None", exc_val: "BaseException | None", exc_tb: "TracebackType | None"
+    ) -> "bool | None":
+        if self._connection is not None:
+            await self._release_connection(self._connection)
+            self._connection = None
+        return None
+
+
+__all__ = (
+    "AiomysqlConnection",
+    "AiomysqlCursor",
+    "AiomysqlDictCursor",
+    "AiomysqlPool",
+    "AiomysqlRawCursor",
+    "AiomysqlSessionContext",
+)

sqlspec-0.45.0/sqlspec/adapters/aiomysql/adk/__init__.py

@@ -0,0 +1,5 @@
+"""aiomysql ADK store for Google Agent Development Kit."""
+
+from sqlspec.adapters.aiomysql.adk.store import AiomysqlADKMemoryStore, AiomysqlADKStore
+
+__all__ = ("AiomysqlADKMemoryStore", "AiomysqlADKStore")