services:
  emb:
    image: "kemingy/mosec-emb-cpu"
    environment:
      - MOSEC_TIMEOUT=30000
    ports:
      - "8080:8000"

  pg:
    image: "tensorchord/pgvecto-rs:pg16-v0.3.0-alpha.1"
    environment:
      - POSTGRES_PASSWORD=password
    ports:
      - "5432:5432"
    volumes:
      - "/tmp/qtext_pgdata:/var/lib/postgresql/data"

  highlight:
    image: "kemingy/colbert-highlight"
    ports:
      - "8081:8000"

  encoder:
    image: "kemingy/cross-encoder"
    environment:
      - MOSEC_TIMEOUT=10000
    ports:
      - "8082:8000"

  sparse:
    image: "kemingy/spladepp"
    environment:
      - MOSEC_TIMEOUT=10000
      - HF_TOKEN=hf_xxxxxxxxx # add your huggingface token here (required by https://huggingface.co/prithivida/Splade_PP_en_v1)
    ports:
      - "8083:8000"

  server:
    build:
      context: ../
      dockerfile: docker/Dockerfile
    ports:
      - "8000:8000"
    depends_on:
      - pg
      - emb
      - highlight
      - encoder
      - sparse