fix pydantic imports

getzep · Oct 5, 2023 · 53b8d10 · 53b8d10
1 parent 237cb88
commit 53b8d10
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/zep_python/user/models.py b/zep_python/user/models.py
@@ -1,8 +1,14 @@
 from datetime import datetime
-from typing import Any, Dict, Optional
+from typing import TYPE_CHECKING, Any, Dict, Optional
 from uuid import UUID
 
-from pydantic import BaseModel
+if TYPE_CHECKING:
+    from pydantic import BaseModel
+else:
+    try:
+        from pydantic.v1 import BaseModel
+    except ImportError:
+        from pydantic import BaseModel
 
 
 class User(BaseModel):