@huyooo/ai-chat-frontend-react 0.2.14 → 0.2.16

package/src/hooks/useChat.ts

@@ -1,1569 +0,0 @@
-/**
- * useChat Hook
- * 管理聊天状态和与后端的通信
- * 
- * 新架构：
- * - 使用 Map 存储每个会话的独立状态
- * - 多会话可同时进行，互不干扰
- * - 切换 tab 不会停止正在进行的请求
- * - ContentPart 数组存储消息内容，支持流式渲染和自定义 UI
- */
-
-import { useState, useCallback, useRef, useMemo, useEffect } from 'react'
-import type {
-  ChatAdapter,
-  ChatEvent as BridgeChatEvent,
-  ChatMode,
-  SessionRecord,
-  MessageRecord,
-  AutoRunConfig,
-} from '@huyooo/ai-chat-bridge-electron/renderer'
-import type { 
-  ChatMessage,
-  ContentPart,
-  TextPart,
-  ThinkingPart,
-  SearchPart,
-  ToolCallPart,
-  ErrorPart,
-  SearchResult,
-} from '../types'
-
-// 扩展事件类型：允许 bridge 端新增事件时前端不被类型卡死
-type ChatEvent = Omit<BridgeChatEvent, 'type'> & {
-  type: BridgeChatEvent['type'] | 'tool_call_output'
-}
-
-// ==================== @ 上下文（文件引用） ====================
-// 约定：AtFilePicker 插入的格式为单行 "@<path>"，通常以换行分隔
-const AT_LINE_RE = /^\s*@\s*(.+?)\s*$/
-const MAX_FILE_CHARS = 20_000
-
-function stripAtContextLines(text: string): string {
-  return text
-    .split('\n')
-    .filter((line) => !AT_LINE_RE.test(line))
-    .join('\n')
-    .trim()
-}
-
-function extractAtContextRefs(text: string): { refs: string[]; cleanedText: string } {
-  const refs: string[] = []
-  const kept: string[] = []
-
-  for (const line of text.split('\n')) {
-    const m = line.match(AT_LINE_RE)
-    if (m?.[1]) {
-      refs.push(m[1])
-    } else {
-      kept.push(line)
-    }
-  }
-
-  return {
-    refs: Array.from(new Set(refs.map((r) => r.trim()).filter(Boolean))),
-    cleanedText: kept.join('\n').trim(),
-  }
-}
-
-async function buildPromptWithAtContext(adapter: ChatAdapter, text: string): Promise<string> {
-  const { refs, cleanedText } = extractAtContextRefs(text)
-  if (refs.length === 0) return text
-
-  const blocks: string[] = []
-
-  for (const ref of refs) {
-    const resolved = (await adapter.resolvePath?.(ref)) || ref
-
-    try {
-      const stat = await adapter.stat?.(resolved)
-      if (stat?.isDirectory) {
-        const items = await adapter.listDir?.(resolved)
-        const listing = (items || [])
-          .slice(0, 200)
-          .map((f) => `${f.isDirectory ? 'DIR ' : 'FILE'} ${f.path}`)
-          .join('\n')
-        blocks.push(
-          `【目录】${resolved}\n` +
-          '```text\n' +
-          (listing || '(空目录/无法列出)') +
-          '\n```'
-        )
-        continue
-      }
-
-      const content = await adapter.readFile?.(resolved)
-      if (typeof content === 'string') {
-        const truncated = content.length > MAX_FILE_CHARS
-          ? content.slice(0, MAX_FILE_CHARS) + `\n\n... (已截断，原始长度 ${content.length})`
-          : content
-        blocks.push(
-          `【文件】${resolved}\n` +
-          '```text\n' +
-          truncated +
-          '\n```'
-        )
-      } else {
-        blocks.push(`【引用】${resolved}\n（无法读取文件内容）`)
-      }
-    } catch (error) {
-      blocks.push(`【引用】${resolved}\n（读取失败：${error instanceof Error ? error.message : String(error)}）`)
-    }
-  }
-
-  const question = cleanedText || stripAtContextLines(text) || text
-  return `以下是用户通过 @ 引用提供的上下文：\n\n${blocks.join('\n\n')}\n\n用户问题：\n${question}`
-}
-
-/** 生成唯一 ID */
-function generateId(): string {
-  return Date.now().toString(36) + Math.random().toString(36).substr(2)
-}
-
-/** 获取消息的纯文本内容 */
-function extractTextContent(parts: ContentPart[]): string {
-  return parts
-    .filter((p): p is TextPart => p.type === 'text')
-    .map(p => p.text)
-    .join('')
-}
-
-/** 判断事件是否需要保存到数据库 */
-function shouldSaveEvent(event: ChatEvent): boolean {
-  return (
-    event.type === 'thinking_end' ||
-    event.type === 'tool_call_result' ||
-    event.type === 'text_delta' ||
-    event.type === 'done' ||
-    event.type === 'error' ||
-    event.type === 'abort'
-  )
-}
-
-/** 解析工具调用结果 */
-function parseToolResult(result: unknown): unknown | null {
-  if (result === undefined || result === null) return null
-  if (typeof result === 'string') {
-    try {
-      return JSON.parse(result)
-    } catch {
-      return result
-    }
-  }
-  return result
-}
-
-/** 转换存储的消息为显示格式 */
-function convertToMessage(record: MessageRecord): ChatMessage {
-  // timestamp 统一为 number（毫秒时间戳），不受 JSON 序列化影响
-  // 如果收到非 number 类型，说明数据异常，直接使用 0
-  const timestamp = typeof record.timestamp === 'number' && Number.isFinite(record.timestamp)
-    ? record.timestamp
-    : 0
-  let parts: ContentPart[] = []
-  
-  if (record.steps) {
-    try {
-      const steps = JSON.parse(record.steps)
-      for (const step of steps) {
-        if (step.type === 'thinking') {
-          parts.push({
-            type: 'thinking',
-            text: step.text || '',
-            status: step.status || 'done',
-            duration: step.duration,
-          })
-        } else if (step.type === 'search') {
-          parts.push({
-            type: 'search',
-            query: step.query,
-            results: step.results,
-            status: step.status || 'done',
-          })
-        } else if (step.type === 'tool_call') {
-          parts.push({
-            type: 'tool_call',
-            id: step.id,
-            name: step.name,
-            args: step.args,
-            result: parseToolResult(step.result),
-            status: step.status || 'done',
-            output: step.output,
-          })
-        } else if (step.type === 'text') {
-          parts.push({
-            type: 'text',
-            text: step.text || '',
-          })
-        } else if (step.type === 'error') {
-          parts.push({
-            type: 'error',
-            message: step.message || '',
-            category: step.category,
-          })
-        }
-      }
-    } catch {
-      // 解析失败，忽略 steps
-    }
-  }
-  
-  if (record.content && !parts.some(p => p.type === 'text')) {
-    parts.push({ type: 'text', text: record.content })
-  }
-  
-  if (parts.length === 0) {
-    parts.push({ type: 'text', text: '' })
-  }
-
-  return {
-    id: record.id,
-    role: record.role,
-    parts,
-    images: record.images ?? [],
-    model: record.model || undefined,
-    mode: record.mode || undefined,
-    webSearchEnabled: record.webSearchEnabled ?? undefined,
-    thinkingEnabled: record.thinkingEnabled ?? undefined,
-    loading: false,
-    timestamp,
-  }
-}
-
-/** 会话状态 */
-interface SessionState {
-  messages: ChatMessage[]
-  isLoading: boolean
-  abortController: AbortController | null
-  /** 当前进行中的 assistant 消息 ID（用于取消/落地状态，避免依赖 message.loading 的不稳定性） */
-  activeAssistantMessageId: string | null
-}
-
-/** 副作用定义 */
-export interface SideEffect {
-  type: string
-  success: boolean
-  data?: unknown
-  message?: string
-}
-
-/** 工具完成事件数据 */
-export interface ToolCompleteEvent {
-  name: string
-  result: unknown
-  /** 
-   * 工具声明的副作用列表
-   * 前端可根据此字段处理通知、刷新文件列表等
-   */
-  sideEffects?: SideEffect[]
-}
-
-export interface UseChatOptions {
-  adapter: ChatAdapter
-  defaultModel?: string
-  defaultMode?: ChatMode
-  onToolComplete?: (event: ToolCompleteEvent) => void
-  autoRunConfig?: AutoRunConfig
-}
-
-/**
- * 聊天状态管理 Hook
- * 
- * 核心架构：
- * - sessionStatesRef: Map<sessionId, SessionState> 存储每个会话的独立状态
- * - 多会话可同时进行请求，互不干扰
- * - 切换 tab 不会停止任何正在进行的请求
- */
-export function useChat(options: UseChatOptions) {
-  const {
-    adapter,
-    defaultModel = 'anthropic/claude-opus-4.5',
-    defaultMode = 'agent',
-    onToolComplete,
-    // autoRunConfig 现在从数据库读取，不再从 props 传入
-  } = options
-
-  // ==================== 会话状态存储 ====================
-  const sessionStatesRef = useRef(new Map<string, SessionState>())
-  const [stateVersion, setStateVersion] = useState(0)
-
-  // ==================== 工具开关配置（从数据库读取） ====================
-  /** 启用的工具名称列表（undefined 表示全部启用） */
-  const [enabledTools, setEnabledTools] = useState<string[] | undefined>(undefined)
-  const enabledToolsRef = useRef<string[] | undefined>(enabledTools)
-  enabledToolsRef.current = enabledTools
-
-  /** 所有可用工具列表（用于设置面板） */
-  const [allTools, setAllTools] = useState<Array<{ name: string; description: string }>>([])
-
-  // ==================== 自动运行配置（从数据库读取） ====================
-  // 默认配置
-  const DEFAULT_AUTO_RUN_CONFIG: AutoRunConfig = {
-    mode: 'run-everything',
-  }
-
-  const [autoRunConfig, setAutoRunConfig] = useState<AutoRunConfig>(DEFAULT_AUTO_RUN_CONFIG)
-  const autoRunConfigRef = useRef(autoRunConfig)
-  autoRunConfigRef.current = autoRunConfig
-
-  /** 从数据库加载自动运行配置 */
-  const loadAutoRunConfig = useCallback(async () => {
-    if (!adapter.getAllSettings) return
-    
-    try {
-      const settings = await adapter.getAllSettings()
-      const configJson = settings['autoRunConfig']
-      
-      if (configJson) {
-        const config = JSON.parse(configJson) as AutoRunConfig
-        // 合并配置，确保新增字段有默认值
-        setAutoRunConfig({ ...DEFAULT_AUTO_RUN_CONFIG, ...config })
-      }
-    } catch (error) {
-      console.error('[useChat] 加载 autoRunConfig 失败:', error)
-    }
-  }, [adapter])
-
-  /** 保存自动运行配置到数据库 */
-  const saveAutoRunConfig = useCallback(async (config: AutoRunConfig) => {
-    if (!adapter.setSetting) return
-    
-    try {
-      await adapter.setSetting('autoRunConfig', JSON.stringify(config))
-      setAutoRunConfig(config)
-    } catch (error) {
-      console.error('[useChat] 保存 autoRunConfig 失败:', error)
-      throw error
-    }
-  }, [adapter])
-
-  /** 从数据库加载工具开关配置 */
-  const loadEnabledTools = useCallback(async () => {
-    if (!adapter.getAllSettings) return
-
-    try {
-      const settings = await adapter.getAllSettings()
-      const toolsJson = settings['enabledTools']
-      if (!toolsJson) {
-        setEnabledTools(undefined)
-        return
-      }
-
-      const parsed = JSON.parse(toolsJson) as unknown
-      if (Array.isArray(parsed) && parsed.every((x) => typeof x === 'string')) {
-        setEnabledTools(parsed)
-      } else {
-        setEnabledTools(undefined)
-      }
-    } catch (error) {
-      console.error('[useChat] 加载 enabledTools 失败:', error)
-    }
-  }, [adapter])
-
-  /** 保存工具开关配置到数据库 */
-  const saveEnabledTools = useCallback(async (tools: string[] | undefined) => {
-    if (!adapter.setSetting) return
-
-    try {
-      if (tools === undefined) {
-        // 全部启用：删除配置（使用默认行为）
-        await adapter.deleteSetting?.('enabledTools')
-        setEnabledTools(undefined)
-        return
-      }
-
-      await adapter.setSetting('enabledTools', JSON.stringify(tools))
-      setEnabledTools(tools)
-    } catch (error) {
-      console.error('[useChat] 保存 enabledTools 失败:', error)
-      throw error
-    }
-  }, [adapter])
-
-  /** 加载所有工具列表 */
-  const loadAllTools = useCallback(async () => {
-    if (!adapter.getAllTools) {
-      console.warn('[useChat] adapter.getAllTools 不存在')
-      return
-    }
-
-    try {
-      const tools = await adapter.getAllTools()
-      setAllTools(tools)
-    } catch (error) {
-      console.error('[useChat] 加载工具列表失败:', error)
-    }
-  }, [adapter])
-
-  // 初始化时加载配置
-  useEffect(() => {
-    loadAutoRunConfig()
-    loadEnabledTools()
-    loadAllTools()
-  }, [loadAutoRunConfig, loadEnabledTools, loadAllTools])
-
-  // 强制重新渲染
-  const forceUpdate = useCallback(() => setStateVersion(v => v + 1), [])
-
-  // 获取或创建会话状态
-  const getSessionState = useCallback((sessionId: string): SessionState => {
-    if (!sessionStatesRef.current.has(sessionId)) {
-      sessionStatesRef.current.set(sessionId, {
-        messages: [],
-        isLoading: false,
-        abortController: null,
-        activeAssistantMessageId: null,
-      })
-    }
-    return sessionStatesRef.current.get(sessionId)!
-  }, [])
-
-  // ==================== 全局状态 ====================
-  const [sessions, setSessions] = useState<SessionRecord[]>([])
-  const [currentSessionId, setCurrentSessionId] = useState<string | null>(null)
-  const currentSessionIdRef = useRef<string | null>(null)
-  
-  // 同步 ref
-  currentSessionIdRef.current = currentSessionId
-
-  // 配置状态
-  const [modeState, setModeState] = useState<ChatMode>(defaultMode)
-  const [modelState, setModelState] = useState(defaultModel)
-  const [webSearchState, setWebSearchState] = useState(true)
-  const [thinkingState, setThinkingState] = useState(true)
-
-  // refs for async access
-  const modeRef = useRef(modeState)
-  const modelRef = useRef(modelState)
-  const webSearchRef = useRef(webSearchState)
-  const thinkingRef = useRef(thinkingState)
-  const sessionsRef = useRef(sessions)
-
-  modeRef.current = modeState
-  modelRef.current = modelState
-  webSearchRef.current = webSearchState
-  thinkingRef.current = thinkingState
-  sessionsRef.current = sessions
-
-  // ==================== 计算属性 ====================
-  const messages = useMemo(() => {
-    // 触发 stateVersion 依赖
-    void stateVersion
-    if (!currentSessionId) return []
-    const state = sessionStatesRef.current.get(currentSessionId)
-    return state?.messages || []
-  }, [currentSessionId, stateVersion])
-
-  const isLoading = useMemo(() => {
-    void stateVersion
-    if (!currentSessionId) return false
-    const state = sessionStatesRef.current.get(currentSessionId)
-    return state?.isLoading || false
-  }, [currentSessionId, stateVersion])
-
-  // ==================== 会话管理 ====================
-  const loadSessions = useCallback(async () => {
-    try {
-      const list = await adapter.getSessions()
-      setSessions(list)
-      if (list.length > 0 && !currentSessionIdRef.current) {
-        // 选择第一个未隐藏的会话，如果都隐藏了则选第一个
-        const firstVisible = list.find(s => !s.hidden) || list[0]
-        setCurrentSessionId(firstVisible.id)
-        // 加载消息
-        const state = getSessionState(firstVisible.id)
-        if (state.messages.length === 0) {
-          const savedMessages = await adapter.getMessages(firstVisible.id)
-          state.messages = savedMessages.map(convertToMessage)
-          forceUpdate()
-        }
-        // 同步配置
-        setModeState(firstVisible.mode)
-        setModelState(firstVisible.model)
-        setWebSearchState(firstVisible.webSearchEnabled)
-        setThinkingState(firstVisible.thinkingEnabled)
-      }
-    } catch (error) {
-      console.error('加载会话失败:', error)
-    }
-  }, [adapter, getSessionState, forceUpdate])
-
-  const switchSession = useCallback(async (sessionId: string) => {
-    if (currentSessionIdRef.current === sessionId) return
-
-    setCurrentSessionId(sessionId)
-    
-    // 无状态架构：不需要同步历史到后端
-    // 发送消息时会传递历史，后端不维护状态
-
-    const state = getSessionState(sessionId)
-
-    if (state.messages.length === 0) {
-      try {
-        const savedMessages = await adapter.getMessages(sessionId)
-        state.messages = savedMessages.map(convertToMessage)
-        forceUpdate()
-      } catch (error) {
-        console.error('加载消息失败:', error)
-        state.messages = []
-        forceUpdate()
-      }
-    }
-
-    const session = sessionsRef.current.find((s) => s.id === sessionId)
-    if (session) {
-      setModeState(session.mode)
-      setModelState(session.model)
-      setWebSearchState(session.webSearchEnabled)
-      setThinkingState(session.thinkingEnabled)
-    }
-  }, [adapter, getSessionState, forceUpdate])
-
-  const createNewSession = useCallback(async () => {
-    try {
-      const session = await adapter.createSession({
-        title: '新对话',
-        model: modelRef.current,
-        mode: modeRef.current,
-        webSearchEnabled: webSearchRef.current,
-        thinkingEnabled: thinkingRef.current,
-      })
-      setSessions(prev => [session, ...prev])
-      
-      sessionStatesRef.current.set(session.id, {
-        messages: [],
-        isLoading: false,
-        abortController: null,
-        activeAssistantMessageId: null,
-      })
-      
-      setCurrentSessionId(session.id)
-      forceUpdate()
-    } catch (error) {
-      console.error('创建会话失败:', error)
-    }
-  }, [adapter, forceUpdate])
-
-  const deleteSession = useCallback(async (sessionId: string) => {
-    try {
-      const state = sessionStatesRef.current.get(sessionId)
-      if (state?.isLoading && state.abortController) {
-        state.abortController.abort()
-        adapter.cancel()
-      }
-      
-      await adapter.deleteSession(sessionId)
-      setSessions(prev => prev.filter((s) => s.id !== sessionId))
-      sessionStatesRef.current.delete(sessionId)
-
-      if (currentSessionIdRef.current === sessionId) {
-        const remainingSessions = sessionsRef.current.filter(s => s.id !== sessionId)
-        if (remainingSessions.length > 0) {
-          setCurrentSessionId(remainingSessions[0].id)
-        } else {
-          setCurrentSessionId(null)
-        }
-      }
-      forceUpdate()
-    } catch (error) {
-      console.error('删除会话失败:', error)
-    }
-  }, [adapter, forceUpdate])
-
-  const hideSession = useCallback(async (sessionId: string, hidden: boolean) => {
-    try {
-      await adapter.updateSession(sessionId, { hidden })
-      setSessions(prev => prev.map((s) =>
-        s.id === sessionId ? { ...s, hidden } : s
-      ))
-    } catch (error) {
-      console.error('更新会话隐藏状态失败:', error)
-    }
-  }, [adapter])
-
-  const clearAllSessions = useCallback(async () => {
-    try {
-      for (const [, state] of sessionStatesRef.current) {
-        if (state.isLoading && state.abortController) {
-          state.abortController.abort()
-        }
-      }
-      adapter.cancel()
-      
-      for (const session of sessionsRef.current) {
-        await adapter.deleteSession(session.id)
-      }
-      setSessions([])
-      sessionStatesRef.current.clear()
-      await createNewSession()
-    } catch (error) {
-      console.error('清空所有会话失败:', error)
-    }
-  }, [adapter, createNewSession])
-
-  const hideOtherSessions = useCallback(async () => {
-    if (!currentSessionIdRef.current) return
-    try {
-      for (const session of sessionsRef.current) {
-        if (session.id !== currentSessionIdRef.current && !session.hidden) {
-          await adapter.updateSession(session.id, { hidden: true })
-        }
-      }
-      setSessions(prev => prev.map((s) =>
-        s.id === currentSessionIdRef.current ? s : { ...s, hidden: true }
-      ))
-    } catch (error) {
-      console.error('隐藏其他会话失败:', error)
-    }
-  }, [adapter])
-
-  const exportCurrentSession = useCallback((): string | null => {
-    if (!currentSessionIdRef.current) return null
-    const session = sessionsRef.current.find((s) => s.id === currentSessionIdRef.current)
-    if (!session) return null
-
-    const state = sessionStatesRef.current.get(currentSessionIdRef.current)
-    const msgs = state?.messages || []
-
-    return JSON.stringify({
-      session: {
-        id: session.id,
-        title: session.title,
-        model: session.model,
-        mode: session.mode,
-        createdAt: session.createdAt,
-        updatedAt: session.updatedAt,
-      },
-      messages: msgs.map((msg) => ({
-        id: msg.id,
-        role: msg.role,
-        parts: msg.parts,
-        model: msg.model,
-        mode: msg.mode,
-        timestamp: msg.timestamp,
-      })),
-      exportedAt: new Date().toISOString(),
-    }, null, 2)
-  }, [])
-
-  const deleteCurrentSession = useCallback(async () => {
-    if (currentSessionIdRef.current) {
-      await deleteSession(currentSessionIdRef.current)
-    }
-  }, [deleteSession])
-
-  // ==================== 消息更新 ====================
-  const updateSessionMessage = useCallback((
-    sessionId: string, 
-    messageIndex: number, 
-    event: ChatEvent
-  ) => {
-    const state = sessionStatesRef.current.get(sessionId)
-    if (!state) return
-
-    const msg = state.messages[messageIndex]
-    if (!msg) return
-
-    const updatedMsg = { ...msg }
-    let parts = [...updatedMsg.parts]
-
-    switch (event.type) {
-      case 'thinking_start': {
-        parts.push({ type: 'thinking', text: '', status: 'running' })
-        break
-      }
-
-      case 'thinking_delta': {
-        const data = event.data as { content: string }
-        // 查找最后一个 running 状态的 thinking part
-        const lastThinkingIndex = parts.findLastIndex(
-          p => p.type === 'thinking' && p.status === 'running'
-        )
-        if (lastThinkingIndex >= 0) {
-          // 追加到现有的 running thinking part
-          const part = parts[lastThinkingIndex] as ThinkingPart
-          parts[lastThinkingIndex] = { ...part, text: part.text + data.content }
-        }
-        // ⚠️ 关键修复：如果没有 running 的 thinking part，忽略此事件（不创建新的）
-        // thinking_start 事件负责创建，thinking_delta 只负责追加
-        break
-      }
-
-      case 'thinking_end': {
-        const data = event.data as { duration: number }
-        const lastThinkingIndex = parts.findLastIndex(
-          p => p.type === 'thinking' && p.status === 'running'
-        )
-        if (lastThinkingIndex >= 0) {
-          const part = parts[lastThinkingIndex] as ThinkingPart
-          parts[lastThinkingIndex] = {
-            ...part,
-            status: 'done',
-            duration: Math.round(data.duration / 1000),
-          }
-        }
-        break
-      }
-
-      case 'search_start': {
-        const data = event.data as { query?: string }
-        const searchPart: SearchPart = { type: 'search', query: data.query, status: 'running' }
-
-        // UX：搜索结果事件可能在 text_delta 之后才到达（Provider 侧以 annotation/结果到达为准）
-        // 若直接 push，会导致"搜索卡片"显示在回答之后。
-        // 这里优先插入到第一个 text part 之前（通常在 thinking 后面），保持视觉顺序符合用户预期。
-        const firstTextIndex = parts.findIndex(p => p.type === 'text')
-        if (firstTextIndex >= 0) {
-          parts.splice(firstTextIndex, 0, searchPart)
-        } else {
-          parts.push(searchPart)
-        }
-        break
-      }
-
-      case 'search_result': {
-        const data = event.data as { results: SearchResult[] }
-        const lastSearchIndex = parts.findLastIndex(p => p.type === 'search')
-        if (lastSearchIndex >= 0) {
-          const part = parts[lastSearchIndex] as SearchPart
-          parts[lastSearchIndex] = { ...part, results: data.results, status: 'done' }
-        }
-        break
-      }
-
-      case 'search_end': {
-        const lastSearchIndex = parts.findLastIndex(
-          p => p.type === 'search' && p.status === 'running'
-        )
-        if (lastSearchIndex >= 0) {
-          const part = parts[lastSearchIndex] as SearchPart
-          parts[lastSearchIndex] = { ...part, status: 'done' }
-        }
-        break
-      }
-
-      case 'tool_approval_request': {
-        const data = event.data as { id: string; name: string; args: Record<string, unknown> }
-        
-        // 如果当前模式是自动执行，直接批准（处理发送消息后切换模式的情况）
-        if (autoRunConfigRef.current.mode === 'run-everything') {
-          adapter.respondToolApproval?.(data.id, true)
-          // 不创建 pending 状态，等待 tool_call_start 事件
-          break
-        }
-        
-        // manual 模式：创建 pending 状态等待用户确认
-        const existingIndex = parts.findIndex(
-          p => p.type === 'tool_call' && (p as ToolCallPart).id === data.id
-        )
-        if (existingIndex >= 0) {
-          // 更新现有 part 为 pending
-          const part = parts[existingIndex] as ToolCallPart
-          parts[existingIndex] = { ...part, status: 'pending' }
-        } else {
-          // 创建新的 pending part
-          parts.push({
-            type: 'tool_call',
-            id: data.id,
-            name: data.name,
-            args: data.args,
-            status: 'pending',
-          })
-        }
-        break
-      }
-
-      case 'tool_call_start': {
-        const data = event.data as { id: string; name: string; args: Record<string, unknown> }
-        // 检查是否已存在 pending 状态的 part
-        const existingIndex = parts.findIndex(
-          p => p.type === 'tool_call' && (p as ToolCallPart).id === data.id
-        )
-        if (existingIndex >= 0) {
-          // 更新现有 part 为 running
-          const part = parts[existingIndex] as ToolCallPart
-          parts[existingIndex] = { ...part, status: 'running' }
-        } else {
-          // 创建新的 running part
-          parts.push({
-            type: 'tool_call',
-            id: data.id,
-            name: data.name,
-            args: data.args,
-            status: 'running',
-          })
-        }
-        break
-      }
-
-      case 'tool_call_result': {
-        const data = event.data as { 
-          id: string
-          name: string
-          result: string
-          success: boolean
-          resultType?: string  // 结果类型（用于生成具体类型的 Part）
-        }
-        
-        let parsedResult: unknown = data.result
-        try {
-          parsedResult = JSON.parse(data.result)
-        } catch {
-          // 保持原始字符串
-        }
-
-        // 判断状态：检查是否是跳过或取消操作
-        const isSkipped = typeof parsedResult === 'object' && parsedResult !== null && 
-          'skipped' in parsedResult && (parsedResult as { skipped?: boolean }).skipped === true
-        const isCancelled = typeof parsedResult === 'object' && parsedResult !== null && 
-          'cancelled' in parsedResult && (parsedResult as { cancelled?: boolean }).cancelled === true
-        const status: 'done' | 'error' | 'cancelled' | 'skipped' = 
-          isSkipped ? 'skipped' : (isCancelled ? 'cancelled' : (data.success ? 'done' : 'error'))
-
-        // 查找对应的 tool_call 并更新状态
-        const toolCallIndex = parts.findIndex(
-          p => p.type === 'tool_call' && (p as ToolCallPart).id === data.id
-        )
-        
-        if (toolCallIndex >= 0) {
-          // 更新 tool_call 状态（不再包含 result 字段）
-          const toolCall = parts[toolCallIndex] as ToolCallPart
-          parts[toolCallIndex] = {
-            ...toolCall,
-            status,
-          }
-        } else {
-          // 如果没有对应的 tool_call，创建一个新的
-          parts.push({
-            type: 'tool_call',
-            id: data.id,
-            name: data.name,
-            args: {},
-            status,
-          })
-        }
-
-        // 如果工具定义了 resultType 且执行成功，生成对应类型的 Part
-        if (data.resultType && data.success && typeof parsedResult === 'object' && parsedResult !== null) {
-          // 找到 tool_call 的位置，在其后插入结果 Part
-          const updatedToolCallIndex = parts.findIndex(
-            p => p.type === 'tool_call' && (p as ToolCallPart).id === data.id
-          )
-          
-          // 创建具体类型的 Part（如 { type: 'weather', city: '北京', ... }）
-          const resultPart = {
-            type: data.resultType,
-            ...parsedResult as Record<string, unknown>,
-          }
-          
-          // 检查是否已存在同类型的结果 Part（避免重复）
-          const existingResultIndex = parts.findIndex(
-            (p, i) => i > updatedToolCallIndex && p.type === data.resultType
-          )
-          
-          if (existingResultIndex >= 0) {
-            // 更新现有的结果 Part
-            parts[existingResultIndex] = resultPart
-          } else {
-            // 插到对应 tool_call 之后
-            const insertAt = updatedToolCallIndex >= 0 ? updatedToolCallIndex + 1 : parts.length
-            parts.splice(insertAt, 0, resultPart)
-          }
-        }
-
-        if (onToolComplete) {
-          // 从事件数据中获取副作用声明
-          const sideEffects = (data as { sideEffects?: SideEffect[] }).sideEffects
-          onToolComplete({
-            name: data.name,
-            result: parsedResult,
-            sideEffects,
-          })
-        }
-        break
-      }
-
-      case 'tool_call_output': {
-        const data = event.data as { id: string; name: string; stream: 'stdout' | 'stderr'; chunk: string }
-        const toolCallIndex = parts.findIndex(
-          p => p.type === 'tool_call' && (p as ToolCallPart).id === data.id
-        )
-        if (toolCallIndex >= 0) {
-          const toolCall = parts[toolCallIndex] as ToolCallPart
-          const prevStdout = toolCall.output?.stdout ?? ''
-          const prevStderr = toolCall.output?.stderr ?? ''
-          const MAX = 120_000
-
-          const nextOutput =
-            data.stream === 'stdout'
-              ? { stdout: (prevStdout + (data.chunk || '')).slice(-MAX), stderr: prevStderr }
-              : { stdout: prevStdout, stderr: (prevStderr + (data.chunk || '')).slice(-MAX) }
-
-          parts[toolCallIndex] = {
-            ...toolCall,
-            output: nextOutput,
-          }
-        }
-        break
-      }
-
-      case 'text_delta': {
-        const data = event.data as { content: string }
-        
-        parts = parts.map(p => 
-          p.type === 'search' && p.status === 'running' 
-            ? { ...p, status: 'done' as const } 
-            : p
-        )
-        
-        // 查找最后一个 text part
-        const lastTextIndex = parts.findLastIndex(p => p.type === 'text')
-        
-        // 检查最后一个 part 是否是已完成的工具调用/思考
-        // 如果是，说明这是新一轮的文本输出，应该创建新的 text part
-        //
-        // 注意：search 结果可能会在文本输出过程中到达（甚至夹在 text_delta 中间），
-        // 如果把 search(done) 也作为“新一轮文本”的边界，会导致文本被拆成多个 TextPart。
-        const lastPart = parts[parts.length - 1]
-        const shouldCreateNew = lastPart && (
-          (lastPart.type === 'tool_call' && ['done', 'error', 'skipped', 'cancelled'].includes((lastPart as ToolCallPart).status)) ||
-          (lastPart.type === 'thinking' && (lastPart as ThinkingPart).status === 'done')
-        )
-        
-        if (shouldCreateNew || lastTextIndex < 0) {
-          // 创建新的 text part
-          parts.push({ type: 'text', text: data.content })
-        } else {
-          // 追加到现有的 text part
-          const part = parts[lastTextIndex] as TextPart
-          parts[lastTextIndex] = { ...part, text: part.text + data.content }
-        }
-        break
-      }
-
-      case 'error': {
-        const data = event.data as { category?: string; message: string; retryable?: boolean }
-        
-        parts = parts.map(p => {
-          if (p.type === 'thinking' && p.status === 'running') return { ...p, status: 'done' as const }
-          if (p.type === 'search' && p.status === 'running') return { ...p, status: 'done' as const }
-          if (p.type === 'tool_call' && p.status === 'running') return { ...p, status: 'error' as const }
-          return p
-        })
-        
-        parts.push({
-          type: 'error',
-          message: data.message,
-          category: data.category,
-          retryable: data.retryable,
-        })
-        
-        updatedMsg.loading = false
-        updatedMsg.error = data
-        break
-      }
-
-      case 'abort': {
-        parts = parts.map(p => {
-          if (p.type === 'thinking' && p.status === 'running') return { ...p, status: 'done' as const }
-          if (p.type === 'search' && p.status === 'running') return { ...p, status: 'done' as const }
-          if (p.type === 'tool_call' && p.status === 'running') return { ...p, status: 'cancelled' as const }
-          return p
-        })
-        updatedMsg.loading = false
-        updatedMsg.aborted = true
-        break
-      }
-
-      case 'done':
-        updatedMsg.loading = false
-        break
-    }
-
-    updatedMsg.parts = parts
-    state.messages = [...state.messages]
-    state.messages[messageIndex] = updatedMsg
-    forceUpdate()
-  }, [onToolComplete, forceUpdate])
-
-  /**
-   * 保存消息到数据库
-   */
-  const saveMessageToDb = useCallback(async (
-    sessionId: string,
-    messageIndex: number,
-    messageId: string
-  ) => {
-    const state = sessionStatesRef.current.get(sessionId)
-    if (!state) return
-    
-    const msg = state.messages[messageIndex]
-    if (!msg) return
-    
-    try {
-      await adapter.updateMessage?.({
-        id: messageId,
-        content: extractTextContent(msg.parts),
-        steps: JSON.stringify(msg.parts),
-      })
-    } catch (error) {
-      console.error('[useChat/react] 保存消息失败:', error)
-    }
-  }, [adapter])
-
-  /**
-   * 统一的事件处理函数：更新UI并保存到数据库
-   * 保证UI更新和DB保存的原子性
-   */
-  const handleEvent = useCallback(async (
-    sessionId: string,
-    messageIndex: number,
-    event: ChatEvent,
-    messageId: string,
-    options: { saveToDb: boolean } = { saveToDb: false }
-  ) => {
-    // 1. 先更新UI
-    updateSessionMessage(sessionId, messageIndex, event)
-    
-    // 2. 如果需要保存，立即保存（保证UI和DB一致）
-    if (options.saveToDb && shouldSaveEvent(event)) {
-      await saveMessageToDb(sessionId, messageIndex, messageId)
-    }
-  }, [updateSessionMessage, saveMessageToDb])
-
-  // ==================== 发送消息 ====================
-  const sendMessage = useCallback(async (text: string, images?: string[]) => {
-    // 允许只发送图片（text 为空但有图片）
-    const hasContent = text.trim() || (images && images.length > 0)
-    if (!hasContent) return
-    
-    let sessionId = currentSessionIdRef.current
-    
-    if (sessionId) {
-      const currentState = sessionStatesRef.current.get(sessionId)
-      if (currentState?.isLoading) {
-        console.warn('[useChat] 当前会话正在进行中，请等待完成')
-        return
-      }
-    }
-
-    if (!sessionId) {
-      try {
-        const session = await adapter.createSession({
-          title: '新对话',
-          model: modelRef.current,
-          mode: modeRef.current,
-        })
-        setSessions(prev => [session, ...prev])
-        sessionStatesRef.current.set(session.id, {
-          messages: [],
-          isLoading: false,
-          abortController: null,
-          activeAssistantMessageId: null,
-        })
-        setCurrentSessionId(session.id)
-        sessionId = session.id
-      } catch (error) {
-        console.error('创建会话失败:', error)
-        return
-      }
-    }
-
-    const state = getSessionState(sessionId)
-
-    const userMsg: ChatMessage = {
-      id: generateId(),
-      role: 'user',
-      parts: [{ type: 'text', text }],
-      images,
-      timestamp: Date.now(),
-    }
-    state.messages = [...state.messages, userMsg]
-    forceUpdate()
-
-    try {
-      // 关键：传入 userMsg.id，确保"重新发送/分叉"能用同一个 id 做锚点更新/删除
-      await adapter.saveMessage({
-        id: userMsg.id,
-        sessionId,
-        role: 'user',
-        content: text,
-        images: images || [],
-      })
-
-      if (state.messages.length === 1) {
-        const title = text.trim() 
-          ? text.slice(0, 20) + (text.length > 20 ? '...' : '')
-          : (images && images.length > 0 ? '图片消息' : '新对话')
-        await adapter.updateSession(sessionId, { title })
-        setSessions(prev => prev.map((s) =>
-          s.id === sessionId ? { ...s, title } : s
-        ))
-      }
-    } catch (error) {
-      console.error('保存消息失败:', error)
-    }
-
-    const assistantMsgIndex = state.messages.length
-    const assistantMsgId = generateId() // 保存消息 ID 用于后续更新
-    const assistantMsg: ChatMessage = {
-      id: assistantMsgId,
-      role: 'assistant',
-      parts: [],
-      model: modelRef.current,
-      mode: modeRef.current,
-      webSearchEnabled: webSearchRef.current,
-      thinkingEnabled: thinkingRef.current,
-      loading: true,
-      timestamp: Date.now(),
-    }
-    state.messages = [...state.messages, assistantMsg]
-    
-    // 创建本次请求专用的 abortController（避免多请求竞态条件）
-    const requestAbortController = new AbortController()
-    state.isLoading = true
-    state.abortController = requestAbortController
-    state.activeAssistantMessageId = assistantMsgId
-    forceUpdate()
-    
-    const sendModel = modelRef.current
-    const sendMode = modeRef.current
-    const sendWebSearch = webSearchRef.current
-    const sendThinking = thinkingRef.current
-    const sendEnabledTools = enabledToolsRef.current
-
-    // 【关键】立即保存助手消息到数据库（初始状态）
-    try {
-      await adapter.saveMessage({
-        id: assistantMsgId,
-        sessionId,
-        role: 'assistant',
-        content: '',
-        model: sendModel,
-        mode: sendMode,
-        webSearchEnabled: sendWebSearch,
-        thinkingEnabled: sendThinking,
-        steps: '[]',
-      })
-    } catch (error) {
-      console.error('创建助手消息失败:', error)
-    }
-
-    // 增量保存函数
-    const saveMessageProgress = async () => {
-      const msg = state.messages[assistantMsgIndex]
-      if (!msg) return
-      try {
-        await adapter.updateMessage?.({
-          id: assistantMsgId,
-          content: extractTextContent(msg.parts),
-          steps: JSON.stringify(msg.parts),
-        })
-      } catch (error) {
-        console.error('更新消息失败:', error)
-      }
-    }
-
-    try {
-      // 构建历史消息（不包括刚添加的用户消息和助手占位消息）
-      const history = state.messages.slice(0, -2).map(msg => ({
-        role: msg.role as 'user' | 'assistant' | 'system' | 'tool',
-        // history 中去掉 @ 引用行（避免后续轮次出现“空引用”干扰）
-        content: stripAtContextLines(extractTextContent(msg.parts)),
-      }))
-      
-      const cleanAutoRunConfig: AutoRunConfig = { ...autoRunConfigRef.current }
-      const promptMessage = await buildPromptWithAtContext(adapter, text)
-
-      for await (const event of adapter.sendMessage(
-        promptMessage,
-        {
-          mode: sendMode,
-          model: sendModel,
-          enableWebSearch: sendWebSearch,
-          thinkingMode: sendThinking ? 'enabled' : 'disabled',
-          enabledTools: sendEnabledTools,
-          autoRunConfig: cleanAutoRunConfig,
-          history, // 传递历史消息
-        },
-        images,
-        sessionId // 传递 sessionId 用于事件过滤
-      )) {
-        // 使用本次请求专用的 abortController 检查（避免被新请求覆盖）
-        if (requestAbortController.signal.aborted) break
-
-        // 统一处理事件：更新UI并保存到数据库
-        await handleEvent(sessionId, assistantMsgIndex, event, assistantMsgId, {
-          saveToDb: true, // 重要事件需要保存
-        })
-
-        if (event.type === 'done' || event.type === 'error') {
-          break
-        }
-      }
-    } catch (error) {
-      console.error('发送消息失败:', error)
-      await handleEvent(
-        sessionId,
-        assistantMsgIndex,
-        {
-          type: 'error',
-          data: { message: error instanceof Error ? error.message : String(error) },
-        },
-        assistantMsgId,
-        { saveToDb: true }
-      )
-    } finally {
-      state.isLoading = false
-      
-      const finalMsg = state.messages[assistantMsgIndex]
-      if (finalMsg) {
-        state.messages = [...state.messages]
-        state.messages[assistantMsgIndex] = { ...finalMsg, loading: false }
-      }
-
-      // 【关键】最终保存一次，确保所有内容都被持久化
-      await saveMessageToDb(sessionId, assistantMsgIndex, assistantMsgId)
-
-      state.abortController = null
-      state.activeAssistantMessageId = null
-      forceUpdate()
-    }
-  }, [adapter, getSessionState, handleEvent, saveMessageToDb, forceUpdate])
-
-  // ==================== 其他方法 ====================
-  const patchPartsAsAborted = useCallback((parts: ContentPart[]): ContentPart[] => {
-    return parts.map((p) => {
-      if (p.type === 'thinking' && (p as ThinkingPart).status === 'running') {
-        return { ...(p as ThinkingPart), status: 'done' as const }
-      }
-      if (p.type === 'search' && (p as SearchPart).status === 'running') {
-        return { ...(p as SearchPart), status: 'done' as const }
-      }
-      if (p.type === 'tool_call') {
-        const tool = p as ToolCallPart
-        if (tool.status === 'running' || tool.status === 'pending') {
-          return { ...tool, status: 'cancelled' as const }
-        }
-      }
-      return p
-    })
-  }, [])
-
-  const cancelActiveAssistantMessage = useCallback((sessionId: string) => {
-    const state = sessionStatesRef.current.get(sessionId)
-    if (!state) return
-    if (!state.activeAssistantMessageId) return
-    const idx = state.messages.findIndex((m) => m.id === state.activeAssistantMessageId)
-    if (idx < 0) return
-    const msg = state.messages[idx]
-    state.messages = [...state.messages]
-    state.messages[idx] = {
-      ...msg,
-      loading: false,
-      aborted: true,
-      parts: patchPartsAsAborted(msg.parts),
-    }
-  }, [patchPartsAsAborted])
-
-  const cancelRequest = useCallback(() => {
-    if (!currentSessionIdRef.current) return
-    const sessionId = currentSessionIdRef.current
-    const state = sessionStatesRef.current.get(sessionId)
-
-    // 先做 UI 即时落地：精准命中本轮 assistant 消息
-    cancelActiveAssistantMessage(sessionId)
-    if (state) {
-      state.abortController?.abort()
-      state.isLoading = false
-      state.activeAssistantMessageId = null
-      forceUpdate()
-    }
-    adapter.cancel()
-  }, [adapter, forceUpdate, cancelActiveAssistantMessage])
-
-  const copyMessage = useCallback(async (messageId: string) => {
-    if (!currentSessionIdRef.current) return
-    const state = sessionStatesRef.current.get(currentSessionIdRef.current)
-    if (!state) return
-
-    const msg = state.messages.find((m) => m.id === messageId)
-    if (!msg) return
-
-    try {
-      const textContent = extractTextContent(msg.parts)
-      await navigator.clipboard.writeText(textContent)
-      
-      state.messages = state.messages.map((m) =>
-        m.id === messageId ? { ...m, copied: true } : m
-      )
-      forceUpdate()
-      
-      setTimeout(() => {
-        const s = sessionStatesRef.current.get(currentSessionIdRef.current!)
-        if (s) {
-          s.messages = s.messages.map((m) =>
-            m.id === messageId ? { ...m, copied: false } : m
-          )
-          forceUpdate()
-        }
-      }, 2000)
-    } catch (err) {
-      console.error('复制失败:', err)
-    }
-  }, [forceUpdate])
-
-  /** 从指定索引重新发送消息（编辑后重发，分叉：删除其后的所有消息并重新生成） */
-  const resendFromIndex = useCallback(async (index: number, text: string) => {
-    if (!currentSessionIdRef.current) return
-    const sessionId = currentSessionIdRef.current
-    const state = sessionStatesRef.current.get(sessionId)
-    if (!state) return
-
-    const targetMsg = state.messages[index]
-    if (!targetMsg || targetMsg.role !== 'user') {
-      // 容错：如果传入的不是 user 消息索引，退化为普通发送
-      sendMessage(text)
-      return
-    }
-
-    // 如果当前会话正在生成，先取消，避免并发造成状态错乱
-    if (state.isLoading) {
-      state.abortController?.abort()
-      adapter.cancel()
-      state.isLoading = false
-      state.abortController = null
-    }
-
-    // 1) UI：保留该条 user 消息，并更新文本；删除其后的所有消息（分叉）
-    const updatedUserMsg: ChatMessage = {
-      ...targetMsg,
-      parts: [{ type: 'text', text }],
-    }
-    state.messages = [...state.messages.slice(0, index), updatedUserMsg]
-    forceUpdate()
-
-    // 2) DB：删除该 user 消息之后的所有消息（用于分叉）
-    // 新架构：只使用 messageId 作为锚点（更稳定），后端用 messageId 查 timestamp 再删除
-    try {
-      await adapter.deleteMessagesAfterMessageId(sessionId, updatedUserMsg.id)
-    } catch (error) {
-      console.error('[useChat/react] deleteMessagesAfterMessageId 失败:', error)
-    }
-
-    // 3) DB：更新该条 user 消息内容（需要保证 saveMessage 使用同一个 id）
-    try {
-      await adapter.updateMessage?.({
-        id: updatedUserMsg.id,
-        content: text,
-      })
-    } catch (error) {
-      console.warn('[useChat/react] updateMessage(user) 失败，已忽略:', error)
-    }
-
-    // 4) 从该 user 消息继续生成 assistant（不再新建 user 消息）
-    const assistantMsgIndex = state.messages.length
-    const assistantMsgId = generateId()
-    const assistantMsg: ChatMessage = {
-      id: assistantMsgId,
-      role: 'assistant',
-      parts: [],
-      model: modelRef.current,
-      mode: modeRef.current,
-      webSearchEnabled: webSearchRef.current,
-      thinkingEnabled: thinkingRef.current,
-      loading: true,
-      timestamp: Date.now(),
-    }
-    state.messages = [...state.messages, assistantMsg]
-
-    const requestAbortController = new AbortController()
-    state.isLoading = true
-    state.abortController = requestAbortController
-    state.activeAssistantMessageId = assistantMsgId
-    forceUpdate()
-
-    // 保存当前配置快照
-    const sendModel = modelRef.current
-    const sendMode = modeRef.current
-    const sendWebSearch = webSearchRef.current
-    const sendThinking = thinkingRef.current
-    const sendEnabledTools = enabledToolsRef.current
-
-    // 立即保存助手消息到数据库（初始状态）
-    try {
-      await adapter.saveMessage({
-        id: assistantMsgId,
-        sessionId,
-        role: 'assistant',
-        content: '',
-        model: sendModel,
-        mode: sendMode,
-        webSearchEnabled: sendWebSearch,
-        thinkingEnabled: sendThinking,
-        steps: '[]',
-      })
-    } catch (error) {
-      console.error('[useChat/react] 创建 assistant 消息失败:', error)
-    }
-
-    // 使用统一的事件处理函数
-
-    try {
-      // history：取该 user 消息之前的消息（不包含当前 user）
-      const history = state.messages.slice(0, index).map(msg => ({
-        role: msg.role as 'user' | 'assistant' | 'system' | 'tool',
-        content: stripAtContextLines(extractTextContent(msg.parts)),
-      }))
-
-      const cleanAutoRunConfig: AutoRunConfig = { ...autoRunConfigRef.current }
-      const images = updatedUserMsg.images
-      const promptMessage = await buildPromptWithAtContext(adapter, text)
-
-      for await (const event of adapter.sendMessage(
-        promptMessage,
-        {
-          mode: sendMode,
-          model: sendModel,
-          enableWebSearch: sendWebSearch,
-          thinkingMode: sendThinking ? 'enabled' : 'disabled',
-          enabledTools: sendEnabledTools,
-          autoRunConfig: cleanAutoRunConfig,
-          history,
-        },
-        images,
-        sessionId
-      )) {
-        if (requestAbortController.signal.aborted) break
-        
-        // 统一处理事件：更新UI并保存到数据库
-        await handleEvent(sessionId, assistantMsgIndex, event, assistantMsgId, {
-          saveToDb: true, // 重要事件需要保存
-        })
-        
-        if (event.type === 'done' || event.type === 'error') break
-      }
-    } catch (error) {
-      console.error('[useChat/react] 分叉重发失败:', error)
-      await handleEvent(
-        sessionId,
-        assistantMsgIndex,
-        {
-          type: 'error',
-          data: { message: error instanceof Error ? error.message : String(error) },
-        },
-        assistantMsgId,
-        { saveToDb: true }
-      )
-    } finally {
-      // 收尾：取消 loading 状态
-      const s = sessionStatesRef.current.get(sessionId)
-      if (s) {
-        s.isLoading = false
-        s.abortController = null
-        s.activeAssistantMessageId = null
-        const finalMsg = s.messages[assistantMsgIndex]
-        if (finalMsg) {
-          s.messages[assistantMsgIndex] = { ...finalMsg, loading: false }
-        }
-        forceUpdate()
-      }
-      // 【关键】最终保存一次，确保所有内容都被持久化
-      await saveMessageToDb(sessionId, assistantMsgIndex, assistantMsgId)
-    }
-  }, [adapter, forceUpdate, sendMessage, handleEvent, saveMessageToDb])
-
-  const regenerateMessage = useCallback((messageIndex: number) => {
-    if (!currentSessionIdRef.current) return
-    const state = sessionStatesRef.current.get(currentSessionIdRef.current)
-    if (!state) return
-
-    if (messageIndex > 0 && state.messages[messageIndex - 1]?.role === 'user') {
-      const userIndex = messageIndex - 1
-      const userMsg = state.messages[userIndex]
-      const userText = extractTextContent(userMsg.parts)
-      void resendFromIndex(userIndex, userText)
-    }
-  }, [resendFromIndex])
-
-  const setWorkingDirectory = useCallback((dir: string) => {
-    if (adapter.setCwd) {
-      adapter.setCwd(dir)
-    }
-  }, [adapter])
-
-  // ==================== 返回 ====================
-  return {
-    sessions,
-    currentSessionId,
-    messages,
-    isLoading,
-    mode: modeState,
-    model: modelState,
-    webSearch: webSearchState,
-    thinking: thinkingState,
-
-    loadSessions,
-    switchSession,
-    createNewSession,
-    deleteSession,
-    deleteCurrentSession,
-    hideSession,
-    clearAllSessions,
-    hideOtherSessions,
-    exportCurrentSession,
-
-    sendMessage,
-    cancelRequest,
-    copyMessage,
-    regenerateMessage,
-    resendFromIndex,
-
-    setMode: (value: ChatMode) => {
-      setModeState(value)
-      if (currentSessionIdRef.current) {
-        adapter.updateSession(currentSessionIdRef.current, { mode: value }).catch((e: Error) =>
-          console.error('更新会话 mode 失败:', e)
-        )
-      }
-    },
-    setModel: (value: string) => {
-      setModelState(value)
-      if (currentSessionIdRef.current) {
-        adapter.updateSession(currentSessionIdRef.current, { model: value }).catch((e: Error) =>
-          console.error('更新会话 model 失败:', e)
-        )
-      }
-    },
-    setWebSearch: (value: boolean) => {
-      setWebSearchState(value)
-      if (currentSessionIdRef.current) {
-        adapter.updateSession(currentSessionIdRef.current, { webSearchEnabled: value }).catch((e: Error) =>
-          console.error('更新会话 webSearchEnabled 失败:', e)
-        )
-      }
-    },
-    setThinking: (value: boolean) => {
-      setThinkingState(value)
-      if (currentSessionIdRef.current) {
-        adapter.updateSession(currentSessionIdRef.current, { thinkingEnabled: value }).catch((e: Error) =>
-          console.error('更新会话 thinkingEnabled 失败:', e)
-        )
-      }
-    },
-
-    setWorkingDirectory,
-
-    // 工具批准对话框
-
-    // 自动运行配置
-    autoRunConfig,
-    loadAutoRunConfig,
-    saveAutoRunConfig,
-
-    // 工具管理
-    enabledTools,
-    allTools,
-    saveEnabledTools,
-  }
-}