feat: add aiomysql adapter (#413)

## Summary

- Add new async MySQL adapter using [aiomysql](https://github.com/aio-libs/aiomysql) (built on PyMySQL)
- Mirrors the existing `asyncmy` adapter structure for consistency
- Extensions (ADK, Litestar, Events) deferred to follow-up

Author: hasansezertasan

.gitignore

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

.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.11"
     hooks:
       - id: ruff
         args: ["--fix"]

docs/reference/adapters/index.rst

@@ -53,7 +53,7 @@ exports a typed config class and a driver implementation.
       :link: mysql
       :link-type: doc
 
-      MySQL via mysql-connector, PyMySQL, and asyncmy.
+      MySQL via mysql-connector, PyMySQL, asyncmy, and aiomysql.
 
    .. grid-item-card:: BigQuery
       :link: bigquery
@@ -151,6 +151,12 @@ Feature Comparison
      - Yes
      -
      -
+   * - aiomysql
+     -
+     - Yes
+     - Yes
+     -
+     -
    * - bigquery
      - Yes
      -

docs/reference/adapters/mysql.rst

@@ -74,3 +74,43 @@ Driver
 .. autoclass:: sqlspec.adapters.asyncmy.AsyncmyDriver
    :members:
    :show-inheritance:
+
+aiomysql
+========
+
+Async MySQL driver (PyMySQL-compatible wire protocol, asyncio-native).
+
+Configuration
+-------------
+
+.. autoclass:: sqlspec.adapters.aiomysql.AiomysqlConfig
+   :members:
+   :show-inheritance:
+
+Driver
+------
+
+.. autoclass:: sqlspec.adapters.aiomysql.AiomysqlDriver
+   :members:
+   :show-inheritance:
+
+Connection Parameters
+---------------------
+
+.. autoclass:: sqlspec.adapters.aiomysql.AiomysqlConnectionParams
+   :members:
+   :show-inheritance:
+
+Pool Parameters
+---------------
+
+.. autoclass:: sqlspec.adapters.aiomysql.AiomysqlPoolParams
+   :members:
+   :show-inheritance:
+
+Driver Features
+---------------
+
+.. autoclass:: sqlspec.adapters.aiomysql.AiomysqlDriverFeatures
+   :members:
+   :show-inheritance:

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.44.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"]
@@ -243,7 +244,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.44.0"
 ignore_missing_files = false
 ignore_missing_version = false
 message = "chore(release): bump to v{new_version}"
@@ -358,6 +359,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 +402,8 @@ module = [
   "orjson",
   "uvicorn.*",
   "uvloop.*",
+  "aiomysql",
+  "aiomysql.*",
   "asyncmy",
   "asyncmy.*",
   "pyarrow",

sqlspec/adapters/adbc/_typing.py

@@ -45,7 +45,7 @@ def __enter__(self) -> "AdbcRawCursor":
     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/adapters/adbc/adk/store.py

@@ -445,7 +445,7 @@ def _create_tables(self) -> None:
                 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 @@ def _create_session(
                 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 @@ def _get_session(self, session_id: str) -> "SessionRecord | None":
                         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 @@ def _update_session_state(self, session_id: str, state: "dict[str, Any]") -> Non
                 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 @@ def _delete_session(self, session_id: str) -> None:
                 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 @@ def _list_sessions(self, app_name: str, user_id: str | None = None) -> "list[Ses
                         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 @@ def _insert_event(self, event_record: "EventRecord") -> None:
                 )
                 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 @@ def _append_event_and_update_state(
                     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 @@ def _get_events(
                         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 @@ def _create_tables(self) -> None:
                 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 @@ def _insert_memory_entries(self, entries: "list[MemoryRecord]", owner_id: "objec
                             raise
                 conn.commit()
             finally:
-                cursor.close()  # type: ignore[no-untyped-call]
+                cursor.close()
 
         return inserted_count
 
@@ -1161,7 +1161,7 @@ def _search_entries(
                     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 @@ def _delete_entries_by_session(self, session_id: str) -> int:
                 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 @@ def _delete_entries_older_than(self, days: int) -> int:
                 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/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",
+)