docker-compose.yml

services:

  ollama:
    image: ollama/ollama
    restart: always
    volumes:
      - ./services/ollama:/root/.ollama
    deploy:
      resources:
        reservations:
          devices:
            - capabilities: [gpu]

  chroma:
    image: chromadb/chroma:latest
    restart: always
    volumes:
      - ./services/webui/vector_db:/chroma/chroma
    environment:
      - IS_PERSISTENT=TRUE

  webui:
    image: ghcr.io/open-webui/open-webui:main
    restart: always
    depends_on:
      - ollama
      - chroma
    ports:
      - 8080:8080
    environment:
      - OLLAMA_BASE_URL=http://ollama:11434
      - WEBUI_SECRET_KEY=${WEBUI_SECRET_KEY}
      - CHROMA_SERVER_HOST=http://chroma:8000
    volumes:
      - ./services/webui:/app/backend/data

  notebook:
    build: ./services/jupyter/
    restart: always
    volumes:
      - ./services/jupyter/notebook:/project
    depends_on:
      - ollama
      - chroma
    ports:
      - 8888:8888