47 lines
1.6 KiB
TypeScript
47 lines
1.6 KiB
TypeScript
import type { BranchGroup, Message } from "./GlobalChatbox.types";
|
|
|
|
export const createId = () =>
|
|
`${Date.now()}-${Math.random().toString(36).slice(2, 8)}`;
|
|
export const PRESET_PROMPTS = [
|
|
"分析当前管网中的水力瓶颈管道，并给出改造建议。",
|
|
"帮我分析当前管网压力异常点，并按风险等级排序。",
|
|
"帮我生成一份今日运行简报，包含问题、原因和建议。",
|
|
"查询关键 SCADA 点位最近 24 小时的异常波动。",
|
|
"排查当前管网爆管风险，并说明优先处置建议。",
|
|
];
|
|
|
|
export const stripMarkdown = (md: string): string =>
|
|
md
|
|
.replace(/```[\s\S]*?```/g, "")
|
|
.replace(/`([^`]+)`/g, "$1")
|
|
.replace(/!\[.*?\]\(.*?\)/g, "")
|
|
.replace(/\[([^\]]+)\]\(.*?\)/g, "$1")
|
|
.replace(/#{1,6}\s+/g, "")
|
|
.replace(/\*\*\*(.+?)\*\*\*/g, "$1")
|
|
.replace(/\*\*(.+?)\*\*/g, "$1")
|
|
.replace(/\*(.+?)\*/g, "$1")
|
|
.replace(/~~(.+?)~~/g, "$1")
|
|
.replace(/>\s+/g, "")
|
|
.replace(/[-*+]\s+/g, "")
|
|
.replace(/\d+\.\s+/g, "")
|
|
.replace(/\n{2,}/g, "\n")
|
|
.replace(/<[^>]+>/g, "")
|
|
.trim();
|
|
|
|
export const cloneMessage = (message: Message): Message => ({
|
|
...message,
|
|
progress: message.progress ? [...message.progress] : undefined,
|
|
artifacts: message.artifacts ? [...message.artifacts] : undefined,
|
|
});
|
|
|
|
export const cloneMessages = (messages: Message[]) => messages.map(cloneMessage);
|
|
|
|
export const cloneBranchGroups = (branchGroups: BranchGroup[]) =>
|
|
branchGroups.map((group) => ({
|
|
...group,
|
|
branches: group.branches.map((branch) => ({
|
|
...branch,
|
|
messages: cloneMessages(branch.messages),
|
|
})),
|
|
}));
|