Use session in document insert

BeanieODM · Sep 5, 2024 · 280e92c · 280e92c
1 parent ca585b1
commit 280e92c
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/beanie/odm/documents.py b/beanie/odm/documents.py
@@ -346,15 +346,20 @@ async def insert(
                         LinkTypes.OPTIONAL_DIRECT,
                     ]:
                         if isinstance(value, Document):
-                            await value.save(link_rule=WriteRules.WRITE)
+                            await value.save(
+                                link_rule=WriteRules.WRITE, session=session
+                            )
                     if field_info.link_type in [
                         LinkTypes.LIST,
                         LinkTypes.OPTIONAL_LIST,
                     ]:
                         if isinstance(value, List):
                             await asyncio.gather(
                                 *[
-                                    obj.save(link_rule=WriteRules.WRITE)
+                                    obj.save(
+                                        link_rule=WriteRules.WRITE,
+                                        session=session,
+                                    )
                                     for obj in value
                                     if isinstance(obj, Document)
                                 ]