79 lines
3.1 KiB
TypeScript
79 lines
3.1 KiB
TypeScript
|
import { aiPrompts } from './aiPrompt'
|
|||
|
|
|
|||
|
|
export type AiInferenceModelModel = {
|
|||
|
|
value: string // AI选项值
|
|||
|
|
label: string // AI选项标签
|
|||
|
|
hasExample: boolean // 是否有示例
|
|||
|
|
mustCharacter: boolean // 是否必须包含角色
|
|||
|
|
systemContent: string // 系统内容
|
|||
|
|
userContent: string // 用户内容
|
|||
|
|
allAndExampleContent: string | null // 所有和示例内容
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* AI选项数据
|
|||
|
|
* @description 该数据用于选择AI选项，包含value、label、hasExample、systemContent、userContent和allAndExampleContent等属性
|
|||
|
|
*/
|
|||
|
|
export const aiOptionsData: AiInferenceModelModel[] = [
|
|||
|
|
{
|
|||
|
|
value: 'NanFengStoryboardMasterScenePrompt',
|
|||
|
|
label: '【NanFeng】场景提示大师（上下文-不包含人物）',
|
|||
|
|
hasExample: false,
|
|||
|
|
mustCharacter: false,
|
|||
|
|
systemContent: aiPrompts.NanFengStoryboardMasterScenePromptSystemContent,
|
|||
|
|
userContent: aiPrompts.NanFengStoryboardMasterScenePromptUserContent,
|
|||
|
|
allAndExampleContent: null
|
|||
|
|
},
|
|||
|
|
{
|
|||
|
|
value: 'NanFengStoryboardMasterSpecialEffects',
|
|||
|
|
label: '【NanFeng】分镜大师-特效增强版（上下文-角色分析-人物固定）',
|
|||
|
|
hasExample: false,
|
|||
|
|
mustCharacter: true,
|
|||
|
|
systemContent: aiPrompts.NanFengStoryboardMasterSpecialEffectsSystemContent,
|
|||
|
|
userContent: aiPrompts.NanFengStoryboardMasterSpecialEffectsUserContent,
|
|||
|
|
allAndExampleContent: null
|
|||
|
|
},
|
|||
|
|
{
|
|||
|
|
value: 'NanFengStoryboardMasterSDEnglish',
|
|||
|
|
label: '【NanFeng】分镜大师-SD英文版（上下文-SD-英文提示词）',
|
|||
|
|
hasExample: false,
|
|||
|
|
mustCharacter: false,
|
|||
|
|
systemContent: aiPrompts.NanFengStoryboardMasterSDEnglishSystemContent,
|
|||
|
|
userContent: aiPrompts.NanFengStoryboardMasterSDEnglishUserContent,
|
|||
|
|
allAndExampleContent: null
|
|||
|
|
},
|
|||
|
|
{
|
|||
|
|
value: 'NanFengStoryboardMasterSingleFrame',
|
|||
|
|
label: '【NanFeng】分镜大师-单帧分镜提示词（上下文-单帧-人物推理）',
|
|||
|
|
hasExample: false,
|
|||
|
|
mustCharacter: false,
|
|||
|
|
systemContent: aiPrompts.NanFengStoryboardMasterSingleFrameSystemContent,
|
|||
|
|
userContent: aiPrompts.NanFengStoryboardMasterSingleFrameUserContent,
|
|||
|
|
allAndExampleContent: null
|
|||
|
|
},
|
|||
|
|
{
|
|||
|
|
value: 'NanFengStoryboardMasterSingleFrameWithCharacter',
|
|||
|
|
label: '【NanFeng】分镜大师-单帧分镜提示词（上下文-单帧-角色分析-人物固定）',
|
|||
|
|
hasExample: false,
|
|||
|
|
mustCharacter: true,
|
|||
|
|
systemContent: aiPrompts.NanFengStoryboardMasterSingleFrameWithCharacterSystemContent,
|
|||
|
|
userContent: aiPrompts.NanFengStoryboardMasterSingleFrameWithCharacterUserContent,
|
|||
|
|
allAndExampleContent: null
|
|||
|
|
}
|
|||
|
|
]
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* 获取AI选项数据
|
|||
|
|
* @description 该函数用于获取AI选项数据，包含value、label、hasExample、systemContent、userContent和allAndExampleContent等属性
|
|||
|
|
* @param id AI选项ID
|
|||
|
|
* @returns AI选项数据
|
|||
|
|
* @throws {Error} 如果没有找到对应的AI选项，则抛出错误
|
|||
|
|
*/
|
|||
|
|
export function GetAIPromptOptionByValue(value: string) {
|
|||
|
|
let aiOptionIndex = aiOptionsData.findIndex((item) => item.value == value)
|
|||
|
|
if (aiOptionIndex == -1) {
|
|||
|
|
throw new Error('没有找到对应的AI选项，请先检查配置')
|
|||
|
|
}
|
|||
|
|
return aiOptionsData[aiOptionIndex]
|
|||
|
|
}
|