services:
  prefect-server:
    image: prefecthq/prefect:3-python3.12
    container_name: prefect-server
    command: prefect server start --host 0.0.0.0
    environment:
      PREFECT_UI_API_URL: http://192.168.124.92:4200/api
      TZ: Asia/Shanghai
    ports:
      - "4200:4200"
    volumes:
      - prefect-data:/root/.prefect
    restart: unless-stopped

  prefect-worker:
    build: .
    container_name: prefect-worker
    environment:
      PREFECT_API_URL: http://prefect-server:4200/api
      TZ: Asia/Shanghai
    depends_on:
      - prefect-server
    volumes:
      # 设备监控(device-check flow):宿主机磁盘使用率 + Docker 容器状态
      - /var/run/docker.sock:/var/run/docker.sock
      - /:/host:ro
    deploy:
      resources:
        reservations:
          devices:
            # GPU 监控(nvidia-smi);依赖宿主机 nvidia-container-toolkit
            - driver: nvidia
              count: all
              capabilities: [gpu]
    restart: unless-stopped

volumes:
  prefect-data: