sqlspec 0.9.1__py3-none-any.whl → 0.10.0__py3-none-any.whl

sqlspec/adapters/psycopg/driver.py

@@ -2,25 +2,28 @@ import logging
 from contextlib import asynccontextmanager, contextmanager
 from typing import TYPE_CHECKING, Any, Optional, Union, cast, overload
 
+from psycopg import AsyncConnection, Connection
 from psycopg.rows import dict_row
 
-from sqlspec.base import AsyncDriverAdapterProtocol, SyncDriverAdapterProtocol, T
+from sqlspec.base import AsyncDriverAdapterProtocol, SyncDriverAdapterProtocol
 from sqlspec.exceptions import SQLParsingError
+from sqlspec.mixins import SQLTranslatorMixin
 from sqlspec.statement import PARAM_REGEX, SQLStatement
 
 if TYPE_CHECKING:
     from collections.abc import AsyncGenerator, Generator, Sequence
 
-    from psycopg import AsyncConnection, Connection
-
-    from sqlspec.typing import ModelDTOT, StatementParameterType
+    from sqlspec.typing import ModelDTOT, StatementParameterType, T
 
 logger = logging.getLogger("sqlspec")
 
-__all__ = ("PsycopgAsyncDriver", "PsycopgSyncDriver")
+__all__ = ("PsycopgAsyncConnection", "PsycopgAsyncDriver", "PsycopgSyncConnection", "PsycopgSyncDriver")
+
+PsycopgSyncConnection = Connection
+PsycopgAsyncConnection = AsyncConnection
 
 
-class PsycopgParameterParser:
+class PsycopgDriverBase:
     dialect: str
 
     def _process_sql_params(
@@ -76,13 +79,17 @@ class PsycopgParameterParser:
         return processed_sql, processed_params
 
 
-class PsycopgSyncDriver(PsycopgParameterParser, SyncDriverAdapterProtocol["Connection"]):
+class PsycopgSyncDriver(
+    PsycopgDriverBase,
+    SQLTranslatorMixin["PsycopgSyncConnection"],
+    SyncDriverAdapterProtocol["PsycopgSyncConnection"],
+):
     """Psycopg Sync Driver Adapter."""
 
-    connection: "Connection"
+    connection: "PsycopgSyncConnection"
     dialect: str = "postgres"
 
-    def __init__(self, connection: "Connection") -> None:
+    def __init__(self, connection: "PsycopgSyncConnection") -> None:
         self.connection = connection
 
     def _process_sql_params(
@@ -92,7 +99,6 @@ class PsycopgSyncDriver(PsycopgParameterParser, SyncDriverAdapterProtocol["Conne
         /,
         **kwargs: Any,
     ) -> "tuple[str, Optional[Union[tuple[Any, ...], list[Any], dict[str, Any]]]]":
-        """Process SQL and parameters, converting :name -> %(name)s if needed."""
         stmt = SQLStatement(sql=sql, parameters=parameters, dialect=self.dialect, kwargs=kwargs or None)
         processed_sql, processed_params = stmt.process()
 
@@ -139,7 +145,7 @@ class PsycopgSyncDriver(PsycopgParameterParser, SyncDriverAdapterProtocol["Conne
 
     @staticmethod
     @contextmanager
-    def _with_cursor(connection: "Connection") -> "Generator[Any, None, None]":
+    def _with_cursor(connection: "PsycopgSyncConnection") -> "Generator[Any, None, None]":
         cursor = connection.cursor(row_factory=dict_row)
         try:
             yield cursor
@@ -154,7 +160,7 @@ class PsycopgSyncDriver(PsycopgParameterParser, SyncDriverAdapterProtocol["Conne
         parameters: "Optional[StatementParameterType]" = None,
         /,
         *,
-        connection: "Optional[Connection]" = None,
+        connection: "Optional[PsycopgSyncConnection]" = None,
         schema_type: None = None,
         **kwargs: Any,
     ) -> "Sequence[dict[str, Any]]": ...
@@ -165,7 +171,7 @@ class PsycopgSyncDriver(PsycopgParameterParser, SyncDriverAdapterProtocol["Conne
         parameters: "Optional[StatementParameterType]" = None,
         /,
         *,
-        connection: "Optional[Connection]" = None,
+        connection: "Optional[PsycopgSyncConnection]" = None,
         schema_type: "type[ModelDTOT]",
         **kwargs: Any,
     ) -> "Sequence[ModelDTOT]": ...
@@ -176,7 +182,7 @@ class PsycopgSyncDriver(PsycopgParameterParser, SyncDriverAdapterProtocol["Conne
         /,
         *,
         schema_type: "Optional[type[ModelDTOT]]" = None,
-        connection: "Optional[Connection]" = None,
+        connection: "Optional[PsycopgSyncConnection]" = None,
         **kwargs: Any,
     ) -> "Sequence[Union[ModelDTOT, dict[str, Any]]]":
         """Fetch data from the database.
@@ -203,7 +209,7 @@ class PsycopgSyncDriver(PsycopgParameterParser, SyncDriverAdapterProtocol["Conne
         parameters: "Optional[StatementParameterType]" = None,
         /,
         *,
-        connection: "Optional[Connection]" = None,
+        connection: "Optional[PsycopgSyncConnection]" = None,
         schema_type: None = None,
         **kwargs: Any,
     ) -> "dict[str, Any]": ...
@@ -214,7 +220,7 @@ class PsycopgSyncDriver(PsycopgParameterParser, SyncDriverAdapterProtocol["Conne
         parameters: "Optional[StatementParameterType]" = None,
         /,
         *,
-        connection: "Optional[Connection]" = None,
+        connection: "Optional[PsycopgSyncConnection]" = None,
         schema_type: "type[ModelDTOT]",
         **kwargs: Any,
     ) -> "ModelDTOT": ...
@@ -224,7 +230,7 @@ class PsycopgSyncDriver(PsycopgParameterParser, SyncDriverAdapterProtocol["Conne
         parameters: "Optional[StatementParameterType]" = None,
         /,
         *,
-        connection: "Optional[Connection]" = None,
+        connection: "Optional[PsycopgSyncConnection]" = None,
         schema_type: "Optional[type[ModelDTOT]]" = None,
         **kwargs: Any,
     ) -> "Union[ModelDTOT, dict[str, Any]]":
@@ -250,7 +256,7 @@ class PsycopgSyncDriver(PsycopgParameterParser, SyncDriverAdapterProtocol["Conne
         parameters: "Optional[StatementParameterType]" = None,
         /,
         *,
-        connection: "Optional[Connection]" = None,
+        connection: "Optional[PsycopgSyncConnection]" = None,
         schema_type: None = None,
         **kwargs: Any,
     ) -> "Optional[dict[str, Any]]": ...
@@ -261,7 +267,7 @@ class PsycopgSyncDriver(PsycopgParameterParser, SyncDriverAdapterProtocol["Conne
         parameters: "Optional[StatementParameterType]" = None,
         /,
         *,
-        connection: "Optional[Connection]" = None,
+        connection: "Optional[PsycopgSyncConnection]" = None,
         schema_type: "type[ModelDTOT]",
         **kwargs: Any,
     ) -> "Optional[ModelDTOT]": ...
@@ -271,7 +277,7 @@ class PsycopgSyncDriver(PsycopgParameterParser, SyncDriverAdapterProtocol["Conne
         parameters: "Optional[StatementParameterType]" = None,
         /,
         *,
-        connection: "Optional[Connection]" = None,
+        connection: "Optional[PsycopgSyncConnection]" = None,
         schema_type: "Optional[type[ModelDTOT]]" = None,
         **kwargs: Any,
     ) -> "Optional[Union[ModelDTOT, dict[str, Any]]]":
@@ -298,7 +304,7 @@ class PsycopgSyncDriver(PsycopgParameterParser, SyncDriverAdapterProtocol["Conne
         parameters: "Optional[StatementParameterType]" = None,
         /,
         *,
-        connection: "Optional[Connection]" = None,
+        connection: "Optional[PsycopgSyncConnection]" = None,
         schema_type: None = None,
         **kwargs: Any,
     ) -> "Any": ...
@@ -309,7 +315,7 @@ class PsycopgSyncDriver(PsycopgParameterParser, SyncDriverAdapterProtocol["Conne
         parameters: "Optional[StatementParameterType]" = None,
         /,
         *,
-        connection: "Optional[Connection]" = None,
+        connection: "Optional[PsycopgSyncConnection]" = None,
         schema_type: "type[T]",
         **kwargs: Any,
     ) -> "T": ...
@@ -319,7 +325,7 @@ class PsycopgSyncDriver(PsycopgParameterParser, SyncDriverAdapterProtocol["Conne
         parameters: "Optional[StatementParameterType]" = None,
         /,
         *,
-        connection: "Optional[Connection]" = None,
+        connection: "Optional[PsycopgSyncConnection]" = None,
         schema_type: "Optional[type[T]]" = None,
         **kwargs: Any,
     ) -> "Union[T, Any]":
@@ -347,7 +353,7 @@ class PsycopgSyncDriver(PsycopgParameterParser, SyncDriverAdapterProtocol["Conne
         parameters: "Optional[StatementParameterType]" = None,
         /,
         *,
-        connection: "Optional[Connection]" = None,
+        connection: "Optional[PsycopgSyncConnection]" = None,
         schema_type: None = None,
         **kwargs: Any,
     ) -> "Optional[Any]": ...
@@ -358,7 +364,7 @@ class PsycopgSyncDriver(PsycopgParameterParser, SyncDriverAdapterProtocol["Conne
         parameters: "Optional[StatementParameterType]" = None,
         /,
         *,
-        connection: "Optional[Connection]" = None,
+        connection: "Optional[PsycopgSyncConnection]" = None,
         schema_type: "type[T]",
         **kwargs: Any,
     ) -> "Optional[T]": ...
@@ -368,7 +374,7 @@ class PsycopgSyncDriver(PsycopgParameterParser, SyncDriverAdapterProtocol["Conne
         parameters: "Optional[StatementParameterType]" = None,
         /,
         *,
-        connection: "Optional[Connection]" = None,
+        connection: "Optional[PsycopgSyncConnection]" = None,
         schema_type: "Optional[type[T]]" = None,
         **kwargs: Any,
     ) -> "Optional[Union[T, Any]]":
@@ -397,7 +403,7 @@ class PsycopgSyncDriver(PsycopgParameterParser, SyncDriverAdapterProtocol["Conne
         parameters: "Optional[StatementParameterType]" = None,
         /,
         *,
-        connection: "Optional[Connection]" = None,
+        connection: "Optional[PsycopgSyncConnection]" = None,
         **kwargs: Any,
     ) -> int:
         """Execute an INSERT, UPDATE, or DELETE query and return the number of affected rows.
@@ -418,7 +424,7 @@ class PsycopgSyncDriver(PsycopgParameterParser, SyncDriverAdapterProtocol["Conne
         parameters: "Optional[StatementParameterType]" = None,
         /,
         *,
-        connection: "Optional[Connection]" = None,
+        connection: "Optional[PsycopgSyncConnection]" = None,
         schema_type: None = None,
         **kwargs: Any,
     ) -> "dict[str, Any]": ...
@@ -429,7 +435,7 @@ class PsycopgSyncDriver(PsycopgParameterParser, SyncDriverAdapterProtocol["Conne
         parameters: "Optional[StatementParameterType]" = None,
         /,
         *,
-        connection: "Optional[Connection]" = None,
+        connection: "Optional[PsycopgSyncConnection]" = None,
         schema_type: "type[ModelDTOT]",
         **kwargs: Any,
     ) -> "ModelDTOT": ...
@@ -439,7 +445,7 @@ class PsycopgSyncDriver(PsycopgParameterParser, SyncDriverAdapterProtocol["Conne
         parameters: "Optional[StatementParameterType]" = None,
         /,
         *,
-        connection: "Optional[Connection]" = None,
+        connection: "Optional[PsycopgSyncConnection]" = None,
         schema_type: "Optional[type[ModelDTOT]]" = None,
         **kwargs: Any,
     ) -> "Optional[Union[dict[str, Any], ModelDTOT]]":
@@ -467,7 +473,7 @@ class PsycopgSyncDriver(PsycopgParameterParser, SyncDriverAdapterProtocol["Conne
         parameters: "Optional[StatementParameterType]" = None,
         /,
         *,
-        connection: "Optional[Connection]" = None,
+        connection: "Optional[PsycopgSyncConnection]" = None,
         **kwargs: Any,
     ) -> str:
         """Execute a script.
@@ -482,18 +488,22 @@ class PsycopgSyncDriver(PsycopgParameterParser, SyncDriverAdapterProtocol["Conne
             return str(cursor.statusmessage) if cursor.statusmessage is not None else "DONE"
 
 
-class PsycopgAsyncDriver(PsycopgParameterParser, AsyncDriverAdapterProtocol["AsyncConnection"]):
+class PsycopgAsyncDriver(
+    PsycopgDriverBase,
+    SQLTranslatorMixin["PsycopgAsyncConnection"],
+    AsyncDriverAdapterProtocol["PsycopgAsyncConnection"],
+):
     """Psycopg Async Driver Adapter."""
 
-    connection: "AsyncConnection"
+    connection: "PsycopgAsyncConnection"
     dialect: str = "postgres"
 
-    def __init__(self, connection: "AsyncConnection") -> None:
+    def __init__(self, connection: "PsycopgAsyncConnection") -> None:
         self.connection = connection
 
     @staticmethod
     @asynccontextmanager
-    async def _with_cursor(connection: "AsyncConnection") -> "AsyncGenerator[Any, None]":
+    async def _with_cursor(connection: "PsycopgAsyncConnection") -> "AsyncGenerator[Any, None]":
         cursor = connection.cursor(row_factory=dict_row)
         try:
             yield cursor
@@ -508,7 +518,7 @@ class PsycopgAsyncDriver(PsycopgParameterParser, AsyncDriverAdapterProtocol["Asy
         parameters: "Optional[StatementParameterType]" = None,
         /,
         *,
-        connection: "Optional[AsyncConnection]" = None,
+        connection: "Optional[PsycopgAsyncConnection]" = None,
         schema_type: None = None,
         **kwargs: Any,
     ) -> "Sequence[dict[str, Any]]": ...
@@ -519,7 +529,7 @@ class PsycopgAsyncDriver(PsycopgParameterParser, AsyncDriverAdapterProtocol["Asy
         parameters: "Optional[StatementParameterType]" = None,
         /,
         *,
-        connection: "Optional[AsyncConnection]" = None,
+        connection: "Optional[PsycopgAsyncConnection]" = None,
         schema_type: "type[ModelDTOT]",
         **kwargs: Any,
     ) -> "Sequence[ModelDTOT]": ...
@@ -529,7 +539,7 @@ class PsycopgAsyncDriver(PsycopgParameterParser, AsyncDriverAdapterProtocol["Asy
         parameters: "Optional[StatementParameterType]" = None,
         /,
         *,
-        connection: "Optional[AsyncConnection]" = None,
+        connection: "Optional[PsycopgAsyncConnection]" = None,
         schema_type: "Optional[type[ModelDTOT]]" = None,
         **kwargs: Any,
     ) -> "Sequence[Union[ModelDTOT, dict[str, Any]]]":
@@ -558,7 +568,7 @@ class PsycopgAsyncDriver(PsycopgParameterParser, AsyncDriverAdapterProtocol["Asy
         parameters: "Optional[StatementParameterType]" = None,
         /,
         *,
-        connection: "Optional[AsyncConnection]" = None,
+        connection: "Optional[PsycopgAsyncConnection]" = None,
         schema_type: None = None,
         **kwargs: Any,
     ) -> "dict[str, Any]": ...
@@ -569,7 +579,7 @@ class PsycopgAsyncDriver(PsycopgParameterParser, AsyncDriverAdapterProtocol["Asy
         parameters: "Optional[StatementParameterType]" = None,
         /,
         *,
-        connection: "Optional[AsyncConnection]" = None,
+        connection: "Optional[PsycopgAsyncConnection]" = None,
         schema_type: "type[ModelDTOT]",
         **kwargs: Any,
     ) -> "ModelDTOT": ...
@@ -579,7 +589,7 @@ class PsycopgAsyncDriver(PsycopgParameterParser, AsyncDriverAdapterProtocol["Asy
         parameters: "Optional[StatementParameterType]" = None,
         /,
         *,
-        connection: "Optional[AsyncConnection]" = None,
+        connection: "Optional[PsycopgAsyncConnection]" = None,
         schema_type: "Optional[type[ModelDTOT]]" = None,
         **kwargs: Any,
     ) -> "Union[ModelDTOT, dict[str, Any]]":
@@ -606,7 +616,7 @@ class PsycopgAsyncDriver(PsycopgParameterParser, AsyncDriverAdapterProtocol["Asy
         parameters: "Optional[StatementParameterType]" = None,
         /,
         *,
-        connection: "Optional[AsyncConnection]" = None,
+        connection: "Optional[PsycopgAsyncConnection]" = None,
         schema_type: None = None,
         **kwargs: Any,
     ) -> "Optional[dict[str, Any]]": ...
@@ -617,7 +627,7 @@ class PsycopgAsyncDriver(PsycopgParameterParser, AsyncDriverAdapterProtocol["Asy
         parameters: "Optional[StatementParameterType]" = None,
         /,
         *,
-        connection: "Optional[AsyncConnection]" = None,
+        connection: "Optional[PsycopgAsyncConnection]" = None,
         schema_type: "type[ModelDTOT]",
         **kwargs: Any,
     ) -> "Optional[ModelDTOT]": ...
@@ -628,7 +638,7 @@ class PsycopgAsyncDriver(PsycopgParameterParser, AsyncDriverAdapterProtocol["Asy
         /,
         *,
         schema_type: "Optional[type[ModelDTOT]]" = None,
-        connection: "Optional[AsyncConnection]" = None,
+        connection: "Optional[PsycopgAsyncConnection]" = None,
         **kwargs: Any,
     ) -> "Optional[Union[ModelDTOT, dict[str, Any]]]":
         """Fetch one row from the database.
@@ -655,7 +665,7 @@ class PsycopgAsyncDriver(PsycopgParameterParser, AsyncDriverAdapterProtocol["Asy
         parameters: "Optional[StatementParameterType]" = None,
         /,
         *,
-        connection: "Optional[AsyncConnection]" = None,
+        connection: "Optional[PsycopgAsyncConnection]" = None,
         schema_type: None = None,
         **kwargs: Any,
     ) -> "Any": ...
@@ -666,7 +676,7 @@ class PsycopgAsyncDriver(PsycopgParameterParser, AsyncDriverAdapterProtocol["Asy
         parameters: "Optional[StatementParameterType]" = None,
         /,
         *,
-        connection: "Optional[AsyncConnection]" = None,
+        connection: "Optional[PsycopgAsyncConnection]" = None,
         schema_type: "type[T]",
         **kwargs: Any,
     ) -> "T": ...
@@ -676,7 +686,7 @@ class PsycopgAsyncDriver(PsycopgParameterParser, AsyncDriverAdapterProtocol["Asy
         parameters: "Optional[StatementParameterType]" = None,
         /,
         *,
-        connection: "Optional[AsyncConnection]" = None,
+        connection: "Optional[PsycopgAsyncConnection]" = None,
         schema_type: "Optional[type[T]]" = None,
         **kwargs: Any,
     ) -> "Union[T, Any]":
@@ -704,7 +714,7 @@ class PsycopgAsyncDriver(PsycopgParameterParser, AsyncDriverAdapterProtocol["Asy
         parameters: "Optional[StatementParameterType]" = None,
         /,
         *,
-        connection: "Optional[AsyncConnection]" = None,
+        connection: "Optional[PsycopgAsyncConnection]" = None,
         schema_type: "Optional[type[T]]" = None,
         **kwargs: Any,
     ) -> "Optional[Union[T, Any]]":
@@ -734,7 +744,7 @@ class PsycopgAsyncDriver(PsycopgParameterParser, AsyncDriverAdapterProtocol["Asy
         parameters: "Optional[StatementParameterType]" = None,
         /,
         *,
-        connection: "Optional[AsyncConnection]" = None,
+        connection: "Optional[PsycopgAsyncConnection]" = None,
         **kwargs: Any,
     ) -> int:
         """Execute an INSERT, UPDATE, or DELETE query and return the number of affected rows.
@@ -760,7 +770,7 @@ class PsycopgAsyncDriver(PsycopgParameterParser, AsyncDriverAdapterProtocol["Asy
         parameters: "Optional[StatementParameterType]" = None,
         /,
         *,
-        connection: "Optional[AsyncConnection]" = None,
+        connection: "Optional[PsycopgAsyncConnection]" = None,
         schema_type: None = None,
         **kwargs: Any,
     ) -> "dict[str, Any]": ...
@@ -771,7 +781,7 @@ class PsycopgAsyncDriver(PsycopgParameterParser, AsyncDriverAdapterProtocol["Asy
         parameters: "Optional[StatementParameterType]" = None,
         /,
         *,
-        connection: "Optional[AsyncConnection]" = None,
+        connection: "Optional[PsycopgAsyncConnection]" = None,
         schema_type: "type[ModelDTOT]",
         **kwargs: Any,
     ) -> "ModelDTOT": ...
@@ -781,7 +791,7 @@ class PsycopgAsyncDriver(PsycopgParameterParser, AsyncDriverAdapterProtocol["Asy
         parameters: "Optional[StatementParameterType]" = None,
         /,
         *,
-        connection: "Optional[AsyncConnection]" = None,
+        connection: "Optional[PsycopgAsyncConnection]" = None,
         schema_type: "Optional[type[ModelDTOT]]" = None,
         **kwargs: Any,
     ) -> "Optional[Union[dict[str, Any], ModelDTOT]]":
@@ -810,7 +820,7 @@ class PsycopgAsyncDriver(PsycopgParameterParser, AsyncDriverAdapterProtocol["Asy
         parameters: "Optional[StatementParameterType]" = None,
         /,
         *,
-        connection: "Optional[AsyncConnection]" = None,
+        connection: "Optional[PsycopgAsyncConnection]" = None,
         **kwargs: Any,
     ) -> str:
         """Execute a script.

sqlspec/adapters/sqlite/__init__.py

@@ -1,7 +1,8 @@
 from sqlspec.adapters.sqlite.config import SqliteConfig
-from sqlspec.adapters.sqlite.driver import SqliteDriver
+from sqlspec.adapters.sqlite.driver import SqliteConnection, SqliteDriver
 
 __all__ = (
     "SqliteConfig",
+    "SqliteConnection",
     "SqliteDriver",
 )

sqlspec/adapters/sqlite/config.py

@@ -1,9 +1,9 @@
+import sqlite3
 from contextlib import contextmanager
 from dataclasses import dataclass, field
-from sqlite3 import Connection
 from typing import TYPE_CHECKING, Any, Literal, Optional, Union
 
-from sqlspec.adapters.sqlite.driver import SqliteDriver
+from sqlspec.adapters.sqlite.driver import SqliteConnection, SqliteDriver
 from sqlspec.base import NoPoolSyncConfig
 from sqlspec.exceptions import ImproperConfigurationError
 from sqlspec.typing import Empty, EmptyType, dataclass_to_dict
@@ -16,7 +16,7 @@ __all__ = ("SqliteConfig",)
 
 
 @dataclass
-class SqliteConfig(NoPoolSyncConfig["Connection", "SqliteDriver"]):
+class SqliteConfig(NoPoolSyncConfig["SqliteConnection", "SqliteDriver"]):
     """Configuration for SQLite database connections.
 
     This class provides configuration options for SQLite database connections, wrapping all parameters
@@ -40,7 +40,7 @@ class SqliteConfig(NoPoolSyncConfig["Connection", "SqliteDriver"]):
     check_same_thread: "Union[bool, EmptyType]" = Empty
     """If True (default), ProgrammingError is raised if the database connection is used by a thread other than the one that created it. If False, the connection may be shared across multiple threads."""
 
-    factory: "Union[type[Connection], EmptyType]" = Empty
+    factory: "Union[type[SqliteConnection], EmptyType]" = Empty
     """A custom Connection class factory. If given, must be a callable that returns a Connection instance."""
 
     cached_statements: "Union[int, EmptyType]" = Empty
@@ -50,7 +50,7 @@ class SqliteConfig(NoPoolSyncConfig["Connection", "SqliteDriver"]):
     """If set to True, database is interpreted as a URI with supported options."""
     driver_type: "type[SqliteDriver]" = field(init=False, default_factory=lambda: SqliteDriver)
     """Type of the driver object"""
-    connection_type: "type[Connection]" = field(init=False, default_factory=lambda: Connection)
+    connection_type: "type[SqliteConnection]" = field(init=False, default_factory=lambda: SqliteConnection)
     """Type of the connection object"""
 
     @property
@@ -61,10 +61,13 @@ class SqliteConfig(NoPoolSyncConfig["Connection", "SqliteDriver"]):
             A string keyed dict of config kwargs for the sqlite3.connect() function.
         """
         return dataclass_to_dict(
-            self, exclude_empty=True, convert_nested=False, exclude={"pool_instance", "driver_type", "connection_type"}
+            self,
+            exclude_empty=True,
+            convert_nested=False,
+            exclude={"pool_instance", "driver_type", "connection_type"},
         )
 
-    def create_connection(self) -> "Connection":
+    def create_connection(self) -> "SqliteConnection":
         """Create and return a new database connection.
 
         Returns:
@@ -73,8 +76,6 @@ class SqliteConfig(NoPoolSyncConfig["Connection", "SqliteDriver"]):
         Raises:
             ImproperConfigurationError: If the connection could not be established.
         """
-        import sqlite3
-
         try:
             return sqlite3.connect(**self.connection_config_dict)  # type: ignore[no-any-return,unused-ignore]
         except Exception as e:
@@ -82,7 +83,7 @@ class SqliteConfig(NoPoolSyncConfig["Connection", "SqliteDriver"]):
             raise ImproperConfigurationError(msg) from e
 
     @contextmanager
-    def provide_connection(self, *args: "Any", **kwargs: "Any") -> "Generator[Connection, None, None]":
+    def provide_connection(self, *args: "Any", **kwargs: "Any") -> "Generator[SqliteConnection, None, None]":
         """Create and provide a database connection.
 
         Yields:
@@ -100,7 +101,7 @@ class SqliteConfig(NoPoolSyncConfig["Connection", "SqliteDriver"]):
         """Create and provide a database connection.
 
         Yields:
-            A DuckDB driver instance.
+            A SQLite driver instance.
 
 
         """