UNPKG

@lobehub/chat

Version:

Lobe Chat - an open-source, high-performance chatbot framework that supports speech synthesis, multimodal, and extensible Function Call plugin system. Supports one-click free deployment of your private ChatGPT/LLM web application.

github.com/lobehub/lobe-chat

lobehub/lobe-chat

75 lines (64 loc) • 2.92 kB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
import { ARTIFACT_TAG_REGEX, ARTIFACT_THINKING_TAG_REGEX } from '@/const/plugin'; /** * Replace all line breaks in the matched `lobeArtifact` tag with an empty string */ export const processWithArtifact = (input: string = '') => { // First remove outer fenced code block if it exists let output = input.replace( /^([\S\s]*?)\s*```[^\n]*\n((?:<lobeThinking>[\S\s]*?<\/lobeThinking>\s*\n\s*)?<lobeArtifact[\S\s]*?<\/lobeArtifact>\s*)\n```\s*([\S\s]*?)$/, (_, before = '', content, after = '') => { return [before.trim(), content.trim(), after.trim()].filter(Boolean).join('\n\n'); }, ); const thinkMatch = ARTIFACT_THINKING_TAG_REGEX.exec(output); // If the input contains the `lobeThinking` tag, replace all line breaks with an empty string if (thinkMatch) { output = output.replace(ARTIFACT_THINKING_TAG_REGEX, (match) => match.replaceAll(/\r?\n|\r/g, ''), ); } // Add empty line between lobeThinking and lobeArtifact if they are adjacent output = output.replace(/(<\/lobeThinking>)\r?\n(<lobeArtifact)/, '$1\n\n$2'); // Remove fenced code block between lobeArtifact and HTML content output = output.replace( /(<lobeArtifact[^>]*>)\s*```[^\n]*\n([\S\s]*?)(```\n)?(<\/lobeArtifact>)/, (_, start, content, __, end) => { if (content.trim().startsWith('<!DOCTYPE html') || content.trim().startsWith('<html')) { return start + content.trim() + end; } return start + content + (__ || '') + end; }, ); // Keep existing code blocks that are not part of lobeArtifact output = output.replace( /^([\S\s]*?)(<lobeThinking>[\S\s]*?<\/lobeThinking>\s*\n\s*<lobeArtifact[\S\s]*?<\/lobeArtifact>)([\S\s]*?)$/, (_, before, content, after) => { return [before.trim(), content.trim(), after.trim()].filter(Boolean).join('\n\n'); }, ); const match = ARTIFACT_TAG_REGEX.exec(output); // If the input contains the `lobeArtifact` tag, replace all line breaks with an empty string if (match) { output = output.replace(ARTIFACT_TAG_REGEX, (match) => match.replaceAll(/\r?\n|\r/g, '')); } // if not match, check if it's start with <lobeArtifact but not closed const regex = /<lobeArtifact\b(?:(?!\/?>)[\S\s])*$/; if (regex.test(output)) { output = output.replace(regex, '<lobeArtifact>'); } return output; }; // 预处理函数：确保 think 标签前后有两个换行符 export const normalizeThinkTags = (input: string) => { return ( input // 确保 <think> 标签前后有两个换行符 .replaceAll(/([^\n])\s*<think>/g, '$1\n\n<think>') .replaceAll(/<think>\s*([^\n])/g, '<think>\n\n$1') // 确保 </think> 标签前后有两个换行符 .replaceAll(/([^\n])\s*<\/think>/g, '$1\n\n</think>') .replaceAll(/<\/think>\s*([^\n])/g, '</think>\n\n$1') // 处理可能产生的多余换行符 .replaceAll(/\n{3,}/g, '\n\n') ); };