remove type check

aws-amplify · Oct 6, 2023 · 04a5224 · 04a5224
1 parent 8ec3c64
commit 04a5224
Showing 1 changed file with 7 additions and 9 deletions.
diff --git a/packages/api-graphql/src/internals/InternalGraphQLAPI.ts b/packages/api-graphql/src/internals/InternalGraphQLAPI.ts
@@ -230,14 +230,6 @@ export class InternalGraphQLAPIClass {
 
 		let customHeadersOptions;
 
-		// TODO: update type:
-		if (customHeaders && variables) {
-			customHeadersOptions = {
-				query: print(query as DocumentNode),
-				variables,
-			};
-		}
-
 		// TODO: Figure what we need to do to remove `!`'s.
 		const headers = {
 			...(!customEndpoint &&
@@ -254,7 +246,13 @@ export class InternalGraphQLAPIClass {
 					: {})) ||
 				{}),
 			// Custom headers included in Amplify configuration:
-			...(customHeaders && (await customHeaders(customHeadersOptions))),
+			...(customHeaders &&
+				(await customHeaders(
+					(customHeadersOptions = {
+						query: print(query as DocumentNode),
+						variables,
+					})
+				))),
 			// Headers from individual calls to `graphql`:
 			...additionalHeaders,
 			// User agent headers: