Update deprecated call of general_plain_validator_function (#897)

BeanieODM · May 1, 2024 · 10cb0c0 · 10cb0c0
1 parent 0e53aa7
commit 10cb0c0
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/beanie/odm/fields.py b/beanie/odm/fields.py
@@ -147,7 +147,7 @@ def __get_pydantic_core_schema__(
             cls, source_type: Any, handler: GetCoreSchemaHandler
         ) -> CoreSchema:  # type: ignore
             return core_schema.json_or_python_schema(
-                python_schema=core_schema.general_plain_validator_function(
+                python_schema=core_schema.with_info_plain_validator_function(
                     cls.validate
                 ),
                 json_schema=str_schema(),
@@ -397,7 +397,7 @@ def __get_pydantic_core_schema__(
             cls, source_type: Any, handler: GetCoreSchemaHandler
         ) -> CoreSchema:  # type: ignore
             return core_schema.json_or_python_schema(
-                python_schema=core_schema.general_plain_validator_function(
+                python_schema=core_schema.with_info_plain_validator_function(
                     cls.build_validation(handler, source_type)
                 ),
                 json_schema=core_schema.typed_dict_schema(
@@ -414,7 +414,7 @@ def __get_pydantic_core_schema__(
                     lambda instance: cls.serialize(instance)  # type: ignore
                 ),
             )
-            return core_schema.general_plain_validator_function(
+            return core_schema.with_info_plain_validator_function(
                 cls.build_validation(handler, source_type)
             )
 
@@ -474,7 +474,7 @@ def validate(v: Union[DBRef, T], field):
         def __get_pydantic_core_schema__(
             cls, source_type: Any, handler: GetCoreSchemaHandler
         ) -> CoreSchema:  # type: ignore
-            return core_schema.general_plain_validator_function(
+            return core_schema.with_info_plain_validator_function(
                 cls.build_validation(handler, source_type)
             )
 
@@ -581,7 +581,7 @@ def validate(v, _):
                 else:
                     return IndexModelField(IndexModel(v))
 
-            return core_schema.general_plain_validator_function(validate)
+            return core_schema.with_info_plain_validator_function(validate)
 
     else: