mirror of
https://github.com/bytedance/deer-flow.git
synced 2026-07-09 16:08:31 +00:00
* feat: add redis stream bridge * Potential fix for pull request finding Co-authored-by: Copilot Autofix powered by AI <175728472+Copilot@users.noreply.github.com> * fix(gateway): address redis stream bridge review Redis was imported eagerly through deerflow.runtime and declared as a hard dependency, which made memory-only installs load redis.asyncio at startup and left the lazy factory import ineffective. Move redis behind an optional extra, remove the public eager re-export, and keep make_stream_bridge as the only runtime import path with an actionable install hint when the extra is missing. Because Docker deployments now default the stream bridge to Redis via DEER_FLOW_STREAM_BRIDGE_REDIS_URL, install the redis extra explicitly in Docker/dev container flows and teach the local uv-extra detector to infer redis from both stream_bridge.type and the Redis URL env var. This keeps Docker working while preserving slim non-Docker installs. Harden the Redis bridge by batching XREAD replay, replacing brittle ResponseError string matching with a single fallback to 0-0 for malformed Last-Event-ID values, documenting connection/retention/fail-hard behavior, and adding fake plus opt-in real Redis coverage for XADD/XREAD, replay, invalid IDs, and MAXLEN trimming. * fix(config): bump config version for stream bridge * fix redis stream bridge terminal handling * fix: repair uv.lock, format redis.py, and align Dockerfile extras test The uv.lock file was missing a closing bracket for the redis extras section, redis.py had a formatting issue caught by ruff, and the Dockerfile extras test did not account for the hardcoded --extra redis flag. Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com> --------- Co-authored-by: Willem Jiang <willem.jiang@gmail.com> Co-authored-by: Copilot Autofix powered by AI <175728472+Copilot@users.noreply.github.com> Co-authored-by: Claude Opus 4.6 <noreply@anthropic.com>
116 lines
5 KiB
Docker
116 lines
5 KiB
Docker
# Backend Dockerfile — multi-stage build
|
|
# Stage 1 (builder): compiles native Python extensions with build-essential
|
|
# Stage 2 (dev): retains toolchain for dev containers (uv sync at startup)
|
|
# Stage 3 (runtime): clean image without compiler toolchain for production
|
|
|
|
# UV source image (override for restricted networks that cannot reach ghcr.io)
|
|
ARG UV_IMAGE=ghcr.io/astral-sh/uv:0.7.20
|
|
FROM ${UV_IMAGE} AS uv-source
|
|
|
|
# ── Stage 1: Builder ──────────────────────────────────────────────────────────
|
|
FROM python:3.12-slim-bookworm AS builder
|
|
|
|
ARG NODE_MAJOR=22
|
|
ARG APT_MIRROR
|
|
ARG UV_INDEX_URL
|
|
# Optional extras to install (e.g. "postgres" for PostgreSQL support)
|
|
# Usage: docker build --build-arg UV_EXTRAS=postgres,discord ...
|
|
ARG UV_EXTRAS
|
|
|
|
# Optionally override apt mirror for restricted networks (e.g. APT_MIRROR=mirrors.aliyun.com)
|
|
RUN if [ -n "${APT_MIRROR}" ]; then \
|
|
sed -i "s|deb.debian.org|${APT_MIRROR}|g" /etc/apt/sources.list.d/debian.sources 2>/dev/null || true; \
|
|
sed -i "s|deb.debian.org|${APT_MIRROR}|g" /etc/apt/sources.list 2>/dev/null || true; \
|
|
fi
|
|
|
|
# Install build tools + Node.js (build-essential needed for native Python extensions)
|
|
RUN apt-get update && apt-get install -y \
|
|
curl \
|
|
build-essential \
|
|
gnupg \
|
|
ca-certificates \
|
|
&& mkdir -p /etc/apt/keyrings \
|
|
&& curl -fsSL https://deb.nodesource.com/gpgkey/nodesource-repo.gpg.key | gpg --dearmor -o /etc/apt/keyrings/nodesource.gpg \
|
|
&& echo "deb [signed-by=/etc/apt/keyrings/nodesource.gpg] https://deb.nodesource.com/node_${NODE_MAJOR}.x nodistro main" > /etc/apt/sources.list.d/nodesource.list \
|
|
&& apt-get update \
|
|
&& apt-get install -y nodejs \
|
|
&& rm -rf /var/lib/apt/lists/*
|
|
|
|
# Install uv (source image overridable via UV_IMAGE build arg)
|
|
COPY --from=uv-source /uv /uvx /usr/local/bin/
|
|
|
|
# Set working directory
|
|
WORKDIR /app
|
|
|
|
# Copy backend source code
|
|
COPY backend ./backend
|
|
|
|
# Install dependencies with cache mount
|
|
# `redis` is always installed because Docker Compose defaults the stream bridge
|
|
# to Redis (DEER_FLOW_STREAM_BRIDGE_REDIS_URL); it is an optional extra elsewhere
|
|
# so slim non-Docker installs do not pull it.
|
|
# When UV_EXTRAS is set (comma- or whitespace-separated), installs optional dependencies.
|
|
RUN --mount=type=cache,target=/root/.cache/uv \
|
|
sh -c 'cd backend && \
|
|
set -f; \
|
|
extras_flags=""; \
|
|
for extra in $(printf "%s" "$UV_EXTRAS" | tr "," " "); do \
|
|
case "$extra" in \
|
|
[!A-Za-z]* | *[!A-Za-z0-9_-]*) \
|
|
echo "UV_EXTRAS entry $extra is invalid (must match [A-Za-z][A-Za-z0-9_-]*)" >&2; \
|
|
exit 1; \
|
|
;; \
|
|
esac; \
|
|
extras_flags="$extras_flags --extra $extra"; \
|
|
done; \
|
|
UV_INDEX_URL=${UV_INDEX_URL:-https://pypi.org/simple} uv sync --extra redis $extras_flags'
|
|
|
|
# UTF-8 locale prevents UnicodeEncodeError on Chinese/emoji content in minimal
|
|
# containers where locale configuration may be missing and the default encoding is not UTF-8.
|
|
ENV LANG=C.UTF-8
|
|
ENV LC_ALL=C.UTF-8
|
|
ENV PYTHONIOENCODING=utf-8
|
|
|
|
# ── Stage 2: Dev ──────────────────────────────────────────────────────────────
|
|
# Retains compiler toolchain from builder so startup-time `uv sync` can build
|
|
# source distributions in development containers.
|
|
FROM builder AS dev
|
|
|
|
# Install Docker CLI (for DooD: allows starting sandbox containers via host Docker socket)
|
|
COPY --from=docker:cli /usr/local/bin/docker /usr/local/bin/docker
|
|
|
|
EXPOSE 8001
|
|
|
|
CMD ["sh", "-c", "cd backend && PYTHONPATH=. uv run uvicorn app.gateway.app:app --host 0.0.0.0 --port 8001"]
|
|
|
|
# ── Stage 3: Runtime ──────────────────────────────────────────────────────────
|
|
# Clean image without build-essential — reduces size (~200 MB) and attack surface.
|
|
FROM python:3.12-slim-bookworm
|
|
|
|
ENV LANG=C.UTF-8
|
|
ENV LC_ALL=C.UTF-8
|
|
ENV PYTHONIOENCODING=utf-8
|
|
|
|
# Copy Node.js runtime from builder (provides npx for MCP servers)
|
|
COPY --from=builder /usr/bin/node /usr/bin/node
|
|
COPY --from=builder /usr/lib/node_modules /usr/lib/node_modules
|
|
RUN ln -s ../lib/node_modules/npm/bin/npm-cli.js /usr/bin/npm \
|
|
&& ln -s ../lib/node_modules/npm/bin/npx-cli.js /usr/bin/npx
|
|
|
|
# Install Docker CLI (for DooD: allows starting sandbox containers via host Docker socket)
|
|
COPY --from=docker:cli /usr/local/bin/docker /usr/local/bin/docker
|
|
|
|
# Install uv (source image overridable via UV_IMAGE build arg)
|
|
COPY --from=uv-source /uv /uvx /usr/local/bin/
|
|
|
|
# Set working directory
|
|
WORKDIR /app
|
|
|
|
# Copy backend with pre-built virtualenv from builder
|
|
COPY --from=builder /app/backend ./backend
|
|
|
|
# Expose Gateway API port.
|
|
EXPOSE 8001
|
|
|
|
# Default command (can be overridden in docker-compose)
|
|
CMD ["sh", "-c", "cd backend && PYTHONPATH=. uv run --no-sync uvicorn app.gateway.app:app --host 0.0.0.0 --port 8001"]
|