Update installpi.sh

installpi.sh

@@ -2,20 +2,22 @@
 set -euo pipefail
 
 # ==============================================================================
-# installpi.sh - 一键全自动安装 pi + llama.cpp (源码编译) + Huihui 模型
+# installpi.sh - 一键全自动安装 pi + llama.cpp (源码编译) + Qwen2.5 模型
 # 特性：无 sudo、无互动、全自动、可断点续传、安装后自动验证
-# 模型：mradermacher/Huihui-Qwen3.5-4B-Claude-4.6-Opus-abliterated-GGUF (Q4_K_M.gguf)
+# 模型：unsloth/GLM-4.7-Flash-GGUF (GLM-4.7-Flash-Q4_K_M.gguf)
 # 启动：taskset -c 0-14 llama-server ... -t 15 --jinja --load-mode mlock -c 65536 (64K上下文)
 # ==============================================================================
 
 HOME_DIR="$HOME"
+# 脚本一开始就把之前安装的 node / pi / llama 路径放进 PATH，避免每次重复装 node
+export PATH="${XDG_DATA_HOME:-$HOME/.local/share}/pi-node/current/bin:$HOME/.local/bin:$HOME/.pi/bin:$PATH"
 LLAMA_SRC_DIR="$HOME_DIR/llama.cpp"          # 源码目录
 LLAMA_BIN="$LLAMA_SRC_DIR/build/bin"         # 编译后二进制目录
 MODEL_DIR="$HOME_DIR/models"
 PI_MODELS_JSON="$HOME_DIR/.pi/agent/models.json"
 LLAMA_PORT=8080
 REPO="unsloth/GLM-4.7-Flash-GGUF"
-MODEL_FILE="https://huggingface.co/unsloth/GLM-4.7-Flash-GGUF/resolve/main/GLM-4.7-Flash-Q4_K_M.gguf"
+MODEL_FILE="GLM-4.7-Flash-Q4_K_M.gguf"
 MODEL_ID="glm-4.7-flash-q4_k_m"
 PROVIDER_ID="llama-cpp-local"
 CONTEXT_SIZE=65536                          # 上下文 64K
@@ -107,20 +109,27 @@ info "npm 版本: $(npm --version)"
 # ==============================================================================
 info "1) 编译 llama.cpp (源码)..."
 
+if [[ "${LLAMA_FORCE_REBUILD:-0}" == "1" ]]; then
+    info "LLAMA_FORCE_REBUILD=1，清理旧 build 目录强制重编..."
+    rm -rf "$LLAMA_SRC_DIR/build"
+fi
+
 if [[ ! -x "$LLAMA_BIN/llama-server" ]]; then
     info "克隆 llama.cpp 仓库 (深度1)..."
     if [[ -d "$LLAMA_SRC_DIR/.git" ]]; then
         info "源码目录已存在，更新..."
         cd "$LLAMA_SRC_DIR"
-        git pull
+        git pull || warn "git pull 失败，继续使用现有源码"
     else
         git clone --depth 1 https://github.com/ggml-org/llama.cpp "$LLAMA_SRC_DIR"
     fi
 
     cd "$LLAMA_SRC_DIR"
-    info "执行 cmake 配置..."
+    info "执行 cmake 配置 (纯 CPU，不开 NATIVE/DL，避免 MoE/多变体后端在纯 CPU 上崩)..."
+    rm -rf build
     cmake -B build \
-        -DGGML_NATIVE=ON \
+        -DGGML_NATIVE=OFF \
+        -DGGML_CPU=ON \
         -DGGML_BACKEND_DL=OFF \
         -DLLAMA_BUILD_TESTS=OFF \
         -DCMAKE_BUILD_TYPE=Release
@@ -135,7 +144,7 @@ if [[ ! -x "$LLAMA_BIN/llama-server" ]]; then
         error "编译失败，找不到 llama-server"
     fi
 else
-    ok "llama-server 已存在，跳过编译"
+    ok "llama-server 已存在，跳过编译 (如需重编: LLAMA_FORCE_REBUILD=1 ./installpi.sh)"
 fi
 
 # 更新 PATH
@@ -153,7 +162,7 @@ done
 info "2) 下载模型 ${REPO} -> ${MODEL_FILE} ..."
 
 MODEL_URL="https://huggingface.co/${REPO}/resolve/main/${MODEL_FILE}"
-MODEL_PATH="$MODEL_DIR/$MODEL_FILE"
+MODEL_PATH="$MODEL_DIR/${MODEL_FILE}"
 
 if [[ ! -f "$MODEL_PATH" ]]; then
     info "开始下载模型 (支持断点续传)..."
@@ -169,7 +178,7 @@ else
 fi
 
 # ==============================================================================
-# 3. 启动 llama-server (使用 taskset 和指定参数，上下文调至极限 128K)
+# 3. 启动 llama-server (使用 taskset 和指定参数，上下文 64K)
 # ==============================================================================
 info "3) 启动 llama-server (port $LLAMA_PORT, taskset 0-14, context $CONTEXT_SIZE)..."
 
@@ -186,20 +195,38 @@ fi
 pkill -f "llama-server.*${LLAMA_PORT}" 2>/dev/null || true
 sleep 1
 
-# 构建命令（注意 taskset 包裹整个 llama-server，prlimit 限制内存 32GB）
+# 按实际 CPU 数决定 taskset 范围 / 线程数 (避免机器核数不足时 taskset 报错)
+NPROC=$(nproc 2>/dev/null || echo 16)
+if [[ "$NPROC" -ge 15 ]]; then
+    TASKSET_ARGS=(-c 0-14)
+    NTHREADS=15
+else
+    TASKSET_ARGS=(-c "0-$((NPROC-1))")
+    NTHREADS="$NPROC"
+fi
+if ! taskset "${TASKSET_ARGS[@]}" true 2>/dev/null; then
+    warn "taskset ${TASKSET_ARGS[*]} 不可用，改用不绑核启动"
+    TASKSET_ARGS=()
+fi
+
+# 构建命令（prlimit 限制内存 32GB；LD_LIBRARY_PATH 已在前面 export，prlimit 会继承）
 CMD=(
-    taskset -c 0-14
+    taskset "${TASKSET_ARGS[@]}"
     prlimit --as=34359738368
     "$LLAMA_BIN/llama-server"
     -m "$MODEL_PATH"
     --alias "$MODEL_ID"
-    -t 15
+    -t "$NTHREADS"
     --jinja
     --load-mode mlock
     -c "$CONTEXT_SIZE"
     --host 127.0.0.1
     --port "$LLAMA_PORT"
 )
+# taskset 参数为空时去掉 taskset 本体
+if [[ "${#TASKSET_ARGS[@]}" -eq 0 ]]; then
+    CMD=(prlimit --as=34359738368 "$LLAMA_BIN/llama-server" -m "$MODEL_PATH" --alias "$MODEL_ID" -t "$NTHREADS" --jinja --load-mode mlock -c "$CONTEXT_SIZE" --host 127.0.0.1 --port "$LLAMA_PORT")
+fi
 
 info "启动命令: ${CMD[*]}"
 touch "$HOME_DIR/llama-server.log" 2>/dev/null || true
@@ -333,7 +360,7 @@ cfg["providers"][provider] = {
     "compat": {"supportsDeveloperRole": True, "supportsReasoningEffort": False},
     "models": [{
         "id": model_id,
-        "name": "Huihui Qwen3.5 4B Claude 4.6 Opus (Local)",
+        "name": "GLM-4.7-Flash Q4_K_M (Local)",
         "reasoning": True,
         "input": ["text"],
         "contextWindow": ctx_size,
@@ -465,7 +492,7 @@ else
     echo -e "\033[33m⚠️  安装完成，但部分验证未通过，请查看上方警告\033[0m"
 fi
 echo "================================================================"
-echo "- 模型: $REPO"
+echo "- 模型: $REPO ($MODEL_FILE)"
 echo "  模型文件: $MODEL_PATH"
 echo "- 内存限制: 32GB"
 echo "- llama-server (编译自源码): $LLAMA_BIN/llama-server"