mu-agent-bridge 1.0.71 → 1.0.78

package/dist/src-ts/plugins/personal-assistant/tools.js

@@ -0,0 +1,167 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.TOOL_NAMES = void 0;
+exports.toolDefinitions = toolDefinitions;
+exports.parseToolArguments = parseToolArguments;
+exports.executeTool = executeTool;
+exports.parseExplicitCommand = parseExplicitCommand;
+exports.stringifyToolResult = stringifyToolResult;
+const ajv_1 = __importDefault(require("ajv"));
+const coordination_1 = require("./coordination");
+exports.TOOL_NAMES = ["agents_list", "agents_call", "aun_identity_list", "sessions_command", "sessions_today", "self_status", "ability_discover", "ability_call", "ability_request_get", "coordination_dispatch", "coordination_status", "owner_context_read"];
+const ajv = new ajv_1.default({ allErrors: false, strict: false });
+const schemas = {
+    agents_list: ajv.compile({ type: "object", additionalProperties: false, properties: {} }),
+    agents_call: ajv.compile({ type: "object", additionalProperties: false, required: ["id", "task"], properties: { id: { type: "string", minLength: 1, maxLength: 128 }, task: { type: "string", minLength: 1, maxLength: 20_000 } } }),
+    aun_identity_list: ajv.compile({ type: "object", additionalProperties: false, properties: {} }),
+    sessions_command: ajv.compile({ type: "object", additionalProperties: false, required: ["command"], properties: { command: { type: "string", minLength: 1, maxLength: 512 } } }),
+    sessions_today: ajv.compile({ type: "object", additionalProperties: false, properties: {} }),
+    self_status: ajv.compile({ type: "object", additionalProperties: false, properties: {} }),
+    ability_discover: ajv.compile({ type: "object", additionalProperties: false, required: ["providerAid"], properties: { providerAid: { type: "string", minLength: 3, maxLength: 256 } } }),
+    ability_call: ajv.compile({ type: "object", additionalProperties: false, required: ["providerAid", "path", "params"], properties: { providerAid: { type: "string", minLength: 3, maxLength: 256 }, path: { type: "string", minLength: 1, maxLength: 256 }, params: { type: "object" } } }),
+    ability_request_get: ajv.compile({ type: "object", additionalProperties: false, required: ["providerAid", "requestId"], properties: { providerAid: { type: "string", minLength: 3, maxLength: 256 }, requestId: { type: "string", minLength: 1, maxLength: 256 } } }),
+    coordination_dispatch: ajv.compile({ type: "object", additionalProperties: false, required: ["agentId", "aid", "task"], properties: { agentId: { type: "string", minLength: 1, maxLength: 64 }, aid: { type: "string", minLength: 3, maxLength: 256 }, task: { type: "string", minLength: 1, maxLength: 20000 } } }),
+    coordination_status: ajv.compile({ type: "object", additionalProperties: false, required: ["aid", "threadId"], properties: { aid: { type: "string", minLength: 3, maxLength: 256 }, threadId: { type: "string", minLength: 1, maxLength: 256 } } }),
+    owner_context_read: ajv.compile({ type: "object", additionalProperties: false, properties: {} }),
+};
+const toolDescriptions = {
+    agents_list: { name: "agents_list", description: "列出当前可调用的 Agent 及其 AID。", parameters: { type: "object", additionalProperties: false, properties: {} } },
+    agents_call: { name: "agents_call", description: "向白名单中的 Agent 发送一项独立任务。", parameters: { type: "object", additionalProperties: false, required: ["id", "task"], properties: { id: { type: "string", minLength: 1, maxLength: 128 }, task: { type: "string", minLength: 1, maxLength: 20_000 } } } },
+    aun_identity_list: { name: "aun_identity_list", description: "列出本机已知 AUN 身份及绑定状态。", parameters: { type: "object", additionalProperties: false, properties: {} } },
+    sessions_command: { name: "sessions_command", description: "执行只读的 /sessions 管理命令。", parameters: { type: "object", additionalProperties: false, required: ["command"], properties: { command: { type: "string", minLength: 1, maxLength: 512 } } } },
+    sessions_today: { name: "sessions_today", description: "生成今天工作的小结。", parameters: { type: "object", additionalProperties: false, properties: {} } },
+    self_status: { name: "self_status", description: "报告个人助理自身状态。", parameters: { type: "object", additionalProperties: false, properties: {} } },
+    ability_discover: { name: "ability_discover", description: "发现已配置 Provider 的只读 AUN Ability。", parameters: { type: "object", additionalProperties: false, required: ["providerAid"], properties: { providerAid: { type: "string", minLength: 3, maxLength: 256 } } } },
+    ability_call: { name: "ability_call", description: "调用无需认证和确认的纯只读 AUN Ability。", parameters: { type: "object", additionalProperties: false, required: ["providerAid", "path", "params"], properties: { providerAid: { type: "string", minLength: 3, maxLength: 256 }, path: { type: "string", minLength: 1, maxLength: 256 }, params: { type: "object" } } } },
+    ability_request_get: { name: "ability_request_get", description: "查询已发送 AUN Ability 的请求状态，不重新发送。", parameters: { type: "object", additionalProperties: false, required: ["providerAid", "requestId"], properties: { providerAid: { type: "string", minLength: 3, maxLength: 256 }, requestId: { type: "string", minLength: 1, maxLength: 256 } } } },
+    coordination_dispatch: { name: "coordination_dispatch", description: "显式向白名单 Agent 派发任务并创建独立 thread。", parameters: { type: "object", additionalProperties: false, required: ["agentId", "aid", "task"], properties: { agentId: { type: "string", minLength: 1, maxLength: 64 }, aid: { type: "string", minLength: 3, maxLength: 256 }, task: { type: "string", minLength: 1, maxLength: 20000 } } } },
+    coordination_status: { name: "coordination_status", description: "读取派发任务对应 thread 的当前状态和历史。", parameters: { type: "object", additionalProperties: false, required: ["aid", "threadId"], properties: { aid: { type: "string", minLength: 3, maxLength: 256 }, threadId: { type: "string", minLength: 1, maxLength: 256 } } } },
+    owner_context_read: { name: "owner_context_read", description: "读取管理员配置的 Owner 共享资料，仅作为参考。", parameters: { type: "object", additionalProperties: false, properties: {} } },
+};
+function toolDefinitions() {
+    return exports.TOOL_NAMES.map((name) => ({ type: "function", function: { ...toolDescriptions[name], parameters: { ...toolDescriptions[name].parameters } } }));
+}
+function parseToolArguments(name, raw) {
+    if (!exports.TOOL_NAMES.includes(name))
+        throw new Error("未知的管理工具");
+    let value;
+    try {
+        value = JSON.parse(raw);
+    }
+    catch {
+        throw new Error("管理工具参数不是有效 JSON");
+    }
+    const validate = schemas[name];
+    if (!validate(value))
+        throw new Error("管理工具参数不符合要求");
+    return value;
+}
+async function executeTool(name, args, management, allowedAgents, runtime = {}) {
+    if (!exports.TOOL_NAMES.includes(name))
+        throw new Error("未知的管理工具");
+    if (!schemas[name](args))
+        throw new Error("管理工具参数不符合要求");
+    switch (name) {
+        case "agents_list": {
+            const agents = await management.listAgents();
+            return allowedAgents?.length && !allowedAgents.includes("*") ? agents.map((agent) => allowedAgents.includes(agent.id) ? agent : { ...agent, callable: false, reason: "未加入个人助理 Agent 白名单" }) : agents;
+        }
+        case "agents_call": {
+            if (allowedAgents?.length && !allowedAgents.includes("*") && !allowedAgents.includes(String(args.id)))
+                throw new Error("目标 Agent 未加入个人助理白名单");
+            const result = await management.callAgent(String(args.id), String(args.task));
+            return {
+                text: result.text,
+                ...((result.attachments?.length || result.images?.length) ? { attachments: (result.attachments?.length ? result.attachments : result.images).map(({ filename, contentType }) => ({ filename, contentType })), attachmentDelivery: "附件将随最终回复自动回传，由通道确认发送结果，无需模型另行读取或发送。" } : {}),
+                ...(result.sessionId ? { sessionId: result.sessionId } : {}),
+            };
+        }
+        case "aun_identity_list": return management.listAunIdentities();
+        case "sessions_command": {
+            const command = String(args.command).trim();
+            if (!/^\/sessions(?:\s|$)/.test(command) || /[\r\n]/.test(command))
+                throw new Error("只支持单行 /sessions 只读命令");
+            return management.runSessionCommand(command);
+        }
+        case "sessions_today": return management.summarizeToday();
+        case "self_status": return { status: "ready", capabilities: exports.TOOL_NAMES };
+        case "ability_discover": {
+            const providerAid = String(args.providerAid);
+            if (!runtime.allowedProviders?.includes(providerAid) || !runtime.remoteAbility)
+                throw new Error("Provider 未加入远程 Ability 白名单");
+            return runtime.remoteAbility.discover(providerAid);
+        }
+        case "ability_call": {
+            const providerAid = String(args.providerAid);
+            if (!runtime.allowedProviders?.includes(providerAid) || !runtime.remoteAbility)
+                throw new Error("Provider 未加入远程 Ability 白名单");
+            return runtime.remoteAbility.call(providerAid, String(args.path), args.params);
+        }
+        case "ability_request_get": {
+            const providerAid = String(args.providerAid);
+            if (!runtime.allowedProviders?.includes(providerAid) || !runtime.remoteAbility)
+                throw new Error("Provider 未加入远程 Ability 白名单");
+            return runtime.remoteAbility.getRequest(providerAid, String(args.requestId));
+        }
+        case "coordination_dispatch": {
+            if (!allowedAgents || (allowedAgents.length > 0 && !allowedAgents.includes("*") && !allowedAgents.includes(String(args.agentId))) || !runtime.selfAid)
+                throw new Error("目标 Agent 未加入个人助理白名单");
+            const abilities = await management.listAgents();
+            const agent = abilities.find((item) => item.id === args.agentId && item.callable && item.aids.includes(String(args.aid)));
+            if (!agent)
+                throw new Error("目标 Agent/AID 当前不可派活");
+            const coordinator = (0, coordination_1.createCoordinator)({ selfAid: runtime.selfAid, ecPath: runtime.ecPath, allowedAgents: [{ id: agent.id, aids: agent.aids }] });
+            const result = await coordinator.dispatch({ agentId: agent.id, aid: String(args.aid), task: String(args.task) });
+            if (runtime.threads && runtime.threads.size >= 128)
+                runtime.threads.delete(runtime.threads.keys().next().value);
+            runtime.threads?.set(result.threadId, result.aid);
+            return result;
+        }
+        case "coordination_status": {
+            if (!runtime.selfAid)
+                throw new Error("个人助理 AID 未配置");
+            if (runtime.threads?.get(String(args.threadId)) !== String(args.aid))
+                throw new Error("thread 不属于本个人助理已派发的任务");
+            const coordinator = (0, coordination_1.createCoordinator)({ selfAid: runtime.selfAid, ecPath: runtime.ecPath, allowedAgents: [] });
+            return coordinator.getStatus({ aid: String(args.aid), threadId: String(args.threadId) });
+        }
+        case "owner_context_read": {
+            if (!runtime.ownerContext)
+                throw new Error("Owner 共享上下文尚未配置");
+            return runtime.ownerContext.get();
+        }
+    }
+}
+function parseExplicitCommand(task) {
+    if (!task.startsWith("/assistant ") && !task.startsWith("/sessions"))
+        return undefined;
+    if (/[\r\n]/.test(task))
+        return { kind: "invalid" };
+    if (task.startsWith("/sessions"))
+        return /^\/sessions(?:\s|$)/.test(task) ? { kind: "sessions", command: task } : { kind: "invalid" };
+    if (task === "/assistant context refresh")
+        return { kind: "context-refresh" };
+    const agentCall = task.match(/^\/assistant agent call ([a-z0-9][a-z0-9-]{0,63}) ([\s\S]{1,20000})$/);
+    if (agentCall)
+        return { kind: "agent-call", agentId: agentCall[1], task: agentCall[2] };
+    const create = task.match(/^\/assistant aid create ([A-Za-z0-9][A-Za-z0-9._-]{2,255})(?: --owner ([A-Za-z0-9][A-Za-z0-9._-]{2,255}))?$/);
+    if (create)
+        return { kind: "create", aid: create[1], ...(create[2] ? { ownerAid: create[2] } : {}) };
+    const bind = task.match(/^\/assistant aid bind ([A-Za-z0-9][A-Za-z0-9._-]{2,255}) ([A-Za-z0-9][A-Za-z0-9._-]{0,127})( --replace)?$/);
+    if (bind)
+        return { kind: "bind", aid: bind[1], agentId: bind[2], replace: Boolean(bind[3]) };
+    return { kind: "invalid" };
+}
+function stringifyToolResult(value) {
+    let text;
+    try {
+        text = typeof value === "string" ? value : JSON.stringify(value);
+    }
+    catch {
+        text = "工具返回结果不可序列化";
+    }
+    return text.length > 64_000 ? `${text.slice(0, 64_000)}\n[结果已截断]` : text;
+}

package/dist/src-ts/plugins/qwenwork/cdp-client.js

@@ -127,19 +127,40 @@ class QwenCdpClient {
         const reply = await this.direct(`这是后台健康检测。请只回复 ${token}，不要添加任何其它文字。`, 45_000, undefined, token);
         verifyQwenHeartbeatReply(token, reply);
     }
-    async createSession(_name) {
+    async createSession(_name, forceNew = false, preserveDraft = false) {
         await this.ensureVisible();
-        await this.recoverDraft();
+        if (!preserveDraft)
+            await this.recoverDraft();
         const snapshot = `(() => ({ url: location.href, messages: document.querySelectorAll('[data-message-id]').length, input: document.querySelector(${JSON.stringify(exports.inputSelector)})?.innerText ?? null, blank: Boolean(document.querySelector('.new-task-model-selector')) }))()`;
         const before = await this.evaluate(snapshot);
         if (before.input?.trim())
             throw new Error("千问输入框含有未发送内容，不能新建会话");
-        if (before.input !== null && before.messages === 0 && before.blank)
-            return;
-        const clicked = await this.evaluate(`(() => { const button = document.querySelector('button[aria-label="新任务"]'); if (!button || button.disabled || button.getAttribute('aria-disabled') === 'true') return false; button.click(); return true; })()`);
+        if (!forceNew && before.input !== null && before.messages === 0 && before.blank)
+            return this.currentSessionId();
+        const clicked = await this.evaluate(`(() => { if (document.querySelector(${JSON.stringify(exports.inputSelector)})?.innerText.trim() || document.querySelector('button[aria-label="停止"],button[aria-label="停止生成"],button[aria-label="Stop generating"]')) throw new Error('千问当前有未发送内容或正在生成回复，不能新建会话'); const button = document.querySelector('button[aria-label="新任务"]'); if (!button || button.disabled || button.getAttribute('aria-disabled') === 'true') return false; button.click(); return true; })()`);
         if (clicked !== true)
             throw new Error("千问新任务入口未找到或不可用");
         await this.waitFor(`(() => { const state = ${snapshot}; return state.input !== null && state.input.trim() === '' && state.messages === 0 && state.url !== ${JSON.stringify(before.url)}; })()`, "千问新会话状态校验失败");
+        return this.currentSessionId();
+    }
+    async currentSessionId() {
+        const href = String(await this.evaluate("location.href"));
+        try {
+            return new URL(href).searchParams.get("chat") ?? undefined;
+        }
+        catch {
+            return undefined;
+        }
+    }
+    async activateSession(sessionId) {
+        if (!sessionId.trim() || sessionId.length > 512 || /[\u0000-\u001f]/.test(sessionId))
+            throw new Error("千问模型会话标识无效");
+        await this.ensureVisible();
+        const target = new URL(String(await this.evaluate("location.href")));
+        target.searchParams.set("chat", sessionId);
+        if (target.href !== String(await this.evaluate("location.href")))
+            await this.evaluate(`(() => { if (document.querySelector(${JSON.stringify(exports.inputSelector)})?.innerText.trim() || document.querySelector('button[aria-label="停止"],button[aria-label="停止生成"],button[aria-label="Stop generating"]')) throw new Error('千问当前有未发送内容或正在生成回复，不能切换会话'); location.href = ${JSON.stringify(target.href)}; })()`);
+        await this.waitFor(`location.href === ${JSON.stringify(target.href)} && Boolean(document.querySelector(${JSON.stringify(exports.inputSelector)}))`, "千问个人助理模型会话恢复失败");
     }
     async connect() {
         if (this.connected())
@@ -204,7 +225,7 @@ class QwenCdpClient {
     async direct(text, timeoutMs = 240_000, signal, expectedReply) {
         return (await this.complete(text, timeoutMs, signal, expectedReply)).text;
     }
-    async complete(text, timeoutMs = 240_000, signal, expectedReply, attachments, onMessage) {
+    async complete(text, timeoutMs = 240_000, signal, expectedReply, attachments, onMessage, preserveDraft = false) {
         let stage = "准备千问办公桌面任务";
         let code = "QWEN_TASK_FAILED";
         try {
@@ -215,7 +236,8 @@ class QwenCdpClient {
                     throw new Error("千问输入框有用户草稿，跳过后台健康检测");
             }
             await this.ensureVisible();
-            await this.recoverDraft();
+            if (!preserveDraft)
+                await this.recoverDraft();
             const ready = await this.evaluate(`(() => { const input = document.querySelector(${JSON.stringify(exports.inputSelector)}); return Boolean(input && !input.innerText.trim() && !document.querySelector('button[aria-label="停止"],button[aria-label="停止生成"],button[aria-label="Stop generating"]')); })()`);
             if (!ready)
                 throw new Error("千问输入框含有未发送内容或正在生成回复，未发送新消息");
@@ -234,15 +256,45 @@ class QwenCdpClient {
             const addedIds = attachments?.length ? await this.injectAttachments(attachments) : [];
             stage = "向千问办公提交任务（尚未确认发送）";
             code = "QWEN_SUBMIT_FAILED";
+            let submitUrl = "";
+            let inserted = false;
+            let submitAttempted = false;
             try {
+                submitUrl = String(await this.evaluate("location.href"));
+                const readyToSubmit = `(() => {
+          const input = document.querySelector(${JSON.stringify(exports.inputSelector)});
+          const button = document.querySelector('button[aria-label="发送消息"]');
+          const generating = document.querySelector('button[aria-label="停止"],button[aria-label="停止生成"],button[aria-label="Stop generating"]');
+          return Boolean(!document.hidden && location.href === ${JSON.stringify(submitUrl)} && input && input.innerText.trim() === ${JSON.stringify(text.trim())}
+            && button && !button.disabled && button.getAttribute('aria-disabled') !== 'true' && !generating);
+        })()`;
                 await this.cdp("Input.insertText", { text });
-                const sent = await this.evaluate(`(() => { const input = document.querySelector(${JSON.stringify(exports.inputSelector)}); const button = document.querySelector('button[aria-label="发送消息"]'); if (!input || input.innerText.trim() !== ${JSON.stringify(text.trim())} || !button || button.disabled || button.getAttribute('aria-disabled') === 'true') return false; button.click(); return true; })()`);
-                if (sent !== true)
-                    throw new Error("千问发送按钮未就绪或输入校验失败，未发送消息");
+                inserted = true;
+                // 只有复查确认尚未点击时才再试一次，不能重发提交结果未知的任务。
+                for (let attempt = 0; attempt < 2; attempt++) {
+                    await this.waitFor(readyToSubmit, "千问发送按钮未就绪或输入校验失败，未发送消息");
+                    submitAttempted = true;
+                    const sent = await this.evaluate(`(() => {
+            if (!${readyToSubmit}) return false;
+            document.querySelector('button[aria-label="发送消息"]').click();
+            return true;
+          })()`);
+                    if (sent === true)
+                        break;
+                    submitAttempted = false;
+                    if (attempt === 1)
+                        throw new Error("千问发送按钮未就绪或输入校验失败，未发送消息");
+                }
                 await this.waitFor(`(() => { const input = document.querySelector(${JSON.stringify(exports.inputSelector)}); return Boolean(input && !input.innerText.trim()); })()`, "千问未确认消息已提交");
             }
             catch (error) {
                 // CDP 断开时清理也可能失败，保留最初的发送错误。
+                if (inserted && !submitAttempted) {
+                    try {
+                        await this.recoverDraft({ text, url: submitUrl });
+                    }
+                    catch { /* 清理失败不能覆盖原错。 */ }
+                }
                 try {
                     await this.cleanupAttachments(addedIds);
                 }
@@ -254,23 +306,38 @@ class QwenCdpClient {
             code = "QWEN_REPLY_FAILED";
             let previousReply = "";
             let stablePolls = 0;
+            let seenGenerating = false;
+            let stoppedPolls = 0;
             const forwarded = new Set();
             const deadline = Date.now() + timeoutMs;
-            while (Date.now() < deadline) {
-                await wait(2_000, signal);
+            while (Date.now() < deadline || signal?.aborted) {
+                // 上游取消不停止桌面生成；确认结束前保留插件队列。
+                await wait(2_000);
+                await this.restoreVisible();
                 const messagesJson = await this.evaluate(`JSON.stringify([...document.querySelectorAll(${JSON.stringify(exports.assistantMessageSelector)})].map(element => ({ id: element.getAttribute('data-message-id'), text: element.innerText })))`);
-                const fresh = freshAssistantMessages(beforeIds, JSON.parse(messagesJson || "[]"));
+                const messages = JSON.parse(messagesJson || "[]");
+                const fresh = freshAssistantMessages(beforeIds, messages);
                 const reply = fresh.at(-1)?.text.trim() ?? "";
                 stablePolls = reply && reply === previousReply ? stablePolls + 1 : 0;
                 previousReply = reply;
                 if (reply && stablePolls >= 3 && isQwenApiUnavailableReply(reply))
                     throw new Error("QWEN_REMOTE_ERROR：千问生成接口不可用；桌面仅返回通用错误，未提供服务端错误码；CDP 仍连接，未重启或重新提交任务");
                 const generating = await this.evaluate(`Boolean(document.querySelector('button[aria-label="停止"],button[aria-label="停止生成"],button[aria-label="Stop generating"]'))`);
+                seenGenerating ||= generating === true;
+                stoppedPolls = seenGenerating && generating === false ? stoppedPolls + 1 : 0;
+                if (signal?.aborted && !reply && stoppedPolls >= 2 && messages.some(message => message.id && !beforeIds.has(message.id))) {
+                    const attachments = expectedReply ? [] : await this.readArtifacts(beforeIds, sessionUrl);
+                    if (attachments.length)
+                        return { text: "", sessionId: new URL(sessionUrl).searchParams.get("chat") ?? undefined, attachments };
+                    code = "QWEN_REQUEST_CANCELLED";
+                    throw new Error("请求已取消，已确认本轮桌面生成结束，未返回正文或附件");
+                }
                 if (!generating && (expectedReply ? isQwenHeartbeatReplyComplete(reply, expectedReply) && stablePolls >= 1 : reply && stablePolls >= 3)) {
                     stage = "读取千问办公已完成任务的产物（尚未上传 AUN FS）";
                     code = "QWEN_ARTIFACT_READ_FAILED";
                     const attachments = expectedReply ? [] : await this.readArtifacts(beforeIds, sessionUrl);
-                    return { text: reply, ...(attachments.length ? { attachments } : {}) };
+                    const sessionId = new URL(sessionUrl).searchParams.get("chat") ?? undefined;
+                    return { text: reply, ...(sessionId ? { sessionId } : {}), ...(attachments.length ? { attachments } : {}) };
                 }
                 if (onMessage && generating && reply && stablePolls >= 1 && !forwarded.has(reply) && !isQwenApiUnavailableReply(reply)) {
                     await onMessage(reply);
@@ -510,9 +577,8 @@ class QwenCdpClient {
         try {
             await this.evaluate(`(async () => {
       if (document.querySelector('button[aria-label="停止"],button[aria-label="停止生成"],button[aria-label="Stop generating"]')) throw new Error('千问桌面端正在生成回复');
-      // 隐藏窗口会暂停消息渲染，使桌面端的提交前检查一直等不到确认。
-      if (document.hidden && typeof window.desktopApi?.focusMainWindow === 'function') await window.desktopApi.focusMainWindow();
     })()`);
+            await this.restoreVisible();
         }
         catch (error) {
             let permission;
@@ -525,8 +591,13 @@ class QwenCdpClient {
             throw error;
         }
     }
-    recoverDraft() {
-        return (0, cdp_draft_1.recoverCdpDraft)(this.evaluate.bind(this), exports.inputSelector, 'button[aria-label="停止"],button[aria-label="停止生成"],button[aria-label="Stop generating"]', this.options.dataDir);
+    async restoreVisible() {
+        await this.evaluate(`(async () => {
+      if (document.hidden && typeof window.desktopApi?.focusMainWindow === 'function') await window.desktopApi.focusMainWindow();
+    })()`);
+    }
+    recoverDraft(expected) {
+        return (0, cdp_draft_1.recoverCdpDraft)(this.evaluate.bind(this), exports.inputSelector, 'button[aria-label="停止"],button[aria-label="停止生成"],button[aria-label="Stop generating"]', this.options.dataDir, expected);
     }
     async cdp(method, params = {}, timeoutMs = this.options.timeoutMs) {
         if (!this.socket)
@@ -588,6 +659,7 @@ class QwenCdpClient {
     async waitFor(expression, message) {
         const deadline = Date.now() + this.options.timeoutMs;
         while (Date.now() < deadline) {
+            await this.restoreVisible();
             if (await this.evaluate(expression))
                 return;
             await wait(200);