ARG PYTHON_VERSION=3.9

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

ARG VENV_PATH=/prod_venv

FROM ${BASE_IMAGE} as builder

ARG POETRY_HOME=/opt/poetry

ARG POETRY_VERSION=1.7.1

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

RUN python3 -m venv ${POETRY_HOME} && ${POETRY_HOME}/bin/pip install poetry==${POETRY_VERSION}

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

ARG VENV_PATH

ENV VIRTUAL_ENV=${VENV_PATH}

RUN python3 -m venv $VIRTUAL_ENV

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

COPY kserve/pyproject.toml kserve/poetry.lock kserve/

RUN cd kserve && poetry install --no-root --no-interaction --no-cache

COPY kserve kserve

RUN cd kserve && poetry install --no-interaction --no-cache

COPY aiffairness/pyproject.toml aiffairness/poetry.lock aiffairness/

RUN cd aiffairness && poetry install --no-root --no-interaction --no-cache

COPY aiffairness aiffairness

RUN cd aiffairness && poetry install --no-interaction --no-cache

FROM ${BASE_IMAGE} as prod

COPY third_party third_party

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 $VIRTUAL_ENV $VIRTUAL_ENV

COPY --from=builder kserve kserve

COPY --from=builder aiffairness aiffairness

USER 1000

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