tmaiplugin/src/aipluginbase.ts

TrainingMaster AIGC Component

import { EventEmitter } from "events"
import { GptBase, GptFactory } from "doomiaichat"

export default abstract class PluginBase extends EventEmitter {
    protected gptInstance: GptBase.default | null;
    /**
     *构造函数
     * @param gpt : 根据配置创建出GPT Instance 
     */
    /**
     * 构造函数
     */
    constructor(gptProvider: string, gptParam: any) {
        super();
        this.gptInstance = GptFactory.createGpt((gptProvider as GptFactory.GptProviderEnum), gptParam.apikey, gptParam);
    }
    /**
     * 执行插件的核心方法
     * @param _text 
     * @param _axiosOption 
     * @returns 
     */
    abstract execute(pluginParam:any):any;
}