Dockerfile

FROM python:3.12-slim

COPY --from=ghcr.io/astral-sh/uv:latest /uv /uvx /bin/

RUN apt-get update && apt-get install -y --no-install-recommends curl && rm -rf /var/lib/apt/lists/*

WORKDIR /app

COPY pyproject.toml uv.lock ./
RUN uv venv .venv && uv sync --frozen --no-dev --no-cache

COPY src/ src/
COPY scripts/ scripts/

# Download data and pre-build DataStore pickle at build time
RUN mkdir -p /tmp/mteb_cache && \
    curl -L -o /tmp/mteb_cache/__cached_results.json.gz \
    "https://raw.githubusercontent.com/embeddings-benchmark/results/cached-data/__cached_results.json.gz" && \
    python -c "import gzip,pathlib; d=gzip.decompress(pathlib.Path('/tmp/mteb_cache/__cached_results.json.gz').read_bytes()); pathlib.Path('/tmp/mteb_cache/__cached_results.json').write_bytes(d)" && \
    rm /tmp/mteb_cache/__cached_results.json.gz && \
    .venv/bin/python scripts/fetch_model_sizes.py && \
    .venv/bin/python scripts/prebuild_data.py && \
    rm /tmp/mteb_cache/__cached_results.json

EXPOSE 8080

ENV PYTHONUNBUFFERED=1

CMD [".venv/bin/python", "-m", "uvicorn", "src.main:app", "--host", "0.0.0.0", "--port", "8080", "--timeout-keep-alive", "120"]