144 lines
6.0 KiB
Bash
144 lines
6.0 KiB
Bash
#!/usr/bin/env bash
|
|
set -euo pipefail
|
|
|
|
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
|
|
|
|
IMAGE_TAG="${IMAGE_TAG:-deerflow-backend-offline:20260505}"
|
|
CONTAINER_NAME="${CONTAINER_NAME:-deerflow-backend}"
|
|
HOST_PORT="${HOST_PORT:-2026}"
|
|
IMAGE_TAR="${IMAGE_TAR:-${SCRIPT_DIR}/deerflow-backend-offline-image-20260505.tar}"
|
|
# Default timezone for the container. The offline image is built from a base
|
|
# layer that ships with UTC, which made scheduled tasks fire at the wrong wall
|
|
# clock and made log timestamps confusing for ops. ``Asia/Shanghai`` keeps the
|
|
# scheduler aligned with the operator's expectation; override with TZ=... when
|
|
# deploying to a different region.
|
|
CONTAINER_TZ="${CONTAINER_TZ:-${TZ:-Asia/Shanghai}}"
|
|
|
|
die() {
|
|
echo "ERROR: $*" >&2
|
|
exit 1
|
|
}
|
|
|
|
require_cmd() {
|
|
command -v "$1" >/dev/null 2>&1 || die "command not found: $1"
|
|
}
|
|
|
|
require_file() {
|
|
[ -f "$1" ] || die "required file not found: $1"
|
|
}
|
|
|
|
require_dir() {
|
|
[ -d "$1" ] || die "required directory not found: $1"
|
|
}
|
|
|
|
require_env_key() {
|
|
local key="$1"
|
|
grep -Eq "^[[:space:]]*${key}=" "${SCRIPT_DIR}/.env" || die "required .env key not found: ${key}"
|
|
}
|
|
|
|
require_cmd docker
|
|
|
|
if [ ! -f "${SCRIPT_DIR}/.env" ]; then
|
|
if [ -f "${SCRIPT_DIR}/.env.example" ]; then
|
|
cp "${SCRIPT_DIR}/.env.example" "${SCRIPT_DIR}/.env"
|
|
echo "Generated .env from .env.example. Please edit .env and run this script again."
|
|
exit 1
|
|
fi
|
|
die "missing .env"
|
|
fi
|
|
|
|
if [ ! -f "${SCRIPT_DIR}/config.yaml" ]; then
|
|
if [ -f "${SCRIPT_DIR}/config.yaml.example" ]; then
|
|
cp "${SCRIPT_DIR}/config.yaml.example" "${SCRIPT_DIR}/config.yaml"
|
|
echo "Generated config.yaml from config.yaml.example. Please edit model/API settings and run this script again."
|
|
exit 1
|
|
fi
|
|
die "missing config.yaml"
|
|
fi
|
|
|
|
require_env_key "AUTH_JWT_SECRET"
|
|
require_env_key "SCHEDULED_TASKS_ENABLED"
|
|
require_env_key "ELEMENT_ENABLED"
|
|
require_file "${SCRIPT_DIR}/config.yaml"
|
|
require_file "${SCRIPT_DIR}/extensions_config.json"
|
|
require_file "${SCRIPT_DIR}/backend/pyproject.toml"
|
|
require_file "${SCRIPT_DIR}/backend/app/gateway/app.py"
|
|
require_file "${SCRIPT_DIR}/backend/packages/harness/deerflow/runtime/scheduler/element.py"
|
|
require_dir "${SCRIPT_DIR}/skills"
|
|
require_dir "${SCRIPT_DIR}/wheelhouse"
|
|
# Knowledge base directory, mounted into the container at /mnt/knowledge_base.
|
|
require_dir "${SCRIPT_DIR}/backend/knowledge/knowledge_base"
|
|
mkdir -p "${SCRIPT_DIR}/data"
|
|
|
|
if ! docker image inspect "${IMAGE_TAG}" >/dev/null 2>&1; then
|
|
require_file "${IMAGE_TAR}"
|
|
echo "Loading Docker image from ${IMAGE_TAR}"
|
|
docker load -i "${IMAGE_TAR}"
|
|
fi
|
|
|
|
if docker ps -a --format '{{.Names}}' | awk -v name="${CONTAINER_NAME}" '$0 == name { found=1 } END { exit(found ? 0 : 1) }'; then
|
|
echo "Removing existing container ${CONTAINER_NAME}"
|
|
docker rm -f "${CONTAINER_NAME}" >/dev/null
|
|
fi
|
|
|
|
echo "Starting backend container ${CONTAINER_NAME}"
|
|
docker run -d \
|
|
--name "${CONTAINER_NAME}" \
|
|
--restart unless-stopped \
|
|
-p "${HOST_PORT}:8001" \
|
|
--env-file "${SCRIPT_DIR}/.env" \
|
|
-e TZ="${CONTAINER_TZ}" \
|
|
-e DEER_FLOW_PROJECT_ROOT=/app \
|
|
-e DEER_FLOW_HOME=/app/backend/.deer-flow \
|
|
-e DEER_FLOW_CONFIG_PATH=/app/config.yaml \
|
|
-e DEER_FLOW_EXTENSIONS_CONFIG_PATH=/app/extensions_config.json \
|
|
-e DEER_FLOW_SKILLS_PATH=/app/skills \
|
|
-e UV_LINK_MODE=copy \
|
|
-v "${SCRIPT_DIR}/backend/app:/app/backend/app" \
|
|
-v "${SCRIPT_DIR}/backend/packages:/app/backend/packages" \
|
|
-v "${SCRIPT_DIR}/backend/scripts:/app/backend/scripts" \
|
|
-v "${SCRIPT_DIR}/backend/pyproject.toml:/app/backend/pyproject.toml:ro" \
|
|
-v "${SCRIPT_DIR}/backend/uv.lock:/app/backend/uv.lock:ro" \
|
|
-v "${SCRIPT_DIR}/backend/langgraph.json:/app/backend/langgraph.json:ro" \
|
|
-v "${SCRIPT_DIR}/config.yaml:/app/config.yaml:ro" \
|
|
-v "${SCRIPT_DIR}/skill_evolution.json:/app/skill_evolution.json:rw" \
|
|
-v "${SCRIPT_DIR}/extensions_config.json:/app/extensions_config.json:rw" \
|
|
-v "${SCRIPT_DIR}/skills:/app/skills:rw" \
|
|
-v "${SCRIPT_DIR}/wheelhouse:/app/wheelhouse:ro" \
|
|
-v "${SCRIPT_DIR}/backend/knowledge/knowledge_base:/mnt/knowledge_base:ro" \
|
|
-v "${SCRIPT_DIR}/data:/app/backend/.deer-flow" \
|
|
"${IMAGE_TAG}" \
|
|
sh -c 'cd backend
|
|
# Apply timezone inside the container. ``TZ`` alone is enough for Python /
|
|
# uvicorn log timestamps, but linking /etc/localtime keeps C-level tools
|
|
# (e.g. anything that reads /etc/localtime directly) on the same clock.
|
|
if [ -n "${TZ:-}" ] && [ -f "/usr/share/zoneinfo/${TZ}" ]; then
|
|
ln -sf "/usr/share/zoneinfo/${TZ}" /etc/localtime 2>/dev/null || true
|
|
echo "${TZ}" > /etc/timezone 2>/dev/null || true
|
|
fi
|
|
if [ "${SCHEDULER_INSTALL_DEPS:-1}" != "0" ]; then
|
|
uv run --no-sync python -c "import croniter" >/dev/null 2>&1 || {
|
|
PYTHON_BIN="$(uv run --no-sync python -c "import sys; print(sys.executable)")"
|
|
uv pip install --python "${PYTHON_BIN}" --no-index --find-links /app/wheelhouse ${SCHEDULER_PIP_PACKAGES:-croniter==6.2.2}
|
|
}
|
|
fi
|
|
if [ "${HINDSIGHT_INSTALL_DEPS:-1}" != "0" ]; then
|
|
uv run --no-sync python -c "import hindsight_client" >/dev/null 2>&1 || {
|
|
PYTHON_BIN="$(uv run --no-sync python -c "import sys; print(sys.executable)")"
|
|
uv pip install --python "${PYTHON_BIN}" --no-index --find-links /app/wheelhouse ${HINDSIGHT_INSTALL_DEPS:-hindsight_client==0.6.2}
|
|
}
|
|
fi
|
|
if [ "${PG_CHECKPOINTER_INSTALL_DEPS:-1}" != "0" ]; then
|
|
uv run --no-sync python -c "import psycopg; import psycopg_pool; import langgraph.checkpoint.postgres; import langgraph.store.postgres" >/dev/null 2>&1 || {
|
|
PYTHON_BIN="$(uv run --no-sync python -c "import sys; print(sys.executable)")"
|
|
uv pip install --python "${PYTHON_BIN}" --no-deps --no-index --find-links /app/wheelhouse ${PG_CHECKPOINTER_PIP_PACKAGES:-langgraph-checkpoint-postgres==3.0.5 psycopg==3.3.3 psycopg-binary==3.3.3 psycopg-pool==3.3.0 orjson==3.11.9 tzdata==2026.2}
|
|
}
|
|
fi
|
|
PYTHONPATH=. uv run --no-sync python -m uvicorn app.gateway.app:app --host 0.0.0.0 --port 8001 --workers ${GATEWAY_WORKERS:-8}'
|
|
|
|
echo
|
|
echo "Backend started."
|
|
echo " URL: http://<server-ip>:${HOST_PORT}"
|
|
echo " Logs: docker logs -f ${CONTAINER_NAME}"
|
|
echo " Stop: ./stop_backend.sh"
|