version: '3.8'

services:
  gateway:
    build: .
    ports:
      - "8080:8080"
      - "8081:8081"
    volumes:
      - ./configs/config.yaml:/etc/edgeai/config.yaml:ro
      - gateway-data:/var/lib/edgeai
    environment:
      - EDGEAI_LOG_LEVEL=info
    depends_on:
      - ollama
    restart: unless-stopped

  ollama:
    image: ollama/ollama:latest
    ports:
      - "11434:11434"
    volumes:
      - ollama-models:/root/.ollama
    deploy:
      resources:
        reservations:
          devices:
            - driver: nvidia
              count: 1
              capabilities: [gpu]
    restart: unless-stopped

  prometheus:
    image: prom/prometheus:latest
    ports:
      - "9090:9090"
    volumes:
      - ./deploy/prometheus.yml:/etc/prometheus/prometheus.yml:ro
    restart: unless-stopped

  grafana:
    image: grafana/grafana:latest
    ports:
      - "3000:3000"
    volumes:
      - grafana-data:/var/lib/grafana
    restart: unless-stopped

volumes:
  gateway-data:
  ollama-models:
  grafana-data: