doomiaichat 1.4.3 → 1.4.5

package/dist/index.d.ts

@@ -39,6 +39,12 @@ export declare class AIChat extends EventEmitter {
      * @param {需要出来的数量} count
      */
     getSimilarityContent(content: string, count?: number, axiosOption?: AxiosRequestConfig): Promise<ChatReponse>;
+    /**
+     * 提取内容的中心思想摘要
+     * @param content
+     * @param axiosOption
+     */
+    getSummaryOfContent(content: string | Array<any>, axiosOption?: AxiosRequestConfig): Promise<SummaryReponse>;
     /**
      * 从指定的文本内容中生成相关的问答
      * @param {*} content
@@ -97,6 +103,17 @@ export interface ChatReponse extends ApiResult {
      */
     'message'?: Array<any>;
 }
+/**
+ * 摘要信息
+ */
+export interface SummaryReponse extends ApiResult {
+    /**
+     * The name of the user in a multi-user chat
+     * @type {Array<any>}
+     * @memberof SummaryReponse
+     */
+    'summary'?: Array<string>;
+}
 /**
  * 调用OpenAI Api的参数约定
  */

package/dist/index.js

@@ -47,7 +47,7 @@ class AIChat extends events_1.EventEmitter {
                 return { successed: false, error: { errcode: 1, errmsg: '聊天机器人无效' } };
             let message = typeof (chatText) == 'string' ?
                 [{ role: 'user', content: chatText }] : chatText;
-            console.log('message', message);
+            // console.log('message', message)
             try {
                 const response = yield this.chatRobot.createChatCompletion({
                     model: (callChatOption === null || callChatOption === void 0 ? void 0 : callChatOption.model) || this.chatModel,
@@ -135,6 +135,36 @@ class AIChat extends events_1.EventEmitter {
             return { successed: true, message: replys };
         });
     }
+    /**
+     * 提取内容的中心思想摘要
+     * @param content
+     * @param axiosOption
+     */
+    getSummaryOfContent(content, axiosOption = {}) {
+        return __awaiter(this, void 0, void 0, function* () {
+            const arrContent = typeof (content) == 'string' ? this.splitLongText(content) : content;
+            let summary = [];
+            while (arrContent.length > 0) {
+                let subarray = arrContent.slice(0, MESSAGE_LENGTH);
+                subarray.push({ role: 'user', content: '根据上述内容精简，提炼摘要内容，并将摘要逐条放入JSON数组结构输出' });
+                let result = yield this.chatRequest(subarray, {}, axiosOption);
+                if (result.successed && result.message) {
+                    try {
+                        let jsonObjItems = JSON.parse(result.message[0].content);
+                        console.log('result.message[0].content', result.message[0].content);
+                        if (Array.isArray(jsonObjItems))
+                            summary = summary.concat(jsonObjItems);
+                    }
+                    catch (error) {
+                        console.log('result.message[0].content', error, result.message[0].content);
+                    }
+                }
+                ////删除已经处理的文本
+                arrContent.splice(0, MESSAGE_LENGTH);
+            }
+            return { successed: true, summary };
+        });
+    }
     /**
      * 从指定的文本内容中生成相关的问答
      * @param {*} content

package/package.json

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

package/src/index.ts

@@ -40,7 +40,7 @@ export class AIChat extends EventEmitter {
 
         let message: Array<ChatCompletionRequestMessage> = typeof(chatText)=='string'?
             [{ role: 'user', content: chatText}] : chatText;
-        console.log('message', message)
+        // console.log('message', message)
         try {
             const response: AxiosResponse<CreateChatCompletionResponse, any>
                 = await this.chatRobot.createChatCompletion({
@@ -123,7 +123,32 @@ export class AIChat extends EventEmitter {
         let replys = result.message.map(item => { return item.message.content.trim(); })
         return { successed: true, message: replys }
     }
-
+    /**
+     * 提取内容的中心思想摘要
+     * @param content 
+     * @param axiosOption 
+     */
+    async getSummaryOfContent(content: string|Array<any>, axiosOption: AxiosRequestConfig = {}): Promise<SummaryReponse> {
+        const arrContent = typeof (content) == 'string' ? this.splitLongText(content) : content;
+        let summary :Array<string> = [];
+        while (arrContent.length > 0 ) {
+            let subarray = arrContent.slice(0, MESSAGE_LENGTH);
+            subarray.push({ role: 'user', content: '根据上述内容精简，提炼摘要内容，并将摘要逐条放入JSON数组结构输出' })
+            let result = await this.chatRequest(subarray, {}, axiosOption);
+            if (result.successed && result.message){
+                try{
+                    let jsonObjItems = JSON.parse(result.message[0].content)
+                    console.log('result.message[0].content', result.message[0].content)
+                    if (Array.isArray(jsonObjItems))  summary = summary.concat(jsonObjItems)
+                }catch(error){
+                    console.log('result.message[0].content',error, result.message[0].content)
+                }
+            }
+            ////删除已经处理的文本
+            arrContent.splice(0, MESSAGE_LENGTH);
+        }
+        return { successed: true, summary }
+    }
     /**
      * 从指定的文本内容中生成相关的问答
      * @param {*} content 
@@ -342,6 +367,18 @@ export interface ChatReponse extends ApiResult {
      */
     'message'?: Array<any>;
 }
+/**
+ * 摘要信息
+ */
+export interface SummaryReponse extends ApiResult {
+
+    /**
+     * The name of the user in a multi-user chat
+     * @type {Array<any>}
+     * @memberof SummaryReponse
+     */
+    'summary'?:Array<string>;
+}
 /**
  * 调用OpenAI Api的参数约定
  */