Function handleGenerateResponse

components/chat-interface.tsx:189–360

(queryText: string, userMessageId: string, assistantMessageId: string)

  // 处理生成回复的逻辑（从handleSubmit中提取出来的核心部分）
  const handleGenerateResponse = async (queryText: string, userMessageId: string, assistantMessageId: string) => {
    try {
      // 创建新的AbortController
      if (abortControllerRef.current) {
        abortControllerRef.current.abort() // 取消之前的请求（如果有）
      }
      abortControllerRef.current = new AbortController()

      setIsStreaming(true)

      // 准备对话历史
      const conversationHistory = messages
        .filter((m) => m.role === "user" || m.role === "assistant")
        .map((m) => ({
          role: m.role,
          content: m.content,
        }))

      // 添加当前用户消息
      conversationHistory.push({
        role: "user",
        content: queryText,
      })

      // 开始流式响应，传入AbortSignal
      const stream = await generateStreamingResponseClient(conversationHistory, abortControllerRef.current.signal)

      if (!stream) {
        throw new Error("无法创建流式响应")
      }

      let fullResponse = ""

      // 处理流式响应
      for await (const chunk of stream) {
        // 如果请求已被取消，停止处理
        if (abortControllerRef.current?.signal.aborted) {
          break
        }

        if (chunk) {
          fullResponse += chunk

          // 更新助手消息的内容
          setMessages((prev) =>
            prev.map((msg) =>
              msg.id === assistantMessageId ? { ...msg, content: fullResponse, isLoading: false } : msg,
            ),
          )
        }
      }

      // 如果请求已被取消，不再继续后续处理
      if (abortControllerRef.current?.signal.aborted) {
        return
      }

      // 流式响应完成后，等待一小段时间确保前端更新