From 53b8d1038631079f188753a170e70c7c983181cf Mon Sep 17 00:00:00 2001
From: Daniel Chalef <131175+danielchalef@users.noreply.github.com>
Date: Thu, 5 Oct 2023 13:51:11 -0700
Subject: [PATCH] fix pydantic imports

---
 zep_python/user/models.py | 10 ++++++++--
 1 file changed, 8 insertions(+), 2 deletions(-)

diff --git a/zep_python/user/models.py b/zep_python/user/models.py
index 6e604e3..830dae0 100644
--- 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):