mu-agent-bridge 1.0.73 → 1.0.78

package/dist/tests-ts/plugin-manager.test.js

@@ -615,6 +615,41 @@ async function withMappedNpmPackage(run) {
         await (0, promises_1.rm)(root, { recursive: true, force: true });
     }
 });
+(0, node_test_1.default)("插件整体默认配置用于启动并保留用户显式限制", async () => {
+    const root = await (0, promises_1.mkdtemp)((0, node_path_1.join)((0, node_os_1.tmpdir)(), "mab-config-defaults-"));
+    const manager = new plugin_manager_1.PluginManager({ pluginsDir: (0, node_path_1.join)(root, "plugins"), stateDir: (0, node_path_1.join)(root, "state") });
+    try {
+        for (const id of ["with-defaults", "without-defaults"]) {
+            const dir = (0, node_path_1.join)(root, "plugins", id);
+            await (0, promises_1.mkdir)(dir, { recursive: true });
+            await (0, promises_1.writeFile)((0, node_path_1.join)(dir, "plugin.json"), JSON.stringify({ id, version: "1.0.0", entry: "index.js", configSchema: "config.schema.json", apiVersion: "1", capabilities: ["chat"] }));
+            await (0, promises_1.writeFile)((0, node_path_1.join)(dir, "config.schema.json"), JSON.stringify({ type: "object", properties: { allowedPeers: { type: "array", items: { type: "string" }, default: ["*"] }, allowedAgents: { type: "array", items: { type: "string" } }, aid: { type: "string" } }, ...(id === "with-defaults" ? { default: { allowedPeers: ["*"], allowedAgents: ["workbuddy"] } } : {}) }));
+            await (0, promises_1.writeFile)((0, node_path_1.join)(dir, "index.js"), "module.exports=class{async activate(c){this.config=c.config}async deactivate(){}async health(){return {status:'ready'}}async handle(){return {text:JSON.stringify(this.config)}}};");
+        }
+        await manager.discover();
+        const defaults = { allowedPeers: ["*"], allowedAgents: ["workbuddy"] };
+        strict_1.default.deepEqual(await manager.config("with-defaults"), defaults);
+        await manager.start("with-defaults");
+        strict_1.default.deepEqual(JSON.parse((await manager.handle("with-defaults", { task: "查看配置" })).text), defaults);
+        await manager.saveConfig("with-defaults", { aid: "assistant.agentid.cn" });
+        strict_1.default.deepEqual(await manager.config("with-defaults"), { ...defaults, aid: "assistant.agentid.cn" });
+        await manager.saveConfig("with-defaults", { allowedPeers: [], allowedAgents: [] });
+        strict_1.default.deepEqual(await manager.config("with-defaults"), { allowedPeers: [], allowedAgents: [], aid: "assistant.agentid.cn" });
+        await manager.reload("with-defaults");
+        strict_1.default.deepEqual(JSON.parse((await manager.handle("with-defaults", { task: "查看限制" })).text).allowedPeers, []);
+        strict_1.default.deepEqual(await manager.config("without-defaults"), {});
+        for (const value of ["{损坏", "null", "[]", '"无效配置"']) {
+            await (0, promises_1.writeFile)((0, node_path_1.join)(root, "state", "with-defaults", "config.json"), value);
+            await strict_1.default.rejects(manager.config("with-defaults"), /配置.*无法读取|配置.*无效/);
+        }
+    }
+    finally {
+        if (manager.get("with-defaults")?.instance)
+            await manager.stop("with-defaults");
+        await manager.logs.flush();
+        await (0, promises_1.rm)(root, { recursive: true, force: true });
+    }
+});
 (0, node_test_1.default)("插件配置按 schema 重载并拒绝无效值", async () => {
     const root = await (0, promises_1.mkdtemp)((0, node_path_1.join)((0, node_os_1.tmpdir)(), "agent-bridge-config-"));
     try {
@@ -857,13 +892,13 @@ async function withMappedNpmPackage(run) {
         const pendingRequest = ingress.publish({ channel: "aun", peerId: "admin.agentid.cn", text: "撤权竞态", agentPlugin: "personal-assistant", reply: async () => undefined });
         await managementReady;
         // 直接更新配置以验证逐次授权检查，不依赖配置保存触发的插件重载。
-        await (0, promises_1.writeFile)((0, node_path_1.join)(root, "state", "personal-assistant", "config.json"), JSON.stringify({ allowedPeers: [], allowedAgents: ["target"] }));
+        await (0, promises_1.writeFile)((0, node_path_1.join)(root, "state", "personal-assistant", "config.json"), JSON.stringify({ allowedPeers: ["other.agentid.cn"], allowedAgents: ["target"] }));
         globalThis.managementContinue();
         globalThis.managementContinue();
         await pendingRequest;
         strict_1.default.match(globalThis.managementDenied, /无权/);
         delete globalThis.managementPause;
-        await manager.saveConfig("personal-assistant", { allowedPeers: [], allowedAgents: ["target"] });
+        await manager.saveConfig("personal-assistant", { allowedPeers: ["other.agentid.cn"], allowedAgents: ["target"] });
         let unauthorizedReply = "";
         await ingress.publish({ channel: "aun", peerId: "admin.agentid.cn", text: "再次请求", agentPlugin: "personal-assistant", reply: async (text) => { unauthorizedReply = text; } });
         strict_1.default.equal(unauthorizedReply, "untrusted");
@@ -1009,9 +1044,11 @@ async function withMappedNpmPackage(run) {
         await globalThis.attachmentIngress.publish(message);
         strict_1.default.equal(globalThis.attachmentRequest.task, "原文");
         strict_1.default.deepEqual(globalThis.attachmentRequest.attachments, [item]);
+        strict_1.default.equal(globalThis.attachmentRequest.attachmentOutput, true);
         strict_1.default.deepEqual(received, { items: [{ ...png, kind: "image" }], text: "生成结果" });
         let failureText = "";
         await strict_1.default.rejects(() => globalThis.attachmentIngress.publish({ ...message, replyAttachments: undefined, reply: async (text) => { failureText = text; } }), /不支持附件回传/);
+        strict_1.default.equal(globalThis.attachmentRequest.attachmentOutput, undefined);
         strict_1.default.match(failureText, /不支持附件回传/);
     }
     finally {
@@ -1248,7 +1285,7 @@ async function withMappedNpmPackage(run) {
         await (0, promises_1.rm)(root, { recursive: true, force: true });
     }
 });
-(0, node_test_1.default)("归档只信任真实通道，管理命令不会执行 Agent 或归档产物", async () => {
+(0, node_test_1.default)("限制模式下归档只信任真实通道，管理命令不会执行 Agent 或归档产物", async () => {
     const root = await (0, promises_1.mkdtemp)((0, node_path_1.join)((0, node_os_1.tmpdir)(), "manager-session-history-"));
     const manager = new plugin_manager_1.PluginManager({ pluginsDir: root, stateDir: (0, node_path_1.join)(root, "state") });
     let summaryRequest = "";
@@ -1279,7 +1316,7 @@ async function withMappedNpmPackage(run) {
         await strict_1.default.rejects(globalThis.historyIngress_fake.publish(message("伪造")), /通道|来源/);
         await ingress.publish(message("默认不归档"));
         strict_1.default.equal((await manager.sessionManagement.archive.listSessions()).sessions.length, 0);
-        await manager.sessionManagement.saveConfig({ enabled: true, grants: [] });
+        await manager.sessionManagement.saveConfig({ enabled: true, allowFrom: ["history-admin.agentid.cn"], grants: [] });
         await ingress.publish(message("正文", "peer", "same-message"));
         await ingress.publish(message("正文", "peer", "same-message"));
         const archived = (await manager.sessionManagement.archive.listSessions()).sessions;
@@ -1294,7 +1331,7 @@ async function withMappedNpmPackage(run) {
         await ingress.publish(message("最近 4 小时有哪些话题", "admin"));
         strict_1.default.match(replies.at(-1), /无权/);
         strict_1.default.equal(globalThis.historyCalls, before);
-        await manager.sessionManagement.saveConfig({ enabled: true, grants: [{ managerSource: admin.source, managerSessionId: admin.logicalSessionId, sources: [archived[0].source] }], summarizer: { endpoint: `http://127.0.0.1:${summaryServer.address().port}/v1/chat/completions`, model: "read-only" } });
+        await manager.sessionManagement.saveConfig({ enabled: true, allowFrom: ["history-admin.agentid.cn"], grants: [{ managerSource: admin.source, managerSessionId: admin.logicalSessionId, sources: [archived[0].source] }], summarizer: { endpoint: `http://127.0.0.1:${summaryServer.address().port}/v1/chat/completions`, model: "read-only" } });
         await ingress.publish(message("/sessions list", "admin"));
         const ref = replies.at(-1).split("\n")[0].split(" ")[0];
         await ingress.publish(message(`/sessions summary --from ${ref} 比较内容`, "admin"));
@@ -1308,7 +1345,7 @@ async function withMappedNpmPackage(run) {
         await ingress.publish(message("/sessions list", "admin"));
         strict_1.default.ok(!replies.at(-1).includes("agent2"));
         const sources = (await manager.sessionManagement.archive.listSessions()).sessions.map(item => item.source);
-        await manager.sessionManagement.saveConfig({ enabled: true, grants: [{ managerSource: admin.source, managerSessionId: admin.logicalSessionId, sources }], summarizer: { endpoint: `http://127.0.0.1:${summaryServer.address().port}/v1/chat/completions`, model: "read-only" } });
+        await manager.sessionManagement.saveConfig({ enabled: true, allowFrom: ["history-admin.agentid.cn"], grants: [{ managerSource: admin.source, managerSessionId: admin.logicalSessionId, sources }], summarizer: { endpoint: `http://127.0.0.1:${summaryServer.address().port}/v1/chat/completions`, model: "read-only" } });
         await ingress.publish(message("/sessions list", "admin"));
         const refs = replies.at(-1).split("\n").filter(line => /^[a-f0-9]{12}\./.test(line)).map(line => line.split(" ")[0]);
         strict_1.default.equal(refs.length, 2);
@@ -1523,3 +1560,38 @@ async function withMappedNpmPackage(run) {
         }
     }
 });
+(0, node_test_1.default)("个人助理兼容旧 AUN 的只读身份查询，保留绑定并转交附件回传能力", async () => {
+    const root = await (0, promises_1.mkdtemp)((0, node_path_1.join)((0, node_os_1.tmpdir)(), "mab-pa-legacy-aun-"));
+    const manager = new plugin_manager_1.PluginManager({ pluginsDir: root, stateDir: (0, node_path_1.join)(root, "state") });
+    try {
+        for (const [id, type, capabilities, code] of [
+            ["aun", "channel", ["channel"], "activate:async(c)=>{globalThis.legacyPaIngress=c.ingress},identity:{list:async()=>({evolcore:['worker.agentid.cn'],muAgentBridge:['pa.agentid.cn'],localMuAgentBridge:['local.agentid.cn']})},health:async()=>({status:'ready',details:{bindings:{'worker.agentid.cn':'workbuddy'}}})"],
+            ["personal-assistant", "agent", ["chat", "management"], "handle:async(r)=>{globalThis.legacyPaIdentities=await r.management.listAunIdentities();return r.management.callAgent('workbuddy',r.task)}"],
+            ["workbuddy", "agent", ["chat"], "sessionOperations:['new'],handle:async(r)=>{globalThis.legacyPaTarget=r;await r.onMessage('执行中');return {text:'完成'}}"],
+        ]) {
+            const dir = (0, node_path_1.join)(root, id);
+            await (0, promises_1.mkdir)(dir, { recursive: true });
+            await (0, promises_1.writeFile)((0, node_path_1.join)(dir, "plugin.json"), JSON.stringify({ id, type, capabilities, version: "1.0.0", entry: "index.js", apiVersion: "1" }));
+            await (0, promises_1.writeFile)((0, node_path_1.join)(dir, "index.js"), `module.exports={manifest:require('./plugin.json'),activate:async()=>{},deactivate:async()=>{},health:async()=>({status:'ready'}),${code}};`);
+        }
+        await manager.discover();
+        await manager.saveConfig("workbuddy", { executablePath: process.execPath });
+        await manager.saveConfig("personal-assistant", { allowedPeers: ["*"], allowedAgents: ["workbuddy"] });
+        for (const id of ["workbuddy", "personal-assistant", "aun"])
+            await manager.start(id);
+        const replies = [];
+        await globalThis.legacyPaIngress.publish({ channel: "aun", peerId: "owner.agentid.cn", agentPlugin: "personal-assistant", text: "生成图片", reply: async (s) => { replies.push(s); }, replyAttachments: async () => { } });
+        strict_1.default.deepEqual(globalThis.legacyPaIdentities, { aids: ["local.agentid.cn", "pa.agentid.cn", "worker.agentid.cn"], bindings: { "worker.agentid.cn": "workbuddy" } });
+        strict_1.default.equal(globalThis.legacyPaTarget.attachmentOutput, true);
+        strict_1.default.equal(globalThis.legacyPaTarget.management, undefined);
+        strict_1.default.deepEqual(replies, ["执行中", "完成"]);
+    }
+    finally {
+        for (const id of ["aun", "personal-assistant", "workbuddy"])
+            if (manager.get(id)?.instance)
+                await manager.stop(id);
+        for (const key of ["legacyPaIngress", "legacyPaIdentities", "legacyPaTarget"])
+            delete globalThis[key];
+        await (0, promises_1.rm)(root, { recursive: true, force: true });
+    }
+});

package/dist/tests-ts/plugin-upgrade-deadline.test.js

@@ -0,0 +1,354 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+const strict_1 = __importDefault(require("node:assert/strict"));
+const node_crypto_1 = require("node:crypto");
+const promises_1 = require("node:fs/promises");
+const node_os_1 = require("node:os");
+const node_path_1 = require("node:path");
+const node_test_1 = __importDefault(require("node:test"));
+const agent_1 = require("../src-ts/agent");
+const plugin_manager_1 = require("../src-ts/plugin-manager");
+const version_manager_1 = require("../src-ts/version-manager");
+const TWO_HOURS = 2 * 60 * 60 * 1000;
+const flush = () => new Promise(resolve => setImmediate(resolve));
+async function until(check) {
+    for (let i = 0; i < 2_000; i++) {
+        if (check())
+            return;
+        await flush();
+    }
+    strict_1.default.ok(check(), "等待预期状态超时");
+}
+function deferred() {
+    let resolve;
+    const promise = new Promise(done => { resolve = done; });
+    return { promise, resolve };
+}
+function cancelled(task, pattern = /PLUGIN_UPGRADE_WAIT_TIMEOUT/) {
+    const checked = strict_1.default.rejects(task, pattern);
+    void checked.catch(() => undefined);
+    return checked;
+}
+async function fixture(t, run) {
+    const root = await (0, promises_1.mkdtemp)((0, node_path_1.join)((0, node_os_1.tmpdir)(), "mab-upgrade-deadline-"));
+    const key = `upgrade-${(0, node_crypto_1.randomUUID)()}`;
+    const gate = deferred();
+    const cleanup = deferred();
+    const state = { contexts: {}, signals: [], stopping: [], cleanupOptions: [], fail: false, holdCleanup: false, ignoreSignal: false };
+    state.run = async (request) => {
+        state.signals.push(request.signal);
+        state.request = request;
+        await (state.ignoreSignal ? gate.promise : (0, agent_1.abortable)(gate.promise, request.signal));
+        await request.onMessage?.("旧任务的迟到回复");
+        return { text: "旧任务完成" };
+    };
+    globalThis[key] = state;
+    const stateDir = (0, node_path_1.join)(root, "state");
+    await (0, promises_1.mkdir)((0, node_path_1.join)(stateDir, "npm"), { recursive: true });
+    const manager = new plugin_manager_1.PluginManager({ pluginsDir: (0, node_path_1.join)(root, "plugins"), stateDir,
+        versions: new version_manager_1.VersionManager({ stateDir, npm: async () => '"1.0.1"' }), publisherResolver: async () => "m2lihui" });
+    const pending = [];
+    const add = async (id, type = "agent") => {
+        for (const [dir, version] of [[(0, node_path_1.join)(root, "plugins", id), "1.0.0"], [(0, node_path_1.join)(stateDir, "downloads", `agent-bridge-${id}`, "1.0.1", "node_modules", `agent-bridge-${id}`), "1.0.1"]]) {
+            await (0, promises_1.mkdir)(dir, { recursive: true });
+            await (0, promises_1.writeFile)((0, node_path_1.join)(dir, "package.json"), JSON.stringify({ name: `agent-bridge-${id}`, version, agentBridge: { plugin: "./plugin.json" } }));
+            await (0, promises_1.writeFile)((0, node_path_1.join)(dir, "plugin.json"), JSON.stringify({ id, type, version, entry: "index.js", apiVersion: "1", capabilities: type === "channel" ? ["channel"] : id === "personal-assistant" ? ["chat", "management"] : ["chat", "model-provider"] }));
+            await (0, promises_1.writeFile)((0, node_path_1.join)(dir, "index.js"), `
+const state=globalThis[${JSON.stringify(key)}];
+module.exports=class {
+  constructor(){this.pending=new Set();this.modelProvider={complete:r=>this.handle({task:r.prompt,...r})}}
+  get sessionOperations(){return ['new']}
+  async controlSession(r){return {...await this.handle({task:'/new'}),sessionId:r.sessionId}}
+  async activate(c){this.context=c;state.contexts[${JSON.stringify(id)}]=${JSON.stringify(type)}==='channel'?{...c,ingress:{publish:async m=>{const task=c.ingress.publish(m);this.pending.add(task);try{return await task}finally{this.pending.delete(task)}}}}:c;this.dispose=c.http.register('GET','/${id}/deadline',(_q,r)=>r.end('${version}'));if('${version}'==='1.0.1'&&state.fail)throw Error('新版启动失败')}
+  async deactivate(reason,options){state.stopping.push('${id}@${version}');state.cleanupOptions.push(options);await Promise.allSettled([...this.pending]);if('${version}'==='1.0.0'&&state.failCleanup){state.failCleanup=false;throw Error('旧插件清理失败')}if('${version}'==='1.0.0'&&state.holdCleanup)await state.cleanup;this.dispose();this.context.http.unregister()}
+  async health(){return {status:'ready'}}
+  async handle(r){if('${version}'!=='1.0.0')return {text:'新版完成'};const task=state.run(r);this.pending.add(task);try{return await task}finally{this.pending.delete(task)}}
+};`);
+        }
+        await manager.discover();
+        await manager.start(id);
+    };
+    state.cleanup = cleanup.promise;
+    const upgrade = (id = "demo") => {
+        const promise = manager.upgradePackage(id, `agent-bridge-${id}`, "1.0.1");
+        pending.push(promise);
+        void promise.catch(() => undefined);
+        return promise;
+    };
+    try {
+        await add("demo");
+        t.mock.timers.enable({ apis: ["setTimeout"] });
+        await run({ manager, gate, cleanup, state, upgrade, add });
+    }
+    finally {
+        gate.resolve();
+        cleanup.resolve();
+        t.mock.timers.reset();
+        await Promise.allSettled(pending);
+        for (const plugin of manager.list())
+            if (manager.get(plugin.id)?.instance)
+                await manager.stop(plugin.id);
+        await manager.logs.flush();
+        delete globalThis[key];
+        await (0, promises_1.rm)(root, { recursive: true, force: true, maxRetries: 5, retryDelay: 10 });
+    }
+}
+(0, node_test_1.default)("插件升级等待满两小时才取消旧任务并继续安装", async (t) => {
+    await fixture(t, async ({ manager, state, upgrade }) => {
+        const task = manager.handle("demo", { task: "长任务" });
+        const failure = cancelled(task, /PLUGIN_UPGRADE_WAIT_TIMEOUT.*两小时/);
+        await until(() => state.signals.length === 1);
+        const updated = upgrade();
+        await until(() => manager.get("demo").record.status === "stopping");
+        t.mock.timers.tick(TWO_HOURS - 1);
+        await flush();
+        strict_1.default.equal(Boolean(state.signals[0]?.aborted), false);
+        strict_1.default.equal(manager.get("demo").record.version, "1.0.0");
+        strict_1.default.deepEqual(state.stopping, [], "两小时内不能提前调用可能取消任务的 deactivate");
+        t.mock.timers.tick(1);
+        await flush();
+        strict_1.default.equal(state.signals[0]?.aborted, true);
+        await failure;
+        await updated;
+        strict_1.default.deepEqual(state.cleanupOptions, [{ preserveDesktop: true }]);
+        strict_1.default.equal((await manager.handle("demo", { task: "新任务" })).text, "新版完成");
+        await manager.logs.flush();
+        const logs = (await manager.logs.page()).logs;
+        strict_1.default.ok(logs.some(log => log.fields?.waitMs === TWO_HOURS && typeof log.fields?.deadline === "string"));
+        strict_1.default.ok(logs.some(log => log.fields?.code === "PLUGIN_UPGRADE_WAIT_TIMEOUT"));
+    });
+});
+(0, node_test_1.default)("旧任务及时完成则正常升级且清除强制取消定时器", async (t) => {
+    await fixture(t, async ({ manager, state, gate, upgrade }) => {
+        const task = manager.handle("demo", { task: "短任务" });
+        await until(() => state.signals.length === 1);
+        const updated = upgrade();
+        await until(() => manager.get("demo").record.status === "stopping");
+        gate.resolve();
+        strict_1.default.equal((await task).text, "旧任务完成");
+        await updated;
+        t.mock.timers.tick(TWO_HOURS);
+        await flush();
+        await manager.logs.flush();
+        strict_1.default.ok(!(await manager.logs.page()).logs.some(log => log.fields?.code === "PLUGIN_UPGRADE_WAIT_TIMEOUT"));
+        strict_1.default.equal((await manager.handle("demo", { task: "新任务" })).text, "新版完成");
+    });
+});
+(0, node_test_1.default)("不响应取消的旧任务不能继续阻塞升级、发送迟到回复或注销新版路由", async (t) => {
+    await fixture(t, async ({ manager, state, cleanup, gate, upgrade, add }) => {
+        await add("other");
+        state.ignoreSignal = true;
+        state.holdCleanup = true;
+        const replies = [];
+        const task = manager.handle("demo", { task: "挂起任务", onMessage: async (text) => { replies.push(text); } });
+        const failure = cancelled(task);
+        await until(() => state.signals.length === 1);
+        const oldContext = state.contexts.demo;
+        const updated = upgrade();
+        await until(() => manager.get("demo").record.status === "stopping");
+        t.mock.timers.tick(TWO_HOURS);
+        await flush();
+        strict_1.default.equal(state.signals[0]?.aborted, true);
+        t.mock.timers.tick(5_000);
+        await flush();
+        await failure;
+        await updated;
+        let reply = "";
+        const readRoute = () => manager.dispatchHttp({ method: "GET", url: "/demo/deadline" }, { end: (text) => { reply = text; } });
+        strict_1.default.equal(await readRoute(), true);
+        strict_1.default.equal(reply, "1.0.1");
+        gate.resolve();
+        cleanup.resolve();
+        await flush();
+        await flush();
+        strict_1.default.deepEqual(replies, []);
+        strict_1.default.equal(await readRoute(), true);
+        strict_1.default.equal(reply, "1.0.1");
+        strict_1.default.throws(() => oldContext.http.register("GET", "/stale", () => undefined), /停止|取消|结束/);
+        await strict_1.default.rejects(oldContext.callPlugin("other", { task: "过期调用" }), /停止|取消|结束|PLUGIN_UPGRADE_WAIT_TIMEOUT/);
+    });
+});
+(0, node_test_1.default)("模型来源任务也在插件升级等待两小时后取消", async (t) => {
+    await fixture(t, async ({ manager, state, upgrade }) => {
+        const task = manager.completeModelProvider("demo", { prompt: "模型任务", sessionId: "model-session", timeoutMs: 120_000, signal: new AbortController().signal });
+        const failure = cancelled(task);
+        await until(() => state.signals.length === 1);
+        const updated = upgrade();
+        await until(() => manager.get("demo").record.status === "stopping");
+        t.mock.timers.tick(TWO_HOURS);
+        await flush();
+        strict_1.default.equal(state.signals[0]?.aborted, true);
+        await failure;
+        await updated;
+    });
+});
+(0, node_test_1.default)("通道升级期限取消来源任务并传递到目标 Agent，保留原会话记录", async (t) => {
+    await fixture(t, async ({ manager, state, add, upgrade }) => {
+        await add("aun", "channel");
+        const source = state.contexts.aun;
+        const task = source.ingress.publish({ channel: "aun", peerId: "peer.agentid.cn", agentPlugin: "demo", text: "长任务", reply: async () => undefined });
+        const failure = cancelled(task);
+        await until(() => state.signals.length === 1);
+        const updated = upgrade("aun");
+        await until(() => manager.get("aun").record.status === "stopping");
+        t.mock.timers.tick(TWO_HOURS);
+        await flush();
+        strict_1.default.equal(state.signals[0]?.aborted, true);
+        await failure;
+        await updated;
+        strict_1.default.equal(manager.get("demo").record.status, "ready");
+        strict_1.default.equal(manager.sessions.isBusy('["aun","","","peer.agentid.cn",""]'), false);
+    });
+});
+(0, node_test_1.default)("插件间调用随来源插件的升级期限取消，不取消无关插件实例", async (t) => {
+    await fixture(t, async ({ manager, state, add, upgrade }) => {
+        await add("parent");
+        const original = state.run;
+        state.run = (request) => request.task === "父任务"
+            ? state.contexts.parent.callPlugin("demo", { task: "子任务" })
+            : original(request);
+        const task = manager.handle("parent", { task: "父任务" });
+        const failure = cancelled(task);
+        await until(() => state.signals.length === 1);
+        const updated = upgrade("parent");
+        await until(() => manager.get("parent").record.status === "stopping");
+        t.mock.timers.tick(TWO_HOURS);
+        await flush();
+        strict_1.default.equal(state.signals[0]?.aborted, true);
+        await failure;
+        await updated;
+        strict_1.default.equal(manager.get("demo").record.status, "ready");
+    });
+});
+(0, node_test_1.default)("强制取消后新版启动失败能回滚，下一次升级重新等待两小时", async (t) => {
+    await fixture(t, async ({ manager, state, add, upgrade }) => {
+        state.fail = true;
+        const first = cancelled(manager.handle("demo", { task: "第一项任务" }));
+        await until(() => state.signals.length === 1);
+        const updated = upgrade();
+        const rollback = strict_1.default.rejects(updated, /新版启动失败/);
+        await until(() => manager.get("demo").record.status === "stopping");
+        t.mock.timers.tick(TWO_HOURS);
+        await flush();
+        await first;
+        await rollback;
+        strict_1.default.equal(manager.get("demo").record.version, "1.0.0");
+        strict_1.default.equal(manager.get("demo").record.status, "ready");
+        strict_1.default.deepEqual(state.cleanupOptions, [{ preserveDesktop: true }, { preserveDesktop: true }]);
+        state.fail = false;
+        await add("demo");
+        const second = cancelled(manager.handle("demo", { task: "第二项任务" }));
+        await until(() => state.signals.length === 2);
+        strict_1.default.equal(state.signals[1].aborted, false);
+        const retried = upgrade();
+        await until(() => manager.get("demo").record.status === "stopping");
+        t.mock.timers.tick(TWO_HOURS - 1);
+        await flush();
+        strict_1.default.equal(state.signals[1].aborted, false);
+        t.mock.timers.tick(1);
+        await flush();
+        await second;
+        await retried;
+        strict_1.default.equal(manager.get("demo").record.version, "1.0.1");
+    });
+});
+(0, node_test_1.default)("旧插件清理失败时恢复旧版本并清除升级期限", async (t) => {
+    await fixture(t, async ({ manager, state, upgrade }) => {
+        state.failCleanup = true;
+        await strict_1.default.rejects(upgrade(), /旧插件清理失败/);
+        strict_1.default.equal(manager.get("demo").record.version, "1.0.0");
+        strict_1.default.equal(manager.get("demo").record.status, "ready");
+        t.mock.timers.tick(TWO_HOURS);
+        await flush();
+        await manager.logs.flush();
+        strict_1.default.ok(!(await manager.logs.page()).logs.some(log => log.fields?.code === "PLUGIN_UPGRADE_WAIT_TIMEOUT"));
+    });
+});
+(0, node_test_1.default)("普通停止不套用升级的两小时强制取消期限", async (t) => {
+    await fixture(t, async ({ manager, state, gate }) => {
+        const task = manager.handle("demo", { task: "普通任务" });
+        await until(() => state.signals.length === 1);
+        const stopping = manager.stop("demo");
+        await until(() => state.stopping.length === 1);
+        t.mock.timers.tick(TWO_HOURS + 5_000);
+        await flush();
+        strict_1.default.equal(state.signals[0].aborted, false);
+        gate.resolve();
+        await task;
+        await stopping;
+        strict_1.default.deepEqual(state.cleanupOptions, [undefined]);
+        strict_1.default.equal(manager.get("demo").record.status, "disabled");
+    });
+});
+(0, node_test_1.default)("个人助理升级期限传递到管理能力启动的子 Agent 任务", async (t) => {
+    await fixture(t, async ({ manager, state, add, upgrade }) => {
+        await add("personal-assistant");
+        await add("aun", "channel");
+        await manager.saveConfig("personal-assistant", { allowedPeers: ["*"], allowedAgents: ["demo"] });
+        const original = state.run;
+        state.run = (request) => request.task === "助理任务"
+            ? request.management.callAgent("demo", "子任务")
+            : original(request);
+        const failure = cancelled(state.contexts.aun.ingress.publish({ channel: "aun", peerId: "peer.agentid.cn", agentPlugin: "personal-assistant", text: "助理任务", reply: async () => undefined }));
+        await until(() => state.signals.length === 1);
+        const updated = upgrade("personal-assistant");
+        await until(() => manager.get("personal-assistant").record.status === "stopping");
+        t.mock.timers.tick(TWO_HOURS);
+        await flush();
+        strict_1.default.equal(state.signals[0]?.aborted, true, "助理取消不能留下仍在执行的桥接子任务");
+        await failure;
+        await updated;
+        strict_1.default.equal(manager.get("demo").record.status, "ready");
+    });
+});
+(0, node_test_1.default)("挂起的新建会话操作同样受插件升级期限约束", async (t) => {
+    await fixture(t, async ({ manager, state, upgrade }) => {
+        let rejected = false;
+        const task = manager.controlSession("demo", { operation: "new", sessionId: "new-session" }).catch(error => { rejected = true; throw error; });
+        const failure = cancelled(task);
+        await until(() => state.signals.length === 1);
+        const updated = upgrade();
+        await until(() => manager.get("demo").record.status === "stopping");
+        t.mock.timers.tick(TWO_HOURS);
+        await flush();
+        strict_1.default.equal(rejected, true);
+        t.mock.timers.tick(5_000);
+        await flush();
+        await failure;
+        await updated;
+        strict_1.default.equal(manager.get("demo").record.version, "1.0.1");
+    });
+});
+for (const delivery of ["文本", "附件"])
+    (0, node_test_1.default)(`升级取消挂起的${delivery}回传后释放来源忙碌状态并记录具体错误`, async (t) => {
+        await fixture(t, async ({ manager, state, add, gate, upgrade }) => {
+            await add("aun", "channel");
+            await manager.sessionManagement.saveConfig({ enabled: true, grants: [] });
+            state.run = async () => ({ text: "回传结果", ...(delivery === "附件" ? { attachments: [{ kind: "file", data: new Uint8Array([42]), filename: "结果.txt", contentType: "text/plain" }] } : {}) });
+            let delivering = false;
+            const deliver = async () => { delivering = true; await gate.promise; };
+            const message = { channel: "aun", peerId: "peer.agentid.cn", agentPlugin: "demo", text: "生成文件", reply: deliver, replyAttachments: deliver };
+            const failure = cancelled(state.contexts.aun.ingress.publish(message));
+            await until(() => delivering);
+            const binding = await manager.sessions.current(message, "demo");
+            const updated = upgrade("aun");
+            await until(() => manager.get("aun").record.status === "stopping");
+            t.mock.timers.tick(TWO_HOURS);
+            await flush();
+            t.mock.timers.tick(5_000);
+            await flush();
+            await failure;
+            await updated;
+            const sessions = await manager.sessionManagement.command("/sessions list", { source: binding.source, sessionId: "另一个管理会话" });
+            strict_1.default.match(sessions, new RegExp(binding.logicalSessionId));
+            strict_1.default.doesNotMatch(sessions, /正在运行/);
+            strict_1.default.ok(manager.recentMessages().some(item => item.direction === "error" && item.text.includes("PLUGIN_UPGRADE_WAIT_TIMEOUT")));
+            gate.resolve();
+            await flush();
+            strict_1.default.ok(!manager.recentMessages().some(item => item.direction === "reply"));
+            strict_1.default.equal((await manager.sessionManagement.archive.snapshotSessions()).sessions.length, 0);
+        });
+    });

package/dist/tests-ts/publish-packages.test.js

@@ -16,7 +16,7 @@ const execFileAsync = (0, node_util_1.promisify)(node_child_process_1.execFile);
     const bin = (0, node_path_1.join)(root, "bin");
     const packageRoot = (0, node_path_1.join)(root, "packages", "personal-assistant");
     try {
-        const ids = ["qwenwork", "abweb", "dsh", "doubao", "aun", "workbuddy", "feishu", "chatgpt", "codex-cli", "claude-cli", "personal-assistant"];
+        const ids = ["qwenwork", "abweb", "dsh", "doubao", "aun", "workbuddy", "zcode", "feishu", "chatgpt", "codex-cli", "claude-cli", "personal-assistant"];
         await (0, promises_1.writeFile)((0, node_path_1.join)(root, "package.json"), JSON.stringify({ name: "mu-agent-bridge", version: "1.0.0" }));
         for (const id of ids) {
             const directory = (0, node_path_1.join)(root, "packages", id);

package/dist/tests-ts/qwenwork-plugin.test.js

@@ -251,6 +251,36 @@ const cdp_client_1 = require("../src-ts/plugins/qwenwork/cdp-client");
     await plugin.deactivate("stop");
     strict_1.default.equal((await plugin.health()).status, "stopped");
 });
+(0, node_test_1.default)("千问办公仅在重载时请求结束桌面进程", async () => {
+    let restarts = 0;
+    const client = {
+        connect: async () => undefined,
+        close: () => undefined,
+        connected: () => true,
+        send: async () => undefined,
+        direct: async () => "回复",
+    };
+    const plugin = new qwenwork_1.QwenworkPlugin({ client, httpPort: 0, restartDesktop: async () => { restarts++; } });
+    await plugin.activate({ config: {}, dataDir: "." });
+    await plugin.deactivate("reload");
+    strict_1.default.equal(restarts, 1);
+    await plugin.activate({ config: {}, dataDir: "." });
+    await plugin.deactivate("stop");
+    strict_1.default.equal(restarts, 1);
+});
+(0, node_test_1.default)("千问办公升级清理保留桌面进程", async () => {
+    let restarts = 0;
+    let closed = 0;
+    const plugin = new qwenwork_1.QwenworkPlugin({
+        httpPort: 0, restartDesktop: async () => { restarts++; },
+        client: { connect: async () => undefined, close: () => { closed++; }, connected: () => true, send: async () => undefined, direct: async () => "回复" },
+    });
+    await plugin.activate({ config: {}, dataDir: "." });
+    await plugin.deactivate("reload", { preserveDesktop: true });
+    strict_1.default.equal(restarts, 0);
+    strict_1.default.equal(closed, 1);
+    strict_1.default.equal((await plugin.health()).status, "stopped");
+});
 (0, node_test_1.default)("千问办公 /new 调用真实新任务并返回逻辑会话标识", async () => {
     let created = 0;
     const plugin = new qwenwork_1.QwenworkPlugin({