pi-web-ui 0.86.2 → 0.87.0

package/dist/server/dsh/dsh-client.js

@@ -379,6 +379,34 @@ export class DshRuntime {
         await this.start();
         return this._request("skills/set-disabled", { skills });
     }
+    // -----------------------------------------------------------------------
+    // Agent 预设 RPC（dsh-web 四模式；roster 缺失 = legacy，list 报 unavailable）
+    // -----------------------------------------------------------------------
+    /** 列出预设名录（字段以运行时树为准；unavailable = 回落 legacy 单组合）。 */
+    async listPresets() {
+        await this.start();
+        return this._request("preset/list", {});
+    }
+    /** 登记创建期预设（新会话首个 prompt 前调用；创建时消费）。 */
+    async assignPreset(sessionId, preset) {
+        await this.start();
+        return this._request("preset/assign", { sessionId, preset });
+    }
+    /** 空白会话切换预设（首轮后锁；走 {ok:false,code,message} 信封，不抛 domain 错）。 */
+    async selectPreset(sessionId, preset) {
+        await this.start();
+        return this._request("preset/select", { sessionId, preset });
+    }
+    /** 读一个会话的权限预设（官方 permissions projection 同源：options + currentValue）。 */
+    async getPermission(sessionId) {
+        await this.start();
+        return this._request("permission/get", { sessionId });
+    }
+    /** 按会话热切换权限预设（写会话日志，无需重启；走 {ok:false,code,message} 信封）。 */
+    async setPermission(sessionId, preset) {
+        await this.start();
+        return this._request("permission/set", { sessionId, preset });
+    }
     /** 优雅关闭：shutdown 握手 → stdin EOF → SIGTERM → SIGKILL 阶梯。 */
     async close() {
         if (this.closed)

package/dist/server/dsh/dsh-sessions.js

@@ -96,6 +96,34 @@ function zstdDecompressAll(buf) {
     }
     return out || buf.toString("utf8");
 }
+/** 会话实际跑的预设：最后一条 agent-preset/selected 事件 > header 记录 > null。
+ *  重启/切换回放时用它恢复 conv 预设（服务内存已丢时唯一的真相）。纯函数。 */
+/** 回放权限预设：permission/preset 事件最后一条（切换即追加，无 header 兜底）。 */
+export function sessionLogPermission(log) {
+    let selected = null;
+    for (const ev of log.events) {
+        if (ev.type === "permission/preset") {
+            const id = ev.data?.preset;
+            if (typeof id === "string" && id)
+                selected = id;
+        }
+    }
+    return selected;
+}
+export function sessionLogPreset(log) {
+    let selected = null;
+    for (const ev of log.events) {
+        if (ev.type === "agent-preset/selected") {
+            const id = ev.data?.agentPreset;
+            if (typeof id === "string" && id)
+                selected = id;
+        }
+    }
+    if (selected)
+        return selected;
+    const h = log.header?.agentPreset;
+    return typeof h === "string" && h ? h : null;
+}
 /** 读一个会话 JSONL 文件 → { header, events }。 */
 export function readSessionLog(file) {
     const raw = readFileSync(file);

package/dist/server/dsh/dsh-usage.js

@@ -0,0 +1,82 @@
+/**
+ * dsh-usage.ts — DSH 的 token 用量 → 底栏统计口径（纯函数，零依赖，可单测）。
+ *
+ * DSH `TokenUsage`（模型调用级，见 @deepseek-ai/dsh-llm）的三个**输入桶互斥**：
+ *   inputTokens       未命中 prompt cache 的输入
+ *   cacheReadTokens   命中 prompt cache 的输入（DeepSeek 从 prompt_tokens 里减出来）
+ *   cacheWriteTokens  写入 / 刷新 cache 的输入
+ *   outputTokens      本次输出
+ * 所以「最近一次请求的输入 = 三桶相加」，而底栏缓存命中率（web/src/cache-stats.ts 的
+ * cacheMetrics）正是按 miss/read/write 归一算的 — 两边口径一致。
+ *
+ * 两条投递路径共用这里的映射：
+ *   - 老运行时（0.1.1-rc.2）：usage 是持久 `assistant/chunk` 事件（逐 chunk 推）
+ *   - 新运行时（0.1.1-rc.2 之后）：usage 随持久 `assistant/message.usage` 落一次，另有
+ *     `agent/assistant-stream` 直播帧（wrapper 转成 `assistant.stream` 通知）
+ * 字段形状两边一致，故收成一个纯函数（单测 tests/unit/dsh-usage.test.ts）。
+ */
+/** 非负整数兜底：缺字段 / 非法值（NaN、负、Infinity）一律算 0。 */
+function count(v) {
+    return typeof v === "number" && Number.isFinite(v) && v > 0 ? Math.round(v) : 0;
+}
+/**
+ * 一次模型调用的 DSH usage → 四桶统计。
+ *
+ * @param usage - `assistant/chunk.usage` / `assistant/message.usage`（结构子集）。
+ * @returns 四桶；非对象（缺 usage）→ null，调用方据此保留旧统计不动。
+ */
+export function normalizeDshUsage(usage) {
+    if (!usage || typeof usage !== "object" || Array.isArray(usage))
+        return null;
+    const u = usage;
+    return {
+        input: count(u.inputTokens),
+        output: count(u.outputTokens),
+        cacheRead: count(u.cacheReadTokens),
+        cacheWrite: count(u.cacheWriteTokens),
+    };
+}
+/** 最近一次请求的 prompt token 数（三个输入桶互斥 → 相加）。 */
+export function dshPromptTokens(tokens) {
+    return tokens.input + tokens.cacheRead + tokens.cacheWrite;
+}
+/**
+ * 当前上下文占用 = 最近一次请求的 prompt + 该次输出（下一步请求会把它带上）。
+ * 还没拿到任何 usage（新对话 / 回放无 usage）→ tokens/percent 为 null，前端显示
+ * `—`（与 pi 引擎一致，而不是骗人的 `0 / 1.0M`）。
+ */
+export function dshContextUsage(tokens, contextWindow) {
+    if (!tokens)
+        return { tokens: null, contextWindow, percent: null };
+    const used = dshPromptTokens(tokens) + tokens.output;
+    if (!(contextWindow > 0))
+        return { tokens: used, contextWindow, percent: null };
+    return { tokens: used, contextWindow, percent: Math.min(100, (used / contextWindow) * 100) };
+}
+/**
+ * 从会话日志（JSONL 回放 / 事件流）里取**最后一次** usage，供切回历史会话时底栏
+ * 立刻有上下文与缓存命中数字。
+ *
+ * 同时认两种形状：新运行时的 `assistant/message.usage`，老运行时的
+ * `assistant/chunk` 里的 `{type:"usage", usage}`。按日志顺序取最后一条（日志即 seq 序）。
+ */
+export function lastUsageFromEvents(events) {
+    let found = null;
+    for (const ev of events) {
+        const data = ev.data ?? {};
+        if (ev.type === "assistant/message") {
+            const usage = normalizeDshUsage(data.usage);
+            if (usage)
+                found = usage;
+        }
+        else if (ev.type === "assistant/chunk") {
+            const chunk = data.chunk;
+            if (chunk?.type === "usage") {
+                const usage = normalizeDshUsage(chunk.usage);
+                if (usage)
+                    found = usage;
+            }
+        }
+    }
+    return found;
+}

package/dist/server/dsh/preset-clones.js

@@ -0,0 +1,260 @@
+/**
+ * preset-clones.ts — DSH Agent 预设的 file: 克隆生成。
+ *
+ * 背景（实测结论，见 tests/scratch/preset-spike-2026-09-15.md 探针2/5）：
+ * shipped 预设（dsh-agent-presets 包内 standard/ptc/minimal/cordis）的 composition
+ * 用裸包名（`@deepseek-ai/dsh-persona` 等），挂载时按 agent scope 的 baseUrl 解析 ——
+ * launcher 式 boot 下 baseUrl = cordis.yml 所在目录，裸包名全部不可解析，四个预设
+ * 全带 broken，mount 直接失败。官方 CLI 靠 PluginPackages resolution 走安装锚点，
+ * pi-web-ui 没装那套。已验证可行的路：把 composition 拷一份，把裸包名改写成指向
+ * 运行时树的绝对 file: URL（clone），roster 用 clone 目录做 system 根并关掉
+ * includeShippedRoot（同名 id 早根优先，shipped 的 broken 条目会被 clone 遮蔽）。
+ *
+ * 本模块是纯 TS（vitest 可测）：发现 shipped 目录 → 逐预设拷贝+改写 → 写
+ * preset-plane.patch.yml（disable 表 + roster，二者原子生效，缺一不可）。
+ * Node 侧（dsh-agent-service）在每次启动运行时前调用；launcher 只消费环境变量，
+ * 不做生成（职责分离；clone 缺失 = 回落 legacy 单组合，见 PI_WEB_DSH_PRESET_PATCH）。
+ */
+import { cpSync, existsSync, mkdirSync, readdirSync, readFileSync, rmSync, writeFileSync } from "node:fs";
+import { dirname, join, resolve } from "node:path";
+import { fileURLToPath, pathToFileURL } from "node:url";
+/** roster 默认预设（与官方 dsh-web-app/cordis.patch.yml 一致）。 */
+export const PRESET_DEFAULT_ID = "standard";
+/** clone 输出目录名（<dataDir>/ 下）。 */
+export const PRESET_CLONES_DIRNAME = "dsh-preset-clones";
+/** preset-plane patch 文件名（<dataDir>/ 下，由本模块生成，launcher 按 env 加载）。 */
+export const PRESET_PLANE_PATCH_FILENAME = "dsh-preset-plane.patch.yml";
+// ---------------------------------------------------------------------------
+// 运行时树发现（server/dsh/runtime/runtime-root.mjs 的 TS 镜像；改一处记得改另一处）
+// ---------------------------------------------------------------------------
+/** 候选 node_modules 根（顺序同 runtime-root.mjs：显式 > 本包 > execPath 邻近 > npm 全局）。 */
+export function runtimeBaseCandidates() {
+    const out = [];
+    const explicit = process.env.PI_WEB_DSH_RUNTIME;
+    if (explicit)
+        out.push(resolve(explicit));
+    out.push(resolve(join(dirname(fileURLToPath(import.meta.url)), "..", "..", "node_modules")));
+    out.push(join(dirname(process.execPath), "node_modules"));
+    try {
+        const prefixes = [
+            process.env.NPM_CONFIG_PREFIX,
+            process.env.npm_config_prefix,
+            join(dirname(process.execPath), ".."),
+        ];
+        for (const prefix of prefixes) {
+            if (!prefix)
+                continue;
+            out.push(join(prefix, "lib", "node_modules"));
+            out.push(join(prefix, "node_modules"));
+        }
+    }
+    catch {
+        /* ignore */
+    }
+    return [...new Set(out)];
+}
+function runtimeBaseFor(root) {
+    const flat = join(root, "@deepseek-ai");
+    if (existsSync(join(flat, "dsh-base", "cordis.patch.yml")) &&
+        existsSync(join(flat, "dsh-app-boot", "lib", "index.js"))) {
+        return resolve(root);
+    }
+    const nested = join(root, "@deepseek-ai", "dsh", "node_modules", "@deepseek-ai");
+    if (existsSync(join(nested, "dsh-base", "cordis.patch.yml")) &&
+        existsSync(join(nested, "dsh-app-boot", "lib", "index.js"))) {
+        return resolve(join(root, "@deepseek-ai", "dsh", "node_modules"));
+    }
+    return null;
+}
+/** shipped 预设源目录（<runtimeBase>/@deepseek-ai/dsh-agent-presets/presets），找不到 = null。 */
+export function findShippedPresetsDir() {
+    for (const root of runtimeBaseCandidates()) {
+        const base = runtimeBaseFor(root);
+        if (!base)
+            continue;
+        const dir = join(base, "@deepseek-ai", "dsh-agent-presets", "presets");
+        if (existsSync(join(dir, "standard", "agent.cordis.yml")))
+            return dir;
+    }
+    return null;
+}
+// ---------------------------------------------------------------------------
+// 裸包名改写（entry 文件解析 + 文本改写；改写是纯函数，可单测）
+// ---------------------------------------------------------------------------
+/**
+ * 把一个裸包名解析到运行时树内的入口文件 file: URL。
+ * discovery 的健康检查要求 file: 指向真实文件（目录 URL 恒判 broken），
+ * mount 侧 Loader 直接 import 该 URL（peer 依赖从运行时树向上解析）。
+ * @param scopeDir 运行时树里 `@deepseek-ai` 作用域的本地目录
+ * @param bare `@deepseek-ai/<pkg>` 或 `@deepseek-ai/<pkg>/<sub>`
+ * @returns 入口文件的 file: URL；解析不出返回 null（调用方保留原文，roster 报 broken）
+ */
+export function resolvePackageEntryFileUrl(scopeDir, bare) {
+    const parts = bare.split("/");
+    if (parts.length < 2 || !parts[0].startsWith("@") || !parts[1])
+        return null;
+    const pkg = parts[1];
+    const sub = parts.length > 2 ? parts.slice(2).join("/") : undefined;
+    const pkgDir = join(scopeDir, pkg);
+    let manifest = {};
+    try {
+        manifest = JSON.parse(readFileSync(join(pkgDir, "package.json"), "utf8"));
+    }
+    catch {
+        return null;
+    }
+    const pickDefault = (v) => {
+        if (typeof v === "string")
+            return v;
+        if (v && typeof v === "object" && typeof v.default === "string") {
+            return v.default;
+        }
+        return null;
+    };
+    const candidates = [];
+    const exportsMap = manifest.exports && typeof manifest.exports === "object" ? manifest.exports : null;
+    if (sub) {
+        const hit = exportsMap?.[`./${sub}`];
+        const mapped = pickDefault(hit);
+        if (mapped)
+            candidates.push(mapped);
+        candidates.push(join("lib", `${sub}.js`), join("lib", sub, "index.js"));
+    }
+    else {
+        const mapped = pickDefault(exportsMap?.["."]);
+        if (mapped)
+            candidates.push(mapped);
+        if (typeof manifest.main === "string" && manifest.main)
+            candidates.push(manifest.main);
+        candidates.push(join("lib", "index.js"));
+    }
+    for (const rel of candidates) {
+        const abs = resolve(pkgDir, rel);
+        if (existsSync(abs))
+            return pathToFileURL(abs).href;
+    }
+    return null;
+}
+/**
+ * 把 composition 文本里的裸包名改写成入口文件 file: URL。
+ * 只动 `name:` 行的 `@deepseek-ai/<pkg[\/sub]>`（单/双引号），`cordis:group`、
+ * `./相对路径`、`!!js` 表达式一律不动。resolve 回 null 的行保留原文。
+ */
+export function rewriteBareNames(yamlText, resolve) {
+    const unresolved = new Set();
+    const text = yamlText.replace(/^([ \t]*name:[ \t]*)(['"])@deepseek-ai\/([^'"\s]+)\2([ \t]*(?:#.*)?)$/gim, (_m, prefix, quote, rest, suffix) => {
+        const url = resolve(`@deepseek-ai/${rest}`);
+        if (!url) {
+            unresolved.add(`@deepseek-ai/${rest}`);
+            return _m;
+        }
+        return `${prefix}${quote}${url}${quote}${suffix}`;
+    });
+    return { text, unresolved: [...unresolved] };
+}
+// ---------------------------------------------------------------------------
+// preset-plane patch（disable 表 + roster；纯函数，可单测）
+// ---------------------------------------------------------------------------
+/** 与官方 dsh-web-app/cordis.patch.yml 同表的 agent-plane disable 行（last-write-wins 整行替换）。 */
+export const AGENT_PLANE_DISABLE_IDS = [
+    "tool-bash",
+    "tool-pwsh",
+    "tool-jobs",
+    "tool-fs",
+    "tool-fs-search",
+    "tool-str-replace-editor",
+    "skill-filesystem",
+    "tool-skill",
+    "command-goal",
+    "tool-goal",
+    "plan-mode",
+    "compaction-basic",
+    "command-compact",
+    "tool-result-pruner",
+    "tool-subagent-control",
+    "tool-subagent-list-agents",
+    "tool-subagent",
+    "tool-subagent-fork",
+    "workflow-worker-thread",
+    "tool-workflow",
+    "tool-ralph",
+    "agent-instructions",
+    "tool-todo",
+    "tool-web",
+];
+/**
+ * 生成 preset-plane patch 文本：先 disable 全部 agent-plane 行（工具下沉到预设），
+ * 再 insert roster（clone 目录做唯一 system 根，关 shipped 根；user 根保留，
+ * `$DSH_HOME/.agent-presets` 的自建预设照常出现）。
+ * 注意：YAML 里 clonesDir 走 JSON 转义（Windows 反斜杠安全）。
+ */
+export function buildPresetPlanePatch(clonesDir) {
+    const disables = AGENT_PLANE_DISABLE_IDS.map((id) => `- id: ${id}\n  disabled: true`).join("\n");
+    return (`# 由 server/dsh/preset-clones.ts 生成（每次 DSH 运行时启动前重写），勿手改。\n` +
+        `# agent plane 下沉到预设：与官方 dsh-web-app/cordis.patch.yml 同表。\n` +
+        `${disables}\n\n` +
+        `# preset roster：clone 目录是唯一 system 根（同名 id 早根优先，shipped 原件被遮蔽）；\n` +
+        `# user 根保留（自建预设照常上架；裸包名在 launcher boot 下不可解析是已知限制，见文档）。\n` +
+        `- insert:\n` +
+        `    - id: agent-presets\n` +
+        `      name: '@deepseek-ai/dsh-agent-presets'\n` +
+        `      config:\n` +
+        `        default: ${PRESET_DEFAULT_ID}\n` +
+        `        includeShippedRoot: false\n` +
+        `        roots:\n` +
+        `          - path: ${JSON.stringify(clonesDir)}\n` +
+        `            trust: system\n`);
+}
+// ---------------------------------------------------------------------------
+// 生成主入口（IO；幂等：先清后写）
+// ---------------------------------------------------------------------------
+/**
+ * 生成 clone + preset-plane patch。成功返回结果；shipped 目录找不到返回 null
+ * （调用方回落 legacy 单组合：不下发 patch，roster 不挂载）。
+ * @param dataDir pi-web-ui 数据目录（输出 <dataDir>/dsh-preset-clones/ + patch 文件）
+ * @param shippedDir 覆盖 shipped 源目录（单测 fixture 用；缺省自动发现）
+ */
+export function generatePresetClones(dataDir, shippedDir) {
+    const src = shippedDir ?? findShippedPresetsDir();
+    if (!src)
+        return null;
+    let ids;
+    try {
+        ids = readdirSync(src, { withFileTypes: true })
+            .filter((e) => e.isDirectory() && /^[a-z0-9][a-z0-9-]*$/.test(e.name))
+            .map((e) => e.name)
+            .filter((id) => existsSync(join(src, id, "agent.cordis.yml")))
+            .sort();
+    }
+    catch {
+        return null;
+    }
+    if (!ids.includes(PRESET_DEFAULT_ID))
+        return null;
+    const clonesDir = join(dataDir, PRESET_CLONES_DIRNAME);
+    rmSync(clonesDir, { recursive: true, force: true });
+    mkdirSync(clonesDir, { recursive: true });
+    // 入口解析基：运行时树里的 `@deepseek-ai` 作用域
+    //（src = <base>/@deepseek-ai/dsh-agent-presets/presets，上两级即作用域）。
+    const scopeDir = join(src, "..", "..");
+    const warnings = [];
+    try {
+        for (const id of ids) {
+            const dst = join(clonesDir, id);
+            cpSync(join(src, id), dst, { recursive: true });
+            const compFile = join(dst, "agent.cordis.yml");
+            const raw = readFileSync(compFile, "utf8");
+            const { text, unresolved } = rewriteBareNames(raw, (bare) => resolvePackageEntryFileUrl(scopeDir, bare));
+            for (const u of unresolved)
+                warnings.push(`${id}: ${u}`);
+            writeFileSync(compFile, text);
+        }
+    }
+    catch {
+        rmSync(clonesDir, { recursive: true, force: true });
+        return null;
+    }
+    const patchFile = join(dataDir, PRESET_PLANE_PATCH_FILENAME);
+    writeFileSync(patchFile, buildPresetPlanePatch(clonesDir));
+    return { clonesDir, patchFile, presetIds: ids, warnings };
+}

package/dist/server/dsh/runtime/custom-prompt.mjs

@@ -0,0 +1,33 @@
+/**
+ * pi-web-ui custom prompt section (host plane).
+ *
+ * pi-web-ui 的自定义系统提示词在预设时代的载体：官方 preset 的 persona 行会
+ * shadow 掉部署人设（standard/ptc/cordis），minimal（complete:true）更压住一切，
+ * 所以沿用 system-prompt 行（DSH_PERSONA）已无法让用户的自定义提示词生效。
+ * 本行在 host plane 注册一个独立 section（deployment:persona-suffix 之后），
+ * standard/ptc/cordis 下随系统提示词下发，minimal 下被 complete 压住（官方语义）。
+ *
+ * 文本来自环境变量 PI_WEB_DSH_CUSTOM_PROMPT（Node 侧 dsh-agent-service 维护，
+ * 变化即重启运行时）；为空时不注册任何 section。无 Config export（loader 允许，
+ * 与 dsh-tool-ask-user 同形）。
+ */
+
+const name = "pi-webui-custom-prompt";
+const inject = ["systemPrompt"];
+
+function apply(ctx) {
+	const text = (process.env.PI_WEB_DSH_CUSTOM_PROMPT ?? "").trim();
+	if (!text) return;
+	const order = ctx.systemPrompt.getSectionOrder("DEPLOYMENT_PERSONA_SUFFIX") + 50;
+	ctx.effect(
+		() =>
+			ctx.systemPrompt.section({
+				name: "pi-webui:custom-prompt",
+				order,
+				text,
+			}),
+		"pi-webui-custom-prompt.section()",
+	);
+}
+
+export { apply, inject, name };