remove pydantic v1 references

Author: sethbuff

aredis_om/__init__.py

@@ -16,3 +16,25 @@
     RedisModelError,
     VectorFieldOptions,
 )
+
+
+__all__ = [
+    "redis",
+    "get_redis_connection",
+    "Field",
+    "HashModel",
+    "JsonModel",
+    "EmbeddedJsonModel",
+    "RedisModel",
+    "FindQuery",
+    "KNNExpression",
+    "VectorFieldOptions",
+    "has_redis_json",
+    "has_redisearch",
+    "MigrationError",
+    "Migrator",
+    "RedisModelError",
+    "NotFoundError",
+    "QueryNotSupportedError",
+    "QuerySyntaxError",
+]

aredis_om/_compat.py

@@ -1 +1,4 @@
 from redis import asyncio as redis
+
+
+__all__ = ["redis"]

aredis_om/async_redis.py

@@ -1,5 +1,4 @@
 from functools import lru_cache
-from typing import List
 
 from aredis_om.connections import get_redis_connection
 

aredis_om/checks.py

@@ -31,8 +31,9 @@
 from types import GeneratorType
 from typing import Any, Callable, Dict, List, Optional, Set, Tuple, Union
 
-from .._compat import ENCODERS_BY_TYPE, BaseModel
-
+from pydantic import BaseModel
+from pydantic.deprecated.json import ENCODERS_BY_TYPE
+from pydantic_core import PydanticUndefined
 
 SetIntStr = Set[Union[int, str]]
 DictIntStrAny = Dict[Union[int, str], Any]
@@ -106,6 +107,7 @@ def jsonable_encoder(
                     or (not isinstance(key, str))
                     or (not key.startswith("_sa"))
                 )
+                and value is not PydanticUndefined
                 and (value is not None or not exclude_none)
                 and ((include and key in include) or not exclude or key not in exclude)
             ):

aredis_om/model/encoders.py

@@ -131,7 +131,7 @@ async def detect_migrations(self):
 
             stored_hash = await conn.get(hash_key)
             if isinstance(stored_hash, bytes):
-                stored_hash = stored_hash.decode('utf-8')
+                stored_hash = stored_hash.decode("utf-8")
 
             schema_out_of_date = current_hash != stored_hash