Fix issue with incorrect body schema (#1299)

uriyyo · Sep 24, 2024 · 774165c · 774165c
1 parent 2959130
commit 774165c
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/fastapi_pagination/api.py b/fastapi_pagination/api.py
@@ -342,8 +342,12 @@ async def _page_ctx_dependency(
 def _bet_body_field(route: APIRoute) -> Optional[Any]:
     try:
         # starting from fastapi 0.113.0 get_body_field changed its signature
-        return get_body_field(flat_dependant=route.dependant, name=route.unique_id, embed_body_fields=True)
-    except TypeError:
+        return get_body_field(
+            flat_dependant=route.dependant,
+            name=route.unique_id,
+            embed_body_fields=route._embed_body_fields,
+        )
+    except (TypeError, AttributeError):
         return get_body_field(dependant=route.dependant, name=route.unique_id)