Function stopGeneration

src/stores/aiChat.ts:1835–1869

(chatKey: string)

  }

  async function stopGeneration(chatKey: string): Promise<boolean> {
    const state = getSessionState(chatKey)
    if (!state || !state.isAIThinking) return false

    state.isAborted = true
    state.isAIThinking = false
    state.isLoadingSource = false
    state.currentToolStatus = null
    setAgentPhase(state, 'aborted')

    // 停止时优先定位真实仍在流式写入的会话缓冲，而不是当前页面正在查看的缓冲。
    const runningBufferKey =
      activeTask.value?.chatKey === chatKey
        ? (activeTask.value.aiChatId ?? DRAFT_AI_CHAT_KEY)
        : getDisplayedBufferKey(state)
    const runningBuffer = state.aiChatBuffers[runningBufferKey]
    const lastMessage = runningBuffer ? runningBuffer.messages[runningBuffer.messages.length - 1] : undefined
    if (lastMessage && lastMessage.role === 'assistant' && lastMessage.isStreaming) {
      lastMessage.isStreaming = false
      lastMessage.content += '\n\n_（已停止生成）_'
    }

    if (state.currentAgentRequestId) {
      try {
        await useAgentStreamService().abort(state.currentAgentRequestId)
      } catch (error) {
        console.error('[AI] 中止 Agent 请求失败:', error)
      }
    }

    state.currentRequestId = ''
    state.currentAgentRequestId = ''
    clearActiveTask(chatKey)
    return true
  }

  async function stopActiveTask(): Promise<boolean> {
    if (!activeTask.value) return false