services:

  init-db:
    build:
        context: .
        dockerfile: initdb.Dockerfile
    container_name: infobot-init
    depends_on:
      db:
        condition: service_healthy
    volumes:
      - ./scripts:/scripts:ro
      - ./data/hf_cache:/root/.cache/huggingface
    environment:
      DB_HOST: db
      DB_PORT: 5432
      DB_USER: infobot
      DB_PASSWORD: infobot
      DB_NAME: infobot
    entrypoint: ["python", "/scripts/init_and_load.py"]
    devices:
      - /dev/nvidia-uvm
      - /dev/nvidia-uvm-tools
      - /dev/nvidia-modeset
      - /dev/nvidiactl
      - /dev/nvidia0
    deploy:
      resources:
        reservations:
          devices:
            - driver: nvidia
              count: all
              capabilities: ["gpu"]
    networks: ["dumant"]


networks:
  dumant:
    name: dumant
    external: true