mlogclub
84 lines
3.5 KiB
TypeScript
84 lines
3.5 KiB
TypeScript
import {
|
|
BookOpenIcon,
|
|
BotIcon,
|
|
ClipboardListIcon,
|
|
FileTextIcon,
|
|
FlagIcon,
|
|
GitBranchIcon,
|
|
HeadphonesIcon,
|
|
HelpCircleIcon,
|
|
MessageCircleIcon,
|
|
PlayCircleIcon,
|
|
SearchIcon,
|
|
SendIcon,
|
|
ShieldCheckIcon,
|
|
TicketIcon,
|
|
UserCheckIcon,
|
|
type LucideIcon,
|
|
} from "lucide-react"
|
|
|
|
export type WorkflowNodeTone =
|
|
| "blue"
|
|
| "cyan"
|
|
| "emerald"
|
|
| "indigo"
|
|
| "amber"
|
|
| "violet"
|
|
| "rose"
|
|
| "slate"
|
|
|
|
export type WorkflowNodeMeta = {
|
|
icon: LucideIcon
|
|
tone: WorkflowNodeTone
|
|
label: string
|
|
}
|
|
|
|
const workflowNodeMetaByType: Record<string, WorkflowNodeMeta> = {
|
|
start: { icon: PlayCircleIcon, tone: "blue", label: "入口" },
|
|
conversation_understanding: { icon: MessageCircleIcon, tone: "indigo", label: "理解" },
|
|
reply_policy: { icon: ShieldCheckIcon, tone: "violet", label: "策略" },
|
|
knowledge_retrieve: { icon: BookOpenIcon, tone: "emerald", label: "知识" },
|
|
answerability_gate: { icon: HelpCircleIcon, tone: "cyan", label: "判断" },
|
|
llm_reply: { icon: BotIcon, tone: "indigo", label: "生成" },
|
|
condition: { icon: GitBranchIcon, tone: "cyan", label: "分支" },
|
|
analyze_conversation: { icon: SearchIcon, tone: "blue", label: "分析" },
|
|
prepare_ticket_draft: { icon: ClipboardListIcon, tone: "amber", label: "工单" },
|
|
human_confirm: { icon: UserCheckIcon, tone: "amber", label: "确认" },
|
|
create_ticket: { icon: TicketIcon, tone: "rose", label: "工单" },
|
|
handoff_to_human: { icon: HeadphonesIcon, tone: "amber", label: "人工" },
|
|
send_reply: { icon: SendIcon, tone: "emerald", label: "发送" },
|
|
end: { icon: FlagIcon, tone: "slate", label: "结束" },
|
|
}
|
|
|
|
export function getWorkflowNodeMeta(type: string): WorkflowNodeMeta {
|
|
return workflowNodeMetaByType[type] ?? { icon: FileTextIcon, tone: "slate", label: "节点" }
|
|
}
|
|
|
|
export function getWorkflowNodeAccentClass(tone: WorkflowNodeTone) {
|
|
const classes: Record<WorkflowNodeTone, string> = {
|
|
blue: "border-blue-200/80 bg-blue-50 text-blue-700 dark:border-blue-900/60 dark:bg-blue-950/50 dark:text-blue-300",
|
|
cyan: "border-cyan-200/80 bg-cyan-50 text-cyan-700 dark:border-cyan-900/60 dark:bg-cyan-950/50 dark:text-cyan-300",
|
|
emerald: "border-emerald-200/80 bg-emerald-50 text-emerald-700 dark:border-emerald-900/60 dark:bg-emerald-950/50 dark:text-emerald-300",
|
|
indigo: "border-indigo-200/80 bg-indigo-50 text-indigo-700 dark:border-indigo-900/60 dark:bg-indigo-950/50 dark:text-indigo-300",
|
|
amber: "border-amber-200/80 bg-amber-50 text-amber-700 dark:border-amber-900/60 dark:bg-amber-950/50 dark:text-amber-300",
|
|
violet: "border-violet-200/80 bg-violet-50 text-violet-700 dark:border-violet-900/60 dark:bg-violet-950/50 dark:text-violet-300",
|
|
rose: "border-rose-200/80 bg-rose-50 text-rose-700 dark:border-rose-900/60 dark:bg-rose-950/50 dark:text-rose-300",
|
|
slate: "border-slate-200/80 bg-slate-50 text-slate-700 dark:border-slate-800 dark:bg-slate-900/60 dark:text-slate-300",
|
|
}
|
|
return classes[tone]
|
|
}
|
|
|
|
export function getWorkflowNodeIconClass(tone: WorkflowNodeTone) {
|
|
const classes: Record<WorkflowNodeTone, string> = {
|
|
blue: "bg-blue-500 text-white shadow-blue-500/20",
|
|
cyan: "bg-cyan-500 text-white shadow-cyan-500/20",
|
|
emerald: "bg-emerald-500 text-white shadow-emerald-500/20",
|
|
indigo: "bg-indigo-500 text-white shadow-indigo-500/20",
|
|
amber: "bg-amber-500 text-white shadow-amber-500/20",
|
|
violet: "bg-violet-500 text-white shadow-violet-500/20",
|
|
rose: "bg-rose-500 text-white shadow-rose-500/20",
|
|
slate: "bg-slate-600 text-white shadow-slate-600/20",
|
|
}
|
|
return classes[tone]
|
|
}
|