From 48e6fb938744ee2b239ca94adc033b93ab7007c0 Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Mon, 29 Jun 2026 18:41:04 -0700 Subject: [PATCH] fix(headroom-proxy): simplify dockerfile, add import smoke test to catch startup failures in build --- prompt_compression/headroom-proxy/Dockerfile | 25 ++------------------ 1 file changed, 2 insertions(+), 23 deletions(-) diff --git a/prompt_compression/headroom-proxy/Dockerfile b/prompt_compression/headroom-proxy/Dockerfile index b2a47d06ecc..9593e9ad787 100644 --- a/prompt_compression/headroom-proxy/Dockerfile +++ b/prompt_compression/headroom-proxy/Dockerfile @@ -1,29 +1,8 @@ -FROM python:3.12-slim-bookworm AS builder - -RUN apt-get update \ - && apt-get install -y --no-install-recommends build-essential curl \ - && rm -rf /var/lib/apt/lists/* - -# 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 \ - && cd /tmp && sha256sum --check 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}" +FROM python:3.12-slim-bookworm RUN pip install --no-cache-dir "headroom-ai[proxy]==0.27.0" - -FROM python:3.12-slim-bookworm - -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 +RUN python -c "from headroom.proxy.server import run_server; print('import ok')" EXPOSE 8787