Fixed imports to avoid pydantic if the version is >=2

event_model/documents/generate/type_wrapper.py

@@ -3,19 +3,43 @@
 """
 
 
+import pkgutil
 from dataclasses import dataclass
 
-try:
+Field = None
+FieldInfo = None
+BaseConfig = None
+BaseModel = None
+create_model = None
+
+pydantic_path = pkgutil.find_loader("pydantic")
+pydantic_version = None
+if pydantic_path:
     import pydantic
 
+    pydantic_version = pydantic.__version__
+
+if pydantic_version and pydantic_version < "2.0":
     Field = pydantic.Field
     FieldInfo = pydantic.fields.FieldInfo
-except ModuleNotFoundError:
+    BaseConfig = pydantic.BaseConfig
+    BaseModel = pydantic.BaseModel
+    create_model = pydantic.create_model
+else:
+
+    def Field(*args, **kwargs):
+        ...
+
+    class FieldInfo:
+        ...
+
+    class BaseConfig:
+        ...
 
-    def Field(*args, **kwargs):  # type: ignore
+    class BaseModel:
         ...
 
-    class FieldInfo:  # type: ignore
+    def create_model(*args, **kwargs):
         ...
 
 

event_model/documents/generate/typeddict_to_schema.py

@@ -7,8 +7,6 @@
 from pathlib import Path
 from typing import Dict, Optional, Tuple, Type, Union
 
-from pydantic import BaseConfig, BaseModel, Field, create_model
-from pydantic.fields import FieldInfo
 from typing_extensions import (
     Annotated,
     NotRequired,
@@ -33,13 +31,21 @@
 from event_model.documents.generate.type_wrapper import (
     ALLOWED_ANNOTATION_ELEMENTS,
     AsRef,
+    BaseConfig,
+    BaseModel,
+    Field,
+    FieldInfo,
+    create_model,
     extra_schema,
+    pydantic_version,
 )
 
 # The hacky indexing on types isn't possible with python < 3.9
 if sys.version_info[:2] < (3, 9):
     raise EnvironmentError("schema generation requires python 3.8 or higher")
 
+if pydantic_version and pydantic_version >= "2.0":
+    raise EnvironmentError(f"schema generation requires pydantic < 2.0 to run, current version {pydantic_version}")
 
 SCHEMA_OUT_DIR = Path("event_model") / SCHEMA_PATH
 

pyproject.toml

@@ -51,7 +51,7 @@ dev = [
     "numpydoc",
 
     # For schema generation.
-    "pydantic",
+    "pydantic<2.0",
 ]
 
 [project.scripts]