flare-chat-core 0.1.1 → 0.2.1

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "flare-chat-core",
   "private": false,
-  "version": "0.1.1",
+  "version": "0.2.1",
   "type": "module",
   "main": "./src/index.js",
   "module": "./src/index.js",

package/scripts/check.sh

@@ -0,0 +1,15 @@
+#!/usr/bin/env sh
+set -eu
+
+SCRIPT_DIR=$(CDPATH= cd -- "$(dirname -- "$0")" && pwd)
+PKG_DIR=$(CDPATH= cd -- "$SCRIPT_DIR/.." && pwd)
+
+level="${1:-quick}"
+case "$level" in
+  quick|full) ;;
+  *) echo "Unsupported level: $level" >&2; exit 2 ;;
+esac
+
+find "$PKG_DIR/src" -type f -name '*.js' | while IFS= read -r file; do
+  node --check "$file"
+done

package/src/chat-core/input/useChatInput.js

@@ -1,5 +1,18 @@
 import { useState, useCallback } from 'react';
 
+function buildAttachmentFile(file, index = 0, status = 'ready') {
+  return {
+    uid: file?.uid || `${Date.now()}-${index}`,
+    name: file?.name || `附件-${index + 1}`,
+    size: Number.isFinite(file?.size) ? file.size : null,
+    type: file?.type || '',
+    originFileObj: file?.originFileObj || file || null,
+    status: file?.status || status,
+    error_message: file?.error_message || '',
+    created_at: file?.created_at || new Date().toISOString(),
+  };
+}
+
 export function useChatInput() {
   const [inputValue, setInputValue] = useState('');
   const [fileList, setFileList] = useState([]);
@@ -23,18 +36,37 @@ export function useChatInput() {
     setIsDragging(false);
 
     const files = Array.from(e.dataTransfer.files);
-    const newFiles = files.map((file, index) => ({
-      uid: `${Date.now()}-${index}`,
-      name: file.name,
-      size: file.size,
-      type: file.type,
-      originFileObj: file,
-    }));
+    const newFiles = files.map((file, index) => buildAttachmentFile(file, index));
 
     setFileList((prev) => [...prev, ...newFiles]);
     return files.length;
   }, []);
 
+  const appendFiles = useCallback((files = [], options = {}) => {
+    const normalized = Array.isArray(files)
+      ? files.map((file, index) => buildAttachmentFile(file, index, options.status || 'ready'))
+      : [];
+    if (normalized.length === 0) {
+      return;
+    }
+    setFileList((prev) => [...prev, ...normalized]);
+  }, []);
+
+  const updateFileStatus = useCallback((uid, status, errorMessage = '') => {
+    if (!uid) {
+      return;
+    }
+    setFileList((prev) => prev.map((file) => (
+      file.uid === uid
+        ? {
+            ...file,
+            status: status || file.status,
+            error_message: errorMessage || '',
+          }
+        : file
+    )));
+  }, []);
+
   const handleDragOver = useCallback((e) => {
     e.preventDefault();
     e.stopPropagation();
@@ -59,6 +91,8 @@ export function useChatInput() {
     handleInputChange,
     handleFileChange,
     handleRemoveFile,
+    appendFiles,
+    updateFileStatus,
     handleDrop,
     handleDragOver,
     handleDragLeave,
@@ -67,4 +101,3 @@ export function useChatInput() {
 }
 
 export default useChatInput;
-

package/src/chat-core/messages/buildTimelineItems.js

@@ -1,8 +1,8 @@
-export function buildTimelineItems(state = {}) {
+export function buildTimelineItems(state = {}, options = {}) {
   const messages = Array.isArray(state.messages) ? state.messages : [];
-  const executionCards = Array.isArray(state.executionCards) ? state.executionCards : [];
   const uiCards = Array.isArray(state.uiCards) ? state.uiCards : [];
   const streaming = state.streaming || {};
+  const loading = Boolean(options?.loading);
 
   const items = [];
 
@@ -21,23 +21,15 @@ export function buildTimelineItems(state = {}) {
       role: msg.role,
       content: msg.content,
       createdAt: msg.created_at,
+      client_request_id: msg.client_request_id || '',
+      intake_session_id: msg.intake_session_id || '',
+      roundKey: msg.round_key || '',
       highlights: msg.highlights ?? [],
       sourceTypes: msg.sourceTypes ?? [],
+      attachments: Array.isArray(msg.attachments) ? msg.attachments : [],
     });
   });
 
-  executionCards
-    .filter(Boolean)
-    .filter((card) => card.step_id !== streaming.executionTrace?.step_id)
-    .forEach((card, index) => {
-      const executionId = card.step_id || `execution-fallback-${index}`;
-      items.push({
-        id: executionId,
-        type: 'execution',
-        executionTrace: card,
-      });
-    });
-
   if (uiCards.length > 0) {
     items.push({
       id: 'conversation-ui-cards',
@@ -54,7 +46,8 @@ export function buildTimelineItems(state = {}) {
     });
   }
 
-  if (streaming.agentStatus || streaming.thinkingTrace || streaming.executionTrace) {
+  const hasThinkingSignal = Boolean(streaming.agentStatus || streaming.thinkingTrace || streaming.executionTrace);
+  if (hasThinkingSignal) {
     items.push({
       id: 'thinking-bubble',
       type: 'thinking',
@@ -62,6 +55,15 @@ export function buildTimelineItems(state = {}) {
       thinkingTrace: streaming.thinkingTrace,
       executionTrace: streaming.executionTrace,
     });
+  } else if (loading && !streaming.content) {
+    items.push({
+      id: 'thinking-bubble-loading',
+      type: 'thinking',
+      agentStatus: {
+        status: 'running',
+        agent: '助手',
+      },
+    });
   }
 
   return {
@@ -71,4 +73,3 @@ export function buildTimelineItems(state = {}) {
 }
 
 export default buildTimelineItems;
-

package/src/chat-core/session/useChatSessionReducer.js

@@ -8,6 +8,7 @@ const MESSAGES_REFRESHED = 'MESSAGES_REFRESHED';
 const MESSAGE_APPENDED = 'MESSAGE_APPENDED';
 const STREAMING_RESET = 'STREAMING_RESET';
 const STREAMING_CHUNK = 'STREAMING_CHUNK';
+const STREAMING_REPLACE = 'STREAMING_REPLACE';
 const AGENT_STATUS_SET = 'AGENT_STATUS_SET';
 const THINKING_TRACE_SET = 'THINKING_TRACE_SET';
 const EXECUTION_TRACE_SET = 'EXECUTION_TRACE_SET';
@@ -19,6 +20,7 @@ const UI_CARDS_SET = 'UI_CARDS_SET';
 const LAST_USER_MESSAGE_SET = 'LAST_USER_MESSAGE_SET';
 const UI_CARD_UPDATED = 'UI_CARD_UPDATED';
 const INSTANCE_PROFILE_SET = 'INSTANCE_PROFILE_SET';
+const SESSION_PROMOTED = 'SESSION_PROMOTED';
 
 const initialStreaming = {
   content: '',
@@ -31,6 +33,7 @@ export const initialState = {
   sessionId: null,
   sessionTitle: '',
   sessionStatus: 'active',
+  sessionDetail: null,
   instanceProfile: null,
   messages: [],
   executionCards: [],
@@ -48,12 +51,19 @@ export const initialState = {
 function reducer(state, { type, payload }) {
   switch (type) {
     case SESSION_LOADED: {
-      const { sessionId, title, status, messages } = payload;
+      const {
+        sessionId,
+        title,
+        status,
+        messages,
+        detail,
+      } = payload;
       return {
         ...state,
         sessionId,
         sessionTitle: title,
         sessionStatus: status || 'active',
+        sessionDetail: (detail && typeof detail === 'object' && !Array.isArray(detail)) ? detail : null,
         instanceProfile: state.instanceProfile,
         messages,
         executionCards: [],
@@ -103,6 +113,40 @@ function reducer(state, { type, payload }) {
     case INSTANCE_PROFILE_SET:
       return { ...state, instanceProfile: payload };
 
+    case SESSION_PROMOTED: {
+      const sessionId = String(payload?.sessionId || '').trim();
+      if (!sessionId) {
+        return state;
+      }
+      const title = String(payload?.title || '').trim();
+      const detailPayload = (
+        payload?.detail
+        && typeof payload.detail === 'object'
+        && !Array.isArray(payload.detail)
+      ) ? payload.detail : null;
+      const nextDetail = detailPayload
+        ? {
+            ...(state.sessionDetail && typeof state.sessionDetail === 'object' ? state.sessionDetail : {}),
+            ...detailPayload,
+            sessionId,
+          }
+        : (
+          state.sessionDetail && typeof state.sessionDetail === 'object'
+            ? {
+                ...state.sessionDetail,
+                sessionId,
+              }
+            : null
+        );
+      return {
+        ...state,
+        sessionId,
+        sessionTitle: title || state.sessionTitle,
+        sessionStatus: String(payload?.status || state.sessionStatus || 'active'),
+        sessionDetail: nextDetail,
+      };
+    }
+
     case EXECUTION_CARD_UPSERTED: {
       const nextCard = payload;
       const cardKey = nextCard.step_id || `${nextCard.agent}-${nextCard.label}-${nextCard.stage}`;
@@ -135,6 +179,15 @@ function reducer(state, { type, payload }) {
         },
       };
 
+    case STREAMING_REPLACE:
+      return {
+        ...state,
+        streaming: {
+          ...state.streaming,
+          content: String(payload || ''),
+        },
+      };
+
     case AGENT_STATUS_SET:
       return {
         ...state,
@@ -223,8 +276,10 @@ export default function useChatSessionReducer(deps = {}) {
         title: detail.title,
         status: detail.status || 'active',
         messages: msgs.messages,
+        detail,
       },
     });
+    return detail;
   }, [sessionAPI, messageAPI]);
 
   const createSession = useCallback(async ({
@@ -242,6 +297,7 @@ export default function useChatSessionReducer(deps = {}) {
         title,
         status: response.status || 'active',
         messages: [],
+        detail: response,
       },
     });
 
@@ -295,6 +351,7 @@ export default function useChatSessionReducer(deps = {}) {
         title,
         status: detail.status || 'active',
         messages: msgs.messages,
+        detail,
       },
     });
 
@@ -311,6 +368,10 @@ export default function useChatSessionReducer(deps = {}) {
     dispatch({ type: TITLE_UPDATED, payload: title });
   }, [sessionAPI]);
 
+  const promoteSession = useCallback((payload = {}) => {
+    dispatch({ type: SESSION_PROMOTED, payload });
+  }, []);
+
   const refreshMessages = useCallback(async (sessionId) => {
     const messageApi = requireAPI(messageAPI, 'messageAPI');
     const response = await messageApi.list(sessionId);
@@ -321,7 +382,14 @@ export default function useChatSessionReducer(deps = {}) {
     dispatch({ type: MESSAGES_REFRESHED, payload: messagesWithIds });
   }, [messageAPI]);
 
-  const appendUserMessage = useCallback((content) => {
+  const appendUserMessage = useCallback((content, metadata = {}) => {
+    const payloadMeta = (
+      metadata
+      && typeof metadata === 'object'
+      && !Array.isArray(metadata)
+    )
+      ? metadata
+      : {};
     dispatch({
       type: MESSAGE_APPENDED,
       payload: {
@@ -329,6 +397,7 @@ export default function useChatSessionReducer(deps = {}) {
         role: 'user',
         content,
         created_at: new Date().toISOString(),
+        ...payloadMeta,
       },
     });
   }, []);
@@ -353,6 +422,10 @@ export default function useChatSessionReducer(deps = {}) {
     dispatch({ type: STREAMING_CHUNK, payload: chunk });
   }, []);
 
+  const replaceStreamContent = useCallback((content) => {
+    dispatch({ type: STREAMING_REPLACE, payload: content });
+  }, []);
+
   const setAgentStatus = useCallback((agentStatus) => {
     dispatch({ type: AGENT_STATUS_SET, payload: agentStatus });
   }, []);
@@ -389,6 +462,7 @@ export default function useChatSessionReducer(deps = {}) {
   return useMemo(() => ({
     sessionId: state.sessionId,
     sessionTitle: state.sessionTitle,
+    sessionDetail: state.sessionDetail,
     instanceProfile: state.instanceProfile,
     messages: state.messages,
     executionCards: state.executionCards,
@@ -403,6 +477,7 @@ export default function useChatSessionReducer(deps = {}) {
     createOrLoadSession,
     resetSession,
     updateTitle,
+    promoteSession,
     refreshMessages,
     setSessionError,
     appendUserMessage,
@@ -411,6 +486,7 @@ export default function useChatSessionReducer(deps = {}) {
     setLastUserMessage,
     resetStreaming,
     appendStreamChunk,
+    replaceStreamContent,
     setAgentStatus,
     setThinkingTrace,
     setExecutionTrace,
@@ -425,6 +501,7 @@ export default function useChatSessionReducer(deps = {}) {
     createOrLoadSession,
     resetSession,
     updateTitle,
+    promoteSession,
     refreshMessages,
     setSessionError,
     appendUserMessage,
@@ -433,6 +510,7 @@ export default function useChatSessionReducer(deps = {}) {
     setLastUserMessage,
     resetStreaming,
     appendStreamChunk,
+    replaceStreamContent,
     setAgentStatus,
     setThinkingTrace,
     setExecutionTrace,

package/src/chat-core/stream/sse-client.js

@@ -13,11 +13,19 @@ function joinUrl(baseUrl, endpoint) {
   return `${baseUrl.replace(/\/+$/, '')}/${endpoint.replace(/^\/+/, '')}`;
 }
 
+function buildClientRequestId() {
+  const random = Math.random().toString(36).slice(2, 10);
+  return `req_${Date.now()}_${random}`;
+}
+
 export class SSEClient {
   constructor({ endpoint = DEFAULT_ENDPOINT, baseUrl = API_BASE_URL } = {}) {
     this.abortController = null;
     this.endpoint = endpoint;
     this.baseUrl = baseUrl;
+    this.activeRequestId = '';
+    this.authReadyPromise = null;
+    this.authReadyAtMs = 0;
   }
 
   async sendMessage(sessionOrParams, contentArg, onEventArg, onCompleteArg, onErrorArg, optionsArg = {}) {
@@ -33,26 +41,124 @@ export class SSEClient {
     const sessionId = params.sessionId;
     const content = params.content;
     const enabledCapabilities = Array.isArray(params.enabledCapabilities) ? params.enabledCapabilities : [];
+    const modeKey = typeof params.modeKey === 'string' ? params.modeKey.trim() : '';
+    const manualModeKey = typeof params.manualModeKey === 'string' ? params.manualModeKey.trim() : '';
+    const payloadExtra = (
+      params.payloadExtra
+      && typeof params.payloadExtra === 'object'
+      && !Array.isArray(params.payloadExtra)
+    )
+      ? params.payloadExtra
+      : {};
     const onEvent = isObjectCall ? contentArg : onEventArg;
     const onComplete = isObjectCall ? onEventArg : onCompleteArg;
     const onError = isObjectCall ? onCompleteArg : onErrorArg;
     const url = params.url || joinUrl(params.baseUrl ?? this.baseUrl, params.endpoint || this.endpoint);
-
+    const command = typeof params.command === 'string' && params.command.trim()
+      ? params.command.trim()
+      : (typeof payloadExtra.command === 'string' && payloadExtra.command.trim() ? payloadExtra.command.trim() : 'send_message');
+    const clientRequestId = typeof params.clientRequestId === 'string' && params.clientRequestId.trim()
+      ? params.clientRequestId.trim()
+      : buildClientRequestId();
+    const contractVersion = typeof params.contractVersion === 'string' && params.contractVersion.trim()
+      ? params.contractVersion.trim()
+      : 'flare.v1';
+    const lastTurnId = typeof params.lastTurnId === 'string' ? params.lastTurnId.trim() : '';
+    const token = typeof params.token === 'string' ? params.token.trim() : '';
+    const onTiming = typeof params.onTiming === 'function' ? params.onTiming : null;
+    const authProvider = (
+      params.auth
+      && typeof params.auth === 'object'
+      && !Array.isArray(params.auth)
+    ) ? params.auth : null;
+
+    const nowMs = Date.now();
+    onTiming?.({
+      point: 't_submit',
+      at_ms: nowMs,
+      request_id: clientRequestId,
+      session_id: String(sessionId || '').trim(),
+    });
+
+    const waitAuthReady = async () => {
+      if (!authProvider || typeof authProvider.ensureReady !== 'function') {
+        return {
+          authorizationToken: token,
+          authReadyAtMs: nowMs,
+        };
+      }
+      if (!this.authReadyPromise) {
+        this.authReadyPromise = Promise.resolve()
+          .then(() => authProvider.ensureReady())
+          .then((resolved) => {
+            this.authReadyAtMs = Date.now();
+            return resolved || null;
+          })
+          .finally(() => {
+            this.authReadyPromise = null;
+          });
+      }
+      const authResult = await this.authReadyPromise;
+      const tokenFromAuthResult = typeof authResult?.token === 'string' ? authResult.token.trim() : '';
+      const tokenFromProvider = typeof authProvider.getToken === 'function'
+        ? String(authProvider.getToken() || '').trim()
+        : '';
+      return {
+        authorizationToken: tokenFromAuthResult || tokenFromProvider || token,
+        authReadyAtMs: this.authReadyAtMs || Date.now(),
+      };
+    };
+
+    const currentRequestId = clientRequestId;
+    if (this.abortController) {
+      this.abortController.abort();
+      this.abortController = null;
+    }
     this.abortController = new AbortController();
+    this.activeRequestId = currentRequestId;
 
     try {
+      const authResolved = await waitAuthReady();
+      onTiming?.({
+        point: 't_auth_ready',
+        at_ms: authResolved.authReadyAtMs || Date.now(),
+        request_id: currentRequestId,
+        session_id: String(sessionId || '').trim(),
+      });
+      if (this.activeRequestId !== currentRequestId) {
+        return;
+      }
       const response = await fetch(url, {
         method: 'POST',
         headers: {
           'Content-Type': 'application/json',
+          ...(authResolved.authorizationToken ? { Authorization: `Bearer ${authResolved.authorizationToken}` } : {}),
         },
         body: JSON.stringify({
+          contract_version: contractVersion,
+          client_request_id: clientRequestId,
+          command,
+          last_turn_id: lastTurnId || null,
           message: content,
           session_id: sessionId,
           enabled_capabilities: enabledCapabilities,
+          ...(modeKey ? { mode: modeKey } : {}),
+          ...(manualModeKey ? { manual_mode: manualModeKey } : {}),
+          payload: {
+            message: content,
+            ...(modeKey ? { mode: modeKey } : {}),
+            ...(manualModeKey ? { manual_mode: manualModeKey } : {}),
+            ...payloadExtra,
+          },
         }),
         signal: this.abortController.signal,
       });
+      onTiming?.({
+        point: 't_stream_connected',
+        at_ms: Date.now(),
+        request_id: currentRequestId,
+        session_id: String(sessionId || '').trim(),
+      });
 
       if (!response.ok) {
         throw new Error(`HTTP ${response.status}: ${response.statusText}`);
@@ -88,13 +194,20 @@ export class SSEClient {
 
           try {
             const eventData = JSON.parse(data);
+            if (this.activeRequestId !== currentRequestId) {
+              return;
+            }
             onEvent?.({
               type: currentEvent,
               data: eventData,
             });
 
-            if (currentEvent === 'complete') {
-              onComplete?.();
+            if (currentEvent === 'complete' || currentEvent === 'done') {
+              await onComplete?.();
+              if (this.activeRequestId === currentRequestId) {
+                this.abortController = null;
+                this.activeRequestId = '';
+              }
               return;
             }
           } catch (error) {
@@ -103,13 +216,21 @@ export class SSEClient {
         }
       }
 
-      onComplete?.();
+      await onComplete?.();
+      if (this.activeRequestId === currentRequestId) {
+        this.abortController = null;
+        this.activeRequestId = '';
+      }
     } catch (error) {
       if (error?.name === 'AbortError') {
         return;
       }
 
       onError?.(error instanceof Error ? error.message : '发送消息失败');
+      if (this.activeRequestId === currentRequestId) {
+        this.abortController = null;
+        this.activeRequestId = '';
+      }
     }
   }
 
@@ -118,6 +239,6 @@ export class SSEClient {
       this.abortController.abort();
       this.abortController = null;
     }
+    this.activeRequestId = '';
   }
 }
-

package/src/chat-core/stream/sse-events.js

@@ -145,6 +145,49 @@ export const parseContent = (raw) => ({
   chunk: raw?.content ?? '',
 });
 
+export const parseAck = (raw) => ({
+  trace_id: String(raw?.trace_id || '').trim(),
+  session_id: String(raw?.session_id || '').trim(),
+  request_id: String(raw?.request_id || '').trim(),
+  mode: String(raw?.mode || '').trim(),
+  ts: Number.isFinite(Number(raw?.ts)) ? Number(raw.ts) : null,
+});
+
+export const parsePhaseEvent = (raw) => ({
+  phase: String(raw?.phase || '').trim(),
+  label: String(raw?.label || '').trim(),
+  status: String(raw?.status || '').trim(),
+  progress: Number.isFinite(Number(raw?.progress)) ? Number(raw.progress) : null,
+  meta: raw?.meta && typeof raw.meta === 'object' && !Array.isArray(raw.meta) ? raw.meta : {},
+  ts: Number.isFinite(Number(raw?.ts)) ? Number(raw.ts) : null,
+});
+
+export const parsePatch = (raw) => ({
+  scope: String(raw?.scope || '').trim(),
+  payload: raw?.payload && typeof raw.payload === 'object' && !Array.isArray(raw.payload) ? raw.payload : {},
+  ts: Number.isFinite(Number(raw?.ts)) ? Number(raw.ts) : null,
+});
+
+export const parseTextDelta = (raw) => ({
+  channel: String(raw?.channel || 'assistant').trim() || 'assistant',
+  delta: String(raw?.delta || ''),
+  ts: Number.isFinite(Number(raw?.ts)) ? Number(raw.ts) : null,
+});
+
+export const parseTextReplace = (raw) => ({
+  channel: String(raw?.channel || 'assistant').trim() || 'assistant',
+  content: String(raw?.content || ''),
+  ts: Number.isFinite(Number(raw?.ts)) ? Number(raw.ts) : null,
+});
+
+export const parseDone = (raw) => ({
+  trace_id: String(raw?.trace_id || '').trim(),
+  session_id: String(raw?.session_id || '').trim(),
+  request_id: String(raw?.request_id || '').trim(),
+  status: String(raw?.status || 'done').trim() || 'done',
+  ts: Number.isFinite(Number(raw?.ts)) ? Number(raw.ts) : null,
+});
+
 export const parseTrace = (raw) => ({
   trace: raw?.trace ?? raw?.detail ?? '',
   summary: raw?.summary ?? raw?.label ?? '',
@@ -169,9 +212,49 @@ function pickFirstArray(payload, keys) {
   return [];
 }
 
+function parseGuidedSession(rawGuidedSession) {
+  const payload = (rawGuidedSession && typeof rawGuidedSession === 'object' && !Array.isArray(rawGuidedSession))
+    ? rawGuidedSession
+    : {};
+  const questionQueue = (payload.question_queue && typeof payload.question_queue === 'object' && !Array.isArray(payload.question_queue))
+    ? payload.question_queue
+    : {};
+  const summary = (payload.summary && typeof payload.summary === 'object' && !Array.isArray(payload.summary))
+    ? payload.summary
+    : {};
+  const decisionPoint = (payload.decision_point && typeof payload.decision_point === 'object' && !Array.isArray(payload.decision_point))
+    ? payload.decision_point
+    : {};
+  return {
+    state: payload?.state ?? '',
+    active: payload?.active === true,
+    current_question: payload?.current_question && typeof payload.current_question === 'object' && !Array.isArray(payload.current_question)
+      ? payload.current_question
+      : null,
+    question_queue: {
+      total: Number.isFinite(Number(questionQueue?.total)) ? Number(questionQueue.total) : 0,
+      answered: Number.isFinite(Number(questionQueue?.answered)) ? Number(questionQueue.answered) : 0,
+      remaining: Number.isFinite(Number(questionQueue?.remaining)) ? Number(questionQueue.remaining) : 0,
+      pending_fields: Array.isArray(questionQueue?.pending_fields) ? questionQueue.pending_fields : [],
+    },
+    summary: {
+      text: summary?.text ?? '',
+      ready: summary?.ready === true,
+    },
+    decision_point: {
+      active: decisionPoint?.active === true,
+      actions: Array.isArray(decisionPoint?.actions) ? decisionPoint.actions : [],
+    },
+  };
+}
+
 export const parseFieldProgress = (raw) => {
   const payload = resolveModeEventPayload(raw);
   return {
+    intake_session_id: payload?.intake_session_id ?? '',
+    intake_session_state: payload?.intake_session_state ?? '',
+    intake_session_completed: payload?.intake_session_completed === true,
+    flow_state: payload?.flow_state ?? payload?.collection_phase ?? '',
     fields: payload?.fields ?? {},
     sources: payload?.sources ?? {},
     field_entries: Array.isArray(payload?.field_entries) ? payload.field_entries : [],
@@ -197,9 +280,29 @@ export const parseFieldProgress = (raw) => {
     required_fields: Array.isArray(payload?.required_fields) ? payload.required_fields : [],
     recommended_fields: Array.isArray(payload?.recommended_fields) ? payload.recommended_fields : [],
     optional_fields: Array.isArray(payload?.optional_fields) ? payload.optional_fields : [],
+    intake_core_fields: Array.isArray(payload?.intake_core_fields) ? payload.intake_core_fields : [],
+    intake_supplementary_fields: Array.isArray(payload?.intake_supplementary_fields) ? payload.intake_supplementary_fields : [],
+    analysis_enrichment_fields: Array.isArray(payload?.analysis_enrichment_fields) ? payload.analysis_enrichment_fields : [],
     field_priorities: payload?.field_priorities && typeof payload.field_priorities === 'object' && !Array.isArray(payload.field_priorities)
       ? payload.field_priorities
       : {},
+    current_question: payload?.current_question && typeof payload.current_question === 'object' && !Array.isArray(payload.current_question)
+      ? payload.current_question
+      : null,
+    question_progress: payload?.question_progress && typeof payload.question_progress === 'object' && !Array.isArray(payload.question_progress)
+      ? payload.question_progress
+      : { current: 0, total: 0 },
+    collection_phase: payload?.collection_phase ?? '',
+    required_missing_count: Number.isFinite(payload?.required_missing_count) ? payload.required_missing_count : null,
+    required_coverage: Number.isFinite(Number(payload?.required_coverage)) ? Number(payload.required_coverage) : 0,
+    total_coverage: Number.isFinite(Number(payload?.total_coverage)) ? Number(payload.total_coverage) : 0,
+    analysis_entry_threshold: Number.isFinite(Number(payload?.analysis_entry_threshold)) ? Number(payload.analysis_entry_threshold) : 0.8,
+    analysis_entry_eligible: payload?.analysis_entry_eligible === true,
+    active_collecting: payload?.active_collecting === true,
+    ready_for_submit: payload?.ready_for_submit === true,
+    decision_point_active: payload?.decision_point_active === true,
+    has_active_question: payload?.has_active_question === true,
+    guided_session: parseGuidedSession(payload?.guided_session),
   };
 };
 
@@ -222,6 +325,9 @@ export const parseRequirementDraft = (raw) => {
     : {};
 
   return {
+    intake_session_id: payload?.intake_session_id ?? '',
+    intake_session_state: payload?.intake_session_state ?? '',
+    intake_session_completed: payload?.intake_session_completed === true,
     mode_key: payload?.mode_key ?? '',
     project_id: payload?.project_id ?? null,
     message_excerpt: payload?.message_excerpt ?? '',
@@ -239,7 +345,11 @@ export const parseNextActions = (raw) => {
     ? payload.actions
     : (Array.isArray(payload?.next_actions) ? payload.next_actions : []);
   return {
+    intake_session_id: payload?.intake_session_id ?? '',
+    intake_session_state: payload?.intake_session_state ?? '',
+    intake_session_completed: payload?.intake_session_completed === true,
     mode_key: payload?.mode_key ?? '',
+    flow_state: payload?.flow_state ?? payload?.collection_phase ?? '',
     project_id: payload?.project_id ?? null,
     message_excerpt: payload?.message_excerpt ?? '',
     actions,
@@ -247,10 +357,37 @@ export const parseNextActions = (raw) => {
     required_missing: Array.isArray(payload?.required_missing) ? payload.required_missing : [],
     recommended_missing: Array.isArray(payload?.recommended_missing) ? payload.recommended_missing : [],
     optional_missing: Array.isArray(payload?.optional_missing) ? payload.optional_missing : [],
+    intake_core_fields: Array.isArray(payload?.intake_core_fields) ? payload.intake_core_fields : [],
+    intake_supplementary_fields: Array.isArray(payload?.intake_supplementary_fields) ? payload.intake_supplementary_fields : [],
+    analysis_enrichment_fields: Array.isArray(payload?.analysis_enrichment_fields) ? payload.analysis_enrichment_fields : [],
     ready_for_sourcing: payload?.ready_for_sourcing ?? null,
     status: payload?.status ?? '',
     reason: payload?.reason ?? '',
+    chooser_required: payload?.chooser_required === true,
+    blocking_reason: payload?.blocking_reason ?? '',
+    blocking: payload?.blocking && typeof payload.blocking === 'object' && !Array.isArray(payload.blocking)
+      ? payload.blocking
+      : null,
+    target_field: payload?.target_field ?? '',
+    degrade_reason: Array.isArray(payload?.degrade_reason) ? payload.degrade_reason : [],
+    current_question: payload?.current_question && typeof payload.current_question === 'object' && !Array.isArray(payload.current_question)
+      ? payload.current_question
+      : null,
+    question_progress: payload?.question_progress && typeof payload.question_progress === 'object' && !Array.isArray(payload.question_progress)
+      ? payload.question_progress
+      : { current: 0, total: 0 },
+    collection_phase: payload?.collection_phase ?? '',
+    required_missing_count: Number.isFinite(payload?.required_missing_count) ? payload.required_missing_count : null,
     render_hint: payload?.render_hint ?? 'next_actions',
+    required_coverage: Number.isFinite(Number(payload?.required_coverage)) ? Number(payload.required_coverage) : 0,
+    total_coverage: Number.isFinite(Number(payload?.total_coverage)) ? Number(payload.total_coverage) : 0,
+    analysis_entry_threshold: Number.isFinite(Number(payload?.analysis_entry_threshold)) ? Number(payload.analysis_entry_threshold) : 0.8,
+    analysis_entry_eligible: payload?.analysis_entry_eligible === true,
+    active_collecting: payload?.active_collecting === true,
+    ready_for_submit: payload?.ready_for_submit === true,
+    decision_point_active: payload?.decision_point_active === true,
+    has_active_question: payload?.has_active_question === true,
+    guided_session: parseGuidedSession(payload?.guided_session),
   };
 };
 
@@ -378,6 +515,151 @@ export const parseEvaluationReportReady = (raw) => {
   };
 };
 
+export const parsePlanBlock = (raw) => {
+  const payload = resolveModeEventPayload(raw);
+  return {
+    ...payload,
+    constraints: Array.isArray(payload?.constraints) ? payload.constraints : [],
+    plan: Array.isArray(payload?.plan) ? payload.plan : [],
+    acceptance: Array.isArray(payload?.acceptance) ? payload.acceptance : [],
+    actions: Array.isArray(payload?.actions) ? payload.actions : [],
+  };
+};
+
+export const parseCanvasState = (raw) => {
+  const payload = resolveModeEventPayload(raw);
+  const canvasState = payload?.canvas_state && typeof payload.canvas_state === 'object' && !Array.isArray(payload.canvas_state)
+    ? payload.canvas_state
+    : {};
+  return {
+    ...payload,
+    canvas_state: {
+      ...canvasState,
+      versions: Array.isArray(canvasState.versions) ? canvasState.versions : [],
+    },
+  };
+};
+
+export const parseCanvasRevision = (raw) => {
+  const payload = resolveModeEventPayload(raw);
+  return {
+    ...payload,
+    actions: Array.isArray(payload?.actions) ? payload.actions : [],
+  };
+};
+
+export const parseKnowledgeSearch = (raw) => {
+  const payload = resolveModeEventPayload(raw);
+  const sourceBreakdown = payload?.source_breakdown && typeof payload.source_breakdown === 'object'
+    && !Array.isArray(payload.source_breakdown)
+    ? payload.source_breakdown
+    : {};
+  return {
+    ...payload,
+    run_id: payload?.run_id ?? '',
+    query: payload?.query ?? '',
+    result_count: Number.isFinite(payload?.result_count) ? payload.result_count : null,
+    source_breakdown: {
+      local: Number.isFinite(sourceBreakdown.local) ? sourceBreakdown.local : 0,
+      mcp: Number.isFinite(sourceBreakdown.mcp) ? sourceBreakdown.mcp : 0,
+      web: Number.isFinite(sourceBreakdown.web) ? sourceBreakdown.web : 0,
+    },
+    results: Array.isArray(payload?.results) ? payload.results : [],
+  };
+};
+
+export const parseKnowledgeCitation = (raw) => {
+  const payload = resolveModeEventPayload(raw);
+  return {
+    ...payload,
+    run_id: payload?.run_id ?? '',
+    citations: Array.isArray(payload?.citations) ? payload.citations : [],
+  };
+};
+
+export const parseOrchestrationStatus = (raw) => {
+  const payload = resolveModeEventPayload(raw);
+  const activeWorkItem = payload?.active_work_item && typeof payload.active_work_item === 'object' && !Array.isArray(payload.active_work_item)
+    ? payload.active_work_item
+    : null;
+  return {
+    trace_id: payload?.trace_id ?? '',
+    intake_session_id: payload?.intake_session_id ?? '',
+    intake_session_state: payload?.intake_session_state ?? '',
+    intake_session_completed: payload?.intake_session_completed === true,
+    current_stage: payload?.current_stage ?? '',
+    detected_intent: payload?.detected_intent && typeof payload.detected_intent === 'object' && !Array.isArray(payload.detected_intent)
+      ? payload.detected_intent
+      : {},
+    confirmed_updates: Array.isArray(payload?.confirmed_updates) ? payload.confirmed_updates : [],
+    open_fields: Array.isArray(payload?.open_fields) ? payload.open_fields : [],
+    search_actions: Array.isArray(payload?.search_actions) ? payload.search_actions : [],
+    workspace_updates: payload?.workspace_updates && typeof payload.workspace_updates === 'object' && !Array.isArray(payload.workspace_updates)
+      ? payload.workspace_updates
+      : {},
+    next_action: payload?.next_action && typeof payload.next_action === 'object' && !Array.isArray(payload.next_action)
+      ? payload.next_action
+      : {},
+    next_actions: Array.isArray(payload?.next_actions) ? payload.next_actions : [],
+    active_collecting: payload?.active_collecting === true,
+    ready_for_submit: payload?.ready_for_submit === true,
+    required_coverage: Number.isFinite(Number(payload?.required_coverage)) ? Number(payload.required_coverage) : 0,
+    total_coverage: Number.isFinite(Number(payload?.total_coverage)) ? Number(payload.total_coverage) : 0,
+    analysis_entry_threshold: Number.isFinite(Number(payload?.analysis_entry_threshold)) ? Number(payload.analysis_entry_threshold) : 0.8,
+    analysis_entry_eligible: payload?.analysis_entry_eligible === true,
+    decision_point_active: payload?.decision_point_active === true,
+    has_active_question: payload?.has_active_question === true,
+    guided_session: parseGuidedSession(payload?.guided_session),
+    active_work_item: activeWorkItem,
+    active_fields: Array.isArray(payload?.active_fields) ? payload.active_fields : [],
+    hypotheses: Array.isArray(payload?.hypotheses) ? payload.hypotheses : [],
+    planner: payload?.planner && typeof payload.planner === 'object' && !Array.isArray(payload.planner)
+      ? payload.planner
+      : {},
+    chooser: payload?.chooser && typeof payload.chooser === 'object' && !Array.isArray(payload.chooser)
+      ? payload.chooser
+      : null,
+    stage_transition: payload?.stage_transition && typeof payload.stage_transition === 'object' && !Array.isArray(payload.stage_transition)
+      ? payload.stage_transition
+      : {},
+    decision_basis: Array.isArray(payload?.decision_basis) ? payload.decision_basis : [],
+    blocking: payload?.blocking && typeof payload.blocking === 'object' && !Array.isArray(payload.blocking)
+      ? payload.blocking
+      : {},
+    node: payload?.node && typeof payload.node === 'object' && !Array.isArray(payload.node)
+      ? payload.node
+      : {},
+    node_id: payload?.node_id ?? '',
+    node_state: payload?.node_state ?? '',
+    node_reason: payload?.node_reason ?? '',
+    needs_user_choice: payload?.needs_user_choice === true,
+    conversation_intent_state: payload?.conversation_intent_state ?? '',
+    session_intent_stage: payload?.session_intent_stage ?? '',
+    intent_type: payload?.intent_type ?? '',
+    procurement_relevance: payload?.procurement_relevance === true,
+    consultative_procurement_qa: payload?.consultative_procurement_qa === true,
+    escalation_target: payload?.escalation_target ?? '',
+    intake_candidate: payload?.intake_candidate === true,
+    clarification_recommended: payload?.clarification_recommended === true,
+    analysis_ready: payload?.analysis_ready === true,
+    draft_seeded: payload?.draft_seeded === true,
+    missing_key_fields: Array.isArray(payload?.missing_key_fields) ? payload.missing_key_fields : [],
+    recommended_next_step: payload?.recommended_next_step ?? '',
+    intent_and_escalation_confidence: Number.isFinite(Number(payload?.intent_and_escalation_confidence))
+      ? Number(payload.intent_and_escalation_confidence)
+      : 0,
+    intent_and_escalation_reason: payload?.intent_and_escalation_reason ?? '',
+    intent_escalation_policy: payload?.intent_escalation_policy && typeof payload.intent_escalation_policy === 'object' && !Array.isArray(payload.intent_escalation_policy)
+      ? payload.intent_escalation_policy
+      : {},
+    intent_escalation_policy_source: payload?.intent_escalation_policy_source ?? '',
+    degrade_reason: Array.isArray(payload?.degrade_reason) ? payload.degrade_reason : [],
+    content_policy: payload?.content_policy && typeof payload.content_policy === 'object' && !Array.isArray(payload.content_policy)
+      ? payload.content_policy
+      : {},
+  };
+};
+
 export const parseWorkspaceActivation = (raw) => ({
   function_type: raw?.function_type ?? '',
   mode: raw?.mode ?? '',
@@ -420,6 +702,166 @@ export const parseInstanceProfile = (raw) => ({
   ui_labels: raw?.instance_profile?.ui_labels ?? raw?.ui_labels ?? {},
 });
 
+export const parseModeRuntime = (raw) => ({
+  mode_key: raw?.mode_key ?? '',
+  mode_state: raw?.mode_state ?? '',
+});
+
+export const parseAgentRuntime = (raw) => ({
+  trace_id: raw?.trace_id ?? '',
+  session_id: raw?.session_id ?? '',
+  intent: raw?.intent ?? '',
+  function_type: raw?.function_type ?? '',
+  mode_key: raw?.mode_key ?? '',
+  mode_state: raw?.mode_state ?? '',
+  agent_step_count: Number.isFinite(raw?.agent_step_count) ? raw.agent_step_count : 0,
+  agent_step: Array.isArray(raw?.agent_step) ? raw.agent_step : [],
+});
+
+export const parseSkillRuntime = (raw) => ({
+  trace_id: raw?.trace_id ?? '',
+  session_id: raw?.session_id ?? '',
+  intent: raw?.intent ?? '',
+  function_type: raw?.function_type ?? '',
+  mode_key: raw?.mode_key ?? '',
+  mode_state: raw?.mode_state ?? '',
+  skill_call_count: Number.isFinite(raw?.skill_call_count) ? raw.skill_call_count : 0,
+  skill_call: Array.isArray(raw?.skill_call) ? raw.skill_call : [],
+});
+
+export const parseModeSwitchReason = (raw) => ({
+  mode_key: raw?.mode_key ?? '',
+  intent: raw?.intent ?? '',
+  source: raw?.source ?? '',
+  previous_mode: raw?.previous_mode ?? '',
+  manual_mode: raw?.manual_mode ?? '',
+  current_mode: raw?.current_mode ?? '',
+  intent_override: raw?.intent_override ?? '',
+  summary: raw?.summary ?? '',
+});
+
 export const parseSSEError = (raw) => ({
   message: raw?.message ?? '发生错误',
 });
+
+export const PATCH_SCOPE_AUTHORITATIVE = Object.freeze([
+  'session',
+  'assistant_reply',
+  'primary_flow',
+  'checkpoint',
+  'intent_state',
+  'question',
+  'confirmed_fields',
+  'missing_fields',
+  'next_actions',
+  'analysis',
+  'observation',
+  'capabilities.field_retrieval',
+  'errors',
+]);
+
+export const PATCH_SCOPE_LEGACY_COMPAT = Object.freeze([
+  'current_question',
+  'recommendation_summary',
+]);
+
+const PATCH_SCOPE_ALLOWLIST = new Set([
+  ...PATCH_SCOPE_AUTHORITATIVE,
+  ...PATCH_SCOPE_LEGACY_COMPAT,
+]);
+
+function getByScopePath(payload, scope) {
+  if (!payload || typeof payload !== 'object' || Array.isArray(payload)) {
+    return undefined;
+  }
+  const segments = String(scope || '').split('.').filter(Boolean);
+  if (!segments.length) {
+    return undefined;
+  }
+  let current = payload;
+  for (const segment of segments) {
+    if (!current || typeof current !== 'object' || Array.isArray(current) || !(segment in current)) {
+      return undefined;
+    }
+    current = current[segment];
+  }
+  return current;
+}
+
+export const parsePatchEvent = (raw) => {
+  const envelope = (raw && typeof raw === 'object' && !Array.isArray(raw)) ? raw : {};
+  const patchScope = Array.isArray(envelope.patch_scope)
+    ? envelope.patch_scope.map((item) => String(item || '').trim()).filter(Boolean)
+    : [];
+  const payload = (envelope.payload && typeof envelope.payload === 'object' && !Array.isArray(envelope.payload))
+    ? envelope.payload
+    : {};
+  const hasPayload = Object.keys(payload).length > 0;
+  const hasUnknownScope = patchScope.some((item) => !(PATCH_SCOPE_ALLOWLIST.has(item) || item.startsWith('confirmed_fields.')));
+  const hasScopePayloadMismatch = patchScope.some((scope) => {
+    const direct = payload[scope];
+    const byPath = getByScopePath(payload, scope);
+    const lastSegment = String(scope).split('.').slice(-1)[0];
+    const leaf = payload[lastSegment];
+    return direct === undefined && byPath === undefined && leaf === undefined;
+  });
+  const legacyPatchScope = patchScope.filter((scope) => PATCH_SCOPE_LEGACY_COMPAT.includes(scope));
+  const authoritativePatchScope = patchScope.filter(
+    (scope) => PATCH_SCOPE_AUTHORITATIVE.includes(scope) || scope.startsWith('confirmed_fields.')
+  );
+  const invalidReason = (() => {
+    if (envelope.message_type !== 'patch_event') {
+      return 'invalid_message_type';
+    }
+    if (!['ack', 'patch', 'error', 'final'].includes(String(envelope.event_type || '').trim())) {
+      return 'invalid_event_type';
+    }
+    if (hasPayload && patchScope.length === 0) {
+      return 'payload_without_patch_scope';
+    }
+    if (hasUnknownScope) {
+      return 'unknown_patch_scope';
+    }
+    if (patchScope.length > 0 && hasScopePayloadMismatch) {
+      return 'patch_scope_payload_mismatch';
+    }
+    return '';
+  })();
+  return {
+    contract_version: String(envelope.contract_version || 'flare.v1'),
+    message_type: 'patch_event',
+    event_type: String(envelope.event_type || '').trim(),
+    sequence: Number.isFinite(Number(envelope.sequence)) ? Number(envelope.sequence) : 0,
+    trace_id: String(envelope.trace_id || '').trim(),
+    client_request_id: String(envelope.client_request_id || '').trim(),
+    intake_session_id: String(
+      payload?.primary_flow?.intake_session_id
+      || payload?.checkpoint?.intake_session_id
+      || payload?.next_actions?.intake_session_id
+      || payload?.question?.intake_session_id
+      || payload?.intake_session_id
+      || ''
+    ).trim(),
+    intake_session_state: String(
+      payload?.primary_flow?.intake_session_state
+      || payload?.next_actions?.intake_session_state
+      || payload?.intake_session_state
+      || ''
+    ).trim(),
+    intake_session_completed: Boolean(
+      payload?.primary_flow?.intake_session_completed
+      || payload?.next_actions?.intake_session_completed
+      || payload?.intake_session_completed
+    ),
+    session: (
+      envelope.session && typeof envelope.session === 'object' && !Array.isArray(envelope.session)
+    ) ? envelope.session : { session_id: '', turn_id: '' },
+    patch_scope: patchScope,
+    authoritative_patch_scope: authoritativePatchScope,
+    legacy_patch_scope: legacyPatchScope,
+    payload,
+    final: envelope.final === true,
+    invalid: Boolean(invalidReason),
+    invalid_reason: invalidReason,
+  };
+};

package/src/chat-core/stream/useSSEStream.js

@@ -1,10 +1,16 @@
-import { useRef, useState, useCallback } from 'react';
+import { useRef, useState, useCallback, useEffect } from 'react';
 import { SSEClient } from './sse-client.js';
 import {
+  parseAck,
   parseAgentStatus,
   parseThinkingTrace,
   parseExecutionTrace,
   parseContent,
+  parseDone,
+  parsePatch,
+  parsePhaseEvent,
+  parseTextDelta,
+  parseTextReplace,
   parseTrace,
   parseFieldProgress,
   parseDoc,
@@ -21,6 +27,17 @@ import {
   parseCapabilitySuggestion,
   parseFieldsUpdated,
   parseInstanceProfile,
+  parseModeRuntime,
+  parseAgentRuntime,
+  parseSkillRuntime,
+  parseModeSwitchReason,
+  parseKnowledgeSearch,
+  parseKnowledgeCitation,
+  parseOrchestrationStatus,
+  parseCanvasState,
+  parseCanvasRevision,
+  parsePlanBlock,
+  parsePatchEvent,
   parseSSEError,
 } from './sse-events.js';
 
@@ -42,19 +59,28 @@ export function useSSEStream(sessionId) {
   const lastMessageRef = useRef({ content: '', handlers: {} });
 
   const send = useCallback(async (content, handlers = {}, options = {}) => {
-    const resolvedSessionId = options.sessionIdOverride || sessionId;
-    if (!resolvedSessionId || !content.trim()) return;
+    const hasSessionOverride = Object.prototype.hasOwnProperty.call(options, 'sessionIdOverride');
+    const resolvedSessionId = hasSessionOverride ? options.sessionIdOverride : sessionId;
+    if (!content.trim()) return;
 
     lastMessageRef.current = { content, handlers, options };
     setError(null);
     accumulatedRef.current = '';
     setLoading(true);
+    let latestSessionId = String(resolvedSessionId || '').trim();
 
     const {
+      onAck,
       onAgentStatus,
       onThinkingTrace,
       onExecutionTrace,
       onContent,
+      onTextReplace,
+      onPhaseStart,
+      onPhaseUpdate,
+      onPhaseEnd,
+      onPatch,
+      onDone,
       onTrace,
       onWorkspaceActivation,
       onUICards,
@@ -69,14 +95,40 @@ export function useSSEStream(sessionId) {
       onCategoryIdentified,
       onFieldsUpdated,
       onInstanceProfile,
+      onModeRuntime,
+      onAgentRuntime,
+      onSkillRuntime,
+      onModeSwitchReason,
+      onKnowledgeSearch,
+      onKnowledgeCitation,
+      onOrchestrationStatus,
+      onCanvasState,
+      onCanvasRevision,
+      onPlanBlock,
       onDoc,
       onCapabilitySuggestion,
+      onPatchEvent,
       onError,
       onComplete,
     } = handlers;
 
     const eventHandler = (event) => {
       switch (event.type) {
+        case 'ack':
+          onAck?.(parseAck(event.data));
+          break;
+        case 'phase.start':
+          onPhaseStart?.(parsePhaseEvent(event.data));
+          break;
+        case 'phase.update':
+          onPhaseUpdate?.(parsePhaseEvent(event.data));
+          break;
+        case 'phase.end':
+          onPhaseEnd?.(parsePhaseEvent(event.data));
+          break;
+        case 'patch':
+          onPatch?.(parsePatch(event.data));
+          break;
         case 'agent_status':
           onAgentStatus?.(parseAgentStatus(event.data));
           break;
@@ -96,6 +148,25 @@ export function useSSEStream(sessionId) {
           onContent?.(chunk);
           break;
         }
+        case 'text.delta': {
+          const parsed = parseTextDelta(event.data);
+          if (parsed.channel === 'assistant' && parsed.delta) {
+            accumulatedRef.current += parsed.delta;
+            onContent?.(parsed.delta);
+          }
+          break;
+        }
+        case 'text.replace': {
+          const parsed = parseTextReplace(event.data);
+          if (parsed.channel === 'assistant') {
+            accumulatedRef.current = parsed.content;
+            onTextReplace?.(parsed.content);
+          }
+          break;
+        }
+        case 'done':
+          onDone?.(parseDone(event.data));
+          break;
         case 'trace':
           onTrace?.(parseTrace(event.data));
           break;
@@ -138,12 +209,52 @@ export function useSSEStream(sessionId) {
         case 'instance_profile':
           onInstanceProfile?.(parseInstanceProfile(event.data));
           break;
+        case 'mode_runtime':
+          onModeRuntime?.(parseModeRuntime(event.data));
+          break;
+        case 'agent_runtime':
+          onAgentRuntime?.(parseAgentRuntime(event.data));
+          break;
+        case 'skill_runtime':
+          onSkillRuntime?.(parseSkillRuntime(event.data));
+          break;
+        case 'mode_switch_reason':
+          onModeSwitchReason?.(parseModeSwitchReason(event.data));
+          break;
+        case 'knowledge_search':
+          onKnowledgeSearch?.(parseKnowledgeSearch(event.data));
+          break;
+        case 'knowledge_citation':
+          onKnowledgeCitation?.(parseKnowledgeCitation(event.data));
+          break;
+        case 'orchestration_status':
+          onOrchestrationStatus?.(parseOrchestrationStatus(event.data));
+          break;
+        case 'canvas_state':
+          onCanvasState?.(parseCanvasState(event.data));
+          break;
+        case 'canvas_revision':
+          onCanvasRevision?.(parseCanvasRevision(event.data));
+          break;
+        case 'plan_block':
+          onPlanBlock?.(parsePlanBlock(event.data));
+          break;
         case 'doc':
           onDoc?.(parseDoc(event.data));
           break;
         case 'capability_suggestion':
           onCapabilitySuggestion?.(parseCapabilitySuggestion(event.data));
           break;
+        case 'patch_event':
+          {
+            const parsedPatchEvent = parsePatchEvent(event.data);
+            const patchSessionId = String(parsedPatchEvent?.session?.session_id || '').trim();
+            if (patchSessionId) {
+              latestSessionId = patchSessionId;
+            }
+            onPatchEvent?.(parsedPatchEvent);
+          }
+          break;
         case 'error':
           onError?.(parseSSEError(event.data));
           break;
@@ -152,9 +263,11 @@ export function useSSEStream(sessionId) {
       }
     };
 
-    const completeHandler = () => {
+    const completeHandler = async () => {
       setLoading(false);
-      onComplete?.(accumulatedRef.current);
+      await onComplete?.(accumulatedRef.current, {
+        sessionId: latestSessionId || null,
+      });
       accumulatedRef.current = '';
     };
 
@@ -167,21 +280,51 @@ export function useSSEStream(sessionId) {
     };
 
     const requestOptions = {
-      sessionId: resolvedSessionId,
+      sessionId: String(resolvedSessionId || '').trim() || null,
       content,
       enabledCapabilities: Array.isArray(options.enabledCapabilities) ? options.enabledCapabilities : [],
+      modeKey: typeof options.modeKey === 'string' ? options.modeKey : '',
+      manualModeKey: typeof options.manualModeKey === 'string' ? options.manualModeKey : '',
+      command: typeof options.command === 'string' ? options.command : '',
+      clientRequestId: typeof options.clientRequestId === 'string' ? options.clientRequestId : '',
+      contractVersion: typeof options.contractVersion === 'string' ? options.contractVersion : '',
+      lastTurnId: typeof options.lastTurnId === 'string' ? options.lastTurnId : '',
+      payloadExtra: (
+        options.payloadExtra
+        && typeof options.payloadExtra === 'object'
+        && !Array.isArray(options.payloadExtra)
+      )
+        ? options.payloadExtra
+        : {},
       endpoint: options.endpoint,
       baseUrl: options.baseUrl,
       url: options.url,
+      token: typeof options.token === 'string' ? options.token : '',
+      auth: (
+        options.auth
+        && typeof options.auth === 'object'
+        && !Array.isArray(options.auth)
+      ) ? options.auth : undefined,
+      onTiming: typeof options.onTiming === 'function' ? options.onTiming : undefined,
     };
 
     if (!requestOptions.enabledCapabilities.length) {
       delete requestOptions.enabledCapabilities;
     }
+    if (!requestOptions.modeKey) delete requestOptions.modeKey;
+    if (!requestOptions.manualModeKey) delete requestOptions.manualModeKey;
+    if (!requestOptions.command) delete requestOptions.command;
+    if (!requestOptions.clientRequestId) delete requestOptions.clientRequestId;
+    if (!requestOptions.contractVersion) delete requestOptions.contractVersion;
+    if (!requestOptions.lastTurnId) delete requestOptions.lastTurnId;
+    if (!Object.keys(requestOptions.payloadExtra || {}).length) delete requestOptions.payloadExtra;
 
     if (!requestOptions.endpoint) delete requestOptions.endpoint;
     if (!requestOptions.baseUrl) delete requestOptions.baseUrl;
     if (!requestOptions.url) delete requestOptions.url;
+    if (!requestOptions.token) delete requestOptions.token;
+    if (!requestOptions.auth) delete requestOptions.auth;
+    if (!requestOptions.onTiming) delete requestOptions.onTiming;
 
     await clientRef.current.sendMessage(
       requestOptions,
@@ -203,6 +346,10 @@ export function useSSEStream(sessionId) {
     accumulatedRef.current = '';
   }, []);
 
+  useEffect(() => () => {
+    clientRef.current.abort();
+  }, []);
+
   return { send, loading, error, retry, abort };
 }