Edits.

Author: WaVEV

django_mongodb_backend/expressions/builtins.py

@@ -172,7 +172,7 @@ def ref(self, compiler, connection, as_path=False):  # noqa: ARG001
 
 @property
 def ref_is_simple_column(self):
-    return isinstance(self.source, Col) and self.source.alias is not None
+    return self.source.is_simple_column
 
 
 def star(self, compiler, connection, as_path=False):  # noqa: ARG001

django_mongodb_backend/fields/embedded_model_array.py

@@ -8,6 +8,7 @@
 from django.utils.functional import cached_property
 
 from .. import forms
+from ..lookups import builtin_lookup_path
 from ..query_utils import process_lhs, process_rhs
 from . import EmbeddedModelField
 from .array import ArrayField, ArrayLenTransform
@@ -141,6 +142,9 @@ def as_mql_expr(self, compiler, connection):
         )
         return {"$anyElementTrue": lhs_mql}
 
+    def as_mql_path(self, compiler, connection):
+        return builtin_lookup_path(self, compiler, connection)
+
 
 @_EmbeddedModelArrayOutputField.register_lookup
 class EmbeddedModelArrayFieldIn(EmbeddedModelArrayFieldBuiltinLookup, lookups.In):

tests/model_fields_/test_embedded_model.py

@@ -225,6 +225,15 @@ def test_nested(self):
         )
         self.assertCountEqual(Book.objects.filter(author__address__city="NYC"), [obj])
 
+    def test_annotate(self):
+        obj = Book.objects.create(
+            author=Author(name="Shakespeare", age=55, address=Address(city="NYC", state="NY"))
+        )
+        book_from_ny = (
+            Book.objects.annotate(city=F("author__address__city")).filter(city="NYC").first()
+        )
+        self.assertCountEqual(book_from_ny.city, obj.author.address.city)
+
 
 class ArrayFieldTests(TestCase):
     @classmethod