services:
  airc:
    build:
      context: .
      dockerfile: Dockerfile
      target: airc
    image: airc
    restart: "no"
    env_file:
      - .env
    devices:
      - /dev/dri:/dev/dri
    depends_on:
      - ollama
    networks:
      - internal
    volumes:
      - ./cache:/root/.cache
      - ./src:/opt/airc/src:rw
    cap_add: # used for running ze-monitor within airc container
      - CAP_DAC_READ_SEARCH # Bypass all filesystem read access checks
      - CAP_PERFMON         # Access to perf_events (vs. overloaded CAP_SYS_ADMIN)
      - CAP_SYS_PTRACE      # PTRACE_MODE_READ_REALCREDS ptrace access mode check

  ollama:
    build:
      context: .
      dockerfile: Dockerfile
      target: ollama
    image: ollama
    restart: "no"
    env_file:
      - .env
    environment:
      - OLLAMA_HOST=0.0.0.0
      - ONEAPI_DEVICE_SELECTOR=level_zero:0
    devices:
      - /dev/dri:/dev/dri
#    ports:
#      - 11434:11434 # ollama serve port
    networks:
      - internal
    volumes:
      - ./cache:/root/.cache   # Cache hub models and neo_compiler_cache
      - ./ollama:/root/.ollama # Cache the ollama models
      - ./src:/opt/airc/src:rw # Live mount src 
    cap_add: # used for running ze-monitor within airc container
      - CAP_DAC_READ_SEARCH # Bypass all filesystem read access checks
      - CAP_PERFMON         # Access to perf_events (vs. overloaded CAP_SYS_ADMIN)
      - CAP_SYS_PTRACE      # PTRACE_MODE_READ_REALCREDS ptrace access mode check

  jupyter:
    build:
      context: .
      dockerfile: Dockerfile
      target: jupyter
    image: jupyter
    env_file:
      - .env
    devices:
      - /dev/dri:/dev/dri
    depends_on:
      - ollama
    ports:
    - 8888:8888 # Jupyter Notebook
    networks:
      - internal
    volumes:
      - ./jupyter:/opt/jupyter:rw
      - ./cache:/root/.cache

networks:
  internal:
    driver: bridge