From 5577fe17bb91460b2a30a7fd1dd3152f7cde2ea7 Mon Sep 17 00:00:00 2001
From: Roe-xin <xiang200308037300@qq.com>
Date: Thu, 8 Jan 2026 15:27:14 +0800
Subject: [PATCH] =?UTF-8?q?fix:=E8=A7=A3=E5=86=B3=E7=94=A8=E6=88=B7?=
 =?UTF-8?q?=E6=B6=88=E6=81=AF=E9=94=99=E4=BD=8D=E7=9A=84bug=20+=20?=
 =?UTF-8?q?=E8=A7=A3=E5=86=B3=E5=86=85=E5=AE=B9=E9=87=8D=E5=A4=8D=E5=B1=95?=
 =?UTF-8?q?=E7=A4=BA=E7=9A=84bug?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/views/messageArea.ts | 13 +++++++++++++
 1 file changed, 13 insertions(+)

diff --git a/src/views/messageArea.ts b/src/views/messageArea.ts
index d4d5519..4357ebe 100644
--- a/src/views/messageArea.ts
+++ b/src/views/messageArea.ts
@@ -742,6 +742,12 @@ export function getMessageAreaScript(): string {
 
       // 添加消息
       function addMessage(text, sender) {
+        // 如果是用户消息，先重置当前分段消息容器
+        // 确保用户消息不会被添加到上一轮 AI 消息容器内部
+        if (sender === 'user' && currentSegmentedMessage) {
+          currentSegmentedMessage = null;
+        }
+
         const div = document.createElement('div');
         div.className = \`message \${sender}-message\`;
 
@@ -919,6 +925,13 @@ export function getMessageAreaScript(): string {
       // 实时更新分段消息（按后端返回顺序）
       function updateSegmentsRealtime(segments, isComplete) {
         console.log('[WebView] updateSegmentsRealtime 被调用, segments:', segments, 'isComplete:', isComplete);
+
+        // 如果是完成标记，不处理 segments，直接返回
+        if (isComplete) {
+          console.log('[WebView] 对话完成，跳过 segments 处理');
+          return;
+        }
+
         if (!segments || segments.length === 0) {
           console.log('[WebView] segments 为空，跳过渲染');
           return;