@botbuildercommunity/middleware-aws-comprehend/lib/sentiment.js

Bot Framework middleware component for the Amazon AWS Comprehend service, including sentiment analysis, language detection, entity extraction, and keyword recognition.

"use strict";
var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
    function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
    return new (P || (P = Promise))(function (resolve, reject) {
        function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
        function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
        function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
        step((generator = generator.apply(thisArg, _arguments || [])).next());
    });
};
Object.defineProperty(exports, "__esModule", { value: true });
const botbuilder_1 = require("botbuilder");
const engine_1 = require("./engine");
/**
 * @module botbuildercommunity/middleware-aws-comprehend
 */
class SentimentAnalysis {
    constructor() {
        this.engine = new engine_1.AWSComprehendEngine();
    }
    onTurn(context, next) {
        return __awaiter(this, void 0, void 0, function* () {
            if (context.activity.type === botbuilder_1.ActivityTypes.Message) {
                const input = {
                    documents: [
                        {
                            'id': '1',
                            'text': context.activity.text
                        }
                    ]
                };
                try {
                    const result = yield this.engine.sentiment(input);
                    const s = result.documents[0].score;
                    const t = result.documents[0].sentiment;
                    context.turnState.set('sentimentScore', s);
                    context.turnState.set('sentimentType', t);
                }
                catch (e) {
                    throw new Error(`Failed to process sentiment on ${context.activity.text}. Error: ${e}`);
                }
            }
            yield next();
        });
    }
}
exports.SentimentAnalysis = SentimentAnalysis;
//# sourceMappingURL=sentiment.js.map