mirror of
https://github.com/open-webui/open-webui.git
synced 2026-06-13 19:20:05 +00:00
Classic298
0cd1529008
start.sh runs with `set -euo pipefail`, but three call sites added in
070ab2650 (refac: reorganize scripts and ci workflows) reference
optional env vars via bash's `,,` lowercase expansion without any
default. Containers that don't set these vars — the default for every
deployment that isn't explicitly opting into Playwright / bundled
Ollama / CUDA — crash on startup with:
start.sh: line 15: WEB_LOADER_ENGINE: unbound variable
(and the same for USE_OLLAMA_DOCKER, USE_CUDA_DOCKER once the first
were set in turn.) Reported in open-webui#24560 by urbenlegend.
The same refactor correctly defaulted every other optional env var
with `${VAR:-…}`. The three `,,` references slipped through because
bash can't combine `:-default` with `,,` in a single substitution —
`${VAR:-default,,}` makes the default literal `,,`, not what's wanted.
Fix: normalise the three vars in a one-line preamble with `${VAR:=}`,
which assigns an empty default if unset. The downstream `${VAR,,}`
expressions stay exactly as Tim wrote them, preserving the file's
visual style and matching the existing `${VAR:-…}` idiom for "this
variable is optional".
108 lines
4.4 KiB
Bash
Executable File
108 lines
4.4 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.
|
|
: "${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 [[ "${WEB_LOADER_ENGINE,,}" == "playwright" ]]; then
|
|
if [[ -z "${PLAYWRIGHT_WS_URL:-}" ]]; then
|
|
echo "Installing Playwright Chromium browser..."
|
|
playwright install chromium
|
|
playwright install-deps chromium
|
|
fi
|
|
python -c "import nltk; nltk.download('punkt_tab')"
|
|
fi
|
|
|
|
# ── Secret key setup ─────────────────────────────────────────────────────────
|
|
|
|
KEY_FILE="${WEBUI_SECRET_KEY_FILE:-.webui_secret_key}"
|
|
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..."
|
|
head -c 12 /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.11/site-packages/torch/lib:/usr/local/lib/python3.11/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:-*}" &
|
|
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")
|
|
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[@]}" |