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 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 custom_tokenizer/pyproject.toml custom_tokenizer/poetry.lock custom_tokenizer/

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

COPY custom_tokenizer custom_tokenizer

RUN cd custom_tokenizer && 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 custom_tokenizer custom_tokenizer

USER 1000

ENTRYPOINT ["python", "-m", "custom_tokenizer.transformer"]