doomiaichat 1.0.0 → 1.0.1

package/package.json

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

package/src/index.ts

@@ -1,7 +1,7 @@
-import { Configuration, OpenAIApi, CreateChatCompletionResponse } from "openai"
+import { Configuration, OpenAIApi, CreateChatCompletionResponse, ChatCompletionRequestMessage } from "openai"
 import { EventEmitter } from "events";
 import { AxiosRequestConfig, AxiosResponse } from "axios";
-export default class AIChat extends EventEmitter {
+export class AIChat extends EventEmitter {
     private readonly chatRobot: OpenAIApi;
     private readonly chatModel: string;
     private readonly maxtoken: number;
@@ -23,22 +23,48 @@ export default class AIChat extends EventEmitter {
      * 向OpenAI发送一个聊天请求
      * @param {*} chatText 
      */
-    public async chatRequest(chatText: string | any, axiosOption: AxiosRequestConfig):Promise<any> {
-        if (!chatText) return { successed: false, errcode: 2, errmsg: '缺失聊天的内容' }
-        if (!this.chatRobot) return { successed: false, errcode: 1, errmsg: '聊天机器人无效' }
-        try{
+    public async chatRequest(chatText: string | Array<any>, axiosOption: AxiosRequestConfig): Promise<ChatReponse> {
+        if (!chatText) return {successed:false, error:{ errcode: 2, errmsg: '缺失聊天的内容' }};
+            if(!this.chatRobot) return {successed: false, error: { errcode: 1, errmsg: '聊天机器人无效' }};
+
+        let message: Array<ChatCompletionRequestMessage> = typeof(chatText)=='string'?
+            [{ role: 'system', content: chatText + '' }] : chatText;
+        try {
             const response: AxiosResponse<CreateChatCompletionResponse, any>
                 = await this.chatRobot.createChatCompletion({
                     model: this.chatModel,
-                    messages: chatText,
+                    messages: message,
                     temperature: this.temperature,
                     max_tokens: this.maxtoken
                     // n: this.replyCounts
                 }, axiosOption);
-            return { successed: true, message: response.data.choices }
-        }catch(error){
-            return { successed: false,error }
+            return {successed:true,message: response.data.choices};
+        } catch (error) {
+            return { successed: false, error }; 
         }
-       
+
     }
 }
+
+/**
+ * Api封装后的返回结果
+ */
+export interface ChatReponse {
+    /**
+     * return the result of api called
+     * @type {boolean}
+     */
+    'successed': boolean
+    /**
+     * The error info 
+     * @type {any}
+     * @memberof ChatReponse
+     */
+    'error'?: any;
+    /**
+     * The name of the user in a multi-user chat
+     * @type {Array<any>}
+     * @memberof ChatReponse
+     */
+    'message'?: Array<any>;
+}