sqlspec 0.38.3__tar.gz → 0.38.4__tar.gz

{sqlspec-0.38.3 → sqlspec-0.38.4}/.gitignore

@@ -61,3 +61,5 @@ specs/
 .playwright-mcp
 .geminiignore
 uv.toml
+.beads/
+.agent/

{sqlspec-0.38.3 → sqlspec-0.38.4}/.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.14.13"
+    rev: "v0.14.14"
     hooks:
       - id: ruff
         args: ["--fix"]

{sqlspec-0.38.3 → sqlspec-0.38.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: sqlspec
-Version: 0.38.3
+Version: 0.38.4
 Summary: SQL Experiments in Python
 Project-URL: Discord, https://discord.gg/litestar
 Project-URL: Issue, https://github.com/litestar-org/sqlspec/issues/

{sqlspec-0.38.3 → sqlspec-0.38.4}/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.38.3"
+version = "0.38.4"
 
 [project.urls]
 Discord = "https://discord.gg/litestar"
@@ -240,7 +240,7 @@ opt_level = "3"   # Maximum optimization (0-3)
 allow_dirty = true
 commit = false
 commit_args = "--no-verify"
-current_version = "0.38.3"
+current_version = "0.38.4"
 ignore_missing_files = false
 ignore_missing_version = false
 message = "chore(release): bump to v{new_version}"

{sqlspec-0.38.3 → sqlspec-0.38.4}/sqlspec/_typing.py

@@ -7,7 +7,7 @@ from dataclasses import dataclass
 from enum import Enum
 from typing import Any, ClassVar, Final, Literal, Protocol, cast, runtime_checkable
 
-from typing_extensions import TypeVar, dataclass_transform
+from typing_extensions import Self, TypeVar, dataclass_transform
 
 from sqlspec.utils.module_loader import dependency_flag, module_available
 
@@ -502,8 +502,8 @@ except ImportError:
         def end(self, end_time: "int | None" = None) -> None:
             return None
 
-        def __enter__(self) -> "Span":
-            return self  # type: ignore[return-value]
+        def __enter__(self) -> Self:
+            return self
 
         def __exit__(self, exc_type: object, exc_val: object, exc_tb: object) -> None:
             return None

{sqlspec-0.38.3 → sqlspec-0.38.4}/sqlspec/adapters/adbc/driver.py

@@ -7,6 +7,8 @@ database dialects, parameter style conversion, and transaction management.
 import contextlib
 from typing import TYPE_CHECKING, Any, Literal, cast
 
+from typing_extensions import Self
+
 from sqlspec.adapters.adbc._typing import AdbcSessionContext
 from sqlspec.adapters.adbc.core import (
     collect_rows,
@@ -83,7 +85,7 @@ class AdbcExceptionHandler:
     def __init__(self) -> None:
         self.pending_exception: Exception | None = None
 
-    def __enter__(self) -> "AdbcExceptionHandler":
+    def __enter__(self) -> Self:
         return self
 
     def __exit__(self, exc_type: Any, exc_val: Any, exc_tb: Any) -> bool:

{sqlspec-0.38.3 → sqlspec-0.38.4}/sqlspec/adapters/aiosqlite/driver.py

@@ -30,6 +30,8 @@ if TYPE_CHECKING:
     from sqlspec.driver import ExecutionResult
     from sqlspec.storage import StorageBridgeJob, StorageDestination, StorageFormat, StorageTelemetry
 
+from typing_extensions import Self
+
 from sqlspec.adapters.aiosqlite._typing import AiosqliteSessionContext
 
 __all__ = ("AiosqliteCursor", "AiosqliteDriver", "AiosqliteExceptionHandler", "AiosqliteSessionContext")
@@ -81,7 +83,7 @@ class AiosqliteExceptionHandler:
     def __init__(self) -> None:
         self.pending_exception: Exception | None = None
 
-    async def __aenter__(self) -> "AiosqliteExceptionHandler":
+    async def __aenter__(self) -> Self:
         return self
 
     async def __aexit__(self, exc_type: Any, exc_val: Any, exc_tb: Any) -> bool:

{sqlspec-0.38.3 → sqlspec-0.38.4}/sqlspec/adapters/asyncmy/driver.py

@@ -39,6 +39,8 @@ if TYPE_CHECKING:
     from sqlspec.driver import ExecutionResult
     from sqlspec.storage import StorageBridgeJob, StorageDestination, StorageFormat, StorageTelemetry
 
+from typing_extensions import Self
+
 from sqlspec.adapters.asyncmy._typing import AsyncmySessionContext
 
 __all__ = ("AsyncmyCursor", "AsyncmyDriver", "AsyncmyExceptionHandler", "AsyncmySessionContext")
@@ -88,7 +90,7 @@ class AsyncmyExceptionHandler:
     def __init__(self) -> None:
         self.pending_exception: Exception | None = None
 
-    async def __aenter__(self) -> "AsyncmyExceptionHandler":
+    async def __aenter__(self) -> Self:
         return self
 
     async def __aexit__(self, exc_type: Any, exc_val: Any, exc_tb: Any) -> bool:

{sqlspec-0.38.3 → sqlspec-0.38.4}/sqlspec/adapters/asyncpg/driver.py

@@ -40,6 +40,8 @@ if TYPE_CHECKING:
     from sqlspec.driver import ExecutionResult
     from sqlspec.storage import StorageBridgeJob, StorageDestination, StorageFormat, StorageTelemetry
 
+from typing_extensions import Self
+
 from sqlspec.adapters.asyncpg._typing import AsyncpgSessionContext
 
 __all__ = ("AsyncpgCursor", "AsyncpgDriver", "AsyncpgExceptionHandler", "AsyncpgSessionContext")
@@ -77,7 +79,7 @@ class AsyncpgExceptionHandler:
     def __init__(self) -> None:
         self.pending_exception: Exception | None = None
 
-    async def __aenter__(self) -> "AsyncpgExceptionHandler":
+    async def __aenter__(self) -> Self:
         return self
 
     async def __aexit__(self, exc_type: Any, exc_val: Any, exc_tb: Any) -> bool:

{sqlspec-0.38.3 → sqlspec-0.38.4}/sqlspec/adapters/bigquery/driver.py

@@ -106,7 +106,7 @@ class BigQueryExceptionHandler:
     def __init__(self) -> None:
         self.pending_exception: Exception | None = None
 
-    def __enter__(self) -> "BigQueryExceptionHandler":
+    def __enter__(self) -> "Self":
         return self
 
     def __exit__(self, exc_type: Any, exc_val: Any, exc_tb: Any) -> bool:

{sqlspec-0.38.3 → sqlspec-0.38.4}/sqlspec/adapters/cockroach_asyncpg/driver.py

@@ -5,6 +5,7 @@ import contextlib
 from typing import TYPE_CHECKING, Any, cast
 
 import asyncpg
+from typing_extensions import Self
 
 from sqlspec.adapters.asyncpg.core import create_mapped_exception, driver_profile
 from sqlspec.adapters.asyncpg.driver import AsyncpgDriver
@@ -40,7 +41,7 @@ class CockroachAsyncpgExceptionHandler:
     def __init__(self) -> None:
         self.pending_exception: Exception | None = None
 
-    async def __aenter__(self) -> "CockroachAsyncpgExceptionHandler":
+    async def __aenter__(self) -> Self:
         return self
 
     async def __aexit__(self, exc_type: Any, exc_val: Any, exc_tb: Any) -> bool:

{sqlspec-0.38.3 → sqlspec-0.38.4}/sqlspec/adapters/cockroach_psycopg/driver.py

@@ -6,6 +6,7 @@ import time
 from typing import TYPE_CHECKING, Any, cast
 
 import psycopg
+from typing_extensions import Self
 
 from sqlspec.adapters.cockroach_psycopg._typing import (
     CockroachAsyncConnection,
@@ -57,7 +58,7 @@ class CockroachPsycopgSyncExceptionHandler:
     def __init__(self) -> None:
         self.pending_exception: Exception | None = None
 
-    def __enter__(self) -> "CockroachPsycopgSyncExceptionHandler":
+    def __enter__(self) -> Self:
         return self
 
     def __exit__(self, exc_type: Any, exc_val: Any, exc_tb: Any) -> bool:
@@ -80,7 +81,7 @@ class CockroachPsycopgAsyncExceptionHandler:
     def __init__(self) -> None:
         self.pending_exception: Exception | None = None
 
-    async def __aenter__(self) -> "CockroachPsycopgAsyncExceptionHandler":
+    async def __aenter__(self) -> Self:
         return self
 
     async def __aexit__(self, exc_type: Any, exc_val: Any, exc_tb: Any) -> bool:

{sqlspec-0.38.3 → sqlspec-0.38.4}/sqlspec/adapters/duckdb/driver.py

@@ -31,6 +31,8 @@ if TYPE_CHECKING:
     from sqlspec.storage import StorageBridgeJob, StorageDestination, StorageFormat, StorageTelemetry
     from sqlspec.typing import ArrowReturnFormat, StatementParameters
 
+from typing_extensions import Self
+
 from sqlspec.adapters.duckdb._typing import DuckDBSessionContext
 
 __all__ = ("DuckDBCursor", "DuckDBDriver", "DuckDBExceptionHandler", "DuckDBSessionContext")
@@ -78,7 +80,7 @@ class DuckDBExceptionHandler:
     def __init__(self) -> None:
         self.pending_exception: Exception | None = None
 
-    def __enter__(self) -> "DuckDBExceptionHandler":
+    def __enter__(self) -> Self:
         return self
 
     def __exit__(self, exc_type: Any, exc_val: Any, exc_tb: Any) -> bool:

{sqlspec-0.38.3 → sqlspec-0.38.4}/sqlspec/adapters/mock/driver.py

@@ -10,6 +10,8 @@ import contextlib
 import sqlite3
 from typing import TYPE_CHECKING, Any
 
+from typing_extensions import Self
+
 from sqlspec.adapters.mock._typing import MockAsyncSessionContext, MockSyncSessionContext
 from sqlspec.adapters.mock.core import (
     build_insert_statement,
@@ -125,7 +127,7 @@ class MockExceptionHandler:
     def __init__(self) -> None:
         self.pending_exception: Exception | None = None
 
-    def __enter__(self) -> "MockExceptionHandler":
+    def __enter__(self) -> Self:
         return self
 
     def __exit__(self, exc_type: Any, exc_val: Any, exc_tb: Any) -> bool:
@@ -148,7 +150,7 @@ class MockAsyncExceptionHandler:
     def __init__(self) -> None:
         self.pending_exception: Exception | None = None
 
-    async def __aenter__(self) -> "MockAsyncExceptionHandler":
+    async def __aenter__(self) -> Self:
         return self
 
     async def __aexit__(self, exc_type: Any, exc_val: Any, exc_tb: Any) -> bool:

{sqlspec-0.38.3 → sqlspec-0.38.4}/sqlspec/adapters/mysqlconnector/driver.py

@@ -41,6 +41,8 @@ if TYPE_CHECKING:
     from sqlspec.driver import ExecutionResult
     from sqlspec.storage import StorageBridgeJob, StorageDestination, StorageFormat, StorageTelemetry
 
+from typing_extensions import Self
+
 from sqlspec.adapters.mysqlconnector._typing import MysqlConnectorAsyncSessionContext, MysqlConnectorSyncSessionContext
 
 __all__ = (
@@ -86,7 +88,7 @@ class MysqlConnectorSyncExceptionHandler:
     def __init__(self) -> None:
         self.pending_exception: Exception | None = None
 
-    def __enter__(self) -> "MysqlConnectorSyncExceptionHandler":
+    def __enter__(self) -> Self:
         return self
 
     def __exit__(self, exc_type: Any, exc_val: Any, exc_tb: Any) -> bool:
@@ -301,7 +303,7 @@ class MysqlConnectorAsyncExceptionHandler:
     def __init__(self) -> None:
         self.pending_exception: Exception | None = None
 
-    async def __aenter__(self) -> "MysqlConnectorAsyncExceptionHandler":
+    async def __aenter__(self) -> Self:
         return self
 
     async def __aexit__(self, exc_type: Any, exc_val: Any, exc_tb: Any) -> bool:

{sqlspec-0.38.3 → sqlspec-0.38.4}/sqlspec/adapters/oracledb/driver.py

@@ -6,6 +6,7 @@ from typing import TYPE_CHECKING, Any, NamedTuple, cast
 
 import oracledb
 from oracledb import AsyncCursor, Cursor
+from typing_extensions import Self
 
 from sqlspec.adapters.oracledb._typing import (
     OracleAsyncConnection,
@@ -278,7 +279,7 @@ class OracleSyncExceptionHandler:
     def __init__(self) -> None:
         self.pending_exception: Exception | None = None
 
-    def __enter__(self) -> "OracleSyncExceptionHandler":
+    def __enter__(self) -> Self:
         return self
 
     def __exit__(self, exc_type: Any, exc_val: Any, exc_tb: Any) -> bool:
@@ -307,7 +308,7 @@ class OracleAsyncExceptionHandler:
     def __init__(self) -> None:
         self.pending_exception: Exception | None = None
 
-    async def __aenter__(self) -> "OracleAsyncExceptionHandler":
+    async def __aenter__(self) -> Self:
         return self
 
     async def __aexit__(self, exc_type: Any, exc_val: Any, exc_tb: Any) -> bool:

{sqlspec-0.38.3 → sqlspec-0.38.4}/sqlspec/adapters/psqlpy/driver.py

@@ -8,6 +8,7 @@ import inspect
 from typing import TYPE_CHECKING, Any, cast
 
 import psqlpy.exceptions
+from typing_extensions import Self
 
 from sqlspec.adapters.psqlpy._typing import PsqlpySessionContext
 from sqlspec.adapters.psqlpy.core import (
@@ -101,7 +102,7 @@ class PsqlpyExceptionHandler:
     def __init__(self) -> None:
         self.pending_exception: Exception | None = None
 
-    async def __aenter__(self) -> "PsqlpyExceptionHandler":
+    async def __aenter__(self) -> Self:
         return self
 
     async def __aexit__(self, exc_type: Any, exc_val: Any, exc_tb: Any) -> bool:

{sqlspec-0.38.3 → sqlspec-0.38.4}/sqlspec/adapters/psycopg/driver.py

@@ -4,6 +4,7 @@ from contextlib import AsyncExitStack, ExitStack
 from typing import TYPE_CHECKING, Any, cast
 
 import psycopg
+from typing_extensions import Self
 
 from sqlspec.adapters.psycopg._typing import (
     PsycopgAsyncConnection,
@@ -146,7 +147,7 @@ class PsycopgSyncExceptionHandler:
     def __init__(self) -> None:
         self.pending_exception: Exception | None = None
 
-    def __enter__(self) -> "PsycopgSyncExceptionHandler":
+    def __enter__(self) -> Self:
         return self
 
     def __exit__(self, exc_type: Any, exc_val: Any, exc_tb: Any) -> bool:
@@ -586,7 +587,7 @@ class PsycopgAsyncExceptionHandler:
     def __init__(self) -> None:
         self.pending_exception: Exception | None = None
 
-    async def __aenter__(self) -> "PsycopgAsyncExceptionHandler":
+    async def __aenter__(self) -> Self:
         return self
 
     async def __aexit__(self, exc_type: Any, exc_val: Any, exc_tb: Any) -> bool:

{sqlspec-0.38.3 → sqlspec-0.38.4}/sqlspec/adapters/pymysql/driver.py

@@ -34,6 +34,8 @@ if TYPE_CHECKING:
     from sqlspec.driver import ExecutionResult
     from sqlspec.storage import StorageBridgeJob, StorageDestination, StorageFormat, StorageTelemetry
 
+from typing_extensions import Self
+
 from sqlspec.adapters.pymysql._typing import PyMysqlSessionContext
 
 __all__ = ("PyMysqlCursor", "PyMysqlDriver", "PyMysqlExceptionHandler", "PyMysqlSessionContext")
@@ -70,7 +72,7 @@ class PyMysqlExceptionHandler:
     def __init__(self) -> None:
         self.pending_exception: Exception | None = None
 
-    def __enter__(self) -> "PyMysqlExceptionHandler":
+    def __enter__(self) -> Self:
         return self
 
     def __exit__(self, exc_type: Any, exc_val: Any, exc_tb: Any) -> bool:

{sqlspec-0.38.3 → sqlspec-0.38.4}/sqlspec/adapters/spanner/driver.py

@@ -5,6 +5,7 @@ from typing import TYPE_CHECKING, Any, Protocol, cast
 
 from google.api_core import exceptions as api_exceptions
 from google.cloud.spanner_v1.transaction import Transaction
+from typing_extensions import Self
 
 from sqlspec.adapters.spanner._typing import SpannerSessionContext
 from sqlspec.adapters.spanner.core import (
@@ -86,7 +87,7 @@ class SpannerExceptionHandler:
     def __init__(self) -> None:
         self.pending_exception: Exception | None = None
 
-    def __enter__(self) -> "SpannerExceptionHandler":
+    def __enter__(self) -> Self:
         return self
 
     def __exit__(self, exc_type: Any, exc_val: Any, exc_tb: Any) -> bool:

{sqlspec-0.38.3 → sqlspec-0.38.4}/sqlspec/adapters/sqlite/driver.py

@@ -4,6 +4,8 @@ import contextlib
 import sqlite3
 from typing import TYPE_CHECKING, Any
 
+from typing_extensions import Self
+
 from sqlspec.adapters.sqlite._typing import SqliteSessionContext
 from sqlspec.adapters.sqlite.core import (
     build_insert_statement,
@@ -85,7 +87,7 @@ class SqliteExceptionHandler:
     def __init__(self) -> None:
         self.pending_exception: Exception | None = None
 
-    def __enter__(self) -> "SqliteExceptionHandler":
+    def __enter__(self) -> Self:
         return self
 
     def __exit__(self, exc_type: Any, exc_val: Any, exc_tb: Any) -> bool:

{sqlspec-0.38.3 → sqlspec-0.38.4}/sqlspec/base.py

@@ -4,6 +4,8 @@ from collections.abc import Awaitable, Coroutine
 from contextlib import AbstractAsyncContextManager, AbstractContextManager
 from typing import TYPE_CHECKING, Any, TypeGuard, cast, overload
 
+from typing_extensions import Self
+
 from sqlspec.config import (
     AsyncConfigT,
     AsyncDatabaseConfig,
@@ -249,7 +251,7 @@ class SQLSpec:
             return f"{config.__class__.__name__}: {exc}"
         return None
 
-    async def __aenter__(self) -> "SQLSpec":
+    async def __aenter__(self) -> Self:
         """Async context manager entry."""
         return self
 

{sqlspec-0.38.3 → sqlspec-0.38.4}/sqlspec/config.py

@@ -1,3 +1,5 @@
+import asyncio
+import threading
 from abc import ABC, abstractmethod
 from collections.abc import Callable
 from inspect import Signature, signature
@@ -1510,7 +1512,7 @@ class NoPoolAsyncConfig(DatabaseConfigProtocol[ConnectionT, None, DriverT]):
 class SyncDatabaseConfig(DatabaseConfigProtocol[ConnectionT, PoolT, DriverT]):
     """Base class for sync database configurations with connection pooling."""
 
-    __slots__ = ("connection_config",)
+    __slots__ = ("_pool_lock", "connection_config")
     is_async: "ClassVar[bool]" = False
     supports_connection_pooling: "ClassVar[bool]" = True
     migration_tracker_type: "ClassVar[type[Any]]" = SyncMigrationTracker
@@ -1549,6 +1551,7 @@ class SyncDatabaseConfig(DatabaseConfigProtocol[ConnectionT, PoolT, DriverT]):
         self.driver_features.setdefault("storage_capabilities", self.storage_capabilities())
         self._promote_driver_feature_hooks()
         self._configure_observability_extensions()
+        self._pool_lock = threading.Lock()
 
     def create_pool(self) -> PoolT:
         """Create and return the connection pool.
@@ -1558,9 +1561,14 @@ class SyncDatabaseConfig(DatabaseConfigProtocol[ConnectionT, PoolT, DriverT]):
         """
         if self.connection_instance is not None:
             return self.connection_instance
-        self.connection_instance = self._create_pool()
-        self.get_observability_runtime().emit_pool_create(self.connection_instance)
-        return self.connection_instance
+
+        with self._pool_lock:
+            if self.connection_instance is not None:
+                return self.connection_instance
+
+            self.connection_instance = self._create_pool()
+            self.get_observability_runtime().emit_pool_create(self.connection_instance)
+            return self.connection_instance
 
     def close_pool(self) -> None:
         """Close the connection pool."""
@@ -1572,9 +1580,7 @@ class SyncDatabaseConfig(DatabaseConfigProtocol[ConnectionT, PoolT, DriverT]):
 
     def provide_pool(self, *args: Any, **kwargs: Any) -> PoolT:
         """Provide pool instance."""
-        if self.connection_instance is None:
-            self.connection_instance = self.create_pool()
-        return self.connection_instance
+        return self.create_pool()
 
     def create_connection(self) -> ConnectionT:
         """Create a database connection."""
@@ -1709,7 +1715,7 @@ class SyncDatabaseConfig(DatabaseConfigProtocol[ConnectionT, PoolT, DriverT]):
 class AsyncDatabaseConfig(DatabaseConfigProtocol[ConnectionT, PoolT, DriverT]):
     """Base class for async database configurations with connection pooling."""
 
-    __slots__ = ("connection_config",)
+    __slots__ = ("_pool_lock", "connection_config")
     is_async: "ClassVar[bool]" = True
     supports_connection_pooling: "ClassVar[bool]" = True
     migration_tracker_type: "ClassVar[type[Any]]" = AsyncMigrationTracker
@@ -1750,6 +1756,7 @@ class AsyncDatabaseConfig(DatabaseConfigProtocol[ConnectionT, PoolT, DriverT]):
         self.driver_features.setdefault("storage_capabilities", self.storage_capabilities())
         self._promote_driver_feature_hooks()
         self._configure_observability_extensions()
+        self._pool_lock = asyncio.Lock()
 
     async def create_pool(self) -> PoolT:
         """Create and return the connection pool.
@@ -1759,9 +1766,14 @@ class AsyncDatabaseConfig(DatabaseConfigProtocol[ConnectionT, PoolT, DriverT]):
         """
         if self.connection_instance is not None:
             return self.connection_instance
-        self.connection_instance = await self._create_pool()
-        self.get_observability_runtime().emit_pool_create(self.connection_instance)
-        return self.connection_instance
+
+        async with self._pool_lock:
+            if self.connection_instance is not None:
+                return self.connection_instance
+
+            self.connection_instance = await self._create_pool()
+            self.get_observability_runtime().emit_pool_create(self.connection_instance)
+            return self.connection_instance
 
     async def close_pool(self) -> None:
         """Close the connection pool."""
@@ -1773,9 +1785,7 @@ class AsyncDatabaseConfig(DatabaseConfigProtocol[ConnectionT, PoolT, DriverT]):
 
     async def provide_pool(self, *args: Any, **kwargs: Any) -> PoolT:
         """Provide pool instance."""
-        if self.connection_instance is None:
-            self.connection_instance = await self.create_pool()
-        return self.connection_instance
+        return await self.create_pool()
 
     async def create_connection(self) -> ConnectionT:
         """Create a database connection."""

{sqlspec-0.38.3 → sqlspec-0.38.4}/sqlspec/driver/_common.py

@@ -10,6 +10,7 @@ from typing import TYPE_CHECKING, Any, ClassVar, Final, Literal, NamedTuple, NoR
 
 from mypy_extensions import mypyc_attr
 from sqlglot import exp
+from typing_extensions import Self
 
 from sqlspec.builder import QueryBuilder
 from sqlspec.core import (
@@ -188,7 +189,7 @@ class SyncExceptionHandler(Protocol):
 
     pending_exception: Exception | None
 
-    def __enter__(self) -> "SyncExceptionHandler": ...
+    def __enter__(self) -> Self: ...
 
     def __exit__(self, exc_type: Any, exc_val: Any, exc_tb: Any) -> bool: ...
 
@@ -203,7 +204,7 @@ class AsyncExceptionHandler(Protocol):
 
     pending_exception: Exception | None
 
-    async def __aenter__(self) -> "AsyncExceptionHandler": ...
+    async def __aenter__(self) -> Self: ...
 
     async def __aexit__(self, exc_type: Any, exc_val: Any, exc_tb: Any) -> bool: ...
 
@@ -388,7 +389,7 @@ class StackExecutionObserver:
         self.span: Any | None = None
         self.started = 0.0
 
-    def __enter__(self) -> "StackExecutionObserver":
+    def __enter__(self) -> Self:
         self.started = perf_counter()
         trace_id, span_id = get_trace_context()
         attributes = {

{sqlspec-0.38.3 → sqlspec-0.38.4}/sqlspec/extensions/litestar/store.py

@@ -12,6 +12,8 @@ from sqlspec.utils.type_guards import has_extension_config
 if TYPE_CHECKING:
     from types import TracebackType
 
+    from typing_extensions import Self
+
 
 ConfigT = TypeVar("ConfigT")
 
@@ -203,7 +205,7 @@ class BaseSQLSpecStore(ABC, Generic[ConfigT]):
         """
         raise NotImplementedError
 
-    async def __aenter__(self) -> "BaseSQLSpecStore":
+    async def __aenter__(self) -> "Self":
         """Enter context manager."""
         return self