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

sqlspec/adapters/sqlite/driver.py

@@ -1,32 +1,39 @@
+import sqlite3
 from contextlib import contextmanager
-from sqlite3 import Connection, Cursor
+from sqlite3 import Cursor
 from typing import TYPE_CHECKING, Any, Optional, Union, cast, overload
 
-from sqlspec.base import SyncDriverAdapterProtocol, T
+from sqlspec.base import SyncDriverAdapterProtocol
+from sqlspec.mixins import SQLTranslatorMixin
 
 if TYPE_CHECKING:
     from collections.abc import Generator, Sequence
 
-    from sqlspec.typing import ModelDTOT, StatementParameterType
+    from sqlspec.typing import ModelDTOT, StatementParameterType, T
 
-__all__ = ("SqliteDriver",)
+__all__ = ("SqliteConnection", "SqliteDriver")
 
+SqliteConnection = sqlite3.Connection
 
-class SqliteDriver(SyncDriverAdapterProtocol["Connection"]):
+
+class SqliteDriver(
+    SQLTranslatorMixin["SqliteConnection"],
+    SyncDriverAdapterProtocol["SqliteConnection"],
+):
     """SQLite Sync Driver Adapter."""
 
-    connection: "Connection"
+    connection: "SqliteConnection"
     dialect: str = "sqlite"
 
-    def __init__(self, connection: "Connection") -> None:
+    def __init__(self, connection: "SqliteConnection") -> None:
         self.connection = connection
 
     @staticmethod
-    def _cursor(connection: "Connection", *args: Any, **kwargs: Any) -> Cursor:
+    def _cursor(connection: "SqliteConnection", *args: Any, **kwargs: Any) -> Cursor:
         return connection.cursor(*args, **kwargs)  # type: ignore[no-any-return]
 
     @contextmanager
-    def _with_cursor(self, connection: "Connection") -> "Generator[Cursor, None, None]":
+    def _with_cursor(self, connection: "SqliteConnection") -> "Generator[Cursor, None, None]":
         cursor = self._cursor(connection)
         try:
             yield cursor
@@ -41,7 +48,7 @@ class SqliteDriver(SyncDriverAdapterProtocol["Connection"]):
         parameters: "Optional[StatementParameterType]" = None,
         /,
         *,
-        connection: "Optional[Connection]" = None,
+        connection: "Optional[SqliteConnection]" = None,
         schema_type: None = None,
         **kwargs: Any,
     ) -> "Sequence[dict[str, Any]]": ...
@@ -52,7 +59,7 @@ class SqliteDriver(SyncDriverAdapterProtocol["Connection"]):
         parameters: "Optional[StatementParameterType]" = None,
         /,
         *,
-        connection: "Optional[Connection]" = None,
+        connection: "Optional[SqliteConnection]" = None,
         schema_type: "type[ModelDTOT]",
         **kwargs: Any,
     ) -> "Sequence[ModelDTOT]": ...
@@ -62,7 +69,7 @@ class SqliteDriver(SyncDriverAdapterProtocol["Connection"]):
         parameters: Optional["StatementParameterType"] = None,
         /,
         *,
-        connection: Optional["Connection"] = None,
+        connection: Optional["SqliteConnection"] = None,
         schema_type: "Optional[type[ModelDTOT]]" = None,
         **kwargs: Any,
     ) -> "Sequence[Union[ModelDTOT, dict[str, Any]]]":
@@ -93,7 +100,7 @@ class SqliteDriver(SyncDriverAdapterProtocol["Connection"]):
         parameters: "Optional[StatementParameterType]" = None,
         /,
         *,
-        connection: "Optional[Connection]" = None,
+        connection: "Optional[SqliteConnection]" = None,
         schema_type: None = None,
         **kwargs: Any,
     ) -> "dict[str, Any]": ...
@@ -104,7 +111,7 @@ class SqliteDriver(SyncDriverAdapterProtocol["Connection"]):
         parameters: "Optional[StatementParameterType]" = None,
         /,
         *,
-        connection: "Optional[Connection]" = None,
+        connection: "Optional[SqliteConnection]" = None,
         schema_type: "type[ModelDTOT]",
         **kwargs: Any,
     ) -> "ModelDTOT": ...
@@ -114,7 +121,7 @@ class SqliteDriver(SyncDriverAdapterProtocol["Connection"]):
         parameters: Optional["StatementParameterType"] = None,
         /,
         *,
-        connection: Optional["Connection"] = None,
+        connection: Optional["SqliteConnection"] = None,
         schema_type: "Optional[type[ModelDTOT]]" = None,
         **kwargs: Any,
     ) -> "Union[ModelDTOT, dict[str, Any]]":
@@ -144,7 +151,7 @@ class SqliteDriver(SyncDriverAdapterProtocol["Connection"]):
         parameters: "Optional[StatementParameterType]" = None,
         /,
         *,
-        connection: "Optional[Connection]" = None,
+        connection: "Optional[SqliteConnection]" = None,
         schema_type: None = None,
         **kwargs: Any,
     ) -> "Optional[dict[str, Any]]": ...
@@ -155,7 +162,7 @@ class SqliteDriver(SyncDriverAdapterProtocol["Connection"]):
         parameters: "Optional[StatementParameterType]" = None,
         /,
         *,
-        connection: "Optional[Connection]" = None,
+        connection: "Optional[SqliteConnection]" = None,
         schema_type: "type[ModelDTOT]",
         **kwargs: Any,
     ) -> "Optional[ModelDTOT]": ...
@@ -165,7 +172,7 @@ class SqliteDriver(SyncDriverAdapterProtocol["Connection"]):
         parameters: Optional["StatementParameterType"] = None,
         /,
         *,
-        connection: Optional["Connection"] = None,
+        connection: Optional["SqliteConnection"] = None,
         schema_type: "Optional[type[ModelDTOT]]" = None,
         **kwargs: Any,
     ) -> "Optional[Union[ModelDTOT, dict[str, Any]]]":
@@ -196,7 +203,7 @@ class SqliteDriver(SyncDriverAdapterProtocol["Connection"]):
         parameters: "Optional[StatementParameterType]" = None,
         /,
         *,
-        connection: "Optional[Connection]" = None,
+        connection: "Optional[SqliteConnection]" = None,
         schema_type: None = None,
         **kwargs: Any,
     ) -> "Any": ...
@@ -207,7 +214,7 @@ class SqliteDriver(SyncDriverAdapterProtocol["Connection"]):
         parameters: "Optional[StatementParameterType]" = None,
         /,
         *,
-        connection: "Optional[Connection]" = None,
+        connection: "Optional[SqliteConnection]" = None,
         schema_type: "type[T]",
         **kwargs: Any,
     ) -> "T": ...
@@ -217,7 +224,7 @@ class SqliteDriver(SyncDriverAdapterProtocol["Connection"]):
         parameters: "Optional[StatementParameterType]" = None,
         /,
         *,
-        connection: "Optional[Connection]" = None,
+        connection: "Optional[SqliteConnection]" = None,
         schema_type: "Optional[type[T]]" = None,
         **kwargs: Any,
     ) -> "Union[T, Any]":
@@ -246,7 +253,7 @@ class SqliteDriver(SyncDriverAdapterProtocol["Connection"]):
         parameters: "Optional[StatementParameterType]" = None,
         /,
         *,
-        connection: "Optional[Connection]" = None,
+        connection: "Optional[SqliteConnection]" = None,
         schema_type: None = None,
         **kwargs: Any,
     ) -> "Optional[Any]": ...
@@ -257,7 +264,7 @@ class SqliteDriver(SyncDriverAdapterProtocol["Connection"]):
         parameters: "Optional[StatementParameterType]" = None,
         /,
         *,
-        connection: "Optional[Connection]" = None,
+        connection: "Optional[SqliteConnection]" = None,
         schema_type: "type[T]",
         **kwargs: Any,
     ) -> "Optional[T]": ...
@@ -267,7 +274,7 @@ class SqliteDriver(SyncDriverAdapterProtocol["Connection"]):
         parameters: "Optional[StatementParameterType]" = None,
         /,
         *,
-        connection: "Optional[Connection]" = None,
+        connection: "Optional[SqliteConnection]" = None,
         schema_type: "Optional[type[T]]" = None,
         **kwargs: Any,
     ) -> "Optional[Union[T, Any]]":
@@ -296,7 +303,7 @@ class SqliteDriver(SyncDriverAdapterProtocol["Connection"]):
         parameters: Optional["StatementParameterType"] = None,
         /,
         *,
-        connection: Optional["Connection"] = None,
+        connection: Optional["SqliteConnection"] = None,
         **kwargs: Any,
     ) -> int:
         """Insert, update, or delete data from the database.
@@ -321,7 +328,7 @@ class SqliteDriver(SyncDriverAdapterProtocol["Connection"]):
         parameters: "Optional[StatementParameterType]" = None,
         /,
         *,
-        connection: "Optional[Connection]" = None,
+        connection: "Optional[SqliteConnection]" = None,
         schema_type: None = None,
         **kwargs: Any,
     ) -> "dict[str, Any]": ...
@@ -332,7 +339,7 @@ class SqliteDriver(SyncDriverAdapterProtocol["Connection"]):
         parameters: "Optional[StatementParameterType]" = None,
         /,
         *,
-        connection: "Optional[Connection]" = None,
+        connection: "Optional[SqliteConnection]" = None,
         schema_type: "type[ModelDTOT]",
         **kwargs: Any,
     ) -> "ModelDTOT": ...
@@ -342,7 +349,7 @@ class SqliteDriver(SyncDriverAdapterProtocol["Connection"]):
         parameters: Optional["StatementParameterType"] = None,
         /,
         *,
-        connection: Optional["Connection"] = None,
+        connection: Optional["SqliteConnection"] = None,
         schema_type: "Optional[type[ModelDTOT]]" = None,
         **kwargs: Any,
     ) -> "Optional[Union[dict[str, Any], ModelDTOT]]":
@@ -386,7 +393,7 @@ class SqliteDriver(SyncDriverAdapterProtocol["Connection"]):
         parameters: Optional["StatementParameterType"] = None,
         /,
         *,
-        connection: Optional["Connection"] = None,
+        connection: Optional["SqliteConnection"] = None,
         **kwargs: Any,
     ) -> str:
         """Execute a script.

sqlspec/base.py

@@ -20,16 +20,14 @@ from typing import (
 
 from sqlspec.exceptions import NotFoundError
 from sqlspec.statement import SQLStatement
-from sqlspec.typing import ModelDTOT, StatementParameterType
+from sqlspec.typing import ConnectionT, ModelDTOT, PoolT, StatementParameterType, T
 from sqlspec.utils.sync_tools import maybe_async_
 
 if TYPE_CHECKING:
     from contextlib import AbstractAsyncContextManager, AbstractContextManager
 
-    from pyarrow import Table as ArrowTable
 
 __all__ = (
-    "AsyncArrowBulkOperationsMixin",
     "AsyncDatabaseConfig",
     "AsyncDriverAdapterProtocol",
     "CommonDriverAttributes",
@@ -39,15 +37,10 @@ __all__ = (
     "NoPoolSyncConfig",
     "SQLSpec",
     "SQLStatement",
-    "SyncArrowBulkOperationsMixin",
     "SyncDatabaseConfig",
     "SyncDriverAdapterProtocol",
 )
 
-T = TypeVar("T")
-ConnectionT = TypeVar("ConnectionT")
-PoolT = TypeVar("PoolT")
-PoolT_co = TypeVar("PoolT_co", covariant=True)
 AsyncConfigT = TypeVar("AsyncConfigT", bound="Union[AsyncDatabaseConfig[Any, Any, Any], NoPoolAsyncConfig[Any, Any]]")
 SyncConfigT = TypeVar("SyncConfigT", bound="Union[SyncDatabaseConfig[Any, Any, Any], NoPoolSyncConfig[Any, Any]]")
 ConfigT = TypeVar(
@@ -558,35 +551,6 @@ class CommonDriverAttributes(Generic[ConnectionT]):
         return stmt.process()
 
 
-class SyncArrowBulkOperationsMixin(Generic[ConnectionT]):
-    """Mixin for sync drivers supporting bulk Apache Arrow operations."""
-
-    __supports_arrow__: "ClassVar[bool]" = True
-
-    @abstractmethod
-    def select_arrow(  # pyright: ignore[reportUnknownParameterType]
-        self,
-        sql: str,
-        parameters: "Optional[StatementParameterType]" = None,
-        /,
-        *,
-        connection: "Optional[ConnectionT]" = None,
-        **kwargs: Any,
-    ) -> "ArrowTable":  # pyright: ignore[reportUnknownReturnType]
-        """Execute a SQL query and return results as an Apache Arrow Table.
-
-        Args:
-            sql: The SQL query string.
-            parameters: Parameters for the query.
-            connection: Optional connection override.
-            **kwargs: Additional keyword arguments to merge with parameters if parameters is a dict.
-
-        Returns:
-            An Apache Arrow Table containing the query results.
-        """
-        raise NotImplementedError
-
-
 class SyncDriverAdapterProtocol(CommonDriverAttributes[ConnectionT], ABC, Generic[ConnectionT]):
     connection: "ConnectionT"
 
@@ -844,35 +808,6 @@ class SyncDriverAdapterProtocol(CommonDriverAttributes[ConnectionT], ABC, Generi
     ) -> str: ...
 
 
-class AsyncArrowBulkOperationsMixin(Generic[ConnectionT]):
-    """Mixin for async drivers supporting bulk Apache Arrow operations."""
-
-    __supports_arrow__: "ClassVar[bool]" = True
-
-    @abstractmethod
-    async def select_arrow(  # pyright: ignore[reportUnknownParameterType]
-        self,
-        sql: str,
-        parameters: "Optional[StatementParameterType]" = None,
-        /,
-        *,
-        connection: "Optional[ConnectionT]" = None,
-        **kwargs: Any,
-    ) -> "ArrowTable":  # pyright: ignore[reportUnknownReturnType]
-        """Execute a SQL query and return results as an Apache Arrow Table.
-
-        Args:
-            sql: The SQL query string.
-            parameters: Parameters for the query.
-            connection: Optional connection override.
-            **kwargs: Additional keyword arguments to merge with parameters if parameters is a dict.
-
-        Returns:
-            An Apache Arrow Table containing the query results.
-        """
-        raise NotImplementedError
-
-
 class AsyncDriverAdapterProtocol(CommonDriverAttributes[ConnectionT], ABC, Generic[ConnectionT]):
     connection: "ConnectionT"
 

sqlspec/exceptions.py

@@ -78,6 +78,15 @@ class SQLParsingError(SQLSpecError):
         super().__init__(message)
 
 
+class SQLConversionError(SQLSpecError):
+    """Issues converting SQL statements."""
+
+    def __init__(self, message: Optional[str] = None) -> None:
+        if message is None:
+            message = "Issues converting SQL statement."
+        super().__init__(message)
+
+
 class ParameterStyleMismatchError(SQLSpecError):
     """Error when parameter style doesn't match SQL placeholder style.
 

sqlspec/extensions/litestar/config.py

@@ -1,10 +1,6 @@
 from dataclasses import dataclass, field
 from typing import TYPE_CHECKING, Callable, Literal, Optional, Union
 
-from sqlspec.base import (
-    ConnectionT,
-    PoolT,
-)
 from sqlspec.exceptions import ImproperConfigurationError
 from sqlspec.extensions.litestar.handlers import (
     autocommit_handler_maker,
@@ -23,13 +19,9 @@ if TYPE_CHECKING:
     from litestar.datastructures.state import State
     from litestar.types import BeforeMessageSendHookHandler, Scope
 
-    from sqlspec.base import (
-        AsyncConfigT,
-        ConnectionT,
-        DriverT,
-        PoolT,
-        SyncConfigT,
-    )
+    from sqlspec.base import AsyncConfigT, DriverT, SyncConfigT
+    from sqlspec.typing import ConnectionT, PoolT
+
 
 CommitMode = Literal["manual", "autocommit", "autocommit_include_redirect"]
 DEFAULT_COMMIT_MODE: CommitMode = "manual"

sqlspec/extensions/litestar/handlers.py

@@ -19,7 +19,8 @@ if TYPE_CHECKING:
     from litestar.datastructures.state import State
     from litestar.types import Message, Scope
 
-    from sqlspec.base import ConnectionT, DatabaseConfigProtocol, DriverT, PoolT
+    from sqlspec.base import DatabaseConfigProtocol, DriverT
+    from sqlspec.typing import ConnectionT, PoolT
 
 
 SESSION_TERMINUS_ASGI_EVENTS = {HTTP_RESPONSE_START, HTTP_DISCONNECT, WEBSOCKET_DISCONNECT, WEBSOCKET_CLOSE}

sqlspec/extensions/litestar/plugin.py

@@ -5,15 +5,14 @@ from litestar.plugins import InitPluginProtocol
 
 from sqlspec.base import (
     AsyncConfigT,
-    ConnectionT,
     DatabaseConfigProtocol,
     DriverT,
-    PoolT,
     SyncConfigT,
 )
 from sqlspec.base import SQLSpec as SQLSpecBase
 from sqlspec.exceptions import ImproperConfigurationError
 from sqlspec.extensions.litestar.config import DatabaseConfig
+from sqlspec.typing import ConnectionT, PoolT
 
 if TYPE_CHECKING:
     from click import Group
@@ -85,6 +84,9 @@ class SQLSpec(InitPluginProtocol, SQLSpecBase):
         )
         for c in self._plugin_configs:
             c.annotation = self.add_config(c.config)
+            app_config.signature_types.append(c.annotation)
+            app_config.signature_types.append(c.config.connection_type)  # type: ignore[union-attr]
+            app_config.signature_types.append(c.config.driver_type)  # type: ignore[union-attr]
             app_config.before_send.append(c.before_send_handler)
             app_config.lifespan.append(c.lifespan_handler)  # pyright: ignore[reportUnknownMemberType]
             app_config.dependencies.update(

sqlspec/mixins.py

@@ -0,0 +1,156 @@
+from abc import abstractmethod
+from typing import (
+    TYPE_CHECKING,
+    Any,
+    ClassVar,
+    Generic,
+    Optional,
+)
+
+from sqlglot import parse_one
+from sqlglot.dialects.dialect import DialectType
+
+from sqlspec.exceptions import SQLConversionError, SQLParsingError
+from sqlspec.typing import ConnectionT, StatementParameterType
+
+if TYPE_CHECKING:
+    from sqlspec.typing import ArrowTable
+
+__all__ = (
+    "AsyncArrowBulkOperationsMixin",
+    "AsyncParquetExportMixin",
+    "SQLTranslatorMixin",
+    "SyncArrowBulkOperationsMixin",
+    "SyncParquetExportMixin",
+)
+
+
+class SyncArrowBulkOperationsMixin(Generic[ConnectionT]):
+    """Mixin for sync drivers supporting bulk Apache Arrow operations."""
+
+    __supports_arrow__: "ClassVar[bool]" = True
+
+    @abstractmethod
+    def select_arrow(  # pyright: ignore[reportUnknownParameterType]
+        self,
+        sql: str,
+        parameters: "Optional[StatementParameterType]" = None,
+        /,
+        *,
+        connection: "Optional[ConnectionT]" = None,
+        **kwargs: Any,
+    ) -> "ArrowTable":  # pyright: ignore[reportUnknownReturnType]
+        """Execute a SQL query and return results as an Apache Arrow Table.
+
+        Args:
+            sql: The SQL query string.
+            parameters: Parameters for the query.
+            connection: Optional connection override.
+            **kwargs: Additional keyword arguments to merge with parameters if parameters is a dict.
+
+        Returns:
+            An Apache Arrow Table containing the query results.
+        """
+        raise NotImplementedError
+
+
+class AsyncArrowBulkOperationsMixin(Generic[ConnectionT]):
+    """Mixin for async drivers supporting bulk Apache Arrow operations."""
+
+    __supports_arrow__: "ClassVar[bool]" = True
+
+    @abstractmethod
+    async def select_arrow(  # pyright: ignore[reportUnknownParameterType]
+        self,
+        sql: str,
+        parameters: "Optional[StatementParameterType]" = None,
+        /,
+        *,
+        connection: "Optional[ConnectionT]" = None,
+        **kwargs: Any,
+    ) -> "ArrowTable":  # pyright: ignore[reportUnknownReturnType]
+        """Execute a SQL query and return results as an Apache Arrow Table.
+
+        Args:
+            sql: The SQL query string.
+            parameters: Parameters for the query.
+            connection: Optional connection override.
+            **kwargs: Additional keyword arguments to merge with parameters if parameters is a dict.
+
+        Returns:
+            An Apache Arrow Table containing the query results.
+        """
+        raise NotImplementedError
+
+
+class SyncParquetExportMixin(Generic[ConnectionT]):
+    """Mixin for sync drivers supporting Parquet export."""
+
+    @abstractmethod
+    def select_to_parquet(
+        self,
+        sql: str,
+        parameters: "Optional[StatementParameterType]" = None,
+        /,
+        *,
+        connection: "Optional[ConnectionT]" = None,
+        **kwargs: Any,
+    ) -> None:
+        """Export a SQL query to a Parquet file."""
+        raise NotImplementedError
+
+
+class AsyncParquetExportMixin(Generic[ConnectionT]):
+    """Mixin for async drivers supporting Parquet export."""
+
+    @abstractmethod
+    async def select_to_parquet(
+        self,
+        sql: str,
+        parameters: "Optional[StatementParameterType]" = None,
+        /,
+        *,
+        connection: "Optional[ConnectionT]" = None,
+        **kwargs: Any,
+    ) -> None:
+        """Export a SQL query to a Parquet file."""
+        raise NotImplementedError
+
+
+class SQLTranslatorMixin(Generic[ConnectionT]):
+    """Mixin for drivers supporting SQL translation."""
+
+    dialect: str
+
+    def convert_to_dialect(
+        self,
+        sql: str,
+        to_dialect: DialectType = None,
+        pretty: bool = True,
+    ) -> str:
+        """Convert a SQL query to a different dialect.
+
+        Args:
+            sql: The SQL query string to convert.
+            to_dialect: The target dialect to convert to.
+            pretty: Whether to pretty-print the SQL query.
+
+        Returns:
+            The converted SQL query string.
+
+        Raises:
+            SQLParsingError: If the SQL query cannot be parsed.
+            SQLConversionError: If the SQL query cannot be converted to the target dialect.
+        """
+        try:
+            parsed = parse_one(sql, dialect=self.dialect)
+        except Exception as e:
+            error_msg = f"Failed to parse SQL: {e!s}"
+            raise SQLParsingError(error_msg) from e
+        if to_dialect is None:
+            to_dialect = self.dialect
+        try:
+            return parsed.sql(dialect=to_dialect, pretty=pretty)
+        except Exception as e:
+            error_msg = f"Failed to convert SQL to {to_dialect}: {e!s}"
+            raise SQLConversionError(error_msg) from e

sqlspec/typing.py

@@ -33,8 +33,26 @@ PYDANTIC_USE_FAILFAST = False  # leave permanently disabled for now
 
 
 T = TypeVar("T")
+ConnectionT = TypeVar("ConnectionT")
+"""Type variable for connection types.
 
-ModelT = TypeVar("ModelT", bound="Union[Struct, BaseModel, DataclassProtocol]")
+:class:`~sqlspec.typing.ConnectionT`
+"""
+PoolT = TypeVar("PoolT")
+"""Type variable for pool types.
+
+:class:`~sqlspec.typing.PoolT`
+"""
+PoolT_co = TypeVar("PoolT_co", covariant=True)
+"""Type variable for covariant pool types.
+
+:class:`~sqlspec.typing.PoolT_co`
+"""
+ModelT = TypeVar("ModelT", bound="Union[dict[str, Any], Struct, BaseModel, DataclassProtocol]")
+"""Type variable for model types.
+
+:class:`dict[str, Any]` | :class:`msgspec.Struct` | :class:`pydantic.BaseModel` | :class:`DataclassProtocol`
+"""
 
 FilterTypeT = TypeVar("FilterTypeT", bound="StatementFilter")
 """Type variable for filter types.

{sqlspec-0.9.1.dist-info → sqlspec-0.10.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: sqlspec
-Version: 0.9.1
+Version: 0.10.0
 Summary: SQL Experiments in Python
 Author-email: Cody Fincher <cody@litestar.dev>
 Maintainer-email: Litestar Developers <hello@litestar.dev>
@@ -41,6 +41,7 @@ Requires-Dist: oracledb; extra == 'oracledb'
 Provides-Extra: orjson
 Requires-Dist: orjson; extra == 'orjson'
 Provides-Extra: performance
+Requires-Dist: msgspec; extra == 'performance'
 Requires-Dist: sqlglot[rs]; extra == 'performance'
 Provides-Extra: psqlpy
 Requires-Dist: psqlpy; extra == 'psqlpy'
@@ -131,7 +132,11 @@ etl_config = sql.add_config(
     )
 )
 with sql.provide_session(etl_config) as session:
-    result = session.select_one("SELECT open_prompt(?)", data.message, schema_type=ChatMessage)
+    result = session.select_one(
+        "SELECT open_prompt(?)",
+        "Can you write a haiku about DuckDB?",
+        schema_type=ChatMessage
+    )
     print(result) # result is a ChatMessage pydantic model
 ```
 
@@ -249,7 +254,7 @@ This list is not final. If you have a driver you'd like to see added, please ope
 | [`oracledb`](https://oracle.github.io/python-oracledb/)                                                      | Oracle     | Async   | ✅         |
 | [`oracledb`](https://oracle.github.io/python-oracledb/)                                                      | Oracle     | Sync    | ✅         |
 | [`duckdb`](https://duckdb.org/)                                                                               | DuckDB     | Sync    | ✅         |
-| [`bigquery`](https://googleapis.dev/python/bigquery/latest/index.html)                                        | BigQuery   | Sync    | 🗓️ |
+| [`bigquery`](https://googleapis.dev/python/bigquery/latest/index.html)                                        | BigQuery   | Sync    | ✅ |
 | [`spanner`](https://googleapis.dev/python/spanner/latest/index.html)                                         | Spanner    | Sync    | 🗓️  |
 | [`sqlserver`](https://docs.microsoft.com/en-us/sql/connect/python/pyodbc/python-sql-driver-for-pyodbc?view=sql-server-ver16) | SQL Server | Sync    | 🗓️  |
 | [`mysql`](https://dev.mysql.com/doc/connector-python/en/connector-python-api-mysql-connector-python.html)     | MySQL      | Sync    | 🗓️  |