update

Author: ymc9

packages/schema/src/plugins/enhancer/policy/utils.ts

@@ -8,6 +8,7 @@ import {
     getAttributeArg,
     getAuthDecl,
     getDataModelAndTypeDefs,
+    getDataModels,
     getEntityCheckerFunctionName,
     getIdFields,
     getLiteral,
@@ -537,16 +538,19 @@ export function generateNormalizedAuthRef(
  * Check if the given enum is referenced in the model
  */
 export function isEnumReferenced(model: Model, decl: Enum): unknown {
-    return streamAllContents(model).some((node) => {
-        if (isDataModelField(node) && node.type.reference?.ref === decl) {
-            // referenced as field type
-            return true;
-        }
-        if (isEnumFieldReference(node) && node.target.ref?.$container === decl) {
-            // enum field is referenced
-            return true;
-        }
-        return false;
+    const dataModels = getDataModels(model);
+    return dataModels.some((dm) => {
+        return streamAllContents(dm).some((node) => {
+            if (isDataModelField(node) && node.type.reference?.ref === decl) {
+                // referenced as field type
+                return true;
+            }
+            if (isEnumFieldReference(node) && node.target.ref?.$container === decl) {
+                // enum field is referenced
+                return true;
+            }
+            return false;
+        });
     });
 }