Add DocumentDB compatibility to fetch_links (#1042)

BeanieODM · Oct 8, 2024 · ac1da37 · ac1da37
1 parent 1a897a8
commit ac1da37
Showing 1 changed file with 8 additions and 8 deletions.
diff --git a/beanie/odm/utils/find.py b/beanie/odm/utils/find.py
@@ -66,7 +66,7 @@ def construct_query(
                     }
                 },
                 {
-                    "$set": {
+                    "$addFields": {
                         link_info.field_name: {
                             "$cond": {
                                 "if": {
@@ -81,7 +81,7 @@ def construct_query(
                         }
                     }
                 },
-                {"$unset": f"_link_{link_info.field_name}"},
+                {"$project": {f"_link_{link_info.field_name}": 0}},
             ]  # type: ignore
             new_depth = (
                 current_depth - 1 if current_depth is not None else None
@@ -122,7 +122,7 @@ def construct_query(
                     }
                 },
                 {
-                    "$set": {
+                    "$addFields": {
                         link_info.field_name: {
                             "$cond": {
                                 "if": {
@@ -137,7 +137,7 @@ def construct_query(
                         }
                     }
                 },
-                {"$unset": f"_link_{link_info.field_name}"},
+                {"$project": {f"_link_{link_info.field_name}": 0}},
             ]
             new_depth = (
                 current_depth - 1 if current_depth is not None else None
@@ -172,7 +172,7 @@ def construct_query(
                     }
                 },
                 {
-                    "$set": {
+                    "$addFields": {
                         link_info.field_name: {
                             "$cond": {
                                 "if": {
@@ -187,7 +187,7 @@ def construct_query(
                         }
                     }
                 },
-                {"$unset": f"_link_{link_info.field_name}"},
+                {"$project": {f"_link_{link_info.field_name}": 0}},
             ]  # type: ignore
             new_depth = (
                 current_depth - 1 if current_depth is not None else None
@@ -231,7 +231,7 @@ def construct_query(
                     }
                 },
                 {
-                    "$set": {
+                    "$addFields": {
                         link_info.field_name: {
                             "$cond": {
                                 "if": {
@@ -246,7 +246,7 @@ def construct_query(
                         }
                     }
                 },
-                {"$unset": f"_link_{link_info.field_name}"},
+                {"$project": {f"_link_{link_info.field_name}": 0}},
             ]
             new_depth = (
                 current_depth - 1 if current_depth is not None else None