ARG PYTHON_VERSION=3.11

ARG BASE_IMAGE=python:${PYTHON_VERSION}-slim-bookworm

ARG VENV_PATH=/prod_venv

FROM ${BASE_IMAGE} AS builder

RUN apt-get update && apt-get install -y --no-install-recommends python3-dev curl build-essential && apt-get clean && \

rm -rf /var/lib/apt/lists/*

RUN curl -LsSf https://astral.sh/uv/install.sh | sh && \

ln -s /root/.local/bin/uv /usr/local/bin/uv

ARG VENV_PATH

ENV VIRTUAL_ENV=${VENV_PATH}

RUN uv venv $VIRTUAL_ENV

ENV PATH="$VIRTUAL_ENV/bin:$PATH"

COPY storage/pyproject.toml storage/uv.lock storage/

COPY kserve/pyproject.toml kserve/uv.lock kserve/

RUN cd kserve && uv sync --active --no-cache

COPY kserve kserve

RUN cd kserve && uv sync --active --no-cache

COPY storage storage

RUN cd storage && uv pip install . --no-cache

COPY lgbserver/pyproject.toml lgbserver/uv.lock lgbserver/

RUN cd lgbserver && uv sync --active --no-cache

COPY lgbserver lgbserver

RUN cd lgbserver && uv sync --active --no-cache

COPY pyproject.toml pyproject.toml

COPY third_party/pip-licenses.py pip-licenses.py

RUN pip install --no-cache-dir tomli

RUN mkdir -p third_party/library && python3 pip-licenses.py

FROM ${BASE_IMAGE} AS prod

RUN apt-get update && apt-get install -y --no-install-recommends \

libgomp1 && \

apt-get clean && \

rm -rf /var/lib/apt/lists/* /tmp/* /var/tmp/*

ARG VENV_PATH

ENV VIRTUAL_ENV=${VENV_PATH}

ENV PATH="${VIRTUAL_ENV}/bin:$PATH"

RUN useradd kserve -m -u 1000 -d /home/kserve

COPY --from=builder --chown=kserve:kserve third_party third_party

COPY --from=builder --chown=kserve:kserve $VIRTUAL_ENV $VIRTUAL_ENV

COPY --from=builder kserve kserve

COPY --from=builder storage storage

COPY --from=builder lgbserver lgbserver

USER 1000

ENV PYTHONPATH=/lgbserver

ENTRYPOINT ["python", "-m", "lgbserver"]