custom sel set result matches shape (#12333)

aws-amplify · Oct 17, 2023 · 8035dfc · 8035dfc
1 parent ce6e561
commit 8035dfc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/packages/api-graphql/src/internals/generateModelsProperty.ts b/packages/api-graphql/src/internals/generateModelsProperty.ts
@@ -129,7 +129,7 @@ export function generateModelsProperty<T extends Record<any, any> = never>(
 								const flattenedResult = flattenItems(data)[key];
 
 								if (options?.selectionSet) {
-									return flattenedResult;
+									return { data: flattenedResult, extensions };
 								} else {
 									// TODO: refactor to avoid destructuring here
 									const [initialized] = initializeModel(