doomiaichat 2.1.0 → 2.3.0

package/dist/openai.js

@@ -8,9 +8,13 @@ var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, ge
         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 });
 const openai_1 = require("openai");
-const events_1 = require("events");
+// import { EventEmitter } from "events";
+const gptbase_1 = __importDefault(require("./gptbase"));
 const SECTION_LENGTH = 1600; ///每2400个字符分成一组
 const MESSAGE_LENGTH = 1; ///每次送8句话给openai 进行解析，送多了，会报错
 //请将答案放在最后,标记为答案：()
@@ -21,7 +25,7 @@ const QUESTION_TEXT_MAPPING = {
     completion: '你是一名专业的出题老师,根据以下内容,请生成@ITEMCOUNT@道填空题和对应答案,输出结果必须是JSON数组并按照[{"question":"","answer":["填空答案1","填空答案2"]}]的结构输出' //请将答案放在最后,标记为答案：()
 };
 const QUESTION_TYPE = ['singlechoice', 'multiplechoice', 'trueorfalse', 'completion'];
-class OpenAIGpt extends events_1.EventEmitter {
+class OpenAIGpt extends gptbase_1.default {
     /**
      *
      * @param apiKey 调用OpenAI 的key
@@ -71,6 +75,40 @@ class OpenAIGpt extends events_1.EventEmitter {
             }
         });
     }
+    /**
+     * 点评问题回答的评价
+     * @param question
+     * @param answer
+     * @param axiosOption
+     */
+    commentQuestionAnswer(question, answer, axiosOption = { timeout: 30000 }) {
+        return __awaiter(this, void 0, void 0, function* () {
+            if (!question || !answer)
+                return { successed: false, error: { errcode: 2, errmsg: '缺失参数' } };
+            let message = [
+                { role: 'system', content: '你是一名专业的培训师。' },
+                { role: 'user', content: `问题题干:“${question}”` },
+                { role: 'user', content: `回答内容:“${answer}”` },
+                { role: 'user', content: `请根据以上的回答内容进行点评，给出一段不超过200字的评语，以及0-100的得分。最终结果按照{"comment":"评语","score":80}的JSON结构输出` }
+            ];
+            const result = yield this.chatRequest(message, {}, axiosOption);
+            if (result.successed && result.message) {
+                let value = result.message[0].message.content.trim();
+                let replyJson = this.fixedJsonString(value);
+                ///能够提取到内容
+                if (replyJson.length)
+                    return Object.assign({ successed: true }, replyJson[0]);
+                ///回答的内容非JSON格式，自己来提取算了
+                console.log('自己组装');
+                let matched = value.match(/\d+分/g), score = 0;
+                if (matched && matched.length) {
+                    score = Number(matched[0].replace('分', ''));
+                }
+                return { successed: true, comment: value, score };
+            }
+            return { successed: false };
+        });
+    }
     /**
      * 判断一句话的表达情绪
      * @param {*} s1
@@ -195,7 +233,7 @@ class OpenAIGpt extends events_1.EventEmitter {
                 //subarray.push({ role: 'user', content:'请根据上述内容,给出一道提问与答案以及答案关键词，按照先问题内容，再标准答案，再关键词的顺序输出,关键词之间用、分开'})
                 subarray.unshift({ role: 'system', content: `你是一位专业培训师,从以下内容中提取${itemCount}条提问及答案，并从答案内容提取出至少2个关键词，最终结果按照[{"question":"提问内容","answer":"答案内容","keywords":["关键词1","关键词2"]}]的JSON数组结构输出。` });
                 //subarray.unshift({ role: 'system', content: `你是一位专业程序开发工程师,根据以下内容,按照[{"question":"问题内容","answer":"答案内容","keywords":["关键词1","关键词2"]}]JSON数组结构,给出${itemCount}条提问问题及答案以及答案关键词` })
-                console.log('subarray', subarray);
+                // console.log('subarray', subarray)
                 let result = yield this.chatRequest(subarray, { replyCounts: 1 }, axiosOption);
                 if (result.successed && result.message) {
                     // console.log('result is ', result.message[0].message.content)
@@ -213,7 +251,7 @@ class OpenAIGpt extends events_1.EventEmitter {
             arrContent = []; /// 释放内存
             ///发出信号，解析完毕
             this.emit('parseover', { type: 'qa', items: faqs });
-            return { successed: true, message: faqs };
+            return { successed: true, message: faqs.slice(0, count) };
         });
     }
     /**
@@ -255,16 +293,18 @@ class OpenAIGpt extends events_1.EventEmitter {
      */
     pickUpFaqContent(messages) {
         let answerString = messages[0].message.content.trim().replace(/\t|\n|\v|\r|\f/g, '');
-        const firstsybmol = answerString.indexOf("["); ////必须过滤出来数组
-        const lastsybmol = answerString.lastIndexOf("]");
-        console.log('answerString', answerString);
-        if (firstsybmol < 0 || lastsybmol < 0 || lastsybmol <= firstsybmol)
+        // const firstsybmol = answerString.indexOf("[");   ////必须过滤出来数组
+        // const lastsybmol = answerString.lastIndexOf("]");
+        // console.log('answerString', answerString)
+        // if (firstsybmol < 0 || lastsybmol < 0 || lastsybmol <= firstsybmol) return [];
+        // answerString = answerString.substr(firstsybmol, lastsybmol - firstsybmol + 1);
+        // console.log('answerString', answerString)
+        let jsonObj = this.fixedJsonString(answerString);
+        if (!jsonObj.length)
             return [];
-        answerString = answerString.substr(firstsybmol, lastsybmol - firstsybmol + 1);
-        console.log('answerString', answerString);
         try {
             //let jsonObj = JSON.parse(answerString);
-            let jsonObj = eval(answerString);
+            //let jsonObj = eval(answerString);
             jsonObj.map((item) => {
                 let realKeyword = [];
                 let keywords = (item.keywords + '').split(',');
@@ -364,29 +404,32 @@ class OpenAIGpt extends events_1.EventEmitter {
      */
     pickUpQuestions(result, count, questiontype, score = 1) {
         let answerString = result[0].message.content.trim().replace(/\t|\n|\v|\r|\f/g, '');
-        const firstsybmol = answerString.indexOf("["); ////必须过滤出来数组
-        const lastsybmol = answerString.lastIndexOf("]");
-        console.log('answerString', answerString);
-        if (firstsybmol < 0 || lastsybmol < 0 || lastsybmol <= firstsybmol)
+        // const firstsybmol = answerString.indexOf("[");   ////必须过滤出来数组
+        // const lastsybmol = answerString.lastIndexOf("]");
+        // console.log('answerString', answerString)
+        // if (firstsybmol < 0 || lastsybmol < 0 || lastsybmol <= firstsybmol) return [];
+        // answerString = answerString.substr(firstsybmol, lastsybmol - firstsybmol + 1);
+        // console.log('answerString', answerString)
+        let jsonObj = this.fixedJsonString(answerString);
+        if (!jsonObj.length)
             return [];
-        answerString = answerString.substr(firstsybmol, lastsybmol - firstsybmol + 1);
-        console.log('answerString', answerString);
         let returnItems = [];
         try {
-            let jsonObj = JSON.parse(answerString);
+            // let jsonObj = JSON.parse(answerString);
             returnItems = jsonObj.map((questionitem) => {
-                console.log('answer item', questionitem);
+                console.log('answer item from jsonObj', questionitem);
                 if (questionitem.choice && Array.isArray(questionitem.choice) && questiontype != 'completion') {
                     questionitem.fullanswer = (questionitem.answer + '').replace(/,|[^ABCDE]/g, '');
                     questionitem.score = score;
                     if (questionitem.choice) {
                         questionitem.choice = questionitem.choice.map((item, index) => {
-                            let seqNo = String.fromCharCode(65 + index);
+                            let seqNo = 'ABCDEFG'[index]; //String.fromCharCode(65 + index);
                             let correctReg = new RegExp(`${seqNo}.|${seqNo}`, 'ig');
+                            // console.log('itemitemitem', item)
                             //let answer = jsonObj.fullanswer
                             return {
                                 id: seqNo,
-                                content: item.replace(correctReg, '').trim(),
+                                content: (item + '').replace(correctReg, '').trim(),
                                 iscorrect: (questionitem.fullanswer || '').indexOf(seqNo) >= 0 ? 1 : 0
                                 //|| jsonObj.fullanswer.indexOf(m))
                             };
@@ -411,7 +454,7 @@ class OpenAIGpt extends events_1.EventEmitter {
                 }
                 ///单选题验证
                 if (questiontype == 'singlechoice') {
-                    let rightAnswer = questionitem.choice ? questionitem.choice.filter(item => { return item.iscorrect === 1; }) : [];
+                    let rightAnswer = questionitem.choice ? questionitem.choice.filter((item) => { return item.iscorrect === 1; }) : [];
                     ///单选题的正确选项大于了1个
                     if (rightAnswer.length != 1 || !questionitem.answer || questionitem.answer.length !== 1)
                         return null;
@@ -421,7 +464,7 @@ class OpenAIGpt extends events_1.EventEmitter {
                 }
                 ///多选题验证
                 if (questiontype == 'multiplechoice') {
-                    let rightAnswer = questionitem.choice ? questionitem.choice.filter(item => { return item.iscorrect === 1; }) : [];
+                    let rightAnswer = questionitem.choice ? questionitem.choice.filter((item) => { return item.iscorrect === 1; }) : [];
                     ///单选题的正确选项大于了1个
                     if (rightAnswer.length === 0 || !questionitem.answer || questionitem.answer.length === 0)
                         return null;
@@ -433,42 +476,55 @@ class OpenAIGpt extends events_1.EventEmitter {
             console.log('error happened:', err);
         }
         return returnItems.filter(i => { return i != null; }).slice(0, count);
-        // let item = result.map(m => {
-        //     ////防止输出的JSON格式不合法
-        //     try {
-        //         let jsonObj = JSON.parse(m.message.content)
-        //         jsonObj.score = score;
-        //         if (jsonObj.choice && Array.isArray(jsonObj.choice) && questiontype != 'completion') {
-        //             jsonObj.fullanswer = (jsonObj.answer + '').replace(/,|[^ABCDE]/g, '');
-        //             jsonObj.choice = jsonObj.choice.map((item:string, index:number) => {
-        //                 let seqNo = String.fromCharCode(65 + index);
-        //                 let correctReg = new RegExp(`${seqNo}.|${seqNo}`, 'ig')
-        //                 //let answer = jsonObj.fullanswer
-        //                 return {
-        //                     id: seqNo,
-        //                     content: item.replace(correctReg, '').trim(),
-        //                     iscorrect: (jsonObj.fullanswer.indexOf(seqNo) >= 0 || jsonObj.fullanswer.indexOf(m)) >= 0 ? 1 : 0
-        //                 }
-        //             })
-        //         }
-        //         switch (questiontype) {
-        //             case 'singlechoice':
-        //                 jsonObj.answer = (jsonObj.answer + '').replace(/,|[^ABCDEFG]/g, '').split('').slice(0, 1);
-        //                 break;
-        //             case 'multiplechoice':
-        //                 jsonObj.answer = (jsonObj.answer + '').replace(/,|[^ABCDEFG]/g, '').split('');
-        //                 break;
-        //             case 'trueorfalse':
-        //                 jsonObj.answer = [(jsonObj.answer + '').indexOf('正确') >= 0 ? 'A' : 'B']
-        //                 break;
-        //         }
-        //         return jsonObj;
-        //     } catch (err) {
-        //         console.log('error happened:', err);
-        //         return null;
-        //     }
-        // })
-        // return item.filter(i => { return i != null; });
+    }
+    /**
+     * 验证JSON字符串是否是真正可转换为JSON的合法格式
+     * 这里只能做一个最简单的处理，就是用两端的符号
+     * @param jsonstr
+     */
+    fixedJsonString(jsonstr) {
+        console.log('original json string:', jsonstr);
+        ///检查返回的是不是一个数组对象（我们需要的是数组对象)
+        let firstBracketSymbol = jsonstr.indexOf("["); ////必须过滤出来数组
+        let lastBracketSymbol = jsonstr.lastIndexOf("]");
+        ///第一个花括号出现的位置，如果花括号出现的位置早于 [ ,则默认返回的对象不是一个数组，仅仅是一个对象，
+        ///则需要我们用中括号包住
+        let firstBraceSymbol = jsonstr.indexOf("{");
+        let lastBraceSymbol = jsonstr.lastIndexOf("}");
+        ///返回的不是一个数组结构的,只是一个{}，我们帮他完成数组拼接
+        if (firstBraceSymbol >= 0 &&
+            firstBraceSymbol < (firstBracketSymbol >= 0 ? firstBracketSymbol : 1000) &&
+            lastBraceSymbol > firstBraceSymbol &&
+            lastBraceSymbol >= 0 && lastBraceSymbol > lastBracketSymbol) {
+            jsonstr = '[' + jsonstr.substr(firstBraceSymbol, lastBraceSymbol - firstBraceSymbol + 1);
+            +']';
+            firstBracketSymbol = 0;
+            lastBracketSymbol = jsonstr.length - 1;
+        }
+        else if (firstBracketSymbol < 0 || lastBracketSymbol < 0 || lastBracketSymbol <= firstBracketSymbol) {
+            return [];
+        }
+        jsonstr = jsonstr.substr(firstBracketSymbol, lastBracketSymbol - firstBracketSymbol + 1);
+        ///尽量处理一些能够一眼识别出来的JSON错误
+        jsonstr = jsonstr.replace(/}{/g, '},{');
+        let mutilitems = jsonstr.split('][');
+        ///确实存在多个数组拼接在一起，中间没有逗号隔开的了
+        let retObject = [];
+        for (let str of mutilitems) {
+            if (!str.startsWith('['))
+                str = '[' + str;
+            if (!str.endsWith(']'))
+                str = str + ']';
+            // console.log('json str', str)
+            try {
+                let jsonObj = eval(str);
+                retObject = retObject.concat(jsonObj);
+            }
+            catch (err) {
+                console.log('json error', str);
+            }
+        }
+        return retObject;
     }
     /**
      * 将一段很长的文本，按1024长度来划分到多个中

package/dist/openaiprovider.d.ts

@@ -1,7 +1,5 @@
-/**
-* 语音转文字服务商工厂
-*/
-import OpenAIGpt from './openai';
+import { ApiCredential } from './baiduai';
+import GptBase from './gptbase';
 /**
  * OpenAI/NLP 的服务提供商 OpenAI,微软，百度文心(待接入)，google(待接入)
  */
@@ -19,4 +17,4 @@ export type AIProviderEnum = typeof AIProviderEnum[keyof typeof AIProviderEnum];
  * @param {*} setting
  * @returns
  */
-export declare function createAIInstance(provider: AIProviderEnum, apikey: string, setting: any): OpenAIGpt | null;
+export declare function createAIInstance(provider: AIProviderEnum, apikey: string | ApiCredential, setting: any): GptBase | null;

package/dist/openaiprovider.js

@@ -9,6 +9,7 @@ exports.createAIInstance = exports.AIProviderEnum = void 0;
 */
 const openai_1 = __importDefault(require("./openai"));
 const azureai_1 = __importDefault(require("./azureai"));
+const baiduai_1 = __importDefault(require("./baiduai"));
 /**
  * OpenAI/NLP 的服务提供商 OpenAI,微软，百度文心(待接入)，google(待接入)
  */
@@ -26,12 +27,15 @@ exports.AIProviderEnum = {
  * @returns
  */
 function createAIInstance(provider, apikey, setting) {
-    let { model, maxtoken, temperature, endpoint, engine, version } = setting;
+    let { model, maxtoken, temperature, endpoint, engine, version } = setting || {};
     switch (provider) {
         case exports.AIProviderEnum.OPENAI:
-            return new openai_1.default(apikey, { model, maxtoken, temperature });
+            return new openai_1.default(apikey + '', { model, maxtoken, temperature });
         case exports.AIProviderEnum.MICROSOFT:
-            return new azureai_1.default(apikey, { endpoint, engine, version }, { model, maxtoken, temperature });
+            return new azureai_1.default(apikey + '', { endpoint, engine, version }, { model, maxtoken, temperature });
+        case exports.AIProviderEnum.BAIDU:
+            let cred = typeof (apikey) === 'string' ? { apikey, securitykey: apikey } : apikey;
+            return new baiduai_1.default(cred);
         default: return null;
     }
 }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "doomiaichat",
-  "version": "2.1.0",
+  "version": "2.3.0",
   "description": "Doomisoft OpenAI",
   "main": "dist/index.js",
   "scripts": {

package/src/azureai.ts

@@ -1,5 +1,4 @@
-import { AzureOpenAIPatameters, ChatReponse, OpenAIApiParameters } from "./declare";
-import axios from 'axios';
+import { AzureOpenAIPatameters, ChatReponse, OpenAIApiParameters, request } from "./declare";
 import OpenAIGpt from "./openai" 
 import { ChatCompletionRequestMessage } from "openai";
 export default class AzureAI extends OpenAIGpt {
@@ -32,13 +31,7 @@ export default class AzureAI extends OpenAIGpt {
 
         let messages: Array<ChatCompletionRequestMessage> = typeof (chatText) == 'string' ?
             [{ role: 'user', content: chatText }] : chatText;
-        //let axios = new Axios(axiosOption)
         try{
-            // const response = await axios.post(this.BaseUrl, {
-            //     messages,
-            //     temperature: Number(paramOption?.temperature || this.temperature),
-            //     max_tokens: Number(paramOption?.maxtoken || this.maxtoken),
-            // })
             let param = {
                 ...axiosOption,
                 method: "post",
@@ -49,9 +42,8 @@ export default class AzureAI extends OpenAIGpt {
                 },
                 url: this.BaseUrl
             };
-            console.log('axiosOption', param)
-
-            const response = await axios(param)
+            // console.log('axiosOption', param)
+            const response =await request(param)
             if (response.data.choices){
                 return { successed: true, message: response.data.choices };
             }

package/src/baiduai.ts

@@ -0,0 +1,106 @@
+import { EmotionResult, SimilarityResult, ChatReponse, SummaryReponse, ExaminationPaperResult, ApiResult, CacheProvider, request, CommentResult } from "./declare";
+import GptBase from "./gptbase"
+const TOKEN_CACHE_KEY = "key:_doomisoft:baiduwenxin:"
+export default class BaiduWenXinAI extends GptBase {
+    protected credential: ApiCredential;
+    private Cacher: CacheProvider|undefined;
+    /**
+     * 
+     * @param credential 调用OpenAI 的key
+     * @param cacher  用作accesstoken的缓存 
+     * @param apiOption  用作accesstoken的缓存 
+     */
+    constructor(credential: ApiCredential, cacher?:CacheProvider) {
+        super();
+        this.credential = credential;
+        this.Cacher = cacher;
+    }
+    /**
+     * 获取调用Api的Token
+     */
+    async getAccessToken():Promise<AccessTokenResult>{
+        ////如果当前应用是一个委托到第三方的
+        ///公众号或小程序，则需要用关联的第三方平台来获取api调用的accesstoken
+        ////如果是需要缓存token,则首先看缓存中是否有AccessToken
+        const accesstoken = this.Cacher ? (await this.Cacher.get(TOKEN_CACHE_KEY+this.credential.apikey)):null;
+        if (accesstoken) return { successed: true, access_token: accesstoken };
+        const option = {
+            url: `https://wenxin.baidu.com/moduleApi/portal/api/oauth/token?grant_type=client_credentials&client_id=${this.credential.apikey}&client_secret=${this.credential.securitykey}`,
+            method:"post",
+        };
+        const tokenData = await request(option);
+        if (tokenData.successed == true && !tokenData.data.code) {
+            ///把api accesstoken缓存起来
+            if (this.Cacher){
+                this.Cacher.set(TOKEN_CACHE_KEY + this.credential.apikey, tokenData.data.data,3600*20);
+            }
+            return { successed: true, access_token: tokenData.data.data };
+        }
+        return { successed: false, error: tokenData.data.msg }
+    }
+    /**
+     * 请求GPT接口
+     */
+    public async chatRequest(chatText: string | Array<any>, _paramOption: any, axiosOption: any = {}): Promise<ApiResult> {
+        console.log('call baidu')
+        if (!chatText) return { successed: false, error: { errcode: 2, errmsg: '缺失聊天的内容' } };
+        const accessToken = await this.getAccessToken();
+        if (!accessToken.successed || !accessToken.access_token) return {successed:false,error:'accesstoken get failed'}
+        try {
+            let param = {
+                ...axiosOption,
+                method: "post",
+                data: {
+                    text: `问题：${chatText}\n回答：`,
+                    seq_len: 512,
+                    topp: 0.5,
+                    penalty_score: 1.2,
+                    min_dec_len: 12,
+                    min_dec_penalty_text: "。?：！[<S>]",
+                    task_prompt: "qa",
+                    mask_type: "paragraph"
+                },
+                url:`https://wenxin.baidu.com/moduleApi/portal/api/rest/1.0/ernie/3.0.25/zeus?access_token=${accessToken.access_token}`
+            };
+            console.log('param', param)
+            const response = await request(param)
+            if (response.successed && !response.data.code) {
+                return { successed: true, ...response.data };
+            }
+            return { successed: false, ...response.data };
+        } catch (error) {
+            console.log('result is error ', error)
+            return { successed: false, error };
+        }
+
+    }
+    commentQuestionAnswer(_question: string, _answer: string, _axiosOption: any): Promise<CommentResult>{
+        throw new Error("Method not implemented.");
+    }
+    getScentenceEmotional(_s1: string, _axiosOption: any): Promise<EmotionResult> {
+        throw new Error("Method not implemented.");
+    }
+    getScentenseSimilarity(_s1: string, _s2: string, _axiosOption: any): Promise<SimilarityResult> {
+        throw new Error("Method not implemented.");
+    }
+    getSimilarityContent(_content: string, _count: number, _axiosOption: any): Promise<ChatReponse> {
+        throw new Error("Method not implemented.");
+    }
+    getSummaryOfContent(_content: string | any[], _axiosOption: any): Promise<SummaryReponse> {
+        throw new Error("Method not implemented.");
+    }
+    generateQuestionsFromContent(_content: string, _count: number, _axiosOption: any): Promise<ChatReponse> {
+        throw new Error("Method not implemented.");
+    }
+    generateExaminationPaperFromContent(_content: string, _paperOption: any, _axiosOption: any): Promise<ExaminationPaperResult> {
+        throw new Error("Method not implemented.");
+    }
+
+}
+export interface AccessTokenResult extends ApiResult {
+    'access_token'?: string
+}
+export interface ApiCredential{
+    'apikey':string,
+    'securitykey': string
+}

package/src/declare.ts

@@ -1,5 +1,6 @@
 
-interface ApiResult {
+import axios from 'axios';
+export interface ApiResult {
     /**
      * return the result of api called
      * @type {boolean}
@@ -97,9 +98,63 @@ export interface SimilarityResult extends ApiResult {
     'value'?: number;    ///相识度的值
 }
 
+/**
+ * 调用OpenAI Api的参数约定
+ */
+export interface CommentResult extends ApiResult {
+    'score'?:number,
+    'comment'?: string;    ///评价内容
+}
 /**
  * 调用OpenAI Api的参数约定
  */
 export interface EmotionResult extends ApiResult {
     'emotion'?: string;    ///情绪
+}
+/**
+ * 远程请求的返回
+ */
+export interface RpcResult extends ApiResult {
+    'data'?: any;
+}
+
+/**
+ * Axios远程请求封装
+ * @param opts 
+ * @returns 
+ */
+export async function request(opts: any = {}): Promise<RpcResult> {
+    if (!opts.data) opts.data = opts.body;
+    try {
+        let result = await axios(opts);
+        return { successed: true, data: result.data }
+    } catch (err) {
+        return { successed: false, error: err }
+    }
+}
+
+
+/**
+ * 数据缓存提供者接口
+ */
+export interface CacheProvider {
+    /**
+     * 缓存数据
+     * @param key 数据的键名称
+     * @param value 数据的键值
+     */
+    set(key: string, value: string | object, exp?: number): void;
+    /**
+     * 从缓存中读取数据
+     * @param key 数据的键名称
+     */
+    get(key: string): Promise<string | null>;
+
+
+    /**
+     * 删除缓存信息
+     * @param key 数据的键名称
+     */
+    delete(key: string): void;
+
 }

package/src/gptbase.ts

@@ -0,0 +1,65 @@
+import { EventEmitter } from "events"
+import { ChatReponse, SummaryReponse, ExaminationPaperResult, EmotionResult, SimilarityResult, ApiResult, CommentResult } from './declare'
+
+export default abstract class GptBase extends EventEmitter {
+    /**
+     * 构造函数
+     */
+    constructor() {
+        super();
+    }
+    /**
+     * 自由聊天模式
+     * @param chatText 
+     * @param _paramOption 
+     * @param axiosOption 
+     */
+    abstract chatRequest(chatText: string | Array<any>, _paramOption: any, axiosOption: any): Promise<ApiResult>;
+    /**
+     * 点评问题回答的评价
+     * @param question  问题题干
+     * @param answer 回答内容
+     * @param axiosOption 
+     */
+    abstract commentQuestionAnswer(question: string, answer: string, axiosOption: any): Promise<CommentResult>;
+    /**
+     * 获取句子的情感
+     * @param s1 
+     * @param axiosOption 
+     */
+    abstract getScentenceEmotional(s1: string, axiosOption: any ): Promise<EmotionResult> ;
+    /**
+     * 获取两段文本的相似度
+     * @param s1 
+     * @param s2 
+     * @param axiosOption 
+     */
+    abstract getScentenseSimilarity(s1: string, s2: string, axiosOption: any): Promise<SimilarityResult>;
+    /**
+     * 获取一段文本的相似内容
+     * @param content 
+     * @param count 
+     * @param axiosOption 
+     */
+    abstract getSimilarityContent(content: string, count: number, axiosOption: any): Promise<ChatReponse>;
+    /**
+     * 获取内容的摘要
+     * @param content 
+     * @param axiosOption 
+     */
+    abstract getSummaryOfContent(content: string | Array<any>, axiosOption: any): Promise<SummaryReponse>;
+    /**
+     * 从内容中提取问题和答案及管件子
+     * @param content 
+     * @param count 
+     * @param axiosOption 
+     */
+    abstract generateQuestionsFromContent(content: string, count: number, axiosOption: any): Promise<ChatReponse>;
+    /**
+     * 从内容中提取单选多选判断填空题
+     * @param content 
+     * @param paperOption 
+     * @param axiosOption 
+     */
+    abstract generateExaminationPaperFromContent(content: string, paperOption: any, axiosOption: any): Promise<ExaminationPaperResult>;
+}

package/src/gptprovider.ts

@@ -0,0 +1,38 @@
+//ts check
+/**
+* 语音转文字服务商工厂
+*/
+import OpenAIGpt from './openai';
+import AzureAI from './azureai'
+import BaiduWenXinAI, { ApiCredential } from './baiduai'
+import GptBase from './gptbase';
+/**
+ * OpenAI/NLP 的服务提供商 OpenAI,微软，百度文心(待接入)，google(待接入)
+ */
+export const GptProviderEnum = {
+    OPENAI: 'openai',
+    MICROSOFT: 'microsoft',
+    BAIDU: 'baidu',
+    GOOGLE:'google'
+} as const;
+export type GptProviderEnum = typeof GptProviderEnum[keyof typeof GptProviderEnum];
+/**
+ * 根据类型创建不同的TTS引擎对象
+ * @param {*} provider 
+ * @param {*} apikey 
+ * @param {*} setting 
+ * @returns 
+ */
+export function createGpt(provider: GptProviderEnum, apikey: string|ApiCredential, setting: any): GptBase | null {
+    let { model, maxtoken, temperature,endpoint,engine,version } = setting || {};
+    switch (provider) {
+        case GptProviderEnum.OPENAI: 
+            return new OpenAIGpt(apikey+'', { model, maxtoken, temperature });
+        case GptProviderEnum.MICROSOFT: 
+            return new AzureAI(apikey+'', { endpoint, engine, version }, { model, maxtoken, temperature }, );
+        case GptProviderEnum.BAIDU:
+            let cred: ApiCredential = typeof (apikey) === 'string' ? { apikey, securitykey: apikey } : apikey
+            return new BaiduWenXinAI(cred);
+        default: return null;
+    }
+};

package/src/index.ts

@@ -1 +1,2 @@
-export * as AIFactory from "./openaiprovider";
+export * as GptFactory from "./gptprovider";
+export * as GptBase from "./gptbase";