Update _utils.py

awtkns · Sep 13, 2023 · 6fbf927 · 6fbf927
1 parent 74aa558
commit 6fbf927
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/fastapi_crudrouter/core/_utils.py b/fastapi_crudrouter/core/_utils.py
@@ -14,7 +14,10 @@ def __init__(self, *args, **kwargs) -> None:  # type: ignore
 
 def get_pk_type(schema: Type[PYDANTIC_SCHEMA], pk_field: str) -> Any:
     try:
-        return schema.__fields__[pk_field].type_
+        if int(pydantic_version.split(".")[0]) >= 2:
+            return schema.model_fields[pk_field].annotation
+        else:
+            return schema.__fields__[pk_field].type_
     except KeyError:
         return int