从零构建ChatGPT：7天完成AI聊天应用

实现流式对话

// app/api/chat/route.ts
import { OpenAI } from 'openai';
import { StreamingTextResponse, OpenAIStream } from 'ai';
import { prisma } from '@/lib/prisma';

const openai = new OpenAI({
  apiKey: process.env.OPENAI_API_KEY,
});

export async function POST(req: Request) {
  const { messages, conversationId, model = 'gpt-3.5-turbo' } = await req.json();
  
  try {
    // 创建流式响应
    const response = await openai.chat.completions.create({
      model,
      messages,
      stream: true,
      temperature: 0.7,
      max_tokens: 2000,
    });
    
    // 转换为可读流
    const stream = OpenAIStream(response, {
      async onCompletion(completion) {
        // 保存消息到数据库
        await prisma.message.create({
          data: {
            role: 'assistant',
            content: completion,
            conversationId,
            model,
            tokens: completion.split(' ').length * 1.3, // 估算
          },
        });
      },
    });
    
    return new StreamingTextResponse(stream);
  } catch (error) {
    console.error('Chat API Error:', error);
    return new Response('Internal Server Error', { status: 500 });
  }
}

// components/chat/ChatInterface.tsx
'use client';

import { useState, useRef, useEffect } from 'react';
import { useChat } from 'ai/react';
import MessageList from './MessageList';
import MessageInput from './MessageInput';
import ModelSelector from './ModelSelector';

export default function ChatInterface() {
  const [selectedModel, setSelectedModel] = useState('gpt-3.5-turbo');
  const messagesEndRef = useRef<HTMLDivElement>(null);
  
  const { messages, input, handleInputChange, handleSubmit, isLoading } = useChat({
    api: '/api/chat',
    body: {
      model: selectedModel,
    },
    onError: (error) => {
      console.error('Chat error:', error);
      toast.error('发送消息失败，请重试');
    },
  });
  
  // 自动滚动到底部
  useEffect(() => {
    messagesEndRef.current?.scrollIntoView({ behavior: 'smooth' });
  }, [messages]);
  
  return (
    <div className="flex flex-col h-screen max-w-4xl mx-auto">
      {/* 头部 */}
      <div className="border-b p-4 flex justify-between items-center">
        <h1 className="text-xl font-semibold">ChatGPT Clone</h1>
        <ModelSelector 
          value={selectedModel} 
          onChange={setSelectedModel}
        />
      </div>
      
      {/* 消息列表 */}
      <div className="flex-1 overflow-y-auto p-4">
        <MessageList messages={messages} />
        <div ref={messagesEndRef} />
      </div>
      
      {/* 输入区域 */}
      <div className="border-t p-4">
        <MessageInput
          value={input}
          onChange={handleInputChange}
          onSubmit={handleSubmit}
          isLoading={isLoading}
        />
      </div>
    </div>
  );
}