feat: stream chat tool progress, fix MCP timeouts, and polish composer disclaimer.

Expose SSE chat progress with Chinese tool labels in the UI, extend MCP tool timeouts for mining workflows, use pre-close before market open, and add the AI disclaimer below the composer panel.

Co-authored-by: Cursor <cursoragent@cursor.com>

Author: Travisun

apps/server/src/index.ts

@@ -1,6 +1,6 @@
 import { randomUUID } from 'node:crypto'
 import Fastify from 'fastify'
-import { AgentEngine, fetchOpenAiModelList, getDataLayerPaths, resolveProjectRoot, type SessionContextRef } from '@inno-a-stock/agent'
+import { AgentEngine, fetchOpenAiModelList, getDataLayerPaths, resolveProjectRoot, type ChatProgressEvent, type SessionContextRef } from '@inno-a-stock/agent'
 import { ResearchHub } from '@inno-a-stock/research-hub'
 import { listTemplates, REGISTRY } from '@inno-a-stock/stock-eval'
 import {
@@ -453,6 +453,46 @@ app.post<{ Params: { id: string }; Body: { message: string; selected_text: strin
   },
 )
 
+app.post<{ Params: { id: string }; Body: { message: string; model?: string } }>(
+  '/api/sessions/:id/chat/stream',
+  async (req, reply) => {
+    if (!req.body?.message?.trim()) return reply.code(400).send({ error: 'message required' })
+
+    reply.hijack()
+    reply.raw.writeHead(200, {
+      'Content-Type': 'text/event-stream; charset=utf-8',
+      'Cache-Control': 'no-cache, no-transform',
+      Connection: 'keep-alive',
+      'X-Accel-Buffering': 'no',
+    })
+
+    const write = (event: ChatProgressEvent) => {
+      reply.raw.write(`data: ${JSON.stringify(event)}\n\n`)
+    }
+
+    try {
+      await agent.chat(
+        req.params.id,
+        req.body.message,
+        req.body.model,
+        { onProgress: write },
+      )
+    } catch (e) {
+      const message = e instanceof Error ? e.message : String(e)
+      write({ type: 'error', message })
+      write({
+        type: 'done',
+        reply: message,
+        tools_used: [],
+        session_id: req.params.id,
+        tool_steps: [],
+      })
+    } finally {
+      reply.raw.end()
+    }
+  },
+)
+
 app.post<{ Params: { id: string }; Body: { message: string; model?: string } }>(
   '/api/sessions/:id/chat',
   async (req, reply) => {

client-ui/src/api/client.ts

@@ -1,4 +1,5 @@
 import type { ApiResponse } from '../types/schemas'
+import type { ChatProgressEvent } from '../types/chatProgress'
 import type { ChatDisplayMessage, EphemeralAskTurn, SessionContextRef, SessionMeta, AvailableModel } from '../types/chat'
 
 /** Vite dev/preview proxies /api → backend (default :8711). */
@@ -546,3 +547,58 @@ export async function sendSessionChat(
     }),
   }, CHAT_REQUEST_TIMEOUT)
 }
+
+export async function streamSessionChat(
+  sessionId: string,
+  message: string,
+  onEvent: (event: ChatProgressEvent) => void,
+  model?: string,
+  signal?: AbortSignal,
+): Promise<void> {
+  const resp = await fetchWithTimeout(`${API_BASE}/sessions/${sessionId}/chat/stream`, {
+    method: 'POST',
+    headers: {
+      'Content-Type': 'application/json',
+      Accept: 'text/event-stream',
+    },
+    body: JSON.stringify({
+      message,
+      ...(model ? { model } : {}),
+    }),
+    signal,
+  }, CHAT_REQUEST_TIMEOUT)
+
+  if (!resp.ok) {
+    const err = await resp.json().catch(() => ({})) as { error?: string }
+    throw new Error(err.error || `API error: ${resp.status}`)
+  }
+  if (!resp.body) throw new Error('流式响应不可用')
+
+  const reader = resp.body.getReader()
+  const decoder = new TextDecoder()
+  let buffer = ''
+
+  while (true) {
+    const { done, value } = await reader.read()
+    if (done) break
+    buffer += decoder.decode(value, { stream: true })
+    const chunks = buffer.split('\n\n')
+    buffer = chunks.pop() ?? ''
+    for (const chunk of chunks) {
+      const line = chunk.split('\n').find(row => row.startsWith('data: '))
+      if (!line) continue
+      try {
+        onEvent(JSON.parse(line.slice(6)) as ChatProgressEvent)
+      } catch {
+        /* ignore malformed chunk */
+      }
+    }
+  }
+
+  if (buffer.trim()) {
+    const line = buffer.split('\n').find(row => row.startsWith('data: '))
+    if (line) {
+      onEvent(JSON.parse(line.slice(6)) as ChatProgressEvent)
+    }
+  }
+}

client-ui/src/chat/ChatApp.tsx

@@ -9,12 +9,13 @@ import WorkspaceSplitDivider from './WorkspaceSplitDivider'
 import {
   listSessions, createSession, getSession, deleteSession, forkSession, clearSessionContext,
   setSessionContext, ephemeralAsk,
-  sendSessionChat, getHealth, listAvailableModels, setSessionModel,
+  streamSessionChat, getHealth, listAvailableModels, setSessionModel,
 } from '../api/client'
 import type {
   ChatDisplayMessage, EphemeralAskTurn, MessageSelection, SessionContextRef, SessionSelectionContextRef,
   SessionMeta, AvailableModel,
 } from '../types/chat'
+import type { ChatLiveTrace } from '../types/chatProgress'
 import { previewSelectionText } from '../utils/formatContextRefPreview'
 import { innoTokens } from '../theme/tokens'
 import { useBreakpoint, useSidebarPreference, useSidebarOverlayMode, useSidebarResizeSync } from '../hooks/useBreakpoint'
@@ -190,6 +191,7 @@ export default function ChatApp() {
   const [contextRef, setContextRef] = useState<SessionContextRef | null>(null)
   const [input, setInput] = useState('')
   const [loading, setLoading] = useState(false)
+  const [liveTrace, setLiveTrace] = useState<ChatLiveTrace | null>(null)
   const [error, setError] = useState('')
   const [availableModels, setAvailableModels] = useState<AvailableModel[]>([])
   const [sessionModel, setSessionModelState] = useState<string | undefined>()
@@ -341,6 +343,7 @@ export default function ChatApp() {
 
     setInput('')
     setLoading(true)
+    setLiveTrace({ steps: [], thinkingLabel: '模型正在思考…' })
     setError('')
 
     const optimistic: ChatDisplayMessage = {
@@ -350,11 +353,52 @@ export default function ChatApp() {
     }
     setMessages(prev => [...prev, optimistic])
 
+    let resolvedSessionId = sessionId
+
     try {
-      const result = await sendSessionChat(sessionId, msg, sessionModel)
-      const sid = result.session_id && result.session_id !== sessionId ? result.session_id : sessionId
-      if (result.session_id && result.session_id !== sessionId) {
-        setActiveId(result.session_id)
+      await streamSessionChat(sessionId, msg, (event) => {
+        if (event.type === 'thinking') {
+          setLiveTrace(prev => ({
+            steps: prev?.steps ?? [],
+            thinkingLabel: event.label,
+            thinkingSnippet: event.snippet ?? prev?.thinkingSnippet,
+          }))
+          return
+        }
+        if (event.type === 'tool_start') {
+          setLiveTrace(prev => ({
+            thinkingLabel: prev?.thinkingLabel,
+            thinkingSnippet: prev?.thinkingSnippet,
+            steps: [...(prev?.steps ?? []), event.step],
+          }))
+          return
+        }
+        if (event.type === 'tool_done') {
+          setLiveTrace(prev => ({
+            thinkingLabel: prev?.thinkingLabel ?? '模型正在整理结果…',
+            thinkingSnippet: prev?.thinkingSnippet,
+            steps: (prev?.steps ?? []).map(step =>
+              step.id === event.step.id ? event.step : step,
+            ),
+          }))
+          return
+        }
+        if (event.type === 'reply') {
+          setLiveTrace(prev => ({
+            steps: prev?.steps ?? [],
+            thinkingLabel: '正在生成回复…',
+            thinkingSnippet: prev?.thinkingSnippet,
+          }))
+          return
+        }
+        if (event.type === 'done') {
+          resolvedSessionId = event.session_id || resolvedSessionId
+        }
+      }, sessionModel)
+
+      const sid = resolvedSessionId
+      if (sid !== sessionId) {
+        setActiveId(sid)
       }
       const fresh = await getSession(sid)
       setMessages(fresh.messages)
@@ -373,6 +417,7 @@ export default function ChatApp() {
         setMessages(prev => prev.slice(0, -1))
       }
     } finally {
+      setLiveTrace(null)
       setLoading(false)
     }
   }
@@ -609,6 +654,7 @@ export default function ChatApp() {
                     contextRef={contextRef}
                     input={input}
                     loading={loading}
+                    liveTrace={liveTrace}
                     error={error}
                     availableModels={availableModels}
                     sessionModel={sessionModel}

client-ui/src/chat/ChatComposer.tsx

@@ -61,6 +61,9 @@ const useStyles = makeStyles({
   },
   panelWrap: {
     position: 'relative',
+    display: 'flex',
+    flexDirection: 'column',
+    alignItems: 'stretch',
     width: '100%',
     boxSizing: 'border-box',
     marginBottom: `calc(-1 * ${innoTokens.chatComposerGroundExtend})`,
@@ -135,6 +138,7 @@ const useStyles = makeStyles({
     fontFamily: 'inherit',
     color: innoTokens.textPrimary,
     padding: 0,
+    margin: 0,
     '::placeholder': {
       color: innoTokens.textTertiary,
     },
@@ -199,6 +203,20 @@ const useStyles = makeStyles({
       to: { opacity: 1 },
     },
   },
+  disclaimer: {
+    position: 'relative',
+    zIndex: 1,
+    display: 'block',
+    flexShrink: 0,
+    textAlign: 'center',
+    fontSize: '12px',
+    lineHeight: 1.45,
+    color: innoTokens.textSecondary,
+    width: '100%',
+    margin: 0,
+    padding: '9px 0 0',
+    userSelect: 'none',
+  },
 })
 
 interface ChatComposerProps {
@@ -329,6 +347,9 @@ export default function ChatComposer({
             </div>
           </div>
         </div>
+        <span className={s.disclaimer}>
+          内容由AI生成，不构成投资建议，请核实重要信息
+        </span>
       </div>
     </div>
   )

client-ui/src/chat/ChatMessageItem.tsx

@@ -7,6 +7,7 @@ import {
 } from '@fluentui/react-icons'
 import type { ChatDisplayMessage } from '../types/chat'
 import MarkdownMessage from './MarkdownMessage'
+import ChatProcessTrace from './ChatProcessTrace'
 import { innoTokens } from '../theme/tokens'
 import { fadeInUp } from '../theme/mixins'
 import { formatFriendlyTime } from '../utils/formatFriendlyTime'
@@ -237,7 +238,21 @@ function ChatMessageItem({ message, index, isMobile = false, onFork }: Props) {
         {isUser
           ? message.content
           : <MarkdownMessage content={message.content} />}
-        {message.toolsUsed && message.toolsUsed.length > 0 && (
+        {message.toolSteps && message.toolSteps.length > 0 && (
+          <details style={{ marginTop: 12 }}>
+            <summary style={{
+              fontSize: 12,
+              color: innoTokens.textTertiary,
+              cursor: 'pointer',
+              userSelect: 'none',
+            }}
+            >
+              {`执行过程（${message.toolSteps.length} 步）`}
+            </summary>
+            <ChatProcessTrace steps={message.toolSteps} />
+          </details>
+        )}
+        {!message.toolSteps?.length && message.toolsUsed && message.toolsUsed.length > 0 && (
           <div className={s.toolTags}>
             {message.toolsUsed.map(t => (
               <Badge key={t} size="small" className={s.toolBadge}>{t}</Badge>