content: ARG PYTHON_VERSION=3.11
  ARG BASE_IMAGE=python:${PYTHON_VERSION}-slim-bookworm
  ARG VENV_PATH=/prod_venv
  
  FROM ${BASE_IMAGE} AS builder
  
  # Required for building packages for arm64 arch
  RUN apt-get update && apt-get install -y --no-install-recommends curl python3-dev build-essential && apt-get clean && \
      rm -rf /var/lib/apt/lists/*
  
  # Install uv
  RUN curl -LsSf https://astral.sh/uv/install.sh | sh && \
      ln -s /root/.local/bin/uv /usr/local/bin/uv
  
  # Setup virtual environment
  ARG VENV_PATH
  ENV VIRTUAL_ENV=${VENV_PATH}
  RUN uv venv $VIRTUAL_ENV
  ENV PATH="$VIRTUAL_ENV/bin:$PATH"
  
  # ------------------ kserve deps ------------------
  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
  
  # ------------------ artexplainer deps ------------------
  COPY artexplainer/pyproject.toml artexplainer/uv.lock artexplainer/
  RUN cd artexplainer && uv sync --active --no-cache
  
  COPY artexplainer artexplainer
  RUN cd artexplainer && uv sync --active --no-cache
  
  # Generate third-party licenses
  COPY pyproject.toml pyproject.toml
  COPY third_party/pip-licenses.py pip-licenses.py
  # TODO: Remove this when upgrading to python 3.11+
  RUN pip install --no-cache-dir tomli
  RUN mkdir -p third_party/library && python3 pip-licenses.py
  
  
  # ------------------ Production stage ------------------
  FROM ${BASE_IMAGE} AS prod
  
  # Activate virtual env
  ARG VENV_PATH
  ENV VIRTUAL_ENV=${VENV_PATH}
  ENV PATH="$VIRTUAL_ENV/bin:$PATH"
  
  COPY third_party third_party
  
  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 artexplainer artexplainer
  
  USER 1000
  ENV PYTHONPATH=/artexplainer
  ENTRYPOINT ["python", "-m", "artserver"]
  