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((?:[\S\s]*?<\/lobeThinking>\s*\n\s*)?\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 // Support both cases: with line break (e.g. from other models) and without (e.g. from Gemini) output = output.replace(/(<\/lobeThinking>)(?:\r?\n)?(]*>)\s*```[^\n]*\n([\S\s]*?)(```\n)?(<\/lobeArtifact>)/, (_, start, content, __, end) => { if (content.trim().startsWith('[\S\s]*?<\/lobeThinking>\s*\n\s*)([\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 )[\S\s])*$/; if (regex.test(output)) { output = output.replace(regex, ''); } return output; }; // 预处理函数：确保 think 标签前后有两个换行符 export const normalizeThinkTags = (input: string) => { return ( input // 确保标签前后有两个换行符 .replaceAll(/([^\n])\s*/g, '$1\n\n') .replaceAll(/\s*([^\n])/g, '\n\n$1') // 确保标签前后有两个换行符 .replaceAll(/([^\n])\s*<\/think>/g, '$1\n\n') .replaceAll(/<\/think>\s*([^\n])/g, '\n\n$1') // 处理可能产生的多余换行符 .replaceAll(/\n{3,}/g, '\n\n') ); };