gmakstutis/LightRAG
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

cherry-pick 8e3497dc

Browse source
This commit is contained in:
Raphaël MANSUY 2025-12-04 19:18:35 +08:00
parent 82a046fa24
commit 22004fee34
1 changed files with 11 additions and 20 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

31
Dockerfile
View file

@ -11,7 +11,7 @@ RUN cd lightrag_webui \
&& bun install --frozen-lockfile \ && bun install --frozen-lockfile \
&& bun run build && bun run build
# Python build stage - using uv for faster package installation # Python build stage - using uv for package installation
FROM ghcr.io/astral-sh/uv:python3.12-bookworm-slim AS builder FROM ghcr.io/astral-sh/uv:python3.12-bookworm-slim AS builder
ENV DEBIAN_FRONTEND=noninteractive ENV DEBIAN_FRONTEND=noninteractive
@ -20,7 +20,7 @@ ENV UV_COMPILE_BYTECODE=1
WORKDIR /app WORKDIR /app
# Install system deps (Rust is required by some wheels) # Install system dependencies required by some wheels
RUN apt-get update \ RUN apt-get update \
&& apt-get install -y --no-install-recommends \ && apt-get install -y --no-install-recommends \
curl \ curl \
@ -39,8 +39,8 @@ COPY pyproject.toml .
COPY setup.py . COPY setup.py .
COPY uv.lock . COPY uv.lock .
# Install base, API, and offline extras without the project to improve caching # Install project dependencies (base + API extras) without the project to improve caching
RUN uv sync --frozen --no-dev --extra api --extra offline --no-install-project --no-editable RUN uv sync --frozen --no-dev --extra api --no-install-project --no-editable
# Copy project sources after dependency layer # Copy project sources after dependency layer
COPY lightrag/ ./lightrag/ COPY lightrag/ ./lightrag/
@ -49,15 +49,9 @@ COPY lightrag/ ./lightrag/
COPY --from=frontend-builder /app/lightrag/api/webui ./lightrag/api/webui COPY --from=frontend-builder /app/lightrag/api/webui ./lightrag/api/webui
# Sync project in non-editable mode and ensure pip is available for runtime installs # Sync project in non-editable mode and ensure pip is available for runtime installs
RUN uv sync --frozen --no-dev --extra api --extra offline --no-editable \ RUN uv sync --frozen --no-dev --extra api --no-editable \
&& /app/.venv/bin/python -m ensurepip --upgrade && /app/.venv/bin/python -m ensurepip --upgrade
# Prepare offline cache directory and pre-populate tiktoken data
# Use uv run to execute commands from the virtual environment
RUN mkdir -p /app/data/tiktoken \
&& uv run lightrag-download-cache --cache-dir /app/data/tiktoken || status=$?; \
if [ -n "${status:-}" ] && [ "$status" -ne 0 ] && [ "$status" -ne 2 ]; then exit "$status"; fi
# Final stage # Final stage
FROM python:3.12-slim FROM python:3.12-slim
@ -79,23 +73,20 @@ COPY uv.lock .
# Ensure the installed scripts are on PATH # Ensure the installed scripts are on PATH
ENV PATH=/app/.venv/bin:/root/.local/bin:$PATH ENV PATH=/app/.venv/bin:/root/.local/bin:$PATH
# Install dependencies with uv sync (uses locked versions from uv.lock) # Sync dependencies inside the final image using uv
# And ensure pip is available for runtime installs # And ensure pip is available for runtime installs
RUN uv sync --frozen --no-dev --extra api --extra offline --no-editable \ RUN uv sync --frozen --no-dev --extra api --no-editable \
&& /app/.venv/bin/python -m ensurepip --upgrade && /app/.venv/bin/python -m ensurepip --upgrade
# Create persistent data directories AFTER package installation # Create persistent data directories
RUN mkdir -p /app/data/rag_storage /app/data/inputs /app/data/tiktoken RUN mkdir -p /app/data/rag_storage /app/data/inputs
# Copy offline cache into the newly created directory # Docker data directories
COPY --from=builder /app/data/tiktoken /app/data/tiktoken
# Point to the prepared cache
ENV TIKTOKEN_CACHE_DIR=/app/data/tiktoken
ENV WORKING_DIR=/app/data/rag_storage ENV WORKING_DIR=/app/data/rag_storage
ENV INPUT_DIR=/app/data/inputs ENV INPUT_DIR=/app/data/inputs
# Expose API port # Expose API port
EXPOSE 9621 EXPOSE 9621
# Set entrypoint
ENTRYPOINT ["python", "-m", "lightrag.api.lightrag_server"] ENTRYPOINT ["python", "-m", "lightrag.api.lightrag_server"]