Fix generate_self_schema for Python 3.12+ (#1299)

pydantic · May 22, 2024 · a762041 · a762041
1 parent 0dcf82a
commit a762041
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/generate_self_schema.py b/generate_self_schema.py
@@ -190,7 +190,12 @@ def all_literal_values(type_: type[core_schema.Literal]) -> list[any]:
 
 def eval_forward_ref(type_: Any) -> Any:
     try:
-        return type_._evaluate(core_schema.__dict__, None, set())
+        try:
+            # Python 3.12+
+            return type_._evaluate(core_schema.__dict__, None, type_params=set(), recursive_guard=set())
+        except TypeError:
+            # Python 3.9+
+            return type_._evaluate(core_schema.__dict__, None, set())
     except TypeError:
         # for Python 3.8
         return type_._evaluate(core_schema.__dict__, None)