version: "3.8"

services:
  auto-sft:
    build:
      context: .
    image: auto-sft:latest
    container_name: auto-sft
    
    # Environment configuration
    env_file:
      - .env
    environment:
      - AUTO_SFT_HOST=0.0.0.0
      - AUTO_SFT_PORT=8765
      - AUTO_SFT_RUNS_DIR=/app/runs
      - AUTO_SFT_EVAL_API_KEY=${AUTO_SFT_EVAL_API_KEY:-}
      - AUTO_SFT_EVAL_API_BASE=${AUTO_SFT_EVAL_API_BASE:-https://openrouter.ai/api/v1}
      - HF_TOKEN=${HF_TOKEN:-}
      - HUGGING_FACE_HUB_TOKEN=${HF_TOKEN:-}
    
    # Port mapping
    ports:
      - "8765:8765"
    
    # Volume mounts for persistence
    volumes:
      - ./runs:/app/runs
      - ${DATA_DIR:-./data}:/app/data
      - ./config:/app/config
      - ./best_configs:/app/best_configs
      - ./finetuned:/app/finetuned
      - hf-cache:/root/.cache/huggingface
    
    # GPU support (Nvidia)
    deploy:
      resources:
        reservations:
          devices:
            - driver: nvidia
              count: 1
              capabilities: [gpu]
    
    # Restart policy
    restart: unless-stopped
    

# Named volumes for caching
volumes:
  hf-cache:
    driver: local