scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-10 22:41:41 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 129

fix: copy local litellm-proxy-extras schema into installed package in Docker runtime stage

Browse source
This commit is contained in:
Ishaan Jaffer 2026-04-01 10:10:13 -07:00
parent 6683c9b601
commit 48e814fc28
1 changed files with 9 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
docker/Dockerfile.database
View file

@ -42,11 +42,6 @@ RUN pip install dist/*.whl
# install dependencies as wheels
RUN pip wheel --no-cache-dir --wheel-dir=/wheels/ -r requirements.txt
# Override litellm-proxy-extras with local version to pick up schema changes not in PyPI release
RUN cd /app/litellm-proxy-extras && python -m build && \
rm -f /wheels/litellm_proxy_extras-*.whl && \
cp dist/litellm_proxy_extras-*.whl /wheels/
# Runtime stage
FROM $LITELLM_RUNTIME_IMAGE AS runtime
@ -89,6 +84,15 @@ 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
# Override litellm-proxy-extras schema with local version to pick up schema changes not in the PyPI release.
# This copies the updated schema.prisma (which includes new columns) directly over the installed package's
# schema file, preventing _resolve_all_migrations from generating DROP COLUMN for new fields.
RUN PKG_DIR=$(python3 -c "import os, litellm_proxy_extras; print(os.path.dirname(litellm_proxy_extras.__file__))") && \
cp /app/litellm-proxy-extras/litellm_proxy_extras/schema.prisma "$PKG_DIR/schema.prisma" && \
if [ -d /app/litellm-proxy-extras/litellm_proxy_extras/migrations ]; then \
cp -r /app/litellm-proxy-extras/litellm_proxy_extras/migrations/. "$PKG_DIR/migrations/"; \
fi
# SECURITY FIX: nodejs-wheel-binaries (pip package used by Prisma) bundles a complete
# npm with old vulnerable deps at /usr/lib/python3.*/site-packages/nodejs_wheel/.
# Patch every copy of tar, glob, and brace-expansion inside that tree.