import Literal from typing or typing_extensions

Author: svlandeg

sqlmodel/_compat.py

@@ -1,3 +1,4 @@
+import sys
 import types
 from contextlib import contextmanager
 from contextvars import ContextVar
@@ -62,6 +63,12 @@ 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

@@ -52,12 +52,13 @@
 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 Literal, TypeAlias, deprecated, get_origin
+from typing_extensions import 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

@@ -1,9 +1,10 @@
-from typing import List, Literal, Optional
+from typing import List, Optional
 
 import pytest
 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
 
 
 def test_should_allow_duplicate_row_if_unique_constraint_is_not_passed(clear_sqlmodel):