simply import Literal from typing_extensions always

Author: svlandeg

pyproject.toml

@@ -109,7 +109,6 @@ disallow_untyped_defs = false
 disallow_untyped_calls = false
 
 [tool.ruff.lint]
-typing-modules = ["sqlmodel._compat"]
 select = [
     "E",  # pycodestyle errors
     "W",  # pycodestyle warnings

sqlmodel/_compat.py

@@ -1,4 +1,3 @@
-import sys
 import types
 from contextlib import contextmanager
 from contextvars import ContextVar
@@ -22,7 +21,7 @@
 from pydantic import VERSION as P_VERSION
 from pydantic import BaseModel
 from pydantic.fields import FieldInfo
-from typing_extensions import Annotated, get_args, get_origin
+from typing_extensions import Annotated, Literal, get_args, get_origin
 
 from .sql.sqltypes import AutoString
 
@@ -65,12 +64,6 @@ def _is_union_type(t: Any) -> bool:
     return t is UnionType or t is Union
 
 
-if sys.version_info >= (3, 9):
-    from typing import Literal
-else:
-    from typing_extensions import Literal
-
-
 finish_init: ContextVar[bool] = ContextVar("finish_init", default=True)
 
 

sqlmodel/main.py

@@ -53,13 +53,12 @@
 from sqlalchemy.orm.instrumentation import is_instrumented
 from sqlalchemy.sql.schema import MetaData
 from sqlalchemy.sql.sqltypes import LargeBinary, Time, Uuid
-from typing_extensions import TypeAlias, deprecated, get_origin
+from typing_extensions import Literal, TypeAlias, deprecated, get_origin
 
 from ._compat import (  # type: ignore[attr-defined]
     IS_PYDANTIC_V2,
     PYDANTIC_MINOR_VERSION,
     BaseConfig,
-    Literal,
     ModelField,
     ModelMetaclass,
     Representation,

tests/test_main.py

@@ -4,7 +4,7 @@
 from sqlalchemy.exc import IntegrityError
 from sqlalchemy.orm import RelationshipProperty
 from sqlmodel import Field, Relationship, Session, SQLModel, create_engine, select
-from sqlmodel._compat import Literal
+from typing_extensions import Literal
 
 
 def test_should_allow_duplicate_row_if_unique_constraint_is_not_passed(clear_sqlmodel):