forked from phoenix/litellm-mirror
97 lines
2.5 KiB
Text
97 lines
2.5 KiB
Text
# Base image for building
|
|
ARG LITELLM_BUILD_IMAGE=python:3.11.8-slim
|
|
|
|
# Runtime image
|
|
ARG LITELLM_RUNTIME_IMAGE=python:3.11.8-slim
|
|
# Builder stage
|
|
FROM $LITELLM_BUILD_IMAGE as builder
|
|
|
|
# Set the working directory to /app
|
|
WORKDIR /app
|
|
|
|
ARG LITELLM_USER=litellm LITELLM_UID=1729
|
|
ARG LITELLM_GROUP=litellm LITELLM_GID=1729
|
|
|
|
RUN groupadd \
|
|
--gid ${LITELLM_GID} \
|
|
${LITELLM_GROUP} \
|
|
&& useradd \
|
|
--create-home \
|
|
--shell /bin/sh \
|
|
--gid ${LITELLM_GID} \
|
|
--uid ${LITELLM_UID} \
|
|
${LITELLM_USER}
|
|
|
|
# Allows user to update python install.
|
|
# This is necessary for prisma.
|
|
RUN chown -R ${LITELLM_USER}:${LITELLM_GROUP} /usr/local/lib/python3.11
|
|
|
|
# Set the HOME var forcefully because of prisma.
|
|
ENV HOME=/home/${LITELLM_USER}
|
|
USER ${LITELLM_USER}
|
|
|
|
# Install build dependencies
|
|
RUN apt-get clean && apt-get update && \
|
|
apt-get install -y gcc python3-dev && \
|
|
rm -rf /var/lib/apt/lists/*
|
|
|
|
RUN pip install --upgrade pip && \
|
|
pip install build
|
|
|
|
# Copy the current directory contents into the container at /app
|
|
COPY . .
|
|
|
|
# Build Admin UI
|
|
RUN chmod +x build_admin_ui.sh && ./build_admin_ui.sh
|
|
|
|
# Build the package
|
|
RUN rm -rf dist/* && python -m build
|
|
|
|
# There should be only one wheel file now, assume the build only creates one
|
|
RUN ls -1 dist/*.whl | head -1
|
|
|
|
# Install the package
|
|
RUN pip install dist/*.whl
|
|
|
|
# install dependencies as wheels
|
|
RUN pip wheel --no-cache-dir --wheel-dir=/wheels/ -r requirements.txt
|
|
|
|
# Runtime stage
|
|
FROM $LITELLM_RUNTIME_IMAGE as runtime
|
|
|
|
WORKDIR /app
|
|
# Copy the current directory contents into the container at /app
|
|
COPY . .
|
|
RUN ls -la /app
|
|
|
|
# Copy the built wheel from the builder stage to the runtime stage; assumes only one wheel file is present
|
|
COPY --from=builder /app/dist/*.whl .
|
|
COPY --from=builder /wheels/ /wheels/
|
|
|
|
# Install the built wheel using pip; again using a wildcard if it's the only file
|
|
RUN pip install *.whl /wheels/* --no-index --find-links=/wheels/ && rm -f *.whl && rm -rf /wheels
|
|
|
|
# install semantic-cache [Experimental]- we need this here and not in requirements.txt because redisvl pins to pydantic 1.0
|
|
RUN pip install redisvl==0.0.7 --no-deps
|
|
|
|
# ensure pyjwt is used, not jwt
|
|
RUN pip uninstall jwt -y
|
|
RUN pip uninstall PyJWT -y
|
|
RUN pip install PyJWT --no-cache-dir
|
|
|
|
# Build Admin UI
|
|
RUN chmod +x build_admin_ui.sh && ./build_admin_ui.sh
|
|
|
|
# Generate prisma client
|
|
RUN prisma generate
|
|
RUN chmod +x entrypoint.sh
|
|
|
|
EXPOSE 4000/tcp
|
|
|
|
# # Set your entrypoint and command
|
|
|
|
ENTRYPOINT ["litellm"]
|
|
|
|
# Append "--detailed_debug" to the end of CMD to view detailed debug logs
|
|
# CMD ["--port", "4000", "--detailed_debug"]
|
|
CMD ["--port", "4000"]
|