Improve replace_with

Aggify · Nov 8, 2023 · 42e49c2 · 42e49c2
1 parent 691e037
commit 42e49c2
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/aggify/aggify.py b/aggify/aggify.py
@@ -752,6 +752,9 @@ def replace_with(
             new_root = {"$replaceWith": name}
         else:
             new_root = {"$replaceWith": {"$mergeObjects": [merge, name]}}
+            self.base_model._fields.update(  # noqa
+                {key: mongoengine_fields.IntField() for key, value in merge.items()}
+            )
         self.pipelines.append(new_root)
 
         return self