mirror of
https://github.com/open-webui/open-webui.git
synced 2026-10-23 03:43:27 -05:00
Classic298
a915df8ce1
The Docker image now runs Python 3.12 on Debian 13 (trixie) instead of Python 3.11 on Debian 12, and the pip install docs recommend 3.12. Image size stays about the same: the slim image is 1 MB smaller and the standard image is about 50 MB larger, mostly from the newer ffmpeg and pandoc that Debian 13 ships. Admins should know that custom functions, tools and anything installed with pip inside the container now run on Python 3.12, so a package that only has builds for 3.11 needs a newer version. Installing with pip still works on both Python 3.11 and 3.12.
113 lines
4.7 KiB
Bash
Executable File
113 lines
4.7 KiB
Bash
Executable File
#!/usr/bin/env bash
|
|
set -euo pipefail
|
|
|
|
# ---------------------------------------------------------------------------
|
|
# Container entry point for Open WebUI.
|
|
# Handles secret key generation, optional Ollama/CUDA/Playwright setup,
|
|
# HuggingFace Space deployment, and launches the uvicorn server.
|
|
# ---------------------------------------------------------------------------
|
|
|
|
# Default optional env vars that we test below with bash's `,,` lowercase
|
|
# expansion. The two can't be combined inline (`${VAR:-default,,}` makes
|
|
# the default literal `,,`), so we normalise once up front and the simple
|
|
# `${VAR,,}` form stays safe under `set -u` everywhere else.
|
|
: "${USE_SLIM_DOCKER:=}" "${WEB_LOADER_ENGINE:=}" "${USE_OLLAMA_DOCKER:=}" "${USE_CUDA_DOCKER:=}"
|
|
|
|
SCRIPT_DIR=$(cd -- "$(dirname -- "${BASH_SOURCE[0]}")" &>/dev/null && pwd)
|
|
cd "$SCRIPT_DIR" || exit 1
|
|
|
|
# ── Playwright browser installation (if configured) ──────────────────────────
|
|
|
|
if [[ "${USE_SLIM_DOCKER,,}" != "true" && "${WEB_LOADER_ENGINE,,}" == "playwright" ]]; then
|
|
if [[ -z "${PLAYWRIGHT_WS_URL:-}" ]]; then
|
|
echo "Installing Playwright Chromium browser..."
|
|
playwright install chromium
|
|
playwright install-deps chromium
|
|
fi
|
|
fi
|
|
|
|
# ── Secret key setup ─────────────────────────────────────────────────────────
|
|
|
|
KEY_FILE="${WEBUI_SECRET_KEY_FILE:-.webui_secret_key}"
|
|
WEBUI_SECRET_KEY_LENGTH="${WEBUI_SECRET_KEY_LENGTH:-24}"
|
|
PORT="${PORT:-8080}"
|
|
HOST="${HOST:-0.0.0.0}"
|
|
|
|
if [[ -z "${WEBUI_SECRET_KEY:-}" && -z "${WEBUI_JWT_SECRET_KEY:-}" ]]; then
|
|
echo "No WEBUI_SECRET_KEY environment variable set, loading from file."
|
|
|
|
if [[ ! -f "$KEY_FILE" ]]; then
|
|
echo "Generating new WEBUI_SECRET_KEY..."
|
|
if ! [[ "$WEBUI_SECRET_KEY_LENGTH" =~ ^[1-9][0-9]*$ ]]; then
|
|
echo "WEBUI_SECRET_KEY_LENGTH must be a positive integer." >&2
|
|
exit 1
|
|
fi
|
|
head -c "$WEBUI_SECRET_KEY_LENGTH" /dev/random | base64 > "$KEY_FILE"
|
|
fi
|
|
|
|
echo "Loading WEBUI_SECRET_KEY from ${KEY_FILE}"
|
|
WEBUI_SECRET_KEY=$(cat "$KEY_FILE")
|
|
fi
|
|
|
|
# ── Ollama (bundled Docker image) ────────────────────────────────────────────
|
|
|
|
if [[ "${USE_OLLAMA_DOCKER,,}" == "true" ]]; then
|
|
echo "Starting bundled ollama serve..."
|
|
ollama serve &
|
|
fi
|
|
|
|
# ── CUDA library paths ──────────────────────────────────────────────────────
|
|
|
|
if [[ "${USE_CUDA_DOCKER,,}" == "true" ]]; then
|
|
echo "CUDA enabled — extending LD_LIBRARY_PATH for torch/cudnn libraries."
|
|
export LD_LIBRARY_PATH="${LD_LIBRARY_PATH:-}:/usr/local/lib/python3.12/site-packages/torch/lib:/usr/local/lib/python3.12/site-packages/nvidia/cudnn/lib"
|
|
fi
|
|
|
|
# ── HuggingFace Space deployment ─────────────────────────────────────────────
|
|
|
|
if [[ -n "${SPACE_ID:-}" ]]; then
|
|
echo "Configuring for HuggingFace Space deployment..."
|
|
|
|
if [[ -n "${ADMIN_USER_EMAIL:-}" && -n "${ADMIN_USER_PASSWORD:-}" ]]; then
|
|
echo "Creating admin user for Space..."
|
|
WEBUI_SECRET_KEY="${WEBUI_SECRET_KEY:-}" \
|
|
uvicorn open_webui.main:app --host "$HOST" --port "$PORT" --forwarded-allow-ips "${FORWARDED_ALLOW_IPS:-*}" --ws-per-message-deflate "${UVICORN_WS_PER_MESSAGE_DEFLATE:-true}" &
|
|
webui_pid=$!
|
|
|
|
echo "Waiting for server to become healthy..."
|
|
until curl -sf "http://localhost:${PORT}/health" > /dev/null 2>&1; do
|
|
sleep 1
|
|
done
|
|
|
|
echo "Registering admin user..."
|
|
curl -sS -X POST "http://localhost:${PORT}/api/v1/auths/signup" \
|
|
-H "Accept: application/json" \
|
|
-H "Content-Type: application/json" \
|
|
-d "{\"email\": \"${ADMIN_USER_EMAIL}\", \"password\": \"${ADMIN_USER_PASSWORD}\", \"name\": \"Admin\"}"
|
|
|
|
echo "Restarting server..."
|
|
kill "$webui_pid"
|
|
wait "$webui_pid" 2>/dev/null || true
|
|
fi
|
|
|
|
export WEBUI_URL="${SPACE_HOST}"
|
|
fi
|
|
|
|
# ── Launch uvicorn ───────────────────────────────────────────────────────────
|
|
|
|
PYTHON_CMD=$(command -v python3 || command -v python)
|
|
UVICORN_WORKERS="${UVICORN_WORKERS:-1}"
|
|
|
|
if [[ "$#" -gt 0 ]]; then
|
|
ARGS=("$@")
|
|
else
|
|
ARGS=(--workers "$UVICORN_WORKERS" --ws-per-message-deflate "${UVICORN_WS_PER_MESSAGE_DEFLATE:-true}")
|
|
fi
|
|
|
|
exec env WEBUI_SECRET_KEY="${WEBUI_SECRET_KEY:-}" \
|
|
"$PYTHON_CMD" -m uvicorn open_webui.main:app \
|
|
--host "$HOST" \
|
|
--port "$PORT" \
|
|
--forwarded-allow-ips "${FORWARDED_ALLOW_IPS:-*}" \
|
|
"${ARGS[@]}"
|