doomiaichat 3.6.0 → 4.0.0

package/dist/baiduai.d.ts

@@ -1,4 +1,4 @@
-import { EmotionResult, SimilarityResult, ChatReponse, SummaryReponse, ExaminationPaperResult, ApiResult, CacheProvider, CommentResult } from "./declare";
+import { ApiResult, CacheProvider } from "./declare";
 import GptBase from "./gptbase";
 export default class BaiduWenXinAI extends GptBase {
     protected credential: ApiCredential;
@@ -14,23 +14,10 @@ export default class BaiduWenXinAI extends GptBase {
      * 获取调用Api的Token
      */
     getAccessToken(): Promise<AccessTokenResult>;
-    /**
-     * 获得文字的向量
-     * @param text
-     */
-    getTextEmbedding(_text: string, _axiosOption: any): Promise<any>;
     /**
      * 请求GPT接口
      */
     chatRequest(chatText: string | Array<any>, _paramOption: any, axiosOption?: any): Promise<ApiResult>;
-    chatRequestInStream(_chatText: string | any[], _paramOption: any, _attach?: any, _axiosOption?: any): void;
-    commentQuestionAnswer(_question: string, _answer: string, _axiosOption: any): Promise<CommentResult>;
-    getScentenceEmotional(_s1: string, _axiosOption: any): Promise<EmotionResult>;
-    getScentenseSimilarity(_s1: string, _s2: string, _axiosOption: any): Promise<SimilarityResult>;
-    getSimilarityContent(_content: string, _count: number, _axiosOption: any): Promise<ChatReponse>;
-    getSummaryOfContent(_content: string | any[], _axiosOption: any): Promise<SummaryReponse>;
-    generateQuestionsFromContent(_content: string, _count: number, _axiosOption: any): Promise<ChatReponse>;
-    generateExaminationPaperFromContent(_content: string, _paperOption: any, _axiosOption: any): Promise<ExaminationPaperResult>;
 }
 export interface AccessTokenResult extends ApiResult {
     'access_token'?: string;

package/dist/baiduai.js

@@ -53,15 +53,6 @@ class BaiduWenXinAI extends gptbase_1.default {
             return { successed: false, error: tokenData.data.msg };
         });
     }
-    /**
-     * 获得文字的向量
-     * @param text
-     */
-    getTextEmbedding(_text, _axiosOption) {
-        return __awaiter(this, void 0, void 0, function* () {
-            return null;
-        });
-    }
     /**
      * 请求GPT接口
      */
@@ -97,29 +88,5 @@ class BaiduWenXinAI extends gptbase_1.default {
             }
         });
     }
-    chatRequestInStream(_chatText, _paramOption, _attach, _axiosOption) {
-        throw new Error("Method not implemented.");
-    }
-    commentQuestionAnswer(_question, _answer, _axiosOption) {
-        throw new Error("Method not implemented.");
-    }
-    getScentenceEmotional(_s1, _axiosOption) {
-        throw new Error("Method not implemented.");
-    }
-    getScentenseSimilarity(_s1, _s2, _axiosOption) {
-        throw new Error("Method not implemented.");
-    }
-    getSimilarityContent(_content, _count, _axiosOption) {
-        throw new Error("Method not implemented.");
-    }
-    getSummaryOfContent(_content, _axiosOption) {
-        throw new Error("Method not implemented.");
-    }
-    generateQuestionsFromContent(_content, _count, _axiosOption) {
-        throw new Error("Method not implemented.");
-    }
-    generateExaminationPaperFromContent(_content, _paperOption, _axiosOption) {
-        throw new Error("Method not implemented.");
-    }
 }
 exports.default = BaiduWenXinAI;

package/dist/declare.js

@@ -28,7 +28,7 @@ function request(opts = {}) {
             return { successed: true, data: result.data };
         }
         catch (err) {
-            return { successed: false, error: err };
+            return { successed: false, error: err, data: err };
         }
     });
 }

package/dist/gptbase.d.ts

@@ -1,6 +1,6 @@
 /// <reference types="node" />
 import { EventEmitter } from "events";
-import { ChatReponse, SummaryReponse, ExaminationPaperResult, EmotionResult, SimilarityResult, ApiResult, CommentResult } from './declare';
+import { ApiResult } from './declare';
 export default abstract class GptBase extends EventEmitter {
     /**
      * 构造函数
@@ -10,7 +10,7 @@ export default abstract class GptBase extends EventEmitter {
      * 获得文字的向量
      * @param text
      */
-    abstract getTextEmbedding(text: string, axiosOption: any): Promise<any>;
+    getTextEmbedding(_text: string, _axiosOption: any): any;
     /**
      * 自由聊天模式
      * @param chatText
@@ -24,52 +24,52 @@ export default abstract class GptBase extends EventEmitter {
      * @param _paramOption
      * @param axiosOption
      */
-    abstract chatRequestInStream(chatText: string | Array<any>, _paramOption: any, attach?: any, axiosOption?: any): any;
+    chatRequestInStream(_chatText: string | Array<any>, _paramOption: any, _attach?: any, _axiosOption?: any): any;
     /**
      * 点评问题回答的评价
      * @param question  问题题干
      * @param answer 回答内容
      * @param axiosOption
      */
-    abstract commentQuestionAnswer(question: string, answer: string, axiosOption: any): Promise<CommentResult>;
+    commentQuestionAnswer(_question: string, _answer: string, _axiosOption: any): any;
     /**
      * 获取句子的情感
      * @param s1
      * @param axiosOption
      */
-    abstract getScentenceEmotional(s1: string, axiosOption: any): Promise<EmotionResult>;
+    getScentenceEmotional(_s1: string, _axiosOption: any): any;
     /**
      * 获取两段文本的相似度
      * @param s1
      * @param s2
      * @param axiosOption
      */
-    abstract getScentenseSimilarity(s1: string, s2: string, axiosOption: any): Promise<SimilarityResult>;
+    getScentenseSimilarity(_s1: string, _s2: string, _axiosOption: any): any;
     /**
      * 获取一段文本的相似内容
      * @param content
      * @param count
      * @param axiosOption
      */
-    abstract getSimilarityContent(content: string, count: number, axiosOption: any): Promise<ChatReponse>;
+    getSimilarityContent(_content: string, _count: number, _axiosOption: any): any;
     /**
      * 获取内容的摘要
      * @param content
      * @param axiosOption
      */
-    abstract getSummaryOfContent(content: string | Array<any>, axiosOption: any): Promise<SummaryReponse>;
+    getSummaryOfContent(_content: string | Array<any>, _axiosOption: any): any;
     /**
      * 从内容中提取问题和答案及管件子
      * @param content
      * @param count
      * @param axiosOption
      */
-    abstract generateQuestionsFromContent(content: string, count: number, everyContentLength: number, axiosOption: any): Promise<ChatReponse>;
+    generateQuestionsFromContent(_content: string, _count: number, _everyContentLength: number, _axiosOption: any): any;
     /**
      * 从内容中提取单选多选判断填空题
      * @param content
      * @param paperOption
      * @param axiosOption
      */
-    abstract generateExaminationPaperFromContent(content: string, paperOption: any, everyContentLength: number, axiosOption: any): Promise<ExaminationPaperResult>;
+    generateExaminationPaperFromContent(_content: string, _paperOption: any, _everyContentLength: number, _axiosOption: any): any;
 }

package/dist/gptbase.js

@@ -8,5 +8,64 @@ class GptBase extends events_1.EventEmitter {
     constructor() {
         super();
     }
+    /**
+     * 获得文字的向量
+     * @param text
+     */
+    getTextEmbedding(_text, _axiosOption) { return null; }
+    /**
+     * 流式的聊天模式
+     * @param chatText
+     * @param _paramOption
+     * @param axiosOption
+     */
+    chatRequestInStream(_chatText, _paramOption, _attach, _axiosOption) { return null; }
+    /**
+     * 点评问题回答的评价
+     * @param question  问题题干
+     * @param answer 回答内容
+     * @param axiosOption
+     */
+    commentQuestionAnswer(_question, _answer, _axiosOption) { return null; }
+    /**
+     * 获取句子的情感
+     * @param s1
+     * @param axiosOption
+     */
+    getScentenceEmotional(_s1, _axiosOption) { return null; }
+    /**
+     * 获取两段文本的相似度
+     * @param s1
+     * @param s2
+     * @param axiosOption
+     */
+    getScentenseSimilarity(_s1, _s2, _axiosOption) { return null; }
+    /**
+     * 获取一段文本的相似内容
+     * @param content
+     * @param count
+     * @param axiosOption
+     */
+    getSimilarityContent(_content, _count, _axiosOption) { return null; }
+    /**
+     * 获取内容的摘要
+     * @param content
+     * @param axiosOption
+     */
+    getSummaryOfContent(_content, _axiosOption) { return null; }
+    /**
+     * 从内容中提取问题和答案及管件子
+     * @param content
+     * @param count
+     * @param axiosOption
+     */
+    generateQuestionsFromContent(_content, _count, _everyContentLength, _axiosOption) { return null; }
+    /**
+     * 从内容中提取单选多选判断填空题
+     * @param content
+     * @param paperOption
+     * @param axiosOption
+     */
+    generateExaminationPaperFromContent(_content, _paperOption, _everyContentLength, _axiosOption) { return null; }
 }
 exports.default = GptBase;

package/dist/gptprovider.d.ts

@@ -8,6 +8,7 @@ export declare const GptProviderEnum: {
     readonly MICROSOFT: "microsoft";
     readonly BAIDU: "baidu";
     readonly GOOGLE: "google";
+    readonly STABILITY: "stability";
 };
 export type GptProviderEnum = typeof GptProviderEnum[keyof typeof GptProviderEnum];
 /**

package/dist/gptprovider.js

@@ -10,6 +10,7 @@ exports.createGpt = exports.GptProviderEnum = void 0;
 */
 const openai_1 = __importDefault(require("./openai"));
 const azureai_1 = __importDefault(require("./azureai"));
+const stabilityai_1 = __importDefault(require("./stabilityai"));
 const baiduai_1 = __importDefault(require("./baiduai"));
 /**
  * OpenAI/NLP 的服务提供商 OpenAI,微软，百度文心(待接入)，google(待接入)
@@ -18,7 +19,8 @@ exports.GptProviderEnum = {
     OPENAI: 'openai',
     MICROSOFT: 'microsoft',
     BAIDU: 'baidu',
-    GOOGLE: 'google'
+    GOOGLE: 'google',
+    STABILITY: 'stability'
 };
 /**
  * 根据类型创建不同的TTS引擎对象
@@ -37,6 +39,8 @@ function createGpt(provider, apikey, setting) {
         case exports.GptProviderEnum.BAIDU:
             let cred = typeof (apikey) === 'string' ? { apikey, securitykey: apikey } : apikey;
             return new baiduai_1.default(cred);
+        case exports.GptProviderEnum.STABILITY:
+            return new stabilityai_1.default(apikey + '', { endpoint, engine }, setting);
         default: return null;
     }
 }

package/dist/stabilityai.d.ts

@@ -0,0 +1,30 @@
+import { ApiResult, AzureOpenAIPatameters } from "./declare";
+import GptBase from "./gptbase";
+export default class StabilityAI extends GptBase {
+    protected readonly apiKey: string;
+    protected readonly apiSetting: AzureOpenAIPatameters;
+    protected readonly apiOption: StabilityOption;
+    /**
+     *
+     * @param apiKey 调用OpenAI 的key
+     * @param azureOption  用作accesstoken的缓存
+     * @param apiOption  用作accesstoken的缓存
+     */
+    constructor(apiKey: string, urlOption: AzureOpenAIPatameters, apiOption?: StabilityOption);
+    /**
+     * 请求Stable作画的接口
+     */
+    chatRequest(chatText: string, paramOption: StabilityOption, axiosOption?: any): Promise<StabilityResult>;
+}
+export interface StabilityOption {
+    'cfg_scale'?: number;
+    'clip_guidance_preset'?: string;
+    'height'?: number;
+    'width'?: number;
+    'samples'?: number;
+    'steps'?: number;
+}
+export interface StabilityResult extends ApiResult {
+    'data'?: any;
+    'type'?: string;
+}

package/dist/stabilityai.js

@@ -0,0 +1,76 @@
+"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 });
+const declare_1 = require("./declare");
+const gptbase_1 = __importDefault(require("./gptbase"));
+class StabilityAI extends gptbase_1.default {
+    /**
+     *
+     * @param apiKey 调用OpenAI 的key
+     * @param azureOption  用作accesstoken的缓存
+     * @param apiOption  用作accesstoken的缓存
+     */
+    constructor(apiKey, urlOption, apiOption = {}) {
+        super();
+        this.apiKey = apiKey;
+        this.apiSetting = urlOption;
+        this.apiOption = apiOption;
+        if (!this.apiSetting.endpoint.toLowerCase().startsWith('https://') &&
+            !this.apiSetting.endpoint.toLowerCase().startsWith('https://')) {
+            this.apiSetting.endpoint = 'https://' + this.apiSetting.endpoint;
+        }
+    }
+    /**
+     * 请求Stable作画的接口
+     */
+    chatRequest(chatText, paramOption, axiosOption = {}) {
+        return __awaiter(this, void 0, void 0, function* () {
+            if (!chatText)
+                return { successed: false, error: { errcode: 2, errmsg: '缺失聊天的内容' } };
+            axiosOption = Object.assign({}, axiosOption, {
+                headers: {
+                    "Content-Type": "application/json",
+                    "Accept": "application/json",
+                    "Authorization": `Bearer ${this.apiKey}`
+                }
+            });
+            try {
+                let param = Object.assign(Object.assign({}, axiosOption), { method: "post", url: `${this.apiSetting.endpoint}/v1/generation/${this.apiSetting.engine}/text-to-image`, data: {
+                        text_prompts: [
+                            {
+                                text: chatText
+                            }
+                        ],
+                        cfg_scale: paramOption.cfg_scale || this.apiOption.cfg_scale || 7,
+                        clip_guidance_preset: paramOption.clip_guidance_preset || this.apiOption.clip_guidance_preset || "FAST_BLUE",
+                        height: paramOption.height || this.apiOption.height || 512,
+                        width: paramOption.width || this.apiOption.width || 512,
+                        samples: paramOption.samples || this.apiOption.samples || 1,
+                        steps: paramOption.steps || this.apiOption.steps || 30,
+                    } });
+                const response = yield (0, declare_1.request)(param);
+                if (response.successed) {
+                    let data = response.data;
+                    return { successed: true, type: 'image', data: data.artifacts, };
+                }
+                return Object.assign({ successed: false }, response.data);
+            }
+            catch (error) {
+                console.log('result is error ', error);
+                return { successed: false, error };
+            }
+        });
+    }
+}
+exports.default = StabilityAI;

package/package.json

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

package/src/baiduai.ts

@@ -1,4 +1,4 @@
-import { EmotionResult, SimilarityResult, ChatReponse, SummaryReponse, ExaminationPaperResult, ApiResult, CacheProvider, request, CommentResult } from "./declare";
+import { ApiResult, CacheProvider, request } from "./declare";
 import GptBase from "./gptbase"
 const TOKEN_CACHE_KEY = "key:_doomisoft:baiduwenxin:"
 export default class BaiduWenXinAI extends GptBase {
@@ -39,13 +39,7 @@ export default class BaiduWenXinAI extends GptBase {
         }
         return { successed: false, error: tokenData.data.msg }
     }
-    /**
-     * 获得文字的向量
-     * @param text 
-     */
-    async getTextEmbedding(_text: string, _axiosOption: any): Promise<any> {
-        return null;
-    }
+   
     /**
      * 请求GPT接口
      */
@@ -82,32 +76,6 @@ export default class BaiduWenXinAI extends GptBase {
         }
 
     }
-
-    chatRequestInStream(_chatText: string | any[], _paramOption: any,_attach?:any, _axiosOption?: any): void {
-        throw new Error("Method not implemented.");
-    }
-    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

package/src/declare.ts

@@ -139,7 +139,7 @@ export async function request(opts: any = {}): Promise<RpcResult> {
         let result = await axios(opts);
         return { successed: true, data: result.data }
     } catch (err) {
-        return { successed: false, error: err }
+        return { successed: false, error: err,data:err }
     }
 }
 

package/src/gptbase.ts

@@ -1,5 +1,5 @@
 import { EventEmitter } from "events"
-import { ChatReponse, SummaryReponse, ExaminationPaperResult, EmotionResult, SimilarityResult, ApiResult, CommentResult } from './declare'
+import { ApiResult } from './declare'
 
 export default abstract class GptBase extends EventEmitter {
     /**
@@ -12,7 +12,7 @@ export default abstract class GptBase extends EventEmitter {
      * 获得文字的向量
      * @param text 
      */
-    abstract getTextEmbedding(text: string, axiosOption: any):Promise<any>;
+    getTextEmbedding(_text: string, _axiosOption: any):any {return null;}
     /**
      * 自由聊天模式
      * @param chatText 
@@ -26,52 +26,52 @@ export default abstract class GptBase extends EventEmitter {
      * @param _paramOption 
      * @param axiosOption 
      */
-    abstract chatRequestInStream(chatText: string | Array<any>, _paramOption: any,attach?:any, axiosOption?: any):any;
+    chatRequestInStream(_chatText: string | Array<any>, _paramOption: any, _attach?: any, _axiosOption?: any): any { return null; }
     /**
      * 点评问题回答的评价
      * @param question  问题题干
      * @param answer 回答内容
      * @param axiosOption 
      */
-    abstract commentQuestionAnswer(question: string, answer: string, axiosOption: any): Promise<CommentResult>;
+    commentQuestionAnswer(_question: string, _answer: string, _axiosOption: any): any { return null; }
     /**
      * 获取句子的情感
      * @param s1 
      * @param axiosOption 
      */
-    abstract getScentenceEmotional(s1: string, axiosOption: any ): Promise<EmotionResult> ;
+    getScentenceEmotional(_s1: string, _axiosOption: any): any { return null; }
     /**
      * 获取两段文本的相似度
      * @param s1 
      * @param s2 
      * @param axiosOption 
      */
-    abstract getScentenseSimilarity(s1: string, s2: string, axiosOption: any): Promise<SimilarityResult>;
+    getScentenseSimilarity(_s1: string, _s2: string, _axiosOption: any): any { return null; }
     /**
      * 获取一段文本的相似内容
      * @param content 
      * @param count 
      * @param axiosOption 
      */
-    abstract getSimilarityContent(content: string, count: number, axiosOption: any): Promise<ChatReponse>;
+    getSimilarityContent(_content: string, _count: number, _axiosOption: any): any { return null; }
     /**
      * 获取内容的摘要
      * @param content 
      * @param axiosOption 
      */
-    abstract getSummaryOfContent(content: string | Array<any>, axiosOption: any): Promise<SummaryReponse>;
+    getSummaryOfContent(_content: string | Array<any>, _axiosOption: any): any { return null; }
     /**
      * 从内容中提取问题和答案及管件子
      * @param content 
      * @param count 
      * @param axiosOption 
      */
-    abstract generateQuestionsFromContent(content: string, count: number, everyContentLength: number, axiosOption: any): Promise<ChatReponse>;
+    generateQuestionsFromContent(_content: string, _count: number, _everyContentLength: number, _axiosOption: any): any { return null; }
     /**
      * 从内容中提取单选多选判断填空题
      * @param content 
      * @param paperOption 
      * @param axiosOption 
      */
-    abstract generateExaminationPaperFromContent(content: string, paperOption: any, everyContentLength: number, axiosOption: any): Promise<ExaminationPaperResult>;
+    generateExaminationPaperFromContent(_content: string, _paperOption: any, _everyContentLength: number, _axiosOption: any): any { return null; }
 }

package/src/gptprovider.ts

@@ -4,6 +4,7 @@
 */
 import OpenAIGpt from './openai';
 import AzureAI from './azureai'
+import StabilityAI from './stabilityai'
 import BaiduWenXinAI, { ApiCredential } from './baiduai'
 import GptBase from './gptbase';
 /**
@@ -13,7 +14,8 @@ export const GptProviderEnum = {
     OPENAI: 'openai',
     MICROSOFT: 'microsoft',
     BAIDU: 'baidu',
-    GOOGLE:'google'
+    GOOGLE:'google',
+    STABILITY:'stability'
 } as const;
 export type GptProviderEnum = typeof GptProviderEnum[keyof typeof GptProviderEnum];
 /**
@@ -33,6 +35,8 @@ export function createGpt(provider: GptProviderEnum, apikey: string|ApiCredentia
         case GptProviderEnum.BAIDU:
             let cred: ApiCredential = typeof (apikey) === 'string' ? { apikey, securitykey: apikey } : apikey
             return new BaiduWenXinAI(cred);
+        case GptProviderEnum.STABILITY:
+            return new StabilityAI(apikey + '', { endpoint, engine }, setting);
         default: return null;
     }
 };

package/src/openai.ts

@@ -46,7 +46,7 @@ export default class OpenAIGpt extends GptBase {
      * 获得文字的向量
      * @param text 
      */
-    async getTextEmbedding(text: string, axiosOption: any): Promise<EmbeddingResult>{
+    override async getTextEmbedding(text: string, axiosOption: any): Promise<EmbeddingResult>{
         if (!text) return { successed: false, error: { errcode: 2, errmsg: 'content required' } };
         if (!this.aiApi) {
             this.aiApi = this.createOpenAI(this.apiKey);
@@ -97,7 +97,7 @@ export default class OpenAIGpt extends GptBase {
      * @param _paramOption 
      * @param axiosOption 
      */
-    async chatRequestInStream(chatText: string | Array<any>, callChatOption: OpenAIApiParameters, attach?: any, axiosOption?: any):Promise<any>{
+    override async chatRequestInStream(chatText: string | Array<any>, callChatOption: OpenAIApiParameters, attach?: any, axiosOption?: any):Promise<any>{
         if (!chatText) this.emit('chaterror', { successed: false, error:'no text in chat'});
         if (!this.aiApi) {
             this.aiApi = this.createOpenAI(this.apiKey);
@@ -162,7 +162,7 @@ export default class OpenAIGpt extends GptBase {
      * @param answer 
      * @param axiosOption 
      */
-    async commentQuestionAnswer(question: string, answer: string, axiosOption: any = { timeout: 30000 }): Promise<CommentResult>{
+    override async commentQuestionAnswer(question: string, answer: string, axiosOption: any = { timeout: 30000 }): Promise<CommentResult>{
         if (!question || !answer) return { successed: false, error: { errcode: 2, errmsg: '缺失参数' } }
         let message = [
             {role:'system',content:'你是一名专业的知识点评师。'},
@@ -192,7 +192,7 @@ export default class OpenAIGpt extends GptBase {
      * @param {*} s1 
      * @param {*} axiosOption 
      */
-    async getScentenceEmotional(s1: string, axiosOption: any = { timeout: 30000 }): Promise<EmotionResult> {
+    override async getScentenceEmotional(s1: string, axiosOption: any = { timeout: 30000 }): Promise<EmotionResult> {
         if (!s1) return { successed: false, error: { errcode: 2, errmsg: '缺失参数' } }
         const emotion = ['愤怒', '威胁', '讽刺', '愧疚', '兴奋', '友好', '消极', '生气', '正常'];
         const messages = [
@@ -215,7 +215,7 @@ export default class OpenAIGpt extends GptBase {
      * @param {*} s1 
      * @param {*} s2 
      */
-    async getScentenseSimilarity(s1: string, s2: string, axiosOption: any = { timeout: 30000 }): Promise<SimilarityResult> {
+    override async getScentenseSimilarity(s1: string, s2: string, axiosOption: any = { timeout: 30000 }): Promise<SimilarityResult> {
         if (!s1 || !s2) return { successed: false, error: { errcode: 2, errmsg: '缺失参数' } }
         const messages = [
             { role: 'system', content: '你是一名专业的语言分析大师' },
@@ -243,7 +243,7 @@ export default class OpenAIGpt extends GptBase {
      * @param {*} content 
      * @param {需要出来的数量} count 
      */
-    async getSimilarityContent(content: string, count: number = 1, axiosOption: any = {}): Promise<ChatReponse> {
+    override async getSimilarityContent(content: string, count: number = 1, axiosOption: any = {}): Promise<ChatReponse> {
         let chnReg: boolean = /([\u4e00-\u9fa5]|[\ufe30-\uffa0])/.test(content)  ///检查源话是否含有中文内容
         let engReg: boolean = /[a-zA-Z]/.test(content)  ///检查源话是否含有英文内容
         ///如果源话是全中文，那么结果中不应该出来英文的相似说法，如果源话是全英文，则结果不能出现全中文的说法
@@ -259,7 +259,7 @@ export default class OpenAIGpt extends GptBase {
      * @param content 
      * @param axiosOption 
      */
-    async getSummaryOfContent(content: string | Array<any>, axiosOption: any = {}): Promise<SummaryReponse> {
+    override async getSummaryOfContent(content: string | Array<any>, axiosOption: any = {}): Promise<SummaryReponse> {
         const arrContent = typeof (content) == 'string' ? this.splitLongText(content) : content;
         let summary: Array<OutlineSummaryItem> = [];
         while (arrContent.length > 0) {
@@ -287,7 +287,7 @@ export default class OpenAIGpt extends GptBase {
      * @param {*} axiosOption 
      * @returns 
      *///并在答案末尾处必须给出答案内容中的关键词
-    async generateQuestionsFromContent(content: string, count: number = 1, everyContentLength: number = SECTION_LENGTH, axiosOption: any = {}): Promise<ChatReponse> {
+    override async generateQuestionsFromContent(content: string, count: number = 1, everyContentLength: number = SECTION_LENGTH, axiosOption: any = {}): Promise<ChatReponse> {
         let arrContent = this.splitLongText(content, everyContentLength || SECTION_LENGTH);
         ///没20句话分为一组，适应大文件内容多次请求组合结果
         ///每一句话需要产生的题目
@@ -369,7 +369,7 @@ export default class OpenAIGpt extends GptBase {
      * @param {*} axiosOption 
      * @returns 
      *///并在答案末尾处必须给出答案内容中的关键词
-    async generateExaminationPaperFromContent(content: string, paperOption: any = {}, everyContentLength: number = SECTION_LENGTH, axiosOption: any = {}): Promise<ExaminationPaperResult> {
+    override async generateExaminationPaperFromContent(content: string, paperOption: any = {}, everyContentLength: number = SECTION_LENGTH, axiosOption: any = {}): Promise<ExaminationPaperResult> {
         let arrContent = this.splitLongText(content, everyContentLength || SECTION_LENGTH);
         let sectionCount: any = {
             singlechoice: (paperOption.singlechoice?.count || 0) / arrContent.length,

package/src/stabilityai.ts

@@ -0,0 +1,82 @@
+import { ApiResult, request, AzureOpenAIPatameters } from "./declare";
+import GptBase from "./gptbase"
+export default class StabilityAI extends GptBase {
+    
+    protected readonly apiKey: string;
+    protected readonly apiSetting: AzureOpenAIPatameters
+    protected readonly apiOption:StabilityOption;
+    /**
+     * 
+     * @param apiKey 调用OpenAI 的key
+     * @param azureOption  用作accesstoken的缓存 
+     * @param apiOption  用作accesstoken的缓存 
+     */
+    constructor(apiKey: string, urlOption: AzureOpenAIPatameters, apiOption: StabilityOption = {}) {
+        super();
+        this.apiKey = apiKey;
+        this.apiSetting = urlOption;
+        this.apiOption = apiOption;
+        if (!this.apiSetting.endpoint.toLowerCase().startsWith('https://') &&
+            !this.apiSetting.endpoint.toLowerCase().startsWith('https://')) {
+            this.apiSetting.endpoint = 'https://' + this.apiSetting.endpoint;
+        }
+    }
+    /**
+     * 请求Stable作画的接口
+     */
+    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:{
+                "Content-Type": "application/json",
+                "Accept": "application/json",
+                "Authorization": `Bearer ${this.apiKey}`
+            }
+        })
+        try {
+            let param = {
+                ...axiosOption,
+                method: "post",
+                url: `${this.apiSetting.endpoint}/v1/generation/${this.apiSetting.engine}/text-to-image`,
+                data: {
+                    text_prompts: [
+                        {
+                            text: chatText
+                        }
+                    ],
+                    cfg_scale: paramOption.cfg_scale || this.apiOption.cfg_scale || 7,
+                    clip_guidance_preset: paramOption.clip_guidance_preset || this.apiOption.clip_guidance_preset || "FAST_BLUE",
+                    height: paramOption.height || this.apiOption.height || 512,
+                    width: paramOption.width || this.apiOption.width || 512,
+                    samples: paramOption.samples || this.apiOption.samples || 1,
+                    steps: paramOption.steps || this.apiOption.steps || 30,
+                },
+                
+            };
+            const response:any = await request(param)
+            if (response.successed) {
+                let data = response.data;
+                return { successed: true, type: 'image', data: data.artifacts, };
+            }
+            return { successed: false, ...response.data };
+        } catch (error) {
+            console.log('result is error ', error)
+            return { successed: false, error };
+        }
+
+    }
+}
+
+export interface StabilityOption {
+    'cfg_scale'?: number,
+    'clip_guidance_preset'?: string,
+    'height'?: number,
+    'width'?: number,
+    'samples'?: number,
+    'steps'?: number,
+}
+
+export interface StabilityResult extends ApiResult {
+    'data'?:any;
+    'type'?:string;
+}