tmaiplugin
Version:
TrainingMaster AIGC Component
42 lines (41 loc) • 1.55 kB
JavaScript
;
Object.defineProperty(exports, "__esModule", { value: true });
exports.createPlugIn = exports.PluginEnum = void 0;
//ts check
/**
* 内容插件生成工厂
*/
const faqplugin_1 = require("./faqplugin");
const questionplugin_1 = require("./questionplugin");
const commentplugin_1 = require("./commentplugin");
const similarityplugin_1 = require("./similarityplugin");
const summaryplugin_1 = require("./summaryplugin");
/**
* OpenAI/NLP 的服务提供商 OpenAI,微软，百度文心(待接入)，google(待接入)
*/
exports.PluginEnum = {
FAQ: 'faq',
QUESTION: 'question',
COMMENT: 'comment',
SIMILARITY: 'similarity',
SUMMARY: 'summary',
};
/**
* 根据类型创建不同的TTS引擎对象
* @param {*} provider
* @param {*} apikey
* @param {*} setting
* @returns
*/
function createPlugIn(provider, gptParam) {
switch (provider) {
case exports.PluginEnum.FAQ: return new faqplugin_1.FaqPlugin(gptParam.provider, gptParam.option);
case exports.PluginEnum.QUESTION: return new questionplugin_1.QuestionPlugin(gptParam.provider, gptParam.option);
case exports.PluginEnum.COMMENT: return new commentplugin_1.CommentPlugin(gptParam.provider, gptParam.option);
case exports.PluginEnum.SIMILARITY: return new similarityplugin_1.SimilarityPlugin(gptParam.provider, gptParam.option);
case exports.PluginEnum.SUMMARY: return new summaryplugin_1.SummaryPlugin(gptParam.provider, gptParam.option);
default: return null;
}
}
exports.createPlugIn = createPlugIn;
;