mirror of
https://github.com/TencentCloud/TencentDB-Agent-Memory
synced 2026-07-13 13:54:29 +00:00
234 lines
7.8 KiB
Bash
234 lines
7.8 KiB
Bash
|
#!/usr/bin/env bash
|
|||
|
|
#
|
|||
|
|
# sync-hermes-llm-to-tdai.sh — 读取系统 Hermes 模型配置，同步到 TDAI Gateway
|
|||
|
|
#
|
|||
|
|
# 前置条件：
|
|||
|
|
# 1. memory-tencentdb-ctl.sh 存在
|
|||
|
|
# 2. ~/.hermes/config.yaml 存在且包含模型配置
|
|||
|
|
#
|
|||
|
|
# 用法：
|
|||
|
|
# bash sync-hermes-llm-to-tdai.sh [--restart] [--dry-run] [--hermes]
|
|||
|
|
#
|
|||
|
|
# 参数语义：
|
|||
|
|
# --restart 仅当 CTL 脚本的 `status` 显示 Gateway 当前为 RUNNING 时，
|
|||
|
|
# 才透传给 `memory-tencentdb-ctl.sh config llm` 执行重启；
|
|||
|
|
# 若当前未运行，则本脚本只同步配置，不拉起新进程。
|
|||
|
|
# --hermes 以 hermes 集成模式调用 CTL（影响 status 判定与配置落点）
|
|||
|
|
# --standalone 以 standalone 模式调用 CTL
|
|||
|
|
# --dry-run 透传给 CTL，预演写配置但不真正落盘
|
|||
|
|
#
|
|||
|
|
# 可选环境变量：
|
|||
|
|
# HERMES_HOME hermes 主目录（默认 ~/.hermes）
|
|||
|
|
# MEMORY_TENCENTDB_ROOT TDAI 统一根目录（默认 ~/.memory-tencentdb）
|
|||
|
|
#
|
|||
|
|
|
|||
|
|
set -euo pipefail
|
|||
|
|
|
|||
|
|
SCRIPT_NAME="sync-hermes-llm-to-tdai"
|
|||
|
|
USER_HOME="${HOME:-$(eval echo "~$(whoami)")}"
|
|||
|
|
|
|||
|
|
# ============================================================
|
|||
|
|
# 路径
|
|||
|
|
# ============================================================
|
|||
|
|
|
|||
|
|
HERMES_HOME="${HERMES_HOME:-$USER_HOME/.hermes}"
|
|||
|
|
HERMES_CONFIG="$HERMES_HOME/config.yaml"
|
|||
|
|
|
|||
|
|
MEMORY_TENCENTDB_ROOT="${MEMORY_TENCENTDB_ROOT:-$USER_HOME/.memory-tencentdb}"
|
|||
|
|
TDAI_INSTALL_DIR="${TDAI_INSTALL_DIR:-$MEMORY_TENCENTDB_ROOT/tdai-memory-openclaw-plugin}"
|
|||
|
|
|
|||
|
|
CTL_SCRIPT="$TDAI_INSTALL_DIR/scripts/memory-tencentdb-ctl.sh"
|
|||
|
|
|
|||
|
|
# ============================================================
|
|||
|
|
# 通用 helpers
|
|||
|
|
# ============================================================
|
|||
|
|
|
|||
|
|
log() { printf '[%s] %s\n' "$SCRIPT_NAME" "$*"; }
|
|||
|
|
warn() { printf '[%s:warn] %s\n' "$SCRIPT_NAME" "$*" >&2; }
|
|||
|
|
die() { printf '[%s:error] %s\n' "$SCRIPT_NAME" "$*" >&2; exit "${2:-1}"; }
|
|||
|
|
|
|||
|
|
should_restart_gateway() {
|
|||
|
|
local ctl_mode_args=()
|
|||
|
|
case "${SYNC_MODE:-standalone}" in
|
|||
|
|
hermes) ctl_mode_args+=("--hermes") ;;
|
|||
|
|
*) ctl_mode_args+=("--standalone") ;;
|
|||
|
|
esac
|
|||
|
|
|
|||
|
|
local status_out
|
|||
|
|
status_out="$(bash "$CTL_SCRIPT" "${ctl_mode_args[@]}" status 2>&1 || true)"
|
|||
|
|
|
|||
|
|
if printf '%s\n' "$status_out" | grep -Eq 'state[[:space:]]*:[[:space:]]*RUNNING'; then
|
|||
|
|
log "通过 ctl status 检测到 Gateway 正在运行，允许执行 --restart"
|
|||
|
|
return 0
|
|||
|
|
fi
|
|||
|
|
|
|||
|
|
warn "ctl status 未显示 Gateway 处于 RUNNING；本次仅同步配置，不执行 --restart，也不会拉起新进程。"
|
|||
|
|
return 1
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
# ============================================================
|
|||
|
|
# Step 1: 检查 memory-tencentdb-ctl.sh 是否存在
|
|||
|
|
# ============================================================
|
|||
|
|
|
|||
|
|
if [[ ! -f "$CTL_SCRIPT" ]]; then
|
|||
|
|
# 也尝试旧路径
|
|||
|
|
_LEGACY_CTL="$USER_HOME/tdai-memory-openclaw-plugin/scripts/memory-tencentdb-ctl.sh"
|
|||
|
|
if [[ -f "$_LEGACY_CTL" ]]; then
|
|||
|
|
CTL_SCRIPT="$_LEGACY_CTL"
|
|||
|
|
warn "使用旧路径: $CTL_SCRIPT"
|
|||
|
|
else
|
|||
|
|
die "memory-tencentdb-ctl.sh 不存在，TDAI 未安装。退出。" 1
|
|||
|
|
fi
|
|||
|
|
fi
|
|||
|
|
|
|||
|
|
log "找到 ctl 脚本: $CTL_SCRIPT"
|
|||
|
|
|
|||
|
|
# ============================================================
|
|||
|
|
# Step 2: 检查 hermes config.yaml
|
|||
|
|
# ============================================================
|
|||
|
|
|
|||
|
|
if [[ ! -f "$HERMES_CONFIG" ]]; then
|
|||
|
|
die "Hermes 配置文件不存在: $HERMES_CONFIG" 1
|
|||
|
|
fi
|
|||
|
|
|
|||
|
|
log "读取 Hermes 配置: $HERMES_CONFIG"
|
|||
|
|
|
|||
|
|
# ============================================================
|
|||
|
|
# Step 3: 从 hermes config.yaml 提取模型配置
|
|||
|
|
# ============================================================
|
|||
|
|
|
|||
|
|
need_cmd() {
|
|||
|
|
command -v "$1" >/dev/null 2>&1 || die "必需命令未找到: $1" 127
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
need_cmd python3
|
|||
|
|
|
|||
|
|
# 使用 python3 安全解析 YAML，提取 model 段的 default / base_url / api_key
|
|||
|
|
read_hermes_model_config() {
|
|||
|
|
python3 - "$HERMES_CONFIG" <<'PYEOF'
|
|||
|
|
import sys, json
|
|||
|
|
|
|||
|
|
config_path = sys.argv[1]
|
|||
|
|
|
|||
|
|
# 尝试使用 PyYAML；如果不存在则手动逐行解析
|
|||
|
|
try:
|
|||
|
|
import yaml
|
|||
|
|
with open(config_path, "r", encoding="utf-8") as f:
|
|||
|
|
cfg = yaml.safe_load(f) or {}
|
|||
|
|
model_section = cfg.get("model", {})
|
|||
|
|
result = {
|
|||
|
|
"model": model_section.get("default", ""),
|
|||
|
|
"base_url": model_section.get("base_url", ""),
|
|||
|
|
"api_key": model_section.get("api_key", ""),
|
|||
|
|
}
|
|||
|
|
print(json.dumps(result))
|
|||
|
|
except ImportError:
|
|||
|
|
# PyYAML 不可用，用简单逐行解析提取 model 段
|
|||
|
|
import re
|
|||
|
|
with open(config_path, "r", encoding="utf-8") as f:
|
|||
|
|
lines = f.readlines()
|
|||
|
|
|
|||
|
|
in_model = False
|
|||
|
|
result = {"model": "", "base_url": "", "api_key": ""}
|
|||
|
|
for line in lines:
|
|||
|
|
stripped = line.rstrip("\n")
|
|||
|
|
# 检测 model: 顶层段
|
|||
|
|
if re.match(r"^model:\s*$", stripped) or re.match(r"^model:\s*#", stripped):
|
|||
|
|
in_model = True
|
|||
|
|
continue
|
|||
|
|
# 进入新的顶层段则退出
|
|||
|
|
if in_model and re.match(r"^[A-Za-z_]", stripped):
|
|||
|
|
break
|
|||
|
|
if in_model:
|
|||
|
|
m = re.match(r"^\s+default:\s*(.+)", stripped)
|
|||
|
|
if m:
|
|||
|
|
result["model"] = m.group(1).strip().strip("'\"")
|
|||
|
|
m = re.match(r"^\s+base_url:\s*(.+)", stripped)
|
|||
|
|
if m:
|
|||
|
|
result["base_url"] = m.group(1).strip().strip("'\"")
|
|||
|
|
m = re.match(r"^\s+api_key:\s*(.+)", stripped)
|
|||
|
|
if m:
|
|||
|
|
result["api_key"] = m.group(1).strip().strip("'\"")
|
|||
|
|
print(json.dumps(result))
|
|||
|
|
PYEOF
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
MODEL_JSON="$(read_hermes_model_config)"
|
|||
|
|
|
|||
|
|
HERMES_MODEL="$(printf '%s' "$MODEL_JSON" | python3 -c 'import json,sys; print(json.load(sys.stdin)["model"])')"
|
|||
|
|
HERMES_BASE_URL="$(printf '%s' "$MODEL_JSON" | python3 -c 'import json,sys; print(json.load(sys.stdin)["base_url"])')"
|
|||
|
|
HERMES_API_KEY="$(printf '%s' "$MODEL_JSON" | python3 -c 'import json,sys; print(json.load(sys.stdin)["api_key"])')"
|
|||
|
|
|
|||
|
|
# 校验必需字段
|
|||
|
|
if [[ -z "$HERMES_MODEL" ]]; then
|
|||
|
|
die "Hermes 配置中 model.default 为空" 1
|
|||
|
|
fi
|
|||
|
|
if [[ -z "$HERMES_BASE_URL" ]]; then
|
|||
|
|
die "Hermes 配置中 model.base_url 为空" 1
|
|||
|
|
fi
|
|||
|
|
if [[ -z "$HERMES_API_KEY" ]]; then
|
|||
|
|
die "Hermes 配置中 model.api_key 为空" 1
|
|||
|
|
fi
|
|||
|
|
|
|||
|
|
log "Hermes 模型配置:"
|
|||
|
|
log " model = $HERMES_MODEL"
|
|||
|
|
log " base_url = $HERMES_BASE_URL"
|
|||
|
|
log " api_key = <${#HERMES_API_KEY} chars>"
|
|||
|
|
|
|||
|
|
# ============================================================
|
|||
|
|
# Step 4: 透传参数（--restart / --dry-run / --hermes）
|
|||
|
|
# ============================================================
|
|||
|
|
|
|||
|
|
CTL_EXTRA_ARGS=()
|
|||
|
|
SYNC_MODE="standalone"
|
|||
|
|
REQUEST_RESTART=0
|
|||
|
|
for arg in "$@"; do
|
|||
|
|
case "$arg" in
|
|||
|
|
--restart)
|
|||
|
|
REQUEST_RESTART=1
|
|||
|
|
;;
|
|||
|
|
--hermes)
|
|||
|
|
SYNC_MODE="hermes"
|
|||
|
|
CTL_EXTRA_ARGS+=("$arg")
|
|||
|
|
;;
|
|||
|
|
--standalone)
|
|||
|
|
SYNC_MODE="standalone"
|
|||
|
|
CTL_EXTRA_ARGS+=("$arg")
|
|||
|
|
;;
|
|||
|
|
--dry-run)
|
|||
|
|
CTL_EXTRA_ARGS+=("$arg")
|
|||
|
|
;;
|
|||
|
|
*)
|
|||
|
|
warn "忽略未知参数: $arg"
|
|||
|
|
;;
|
|||
|
|
esac
|
|||
|
|
done
|
|||
|
|
|
|||
|
|
if [[ $REQUEST_RESTART -eq 1 ]]; then
|
|||
|
|
# 新语义：--restart 只在 Gateway 当前已运行时才生效。
|
|||
|
|
# 这里复用 CTL 的 status 输出作为唯一事实来源，避免本脚本
|
|||
|
|
# 自己维护另一套“端口监听/健康检查”判定逻辑。
|
|||
|
|
if should_restart_gateway; then
|
|||
|
|
CTL_EXTRA_ARGS+=("--restart")
|
|||
|
|
else
|
|||
|
|
log "已按新逻辑跳过 --restart。"
|
|||
|
|
fi
|
|||
|
|
fi
|
|||
|
|
|
|||
|
|
# ============================================================
|
|||
|
|
# Step 5: 调用 memory-tencentdb-ctl.sh config llm 同步配置
|
|||
|
|
# ============================================================
|
|||
|
|
|
|||
|
|
log "调用 memory-tencentdb-ctl.sh config llm 同步配置到 TDAI ..."
|
|||
|
|
|
|||
|
|
bash "$CTL_SCRIPT" "${CTL_EXTRA_ARGS[@]+"${CTL_EXTRA_ARGS[@]}"}" \
|
|||
|
|
config llm \
|
|||
|
|
--api-key "$HERMES_API_KEY" \
|
|||
|
|
--base-url "$HERMES_BASE_URL" \
|
|||
|
|
--model "$HERMES_MODEL"
|
|||
|
|
|
|||
|
|
log "同步完成。"
|
|||
|
|
log ""
|
|||
|
|
log "如需立即重启 Gateway 使配置生效，请重新运行并追加 --restart："
|
|||
|
|
log " bash $0 --restart"
|