@aws-amplify/predictions 6.0.22-unstable.bd1fefd.0 → 6.0.22

package/src/providers/AmazonAIConvertPredictionsProvider.ts

@@ -1,7 +1,5 @@
 // Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
 // SPDX-License-Identifier: Apache-2.0
-import { Buffer } from 'buffer';
-
 import { Amplify, ConsoleLogger, fetchAuthSession } from '@aws-amplify/core';
 import {
 	AWSCredentials,
@@ -20,7 +18,7 @@ import {
 	MessageHeaderValue,
 } from '@smithy/eventstream-codec';
 import { fromUtf8, toUtf8 } from '@smithy/util-utf8';
-
+import { Buffer } from 'buffer';
 import { PredictionsValidationErrorCode } from '../errors/types/validation';
 import { assertValidationError } from '../errors/utils/assertValidationError';
 import {
@@ -61,15 +59,12 @@ export class AmazonAIConvertPredictionsProvider {
 
 		if (isTranslateTextInput(input)) {
 			logger.debug('translateText');
-
 			return this.translateText(input);
 		} else if (isTextToSpeechInput(input)) {
 			logger.debug('textToSpeech');
-
 			return this.convertTextToSpeech(input);
 		} else {
 			logger.debug('textToSpeech');
-
 			return this.convertSpeechToText(input);
 		}
 	}
@@ -121,7 +116,6 @@ export class AmazonAIConvertPredictionsProvider {
 			Text: input.translateText?.source?.text,
 		});
 		const data = await this.translateClient.send(translateTextCommand);
-
 		return {
 			text: data.TranslatedText,
 			language: data.TargetLanguageCode,
@@ -167,6 +161,7 @@ export class AmazonAIConvertPredictionsProvider {
 			VoiceId: voiceId,
 			TextType: 'text',
 			SampleRate: '24000',
+			// tslint:disable-next-line: align
 		});
 		const data = await this.pollyClient.send(synthesizeSpeechCommand);
 		const response = new Response(data.AudioStream as ReadableStream);
@@ -175,7 +170,6 @@ export class AmazonAIConvertPredictionsProvider {
 			type: data.ContentType,
 		});
 		const url = URL.createObjectURL(blob);
-
 		return {
 			speech: { url },
 			audioStream: arrayBuffer,
@@ -224,7 +218,6 @@ export class AmazonAIConvertPredictionsProvider {
 			raw: source.bytes,
 			languageCode: language,
 		});
-
 		return {
 			transcription: {
 				fullText,
@@ -268,7 +261,6 @@ export class AmazonAIConvertPredictionsProvider {
 				}
 			}
 		}
-
 		return decodedMessage;
 	}
 
@@ -277,7 +269,7 @@ export class AmazonAIConvertPredictionsProvider {
 		raw,
 		languageCode,
 	}: TranscribeData): Promise<string> {
-		return new Promise((resolve, reject) => {
+		return new Promise((res, rej) => {
 			let fullText = '';
 			connection.onmessage = message => {
 				try {
@@ -290,19 +282,18 @@ export class AmazonAIConvertPredictionsProvider {
 					}
 				} catch (err: unknown) {
 					logger.debug(err);
-					reject(err);
+					rej(err);
 				}
 			};
 
 			connection.onerror = errorEvent => {
 				logger.debug({ errorEvent });
-				reject(new Error('failed to transcribe, network error'));
+				rej('failed to transcribe, network error');
 			};
 
 			connection.onclose = closeEvent => {
 				logger.debug({ closeEvent });
-
-				resolve(fullText.trim());
+				return res(fullText.trim());
 			};
 
 			logger.debug({ raw });
@@ -375,7 +366,6 @@ export class AmazonAIConvertPredictionsProvider {
 			const s = Math.max(-1, Math.min(1, input[i]));
 			view.setInt16(offset, s < 0 ? s * 0x8000 : s * 0x7fff, true);
 		}
-
 		return buffer;
 	}
 
@@ -403,8 +393,8 @@ export class AmazonAIConvertPredictionsProvider {
 		let offsetBuffer = 0;
 		while (offsetResult < result.length) {
 			const nextOffsetBuffer = Math.round((offsetResult + 1) * sampleRateRatio);
-			let accum = 0;
-			let count = 0;
+			let accum = 0,
+				count = 0;
 			for (
 				let i = offsetBuffer;
 				i < nextOffsetBuffer && i < buffer.length;
@@ -430,7 +420,7 @@ export class AmazonAIConvertPredictionsProvider {
 		region: string;
 		languageCode: string;
 	}): Promise<WebSocket> {
-		return new Promise((resolve, _reject) => {
+		return new Promise(async (res, rej) => {
 			const signedUrl = this.generateTranscribeUrl({
 				credentials,
 				region,
@@ -443,7 +433,7 @@ export class AmazonAIConvertPredictionsProvider {
 			connection.binaryType = 'arraybuffer';
 			connection.onopen = () => {
 				logger.debug('connected');
-				resolve(connection);
+				res(connection);
 			};
 		});
 	}

package/src/providers/AmazonAIIdentifyPredictionsProvider.ts

@@ -26,7 +26,6 @@ import {
 	DetectDocumentTextCommandInput,
 	TextractClient,
 } from '@aws-sdk/client-textract';
-
 import { PredictionsValidationErrorCode } from '../errors/types/validation';
 import { assertValidationError } from '../errors/utils/assertValidationError';
 import {
@@ -51,8 +50,12 @@ import {
 	isStorageSource,
 	isValidIdentifyInput,
 } from '../types';
-import { BlockList, Image, TextDetectionList } from '../types/AWSTypes';
-
+import {
+	BlockList,
+	Document,
+	Image,
+	TextDetectionList,
+} from '../types/AWSTypes';
 import {
 	categorizeRekognitionBlocks,
 	categorizeTextractBlocks,
@@ -81,15 +84,12 @@ export class AmazonAIIdentifyPredictionsProvider {
 
 		if (isIdentifyTextInput(input)) {
 			logger.debug('identifyText');
-
 			return this.identifyText(input);
 		} else if (isIdentifyLabelsInput(input)) {
 			logger.debug('identifyLabels');
-
 			return this.identifyLabels(input);
 		} else {
 			logger.debug('identifyEntities');
-
 			return this.identifyEntities(input);
 		}
 	}
@@ -102,7 +102,7 @@ export class AmazonAIIdentifyPredictionsProvider {
 	 * @return {Promise<Image>} - Promise resolving to the converted source object.
 	 */
 	private configureSource(source: IdentifySource): Promise<Image> {
-		return new Promise((resolve, reject) => {
+		return new Promise((res, rej) => {
 			if (isStorageSource(source)) {
 				const storageConfig = {
 					accessLevel: source.level,
@@ -114,44 +114,37 @@ export class AmazonAIIdentifyPredictionsProvider {
 						const parser =
 							/https:\/\/([a-zA-Z0-9%\-_.]+)\.s3\.[A-Za-z0-9%\-._~]+\/([a-zA-Z0-9%\-._~/]+)\?/;
 						const parsedURL = value.url.toString().match(parser) ?? '';
-						if (parsedURL.length < 3)
-							reject(new Error('Invalid S3 key was given.'));
-						resolve({
+						if (parsedURL.length < 3) rej('Invalid S3 key was given.');
+						res({
 							S3Object: {
 								Bucket: parsedURL[1],
 								Name: decodeURIComponent(parsedURL[2]),
 							},
 						});
 					})
-					.catch(err => {
-						reject(err);
-					});
+					.catch(err => rej(err));
 			} else if (isFileSource(source)) {
 				blobToArrayBuffer(source.file)
 					.then(buffer => {
-						resolve({ Bytes: new Uint8Array(buffer) });
+						res({ Bytes: new Uint8Array(buffer) });
 					})
-					.catch(err => {
-						reject(err);
-					});
+					.catch(err => rej(err));
 			} else if (isIdentifyBytesSource(source)) {
-				const { bytes } = source;
+				const bytes = source.bytes;
 				if (bytes instanceof Blob) {
 					blobToArrayBuffer(bytes)
 						.then(buffer => {
-							resolve({ Bytes: new Uint8Array(buffer) });
+							res({ Bytes: new Uint8Array(buffer) });
 						})
-						.catch(err => {
-							reject(err);
-						});
+						.catch(err => rej(err));
 				}
 				if (bytes instanceof ArrayBuffer || bytes instanceof Buffer) {
-					resolve({ Bytes: new Uint8Array(bytes) } as Image);
+					res({ Bytes: new Uint8Array(bytes) } as Image);
 				}
 				// everything else can be directly passed to Rekognition / Textract.
-				resolve({ Bytes: bytes } as Image);
+				res({ Bytes: bytes } as Image);
 			} else {
-				reject(new Error('Input source is not configured correctly.'));
+				rej('Input source is not configured correctly.');
 			}
 		});
 	}
@@ -186,8 +179,9 @@ export class AmazonAIIdentifyPredictionsProvider {
 			credentials,
 			customUserAgent: _getPredictionsIdentifyAmplifyUserAgent(),
 		});
+		let inputDocument: Document;
 
-		const inputDocument = await this.configureSource(input.text?.source);
+		inputDocument = await this.configureSource(input.text?.source);
 
 		// get default value if format isn't specified in the input.
 		const format = input.text?.format ?? configFormat;
@@ -243,7 +237,6 @@ export class AmazonAIIdentifyPredictionsProvider {
 
 			const analyzeDocumentCommand = new AnalyzeDocumentCommand(param);
 			const { Blocks } = await this.textractClient.send(analyzeDocumentCommand);
-
 			return categorizeTextractBlocks(Blocks as BlockList);
 		}
 	}
@@ -292,15 +285,14 @@ export class AmazonAIIdentifyPredictionsProvider {
 			data.forEach(val => {
 				identifyResult = { ...identifyResult, ...val };
 			});
-
 			return identifyResult;
 		});
 	}
 
 	/**
 	 * Calls Rekognition.detectLabels and organizes the returned data.
-	 * @param param - parameters as {@link DetectLabelsCommandInput} to be passed onto Rekognition
-	 * @return a promise resolving to organized detectLabels response as {@link IdentifyLabelsOutput}.
+	 * @param {DetectLabelsInput} param - parameter to be passed onto Rekognition
+	 * @return {Promise<IdentifyLabelsOutput>} - Promise resolving to organized detectLabels response.
 	 */
 	private async detectLabels(
 		param: DetectLabelsCommandInput,
@@ -314,7 +306,6 @@ export class AmazonAIIdentifyPredictionsProvider {
 					instance =>
 						makeCamelCase(instance.BoundingBox) as BoundingBox | undefined,
 				) || [];
-
 			return {
 				name: label.Name,
 				boundingBoxes: boxes,
@@ -324,14 +315,13 @@ export class AmazonAIIdentifyPredictionsProvider {
 				},
 			};
 		});
-
 		return { labels: detectLabelData };
 	}
 
 	/**
 	 * Calls Rekognition.detectModerationLabels and organizes the returned data.
-	 * @param param parameter to be passed onto Rekognition as {@link DetectModerationLabelsCommandInput}
-	 * @return a promise resolving to organized detectModerationLabels response as {@link IdentifyLabelsOutput}.
+	 * @param {Rekognition.DetectLabelsRequest} param - Parameter to be passed onto Rekognition
+	 * @return {Promise<IdentifyLabelsOutput>} - Promise resolving to organized detectModerationLabels response.
 	 */
 	private async detectModerationLabels(
 		param: DetectModerationLabelsCommandInput,
@@ -352,8 +342,8 @@ export class AmazonAIIdentifyPredictionsProvider {
 	/**
 	 * Identify faces within an image that is provided as input, and match faces from a collection
 	 * or identify celebrities.
-	 * @param input - object of {@link IdentifyEntitiesInput} containing the source image and face match options.
-	 * @return a promise resolving to identify results as {@link IdentifyEntitiesOutput}.
+	 * @param {IdentifyEntityInput} input - object containing the source image and face match options.
+	 * @return {Promise<IdentifyEntityOutput>} Promise resolving to identify results.
 	 */
 	protected async identifyEntities(
 		input: IdentifyEntitiesInput,
@@ -411,7 +401,6 @@ export class AmazonAIIdentifyPredictionsProvider {
 							},
 						}) as IdentifyEntity,
 				) ?? [];
-
 			return { entities: faces };
 		} else if (
 			isIdentifyFromCollection(input.entities) &&
@@ -436,7 +425,6 @@ export class AmazonAIIdentifyPredictionsProvider {
 					const externalImageId = match.Face?.ExternalImageId
 						? this.decodeExternalImageId(match.Face.ExternalImageId)
 						: undefined;
-
 					return {
 						boundingBox: makeCamelCase(match.Face?.BoundingBox),
 						metadata: {
@@ -445,7 +433,6 @@ export class AmazonAIIdentifyPredictionsProvider {
 						},
 					} as IdentifyEntity;
 				}) ?? [];
-
 			return { entities: faces };
 		} else {
 			const detectFacesCommand = new DetectFacesCommand(param);
@@ -471,7 +458,6 @@ export class AmazonAIIdentifyPredictionsProvider {
 					faceAttributes.emotions = detail.Emotions?.map(
 						emotion => emotion.Type,
 					);
-
 					return {
 						boundingBox: makeCamelCase(detail.BoundingBox),
 						landmarks: makeCamelCaseArray(detail.Landmarks),
@@ -483,7 +469,6 @@ export class AmazonAIIdentifyPredictionsProvider {
 						},
 					} as IdentifyEntity;
 				}) ?? [];
-
 			return { entities: faces };
 		}
 	}

package/src/providers/AmazonAIInterpretPredictionsProvider.ts

@@ -6,6 +6,7 @@ import {
 	PredictionsAction,
 	getAmplifyUserAgentObject,
 } from '@aws-amplify/core/internals/utils';
+
 import {
 	ComprehendClient,
 	DetectDominantLanguageCommand,
@@ -16,7 +17,6 @@ import {
 	Entity,
 	SyntaxToken,
 } from '@aws-sdk/client-comprehend';
-
 import { PredictionsValidationErrorCode } from '../errors/types/validation';
 import { assertValidationError } from '../errors/utils/assertValidationError';
 import {
@@ -63,9 +63,9 @@ export class AmazonAIInterpretPredictionsProvider {
 		const { text: textSource } = input;
 		const { source, type = defaultType } = textSource;
 		const { text } = source;
-		let sourceLanguage;
+		let language;
 		if (isInterpretTextOthers(textSource)) {
-			sourceLanguage = (textSource as InterpretTextOthers).source.language;
+			language = (textSource as InterpretTextOthers).source.language;
 		}
 
 		this.comprehendClient = new ComprehendClient({
@@ -79,7 +79,7 @@ export class AmazonAIInterpretPredictionsProvider {
 
 		const doAll = type === 'all';
 
-		let languageCode = sourceLanguage;
+		let languageCode = language;
 		if (doAll || type === 'language') {
 			const languageDetectionParams = {
 				Text: text,
@@ -87,7 +87,7 @@ export class AmazonAIInterpretPredictionsProvider {
 			languageCode = await this.detectLanguage(languageDetectionParams);
 		}
 
-		let entitiesPromise: Promise<TextEntities[]> | undefined;
+		let entitiesPromise: Promise<Array<TextEntities>> | undefined;
 		if (doAll || type === 'entities') {
 			assertValidationError(
 				!!languageCode,
@@ -113,7 +113,7 @@ export class AmazonAIInterpretPredictionsProvider {
 			sentimentPromise = this.detectSentiment(sentimentParams);
 		}
 
-		let syntaxPromise: Promise<TextSyntax[]> | undefined;
+		let syntaxPromise: Promise<Array<TextSyntax>> | undefined;
 		if (doAll || type === 'syntax') {
 			assertValidationError(
 				!!languageCode,
@@ -145,7 +145,6 @@ export class AmazonAIInterpretPredictionsProvider {
 			syntaxPromise,
 			keyPhrasesPromise,
 		]);
-
 		return {
 			textInterpretation: {
 				keyPhrases,
@@ -161,9 +160,8 @@ export class AmazonAIInterpretPredictionsProvider {
 		try {
 			const detectKeyPhrasesCommand = new DetectKeyPhrasesCommand(params);
 			const data = await this.comprehendClient!.send(detectKeyPhrasesCommand);
-			const { KeyPhrases: keyPhrases = [] } = data || {};
-
-			return keyPhrases.map(({ Text: text }) => {
+			const { KeyPhrases = [] } = data || {};
+			return KeyPhrases.map(({ Text: text }) => {
 				return { text };
 			});
 		} catch (err: any) {
@@ -178,12 +176,11 @@ export class AmazonAIInterpretPredictionsProvider {
 		}
 	}
 
-	private async detectSyntax(params: DetectParams): Promise<TextSyntax[]> {
+	private async detectSyntax(params: DetectParams): Promise<Array<TextSyntax>> {
 		try {
 			const detectSyntaxCommand = new DetectSyntaxCommand(params);
 			const data = await this.comprehendClient!.send(detectSyntaxCommand);
 			const { SyntaxTokens = [] } = data || {};
-
 			return this.serializeSyntaxFromComprehend(SyntaxTokens);
 		} catch (err: any) {
 			if (err.code === 'AccessDeniedException') {
@@ -197,7 +194,9 @@ export class AmazonAIInterpretPredictionsProvider {
 		}
 	}
 
-	private serializeSyntaxFromComprehend(tokens: SyntaxToken[]): TextSyntax[] {
+	private serializeSyntaxFromComprehend(
+		tokens: SyntaxToken[],
+	): Array<TextSyntax> {
 		let response: TextSyntax[] = [];
 		if (tokens && Array.isArray(tokens)) {
 			response = tokens.map(
@@ -206,7 +205,6 @@ export class AmazonAIInterpretPredictionsProvider {
 				},
 			);
 		}
-
 		return response;
 	}
 
@@ -222,8 +220,7 @@ export class AmazonAIInterpretPredictionsProvider {
 					Neutral: neutral = 0,
 					Mixed: mixed = 0,
 				} = {},
-			} = data ?? {};
-
+			} = ({} = data);
 			return { predominant, positive, negative, neutral, mixed };
 		} catch (err: any) {
 			if (err.code === 'AccessDeniedException') {
@@ -237,12 +234,13 @@ export class AmazonAIInterpretPredictionsProvider {
 		}
 	}
 
-	private async detectEntities(params: DetectParams): Promise<TextEntities[]> {
+	private async detectEntities(
+		params: DetectParams,
+	): Promise<Array<TextEntities>> {
 		try {
 			const detectEntitiesCommand = new DetectEntitiesCommand(params);
 			const data = await this.comprehendClient!.send(detectEntitiesCommand);
 			const { Entities = [] } = data || {};
-
 			return this.serializeEntitiesFromComprehend(Entities);
 		} catch (err: any) {
 			if (err.code === 'AccessDeniedException') {
@@ -256,14 +254,13 @@ export class AmazonAIInterpretPredictionsProvider {
 		}
 	}
 
-	private serializeEntitiesFromComprehend(data: Entity[]): TextEntities[] {
+	private serializeEntitiesFromComprehend(data: Entity[]): Array<TextEntities> {
 		let response: TextEntities[] = [];
 		if (data && Array.isArray(data)) {
 			response = data.map(({ Type: type, Text: text }) => {
 				return { type, text };
 			});
 		}
-
 		return response;
 	}
 
@@ -275,8 +272,7 @@ export class AmazonAIInterpretPredictionsProvider {
 			const data = await this.comprehendClient!.send(
 				detectDominantLanguageCommand,
 			);
-			const { Languages: [{ LanguageCode }] = [{ LanguageCode: undefined }] } =
-				data ?? {};
+			const { Languages: [{ LanguageCode }] = [{}] } = ({} = data || {});
 			assertValidationError(
 				!!LanguageCode,
 				PredictionsValidationErrorCode.NoLanguage,

package/src/providers/IdentifyTextUtils.ts

@@ -11,7 +11,6 @@ import {
 	TableCell,
 } from '../types';
 import { Block, BlockList, TextDetectionList } from '../types/AWSTypes';
-
 import { makeCamelCase, makeCamelCaseArray } from './Utils';
 
 function getBoundingBox(geometry?: Geometry): BoundingBox | undefined {
@@ -20,7 +19,6 @@ function getBoundingBox(geometry?: Geometry): BoundingBox | undefined {
 
 function getPolygon(geometry?: Geometry): Polygon | undefined {
 	if (!geometry?.Polygon) return undefined;
-
 	return makeCamelCaseArray(Array.from(geometry.Polygon)) as Polygon;
 }
 
@@ -71,7 +69,6 @@ export function categorizeRekognitionBlocks(
 		0,
 		response.text.fullText.length - 1,
 	);
-
 	return response;
 }
 
@@ -105,9 +102,9 @@ export function categorizeTextractBlocks(
 	 * Note that we do not map `WORD` and `TABLE` in `blockMap` because they will not be referenced by any other
 	 * block except the Page block.
 	 */
-	const tableBlocks: BlockList = [];
-	const keyValueBlocks: BlockList = [];
-	const blockMap: Record<string, Block> = {};
+	const tableBlocks: BlockList = Array();
+	const keyValueBlocks: BlockList = Array();
+	const blockMap: { [id: string]: Block } = {};
 
 	blocks.forEach(block => {
 		switch (block.BlockType) {
@@ -133,8 +130,9 @@ export function categorizeTextractBlocks(
 					blockMap[block.Id] = block;
 				}
 				break;
-			case 'SELECTION_ELEMENT': {
-				const selectionStatus = block.SelectionStatus === 'SELECTED';
+			case 'SELECTION_ELEMENT':
+				const selectionStatus =
+					block.SelectionStatus === 'SELECTED' ? true : false;
 				if (!response.text.selections) response.text.selections = [];
 				response.text.selections.push({
 					selected: selectionStatus,
@@ -145,7 +143,6 @@ export function categorizeTextractBlocks(
 					blockMap[block.Id] = block;
 				}
 				break;
-			}
 			case 'TABLE':
 				tableBlocks.push(block);
 				break;
@@ -169,14 +166,14 @@ export function categorizeTextractBlocks(
 
 	// Post-process complex structures if they exist.
 	if (tableBlocks.length !== 0) {
-		const tableResponse: Table[] = [];
+		const tableResponse: Table[] = Array();
 		tableBlocks.forEach(table => {
 			tableResponse.push(constructTable(table, blockMap));
 		});
 		response.text.tables = tableResponse;
 	}
 	if (keyValueBlocks.length !== 0) {
-		const keyValueResponse: KeyValue[] = [];
+		const keyValueResponse: KeyValue[] = Array();
 		keyValueBlocks.forEach(keyValue => {
 			// We need the KeyValue blocks of EntityType = `KEY`, which has both key and value references.
 			if (keyValue.EntityTypes) {
@@ -188,7 +185,6 @@ export function categorizeTextractBlocks(
 		});
 		response.text.keyValues = keyValueResponse;
 	}
-
 	return response;
 }
 
@@ -197,8 +193,12 @@ export function categorizeTextractBlocks(
  * @param {Block} table - Table block that has references (`Relationships`) to its cells
  * @param {[id: string]: Block} blockMap - Maps block Ids to blocks.
  */
-function constructTable(table: Block, blockMap: Record<string, Block>): Table {
-	const tableMatrix: TableCell[][] = [];
+function constructTable(
+	table: Block,
+	blockMap: { [key: string]: Block },
+): Table {
+	let tableMatrix: TableCell[][];
+	tableMatrix = [];
 	// visit each of the cell associated with the table's relationship.
 	for (const tableRelation of table.Relationships ?? []) {
 		for (const cellId of tableRelation.Ids ?? []) {
@@ -225,7 +225,6 @@ function constructTable(table: Block, blockMap: Record<string, Block>): Table {
 	const columnSize = tableMatrix[0].length;
 	const boundingBox = getBoundingBox(table.Geometry);
 	const polygon = getPolygon(table.Geometry);
-
 	// Note that we leave spanned cells undefined for distinction
 	return {
 		size: { rows: rowSize, columns: columnSize },
@@ -242,11 +241,11 @@ function constructTable(table: Block, blockMap: Record<string, Block>): Table {
  */
 function constructKeyValue(
 	keyBlock: Block,
-	blockMap: Record<string, Block>,
+	blockMap: { [key: string]: Block },
 ): KeyValue {
-	let keyText = '';
-	let valueText = '';
-	let valueSelected = false;
+	let keyText: string = '';
+	let valueText: string = '';
+	let valueSelected: boolean = false;
 	for (const keyValueRelation of keyBlock.Relationships ?? []) {
 		if (keyValueRelation.Type === 'CHILD') {
 			// relation refers to key
@@ -262,7 +261,6 @@ function constructKeyValue(
 			}
 		}
 	}
-
 	return {
 		key: keyText,
 		value: { text: valueText, selected: valueSelected },
@@ -278,10 +276,10 @@ function constructKeyValue(
  */
 function extractContentsFromBlock(
 	block: Block,
-	blockMap: Record<string, Block>,
+	blockMap: { [id: string]: Block },
 ): Content {
-	let words = '';
-	let isSelected = false;
+	let words: string = '';
+	let isSelected: boolean = false;
 
 	if (!block.Relationships) {
 		// some block might have no content
@@ -293,12 +291,11 @@ function extractContentsFromBlock(
 			if (contentBlock.BlockType === 'WORD') {
 				words += contentBlock.Text + ' ';
 			} else if (contentBlock.BlockType === 'SELECTION_ELEMENT') {
-				isSelected = contentBlock.SelectionStatus === 'SELECTED';
+				isSelected = contentBlock.SelectionStatus === 'SELECTED' ? true : false;
 			}
 		}
 	}
 
 	words = words.substr(0, words.length - 1); // remove trailing space.
-
 	return { text: words, selected: isSelected };
 }