359 lines
9.7 KiB
TypeScript
359 lines
9.7 KiB
TypeScript
import { AppModuleItemTypeEnum, SystemInputEnum, SpecificInputEnum } from '../app';
|
||
import { FlowModuleTypeEnum, FlowInputItemTypeEnum, FlowOutputItemTypeEnum } from './index';
|
||
import type { AppModuleTemplateItemType } from '@/types/app';
|
||
import { chatModelList } from '@/store/static';
|
||
import {
|
||
Input_Template_History,
|
||
Input_Template_TFSwitch,
|
||
Input_Template_UserChatInput
|
||
} from './inputTemplate';
|
||
import { rawSearchKey } from '../chat';
|
||
|
||
export const VariableInputModule: AppModuleTemplateItemType = {
|
||
logo: '/imgs/module/userGuide.png',
|
||
name: '开场引导',
|
||
intro: '可以在每个新对话开始前，给用户发送一段开场白，或要求用户填写一些内容作为本轮对话的变量。',
|
||
type: AppModuleItemTypeEnum.userGuide,
|
||
flowType: FlowModuleTypeEnum.userGuide,
|
||
inputs: [
|
||
{
|
||
key: SystemInputEnum.welcomeText,
|
||
type: FlowInputItemTypeEnum.input,
|
||
label: '开场白'
|
||
},
|
||
{
|
||
key: SystemInputEnum.variables,
|
||
type: FlowInputItemTypeEnum.systemInput,
|
||
label: '变量输入',
|
||
value: []
|
||
}
|
||
],
|
||
outputs: []
|
||
};
|
||
export const UserInputModule: AppModuleTemplateItemType = {
|
||
logo: '/imgs/module/userChatInput.png',
|
||
name: '用户问题',
|
||
intro: '用户输入的内容。该模块通常作为应用的入口，用户在发送消息后会首先执行该模块。',
|
||
type: AppModuleItemTypeEnum.initInput,
|
||
flowType: FlowModuleTypeEnum.questionInputNode,
|
||
url: '/openapi/modules/init/userChatInput',
|
||
inputs: [
|
||
{
|
||
key: SystemInputEnum.userChatInput,
|
||
type: FlowInputItemTypeEnum.systemInput,
|
||
label: '用户问题'
|
||
}
|
||
],
|
||
outputs: [
|
||
{
|
||
key: SystemInputEnum.userChatInput,
|
||
label: '用户问题',
|
||
type: FlowOutputItemTypeEnum.source,
|
||
targets: []
|
||
}
|
||
]
|
||
};
|
||
export const HistoryModule: AppModuleTemplateItemType = {
|
||
logo: '/imgs/module/history.png',
|
||
name: '聊天记录',
|
||
intro: '用户输入的内容。该模块通常作为应用的入口，用户在发送消息后会首先执行该模块。',
|
||
type: AppModuleItemTypeEnum.initInput,
|
||
flowType: FlowModuleTypeEnum.historyNode,
|
||
url: '/openapi/modules/init/history',
|
||
inputs: [
|
||
{
|
||
key: 'maxContext',
|
||
type: FlowInputItemTypeEnum.numberInput,
|
||
label: '最长记录数',
|
||
value: 4,
|
||
min: 0,
|
||
max: 50
|
||
},
|
||
{
|
||
key: SystemInputEnum.history,
|
||
type: FlowInputItemTypeEnum.hidden,
|
||
label: '聊天记录'
|
||
}
|
||
],
|
||
outputs: [
|
||
{
|
||
key: SystemInputEnum.history,
|
||
label: '聊天记录',
|
||
type: FlowOutputItemTypeEnum.source,
|
||
targets: []
|
||
}
|
||
]
|
||
};
|
||
|
||
const defaultModel = chatModelList[0];
|
||
export const ChatModule: AppModuleTemplateItemType = {
|
||
logo: '/imgs/module/AI.png',
|
||
name: 'AI 对话',
|
||
intro: 'AI 大模型对话',
|
||
flowType: FlowModuleTypeEnum.chatNode,
|
||
type: AppModuleItemTypeEnum.http,
|
||
url: '/openapi/modules/chat/gpt',
|
||
inputs: [
|
||
{
|
||
key: 'model',
|
||
type: FlowInputItemTypeEnum.custom,
|
||
label: '对话模型',
|
||
value: defaultModel?.model,
|
||
list: chatModelList.map((item) => ({ label: item.name, value: item.model }))
|
||
},
|
||
{
|
||
key: 'temperature',
|
||
type: FlowInputItemTypeEnum.slider,
|
||
label: '温度',
|
||
value: 0,
|
||
min: 0,
|
||
max: 10,
|
||
step: 1,
|
||
markList: [
|
||
{ label: '严谨', value: 0 },
|
||
{ label: '发散', value: 10 }
|
||
]
|
||
},
|
||
{
|
||
key: 'maxToken',
|
||
type: FlowInputItemTypeEnum.slider,
|
||
label: '回复上限',
|
||
value: defaultModel ? defaultModel.contextMaxToken / 2 : 2000,
|
||
min: 0,
|
||
max: defaultModel?.contextMaxToken || 4000,
|
||
step: 50,
|
||
markList: [
|
||
{ label: '0', value: 0 },
|
||
{
|
||
label: `${defaultModel?.contextMaxToken || 4000}`,
|
||
value: defaultModel?.contextMaxToken || 4000
|
||
}
|
||
]
|
||
},
|
||
{
|
||
key: 'systemPrompt',
|
||
type: FlowInputItemTypeEnum.textarea,
|
||
label: '系统提示词',
|
||
description:
|
||
'模型固定的引导词，通过调整该内容，可以引导模型聊天方向。该内容会被固定在上下文的开头。',
|
||
placeholder:
|
||
'模型固定的引导词，通过调整该内容，可以引导模型聊天方向。该内容会被固定在上下文的开头。',
|
||
value: ''
|
||
},
|
||
{
|
||
key: 'limitPrompt',
|
||
type: FlowInputItemTypeEnum.textarea,
|
||
label: '限定词',
|
||
description:
|
||
'限定模型对话范围，会被放置在本次提问前，拥有强引导和限定性。例如:\n1. 知识库是关于 Laf 的介绍，参考知识库回答问题，与 "Laf" 无关内容，直接回复: "我不知道"。\n2. 你仅回答关于 "xxx" 的问题，其他问题回复: "xxxx"',
|
||
placeholder:
|
||
'限定模型对话范围，会被放置在本次提问前，拥有强引导和限定性。例如:\n1. 知识库是关于 Laf 的介绍，参考知识库回答问题，与 "Laf" 无关内容，直接回复: "我不知道"。\n2. 你仅回答关于 "xxx" 的问题，其他问题回复: "xxxx"',
|
||
value: ''
|
||
},
|
||
Input_Template_TFSwitch,
|
||
{
|
||
key: 'quotePrompt',
|
||
type: FlowInputItemTypeEnum.target,
|
||
label: '引用内容'
|
||
},
|
||
Input_Template_History,
|
||
Input_Template_UserChatInput
|
||
],
|
||
outputs: [
|
||
{
|
||
key: SpecificInputEnum.answerText,
|
||
label: '模型回复',
|
||
description: '直接响应，无需配置',
|
||
type: FlowOutputItemTypeEnum.hidden,
|
||
targets: []
|
||
}
|
||
]
|
||
};
|
||
|
||
export const KBSearchModule: AppModuleTemplateItemType = {
|
||
logo: '/imgs/module/db.png',
|
||
name: '知识库搜索',
|
||
intro: '去知识库中搜索对应的答案。可作为 AI 对话引用参考。',
|
||
flowType: FlowModuleTypeEnum.kbSearchNode,
|
||
type: AppModuleItemTypeEnum.http,
|
||
url: '/openapi/modules/kb/search',
|
||
inputs: [
|
||
{
|
||
key: 'kb_ids',
|
||
type: FlowInputItemTypeEnum.custom,
|
||
label: '关联的知识库',
|
||
value: [],
|
||
list: []
|
||
},
|
||
{
|
||
key: 'similarity',
|
||
type: FlowInputItemTypeEnum.slider,
|
||
label: '相似度',
|
||
value: 0.8,
|
||
min: 0,
|
||
max: 1,
|
||
step: 0.01,
|
||
markList: [
|
||
{ label: '0', value: 0 },
|
||
{ label: '1', value: 1 }
|
||
]
|
||
},
|
||
{
|
||
key: 'limit',
|
||
type: FlowInputItemTypeEnum.slider,
|
||
label: '单次搜索上限',
|
||
value: 5,
|
||
min: 1,
|
||
max: 20,
|
||
step: 1,
|
||
markList: [
|
||
{ label: '1', value: 1 },
|
||
{ label: '20', value: 20 }
|
||
]
|
||
},
|
||
Input_Template_TFSwitch,
|
||
Input_Template_UserChatInput
|
||
],
|
||
outputs: [
|
||
{
|
||
key: rawSearchKey,
|
||
label: '源搜索数据',
|
||
type: FlowOutputItemTypeEnum.hidden,
|
||
response: true,
|
||
targets: []
|
||
},
|
||
{
|
||
key: 'isEmpty',
|
||
label: '搜索结果为空',
|
||
type: FlowOutputItemTypeEnum.source,
|
||
targets: []
|
||
},
|
||
{
|
||
key: 'quotePrompt',
|
||
label: '引用内容',
|
||
description: '搜索结果为空时不返回',
|
||
type: FlowOutputItemTypeEnum.source,
|
||
targets: []
|
||
}
|
||
]
|
||
};
|
||
|
||
export const AnswerModule: AppModuleTemplateItemType = {
|
||
logo: '/imgs/module/reply.png',
|
||
name: '指定回复',
|
||
intro: '该模块可以直接回复一段指定的内容。常用于引导、提示。',
|
||
type: AppModuleItemTypeEnum.answer,
|
||
flowType: FlowModuleTypeEnum.answerNode,
|
||
inputs: [
|
||
Input_Template_TFSwitch,
|
||
{
|
||
key: SpecificInputEnum.answerText,
|
||
value: '',
|
||
type: FlowInputItemTypeEnum.input,
|
||
label: '回复的内容'
|
||
}
|
||
],
|
||
outputs: []
|
||
};
|
||
export const TFSwitchModule: AppModuleTemplateItemType = {
|
||
logo: '',
|
||
name: 'TF开关',
|
||
intro: '可以判断输入的内容为 True 或者 False，从而执行不同操作。',
|
||
type: AppModuleItemTypeEnum.switch,
|
||
flowType: FlowModuleTypeEnum.tfSwitchNode,
|
||
inputs: [
|
||
{
|
||
key: SystemInputEnum.switch,
|
||
type: FlowInputItemTypeEnum.target,
|
||
label: '输入'
|
||
}
|
||
],
|
||
outputs: [
|
||
{
|
||
key: 'true',
|
||
label: 'True',
|
||
type: FlowOutputItemTypeEnum.source,
|
||
targets: []
|
||
},
|
||
{
|
||
key: 'false',
|
||
label: 'False',
|
||
type: FlowOutputItemTypeEnum.source,
|
||
targets: []
|
||
}
|
||
]
|
||
};
|
||
export const RecognizeIntentionModule: AppModuleTemplateItemType = {
|
||
logo: '/imgs/module/cq.png',
|
||
name: '意图识别',
|
||
intro: '可以判断用户问题属于哪方面问题，从而执行不同的操作。',
|
||
type: AppModuleItemTypeEnum.http,
|
||
url: '/openapi/modules/agent/recognizeIntention',
|
||
flowType: FlowModuleTypeEnum.recognizeIntention,
|
||
inputs: [
|
||
{
|
||
key: 'systemPrompt',
|
||
type: FlowInputItemTypeEnum.textarea,
|
||
label: '系统提示词',
|
||
description:
|
||
'你可以添加一些特定内容的介绍，从而更好的识别用户的问题类型。这个内容通常是给模型介绍一个它不知道的内容。',
|
||
placeholder: '例如: \n1. Laf 是一个云函数开发平台……\n2. Sealos 是一个集群操作系统',
|
||
value: ''
|
||
},
|
||
Input_Template_History,
|
||
Input_Template_UserChatInput,
|
||
{
|
||
key: 'agents',
|
||
type: FlowInputItemTypeEnum.custom,
|
||
label: '',
|
||
value: [
|
||
{
|
||
value: '',
|
||
key: 'fasw'
|
||
},
|
||
{
|
||
value: '',
|
||
key: 'fqsw'
|
||
}
|
||
]
|
||
}
|
||
],
|
||
outputs: [
|
||
{
|
||
key: 'fasw',
|
||
label: '',
|
||
type: FlowOutputItemTypeEnum.hidden,
|
||
targets: []
|
||
},
|
||
{
|
||
key: 'fqsw',
|
||
label: '',
|
||
type: FlowOutputItemTypeEnum.hidden,
|
||
targets: []
|
||
}
|
||
]
|
||
};
|
||
|
||
export const ModuleTemplates = [
|
||
{
|
||
label: '输入模块',
|
||
list: [UserInputModule, HistoryModule, VariableInputModule]
|
||
},
|
||
{
|
||
label: '对话模块',
|
||
list: [ChatModule]
|
||
},
|
||
{
|
||
label: '知识库模块',
|
||
list: [KBSearchModule]
|
||
},
|
||
{
|
||
label: '工具',
|
||
list: [AnswerModule]
|
||
},
|
||
{
|
||
label: 'Agent',
|
||
list: [RecognizeIntentionModule]
|
||
}
|
||
];
|