mu-agent-bridge 1.0.34 → 1.0.39

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

@@ -0,0 +1,172 @@
+"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_test_1 = __importDefault(require("node:test"));
+const node_vm_1 = require("node:vm");
+const session_history_1 = require("../src-ts/plugins/doubao/session-history");
+const cdp_client_1 = require("../src-ts/plugins/doubao/cdp-client");
+const rawConversation = {
+    conversation_id: "conv-1", name: "今天的问题", create_time: "1792555702", update_time: "1792555770",
+    latest_index: "2", messages: [
+        { message_id: "user-1", user_type: 2, index_in_conv: "1", create_time: "1792555702", tts_content: "修复图片回传" },
+        { message_id: "assistant-1", user_type: 1, index_in_conv: "2", create_time: "1792555770", tts_content: "已修复并验证" },
+    ],
+};
+(0, node_test_1.default)("豆包只读历史脚本解析原生会话目录、Unix 时间与角色文本", async () => {
+    const require = (id) => {
+        strict_1.default.equal(id, 55807);
+        return { V: async (query) => {
+                strict_1.default.equal(query.conv_version, "cursor-1");
+                strict_1.default.equal(query.message_count_per_conv, 100);
+                return { code: 0, data: { cells: [{ conversation: rawConversation }], has_more: false, next_cursor: "" } };
+            } };
+    };
+    const chunks = [];
+    chunks.push = (...items) => {
+        for (const item of items) {
+            const install = item[2];
+            install(require);
+        }
+        return items.length;
+    };
+    const window = { "@flow-web/desktop:stable": chunks };
+    const execute = (0, node_vm_1.runInNewContext)(`(${cdp_client_1.doubaoHistoryPageScript})`, { window });
+    const page = await execute("cursor-1", 20);
+    strict_1.default.equal(page.complete, true);
+    strict_1.default.equal(JSON.stringify(page.conversations[0].messages.map(({ id, role, text }) => ({ id, role, text }))), JSON.stringify([
+        { id: "user-1", role: "user", text: "修复图片回传" },
+        { id: "assistant-1", role: "assistant", text: "已修复并验证" },
+    ]));
+    strict_1.default.equal(page.conversations[0].messages[0].time, new Date(1_792_555_702_000).toISOString());
+    strict_1.default.equal(JSON.stringify(page.missingReasons), "[]");
+});
+(0, node_test_1.default)("豆包历史 provider 使用原生正文游标分页并校验会话 ID", async () => {
+    let next;
+    const client = {
+        connected: () => true,
+        listHistory: async (cursor) => {
+            next = cursor;
+            return { conversations: [{ id: "conv-1", title: "会话", messages: [], truncated: true }], nextCursor: "next", complete: false, missingReasons: [] };
+        },
+        readHistory: async (id, cursor) => {
+            strict_1.default.equal(id, "conv-1");
+            strict_1.default.equal(cursor, undefined);
+            return { conversationId: id, messages: [{ id: "m-1", role: "user", text: "正文", time: new Date().toISOString() }], nextCursor: "42", hasMore: true, complete: true, missingReasons: [] };
+        },
+    };
+    const provider = new session_history_1.DoubaoSessionHistoryProvider({ scope: "doubao:9227", client: () => client });
+    strict_1.default.equal(provider.mode, "all");
+    const listed = await provider.listSessions({ limit: 20, cursor: "before" });
+    strict_1.default.equal(next, "before");
+    strict_1.default.equal(listed.supported && listed.nextCursor, "next");
+    strict_1.default.equal(listed.supported && listed.complete, false);
+    const result = await provider.readSession({ sessionId: "conv-1", limit: 100 });
+    strict_1.default.equal(result.supported && result.complete, true);
+    strict_1.default.equal(result.supported && result.nextCursor, "42");
+    strict_1.default.equal(result.supported && result.turns.length, 1);
+});
+(0, node_test_1.default)("豆包正文分页脚本使用原生单会话链游标并排除思考字段", async () => {
+    let nativeQuery;
+    const require = (id) => {
+        strict_1.default.equal(id, 77049);
+        return { J: async (query) => {
+                nativeQuery = query;
+                return { code: 0, data: { messages: [
+                            { conversation_id: "conv-1", message_id: "m-1", user_type: 1, create_time: "1792555770", thinking_content: "不可见思考", content_block: [{ thinking_content: { text_block: { text: "不可见嵌套思考" } }, reasoning_block: { text_block: { text: "不可见推理" } }, text_block: { text: "可见正文" } }] },
+                        ], has_more: true, next_index: "41" } };
+            } };
+    };
+    const chunks = [];
+    chunks.push = (...items) => { for (const item of items)
+        item[2](require); return items.length; };
+    const execute = (0, node_vm_1.runInNewContext)(`(${cdp_client_1.doubaoHistoryReadPageScript})`, { window: { "@flow-web/desktop:stable": chunks } });
+    const page = await execute("conv-1", "50", 50);
+    strict_1.default.equal(page.error, undefined);
+    strict_1.default.equal(nativeQuery?.conversation_id, "conv-1");
+    strict_1.default.equal(nativeQuery?.anchor_index, 50);
+    strict_1.default.equal(nativeQuery?.direction, 1);
+    strict_1.default.equal(JSON.stringify(page.messages.map(({ id, role, text }) => ({ id, role, text }))), JSON.stringify([{ id: "m-1", role: "assistant", text: "可见正文" }]));
+    strict_1.default.equal(page.nextCursor, "41");
+    strict_1.default.equal(page.complete, true);
+});
+(0, node_test_1.default)("豆包今日正文先按时间排除旧附件消息，未知时间仍报告不完整", async () => {
+    const messages = [
+        { conversation_id: "conv-1", message_id: "m-today", user_type: 1, create_time: "1792555770", tts_content: "今天的答复" },
+        { conversation_id: "conv-1", message_id: "m-old", user_type: 2, create_time: "1792497638", content_block: [{ content: { thinking_block: {}, creation_block: { creations: [{ image: {} }] } } }] },
+    ];
+    const require = (_id) => ({ J: async () => ({ code: 0, data: { messages, has_more: true, next_index: "41" } }) });
+    const chunks = [];
+    chunks.push = (...items) => { for (const item of items)
+        item[2](require); return items.length; };
+    const execute = (0, node_vm_1.runInNewContext)(`(${cdp_client_1.doubaoHistoryReadPageScript})`, { window: { "@flow-web/desktop:stable": chunks } });
+    const page = await execute("conv-1", "", 100, "2026-10-21T00:00:00+08:00");
+    strict_1.default.equal(page.complete, true);
+    strict_1.default.equal(page.reachedSince, true);
+    strict_1.default.equal(page.messages.length, 1);
+    strict_1.default.equal(page.messages[0].text, "今天的答复");
+    messages.push({ conversation_id: "conv-1", message_id: "m-unknown", user_type: 1, thinking_content: "仅思考" });
+    const incomplete = await execute("conv-1", "", 100, "2026-10-21T00:00:00+08:00");
+    strict_1.default.equal(incomplete.complete, false);
+    strict_1.default.ok(incomplete.missingReasons.some(reason => reason.includes("时间")));
+});
+(0, node_test_1.default)("豆包今日 provider 转交起点并在原生已越过起点时结束正文分页", async () => {
+    const since = "2026-10-21T00:00:00+08:00";
+    const client = {
+        connected: () => true,
+        listHistory: async () => ({ conversations: [{ id: "conv-1", title: "会话", messages: [], truncated: true }], complete: true, missingReasons: [] }),
+        readHistory: async (id, _cursor, _limit, nativeSince) => {
+            strict_1.default.equal(nativeSince, since);
+            return { conversationId: id, messages: [{ id: "m-1", role: "assistant", text: "今日答复", time: "2026-10-21T01:00:00Z" }], hasMore: true, nextCursor: "42", reachedSince: true, complete: true, missingReasons: [] };
+        },
+    };
+    const provider = new session_history_1.DoubaoSessionHistoryProvider({ scope: "doubao:9227", client: () => client });
+    await provider.listSessions({ limit: 20 });
+    const result = await provider.readSession({ sessionId: "conv-1", limit: 100, since });
+    strict_1.default.equal(result.supported && result.complete, true);
+    strict_1.default.equal(result.supported && result.nextCursor, undefined);
+    strict_1.default.equal(result.supported && result.turns.length, 1);
+});
+(0, node_test_1.default)("豆包历史脚本重复调用兼容 webpack 已安装 chunk 缓存", async () => {
+    const installed = new Set();
+    let calls = 0;
+    const require = (_id) => ({ V: async () => {
+            calls++;
+            return { code: 0, data: { cells: [{ conversation: rawConversation }], has_more: false } };
+        } });
+    const chunks = [];
+    chunks.push = (...items) => {
+        for (const item of items) {
+            const [ids, , install] = item;
+            // webpack 仅在存在未安装的 chunk 时执行 runtime callback。
+            if (ids.some(id => !installed.has(id)))
+                install(require);
+            for (const id of ids)
+                installed.add(id);
+        }
+        return items.length;
+    };
+    const execute = (0, node_vm_1.runInNewContext)(`(${cdp_client_1.doubaoHistoryPageScript})`, { window: { "@flow-web/desktop:stable": chunks } });
+    const first = await execute("", 20);
+    const second = await execute("", 20);
+    strict_1.default.equal(first.error, undefined);
+    strict_1.default.equal(second.error, undefined);
+    strict_1.default.equal(second.conversations?.length, 1);
+    strict_1.default.equal(calls, 2);
+});
+(0, node_test_1.default)("豆包历史目录 has_more 但游标缺失时报告不完整", async () => {
+    const require = (_id) => ({ V: async () => ({ code: 0, data: { cells: [], has_more: true, next_cursor: "" } }) });
+    const chunks = [];
+    chunks.push = (...items) => {
+        for (const item of items)
+            item[2](require);
+        return items.length;
+    };
+    const execute = (0, node_vm_1.runInNewContext)(`(${cdp_client_1.doubaoHistoryPageScript})`, { window: { "@flow-web/desktop:stable": chunks } });
+    const page = await execute("", 20);
+    strict_1.default.equal(page.complete, false);
+    strict_1.default.equal(page.nextCursor, undefined);
+    strict_1.default.ok(page.missingReasons.some(reason => reason.includes("游标")));
+});