fastapi-ai-chat-module-skill

FastAPI AI Chat Module Skill

面向已有 FastAPI 项目的开发者，快速集成 AI 聊天能力。

能力清单

触发场景

用户说"帮我加 AI 聊天"或"集成 AI 对话"时触发。

核心实现

依赖配置

pip install openai anthropic httpx sse-starlette

配置

# config.py
from pydantic_settings import BaseSettings

class Settings(BaseSettings):
    openai_api_key: str = ""
    openai_base_url: str = "https://api.openai.com/v1"
    openai_model: str = "gpt-4"
    
    # 会话配置
    chat_max_history: int = 10
    chat_max_tokens: int = 4000

settings = Settings()

数据模型

# models.py
from sqlalchemy import Column, BigInteger, String, Text, Integer, DateTime, Enum as SQLEnum
from sqlalchemy.sql import func
from database import Base
import enum

class MessageRole(str, enum.Enum):
    USER = "user"
    ASSISTANT = "assistant"
    SYSTEM = "system"

class ChatModel(str, enum.Enum):
    GPT_3_5 = "gpt-3.5-turbo"
    GPT_4 = "gpt-4"
    CLAUDE = "claude-3"

class ChatSession(Base):
    __tablename__ = "wg_chat_session"
    
    id = Column(BigInteger, primary_key=True, autoincrement=True)
    user_id = Column(String(50), nullable=False, index=True)
    title = Column(String(200))
    model = Column(SQLEnum(ChatModel), default=ChatModel.GPT_3_5)
    status = Column(Integer, default=1)
    created_at = Column(DateTime, server_default=func.now())
    updated_at = Column(DateTime, server_default=func.now(), onupdate=func.now())

class ChatMessage(Base):
    __tablename__ = "wg_chat_message"
    
    id = Column(BigInteger, primary_key=True, autoincrement=True)
    session_id = Column(BigInteger, nullable=False, index=True)
    role = Column(SQLEnum(MessageRole), nullable=False)
    content = Column(Text, nullable=False)
    token_count = Column(Integer, default=0)
    created_at = Column(DateTime, server_default=func.now())