scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-07 02:59:05 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 103

fix(headroom-proxy): use python:3.12 builder with pinned verified rustup, copy site-packages to final stage

Browse source
This commit is contained in:
Krrish Dholakia 2026-06-29 18:29:50 -07:00
parent 3c7daf6e6c
commit 9a394427e7
1 changed files with 16 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

22
prompt_compression/headroom-proxy/Dockerfile
View file

@ -1,19 +1,29 @@
FROM rust:1.87-slim-bookworm AS builder
FROM python:3.12-slim-bookworm AS builder
RUN apt-get update \
&& apt-get install -y --no-install-recommends python3 python3-pip python3-venv build-essential \
&& apt-get install -y --no-install-recommends build-essential curl \
&& rm -rf /var/lib/apt/lists/*
RUN python3 -m venv /opt/venv
ENV PATH="/opt/venv/bin:$PATH"
# Download rustup-init and verify its checksum using the official sidecar.
# Pinned to rustup 1.28.1 on x86_64 linux.
RUN curl -Lo /tmp/rustup-init \
https://static.rust-lang.org/rustup/archive/1.28.1/x86_64-unknown-linux-gnu/rustup-init \
&& curl -Lo /tmp/rustup-init.sha256 \
https://static.rust-lang.org/rustup/archive/1.28.1/x86_64-unknown-linux-gnu/rustup-init.sha256 \
&& sha256sum --check /tmp/rustup-init.sha256 \
&& chmod +x /tmp/rustup-init \
&& /tmp/rustup-init -y --profile minimal --default-toolchain 1.87.0 --no-modify-path \
&& rm /tmp/rustup-init /tmp/rustup-init.sha256
ENV PATH="/root/.cargo/bin:${PATH}"
RUN pip install --no-cache-dir "headroom-ai[proxy]==0.27.0"
FROM python:3.12-slim-bookworm
COPY --from=builder /opt/venv /opt/venv
ENV PATH="/opt/venv/bin:$PATH"
COPY --from=builder /usr/local/lib/python3.12/site-packages /usr/local/lib/python3.12/site-packages
COPY --from=builder /usr/local/bin/headroom /usr/local/bin/headroom
EXPOSE 8787