wechaty-qnamaker/dist/src/asker.js

QnAMaker.ai Plugin for Wechaty

"use strict";
var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
    if (k2 === undefined) k2 = k;
    Object.defineProperty(o, k2, { enumerable: true, get: function() { return m[k]; } });
}) : (function(o, m, k, k2) {
    if (k2 === undefined) k2 = k;
    o[k2] = m[k];
}));
var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
    Object.defineProperty(o, "default", { enumerable: true, value: v });
}) : function(o, v) {
    o["default"] = v;
});
var __importStar = (this && this.__importStar) || function (mod) {
    if (mod && mod.__esModule) return mod;
    var result = {};
    if (mod != null) for (var k in mod) if (k !== "default" && Object.hasOwnProperty.call(mod, k)) __createBinding(result, mod, k);
    __setModuleDefault(result, mod);
    return result;
};
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 });
exports.asker = void 0;
/**
 * QuickStart: QnA Maker client library
 *  https://docs.microsoft.com/en-us/azure/cognitive-services/qnamaker/quickstarts/quickstart-sdk?tabs=visual-studio&pivots=programming-language-javascript#qnamakerruntimeclient-object-model
 *
 * Azure-Samples/cognitive-services-quickstart-code
 *  https://github.com/Azure-Samples/cognitive-services-quickstart-code/blob/master/javascript/QnAMaker/sdk/qnamaker_quickstart.js
 */
const msRest = __importStar(require("@azure/ms-rest-js"));
const runtime = __importStar(require("@azure/cognitiveservices-qnamaker-runtime"));
const wechaty_1 = require("wechaty");
function asker(options) {
    wechaty_1.log.verbose('WechatyQnAMaker', 'asker(%s)', JSON.stringify(options));
    const customHeaders = { Authorization: `EndpointKey ${options.endpointKey}` };
    const queryingURL = `https://${options.resourceName}.azurewebsites.net`;
    const queryRuntimeCredentials = new msRest.ApiKeyCredentials({
        inHeader: { 'Ocp-Apim-Subscription-Key': options.endpointKey },
    });
    const runtimeClient = new runtime.QnAMakerRuntimeClient(queryRuntimeCredentials, queryingURL);
    return function ask(question, dto = {}) {
        return __awaiter(this, void 0, void 0, function* () {
            wechaty_1.log.verbose('WechatyQnAMaker', 'ask(%s, %s)', question, JSON.stringify(dto));
            const normalizedQueryDto = Object.assign(Object.assign({ scoreThreshold: options.scoreThreshold }, dto), { question });
            wechaty_1.log.verbose('WechatyQnAMaker', 'ask() normalizedQueryDto: %s', JSON.stringify(normalizedQueryDto));
            const requestQuery = yield runtimeClient.runtime.generateAnswer(options.knowledgeBaseId, normalizedQueryDto, { customHeaders });
            // console.info(JSON.stringify(requestQuery))
            const answers = requestQuery.answers || [];
            return answers.filter(item => item.score);
        });
    };
}
exports.asker = asker;
//# sourceMappingURL=asker.js.map