doomiaichat 5.1.0 → 6.0.0

package/src/openaibase.ts

@@ -0,0 +1,30 @@
+// import { Configuration, OpenAIApi, ChatCompletionRequestMessage } from "azure-openai"
+import { OpenAIApiParameters } from "./declare";
+import GptBase from "./gptbase"
+export default abstract class OpenAIBase<T> extends GptBase {
+    protected readonly apiKey: string;
+    protected readonly chatModel: string;
+    protected readonly maxtoken: number;
+    protected readonly top_p: number;
+    protected readonly presence_penalty: number;
+    protected readonly frequency_penalty: number;
+    protected readonly temperature: number;
+    protected readonly embeddingmodel: string;
+    protected aiApi: T | undefined;//OpenAIApi | undefined;
+    constructor(apiKey: string, apiOption: OpenAIApiParameters = {}) {
+        super();
+        this.apiKey = apiKey;
+        this.chatModel = apiOption.model || 'gpt-3.5-turbo';
+        this.maxtoken = apiOption.maxtoken || 2048;
+        this.top_p = apiOption.top_p || 0.95;
+        this.temperature = apiOption.temperature || 0.9;
+        this.presence_penalty = apiOption.presence_penalty || 0;
+        this.frequency_penalty = apiOption.frequency_penalty || 0;
+        this.embeddingmodel = apiOption.embedding || 'text-embedding-ada-002';
+    }
+    /**
+     * 初始化OpenAI 的聊天对象Api
+     */
+    abstract createOpenAI(apiKey: string): T ;
+
+}

package/src/openaiproxy.ts

@@ -0,0 +1,89 @@
+import { ApiResult, OpenAIApiParameters, ProxyPatameters } from './declare';
+import OpenAIGpt from './openai';
+import axios  from 'axios';
+const ERROR_RESPONSE: string[] = ['[AUTHORIZATION NEEDED]', '[AUTHORIZATION ERROR]', '[BODY ERROR]','[REQUEST ERROR]']
+export default class OpenAIProxy extends OpenAIGpt {
+    protected readonly proxySetting: ProxyPatameters;
+    constructor(apiKey: string, proxyOption: ProxyPatameters, apiOption: OpenAIApiParameters = {}) {
+        super(apiKey, apiOption);
+        this.proxySetting = proxyOption;
+    }
+    /**
+     * 重写chatRequest方法
+     * @param chatText 
+     * @param callChatOption 
+     * @param axiosOption 
+     */
+    override async chatRequest(chatText: string | any[], _paramOption: any, axiosOption?: any): Promise<ApiResult> {
+        const opts: any = {
+            headers: {
+                'Content-Type': 'application/json',
+                'authorization': `Bearer ${this.apiKey}`
+            },
+            method: 'post',
+            url: this.proxySetting.serviceurl,
+            data: {
+                chatText,
+                option: _paramOption
+            },
+            ...axiosOption
+        }
+        const requestResult = await axios(opts);
+        return requestResult.data as ApiResult;
+    }
+    /**
+     * 重写chatRequestInStream方法
+     * @param chatText 
+     * @param callChatOption 
+     * @param attach 
+     * @param axiosOption 
+     */
+    override async chatRequestInStream(chatText:Array<any>, callChatOption: OpenAIApiParameters, attach?: any, axiosOption?: any): Promise<any> {
+        // const decoder = new TextDecoder();
+        //overContent,
+        let  streamText,requestid = Math.ceil(Math.random() * (new Date().getTime() * Math.random()) / 1000);
+        const opts:any = {
+            headers: {
+                'Content-Type': 'application/json',
+                'authorization': `Bearer ${this.apiKey}`
+            },
+            method: 'post',
+            url: this.proxySetting.serviceurl +'/stream',
+            data: {
+                messages:chatText,
+                option: callChatOption,
+                axiosOption
+            },
+            responseType: 'stream',
+        }
+        axios(opts)
+        .then(res => {
+            res.data.on('data', (chunk:any) => { 
+                streamText = chunk.toString(); //decoder.decode(chunk);
+                if (streamText){
+                    ///请求的响应发生了错误
+                    if (ERROR_RESPONSE.includes(streamText)) {
+                        return this.emit('requesterror', { successed: false, requestid, error: 'Request Remote OpenAI Error : ' + streamText });
+                    }
+                    ///已经全部结束了
+                    if (streamText==='[END]') {
+                        return this.emit('chatdone', streamText || {})
+                    } else{
+                        ///持续的文字输出中
+                        try{
+                            //streamText = JSON.parse(streamText);
+                            // overContent = JSON.parse(streamText);
+                            return this.emit('chattext', Object.assign(streamText, attach));
+                        }catch(errParse){
+                            return this.emit('chaterror', { successed: false, requestid, error: 'JSON parse stream message', streamText  });
+                        }
+                    }
+                }
+                return;
+            });
+            //res.data.on('end', () => { this.emit('chatdone', Object.assign(streamText, attach)); });
+        }).catch(err=>{
+            this.emit('requesterror', { successed: false, requestid, error: 'Axios Error : ' + err });
+        })
+    }
+}

package/src/stabilityai.ts

@@ -1,4 +1,4 @@
-import { ApiResult, request, AzureOpenAIPatameters } from "./declare";
+import { request, AzureOpenAIPatameters, StabilityOption, StabilityResult } from "./declare";
 import GptBase from "./gptbase"
 export default class StabilityAI extends GptBase {
     
@@ -65,24 +65,3 @@ export default class StabilityAI extends GptBase {
 
     }
 }
-
-export interface StabilityOption {
-    'cfg_scale'?: number,
-    'clip_guidance_preset'?: string,
-    'height'?: number,
-    'width'?: number,
-    'samples'?: number,
-    'seed'?:number,
-    'steps'?: number,
-    'sampler'?:string,
-    'negative'?:string,
-    'engine'?:string,
-    'endpoint'?:string
-    'denoising_strength'?:number,
-    'hr_scale'?:number
-}
-
-export interface StabilityResult extends ApiResult {
-    'data'?:any;
-    'type'?:string;
-}

package/src/stabilityplusai.ts

@@ -1,11 +1,23 @@
 
-import { request } from "./declare";
-import StabilityAI, { StabilityOption, StabilityResult } from "./stabilityai"
-export default class StabilityPlusAI extends StabilityAI {
+import { AzureOpenAIPatameters, StabilityOption, StabilityResult, request } from "./declare";
+import GptBase from "./gptbase";
+export default class StabilityPlusAI extends GptBase {
+    protected readonly apiKey: string;
+    // protected readonly apiSetting: AzureOpenAIPatameters
+    protected readonly apiOption: StabilityOption;
+    constructor(apiKey: string, _urlOption: AzureOpenAIPatameters, apiOption: StabilityOption = {}) {
+        super();
+        this.apiKey = apiKey;
+        // this.apiSetting = urlOption;
+        this.apiOption = apiOption;
+        // if (!this.apiSetting.endpoint.toLowerCase().startsWith('http')) {
+        //     this.apiSetting.endpoint = 'https://' + this.apiSetting.endpoint;
+        // }
+    }
     /**
      * 请求Stable作画的接口
      */
-    public override async chatRequest(chatText: string, paramOption: StabilityOption, axiosOption: any = {}): Promise<StabilityResult> {
+    public async chatRequest(chatText: string, paramOption: StabilityOption, axiosOption: any = {}): Promise<StabilityResult> {
         if (!chatText) return { successed: false, error: { errcode: 2, errmsg: '缺失聊天的内容' } };
         axiosOption = Object.assign({}, axiosOption, {
             headers: {
@@ -14,10 +26,9 @@ export default class StabilityPlusAI extends StabilityAI {
             }
         })
         try {
-            let param = {
+            const requestOption = {
                 ...axiosOption,
-                method: "post",
-                url: `${paramOption.endpoint || this.apiSetting.endpoint}/sdapi/v1/txt2img`,
+                method: "POST",
                 data: {
                     "enable_hr": false,
                     "denoising_strength": paramOption.denoising_strength || this.apiOption.denoising_strength ||  0.5,
@@ -46,25 +57,16 @@ export default class StabilityPlusAI extends StabilityAI {
                     "tiling": false,
                     "do_not_save_samples": false,
                     "do_not_save_grid": false,
-                    "negative_prompt": paramOption.negative || '',
-                    // "eta": 0,
-                    // "s_churn": 0,
-                    // "s_tmax": 0,
-                    // "s_tmin": 0,
-                    // "s_noise": 1,
-                    // "script_args": [],
-                    // "sampler_index": paramOption.sampler || this.apiOption.sampler || "Euler", //"Euler",
-                    // "send_images": true,
-                    // "save_images": false,
-                    // "alwayson_scripts": {}
+                    "negative_prompt": paramOption.negative || ''
                 },
-
+                url: `${paramOption.endpoint}/sdapi/v1/txt2img`,
             };
-            const response: any = await request(param)
+            // console.log('stablity param', requestOption);
+            const response: any = await request(requestOption)
             if (response.successed) {
                 return { successed: true, type: 'image', data: response.data.images, };
             }
-            console.log('response result ', response.data)
+            // console.log('response result ', response.data)
             return { successed: false, ...response.data };
         } catch (error) {
             console.log('result is error ', error)