ds-markdown 0.0.10-beta.2 → 0.0.10-beta.4

package/dist/cjs/MarkdownCMD/index.js

@@ -10,7 +10,8 @@ const classnames_1 = __importDefault(require("classnames"));
 const compiler_js_1 = require("../utils/compiler.js");
 const constant_js_1 = require("../constant.js");
 const deepClone_js_1 = __importDefault(require("../utils/methods/deepClone.js"));
-const MarkdownCMD = (0, react_1.forwardRef)(({ interval = 30, isClosePrettyTyped = false, onEnd, onStart, onTypedChar }, ref) => {
+const useTypingTask_js_1 = require("../hooks/useTypingTask.js");
+const MarkdownCMD = (0, react_1.forwardRef)(({ interval = 30, onEnd, onStart, onTypedChar, timerType = 'requestAnimationFrame' }, ref) => {
     /** 当前需要打字的内容 */
     const charsRef = (0, react_1.useRef)([]);
     /**
@@ -18,23 +19,6 @@ const MarkdownCMD = (0, react_1.forwardRef)(({ interval = 30, isClosePrettyTyped
      * 如果打字已经完全结束，则不会再触发打字效果
      */
     const isWholeTypedEndRef = (0, react_1.useRef)(false);
-    /** 已经打过的字 */
-    const typedCharsRef = (0, react_1.useRef)(undefined);
-    /** 是否卸载 */
-    const isUnmountRef = (0, react_1.useRef)(false);
-    /** 是否正在打字 */
-    const isTypedRef = (0, react_1.useRef)(false);
-    /** 打字结束回调, */
-    const onEndRef = (0, react_1.useRef)(onEnd);
-    onEndRef.current = onEnd;
-    /** 打字开始回调 */
-    const onStartRef = (0, react_1.useRef)(onStart);
-    onStartRef.current = onStart;
-    /** 打字过程中回调 */
-    const onTypedCharRef = (0, react_1.useRef)(onTypedChar);
-    onTypedCharRef.current = onTypedChar;
-    /** 打字定时器 */
-    const timerRef = (0, react_1.useRef)(null);
     /**
      * 稳定段落
      * 稳定段落是已经打过字，并且不会再变化的段落
@@ -44,174 +28,22 @@ const MarkdownCMD = (0, react_1.forwardRef)(({ interval = 30, isClosePrettyTyped
     const [currentSegment, setCurrentSegment] = (0, react_1.useState)(undefined);
     /** 当前段落引用 */
     const currentParagraphRef = (0, react_1.useRef)(undefined);
-    currentParagraphRef.current = currentSegment;
-    /** 清除打字定时器 */
-    const clearTimer = () => {
-        if (timerRef.current) {
-            clearTimeout(timerRef.current);
-            timerRef.current = null;
-        }
-        isTypedRef.current = false;
-    };
-    (0, react_1.useEffect)(() => {
-        isUnmountRef.current = false;
-        return () => {
-            isUnmountRef.current = true;
-        };
-    }, []);
-    /** 思考段落 */
-    const thinkingParagraphs = (0, react_1.useMemo)(() => stableSegments.filter((paragraph) => paragraph.answerType === 'thinking'), [stableSegments]);
-    /** 回答段落 */
-    const answerParagraphs = (0, react_1.useMemo)(() => stableSegments.filter((paragraph) => paragraph.answerType === 'answer'), [stableSegments]);
-    /**
-     * 记录打过的字
-     * @param char 当前字符
-     * @returns
-     */
-    const recordTypedChars = (char) => {
-        let prevStr = '';
-        if (!typedCharsRef.current || typedCharsRef.current.answerType !== char.answerType) {
-            typedCharsRef.current = {
-                typedContent: char.content,
-                answerType: char.answerType,
-                prevStr: '',
-            };
-        }
-        else {
-            prevStr = typedCharsRef.current.typedContent;
-            typedCharsRef.current.typedContent += char.content;
-            typedCharsRef.current.prevStr = prevStr;
-        }
-        return {
-            prevStr,
-            nextStr: typedCharsRef.current?.typedContent || '',
-        };
-    };
+    // currentParagraphRef.current = currentSegment;
     /**
-     * 触发打字开始回调
-     * @param char 当前字符
+     * 处理字符显示逻辑
      */
-    const triggerOnStart = (char) => {
-        const onStartFn = onStartRef.current;
-        if (!onStartFn) {
-            return;
-        }
-        const { prevStr } = recordTypedChars(char);
-        onStartRef.current?.({
-            currentIndex: prevStr.length,
-            currentChar: char.content,
-            answerType: char.answerType,
-            prevStr,
-        });
-    };
-    /**
-     * 触发打字结束回调
-     */
-    const triggerOnEnd = () => {
-        const onEndFn = onEndRef.current;
-        if (!onEndFn) {
-            return;
-        }
-        onEndFn({
-            str: typedCharsRef.current?.typedContent,
-            answerType: typedCharsRef.current?.answerType,
-        });
-    };
-    /**
-     * 触发打字过程中回调
-     * @param char 当前字符
-     * @param isStartPoint 是否是开始打字(第一个字)
-     */
-    const triggerOnTypedChar = (char, isStartPoint = false) => {
-        const onTypedCharFn = onTypedCharRef.current;
-        if (!isStartPoint) {
-            recordTypedChars(char);
-        }
-        if (!onTypedCharFn) {
-            return;
-        }
-        onTypedCharFn({
-            currentIndex: typedCharsRef.current?.prevStr.length || 0,
-            currentChar: char.content,
-            answerType: char.answerType,
-            prevStr: typedCharsRef.current?.prevStr || '',
-        });
-    };
-    /** 开始打字任务 */
-    const startTypedTask = () => {
-        if (isTypedRef.current) {
-            return;
-        }
-        const chars = charsRef.current;
-        /** 停止打字 */
-        const stopTyped = () => {
-            isTypedRef.current = false;
-            if (timerRef.current) {
-                clearTimeout(timerRef.current);
-                timerRef.current = null;
-            }
-            triggerOnEnd();
-        };
-        /** 打下一个字 */
-        const nextTyped = () => {
-            if (chars.length === 0) {
-                stopTyped();
-                return;
-            }
-            timerRef.current = setTimeout(startTyped, interval);
-        };
-        /**
-         * 开始打字
-         * @param isStartPoint 是否是开始打字
-         */
-        function startTyped(isStartPoint = false) {
-            if (isUnmountRef.current) {
-                return;
-            }
-            isTypedRef.current = true;
-            const char = chars.shift();
-            if (char === undefined) {
-                stopTyped();
-                return;
-            }
-            if (isStartPoint) {
-                triggerOnStart(char);
-                triggerOnTypedChar(char, isStartPoint);
-            }
-            else {
-                triggerOnTypedChar(char);
-            }
-            const currentSegment = currentParagraphRef.current;
-            /** 如果碰到 space，和split_segment 则需要处理成两个段落 */
-            if (char.contentType === 'space' || char.contentType === 'split_segment') {
-                if (currentSegment) {
-                    setStableSegments((prev) => {
-                        const newParagraphs = [...prev];
-                        // 放入到稳定队列
-                        if (currentSegment) {
-                            newParagraphs.push({ ...currentSegment, isTyped: false });
-                        }
-                        if (char.contentType === 'space') {
-                            newParagraphs.push({
-                                content: '',
-                                isTyped: false,
-                                type: 'br',
-                                answerType: char.answerType,
-                                tokensReference: {
-                                    [char.tokenId]: {
-                                        startIndex: 0,
-                                        raw: char.content,
-                                    },
-                                },
-                            });
-                        }
-                        return newParagraphs;
-                    });
-                    setCurrentSegment(undefined);
-                }
-                else {
-                    setStableSegments((prev) => {
-                        const newParagraphs = [...prev];
+    const processCharDisplay = (char) => {
+        const currentSegment = currentParagraphRef.current;
+        /** 如果碰到 space，和split_segment 则需要处理成两个段落 */
+        if (char.contentType === 'space' || char.contentType === 'split_segment') {
+            if (currentSegment) {
+                setStableSegments((prev) => {
+                    const newParagraphs = [...prev];
+                    // 放入到稳定队列
+                    if (currentSegment) {
+                        newParagraphs.push({ ...currentSegment, isTyped: false });
+                    }
+                    if (char.contentType === 'space') {
                         newParagraphs.push({
                             content: '',
                             isTyped: false,
@@ -224,176 +56,332 @@ const MarkdownCMD = (0, react_1.forwardRef)(({ interval = 30, isClosePrettyTyped
                                 },
                             },
                         });
-                        return newParagraphs;
-                    });
-                }
-                nextTyped();
-                return;
-            }
-            // 处理当前段落
-            let _currentParagraph = currentSegment;
-            const newCurrentParagraph = {
-                content: '',
-                isTyped: false,
-                type: 'text',
-                answerType: char.answerType,
-                tokensReference: {},
-            };
-            if (!_currentParagraph) {
-                // 如果当前没有段落，则直接设置为当前段落
-                _currentParagraph = newCurrentParagraph;
+                    }
+                    return newParagraphs;
+                });
+                setCurrentSegment(() => undefined);
+                currentParagraphRef.current = undefined;
             }
-            else if (currentSegment && currentSegment?.answerType !== char.answerType) {
-                // 如果当前段落和当前字符的回答类型不一致，则需要处理成两个段落
+            else {
                 setStableSegments((prev) => {
                     const newParagraphs = [...prev];
-                    newParagraphs.push({ ...currentSegment, isTyped: false });
+                    newParagraphs.push({
+                        content: '',
+                        isTyped: false,
+                        type: 'br',
+                        answerType: char.answerType,
+                        tokensReference: {
+                            [char.tokenId]: {
+                                startIndex: 0,
+                                raw: char.content,
+                            },
+                        },
+                    });
                     return newParagraphs;
                 });
-                _currentParagraph = newCurrentParagraph;
-                setCurrentSegment(_currentParagraph);
             }
-            setCurrentSegment((prev) => {
-                const tokensReference = (0, deepClone_js_1.default)(_currentParagraph.tokensReference);
-                if (tokensReference[char.tokenId]) {
-                    tokensReference[char.tokenId].raw += char.content;
-                    tokensReference[char.tokenId].startIndex = prev?.content?.length || 0;
-                }
-                else {
-                    tokensReference[char.tokenId] = {
-                        startIndex: prev?.content?.length || 0,
-                        raw: char.content,
-                    };
-                }
-                return {
-                    ..._currentParagraph,
-                    tokensReference,
-                    content: (prev?.content || '') + char.content,
-                    isTyped: true,
-                };
+            return;
+        }
+        // 处理当前段落
+        const newCurrentParagraph = {
+            content: '',
+            isTyped: false,
+            type: 'text',
+            answerType: char.answerType,
+            tokensReference: {},
+        };
+        let _currentParagraph = currentSegment;
+        if (!_currentParagraph) {
+            // 如果当前没有段落，则直接设置为新当前段落
+            _currentParagraph = newCurrentParagraph;
+        }
+        else if (currentSegment && currentSegment?.answerType !== char.answerType) {
+            // 如果当前段落和当前字符的回答类型不一致，则需要处理成两个段落
+            setStableSegments((prev) => {
+                const newParagraphs = [...prev];
+                newParagraphs.push({ ...currentSegment, isTyped: false });
+                return newParagraphs;
             });
-            nextTyped();
+            _currentParagraph = newCurrentParagraph;
         }
-        startTyped(true);
+        const tokensReference = (0, deepClone_js_1.default)(_currentParagraph.tokensReference);
+        if (tokensReference[char.tokenId]) {
+            tokensReference[char.tokenId].raw += char.content;
+            tokensReference[char.tokenId].startIndex = currentSegment?.content?.length || 0;
+        }
+        else {
+            tokensReference[char.tokenId] = {
+                startIndex: currentSegment?.content?.length || 0,
+                raw: char.content,
+            };
+        }
+        const newCurrentSegment = {
+            ..._currentParagraph,
+            tokensReference,
+            content: (currentSegment?.content || '') + char.content,
+            isTyped: true,
+        };
+        currentParagraphRef.current = newCurrentSegment;
+        setCurrentSegment(() => newCurrentSegment);
     };
+    /** 思考段落 */
+    const thinkingParagraphs = (0, react_1.useMemo)(() => stableSegments.filter((paragraph) => paragraph.answerType === 'thinking'), [stableSegments]);
+    /** 回答段落 */
+    const answerParagraphs = (0, react_1.useMemo)(() => stableSegments.filter((paragraph) => paragraph.answerType === 'answer'), [stableSegments]);
+    // 使用新的打字任务 hook
+    const typingTask = (0, useTypingTask_js_1.useTypingTask)({
+        timerType,
+        interval,
+        charsRef,
+        onEnd,
+        onStart,
+        onTypedChar,
+        processCharDisplay,
+    });
     const lastSegmentRawRef = (0, react_1.useRef)({
         thinking: '',
         answer: '',
         thinkingReference: null,
         answerReference: null,
+        thinkingBuffer: '',
+        answerBuffer: '',
     });
-    (0, react_1.useImperativeHandle)(ref, () => ({
-        /**
-         * 添加内容
-         * @param content 内容 {string}
-         * @param answerType 回答类型 {AnswerType}
-         */
-        push: (content, answerType) => {
-            const lastSegmentReference = lastSegmentRawRef.current[`${answerType}Reference`];
-            if (isWholeTypedEndRef.current) {
-                if (constant_js_1.__DEV__) {
-                    console.warn('打字已经完全结束，不能再添加新的内容');
-                }
-                return;
+    /**
+     * 检测当前内容是否在安全的 Markdown 边界
+     */
+    const isAtSafeMarkdownBoundary = (content) => {
+        if (!content.trim())
+            return true;
+        const lines = content.split('\n');
+        const lastLine = lines[lines.length - 1];
+        // 如果以换行符结尾，通常是安全的
+        if (content.endsWith('\n'))
+            return true;
+        // 检查最后一行是否是完整的语法结构
+        const patterns = [
+            /^#+\s+.+$/, // 完整标题: "## 标题"
+            /^\s*(\d+\.|\*|\+|-)\s+.+$/, // 完整列表项: "1. 项目" 或 "- 项目"
+            /^\s*```\s*$/, // 代码块结束: "```"
+            /^\s*```\w*\s*$/, // 代码块开始: "```js"
+            /^\s*>.*$/, // 引用: "> 内容"
+            /^\s*\|.*\|\s*$/, // 完整表格行: "| 列1 | 列2 |"
+            /^.*[.!?。！？]\s*$/, // 以句号等结尾的句子
+        ];
+        return patterns.some((pattern) => pattern.test(lastLine));
+    };
+    /**
+     * 查找最近的安全分割点
+     */
+    const findLastSafeBoundary = (content) => {
+        const lines = content.split('\n');
+        // 从后往前找最后一个完整的行
+        for (let i = lines.length - 1; i >= 0; i--) {
+            const currentContent = lines.slice(0, i + 1).join('\n');
+            if (i < lines.length - 1) {
+                // 有后续行，说明当前行以换行结尾，通常是安全的
+                return currentContent.length + 1; // +1 for \n
             }
-            let currentLastSegmentReference = null;
-            let currentLastSegmentRaw = '';
-            let lastSegmentRaw = '';
-            if (lastSegmentReference) {
-                lastSegmentRaw = lastSegmentReference.noTrimEndRaw || lastSegmentReference.raw;
-                currentLastSegmentRaw = lastSegmentRaw + content;
+            const line = lines[i];
+            // 检查当前行是否是完整的结构
+            if (/^#+\s+.+$/.test(line) || // 完整标题
+                /^\s*(\d+\.|\*|\+|-)\s+.+$/.test(line) || // 完整列表项
+                /^\s*```\s*$/.test(line) || // 代码块标记
+                /^\s*>.*$/.test(line) || // 引用行
+                /^.*[.!?。！？]\s*$/.test(line)) {
+                // 完整句子
+                return currentContent.length;
             }
-            else {
-                currentLastSegmentRaw = content;
+        }
+        return 0; // 没找到安全点
+    };
+    /**
+     * 同步处理带缓冲的内容推送
+     */
+    const processBufferedPush = (content, answerType) => {
+        const bufferKey = `${answerType}Buffer`;
+        const lastSegmentRef = lastSegmentRawRef.current;
+        // 将内容添加到缓冲区
+        lastSegmentRef[bufferKey] += content;
+        // 检查当前是否在安全边界
+        if (isAtSafeMarkdownBoundary(lastSegmentRef[bufferKey])) {
+            // 在安全边界，直接处理所有内容
+            const bufferedContent = lastSegmentRef[bufferKey];
+            lastSegmentRef[bufferKey] = '';
+            processPushInternal(bufferedContent, answerType);
+        }
+        else {
+            // 不在安全边界，找到最后一个安全分割点
+            const safeBoundary = findLastSafeBoundary(lastSegmentRef[bufferKey]);
+            if (safeBoundary > 0) {
+                // 有安全分割点，处理安全部分，保留其余部分
+                const safeContent = lastSegmentRef[bufferKey].substring(0, safeBoundary);
+                const remainingContent = lastSegmentRef[bufferKey].substring(safeBoundary);
+                lastSegmentRef[bufferKey] = remainingContent;
+                processPushInternal(safeContent, answerType);
             }
-            // debugger;
-            const tokens = (0, compiler_js_1.compiler)(currentLastSegmentRaw);
-            // 如果最后一个token是space，则把lastSegmentRaw设置为空
-            if (tokens[tokens.length - 1].type === 'space') {
-                currentLastSegmentReference = null;
+            // 如果没有安全分割点，继续缓冲等待更多内容
+        }
+    };
+    /**
+     * 内部推送处理逻辑
+     */
+    const processPushInternal = (content, answerType) => {
+        if (content.length === 0) {
+            return;
+        }
+        const lastSegmentReference = lastSegmentRawRef.current[`${answerType}Reference`];
+        if (isWholeTypedEndRef.current) {
+            if (constant_js_1.__DEV__) {
+                console.warn('打字已经完全结束，不能再添加新的内容');
             }
-            else {
-                currentLastSegmentReference = tokens[tokens.length - 1];
+            return;
+        }
+        let currentLastSegmentReference = null;
+        let currentLastSegmentRaw = '';
+        let lastSegmentRaw = '';
+        if (lastSegmentReference) {
+            lastSegmentRaw = lastSegmentReference.noTrimEndRaw || lastSegmentReference.raw;
+            currentLastSegmentRaw = lastSegmentRaw + content;
+        }
+        else {
+            currentLastSegmentRaw = content;
+        }
+        const tokens = (0, compiler_js_1.compiler)(currentLastSegmentRaw);
+        // 如果最后一个token是space，则把lastSegmentRaw设置为空
+        if (tokens[tokens.length - 1].type === 'space') {
+            currentLastSegmentReference = null;
+        }
+        else {
+            currentLastSegmentReference = tokens[tokens.length - 1];
+        }
+        const pushAndSplitSegment = (raw, currenIndex, segmentTokenId) => {
+            const currentToken = tokens[currenIndex];
+            if (currenIndex > 0) {
+                const prevToken = tokens[currenIndex - 1];
+                if (prevToken.type !== 'space' && currentToken.type !== 'space') {
+                    charsRef.current.push({ content: '', answerType, contentType: 'split_segment', tokenId: currentToken.id });
+                }
             }
-            const pushAndSplitSegment = (raw, currenIndex) => {
-                const currentToken = tokens[currenIndex];
-                if (currenIndex > 0) {
-                    const prevToken = tokens[currenIndex - 1];
-                    if (prevToken.type !== 'space' && currentToken.type !== 'space') {
-                        charsRef.current.push({ content: '', answerType, contentType: 'split_segment', tokenId: currentToken.id });
-                    }
+            charsRef.current.push(...raw.split('').map((char) => ({ content: char, answerType, contentType: 'segment', tokenId: segmentTokenId })));
+        };
+        if (!lastSegmentReference) {
+            tokens.forEach((token, i) => {
+                if (token.type === 'space') {
+                    charsRef.current.push({ content: token.raw, answerType, contentType: 'space', tokenId: token.id });
                 }
-                charsRef.current.push(...raw.split('').map((char) => ({ content: char, answerType, contentType: 'segment', tokenId: currentToken.id })));
-            };
-            if (!lastSegmentReference) {
-                tokens.forEach((token, i) => {
-                    if (token.type === 'space') {
-                        charsRef.current.push({ content: token.raw, answerType, contentType: 'space', tokenId: token.id });
+                else {
+                    pushAndSplitSegment(token.raw, i, token.id);
+                }
+            });
+        }
+        else {
+            let str = '';
+            let firstSpaceIndex = -1;
+            let nextTokenIndex = lastSegmentRaw.length;
+            for (let i = 0; i < tokens.length; i++) {
+                const token = tokens[i];
+                if (token.type === 'space') {
+                    if (firstSpaceIndex === -1) {
+                        firstSpaceIndex = str.length;
                     }
-                    else {
-                        pushAndSplitSegment(token.raw, i);
-                    }
-                });
-            }
-            else {
-                let str = '';
-                let firstSpaceIndex = -1;
-                let nextTokenIndex = lastSegmentRaw.length;
-                for (let i = 0; i < tokens.length; i++) {
-                    const token = tokens[i];
-                    if (token.type === 'space') {
-                        if (firstSpaceIndex === -1) {
-                            firstSpaceIndex = str.length;
-                        }
-                        str += token.raw;
-                        if (lastSegmentRaw.length > firstSpaceIndex) {
-                            // 如果lastSegmentRaw的长度大于firstSpaceIndex，则需要将当前设置为 segment
-                            charsRef.current.push(...token.raw.split('').map((char) => ({ content: char, answerType, contentType: 'segment', tokenId: token.id })));
-                        }
-                        else {
-                            charsRef.current.push({ content: token.raw, answerType, contentType: 'space', tokenId: token.id });
-                        }
+                    str += token.raw;
+                    if (lastSegmentRaw.length > firstSpaceIndex) {
+                        // 如果lastSegmentRaw的长度大于firstSpaceIndex，则需要将当前设置为 segment
+                        charsRef.current.push(...token.raw.split('').map((char) => ({ content: char, answerType, contentType: 'segment', tokenId: token.id })));
                     }
                     else {
-                        str += token.noTrimEndRaw || token.raw;
-                        const realRaw = str.slice(nextTokenIndex);
-                        if (realRaw.length > 0) {
-                            pushAndSplitSegment(realRaw, i);
-                        }
+                        charsRef.current.push({ content: token.raw, answerType, contentType: 'space', tokenId: token.id });
                     }
-                    nextTokenIndex = str.length;
                 }
+                else {
+                    str += token.noTrimEndRaw || token.raw;
+                    const realRaw = str.slice(nextTokenIndex);
+                    if (realRaw.length > 0) {
+                        pushAndSplitSegment(realRaw, i, lastSegmentReference.id);
+                    }
+                }
+                nextTokenIndex = str.length;
+            }
+        }
+        lastSegmentRawRef.current[`${answerType}Reference`] = currentLastSegmentReference;
+        if (!typingTask.isTyping()) {
+            typingTask.start();
+        }
+    };
+    /**
+     * 强制刷新缓冲区内容
+     */
+    const flushBuffer = (answerType) => {
+        const lastSegmentRef = lastSegmentRawRef.current;
+        if (answerType) {
+            // 刷新指定类型的缓冲区
+            const bufferKey = `${answerType}Buffer`;
+            const bufferedContent = lastSegmentRef[bufferKey];
+            if (bufferedContent) {
+                lastSegmentRef[bufferKey] = '';
+                processPushInternal(bufferedContent, answerType);
+            }
+        }
+        else {
+            // 刷新所有缓冲区
+            if (lastSegmentRef.thinkingBuffer) {
+                const content = lastSegmentRef.thinkingBuffer;
+                lastSegmentRef.thinkingBuffer = '';
+                processPushInternal(content, 'thinking');
             }
-            lastSegmentRawRef.current[`${answerType}Reference`] = currentLastSegmentReference;
-            if (!isTypedRef.current) {
-                startTypedTask();
+            if (lastSegmentRef.answerBuffer) {
+                const content = lastSegmentRef.answerBuffer;
+                lastSegmentRef.answerBuffer = '';
+                processPushInternal(content, 'answer');
             }
+        }
+    };
+    (0, react_1.useImperativeHandle)(ref, () => ({
+        /**
+         * 添加内容
+         * @param content 内容 {string}
+         * @param answerType 回答类型 {AnswerType}
+         */
+        push: (content, answerType) => {
+            processBufferedPush(content, answerType);
         },
         /**
          * 清除打字任务
          */
         clear: () => {
-            clearTimer();
+            typingTask.stop();
             charsRef.current = [];
             setStableSegments([]);
             setCurrentSegment(undefined);
             isWholeTypedEndRef.current = false;
-            lastSegmentRawRef.current = {
-                thinking: '',
-                answer: '',
-                thinkingReference: null,
-                answerReference: null,
-            };
+            currentParagraphRef.current = undefined;
+            typingTask.clear();
+            // 清理缓冲区
+            const lastSegmentRef = lastSegmentRawRef.current;
+            lastSegmentRef.thinkingBuffer = '';
+            lastSegmentRef.answerBuffer = '';
+            lastSegmentRef.thinkingReference = null;
+            lastSegmentRef.answerReference = null;
         },
         /**
          * 主动触发打字结束
          */
         triggerWholeEnd: () => {
+            // 先刷新所有缓冲区内容
+            flushBuffer();
             isWholeTypedEndRef.current = true;
-            if (!isTypedRef.current) {
-                triggerOnEnd();
+            if (!typingTask.isTyping()) {
+                // 这里需要手动触发结束回调，因为 hook 中的 triggerOnEnd 不能直接调用
+                onEnd?.({
+                    str: undefined,
+                    answerType: undefined,
+                });
             }
         },
+        /**
+         * 刷新缓冲区 (新增方法)
+         */
+        flushBuffer,
     }));
     const getParagraphs = (paragraphs, answerType) => {
         return ((0, jsx_runtime_1.jsxs)("div", { className: `ds-markdown-paragraph ds-typed-${answerType}`, children: [paragraphs.map((paragraph, index) => {