pi-web-ui 0.91.0 → 0.92.0

package/dist/server/conversation-touches.js

@@ -0,0 +1,357 @@
+// ---------------------------------------------------------------------------
+// conversation-touches.ts — 会话「触碰文件集」提取（纯函数，零 node 依赖）
+// ---------------------------------------------------------------------------
+// 背景：同项目多对话并行时，服务端完全知道「谁碰过哪些文件」（实时消息与落盘
+// 转录里都有工具调用记录），但既没算也没给 AI，导致 AI 只能靠猜冲突。本模块
+// 只实现一次触碰集解析器，三处复用：并行提醒（agent-service.ts）、
+// conversation_read 的 action:"files" 与 action:"status"。
+//
+// 输入形状（两条路都已核实，见下「数据来源」）：
+//   - 实时路：session.agent.state.messages（SDK AgentMessage），assistant 消息
+//     的 content 里是 {type:"toolCall", id, name, arguments} 块；
+//   - 历史路：转录 jsonl 的 message.content，原样保留同一形状（parseTranscript
+//     Lines 透传 content）。
+// 注意：textOfContent 只在「渲染文本」里丢了参数（仍渲染 `[tool call: name]`），
+// content 结构里的 arguments 一直都在。所以 toolCalls 字段是 additive 的便捷
+// 访问点（一处解析、多处复用，并行 toolCall 也能表示），不是新数据源；取不到
+// 时回落扫 content，再取不到就给空集 —— 绝不伪造。
+//
+// 提取规则（只算写，不算读 —— 这是准确性的关键）：
+//   写工具白名单：edit / write / edit_soft（参数 path；兼容 file_path/filePath
+//   别名）。白名单之外的一切工具调用一律忽略，所以 read/ls/glob/grep 等读操作
+//   天然不算 —— 不需要也不维护「读工具黑名单」。
+//   bash 高置信度写入迹象（拿不准就丢，宁可漏报不要假阳性）：
+//     - 重定向 `>` / `>>` 的目标词（含 `2> file` 这类 fd 前缀；排除 `>&N`、
+//       /dev/null、NUL、&N；heredoc `<<`/`<<<` 不算写入）；
+//     - rm / rmdir / mkdir / touch 的非 flag 参数；
+//     - cp / mv 只取最后一个非 flag 参数（目标端；源端是读/被删，不算写）。
+//   解析是轻量分词器（引号/转义感知，`;`/`&&`/`||`/`|`/`&`/`(`/`)` 切分命令
+//   段），不是完整 shell 解析器。
+//
+// 已知局限（如实记录，调用方展示时不要夸大）：
+//   1. 压缩（compaction）后旧消息被摘要替代，压缩前的触碰会丢失 —— 触碰集是
+//      「当前可见转录」的下界，不是全集。
+//   2. 相对路径按字面记录（edit 常给绝对路径、bash 常给相对路径），跨会话交集
+//      可能因此漏报；但绝不做后缀模糊匹配（同名不同目录会假阳性，比漏报更糟）。
+//   3. 路径归一化只做：去首尾空白、反斜杠转斜杠、去 `./` 前缀。Windows 大小写
+//      不敏感不做归一。
+//   4. powershell/cmd 风格写入（Set-Content/Out-File/del/copy）、git apply、
+//      tee 落盘、xargs/timeout/nice 等包装词之后的子命令 —— 一律不提取
+//      （sudo/env/command/time/nohup/doas 这几个常见包装词能跳过）。
+//   5. 引号内的 `>`（如 echo "a > b"）不提取；但 `eval`/`sh -c` 套娃串内的重定
+//      向会被当成普通命令解析 —— 置信度下降，属已知噪声源。
+// ---------------------------------------------------------------------------
+/** 工具调用的参数键：SDK 用 arguments，兼容 input/args 两种历史写法。 */
+function argsOfBlock(blk) {
+    if (typeof blk.arguments !== "undefined")
+        return blk.arguments;
+    if (typeof blk.input !== "undefined")
+        return blk.input;
+    return blk.args;
+}
+/** content 块数组里工具调用的 (name, args) 对（非 toolCall 块跳过）。 */
+export function toolCallRefsOfContent(content) {
+    if (!Array.isArray(content))
+        return [];
+    const out = [];
+    for (const b of content) {
+        if (!b || typeof b !== "object")
+            continue;
+        const blk = b;
+        if (blk.type !== "toolCall" || typeof blk.name !== "string" || !blk.name)
+            continue;
+        const args = argsOfBlock(blk);
+        if (typeof args === "undefined")
+            out.push({ name: blk.name });
+        else
+            out.push({ name: blk.name, args });
+    }
+    return out;
+}
+/** 一条消息的全部工具调用：优先已解析的 toolCalls 字段，回落扫 content。 */
+export function toolCallsOf(m) {
+    if (!m || typeof m !== "object")
+        return [];
+    if (Array.isArray(m.toolCalls) && m.toolCalls.length > 0) {
+        return m.toolCalls.filter((t) => !!t && typeof t === "object" && typeof t.name === "string");
+    }
+    return toolCallRefsOfContent(m.content);
+}
+/** 写工具白名单：只有这些工具的目标路径才算「写」。 */
+const WRITE_TOOL_NAMES = new Set(["edit", "write", "edit_soft"]);
+function pathOfWriteArgs(args) {
+    if (!args || typeof args !== "object")
+        return undefined;
+    const a = args;
+    for (const k of ["path", "file_path", "filePath"]) {
+        const v = a[k];
+        if (typeof v === "string" && v.trim())
+            return v.trim();
+    }
+    return undefined;
+}
+/** 路径轻归一（见文件头局限 3）：只做无损、低风险的规范化。 */
+export function normalizeTouchPath(p) {
+    let s = p.trim().replace(/\\/g, "/");
+    while (s.startsWith("./"))
+        s = s.slice(2);
+    return s;
+}
+// ---------------------------------------------------------------------------
+// bash 命令解析（高置信度写入迹象，见文件头规则）
+// ---------------------------------------------------------------------------
+/** shell 分词：引号/反斜杠转义感知，引号剥掉（`echo "a > b"` 的 > 不会误报）。 */
+function tokenizeShell(cmd) {
+    const toks = [];
+    let cur = "";
+    let quote = null;
+    let i = 0;
+    const push = () => {
+        if (cur !== "") {
+            toks.push(cur);
+            cur = "";
+        }
+    };
+    while (i < cmd.length) {
+        const ch = cmd[i];
+        if (quote) {
+            if (ch === "\\" && quote === '"' && i + 1 < cmd.length) {
+                cur += cmd[i + 1];
+                i += 2;
+                continue;
+            }
+            if (ch === quote)
+                quote = null;
+            else
+                cur += ch;
+            i++;
+            continue;
+        }
+        if (ch === "\\" && i + 1 < cmd.length) {
+            cur += cmd[i + 1];
+            i += 2;
+            continue;
+        }
+        if (ch === "'" || ch === '"') {
+            quote = ch;
+            i++;
+            continue;
+        }
+        if (ch === " " || ch === "\t" || ch === "\r" || ch === "\n") {
+            push();
+            i++;
+            continue;
+        }
+        // 多字符操作符先行：>>、<<、&&、||、2>、1>> 等。
+        const three = cmd.slice(i, i + 3);
+        const two = cmd.slice(i, i + 2);
+        if (/^\d>>/.test(three) || three === "<<<") {
+            push();
+            toks.push(three);
+            i += 3;
+            continue;
+        }
+        if (two === ">>" || two === "<<" || two === "&&" || two === "||" || /^\d>$/.test(two) || /^\d</.test(two)) {
+            push();
+            toks.push(two);
+            i += 2;
+            continue;
+        }
+        if (ch === ">" || ch === "<" || ch === ";" || ch === "|" || ch === "&" || ch === "(" || ch === ")") {
+            push();
+            toks.push(ch);
+            i++;
+            continue;
+        }
+        cur += ch;
+        i++;
+    }
+    push();
+    return toks;
+}
+const SEGMENT_SEPS = new Set([";", "&&", "||", "|", "&", "(", ")"]);
+/** 命令段首可跳过的包装词（之后仍是命令位置）。 */
+const WRAPPER_WORDS = new Set(["sudo", "env", "command", "time", "nohup", "doas"]);
+const MULTI_TARGET_WRITES = new Set(["rm", "rmdir", "mkdir", "touch"]);
+/** 只算目标端（最后一个非 flag 参数）的命令。 */
+const DEST_ONLY_WRITES = new Set(["cp", "mv"]);
+/** 重定向目标黑名单：不是文件。 */
+function isNonFileTarget(t) {
+    if (t === "/dev/null" || t.toLowerCase() === "nul")
+        return true;
+    if (/^&\d*$/.test(t))
+        return true;
+    if (t === "-")
+        return true;
+    return false;
+}
+/** bash 命令 → 高置信度写入路径（去重前，调用方计数）。 */
+export function bashWriteTargets(command) {
+    if (typeof command !== "string" || !command.trim())
+        return [];
+    const toks = tokenizeShell(command);
+    const out = [];
+    // 命令段内状态：cmd = 当前命令词（null = 还在找命令词），args = 该段非 flag 参数。
+    let cmd = null;
+    let pendingMulti = [];
+    let pendingDest = null;
+    let afterDoubleDash = false;
+    let skipNext = false;
+    const flush = () => {
+        out.push(...pendingMulti);
+        if (pendingDest !== null)
+            out.push(pendingDest);
+        pendingMulti = [];
+        pendingDest = null;
+    };
+    for (let i = 0; i < toks.length; i++) {
+        const t = toks[i];
+        if (skipNext) {
+            skipNext = false;
+            continue;
+        }
+        if (SEGMENT_SEPS.has(t)) {
+            flush();
+            cmd = null;
+            afterDoubleDash = false;
+            continue;
+        }
+        // 重定向：`[fd]>[>] 目标`；heredoc 一律跳过。分隔符不能当目标吃掉
+        // （`echo a > ; rm x` 里 `;` 必须保留，否则后半段命令会错位）。
+        if (/^(\d*)(>>?)$/.test(t) || t === "<<" || t === "<<<" || /^(\d*)<<<?$/.test(t)) {
+            const next = toks[i + 1];
+            if (typeof next !== "undefined" && !SEGMENT_SEPS.has(next)) {
+                if (t.includes(">") && !isNonFileTarget(next))
+                    out.push(next);
+                skipNext = true;
+            }
+            continue;
+        }
+        if (t === "--") {
+            afterDoubleDash = true;
+            continue;
+        }
+        if (cmd === null) {
+            if (WRAPPER_WORDS.has(t))
+                continue;
+            if (t.includes("=") && !t.startsWith("-"))
+                continue; // VAR=val 前缀赋值
+            cmd = t;
+            if (cmd === "xargs")
+                cmd = "\0xargs-opaque"; // xargs 后的是子命令+参数，别当文件（局限 4）
+            continue;
+        }
+        if (cmd === "\0xargs-opaque")
+            continue;
+        if (!afterDoubleDash && t.startsWith("-") && t.length > 1)
+            continue; // flag
+        if (MULTI_TARGET_WRITES.has(cmd))
+            pendingMulti.push(t);
+        else if (DEST_ONLY_WRITES.has(cmd))
+            pendingDest = t; // 只留最后一个
+    }
+    flush();
+    return out.filter((p) => p !== "" && !isNonFileTarget(p));
+}
+/** 转录消息 → 触碰文件集（按次数降序，次数相同按最后时间降序）。 */
+export function extractTouches(messages, opts) {
+    if (!Array.isArray(messages) || messages.length === 0)
+        return [];
+    const max = Math.max(1, Math.floor(opts?.maxMessages ?? 500));
+    const slice = messages.slice(Math.max(0, messages.length - max));
+    const acc = new Map();
+    const bump = (raw, ts) => {
+        const path = normalizeTouchPath(raw);
+        if (!path)
+            return;
+        const e = acc.get(path);
+        if (e) {
+            e.count++;
+            if (ts > e.lastTs)
+                e.lastTs = ts;
+        }
+        else {
+            acc.set(path, { count: 1, lastTs: ts });
+        }
+    };
+    for (const m of slice) {
+        if (!m || typeof m !== "object")
+            continue;
+        const ts = typeof m.timestamp === "number" && Number.isFinite(m.timestamp) ? m.timestamp : 0;
+        for (const call of toolCallsOf(m)) {
+            const name = call.name;
+            if (WRITE_TOOL_NAMES.has(name)) {
+                const p = pathOfWriteArgs(call.args);
+                if (p)
+                    bump(p, ts);
+                continue;
+            }
+            if (name === "bash" && call.args && typeof call.args === "object") {
+                const c = call.args.command;
+                if (typeof c === "string") {
+                    for (const p of bashWriteTargets(c))
+                        bump(p, ts);
+                }
+            }
+        }
+        // `!` 命令 / bashExecution 转录消息：命令在顶层 command 字段。
+        if (m.role === "bashExecution" && typeof m.command === "string") {
+            for (const p of bashWriteTargets(m.command))
+                bump(p, ts);
+        }
+    }
+    return [...acc.entries()]
+        .map(([path, e]) => ({ path, count: e.count, lastTs: e.lastTs }))
+        .sort((a, b) => b.count - a.count || b.lastTs - a.lastTs);
+}
+/** 两个触碰集求并（sidecar 合并用）：sidecar 是历史快照、live 是当前全量，
+ *  两者消息集重叠，所以 count 取大不相加 —— 相加会把同一批触碰算两遍。
+ *  压缩后次数只能保证是下界（presence 与 lastTs 精确，交集计算不受影响）。 */
+export function unionTouchLists(a, b) {
+    const acc = new Map();
+    for (const f of [...(Array.isArray(a) ? a : []), ...(Array.isArray(b) ? b : [])]) {
+        if (!f || typeof f.path !== "string")
+            continue;
+        const e = acc.get(f.path);
+        const count = typeof f.count === "number" && Number.isFinite(f.count) ? f.count : 0;
+        const lastTs = typeof f.lastTs === "number" && Number.isFinite(f.lastTs) ? f.lastTs : 0;
+        if (e) {
+            if (count > e.count)
+                e.count = count;
+            if (lastTs > e.lastTs)
+                e.lastTs = lastTs;
+        }
+        else {
+            acc.set(f.path, { count, lastTs });
+        }
+    }
+    return [...acc.entries()]
+        .map(([path, e]) => ({ path, count: e.count, lastTs: e.lastTs }))
+        .sort((x, y) => y.count - x.count || y.lastTs - x.lastTs);
+}
+/** 两个触碰集的交集：取 a 中同时出现在 b 里的条目（计数/时间沿用 a 的）。 */
+export function intersectTouches(a, b) {
+    if (!Array.isArray(a) || !Array.isArray(b) || a.length === 0 || b.length === 0)
+        return [];
+    const inB = new Set(b.map((f) => f.path));
+    return a.filter((f) => inB.has(f.path));
+}
+/** 单条触碰格式化（供提醒用）：条目超长时优先保路径完整 —— 路径永不截断
+ *  （半截路径不可用，还可能误导 AI 去碰错文件），只丢 `×N` 后缀。 */
+export function formatTouchEntry(f, perItemCap = 60) {
+    const full = f.count > 1 ? `${f.path} ×${f.count}` : f.path;
+    if (full.length <= perItemCap)
+        return full;
+    return f.path;
+}
+/** 触碰集紧凑格式化：前 N 条完整路径 + `… (+K)` 计数（静默丢弃是大忌）。 */
+export function formatTouchesCompact(files, opts) {
+    if (!Array.isArray(files) || files.length === 0)
+        return "";
+    const maxItems = Math.max(1, Math.floor(opts?.maxItems ?? 3));
+    const cap = Math.max(10, Math.floor(opts?.perItemCap ?? 60));
+    const shown = files.slice(0, maxItems).map((f) => formatTouchEntry(f, cap));
+    if (files.length > maxItems)
+        shown.push(`… (+${files.length - maxItems})`);
+    return shown.join("、");
+}

package/dist/server/dsh/dsh-agent-service.js

@@ -2590,6 +2590,8 @@ export class DshClientSession {
             terminalToolsEnabled: this.settings.terminalToolsEnabled,
             terminalBash: this.settings.terminalBash,
             terminalBashIdleMs: this.settings.terminalBashIdleMs,
+            // DSH 引擎无 customTool 注册面（工具来自 shipped preset），read 目录覆盖面不存在。
+            readDirEnabled: true,
             editSoftEnabled: this.settings.editSoftEnabled,
             // DSH 无独立重试配置（pi 引擎才暴露），保持默认。
             retryMaxAttempts: DEFAULT_RETRY_MAX_ATTEMPTS,
@@ -3880,6 +3882,11 @@ export class DshClientSession {
         this.emit({ type: "notice", level: "error", text: error, textEn: errorEn });
         this.emit({ type: "clone_provider_result", reqId, ok: false, error });
     }
+    async enrichModels(reqId, _ids, _hints) {
+        const error = pick(this.getLang(), "DSH 引擎不支持自定义 provider", "The DSH engine does not support custom providers", "dsh.provider.custom.unsupported");
+        this.emit({ type: "enrich_models_result", reqId, ok: false, error });
+    }
+    abortEnrichModels(_reqId) { }
     // -----------------------------------------------------------------------
     // 其他
     // -----------------------------------------------------------------------

package/dist/server/index.js

@@ -16,7 +16,7 @@
  *   all share one conversation list per project.
  *   PI_CODING_AGENT_DIR  pi config dir (auth/models/skills) — passed to the SDK
  */
-import { existsSync, readFileSync } from "node:fs";
+import { existsSync, mkdirSync, readFileSync } from "node:fs";
 import { stat } from "node:fs/promises";
 import { createServer, request as proxyRequest } from "node:http";
 import { createConnection } from "node:net";
@@ -32,7 +32,7 @@ import { PROTOCOL_VERSION } from "./protocol-version.js";
 import { AgentService, workspacePath, QuiesceRejectedError } from "./agent-service.js";
 import { WS_MAX_PAYLOAD_BYTES, isAbsoluteWirePath, wireToAbs } from "./files-service.js";
 import { registerFileTransferRoutes } from "./file-transfer-routes.js";
-import { previewKind } from "./text-sniff.js";
+import { isAudioFile, previewKind } from "./text-sniff.js";
 import { startControlServer } from "./control-socket.js";
 import { scheduleUploadCleanup } from "./uploads.js";
 import { ensureWindowsBash, windowsBashDir } from "./ensure-bash.js";
@@ -81,6 +81,18 @@ function cliFlag(name) {
 const PORT = Number(cliFlag("--port") ?? process.env.PI_WEB_PORT ?? 8787);
 const CWD = resolve(cliFlag("--cwd") ?? process.env.PI_WEB_CWD ?? process.cwd());
 const DATA_DIR = resolve(cliFlag("--data-dir") ?? process.env.PI_WEB_DATA_DIR ?? join(homedir(), ".pi-web"));
+// The data dir is where the control socket, client state, plugins, themes and
+// uploads live, but nothing guarantees it exists on a first run (a fresh
+// `server install` never creates it). The control socket binds at startup —
+// before any store gets a chance to mkdir its own subdirectory — and bind()
+// into a missing directory fails with EACCES, silently disabling the whole
+// control channel (status/quiesce) for that process. Create it up front.
+try {
+    mkdirSync(DATA_DIR, { recursive: true });
+}
+catch (err) {
+    console.warn(`[data] 无法创建数据目录 ${DATA_DIR}: ${err.message}`);
+}
 // Dev-no-cache setting read without a ClientStateStore instance (the index.html
 // route runs before any client attaches). Reads the same global settings blob.
 function readDevNoCacheSetting() {
@@ -279,7 +291,9 @@ app.get("/api/health", (_req, res) => {
  *
  * Media preview (no download param): only image/video kinds are served —
  * text goes over the WebSocket, and exe/jar/etc. are never exposed here.
- * express's sendFile handles Range requests, so video seeking works.
+ * Audio files (isAudioFile: browser-playable containers only) are allowed too
+ * so present_files cards and the preview dialog can inline-play them.
+ * express's sendFile handles Range requests, so video/audio seeking works.
  *
  * Download (?download=1): any file kind is served with
  * Content-Disposition: attachment so the browser saves it instead of
@@ -316,7 +330,8 @@ app.get("/api/file", async (req, res) => {
         // allowlist them explicitly here.
         const lower = name.toLowerCase();
         const isHtmlPreview = lower.endsWith(".html") || lower.endsWith(".htm") || lower.endsWith(".xhtml");
-        if (!isDownload && kind !== "image" && kind !== "video" && !isHtmlPreview) {
+        const isAudioPreview = isAudioFile(name);
+        if (!isDownload && kind !== "image" && kind !== "video" && !isHtmlPreview && !isAudioPreview) {
             res.status(400).end("not a previewable media file");
             return;
         }
@@ -1609,6 +1624,12 @@ wss.on("connection", (ws) => {
             case "set_model":
                 void cs.setModel(msg.modelId);
                 break;
+            case "set_default_model":
+                void cs.setDefaultModel?.(msg.modelId);
+                break;
+            case "clear_default_model":
+                cs.clearDefaultModel?.();
+                break;
             case "set_thinking":
                 cs.setThinking(msg.level);
                 break;
@@ -1727,6 +1748,12 @@ wss.on("connection", (ws) => {
             case "clone_provider":
                 void cs.cloneProvider(msg.provider, msg.reqId);
                 break;
+            case "enrich_models":
+                void cs.enrichModels(msg.reqId, msg.ids, msg.hints);
+                break;
+            case "abort_enrich_models":
+                cs.abortEnrichModels(msg.reqId);
+                break;
             case "list_provider_keys":
                 cs.listProviderKeys();
                 break;
@@ -1814,6 +1841,7 @@ wss.on("connection", (ws) => {
                     terminalToolsEnabled: msg.terminalToolsEnabled,
                     terminalBash: msg.terminalBash,
                     terminalBashIdleMs: msg.terminalBashIdleMs,
+                    readDirEnabled: msg.readDirEnabled,
                     editSoftEnabled: msg.editSoftEnabled,
                     questionnaireEnabled: msg.questionnaireEnabled,
                     parallelReminderEnabled: msg.parallelReminderEnabled,

package/dist/server/model-admin.js

@@ -13,6 +13,7 @@
  */
 import { mkdirSync, readFileSync, writeFileSync } from "node:fs";
 import { dirname, join } from "node:path";
+import { enrichBatch } from "./model-enrich.js";
 import { pick } from "./i18n.js";
 import { ProviderOAuthFlowManager } from "./provider-oauth-flow.js";
 /** Drop provisional overlay rows once the official catalog lists the same id.
@@ -196,9 +197,13 @@ function parseOpenAiModel(m) {
         boolMeta(r.supports_vision) === true ||
         boolMeta(r.vision) === true ||
         strArrMeta(r.input)?.includes("image") === true;
+    // 名字兜底：很多端点（如 Antigravity-Manager 的 /v1/models）只给 id 不给
+    // 元数据；id 含 thinking 即视为思考模型（与反代自身的 is_thinking_model
+    // 同口径）。只增不减：取消勾选靠手填（编辑页/刷新合并都只补缺）。
     const reasoning = boolMeta(r.reasoning) === true ||
         boolMeta(r.supports_reasoning) === true ||
-        modalities?.includes("reasoning") === true;
+        modalities?.includes("reasoning") === true ||
+        id.toLowerCase().includes("thinking");
     const contextWindow = numMeta(r.context_window) ?? numMeta(r.context_length) ?? numMeta(r.max_model_len) ?? numMeta(r.max_context_length);
     const maxTokens = numMeta(r.max_tokens) ?? numMeta(r.max_output_tokens) ?? numMeta(r.max_completion_tokens);
     return {
@@ -218,9 +223,12 @@ function parseGoogleModel(m) {
     const rawName = typeof r.name === "string" ? r.name : "";
     const id = rawName.replace(/^models\//, "");
     const displayName = typeof r.displayName === "string" ? r.displayName : undefined;
+    // 与 OpenAI 形状同口径的名字兜底（见 parseOpenAiModel）。
+    const reasoning = id.toLowerCase().includes("thinking");
     return {
         id,
         ...(displayName && displayName !== id ? { name: displayName } : {}),
+        ...(reasoning ? { reasoning: true } : {}),
         ...(numMeta(r.inputTokenLimit) ? { contextWindow: numMeta(r.inputTokenLimit) } : {}),
         ...(numMeta(r.outputTokenLimit) ? { maxTokens: numMeta(r.outputTokenLimit) } : {}),
     };
@@ -228,6 +236,7 @@ function parseGoogleModel(m) {
 export class ModelAdminService {
     host;
     oauthFlows;
+    activeEnrichAbort = new Map();
     constructor(host) {
         this.host = host;
         this.oauthFlows = new ProviderOAuthFlowManager({
@@ -251,6 +260,25 @@ export class ModelAdminService {
     }
     dispose() {
         this.oauthFlows.dispose();
+        for (const ac of this.activeEnrichAbort.values()) {
+            ac.abort();
+        }
+        this.activeEnrichAbort.clear();
+    }
+    abortEnrichModels(reqId) {
+        if (reqId !== undefined) {
+            const ac = this.activeEnrichAbort.get(reqId);
+            if (ac) {
+                ac.abort();
+                this.activeEnrichAbort.delete(reqId);
+            }
+        }
+        else {
+            for (const ac of this.activeEnrichAbort.values()) {
+                ac.abort();
+            }
+            this.activeEnrichAbort.clear();
+        }
     }
     async logoutProviderOAuth(provider) {
         const providerId = provider.trim();
@@ -880,6 +908,94 @@ export class ModelAdminService {
         }
         this.host.flushSnapshot();
     }
+    /**
+     * Enrich custom-provider DRAFT rows with public catalog params
+     * (OpenRouter primary, models.dev secondary) — enrich_models_result.
+     * Nothing is saved here; the UI fills blanks from the result.
+     */
+    async enrichModels(reqId, ids, hints, lang) {
+        const l = lang?.() ?? "en";
+        const ac = new AbortController();
+        this.activeEnrichAbort.set(reqId, ac);
+        try {
+            const cleanIds = [...new Set((ids ?? []).map((s) => (s ?? "").trim()).filter(Boolean))].slice(0, 100);
+            if (cleanIds.length === 0) {
+                throw new Error(pick(l, "没有可补的模型 id", "No model ids to enrich", "models.enrich.empty"));
+            }
+            const cleanHints = {};
+            for (const [k, v] of Object.entries(hints ?? {})) {
+                if (k.trim() && (v ?? "").trim())
+                    cleanHints[k.trim()] = (v ?? "").trim();
+            }
+            const results = await enrichBatch(cleanIds, cleanHints, {
+                lang: (l === "zh" ? "zh" : "en"),
+                signal: ac.signal,
+                onProgress: (p) => {
+                    this.host.emit({
+                        type: "enrich_models_progress",
+                        reqId,
+                        phase: p.phase,
+                        current: p.current,
+                        total: p.total,
+                        message: p.message,
+                    });
+                },
+            });
+            const matched = results.filter((r) => r.status === "matched").length;
+            const suggested = results.filter((r) => r.status === "suggested").length;
+            this.host.emit({ type: "enrich_models_result", reqId, ok: true, results });
+            this.host.emit({
+                type: "notice",
+                level: "info",
+                text: `🔍 已补 ${matched} 个${suggested ? `，另有 ${suggested} 个只找到相近家族（见建议）` : ""}，共 ${results.length} 个`,
+                textEn: `🔍 Enriched ${matched}${suggested ? `, ${suggested} with family suggestions` : ""} of ${results.length}`,
+            });
+        }
+        catch (err) {
+            const isAbort = err.message === "aborted" || ac.signal.aborted;
+            const partial = err.partialResults;
+            if (isAbort) {
+                const hasPartial = Array.isArray(partial) && partial.length > 0;
+                if (hasPartial) {
+                    const matched = partial.filter((r) => r.status === "matched").length;
+                    this.host.emit({ type: "enrich_models_result", reqId, ok: true, results: partial });
+                    this.host.emit({
+                        type: "notice",
+                        level: "warning",
+                        text: `⏹ 已中断补参数（已保留中断前匹配的 ${matched} 个模型）`,
+                        textEn: `⏹ Model enrichment aborted (kept ${matched} models matched before abort)`,
+                    });
+                }
+                else {
+                    this.host.emit({
+                        type: "enrich_models_result",
+                        reqId,
+                        ok: false,
+                        error: pick(l, "已取消补参数", "Model enrichment cancelled", "models.enrich.cancelled"),
+                    });
+                    this.host.emit({
+                        type: "notice",
+                        level: "warning",
+                        text: pick(l, "已取消补参数", "Model enrichment cancelled", "models.enrich.cancelled"),
+                    });
+                }
+            }
+            else {
+                const error = err.message;
+                this.host.emit({ type: "enrich_models_result", reqId, ok: false, error });
+                this.host.emit({
+                    type: "notice",
+                    level: "error",
+                    text: `补参数失败：${error}`,
+                    textEn: `Enrich failed: ${error}`,
+                });
+            }
+        }
+        finally {
+            this.activeEnrichAbort.delete(reqId);
+        }
+        this.host.flushSnapshot();
+    }
     /** Enumerate pi's built-in providers with auth capabilities and status. */
     async listProviders() {
         const mr = this.host.modelRuntime();
@@ -1074,9 +1190,12 @@ export class ModelAdminService {
             boolMeta(r.supports_vision) === true ||
             boolMeta(r.vision) === true ||
             strArrMeta(r.input)?.includes("image") === true;
+        // 名字兜底（同模块级 parseOpenAiModel）：只给 id 的端点靠 id 含 thinking
+        // 视为思考模型，只增不减，手填合并只补缺。
         const reasoning = boolMeta(r.reasoning) === true ||
             boolMeta(r.supports_reasoning) === true ||
-            modalities?.includes("reasoning") === true;
+            modalities?.includes("reasoning") === true ||
+            id.toLowerCase().includes("thinking");
         const contextWindow = numMeta(r.context_window) ??
             numMeta(r.context_length) ??
             numMeta(r.max_model_len) ??
@@ -1099,9 +1218,12 @@ export class ModelAdminService {
         const rawName = typeof r.name === "string" ? r.name : "";
         const id = rawName.replace(/^models\//, "");
         const displayName = typeof r.displayName === "string" ? r.displayName : undefined;
+        // 与 OpenAI 形状同口径的名字兜底（见模块级 parseOpenAiModel）。
+        const reasoning = id.toLowerCase().includes("thinking");
         return {
             id,
             ...(displayName && displayName !== id ? { name: displayName } : {}),
+            ...(reasoning ? { reasoning: true } : {}),
             ...(numMeta(r.inputTokenLimit) ? { contextWindow: numMeta(r.inputTokenLimit) } : {}),
             ...(numMeta(r.outputTokenLimit) ? { maxTokens: numMeta(r.outputTokenLimit) } : {}),
         };