@veltra/ai 2.1.5 → 2.1.7

package/src/chat/use-chat.ts

@@ -1,16 +1,21 @@
 import { useModel } from '@veltra/compositions'
 import { reactive, ref, watch, type Ref } from 'vue'
 
-import { createBuiltinTools } from '../tools'
+import type { AskQuestionItem } from '../tools'
 import type { AiChatEmits, AiChatProps } from '../types'
+import type { ChatPendingApproval } from './fold'
+import { isServerTransport, type ChatSessionTransport } from './session'
 import type {
   ChatAttachment,
+  ChatJob,
   ChatMessage,
   ChatQueuedMessage,
   ChatTokenUsage,
   ChatTool,
   ChatToolCall
 } from './types'
+import { addTokenUsage, resolveTools, serializeResult } from './use-chat-helpers'
+import { createSessionRuntime } from './use-chat-session'
 
 export interface UseChatOptions {
   props: AiChatProps
@@ -23,47 +28,10 @@ const uid = () => {
     : `${Date.now()}-${Math.random().toString(36).slice(2)}`
 }
 
-function serializeResult(result: unknown): string {
-  if (typeof result === 'string') return result
-  try {
-    return JSON.stringify(result) ?? 'null'
-  } catch {
-    return String(result)
-  }
-}
-
-/** 内置工具 + 用户工具，同名时内置优先 */
-function resolveTools(userTools?: ChatTool[]): ChatTool[] {
-  const builtins = createBuiltinTools()
-  const names = new Set(builtins.map((t) => t.name))
-  return [...builtins, ...(userTools ?? []).filter((t) => !names.has(t.name))]
-}
-
-function mergeOptionalCount(a?: number, b?: number): number | undefined {
-  if (a == null && b == null) return undefined
-  return (a ?? 0) + (b ?? 0)
-}
-
-/** 累加两次 usage；缓存字段只在至少一侧有值时保留 */
-function addTokenUsage(a: ChatTokenUsage, b: ChatTokenUsage): ChatTokenUsage {
-  const usage: ChatTokenUsage = {
-    promptTokens: a.promptTokens + b.promptTokens,
-    completionTokens: a.completionTokens + b.completionTokens,
-    totalTokens: a.totalTokens + b.totalTokens
-  }
-  const cacheHitTokens = mergeOptionalCount(a.cacheHitTokens, b.cacheHitTokens)
-  if (cacheHitTokens != null) usage.cacheHitTokens = cacheHitTokens
-  const cacheMissTokens = mergeOptionalCount(a.cacheMissTokens, b.cacheMissTokens)
-  if (cacheMissTokens != null) usage.cacheMissTokens = cacheMissTokens
-  const cacheCreationTokens = mergeOptionalCount(a.cacheCreationTokens, b.cacheCreationTokens)
-  if (cacheCreationTokens != null) usage.cacheCreationTokens = cacheCreationTokens
-  return usage
-}
-
 /**
  * AI 对话核心状态机：消息管理、流式追加、工具调用循环编排。
  * 与 UI 解耦，ai-chat.vue 只负责渲染与交互转发。
- * 始终注入内置工具（如 askQuestion），无需经 props 传入。
+ * 函数型 transport 注入内置工具；session transport 不注入、不执行客户端工具。
  */
 export function useChat(options: UseChatOptions) {
   const { props, emit } = options
@@ -93,6 +61,28 @@ export function useChat(options: UseChatOptions) {
   /** 最近一轮用户对话（含工具多轮请求）的 token；该轮无 usage 时为 null */
   const lastTurnUsage = ref<ChatTokenUsage | null>(null)
 
+  /** session 下由 jobs/snapshot 整体替换；函数 transport 保持空数组 */
+  const jobs = ref<ChatJob[]>([])
+  const projections = ref<Record<string, unknown>>({})
+  const title = ref<string | null>(null)
+  const pendingApprovals = ref<ChatPendingApproval[]>([])
+  const pendingQuestion = ref<{ questions: AskQuestionItem[]; rpcId: string } | null>(null)
+
+  const sessionRuntime = createSessionRuntime({
+    messages,
+    queue,
+    jobs,
+    tokenUsage,
+    projections,
+    title,
+    running,
+    pendingApprovals,
+    pendingQuestion
+  })
+
+  const session = (): ChatSessionTransport | null =>
+    isServerTransport(props.transport) ? props.transport : null
+
   let abortController: AbortController | null = null
   /** needsConfirm 工具的挂起确认器，key 为 toolCallId */
   const confirmResolvers = new Map<string, (approved: boolean) => void>()
@@ -129,6 +119,25 @@ export function useChat(options: UseChatOptions) {
     { immediate: true }
   )
 
+  // session：open + fetchHistory 播种；更换或卸载时 disposer
+  watch(
+    () => props.transport,
+    (transport, _prev, onCleanup) => {
+      if (!isServerTransport(transport)) return
+      sessionRuntime.attach(transport, onCleanup)
+    },
+    { immediate: true }
+  )
+
+  // session 且已 open 时，模型切换走 selectModel
+  watch(model, (modelId) => {
+    const current = session()
+    if (!sessionRuntime.opened || !current || !modelId) return
+    const option = props.models?.find((m) => m.id === modelId)
+    if (!option) return
+    void current.selectModel(option.providerId, modelId)
+  })
+
   /**
    * 向父组件同步消息快照（流式 delta 过于频繁，不同步，只在关键节点同步）。
    * 必须经 setter 赋值触发 emit，而非直接 emit 切片：setter 让本地数组与下发的
@@ -146,6 +155,11 @@ export function useChat(options: UseChatOptions) {
    */
   const send = (content: string, attachments?: ChatAttachment[]) => {
     if (!content.trim() && !attachments?.length) return
+    const current = session()
+    if (current) {
+      void current.send(content, attachments)
+      return
+    }
     enqueue(content, attachments)
   }
 
@@ -159,6 +173,7 @@ export function useChat(options: UseChatOptions) {
     beforeId?: string
   ): ChatQueuedMessage => {
     const item: ChatQueuedMessage = { id: uid(), content, attachments }
+    if (session()) return item
     const anchorIndex = beforeId ? queue.value.findIndex((q) => q.id === beforeId) : -1
     if (anchorIndex === -1) {
       queue.value.push(item)
@@ -177,6 +192,7 @@ export function useChat(options: UseChatOptions) {
 
   /** 立即执行队列中的某条：中断当前会话，该条插队为下一条（其余保持原顺序） */
   const startQueued = (id: string) => {
+    if (session()) return
     const item = removeQueued(id)
     if (!item) return
 
@@ -193,6 +209,7 @@ export function useChat(options: UseChatOptions) {
 
   /** 从队列移除某条（返回被移除项；编辑场景由 UI 取回内容） */
   const removeQueued = (id: string): ChatQueuedMessage | undefined => {
+    if (session()) return undefined
     const index = queue.value.findIndex((item) => item.id === id)
     if (index === -1) return undefined
     const [item] = queue.value.splice(index, 1)
@@ -287,9 +304,10 @@ export function useChat(options: UseChatOptions) {
 
   /** 单轮生成；存在工具调用时执行后递归进入下一轮（递归避免 await-in-loop） */
   const runRound = async (controller: AbortController, depth: number): Promise<void> => {
-    if (!props.transport) return
+    const transport = props.transport
+    if (typeof transport !== 'function') return
     const { signal } = controller
-    const tools = resolveTools(props.tools)
+    const tools = resolveTools(props.tools as ChatTool[] | undefined)
 
     const assistant = reactive<ChatMessage>({
       id: uid(),
@@ -303,7 +321,7 @@ export function useChat(options: UseChatOptions) {
     let requestError: Error | null = null
 
     try {
-      await props.transport(
+      await transport(
         {
           // 排除刚追加的 assistant 占位消息
           messages: messages.value.slice(0, -1),
@@ -395,17 +413,34 @@ export function useChat(options: UseChatOptions) {
 
   /** 中断当前生成，挂起的工具确认按拒绝处理 */
   const abort = () => {
+    const current = session()
+    if (current) {
+      void current.cancel()
+      return
+    }
     abortController?.abort()
   }
 
-  /** 响应 needsConfirm 工具的用户确认 */
+  /** 响应 needsConfirm 工具的用户确认；session 下按 callId 找到 rpcId 后 respond */
   const respondToolCall = (toolCallId: string, approved: boolean) => {
+    const current = session()
+    if (current) {
+      const pending = pendingApprovals.value.find((a) => a.callId === toolCallId)
+      if (pending) void current.respond(pending.rpcId, approved)
+      return
+    }
     confirmResolvers.get(toolCallId)?.(approved)
   }
 
+  /** session 审批横幅 / 提问提交：直接 respond(rpcId) */
+  const respondSession = (rpcId: string, ok: boolean, value?: unknown) => {
+    const current = session()
+    if (current) void current.respond(rpcId, ok, value)
+  }
+
   /** 重新生成：移除最后一条用户消息之后的所有消息，重新跑对话循环 */
   const regenerate = () => {
-    if (running.value) return
+    if (session() || running.value) return
 
     const list = messages.value
     let lastUserIndex = -1
@@ -424,6 +459,11 @@ export function useChat(options: UseChatOptions) {
 
   /** 清空消息、待发送队列与 token 统计，生成中则先中断 */
   const clear = () => {
+    if (session()) {
+      sessionRuntime.resetLocal()
+      lastTurnUsage.value = null
+      return
+    }
     abort()
     queue.value = []
     messages.value = []
@@ -438,13 +478,19 @@ export function useChat(options: UseChatOptions) {
     reasoningLevel,
     running,
     queue,
+    jobs,
     tokenUsage,
     lastTurnUsage,
+    projections,
+    title,
+    pendingApprovals,
+    pendingQuestion,
     send,
     abort,
     regenerate,
     clear,
     respondToolCall,
+    respondSession,
     enqueue,
     startQueued,
     removeQueued

package/src/components/ai-chat/__test__/ai-chat.test.ts

@@ -1,9 +1,29 @@
 import { describe, expect, it, vi } from 'vitest'
-import { createApp, defineComponent, h, nextTick, ref, type PropType } from 'vue'
-
-import type { ChatMessage, ChatTool, ChatToolCall, ChatTransport } from '../../../chat/types'
+import {
+  createApp,
+  defineComponent,
+  h,
+  nextTick,
+  reactive,
+  ref,
+  type Component,
+  type PropType
+} from 'vue'
+
+import {
+  createServerTransport,
+  type ChatSessionAdapter,
+  type ChatSessionEvent
+} from '../../../chat/session'
+import type {
+  ChatMessage,
+  ChatTool,
+  ChatToolCall,
+  ChatToolMeta,
+  ChatTransport
+} from '../../../chat/types'
 import type { ChatModelOption } from '../../../providers'
-import type { AiChatExposed } from '../../../types'
+import type { AiChatExposed, AiChatProps } from '../../../types'
 import UAiChat from '../ai-chat.vue'
 
 // happy-dom 环境下用桩组件替换 MarkdownRender，仅验证组件自身逻辑
@@ -17,13 +37,16 @@ vi.mock('markstream-vue', () => ({
 }))
 
 function mountAiChat(options: {
-  transport: ChatTransport
-  tools?: ChatTool[]
+  transport: AiChatProps['transport']
+  tools?: AiChatProps['tools']
   welcome?: string | string[]
   models?: ChatModelOption[]
   model?: string
   reasoningLevel?: string
   tokenUsageDetail?: boolean
+  messages?: ChatMessage[]
+  toolIcons?: Record<string, Component>
+  readonly?: boolean
   slots?: Record<string, (scope: any) => any>
 }) {
   const host = document.createElement('div')
@@ -43,7 +66,10 @@ function mountAiChat(options: {
           models: options.models,
           model: options.model,
           reasoningLevel: options.reasoningLevel,
-          tokenUsageDetail: options.tokenUsageDetail
+          tokenUsageDetail: options.tokenUsageDetail,
+          messages: options.messages,
+          toolIcons: options.toolIcons,
+          readonly: options.readonly
         },
         options.slots
       )
@@ -1251,4 +1277,376 @@ describe('UAiChat', () => {
     expect(noCache.host.querySelector('.u-ai-chat__input-usage')?.textContent).not.toContain('缓存')
     noCache.unmount()
   })
+
+  const UNKNOWN_TOOL = 'totally-unknown-tool-xyz'
+
+  function seedUnknownCall(init: Partial<ChatToolCall> = {}) {
+    const call = reactive<ChatToolCall>({
+      id: 'c1',
+      name: UNKNOWN_TOOL,
+      arguments: '{"q":"hi"}',
+      status: 'pending',
+      ...init
+    })
+    const messages: ChatMessage[] = [
+      { id: 'u1', role: 'user', content: 'go' },
+      { id: 'a1', role: 'assistant', content: '', status: 'done', toolCalls: [call] }
+    ]
+    return { call, messages }
+  }
+
+  async function expandToolCard(host: HTMLElement) {
+    const header = host.querySelector<HTMLElement>('.u-ai-chat__tool-call .u-collapse__header')
+    header?.dispatchEvent(new MouseEvent('click', { bubbles: true }))
+    await nextTick()
+    await nextTick()
+  }
+
+  function assertGenericCard(host: HTMLElement, status: 'pending' | 'success' | 'error') {
+    const card = host.querySelector(`.u-ai-chat__tool-call.is-${status}`)
+    expect(card).toBeTruthy()
+    expect(host.querySelector('.u-ai-chat__tool-call-name')?.textContent).toBe(UNKNOWN_TOOL)
+    expect(host.querySelector('.u-ai-chat__tool-call-status')).toBeTruthy()
+    expect(host.textContent).not.toContain('未找到')
+  }
+
+  it('未注册工具名 pending → success 走出通用卡片，无空白、无未找到工具文案', async () => {
+    const { call, messages } = seedUnknownCall()
+    const { host, unmount } = mountAiChat({ transport: () => {}, messages })
+    await nextTick()
+
+    assertGenericCard(host, 'pending')
+    expect(host.querySelector('.u-ai-chat__tool-call-name')?.classList.contains('u-shine')).toBe(
+      true
+    )
+    expect(host.textContent).toContain('"q": "hi"')
+
+    call.status = 'success'
+    call.result = '{"ok":true}'
+    await nextTick()
+    assertGenericCard(host, 'success')
+
+    await expandToolCard(host)
+    expect(host.textContent).toContain('"ok": true')
+    unmount()
+  })
+
+  it('未注册工具名 pending → error 走出通用卡片，无空白、无未找到工具文案', async () => {
+    const { call, messages } = seedUnknownCall()
+    const { host, unmount } = mountAiChat({ transport: () => {}, messages })
+    await nextTick()
+    assertGenericCard(host, 'pending')
+
+    call.status = 'error'
+    call.error = 'boom'
+    await nextTick()
+    assertGenericCard(host, 'error')
+
+    await expandToolCard(host)
+    expect(host.textContent).toContain('boom')
+    expect(host.textContent).toContain('错误')
+    unmount()
+  })
+
+  it('通用卡片展示 view，长 JSON 可截断并展开看全文', async () => {
+    const long = 'x'.repeat(500)
+    const { call, messages } = seedUnknownCall({
+      status: 'success',
+      result: JSON.stringify({ data: long }),
+      view: { kind: 'diff', body: 'patch-xyz' }
+    })
+    const { host, unmount } = mountAiChat({ transport: () => {}, messages })
+    await nextTick()
+    await expandToolCard(host)
+
+    expect(host.textContent).toContain('视图')
+    expect(host.textContent).toContain('patch-xyz')
+    expect(host.textContent).not.toContain(long)
+    const toggle = [
+      ...host.querySelectorAll<HTMLElement>('.u-ai-chat__tool-call-json-toggle')
+    ].find((el) => el.textContent?.includes('展开'))
+    expect(toggle).toBeTruthy()
+    toggle!.dispatchEvent(new MouseEvent('click', { bubbles: true }))
+    await nextTick()
+    expect(host.textContent).toContain(long)
+    unmount()
+  })
+
+  it('toolIcons 可覆盖未知名工具图标', async () => {
+    const OverrideIcon = defineComponent({
+      setup: () => () => h('svg', { class: 'override-tool-icon' })
+    })
+    const { messages } = seedUnknownCall()
+    const { host, unmount } = mountAiChat({
+      transport: () => {},
+      messages,
+      toolIcons: { [UNKNOWN_TOOL]: OverrideIcon }
+    })
+    await nextTick()
+    expect(host.querySelector('.override-tool-icon')).toBeTruthy()
+    unmount()
+  })
+
+  it('session 下不注入内置 askQuestion，无 meta 时走通用视图', async () => {
+    const history: ChatSessionEvent[] = [
+      {
+        type: 'tool/call',
+        callId: 'c1',
+        name: 'askQuestion',
+        arguments: '{"questions":[{"question":"Q?"}]}',
+        seq: 1
+      }
+    ]
+    const { host, unmount } = mountAiChat({
+      transport: createServerTransport({
+        subscribe: () => () => {},
+        send: async () => {},
+        cancel: async () => {},
+        respond: async () => {},
+        fetchHistory: async () => ({ events: history, hasMore: false }),
+        selectModel: async () => {}
+      })
+    })
+    await vi.waitFor(() => {
+      expect(host.querySelector('.u-ai-chat__tool-call-name')?.textContent).toBe('askQuestion')
+    })
+    expect(host.querySelector('.u-ai-chat__ask-question')).toBeFalsy()
+    unmount()
+  })
+
+  it('session 下 meta.render 优先于通用视图', async () => {
+    const meta: ChatToolMeta = {
+      name: 'server-tool',
+      label: '服务端工具',
+      render: defineComponent({
+        props: { toolCall: { type: Object as PropType<ChatToolCall>, required: true } },
+        setup: () => () => h('div', { class: 'meta-render' }, 'meta-body')
+      })
+    }
+    const history: ChatSessionEvent[] = [
+      { type: 'tool/call', callId: 'c1', name: 'server-tool', arguments: '{}', seq: 1 }
+    ]
+    const { host, unmount } = mountAiChat({
+      transport: createServerTransport({
+        subscribe: () => () => {},
+        send: async () => {},
+        cancel: async () => {},
+        respond: async () => {},
+        fetchHistory: async () => ({ events: history, hasMore: false }),
+        selectModel: async () => {}
+      }),
+      tools: [meta]
+    })
+    await vi.waitFor(() => {
+      expect(host.querySelector('.u-ai-chat__tool-call-name')?.textContent).toBe('服务端工具')
+    })
+    expect(host.querySelector('.meta-render')?.textContent).toBe('meta-body')
+    expect(host.textContent).not.toContain('参数')
+    unmount()
+  })
+})
+
+function createFakeAdapter(options?: {
+  history?: ChatSessionEvent[]
+}): ChatSessionAdapter & { emit: (event: ChatSessionEvent) => void } {
+  let listener: { onEvent(event: ChatSessionEvent): void } | undefined
+  const adapter: ChatSessionAdapter = {
+    subscribe(handlers) {
+      listener = handlers
+      return () => {
+        listener = undefined
+      }
+    },
+    send: vi.fn(async () => {}),
+    cancel: vi.fn(async () => {}),
+    respond: vi.fn(async () => {}),
+    fetchHistory: vi.fn(async () => ({ events: options?.history ?? [], hasMore: false })),
+    selectModel: vi.fn(async () => {})
+  }
+  return Object.assign(adapter, {
+    emit(event: ChatSessionEvent) {
+      listener?.onEvent(event)
+    }
+  })
+}
+
+async function mountSession(options?: {
+  history?: ChatSessionEvent[]
+  readonly?: boolean
+  welcome?: string | string[]
+}) {
+  const adapter = createFakeAdapter({ history: options?.history })
+  const mounted = mountAiChat({
+    transport: createServerTransport(adapter),
+    readonly: options?.readonly,
+    welcome: options?.welcome
+  })
+  await vi.waitFor(() => {
+    expect(adapter.fetchHistory).toHaveBeenCalled()
+  })
+  return { ...mounted, adapter }
+}
+
+const click = (el: HTMLElement) => el.dispatchEvent(new MouseEvent('click', { bubbles: true }))
+
+describe('UAiChat P4 作业条 / 审批 / 只读 / 提问', () => {
+  it('无 jobs 不渲染作业条；有 jobs 显示图标、文案、状态点，进行中扫光', async () => {
+    const { host, adapter, unmount } = await mountSession()
+    expect(host.querySelector('.u-ai-chat__job-bar')).toBeFalsy()
+
+    adapter.emit({
+      type: 'jobs/snapshot',
+      jobs: [{ id: 'j1', kind: 'bash', label: '跑测试', status: 'running' }]
+    })
+    await nextTick()
+
+    expect(host.querySelector('.u-ai-chat__job-bar')).toBeTruthy()
+    expect(host.querySelector('.u-ai-chat__job-bar-label')?.textContent).toContain('跑测试')
+    expect(host.querySelector('.u-ai-chat__job-bar-label')?.classList.contains('u-shine')).toBe(
+      true
+    )
+    expect(host.querySelector('.u-ai-chat__job-bar-dot.is-running')).toBeTruthy()
+
+    host.querySelector<HTMLElement>('.u-ai-chat__job-bar-head')!.click()
+    await nextTick()
+    expect(host.querySelector('.u-ai-chat__job-bar-item')).toBeFalsy()
+    unmount()
+  })
+
+  it('审批横幅仅无 callId 出现；有 callId 走工具卡确认', async () => {
+    const { host, adapter, unmount } = await mountSession()
+
+    adapter.emit({ type: 'tool/call', callId: 'c1', name: 'bash', arguments: '{}', seq: 1 })
+    adapter.emit({
+      type: 'approval/requested',
+      approvalId: 'a1',
+      toolName: 'bash',
+      callId: 'c1',
+      rpcId: 'rpc-card'
+    })
+    await nextTick()
+
+    expect(host.querySelector('.u-ai-chat__approval-banner-card')).toBeFalsy()
+    expect(host.querySelector('.u-ai-chat__tool-call-confirm')).toBeTruthy()
+
+    adapter.emit({
+      type: 'approval/requested',
+      approvalId: 'a2',
+      toolName: 'deploy',
+      reason: '发布到生产',
+      rpcId: 'rpc-banner'
+    })
+    await nextTick()
+
+    expect(host.querySelector('.u-ai-chat__approval-banner-card')).toBeTruthy()
+    expect(host.textContent).toContain('发布到生产')
+
+    const allow = [
+      ...host.querySelectorAll<HTMLElement>('.u-ai-chat__approval-banner-card button')
+    ].find((btn) => btn.textContent?.includes('允许'))!
+    click(allow)
+    await vi.waitFor(() => {
+      expect(adapter.respond).toHaveBeenCalledWith('rpc-banner', true, undefined)
+    })
+    unmount()
+  })
+
+  it('readonly 不展示输入区，欢迎语点击不发送，队列无插队编辑移除', async () => {
+    const { host, adapter, unmount } = await mountSession({ readonly: true, welcome: '点我发送' })
+    expect(host.querySelector('.u-ai-chat__input-area')).toBeFalsy()
+
+    host.querySelector<HTMLElement>('.u-ai-chat__welcome-item')!.click()
+    await nextTick()
+    expect(host.querySelector('.u-ai-chat__message--user')).toBeFalsy()
+
+    adapter.emit({ type: 'queue/snapshot', items: [{ id: 'q1', content: '排队中' }] })
+    await nextTick()
+    expect(host.querySelector('.u-ai-chat__queue-text')?.textContent).toContain('排队中')
+    expect(host.querySelector('.u-ai-chat__queue-actions')).toBeFalsy()
+    unmount()
+  })
+
+  it('非只读客户端输入区与欢迎点击发送与现状一致', async () => {
+    const transport: ChatTransport = (_req, handlers) => {
+      handlers.onTextDelta('收到')
+    }
+    const { host, unmount } = mountAiChat({ transport, welcome: '快捷提问' })
+    expect(host.querySelector('.u-ai-chat__input-area')).toBeTruthy()
+    host.querySelector<HTMLElement>('.u-ai-chat__welcome-item')!.click()
+    await nextTick()
+    expect(
+      host.querySelector('.u-ai-chat__message--user .u-ai-chat__message-bubble')?.textContent
+    ).toContain('快捷提问')
+    unmount()
+  })
+
+  it('ask-question 客户端提交走 resolveAskQuestion', async () => {
+    const rounds: ChatMessage[][] = []
+    let round = 0
+    const transport: ChatTransport = (req, handlers) => {
+      round++
+      rounds.push(req.messages)
+      if (round === 1) {
+        handlers.onToolCall?.({
+          id: 'q1',
+          name: 'askQuestion',
+          arguments: JSON.stringify({ questions: [{ question: 'Q？', options: ['A'] }] })
+        })
+      } else {
+        handlers.onTextDelta('明白了')
+      }
+    }
+    const { host, chat, unmount } = mountAiChat({ transport })
+    chat.value?.send('提问')
+    await vi.waitFor(() => {
+      expect(host.querySelector('.u-ai-chat__ask-question-q')?.textContent).toContain('Q？')
+    })
+    click(
+      [...host.querySelectorAll<HTMLElement>('.u-ai-chat__ask-question-opt')].find((el) =>
+        el.textContent?.includes('A')
+      )!
+    )
+    await nextTick()
+    click(
+      [...host.querySelectorAll<HTMLElement>('.u-ai-chat__ask-question-actions button')].find(
+        (btn) => btn.textContent?.includes('提交')
+      )!
+    )
+    await vi.waitFor(() => {
+      expect(rounds.length).toBe(2)
+    })
+    const toolMessage = rounds[1]![rounds[1]!.length - 1]!
+    expect(toolMessage.role).toBe('tool')
+    expect(JSON.parse(toolMessage.content)).toEqual({ answers: [{ question: 'Q？', answer: 'A' }] })
+    unmount()
+  })
+
+  it('ask-question session 提交走 session.respond', async () => {
+    const { host, adapter, unmount } = await mountSession()
+    adapter.emit({
+      type: 'question/requested',
+      rpcId: 'rpc-q',
+      questions: [{ question: '选一个', options: ['A', 'B'] }]
+    })
+    await nextTick()
+    expect(host.querySelector('.u-ai-chat__ask-question-q')?.textContent).toContain('选一个')
+
+    click(
+      [...host.querySelectorAll<HTMLElement>('.u-ai-chat__ask-question-opt')].find((el) =>
+        el.textContent?.includes('A')
+      )!
+    )
+    await nextTick()
+    click(
+      [...host.querySelectorAll<HTMLElement>('.u-ai-chat__ask-question-actions button')].find(
+        (btn) => btn.textContent?.includes('提交')
+      )!
+    )
+    await vi.waitFor(() => {
+      expect(adapter.respond).toHaveBeenCalledWith('rpc-q', true, [
+        { question: '选一个', answer: 'A' }
+      ])
+    })
+    unmount()
+  })
 })