fix: align mobile AI stream recovery

Author: Jinyu Xu

src/api/index.js

@@ -771,26 +771,36 @@ export const aiChatApi = {
       return event
     }
 
-    while (!streamComplete) {
-      const { value, done } = await reader.read()
-      if (done) break
-      buffer += decoder.decode(value, { stream: true })
-      const parts = buffer.split(/\r?\n\r?\n/)
-      buffer = parts.pop() || ''
-      for (const part of parts) {
-        if (await handlePart(part) === 'done') {
-          streamComplete = true
+    try {
+      while (!streamComplete) {
+        const { value, done } = await reader.read()
+        if (done) {
+          buffer += decoder.decode()
           break
         }
+        buffer += decoder.decode(value, { stream: true })
+        const parts = buffer.split(/\r?\n\r?\n/)
+        buffer = parts.pop() || ''
+        for (const part of parts) {
+          if (await handlePart(part) === 'done') {
+            streamComplete = true
+            break
+          }
+        }
+        if (streamComplete) {
+          try {
+            await reader.cancel()
+          } catch (_) {}
+        }
       }
-      if (streamComplete) {
-        try {
-          await reader.cancel()
-        } catch (_) {}
+      if (!streamComplete && buffer.trim()) {
+        streamComplete = await handlePart(buffer) === 'done'
       }
-    }
-    if (!streamComplete && buffer.trim()) {
-      streamComplete = await handlePart(buffer) === 'done'
+    } catch (error) {
+      try {
+        await reader.cancel()
+      } catch (_) {}
+      throw error
     }
     return { completed: streamComplete }
   },

src/views/ai-hub/index.vue

@@ -122,6 +122,10 @@
                     </div>
                   </template>
                 </div>
+                <div v-if="msg.streamWarning" class="stream-warning">
+                  <van-icon name="warning-o" />
+                  <span>{{ msg.streamWarning }}</span>
+                </div>
                 <van-loading v-if="msg.loading" size="18" />
               </div>
               <div v-if="msg.role === 'assistant' && msg.content && !msg.loading" class="bubble-tools">
@@ -339,6 +343,8 @@ const COPY = {
     strategyPromptNeeded: '请先写一点策略想法',
     generateFailed: '生成失败',
     streamInterrupted: '连接中断，已保留当前内容，请重试。',
+    streamIncomplete: '响应未正常结束，请重试。',
+    outputLimit: '回答已达到输出上限，当前内容可能不完整。',
     desktopOnly: '手机端仅支持使用与监控，请在电脑端完成代码编辑或回测。',
     taskDiagnose: '诊断标的',
     taskDiagnoseDesc: '趋势、量能、支撑阻力和风险',
@@ -384,6 +390,8 @@ const COPY = {
     strategyPromptNeeded: '請先寫一點策略想法',
     generateFailed: '生成失敗',
     streamInterrupted: '連線中斷，已保留目前內容，請重試。',
+    streamIncomplete: '回應未正常結束，請重試。',
+    outputLimit: '回答已達到輸出上限，目前內容可能不完整。',
     desktopOnly: '手機端僅支援使用與監控，請在電腦端完成程式碼編輯或回測。',
     taskDiagnose: '診斷標的',
     taskDiagnoseDesc: '趨勢、量能、支撐阻力和風險',
@@ -429,6 +437,8 @@ const COPY = {
     strategyPromptNeeded: 'Write a short strategy idea first',
     generateFailed: 'Generation failed',
     streamInterrupted: 'The connection was interrupted. The current response was kept; please retry.',
+    streamIncomplete: 'The response did not finish correctly. Please retry.',
+    outputLimit: 'The response reached the output limit and may be incomplete.',
     desktopOnly: 'Mobile supports usage and monitoring only. Use desktop for code editing or backtesting.',
     taskDiagnose: 'Diagnose',
     taskDiagnoseDesc: 'Trend, volume, levels, and risk',
@@ -474,6 +484,8 @@ const COPY = {
     strategyPromptNeeded: 'まず戦略アイデアを入力してください',
     generateFailed: '生成に失敗しました',
     streamInterrupted: '接続が中断されました。現在の内容を保持しました。再試行してください。',
+    streamIncomplete: '応答が正常に完了しませんでした。もう一度お試しください。',
+    outputLimit: '出力上限に達したため、回答が不完全な可能性があります。',
     desktopOnly: 'モバイルは利用と監視専用です。コード編集やバックテストはデスクトップで行ってください。',
     taskDiagnose: '銘柄診断',
     taskDiagnoseDesc: 'トレンド、出来高、重要水準、リスク',
@@ -519,6 +531,8 @@ const COPY = {
     strategyPromptNeeded: '먼저 전략 아이디어를 입력하세요',
     generateFailed: '생성 실패',
     streamInterrupted: '연결이 중단되었습니다. 현재 내용을 유지했으니 다시 시도해 주세요.',
+    streamIncomplete: '응답이 정상적으로 완료되지 않았습니다. 다시 시도해 주세요.',
+    outputLimit: '출력 한도에 도달하여 답변이 불완전할 수 있습니다.',
     desktopOnly: '모바일은 사용 및 모니터링 전용입니다. 코드 편집과 백테스트는 데스크톱에서 진행하세요.',
     taskDiagnose: '종목 진단',
     taskDiagnoseDesc: '추세, 거래량, 레벨, 리스크',
@@ -893,9 +907,11 @@ export default {
     },
     async sendMessageStream(payload, pendingMsg) {
       let hasContent = false
+      let streamAccepted = false
       try {
-        await aiChatApi.streamMessage(payload, async (event, data) => {
-          if (event === 'meta') {
+        const streamResult = await aiChatApi.streamMessage(payload, async (event, data) => {
+          if (event === 'accepted' || event === 'meta') {
+            streamAccepted = true
             this.sessionId = data?.session_id || this.sessionId
             return
           }
@@ -909,7 +925,29 @@ export default {
             await this.revealStreamText(pendingMsg, text)
             return
           }
+          if (event === 'replace') {
+            const text = this.extractStreamText(data)
+            if (!text) throw new Error(this.text.generateFailed)
+            hasContent = true
+            this.updatePendingMessage(pendingMsg, {
+              content: text,
+              loading: false,
+              streamWarning: ''
+            })
+            return
+          }
+          if (event === 'warning') {
+            streamAccepted = true
+            this.updatePendingMessage(pendingMsg, {
+              loading: false,
+              streamWarning: data?.code === 'output_limit'
+                ? this.text.outputLimit
+                : (data?.msg || this.text.streamIncomplete)
+            })
+            return
+          }
           if (event === 'done') {
+            streamAccepted = true
             this.sessionId = data?.session_id || this.sessionId
             this.updatePendingMessage(pendingMsg, { id: data?.message_id || pendingMsg.id })
             const finalText = this.extractStreamText(data)
@@ -924,12 +962,19 @@ export default {
             return
           }
           if (event === 'error') {
-            throw new Error(data?.msg || data?.message || this.text.generateFailed)
+            streamAccepted = true
+            const streamError = new Error(data?.msg || data?.message || this.text.generateFailed)
+            streamError.streamErrorType = data?.error_type || ''
+            throw streamError
           }
         })
+        if (!streamResult?.completed) {
+          throw new Error(this.text.streamIncomplete)
+        }
       } catch (error) {
         if (error && typeof error === 'object') {
           error.streamHasContent = hasContent
+          error.streamAccepted = streamAccepted
         }
         throw error
       }
@@ -942,9 +987,17 @@ export default {
         await this.sendMessageStream(payload, pendingMsg)
         return
       } catch (error) {
-        if (error?.streamHasContent) {
-          this.updatePendingMessage(pendingMsg, { loading: false })
-          showToast({ message: this.text.streamInterrupted, type: 'fail' })
+        if (error?.streamAccepted || error?.streamHasContent) {
+          const hasContent = Boolean(String(pendingMsg.content || '').trim()) && pendingMsg.content !== this.text.sending
+          this.updatePendingMessage(pendingMsg, {
+            content: hasContent ? pendingMsg.content : (error?.message || this.text.generateFailed),
+            loading: false,
+            streamWarning: hasContent ? this.text.streamInterrupted : ''
+          })
+          showToast({
+            message: hasContent ? this.text.streamInterrupted : (error?.message || this.text.generateFailed),
+            type: 'fail'
+          })
           return
         }
         this.updatePendingMessage(pendingMsg, {
@@ -2008,6 +2061,20 @@ export default {
   line-height: 1.72;
 }
 
+.stream-warning {
+  display: flex;
+  align-items: flex-start;
+  gap: 6px;
+  margin-top: 10px;
+  padding: 8px 10px;
+  border: 1px solid rgba(251, 191, 36, 0.28);
+  border-radius: 10px;
+  color: #fbbf24;
+  background: rgba(251, 191, 36, 0.08);
+  font-size: 12px;
+  line-height: 1.45;
+}
+
 .markdown-body :deep(p) {
   margin: 0 0 10px;
 }

tests/unit/mobile-usage-boundary.test.mjs

@@ -206,7 +206,10 @@ test('AI composer stays at the bottom and uses a text send action', () => {
   assert.match(api, /sendMessage: \(payload\) => http\.post\('\/api\/ai\/chat\/message', payload, \{ timeout: 600000 \}\)/)
   assert.match(api, /streamMessage:[\s\S]*\/api\/ai\/chat\/message\/stream/)
   assert.match(api, /if \(await handlePart\(part\) === 'done'\)[\s\S]*await reader\.cancel\(\)/)
-  assert.match(aiHub, /error\.streamHasContent = hasContent[\s\S]*if \(error\?\.streamHasContent\)[\s\S]*return/)
+  assert.match(aiHub, /if \(!streamResult\?\.completed\)[\s\S]*throw new Error\(this\.text\.streamIncomplete\)/)
+  assert.match(aiHub, /error\.streamAccepted = streamAccepted[\s\S]*if \(error\?\.streamAccepted \|\| error\?\.streamHasContent\)[\s\S]*return/)
+  assert.match(aiHub, /event === 'replace'[\s\S]*content: text/)
+  assert.match(aiHub, /event === 'warning'[\s\S]*streamWarning/)
 })
 
 test('signal chart supports mobile history navigation and candle inspection', () => {