docker-compose.yml

version: '3.8'

services:
  # FastAPI 后端服务
  api:
    build:
      context: ./backend
      dockerfile: Dockerfile
    container_name: ai-kb-api
    ports:
      - "8000:8000"
    environment:
      - APP_ENV=development
      - DATABASE_URL=postgresql+asyncpg://postgres:postgres@db:5432/ai_knowledge_base
      - REDIS_URL=redis://redis:6379/0
      - CELERY_BROKER_URL=redis://redis:6379/1
      - CELERY_RESULT_BACKEND=redis://redis:6379/2
      - CHROMADB_URL=http://chromadb:8000
      - DASHSCOPE_API_KEY=sk-0a68712def5c4a759b450f5f02630d6d
      - API_BASE_URL=https://dashscope.aliyuncs.com/compatible-mode/v1
      - EMBEDDING_MODEL=text-embedding-v3
      - LLM_MODEL=qwen-plus
      - MINIO_ENDPOINT=minio:9000
      - MINIO_ACCESS_KEY=admin
      - MINIO_SECRET_KEY=admin232918
      - MINIO_BUCKET=documents
      - MINIO_SECURE=false
      - STORAGE_TYPE=minio
      - JWT_SECRET_KEY=docker-development-secret-key-change-in-production-min-32-chars
      - JWT_ALGORITHM=HS256
      - JWT_EXPIRE_DAYS=30
      - PASSWORD_MIN_LENGTH=8
      - PASSWORD_REQUIRE_LETTER=true
      - PASSWORD_REQUIRE_NUMBER=true
      - ADMIN_USERNAME=admin
      - ADMIN_EMAIL=admin@example.com
      - ADMIN_PASSWORD=admin123456
      # SMTP 邮件服务配置
      - SMTP_HOST=smtp.gmail.com
      - SMTP_PORT=587
      - SMTP_USER=your-email@gmail.com
      - SMTP_PASSWORD=your-app-password
      # 验证码配置
      - VERIFICATION_CODE_LENGTH=6
      - VERIFICATION_CODE_EXPIRE=300
      - VERIFICATION_CODE_RATE_LIMIT=5
      # 加密配置
      - ENCRYPTION_KEY=M7YA5TTt7ltgC074c2kWXa9ZppcN1Os5OEKjMF0CWV0=
    volumes:
      - ./backend:/app
      - ./data:/app/data
    depends_on:
      - db
      - redis
      - chromadb
      - minio
    networks:
      - ai-kb-network

  # PostgreSQL 数据库
  db:
    image: postgres:15-alpine
    container_name: ai-kb-db
    environment:
      - POSTGRES_USER=postgres
      - POSTGRES_PASSWORD=postgres
      - POSTGRES_DB=ai_knowledge_base
    ports:
      - "5432:5432"
    volumes:
      - postgres_data:/var/lib/postgresql/data
    networks:
      - ai-kb-network

  # Redis 服务
  redis:
    image: redis:7-alpine
    container_name: ai-kb-redis
    ports:
      - "6379:6379"
    volumes:
      - redis_data:/data
    networks:
      - ai-kb-network

  # MinIO 对象存储
  minio:
    image: minio/minio:latest
    container_name: ai-kb-minio
    ports:
      - "9000:9000" # API 端口
      - "9001:9001" # Console 端口
    environment:
      - MINIO_ROOT_USER=admin
      - MINIO_ROOT_PASSWORD=admin232918
    volumes:
      - minio_data:/data
    command: server /data --console-address ":9001"
    networks:
      - ai-kb-network
    healthcheck:
      test: [ "CMD", "curl", "-f", "http://localhost:9000/minio/health/live" ]
      interval: 10s
      timeout: 5s
      retries: 3

  # ChromaDB 向量数据库
  chromadb:
    image: chromadb/chroma:latest
    container_name: ai-kb-chromadb
    ports:
      - "8001:8000"
    volumes:
      - chroma_data:/chroma/chroma
    environment:
      - IS_PERSISTENT=TRUE
      - ANONYMIZED_TELEMETRY=FALSE
    networks:
      - ai-kb-network
    healthcheck:
      test: [ "CMD", "curl", "-f", "http://localhost:8000/api/v1" ]
      interval: 10s
      timeout: 5s
      retries: 3

  # Celery Worker
  celery_worker:
    build:
      context: ./backend
      dockerfile: Dockerfile
    container_name: ai-kb-celery-worker
    command: uv run celery -A app.tasks worker --loglevel=info --concurrency=2
    environment:
      - APP_ENV=development
      - DATABASE_URL=postgresql+asyncpg://postgres:postgres@db:5432/ai_knowledge_base
      - REDIS_URL=redis://redis:6379/0
      - CELERY_BROKER_URL=redis://redis:6379/1
      - CELERY_RESULT_BACKEND=redis://redis:6379/2
      - CHROMADB_URL=http://chromadb:8000
      - DASHSCOPE_API_KEY=sk-0a68712def5c4a759b450f5f02630d6d
      - API_BASE_URL=https://dashscope.aliyuncs.com/compatible-mode/v1
      - EMBEDDING_MODEL=text-embedding-v3
      - LLM_MODEL=qwen-plus
      - MINIO_ENDPOINT=minio:9000
      - MINIO_ACCESS_KEY=admin
      - MINIO_SECRET_KEY=admin232918
      - MINIO_BUCKET=documents
      - MINIO_SECURE=false
      - STORAGE_TYPE=minio
      - JWT_SECRET_KEY=docker-development-secret-key-change-in-production-min-32-chars
      - JWT_ALGORITHM=HS256
      - JWT_EXPIRE_DAYS=30
      - PASSWORD_MIN_LENGTH=8
      - PASSWORD_REQUIRE_LETTER=true
      - PASSWORD_REQUIRE_NUMBER=true
      - ADMIN_USERNAME=admin
      - ADMIN_EMAIL=admin@example.com
      - ADMIN_PASSWORD=admin123456
      # SMTP 邮件服务配置
      - SMTP_HOST=smtp.gmail.com
      - SMTP_PORT=587
      - SMTP_USER=your-email@gmail.com
      - SMTP_PASSWORD=your-app-password
      # 验证码配置
      - VERIFICATION_CODE_LENGTH=6
      - VERIFICATION_CODE_EXPIRE=300
      - VERIFICATION_CODE_RATE_LIMIT=5
      # 加密配置
      - ENCRYPTION_KEY=M7YA5TTt7ltgC074c2kWXa9ZppcN1Os5OEKjMF0CWV0=
    volumes:
      - ./backend:/app
      - ./data:/app/data
    depends_on:
      - redis
      - db
      - chromadb
      - minio
    networks:
      - ai-kb-network

  # React 前端服务
  frontend:
    build:
      context: ./frontend
      dockerfile: Dockerfile
    container_name: ai-kb-frontend
    ports:
      - "3000:80"
    environment:
      - VITE_API_BASE_URL=http://localhost:8000
      - VITE_APP_TITLE=AI Knowledge Base
    depends_on:
      - api
    networks:
      - ai-kb-network
    healthcheck:
      test: ["CMD", "wget", "--no-verbose", "--tries=1", "--spider", "http://localhost:80"]
      interval: 30s
      timeout: 10s
      retries: 3

volumes:
  postgres_data:
  redis_data:
  chroma_data:
  minio_data:


networks:
  ai-kb-network:
    driver: bridge