mu-agent-bridge 1.0.34 → 1.0.39

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

@@ -39,11 +39,128 @@ Object.defineProperty(exports, "__esModule", { value: true });
 const strict_1 = __importDefault(require("node:assert/strict"));
 const node_test_1 = __importDefault(require("node:test"));
 const node_vm_1 = require("node:vm");
+const promises_1 = require("node:fs/promises");
+const node_os_1 = require("node:os");
+const node_path_1 = require("node:path");
 const doubao = __importStar(require("../src-ts/plugins/doubao/cdp-client"));
 const doubao_1 = require("../src-ts/plugins/doubao");
 const cdp_client_1 = require("../src-ts/plugins/doubao/cdp-client");
+const attachments_1 = require("../src-ts/plugins/doubao/attachments");
 const imageUrl = "https://p3-flow-imagex-sign.byteimg.com/tos-cn-i-a9rns2rl98/rc_gen_image/test.png?signature=test";
 const png = Buffer.from("iVBORw0KGgoAAAANSUhEUgAAAAEAAAABCAQAAAC1HAwCAAAAC0lEQVR42mP8/x8AAwMCAO+jRZkAAAAASUVORK5CYII=", "base64");
+function deferred() {
+    let resolve;
+    let reject;
+    const promise = new Promise((done, fail) => { resolve = done; reject = fail; });
+    return { promise, resolve, reject };
+}
+(0, node_test_1.default)("豆包原生上传保护已有草稿，并只清理本次唯一文件名", () => {
+    const items = [{ name: "marker.txt", uploadName: "unique-marker.txt", type: "text/plain", size: 6, base64: "bWFya2Vy" }];
+    const input = { dispatchEvent() { strict_1.default.equal(this.files[0].name, "unique-marker.txt"); } };
+    let draft = "未发送正文";
+    let area;
+    const page = { document: { querySelector(selector) { return selector.includes("contenteditable") ? { textContent: draft } : selector.includes("attachment_area") ? area : input; } },
+        atob, Uint8Array, Event, File: class {
+            name;
+            constructor(_parts, name) {
+                this.name = name;
+            }
+        },
+        DataTransfer: class {
+            files = [];
+            items = { add: (file) => this.files.push(file) };
+        },
+        MouseEvent: class {
+        }, };
+    strict_1.default.match((0, node_vm_1.runInNewContext)((0, attachments_1.buildUploadInjection)(items), page), /未发送草稿/);
+    draft = "";
+    strict_1.default.equal((0, node_vm_1.runInNewContext)((0, attachments_1.buildUploadInjection)(items), page), "started");
+    strict_1.default.equal(Object.hasOwn(input, "files"), false);
+    const deleted = [];
+    area = { __reactFiberTest: { memoizedProps: { attachmentStates: [{ fileName: "unique-marker.txt" }, { fileName: "用户新上传.txt" }] } }, querySelectorAll: () => [0, 1].map(index => ({ dispatchEvent() { deleted.push(index); } })) };
+    strict_1.default.match((0, node_vm_1.runInNewContext)((0, attachments_1.buildUploadInjection)(items), page), /未发送草稿/);
+    (0, node_vm_1.runInNewContext)(`(${attachments_1.cleanupUploadScript})(["unique-marker.txt"])`, page);
+    strict_1.default.deepEqual(deleted, [0]);
+});
+(0, node_test_1.default)("豆包附件按原生附件块传递文件和图片，保留独立文本消息", () => {
+    const uploaded = [
+        { type: "file", fileName: "marker.txt", fileKey: "tos/file.txt", localKey: "file-key", size: 27, parseState: 1 },
+        { type: "image", fileName: "colors.png", fileKey: "tos/image.png", localKey: "image-key", size: 100, imageWidth: 64, imageHeight: 64 },
+    ];
+    const body = JSON.parse(cdp_client_1.requestBody("读取附件", "conversation", "bot", [], uploaded));
+    strict_1.default.equal(body.messages.length, 2);
+    const block = body.messages[0].content_block[0];
+    strict_1.default.equal(block.block_type, 10052);
+    const [file, image] = block.content.attachment_block.attachments;
+    strict_1.default.deepEqual(file.file, { uri: "tos/file.txt", url: "", file_type: 0, name: "marker.txt", size: 27 });
+    strict_1.default.equal(file.type, 3);
+    strict_1.default.equal(file.parse_state, 1);
+    strict_1.default.equal(image.type, 1);
+    strict_1.default.equal(image.image.uri, "tos/image.png");
+    strict_1.default.equal(image.image.image_ori.width, 64);
+    strict_1.default.equal(body.messages[1].content_block[0].content.text_block.text, "读取附件");
+});
+(0, node_test_1.default)("豆包插件声明附件输入并向完整客户端传递附件，旧客户端明确拒绝", async () => {
+    const attachment = { kind: "file", data: Buffer.from("marker"), filename: "marker.txt", contentType: "text/plain" };
+    let received;
+    const client = { async connect() { }, async close() { }, connected: () => true, direct: async () => "文本", complete: async (_text, _signal, items) => { received = items; return { text: "已读取" }; } };
+    const plugin = new doubao_1.DoubaoPlugin(client);
+    try {
+        await plugin.activate({ config: {} });
+        strict_1.default.equal(plugin.attachmentInput, true);
+        await plugin.handle({ task: "读取附件", attachments: [attachment] });
+        strict_1.default.deepEqual(received, [attachment]);
+    }
+    finally {
+        await plugin.deactivate();
+    }
+    const legacy = new doubao_1.DoubaoPlugin({ ...client, complete: undefined });
+    try {
+        await legacy.activate({ config: {} });
+        await strict_1.default.rejects(legacy.handle({ task: "读取附件", attachments: [attachment] }), /不支持附件/);
+    }
+    finally {
+        await legacy.deactivate();
+    }
+});
+(0, node_test_1.default)("豆包在访问页面前明确拒绝视频与不支持的文件类型", async () => {
+    const client = new cdp_client_1.DoubaoCdpClient();
+    for (const [filename, contentType, error] of [["clip.mp4", "video/mp4", /不支持视频/], ["archive.zip", "application/zip", /不支持.*文件类型/]]) {
+        await strict_1.default.rejects(client.complete("读取附件", undefined, [{ kind: "file", data: Buffer.from("data"), filename, contentType }]), error);
+    }
+});
+(0, node_test_1.default)("豆包上传和清理同时失败时保留两个错误", async () => {
+    const client = new cdp_client_1.DoubaoCdpClient({ timeoutMs: 3000 });
+    let uploadName = "";
+    client.evaluate = async (expression) => {
+        if (expression.includes("const transfer = new DataTransfer")) {
+            uploadName = expression.match(/"uploadName":"([^"]+)"/)[1];
+            return "started";
+        }
+        if (expression.includes("names.includes"))
+            throw new Error("清理测试失败");
+        return [{ fileName: uploadName, size: 6, status: "Error" }];
+    };
+    await strict_1.default.rejects(client.uploadAttachments([{ kind: "file", data: Buffer.from("marker"), filename: "marker.txt", contentType: "text/plain" }], new AbortController().signal), (error) => {
+        strict_1.default.ok(error instanceof AggregateError);
+        strict_1.default.match(String(error.errors[0]), /上传或解析失败/);
+        strict_1.default.match(String(error.errors[1]), /清理测试失败/);
+        return true;
+    });
+});
+(0, node_test_1.default)("豆包上传启动阶段异常也清理本次附件", async () => {
+    const client = new cdp_client_1.DoubaoCdpClient();
+    let cleaned = false;
+    client.evaluate = async (expression) => {
+        if (expression.includes("names.includes")) {
+            cleaned = true;
+            return "cleaned";
+        }
+        throw new Error("上传启动失败");
+    };
+    await strict_1.default.rejects(client.uploadAttachments([{ kind: "file", data: Buffer.from("marker"), filename: "marker.txt", contentType: "text/plain" }], new AbortController().signal), /上传启动失败/);
+    strict_1.default.equal(cleaned, true);
+});
 (0, node_test_1.default)("豆包只提取生成完成的图片并去重，排除会话头像", () => {
     const extract = new Function(`return (${doubao.responseImagesScript})`)();
     const image = { status: 2, image_ori: { url: imageUrl, width: 2048, height: 2048 } };
@@ -51,6 +168,179 @@ const png = Buffer.from("iVBORw0KGgoAAAANSUhEUgAAAAEAAAABCAQAAAC1HAwCAAAAC0lEQVR
     strict_1.default.deepEqual(extract({ avatar: { image_ori: { url: "avatar" } }, patch_op: [{ patch_value: { content_block: [block] } }] }), [imageUrl]);
     strict_1.default.deepEqual(extract(null), []);
 });
+(0, node_test_1.default)("豆包只收集本轮原生块中完成的图片、文件和视频，不读取正文地址或本地路径", () => {
+    const extract = new Function(`return (${doubao.responseArtifactsScript})`)();
+    const image = { type: 1, id: "image-1", image: { status: 2, image_ori: { url: imageUrl } } };
+    const video = { type: 2, id: "video-1", video: { status: 3, vid: "video-id", download_url: "" } };
+    const imageBlock = { block_type: 2074, is_finish: true, content: { creation_block: { creations: [image, video, video, { ...video, id: "pending", video: { status: 2, vid: "pending-id" } }] } } };
+    const fileBlock = { block_type: 10020, block_id: "file-1", is_finish: true, content: { file_block: { name: "output.txt", uri: "tos/output.txt", url: "", path: "C:/private/output.txt", type: "txt" } } };
+    const result = extract({ patch_op: [{ patch_value: { content_block: [imageBlock, fileBlock, fileBlock, { ...fileBlock, is_finish: false }, { ...fileBlock, content: { file_block: { name: "private.txt", path: "C:/private/private.txt" } } }, { block_type: 10000, content: { text_block: { text: "https://example.com/fake.mp4" } } }] } }] });
+    strict_1.default.deepEqual(result, {
+        artifacts: [{ kind: "image", url: imageUrl }, { kind: "video", vid: "video-id" }, { kind: "file", filename: "output.txt", uri: "tos/output.txt" }],
+        pending: true,
+    });
+    strict_1.default.deepEqual(extract({ block_type: 10020, is_finish: true, content_obj: { name: "second.csv", url: "https://example.com/second.csv", type: "csv" } }).artifacts, [{ kind: "file", filename: "second.csv", url: "https://example.com/second.csv" }]);
+    strict_1.default.deepEqual(extract({ ...imageBlock, is_finish: false }).artifacts, []);
+    strict_1.default.deepEqual(extract({ ...fileBlock, is_finish: false }), { artifacts: [], pending: true });
+    strict_1.default.deepEqual(extract({ block_type: 2074, is_finish: true, content_obj: { creations: [{ type: 2, video: { status: 4, vid: "failed" } }] } }), { artifacts: [], pending: false, failed: true });
+    strict_1.default.deepEqual(extract(null), { artifacts: [], pending: false });
+});
+(0, node_test_1.default)("豆包异步产物只匹配同会话、本轮回复或原始问题引用", () => {
+    const file = { block_type: 10020, is_finish: true, content_obj: { name: "result.txt", uri: "own-file" } };
+    const question = "59158521582734851";
+    const messages = [
+        { message_id: "old", conversation_id: "own", ext: { reference_info: '{"referenced_message_id":1}' }, content_block: [file] },
+        { message_id: "other", conversation_id: "other", ext: { reference_info: `{"referenced_message_id":${question}}` }, content_block: [file] },
+        { message_id: "later", conversation_id: "own", ext: { reference_info: `{"referenced_message_id":${question}}` }, content_block: [file] },
+    ];
+    const page = { location: { pathname: "/chat/own" }, document: { querySelectorAll: () => messages.map(message => ({ __reactFiberTest: { memoizedProps: { message } } })) } };
+    const result = (0, node_vm_1.runInNewContext)(`(${doubao.currentResponseArtifactsScript})("own", ["${question}"], [])`, page);
+    strict_1.default.deepEqual(JSON.parse(JSON.stringify(result)), { artifacts: [{ kind: "file", filename: "result.txt", uri: "own-file" }], pending: false });
+    const unmatched = (0, node_vm_1.runInNewContext)(`(${doubao.currentResponseArtifactsScript})("own", ["different"], [])`, page);
+    strict_1.default.deepEqual(JSON.parse(JSON.stringify(unmatched)), { artifacts: [], pending: false });
+});
+(0, node_test_1.default)("豆包文件与视频下载限制可信域名、类型、字节和总预算", async () => {
+    const download = doubao.downloadDoubaoArtifact;
+    strict_1.default.equal(typeof download, "function");
+    const videoUrl = "https://v11-vdl.doubao.com/test.mp4";
+    const fileUrl = "https://p3-flow-sign.byteimg.com/test.txt";
+    const mp4 = Buffer.from("000000186674797069736f6d0000020069736f6d6d703431", "hex");
+    const fetchVideo = async (_url, init) => { strict_1.default.equal(init.redirect, "error"); return new Response(mp4, { headers: { "content-type": "video/mp4" } }); };
+    const video = await download({ kind: "video", url: videoUrl }, fetchVideo);
+    strict_1.default.equal(video.kind, "file");
+    strict_1.default.equal(video.contentType, "video/mp4");
+    strict_1.default.deepEqual(video.data, mp4);
+    strict_1.default.match(video.filename, /\.mp4$/);
+    const file = await download({ kind: "file", url: fileUrl, filename: "../result.txt" }, async () => new Response("marker", { headers: { "content-type": "text/plain;charset=utf-8" } }));
+    strict_1.default.equal(file.filename, "result.txt");
+    strict_1.default.equal(file.contentType, "text/plain");
+    for (const url of ["http://127.0.0.1/test", "https://example.com/test", "https://v11-vdl.doubao.com.evil.com/test", "https://user:pass@p3-flow-sign.byteimg.com/test", "https://p3-flow-sign.byteimg.com:444/test"]) {
+        await strict_1.default.rejects(download({ kind: "file", url, filename: "result.txt" }, () => { strict_1.default.fail("非法地址不得请求"); }), /地址/);
+    }
+    await strict_1.default.rejects(download({ kind: "video", url: videoUrl }, async () => new Response("html", { headers: { "content-type": "video/mp4" } })), /内容与类型/);
+    await strict_1.default.rejects(download({ kind: "video", url: videoUrl }, async () => new Response(mp4, { headers: { "content-type": "text/html" } })), /类型/);
+    await strict_1.default.rejects(download({ kind: "file", url: fileUrl, filename: "result.txt" }, async () => new Response("marker", { headers: { "content-type": "text/plain", "content-length": "40000000" } })), /过大/);
+    await strict_1.default.rejects(download({ kind: "file", url: fileUrl, filename: "result.txt" }, async () => new Response("marker", { headers: { "content-type": "text/plain" } }), undefined, 3), /过大/);
+});
+(0, node_test_1.default)("豆包完整回复等待本轮异步视频，并将图片、文件、视频统一回传", async () => {
+    const videoUrl = "https://v26-show.douyinvod.com/video.mp4", fileUrl = "https://p3-flow-sign.byteimg.com/output.txt";
+    const mp4 = Buffer.from("000000186674797069736f6d0000020069736f6d6d703431", "hex");
+    const image = { block_type: 2074, is_finish: true, content: { creation_block: { creations: [{ type: 1, image: { status: 2, image_ori: { url: imageUrl } } }] } } };
+    const file = { block_type: 10020, is_finish: true, content: { file_block: { uri: "own-file", name: "output.txt" } } };
+    const video = { block_type: 2074, is_finish: true, content_obj: { creations: [{ type: 2, video: { status: 3, vid: "own-video" } }] } };
+    const stream = [
+        ["STREAM_MSG_NOTIFY", { meta: { conversation_id: "other" }, content: { content_block: [{ ...file, content: { file_block: { uri: "other-file", name: "other.txt" } } }] } }],
+        ["SSE_ACK", { ack_client_meta: { conversation_id: "own" }, query_list: [{ question_id: "59158521582734851" }] }],
+        ["STREAM_MSG_NOTIFY", { meta: { message_id: "reply", conversation_id: "own", user_type: 2 }, content: { content_block: [image, file], ext: { has_video_gen: "1" }, tts_content: "已提交生成" } }],
+        ["SSE_REPLY_END", {}],
+    ].map(([event, data]) => `event: ${event}\ndata: ${JSON.stringify(data).replace(/"question_id":"(\d+)"/g, '"question_id":$1')}\n\n`).join("");
+    const calls = [];
+    const message = { message_id: "later", conversation_id: "own", ext: { reference_info: '{"referenced_message_id":59158521582734851}' }, content_block: [video] };
+    const page = { window: {}, navigator: { userAgent: "Chrome/147.0.0.0 SamanthaDoubao/2.32.6" }, location: { pathname: "/chat/own" }, performance: { getEntriesByType: () => [] }, URL, AbortController, TextDecoder,
+        document: { querySelectorAll: () => [{ __reactFiberTest: { memoizedProps: { message } } }] },
+        fetch: async (input, init) => {
+            const path = new URL(input).pathname;
+            if (path === "/chat/completion")
+                return new Response(stream);
+            calls.push(path);
+            if (path === "/alice/message/get_file_url") {
+                strict_1.default.deepEqual(JSON.parse(String(init.body)), { uris: ["own-file"], type: "file" });
+                return new Response(JSON.stringify({ code: 0, data: { file_urls: [{ uri: "own-file", main_url: fileUrl }] } }));
+            }
+            strict_1.default.equal(path, "/samantha/video/get_play_info");
+            strict_1.default.deepEqual(JSON.parse(String(init.body)), { vid: "own-video" });
+            return new Response(JSON.stringify({ code: 0, data: { play_infos: [{ main: videoUrl }] } }));
+        },
+    };
+    const client = new cdp_client_1.DoubaoCdpClient({ conversationId: "own", timeoutMs: 5000, fetchImpl: async (input) => new Response(String(input) === imageUrl ? png : String(input) === videoUrl ? mp4 : "marker", { headers: { "content-type": String(input) === imageUrl ? "image/png" : String(input) === videoUrl ? "video/mp4" : "text/plain" } }) });
+    client.connect = async () => { };
+    client.evaluate = async (expression) => (0, node_vm_1.runInNewContext)(expression, page);
+    const result = await client.complete("生成产物");
+    strict_1.default.equal(result.images.length, 1);
+    strict_1.default.deepEqual(result.attachments.map((item) => item.contentType).sort(), ["image/png", "text/plain", "video/mp4"]);
+    strict_1.default.deepEqual(calls.sort(), ["/alice/message/get_file_url", "/samantha/video/get_play_info"]);
+});
+(0, node_test_1.default)("豆包并发请求串行完成上传、回复和下载后再开始下一轮", { timeout: 10_000 }, async () => {
+    const uploadStarted = deferred(), uploadGate = deferred();
+    const requestStarted = deferred(), requestGate = deferred();
+    const downloadStarted = deferred(), downloadGate = deferred();
+    const uploads = [], requests = [];
+    const block = { block_type: 2074, is_finish: true, content: { creation_block: { creations: [{ image: { status: 2, image_ori: { url: imageUrl } } }] } } };
+    const page = {
+        window: {}, URL, AbortController, TextDecoder,
+        navigator: { userAgent: "Chrome/147.0.0.0 SamanthaDoubao/2.32.6" }, performance: { getEntriesByType: () => [] },
+        fetch: async (_url, init) => {
+            const text = JSON.parse(String(init.body)).messages.at(-1).content_block[0].content.text_block.text;
+            requests.push(text);
+            if (text === "请求甲") {
+                requestStarted.resolve();
+                await requestGate.promise;
+            }
+            return new Response(`event: STREAM_MSG_NOTIFY\ndata: ${JSON.stringify({ content: { content_block: text === "请求甲" ? [block] : [{ content: { text_block: { text: "第二轮回复" } } }] } })}\n\nevent: CHUNK_DELTA\ndata: ${JSON.stringify({ text: text === "请求甲" ? "第一轮回复" : "" })}\n\n`);
+        },
+    };
+    const client = new cdp_client_1.DoubaoCdpClient({ conversationId: "test", timeoutMs: 5000, fetchImpl: async () => {
+            downloadStarted.resolve();
+            await downloadGate.promise;
+            return new Response(png, { headers: { "content-type": "image/png" } });
+        } });
+    client.connect = async () => { };
+    client.evaluate = async (expression) => (0, node_vm_1.runInNewContext)(expression, page);
+    client.uploadAttachments = async (items) => {
+        uploads.push(items[0].filename);
+        if (items[0].filename === "甲.txt") {
+            uploadStarted.resolve();
+            await uploadGate.promise;
+        }
+        return [];
+    };
+    const attachment = (filename) => [{ kind: "file", data: Buffer.from("marker"), contentType: "text/plain", filename }];
+    const first = client.complete("请求甲", undefined, attachment("甲.txt"));
+    await uploadStarted.promise;
+    const second = client.complete("请求乙", undefined, attachment("乙.txt"));
+    await new Promise(setImmediate);
+    const duringUpload = { uploads: [...uploads], requests: [...requests] };
+    uploadGate.resolve();
+    await requestStarted.promise;
+    requestGate.resolve();
+    await downloadStarted.promise;
+    const duringDownload = { uploads: [...uploads], requests: [...requests] };
+    downloadGate.resolve();
+    const results = await Promise.all([first, second]);
+    strict_1.default.deepEqual(duringUpload, { uploads: ["甲.txt"], requests: [] });
+    strict_1.default.deepEqual(duringDownload, { uploads: ["甲.txt"], requests: ["请求甲"] });
+    strict_1.default.deepEqual(results.map(item => item.text), ["第一轮回复", "第二轮回复"]);
+    strict_1.default.equal(results[0].attachments.length, 1);
+    strict_1.default.equal(results[1].attachments, undefined);
+});
+for (const late of ["成功", "失败"]) {
+    (0, node_test_1.default)(`豆包前轮超时后的晚${late}回调不覆盖下一轮结果`, { timeout: 10_000 }, async () => {
+        const firstResponse = deferred(), nextResponse = deferred(), nextStarted = deferred();
+        let requests = 0;
+        const page = {
+            window: {}, URL, AbortController, TextDecoder,
+            navigator: { userAgent: "Chrome/147.0.0.0 SamanthaDoubao/2.32.6" }, performance: { getEntriesByType: () => [] },
+            fetch: () => { if (++requests === 1)
+                return firstResponse.promise; nextStarted.resolve(); return nextResponse.promise; },
+        };
+        const client = new cdp_client_1.DoubaoCdpClient({ conversationId: "test", timeoutMs: 20 });
+        client.connect = async () => { };
+        client.evaluate = async (expression) => (0, node_vm_1.runInNewContext)(expression, page);
+        await strict_1.default.rejects(client.complete("请求甲"), /等待豆包回复超时/);
+        const next = client.complete("请求乙");
+        await nextStarted.promise;
+        if (late === "成功")
+            firstResponse.resolve(new Response('event: CHUNK_DELTA\ndata: {"text":"过期回复"}\n\n'));
+        else
+            firstResponse.reject(new Error("过期错误"));
+        await new Promise(setImmediate);
+        const resultAfterLateCallback = page.window.__mab_doubao_result;
+        nextResponse.resolve(new Response('event: CHUNK_DELTA\ndata: {"text":"本轮回复"}\n\n'));
+        const result = await next;
+        strict_1.default.equal(resultAfterLateCallback, null);
+        strict_1.default.equal(result.text, "本轮回复");
+    });
+}
 (0, node_test_1.default)("豆包图片结果仍清理回复中重复的用户问题", () => {
     strict_1.default.equal(doubao.cleanDoubaoReply("生成宇航员橘猫图片已为你生成。", "生成宇航员橘猫图片"), "已为你生成。");
     strict_1.default.equal(doubao.cleanDoubaoReply("已生成图片。", "生成宇航员橘猫图片"), "已生成图片。");
@@ -62,8 +352,28 @@ const png = Buffer.from("iVBORw0KGgoAAAANSUhEUgAAAAEAAAABCAQAAAC1HAwCAAAAC0lEQVR
     const original = globalThis.WebSocket;
     const block = { block_type: 2074, is_finish: true, content: { creation_block: { creations: [{ image: { status: 2, image_ori: { url: imageUrl } } }] } } };
     let stream = `event: STREAM_CHUNK\ndata: ${JSON.stringify({ patch_op: [{ patch_value: { content_block: [block], tts_content: "生成宇航员橘猫图片已为你生成。" } }] })}\n\nevent: SSE_REPLY_END\ndata: {}\n\n`;
-    const page = { window: {}, performance: { getEntriesByType: () => [] }, URL, AbortController, TextDecoder, fetch: async () => new Response(stream) };
+    const requests = [];
+    const bodies = [];
+    let draft = "", messages = 2;
+    const page = {
+        location: { pathname: "/chat/123", href: "chrome://doubao-chat/chat/123" },
+        document: {
+            querySelector: (selector) => selector.includes("create_conversation_button")
+                ? { getAttribute() { return null; }, click() { page.location.pathname = "/chat"; messages = 0; } }
+                : { get innerText() { return draft; } },
+            querySelectorAll: () => Array.from({ length: messages }),
+        },
+        window: {}, navigator: { userAgent: "Chrome/147.0.0.0 SamanthaDoubao/2.32.6" },
+        performance: { getEntriesByType: () => [
+                { name: "https://www.doubao.com/settings?pc_version=2.32.6&device_id=current-device&tea_uuid=current-device&web_id=current-web&web_tab_id=current-tab" },
+                { name: "https://www.doubao.com/chat/completion?pc_version=2.30.4&device_id=old-device" },
+                { name: "https://evil.example/chat/completion?device_id=untrusted" },
+            ] },
+        URL, AbortController, TextDecoder,
+        fetch: async (address, options) => { requests.push(new URL(address)); bodies.push(JSON.parse(options.body)); return new Response(stream); },
+    };
     class TestWebSocket {
+        readyState = 1;
         onopen;
         onmessage;
         constructor() { queueMicrotask(() => this.onopen?.()); }
@@ -98,6 +408,12 @@ const png = Buffer.from("iVBORw0KGgoAAAANSUhEUgAAAAEAAAABCAQAAAC1HAwCAAAAC0lEQVR
         strict_1.default.deepEqual(Buffer.from(result.images[0].data), png);
         strict_1.default.equal(result.images[0].contentType, "image/png");
         strict_1.default.equal(downloads, 1);
+        strict_1.default.equal(requests[0].origin, "https://www.doubao.com");
+        strict_1.default.equal(requests[0].searchParams.get("pc_version"), "2.32.6");
+        strict_1.default.equal(requests[0].searchParams.get("doubao_pc_version"), "2.32.6");
+        strict_1.default.equal(requests[0].searchParams.get("device_id"), "current-device");
+        strict_1.default.equal(requests[0].searchParams.get("web_id"), "current-web");
+        page.navigator.userAgent = "Chrome/148.0.1.2 SamanthaDoubao/2.33.0";
         const chunks = ["天空是蓝色。海水也是", "蓝色", "，这是光散射的结果。"];
         for (const event of ["CHUNK_DELTA", "STREAM_CHUNK"]) {
             stream = chunks.map(text => `event: ${event}\ndata: ${JSON.stringify(event === "CHUNK_DELTA" ? { text } : { patch_op: [{ patch_value: { tts_content: text } }] })}\n\n`).join("");
@@ -105,7 +421,27 @@ const png = Buffer.from("iVBORw0KGgoAAAANSUhEUgAAAAEAAAABCAQAAAC1HAwCAAAAC0lEQVR
             const reply = await client.complete("请继续");
             strict_1.default.equal(reply.text, chunks.join(""), event);
             strict_1.default.equal(reply.images?.length, 1);
+            strict_1.default.equal(requests.at(-1)?.searchParams.get("pc_version"), "2.33.0");
+            strict_1.default.equal(requests.at(-1)?.searchParams.get("chromium_version"), "148.0.1.2");
         }
+        strict_1.default.equal(typeof client.createSession, "function");
+        draft = "用户尚未发送的内容";
+        await strict_1.default.rejects(client.createSession(), /未发送内容/);
+        draft = "";
+        const count = bodies.length;
+        await client.createSession();
+        strict_1.default.equal(page.location.pathname, "/chat");
+        strict_1.default.equal(bodies.length, count, "新建入口不发送占位消息");
+        stream = 'event: SSE_ACK\ndata: {"ack_client_meta":{"conversation_id":"456"}}\n\nevent: CHUNK_DELTA\ndata: {"text":"新会话已回复"}\n\n';
+        const fresh = await client.complete("新会话首条消息", undefined, undefined, "");
+        strict_1.default.equal(fresh.sessionId, "456");
+        strict_1.default.equal(bodies.at(-1).client_meta.conversation_id, "");
+        strict_1.default.match(bodies.at(-1).client_meta.local_conversation_id, /^local_/);
+        strict_1.default.equal(bodies.at(-1).option.need_create_conversation, true);
+        strict_1.default.equal(bodies.at(-1).option.conversation_init_option.need_ack_conversation, true);
+        await client.complete("继续回复", undefined, undefined, fresh.sessionId);
+        strict_1.default.equal(bodies.at(-1).client_meta.conversation_id, "456");
+        strict_1.default.equal(bodies.at(-1).option.need_create_conversation, false);
     }
     finally {
         await client.close();
@@ -177,10 +513,30 @@ const png = Buffer.from("iVBORw0KGgoAAAANSUhEUgAAAAEAAAABCAQAAAC1HAwCAAAAC0lEQVR
 (0, node_test_1.default)("豆包错误兼容旧版 code 字段", () => {
     strict_1.default.equal((0, cdp_client_1.doubaoStreamErrorMessage)(JSON.stringify({ code: "E_LIMIT" })), "豆包平台返回流式错误（错误码 E_LIMIT）");
 });
+(0, node_test_1.default)("豆包风控滑块验证优先于 rate limited 文案，且不泄露验证令牌", () => {
+    const decision = { type: "verify", subtype: "slide", detail: "private-challenge", log_id: "private-log" };
+    for (const value of [decision, JSON.stringify(decision)]) {
+        const message = (0, cdp_client_1.doubaoStreamErrorMessage)(JSON.stringify({ error_code: 710022004, error_msg: "rate limited", extra: { decision: value } }));
+        strict_1.default.match(message, /710022004.*滑块验证.*豆包桌面端/);
+        strict_1.default.doesNotMatch(message, /限流|rate limited|private-/);
+    }
+    strict_1.default.doesNotThrow(() => (0, cdp_client_1.doubaoStreamErrorMessage)('{"extra":{"decision":"无效 JSON"}}'));
+});
+(0, node_test_1.default)("豆包新建会话协议区分空会话和已有会话", () => {
+    const fresh = JSON.parse((0, cdp_client_1.requestBody)("你好", "", "bot"));
+    strict_1.default.equal(fresh.option.need_create_conversation, true);
+    strict_1.default.equal(fresh.option.conversation_init_option.need_ack_conversation, true);
+    strict_1.default.match(fresh.client_meta.local_conversation_id, /^local_/);
+    strict_1.default.notEqual(fresh.client_meta.local_conversation_id, JSON.parse((0, cdp_client_1.requestBody)("你好", "", "bot")).client_meta.local_conversation_id);
+    const existing = JSON.parse((0, cdp_client_1.requestBody)("继续", "123", "bot"));
+    strict_1.default.equal(existing.option.need_create_conversation, false);
+    strict_1.default.equal(existing.client_meta.local_conversation_id, undefined);
+});
 (0, node_test_1.default)("停止豆包客户端时向页面发送请求取消指令", async () => {
     const original = globalThis.WebSocket;
     const messages = [];
     class TestWebSocket {
+        readyState = 1;
         onopen;
         onmessage;
         constructor(_url) { queueMicrotask(() => this.onopen?.()); }
@@ -198,7 +554,7 @@ const png = Buffer.from("iVBORw0KGgoAAAANSUhEUgAAAAEAAAABCAQAAAC1HAwCAAAAC0lEQVR
         const pending = client.direct("停止测试").catch((error) => String(error));
         await new Promise((resolve) => setImmediate(resolve));
         await client.cancel();
-        strict_1.default.match(messages.map((message) => JSON.parse(message).params?.expression ?? "").join("\n"), /new AbortController\(\).*signal: window\.__mab_doubao_controller\.signal/s);
+        strict_1.default.match(messages.map((message) => JSON.parse(message).params?.expression ?? "").join("\n"), /new AbortController\(\).*signal: controller\.signal/s);
         strict_1.default.match(messages.at(-1) ?? "", /__mab_doubao_controller\?\.abort\(\)/);
         strict_1.default.match(await pending, /请求已取消/);
         client.close();
@@ -207,3 +563,194 @@ const png = Buffer.from("iVBORw0KGgoAAAANSUhEUgAAAAEAAAABCAQAAAC1HAwCAAAAC0lEQVR
         globalThis.WebSocket = original;
     }
 });
+(0, node_test_1.default)("豆包并发连接复用同一次页面发现和 WebSocket，活动命令保持可用", async () => {
+    const original = globalThis.WebSocket;
+    const sockets = [], secondDiscovery = deferred();
+    let discoveries = 0, finishActive;
+    class TestWebSocket {
+        readyState = 0;
+        onopen;
+        onclose;
+        onmessage;
+        constructor(_url) { sockets.push(this); if (sockets.length === 1)
+            queueMicrotask(() => this.open()); }
+        open() { this.readyState = 1; this.onopen?.(); }
+        send(raw) {
+            const request = JSON.parse(raw);
+            const reply = () => this.onmessage?.({ data: JSON.stringify({ id: request.id, result: { result: { value: "活动页面回复" } } }) });
+            if (request.params.expression === "活动命令")
+                finishActive = reply;
+            else
+                queueMicrotask(reply);
+        }
+        close() { this.readyState = 3; this.onclose?.(); }
+    }
+    globalThis.WebSocket = TestWebSocket;
+    const client = new cdp_client_1.DoubaoCdpClient({ fetchImpl: async () => {
+            if (++discoveries === 2)
+                await secondDiscovery.promise;
+            return new Response(JSON.stringify([{ type: "page", url: "doubao://doubao-chat/chat/test", webSocketDebuggerUrl: "ws://test" }]));
+        } });
+    try {
+        const first = client.connect(), second = client.connect();
+        await first;
+        const active = client.evaluate("活动命令");
+        secondDiscovery.resolve();
+        await new Promise(setImmediate);
+        const readyDuringReconnect = client.connected();
+        const followup = client.evaluate("后续命令").catch((error) => error.message);
+        sockets[1]?.open();
+        await second;
+        finishActive();
+        strict_1.default.equal(await active, "活动页面回复");
+        strict_1.default.equal(await followup, "活动页面回复");
+        strict_1.default.equal(readyDuringReconnect, true);
+        strict_1.default.equal(discoveries, 1);
+        strict_1.default.equal(sockets.length, 1);
+    }
+    finally {
+        await client.close();
+        globalThis.WebSocket = original;
+    }
+});
+(0, node_test_1.default)("豆包更新断开 CDP 后清理待响应命令，下次请求连接新页面", async () => {
+    const original = globalThis.WebSocket;
+    const sockets = [];
+    const prompts = [];
+    class TestWebSocket {
+        url;
+        readyState = 0;
+        onopen;
+        onclose;
+        onmessage;
+        constructor(url) {
+            this.url = url;
+            sockets.push(this);
+            queueMicrotask(() => { this.readyState = 1; this.onopen?.(); });
+        }
+        send(raw) {
+            const request = JSON.parse(raw);
+            const expression = request.params.expression;
+            if (expression === "等待旧页面")
+                return;
+            if (expression.includes("new AbortController()"))
+                prompts.push(expression);
+            const value = expression === "window.__mab_doubao_result" ? JSON.stringify({ ok: true, reply: "连接已恢复" }) : "started";
+            queueMicrotask(() => this.onmessage?.({ data: JSON.stringify({ id: request.id, result: { result: { value } } }) }));
+        }
+        close() { this.readyState = 3; this.onclose?.(); }
+    }
+    globalThis.WebSocket = TestWebSocket;
+    let discoveries = 0;
+    const client = new cdp_client_1.DoubaoCdpClient({ timeoutMs: 2000, fetchImpl: async () => new Response(JSON.stringify([
+            { type: "page", url: "doubao://doubao-chat/chat/test", webSocketDebuggerUrl: `ws://page-${++discoveries}` },
+        ])) });
+    try {
+        await client.connect();
+        const pending = client.evaluate("等待旧页面");
+        const rejected = strict_1.default.rejects(pending, /豆包 CDP 已断开/);
+        sockets[0].close();
+        strict_1.default.equal(client.connected(), false);
+        await rejected;
+        strict_1.default.equal(await client.direct("更新后验证"), "连接已恢复");
+        strict_1.default.equal(sockets[1].url, "ws://page-2");
+        strict_1.default.equal(client.connected(), true);
+        strict_1.default.equal(prompts.length, 1);
+    }
+    finally {
+        await client.close();
+        globalThis.WebSocket = original;
+    }
+});
+(0, node_test_1.default)("豆包插件在下一次任务前重新执行连接和启动流程", async () => {
+    let connected = false, connections = 0;
+    const client = {
+        async connect() { connections++; connected = true; },
+        async close() { connected = false; }, connected: () => connected,
+        async direct() { strict_1.default.ok(connected, "发送前应恢复连接"); return "新版已连接"; },
+    };
+    const plugin = new doubao_1.DoubaoPlugin(client);
+    try {
+        await plugin.activate({ config: {} });
+        connected = false;
+        strict_1.default.equal((await plugin.health()).status, "degraded");
+        strict_1.default.deepEqual(await plugin.handle({ task: "升级后测试" }), { text: "新版已连接" });
+        strict_1.default.equal(connections, 2);
+        strict_1.default.equal((await plugin.health()).status, "ready");
+    }
+    finally {
+        await plugin.deactivate();
+    }
+});
+(0, node_test_1.default)("豆包 /new 切换独立会话，重载后仍使用服务端会话标识", async () => {
+    const dataDir = await (0, promises_1.mkdtemp)((0, node_path_1.join)((0, node_os_1.tmpdir)(), "doubao-sessions-"));
+    const targets = [];
+    let created = 0, failCreate = false;
+    const client = {
+        async connect() { }, async close() { }, connected: () => true, async direct() { return "旧接口"; },
+        async createSession() { if (failCreate)
+            throw new Error("未发送内容"); created++; },
+        async complete(_text, _signal, _attachments, conversationId) {
+            targets.push(conversationId);
+            return { text: "回复", sessionId: conversationId || String(100 + targets.length) };
+        },
+    };
+    let plugin = new doubao_1.DoubaoPlugin(client);
+    try {
+        await plugin.activate({ config: {}, dataDir });
+        strict_1.default.deepEqual(plugin.sessionOperations, ["new"]);
+        const first = await plugin.handle({ task: "首条消息", sessionId: "peer-a" });
+        strict_1.default.equal(first.sessionId, "peer-a");
+        strict_1.default.equal(targets.at(-1), "");
+        await plugin.handle({ task: "继续", sessionId: "peer-a" });
+        strict_1.default.equal(targets.at(-1), "101");
+        strict_1.default.deepEqual(await plugin.controlSession({ operation: "new", sessionId: "peer-b", name: "新会话" }), { text: "豆包已创建新会话", sessionId: "peer-b" });
+        strict_1.default.equal(created, 1);
+        await plugin.deactivate();
+        plugin = new doubao_1.DoubaoPlugin(client);
+        await plugin.activate({ config: {}, dataDir });
+        await plugin.handle({ task: "新会话消息", sessionId: "peer-b", newSession: true });
+        strict_1.default.equal(targets.at(-1), "");
+        await plugin.handle({ task: "继续新会话", sessionId: "peer-b" });
+        strict_1.default.equal(targets.at(-1), "103");
+        await plugin.handle({ task: "回到旧会话", sessionId: "peer-a" });
+        strict_1.default.equal(targets.at(-1), "101");
+        failCreate = true;
+        await strict_1.default.rejects(plugin.controlSession({ operation: "new", sessionId: "peer-c" }), /未发送内容/);
+        await plugin.handle({ task: "失败后继续原会话", sessionId: "peer-b" });
+        strict_1.default.equal(targets.at(-1), "103");
+        const saved = await (0, promises_1.readFile)((0, node_path_1.join)(dataDir, "sessions.json"), "utf8");
+        strict_1.default.doesNotMatch(saved, /首条消息|新会话消息|回到旧会话/);
+    }
+    finally {
+        await plugin.deactivate();
+        await (0, promises_1.rm)(dataDir, { recursive: true, force: true });
+    }
+});
+(0, node_test_1.default)("豆包跨来源请求串行执行，忙时不允许新建会话", async () => {
+    let release;
+    const blocked = new Promise(resolve => { release = resolve; });
+    let started;
+    const running = new Promise(resolve => { started = resolve; });
+    let calls = 0, active = 0, peak = 0;
+    const plugin = new doubao_1.DoubaoPlugin({
+        async connect() { }, async close() { }, connected: () => true, async direct() { return ""; }, async createSession() { },
+        async complete() { calls++; active++; peak = Math.max(peak, active); started(); if (calls === 1)
+            await blocked; active--; return { text: "回复", sessionId: String(calls) }; },
+    });
+    try {
+        await plugin.activate({ config: {} });
+        const first = plugin.handle({ task: "一", sessionId: "a" });
+        const second = plugin.handle({ task: "二", sessionId: "b" });
+        await running;
+        await strict_1.default.rejects(plugin.controlSession({ operation: "new", sessionId: "c" }), /当前忙/);
+        release();
+        await Promise.all([first, second]);
+        strict_1.default.equal(peak, 1);
+        strict_1.default.equal(calls, 2);
+    }
+    finally {
+        release();
+        await plugin.deactivate();
+    }
+});