revert auth change

aws-amplify · Sep 3, 2024 · bf0e749 · bf0e749
1 parent e604c2c
commit bf0e749
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/packages/amplify-graphql-auth-transformer/src/graphql-auth-transformer.ts b/packages/amplify-graphql-auth-transformer/src/graphql-auth-transformer.ts
@@ -1532,7 +1532,7 @@ export class AuthTransformer extends TransformerAuthBase implements TransformerA
     allowedFields: Set<string>,
     fields: readonly string[],
   ): void => {
-    const typeDefinitions = ctx.output.getTypeDefinitionsOfKind('ObjectTypeDefinition') as ObjectTypeDefinitionNode[];
+    const typeDefinitions = ctx.inputDocument.definitions.filter((it) => it.kind === 'ObjectTypeDefinition') as ObjectTypeDefinitionNode[];
 
     this.addAutoGeneratedHasManyFields(ctx, typeDefinitions, def, allowedFields);
     this.addAutoGeneratedHasOneFields(ctx, typeDefinitions, fields, def, allowedFields);