mirror of
https://github.com/ThearchyHelios/AgentLab.git
synced 2026-11-15 00:41:59 +00:00
ThearchyHelios
202 lines
10 KiB
Bash
Executable File
202 lines
10 KiB
Bash
Executable File
#!/usr/bin/env bash
|
||
# 一条命令把前后端都拉起来。Ctrl-C 一起退出。
|
||
set -euo pipefail
|
||
|
||
usage() {
|
||
cat <<'USAGE'
|
||
用法：./scripts/dev.sh [--host] [--postgres]
|
||
|
||
--host 前端也在局域网上监听（同一网络里的手机、别的电脑能打开）。后端仍只听 127.0.0.1：
|
||
页面里的 /api 由前端开发服务器在本机转发，局域网里的设备不需要、也不该直连后端。
|
||
注意：AgentLab 没有登录，同一网络里打得开页面的人都能用它——跑工作流、花你配的
|
||
模型额度、读数据源、执行沙箱代码。只在信得过的网络里开。
|
||
--postgres 应用库用 PostgreSQL：起（已经有就复用）一个本地开发用的 PostgreSQL 容器，端口只开在 127.0.0.1，
|
||
后端照它设 AGENTLAB_DATABASE_URL。密码第一次自动生成，在 data/.postgres-dev/ 里；库的数据在 docker
|
||
数据卷里，Ctrl-C 只停前后端，库的容器留着下次复用（不要了：docker rm -f 容器名 && docker volume rm 卷名）。
|
||
不加就和原来一样：SQLite，不用 docker。下面几项可以改缺省（环境变量，或者写在 .env / backend/.env 里，
|
||
环境变量优先）：
|
||
AGENTLAB_DEV_PG_CONTAINER 容器名，缺省 agentlab-dev-postgres
|
||
AGENTLAB_DEV_PG_PORT 本机端口，缺省 55432
|
||
AGENTLAB_DEV_PG_VOLUME 数据卷，缺省 agentlab-dev-pgdata
|
||
AGENTLAB_DEV_PG_IMAGE 镜像，缺省 postgres:16
|
||
USAGE
|
||
}
|
||
|
||
# 参数先认完再干活：以前不读参数，--host 这类写了也被悄悄忽略
|
||
WEB_ARGS=()
|
||
WITH_PG=""
|
||
for arg in "$@"; do
|
||
case "$arg" in
|
||
--host) WEB_ARGS+=(--host) ;;
|
||
--postgres) WITH_PG=1 ;;
|
||
-h|--help) usage; exit 0 ;;
|
||
*) echo "认不出的参数：$arg"; echo; usage; exit 2 ;;
|
||
esac
|
||
done
|
||
|
||
ROOT="$(cd "$(dirname "${BASH_SOURCE[0]}")/.." && pwd)"
|
||
API_PORT="${AGENTLAB_PORT:-8000}"
|
||
WEB_PORT="${AGENTLAB_WEB_PORT:-5273}"
|
||
|
||
CONDA_ENV="${AGENTLAB_CONDA_ENV:-agentlab}"
|
||
|
||
command -v conda >/dev/null || { echo "缺少 conda：https://conda-forge.org/miniforge/"; exit 1; }
|
||
command -v pnpm >/dev/null || { echo "缺少 pnpm：npm i -g pnpm"; exit 1; }
|
||
|
||
# 环境不存在就按 environment.yml 建一个
|
||
if ! conda env list | awk '{print $1}' | grep -qx "$CONDA_ENV"; then
|
||
echo "==> 创建 conda 环境 $CONDA_ENV"
|
||
conda env create -f "$ROOT/environment.yml"
|
||
fi
|
||
PY_BIN="$(conda run -n "$CONDA_ENV" python -c 'import sys; print(sys.executable)' 2>/dev/null)"
|
||
[ -x "$PY_BIN" ] || { echo "conda 环境 $CONDA_ENV 不可用，试试：conda env create -f environment.yml"; exit 1; }
|
||
|
||
echo "==> 后端解释器 $PY_BIN"
|
||
[ -d "$ROOT/frontend/node_modules" ] || (cd "$ROOT/frontend" && pnpm install)
|
||
|
||
|
||
# 端口被别人占着就往后挪。踩过一次坑：8000 被另一个项目占住时，uvicorn
|
||
# 起不来直接退出，而下面的健康检查去探 8000 又被那个项目答应了，脚本以为
|
||
# 后端就绪、继续起前端——于是前端连着别人的服务，界面显示的全是陈年旧数据。
|
||
port_taken() { lsof -nP -iTCP:"$1" -sTCP:LISTEN >/dev/null 2>&1; }
|
||
if [ -z "${AGENTLAB_PORT:-}" ]; then
|
||
while port_taken "$API_PORT"; do
|
||
echo "==> 端口 $API_PORT 已被占用，换 $((API_PORT + 1))"
|
||
API_PORT=$((API_PORT + 1))
|
||
done
|
||
elif port_taken "$API_PORT"; then
|
||
echo "端口 $API_PORT 已被占用（AGENTLAB_PORT 指定的），换一个或停掉占用方"
|
||
lsof -nP -iTCP:"$API_PORT" -sTCP:LISTEN | tail -n +2 | head -3
|
||
exit 1
|
||
fi
|
||
# 前端代理要跟着后端端口走，否则请求会打到占端口的那个服务上
|
||
export AGENTLAB_PORT="$API_PORT"
|
||
|
||
# ---- --postgres：本地开发用的 PostgreSQL ----
|
||
# 这几项（AGENTLAB_DEV_PG_*，还有数据目录 AGENTLAB_DATA_DIR）和后端的配置一样认 .env：环境变量 > backend/.env > .env > 缺省
|
||
# （后端的 settings 读这两个文件，后一个优先）。写法照后端的 dotenv：KEY=value、KEY = value、export KEY=value、引号、行尾 # 注释。
|
||
# 不 source：.env 是给后端按 KEY=VALUE 读的，不是 shell（值里的 $、空格、反引号不该被执行）
|
||
dotenv_value() {
|
||
local key="$1" file line name val="" found=""
|
||
for file in "$ROOT/.env" "$ROOT/backend/.env"; do
|
||
[ -f "$file" ] || continue
|
||
while IFS= read -r line || [ -n "$line" ]; do
|
||
line="${line%$'\r'}"
|
||
line="${line#"${line%%[![:space:]]*}"}"
|
||
line="${line#export }"
|
||
case "$line" in *=*) ;; *) continue ;; esac
|
||
name="${line%%=*}"
|
||
name="${name%"${name##*[![:space:]]}"}"
|
||
[ "$name" = "$key" ] || continue
|
||
val="${line#*=}"
|
||
val="${val#"${val%%[![:space:]]*}"}"
|
||
found=1
|
||
done < "$file"
|
||
done
|
||
[ -n "$found" ] || return 1
|
||
val="${val%"${val##*[![:space:]]}"}"
|
||
case "$val" in
|
||
\"*\"|\'*\') val="${val:1:${#val}-2}" ;;
|
||
*) val="${val%%[[:space:]]#*}"; val="${val%"${val##*[![:space:]]}"}" ;;
|
||
esac
|
||
printf '%s' "$val"
|
||
}
|
||
dotenv_setting() {
|
||
local value
|
||
value="$(printenv "$1" 2>/dev/null || true)"
|
||
[ -n "$value" ] || value="$(dotenv_value "$1" || true)"
|
||
printf '%s' "${value:-$2}"
|
||
}
|
||
if [ -n "$WITH_PG" ]; then
|
||
PG_CONTAINER="$(dotenv_setting AGENTLAB_DEV_PG_CONTAINER agentlab-dev-postgres)"
|
||
PG_PORT="$(dotenv_setting AGENTLAB_DEV_PG_PORT 55432)"
|
||
PG_VOLUME="$(dotenv_setting AGENTLAB_DEV_PG_VOLUME agentlab-dev-pgdata)"
|
||
PG_IMAGE="$(dotenv_setting AGENTLAB_DEV_PG_IMAGE postgres:16)"
|
||
# 密码文件放在后端实际用的数据目录里：和后端一样读 AGENTLAB_DATA_DIR，相对路径按后端的工作目录（backend/）算
|
||
DEV_DATA="$(dotenv_setting AGENTLAB_DATA_DIR "$ROOT/data")"
|
||
case "$DEV_DATA" in /*) ;; *) DEV_DATA="$ROOT/backend/$DEV_DATA" ;; esac
|
||
case "$PG_PORT" in ''|*[!0-9]*) echo "AGENTLAB_DEV_PG_PORT 要是端口号：$PG_PORT"; exit 2 ;; esac
|
||
command -v docker >/dev/null || { echo "--postgres 要用 docker，没找到它"; exit 1; }
|
||
docker version --format '{{.Server.Version}}' >/dev/null 2>&1 || { echo "连不上 docker 服务，先把 Docker 启动起来"; exit 1; }
|
||
# 密码：第一次生成（0600），之后沿用；pgpass 给后端连库用（PGPASSFILE），密码不进命令行和环境变量
|
||
(cd "$ROOT/backend" && "$PY_BIN" -m app.ops postgres-secrets --data "$DEV_DATA" --dir .postgres-dev \
|
||
--host 127.0.0.1 --port "$PG_PORT") >/dev/null
|
||
if docker container inspect "$PG_CONTAINER" >/dev/null 2>&1; then
|
||
role="$(docker inspect -f '{{index .Config.Labels "agentlab.role"}}' "$PG_CONTAINER")"
|
||
[ "$role" = dev-postgres ] || { echo "已经有一个叫 $PG_CONTAINER 的容器，不是这个脚本起的。换个名字：AGENTLAB_DEV_PG_CONTAINER=…"; exit 1; }
|
||
published="$(docker inspect -f '{{range $p, $b := .HostConfig.PortBindings}}{{range $b}}{{.HostPort}}{{end}}{{end}}' "$PG_CONTAINER")"
|
||
[ "$published" = "$PG_PORT" ] || { echo "容器 $PG_CONTAINER 开的是本机端口 ${published}，不是 ${PG_PORT}：AGENTLAB_DEV_PG_PORT=$published 再来，或者删掉容器重建"; exit 1; }
|
||
if [ "$(docker inspect -f '{{.State.Running}}' "$PG_CONTAINER")" != true ]; then
|
||
echo "==> 启动开发用的 PostgreSQL（${PG_CONTAINER}）"
|
||
docker start "$PG_CONTAINER" >/dev/null
|
||
else
|
||
echo "==> 复用开发用的 PostgreSQL（${PG_CONTAINER}）"
|
||
fi
|
||
else
|
||
port_taken "$PG_PORT" && { echo "端口 $PG_PORT 已被占用：换一个 AGENTLAB_DEV_PG_PORT=…"; exit 1; }
|
||
echo "==> 起开发用的 PostgreSQL（${PG_CONTAINER}，${PG_IMAGE}，127.0.0.1:${PG_PORT}，数据卷 ${PG_VOLUME}）"
|
||
docker run -d --name "$PG_CONTAINER" --label agentlab.role=dev-postgres \
|
||
-e POSTGRES_USER=agentlab -e POSTGRES_DB=agentlab -e POSTGRES_PASSWORD_FILE=/run/agentlab/password \
|
||
-v "$PG_VOLUME:/var/lib/postgresql/data" -v "$DEV_DATA/.postgres-dev/password:/run/agentlab/password:ro" \
|
||
-p "127.0.0.1:$PG_PORT:5432" "$PG_IMAGE" >/dev/null
|
||
fi
|
||
# 走 TCP 问：第一次初始化时临时起的那个只听本地套接字，问得通才是正式起好了
|
||
ready=""
|
||
for _ in $(seq 1 60); do
|
||
if docker exec "$PG_CONTAINER" pg_isready -q -h 127.0.0.1 -U agentlab -d agentlab 2>/dev/null; then ready=1; break; fi
|
||
[ "$(docker inspect -f '{{.State.Running}}' "$PG_CONTAINER" 2>/dev/null)" = true ] || break
|
||
sleep 1
|
||
done
|
||
[ -n "$ready" ] || { docker logs --tail 30 "$PG_CONTAINER" 2>&1 || true; echo "开发用的 PostgreSQL 没起来，见上面的日志"; exit 1; }
|
||
export AGENTLAB_DATABASE_URL="postgresql+asyncpg://agentlab@127.0.0.1:$PG_PORT/agentlab"
|
||
export PGPASSFILE="$DEV_DATA/.postgres-dev/pgpass"
|
||
echo "==> 应用库 PostgreSQL 127.0.0.1:$PG_PORT/agentlab（容器 ${PG_CONTAINER}）"
|
||
fi
|
||
|
||
cleanup() {
|
||
echo
|
||
echo "==> 停止"
|
||
# 杀掉整个进程组，避免 uvicorn / vite 变成孤儿进程
|
||
[ -n "${API_PID:-}" ] && kill "$API_PID" 2>/dev/null || true
|
||
[ -n "${WEB_PID:-}" ] && kill "$WEB_PID" 2>/dev/null || true
|
||
wait 2>/dev/null || true
|
||
}
|
||
trap cleanup EXIT INT TERM
|
||
|
||
echo "==> 后端 http://127.0.0.1:$API_PORT"
|
||
# 开发时一个进程、改了代码自动重启（--reload 先停旧的再起新的，SQLite 的单进程守卫不会挡住它）。要试多进程用
|
||
# scripts/prod.sh --workers N（应用库要是 PostgreSQL），或者 scripts/test-multiprocess.sh 跑多进程端到端
|
||
(cd "$ROOT/backend" && "$PY_BIN" -m uvicorn app.main:app --host 127.0.0.1 --port "$API_PORT" --reload) &
|
||
API_PID=$!
|
||
|
||
# 等后端起来再起前端，省得前端第一次请求就打空。
|
||
# 光看"有没有响应"不够——得确认答应的是 agentlab 自己，不是碰巧占着
|
||
# 这个端口的别的服务；顺便看住后端进程别悄悄死了。
|
||
ready=""
|
||
for _ in $(seq 1 60); do
|
||
if ! kill -0 "$API_PID" 2>/dev/null; then
|
||
echo "后端启动失败，上面应该有它的报错"
|
||
exit 1
|
||
fi
|
||
if curl -sf "http://127.0.0.1:$API_PORT/api/health" 2>/dev/null | grep -q '"agentlab"'; then
|
||
ready=1; break
|
||
fi
|
||
sleep 0.5
|
||
done
|
||
[ -n "$ready" ] || { echo "后端 30s 内没就绪（或 $API_PORT 上答话的不是 agentlab）"; exit 1; }
|
||
|
||
echo "==> 前端 http://localhost:$WEB_PORT"
|
||
(cd "$ROOT/frontend" && pnpm dev --port "$WEB_PORT" ${WEB_ARGS[@]+"${WEB_ARGS[@]}"}) &
|
||
WEB_PID=$!
|
||
|
||
echo
|
||
echo " 界面 http://localhost:$WEB_PORT"
|
||
if [ ${#WEB_ARGS[@]} -gt 0 ]; then
|
||
# 本机在局域网里的地址：macOS 取 en0 / en1，Linux 取 hostname -I 的第一个
|
||
LAN_IP="$(ipconfig getifaddr en0 2>/dev/null || ipconfig getifaddr en1 2>/dev/null || hostname -I 2>/dev/null | awk '{print $1}' || true)"
|
||
echo " 局域网 http://${LAN_IP:-<本机 IP>}:$WEB_PORT （没有登录：同一网络里打得开页面的人都能用它）"
|
||
fi
|
||
echo " 接口 http://127.0.0.1:$API_PORT/docs"
|
||
echo
|
||
wait
|