Edits

Author: WaVEV

django_mongodb_backend/indexes.py

@@ -127,7 +127,7 @@ def check(self, model, connection):
 
     def search_index_data_types(self, db_type):
         """
-        Map a model field's internal type to search index type.
+        Map a model field's type to search index type.
         Reference: https://www.mongodb.com/docs/atlas/atlas-search/define-field-mappings/#data-types
         """
         if db_type in {"double", "int", "long"}:
@@ -183,11 +183,11 @@ def check(self, model, connection):
         errors = super().check(model, connection)
         expected_similarities = 0
         for field_name, _ in self.fields_orders:
-            field_ = model._meta.get_field(field_name)
-            if isinstance(field_, ArrayField):
+            field = model._meta.get_field(field_name)
+            if isinstance(field, ArrayField):
                 expected_similarities += 1
                 try:
-                    int(field_.size)
+                    int(field.size)
                 except (ValueError, TypeError):
                     errors.append(
                         Error(
@@ -196,27 +196,27 @@ def check(self, model, connection):
                             id=f"{self._error_id_prefix}.E002",
                         )
                     )
-                if not isinstance(field_.base_field, FloatField | IntegerField):
+                if not isinstance(field.base_field, FloatField | IntegerField):
                     errors.append(
                         Error(
                             "VectorSearchIndex requires the base field of "
-                            f"ArrayField '{field_.name}' to be FloatField or "
+                            f"ArrayField '{field.name}' to be FloatField or "
                             "IntegerField but is "
-                            f"{field_.base_field.get_internal_type()}.",
+                            f"{field.base_field.get_internal_type()}.",
                             obj=model,
                             id=f"{self._error_id_prefix}.E003",
                         )
                     )
             else:
-                search_type = self.search_index_data_types(field_.db_type(connection))
+                search_type = self.search_index_data_types(field.db_type(connection))
                 # filter - for fields that contain boolean, date, objectId,
                 # numeric, string, or UUID values. Reference:
                 # https://www.mongodb.com/docs/atlas/atlas-vector-search/vector-search-type/#atlas-vector-search-index-fields
                 if search_type not in {"number", "string", "boolean", "objectId", "uuid", "date"}:
                     errors.append(
                         Error(
                             "VectorSearchIndex does not support "
-                            f"{field_.get_internal_type()} '{field_name}'.",
+                            f"{field.get_internal_type()} '{field_name}'.",
                             obj=model,
                             id=f"{self._error_id_prefix}.E004",
                         )