services:
  ollama:
    image: ollama/ollama:latest
    container_name: ollama
    restart: unless-stopped

    # NVIDIA via Docker Compose v2
    gpus: all

    ports:
      - "${OLLAMA_PORT:-11434}:11434"

    volumes:
      - "${OLLAMA_DATA_DIR:-/datadisk/docker/ollama}:/root/.ollama"

  open-webui:
    image: ghcr.io/open-webui/open-webui:main
    container_name: open-webui
    restart: unless-stopped
    depends_on:
      - ollama
    environment:
      - OLLAMA_BASE_URL=http://ollama:11434
    ports:
      - "${OPENWEBUI_PORT:-3000}:8080"
    volumes:
      - "${OPENWEBUI_DATA_DIR:-/datadisk/docker/open-webui}:/app/backend/data"