tmaiplugin/dist/summaryplugin.js

TrainingMaster AIGC Component

"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());
    });
};
var __importDefault = (this && this.__importDefault) || function (mod) {
    return (mod && mod.__esModule) ? mod : { "default": mod };
};
Object.defineProperty(exports, "__esModule", { value: true });
exports.SummaryPlugin = void 0;
const stringutil_1 = require("./util/stringutil");
const aipluginbase_1 = __importDefault(require("./aipluginbase"));
const ROLE_DEFINE = ['你是一名专业的内容分析专家，擅长提取篇幅较长内容的重点，形成摘要',
    '你是一位程序设计专家，特别擅长数据分析，内容组织并进行结构化设计，形成Json格式的数据'];
const PROMPT = [`请分析以下内容并进行精简，按照内容提炼题纲以及小标题，形成层级式摘要，内容尽量精简
原文内容如下："
{{CONTENT}}
"`,
    `请分析内容结构化,按照[{outline:"提纲标题","summary":["摘要内容1","摘要内容2","摘要内容3"]}]的标准Json结构输出。
内容如下："
{{CONTENT}}
"`
];
/**
 * 生成摘要插件
 */
class SummaryPlugin extends aipluginbase_1.default {
    /**
     * 从指定的文本内容中生成相关的问答
     * @param {*} content
     * @param {*} count
     * @param {*} axiosOption
     * @returns
     */ //并在答案末尾处必须给出答案内容中的关键词
    execute(params) {
        return __awaiter(this, void 0, void 0, function* () {
            let { content, sectionlength, axios } = params;
            sectionlength = sectionlength || 1024;
            if (!this.gptInstance)
                return { successed: false };
            if (!content)
                return { successed: false, error: { errcode: 2, errmsg: '缺失参数' } };
            let arrContent = (0, stringutil_1.splitLongText)(content, sectionlength);
            let summary = [];
            while (arrContent.length > 0) {
                let subarray = [
                    { role: 'system', content: ROLE_DEFINE[0] },
                    { role: 'user', content: PROMPT[0].replace('{{CONTENT}}', arrContent.slice(0, 1)[0]) },
                ];
                console.log('Summary  Prompt:', subarray);
                let result = yield this.gptInstance.chatRequest(subarray, { replyCounts: 1 }, axios);
                ///如果请求发生了网络错误(不是内容合规问题），则再重试一次，如果任然有错则放弃
                if (!result.successed && result.error != 'content_filter') {
                    console.log('network error,retry onemore time');
                    result = yield this.gptInstance.chatRequest(subarray, { replyCounts: 1 }, axios);
                }
                if (result.successed && result.message) {
                    let answerString = result.message[0].message.content.trim();
                    let orgJsonPrompt = [
                        { role: 'system', content: ROLE_DEFINE[1] },
                        { role: 'user', content: PROMPT[1].replace('{{CONTENT}}', answerString) }
                    ];
                    console.log('orgJsonPrompt', orgJsonPrompt);
                    let fixedJsonResult = yield this.gptInstance.chatRequest(orgJsonPrompt, { replyCounts: 1 }, {});
                    if (fixedJsonResult.successed) {
                        answerString = fixedJsonResult.message[0].message.content.trim().replace(/\t|\n|\v|\r|\f/g, '');
                    }
                    let jsonObj = fixedJsonResult.successed ? (0, stringutil_1.fixedJsonString)(answerString) : [];
                    ///能够提取到内容
                    if (jsonObj.length)
                        summary = summary.concat(jsonObj);
                }
                ////删除已经处理的文本
                arrContent.splice(0, 1);
            }
            return { successed: true, article: summary };
        });
    }
}
exports.SummaryPlugin = SummaryPlugin;