verify-answer-stream.mjs 30 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616
  1. /**
  2. * 验证 F073「正文真实增量流式输出」的两块:
  3. * 第 1 部分 —— **纯逻辑**(不必联网):增量规整器的前缀不变式 + 流状态机
  4. * 第 2 部分 —— **协调器级假流**:假 SSE 流喂真实协调器,断言事件通道与去重
  5. *
  6. * 为什么规整器要这么较真:它手里攥着一个不变式——
  7. * **任一时刻「已发出的增量拼接」=== `normalizeChatMarkdown(已输入的全部文本)`**
  8. * 只要不变式成立,后端 `answer_end` 的「整块替换」在正常路径上就是纯尾缀追加
  9. * (不会触发整段重建 → 不会让「思考中」卡片闪现重播)。破了它,观感立刻劣化。
  10. *
  11. * 怎么跑(先重新打包,必须带 --loader:.png=dataurl):
  12. * npx esbuild harness/tools/_entry-coordinator.ts --bundle --format=esm \
  13. * --outfile=harness/tools/_coordinator.mjs --alias:@=./src \
  14. * --define:import.meta.env='{}' --loader:.png=dataurl
  15. * node harness/tools/verify-answer-stream.mjs
  16. */
  17. import {
  18. ApiChatCoordinator,
  19. applyAnswerStreamAbort,
  20. applyAnswerStreamEnd,
  21. applyAnswerStreamEvent,
  22. createAnswerStreamState,
  23. composeStreamedContent,
  24. createIncrementalMarkdownNormalizer,
  25. normalizeChatMarkdown,
  26. parseAnswerStreamEvent,
  27. shouldKeepAnswerOverview,
  28. } from './_coordinator.mjs';
  29. let pass = 0;
  30. let fail = 0;
  31. const check = (name, cond, extra = '') => {
  32. if (cond) {
  33. pass++;
  34. console.log(` ok ${name}`);
  35. } else {
  36. fail++;
  37. console.log(` FAIL ${name}${extra === '' ? '' : ` → ${JSON.stringify(extra)}`}`);
  38. }
  39. };
  40. /* ================================================================== *
  41. * 第 1 部分:纯逻辑
  42. * ================================================================== */
  43. console.log('【1】增量规整器:前缀不变式');
  44. /** 把一串输入分片喂给规整器,返回逐片输出 */
  45. const feed = (chunks) => {
  46. const normalizer = createIncrementalMarkdownNormalizer();
  47. const out = [];
  48. for (const chunk of chunks) out.push(normalizer.push(chunk));
  49. return { out, normalizer };
  50. };
  51. /**
  52. * 不变式(流中):**已发出的文本必须是「规整后全文」的前缀**,
  53. * 且未发出的差额只能是「**给当前行准备的空行分隔符(可能 1~2 个 \n)+ 当前行的尾巴**」——
  54. * 也就是「为了判断该行是不是列表项而短暂扣住的一小段」,行尾一到就释放。
  55. *
  56. * 这个性质正好保证 `answer_end` 的「整块替换」= 纯尾缀追加(不会整段重建)。
  57. */
  58. const isPrefixWithHeldLineTail = (got, want) => {
  59. if (!want.startsWith(got)) return false;
  60. const held = want.slice(got.length).replace(/^\n+/, ''); // 去掉随行一起扣住的前置空行
  61. return !held.includes('\n');
  62. };
  63. /** 不变式:每喂一片就核对一次 */
  64. const checkInvariant = (label, chunks, { finish = false } = {}) => {
  65. const normalizer = createIncrementalMarkdownNormalizer();
  66. let want = '';
  67. let got = '';
  68. let ok = true;
  69. let at = 0;
  70. for (let i = 0; i < chunks.length; i++) {
  71. want += chunks[i];
  72. got += normalizer.push(chunks[i]);
  73. if (!isPrefixWithHeldLineTail(got, normalizeChatMarkdown(want))) {
  74. ok = false;
  75. at = i;
  76. break;
  77. }
  78. }
  79. if (ok && finish) {
  80. got += normalizer.finish();
  81. if (got !== normalizeChatMarkdown(want)) ok = false;
  82. }
  83. check(
  84. `不变式:${label}`,
  85. ok,
  86. ok ? '' : { 第几片: at, 期望: normalizeChatMarkdown(want).slice(-40), 实际: got.slice(-40) }
  87. );
  88. return ok;
  89. };
  90. checkInvariant('整段一次 push', ['第一行\n第二行\n第三行']);
  91. checkInvariant('逐字 push(含换行)', [...'第一行\n第二行\n1. a\n2. b']);
  92. checkInvariant('双换行(空行)', ['段落一\n\n段落二']);
  93. checkInvariant('连续有序列表不插空行', ['1. 甲\n2. 乙\n3. 丙']);
  94. checkInvariant('连续无序列表不插空行', ['- 甲\n- 乙']);
  95. checkInvariant('列表后接说明行', ['- 甲\n- 乙\n以上就是全部']);
  96. checkInvariant('伪列表符(2倍)不误判', ['2倍\n3. 真列表']);
  97. checkInvariant('行首恰为列表前缀(- )', ['前一行\n- ']);
  98. checkInvariant('行首数字未定型(12)', ['前一行\n12']);
  99. checkInvariant('行首数字带点(12.)', ['前一行\n12.']);
  100. checkInvariant('末尾无换行', ['段一\n段二(未完']);
  101. checkInvariant('以换行结尾', ['段一\n']);
  102. checkInvariant('空串与纯空白', ['', ' ', '\n', ' \n']);
  103. checkInvariant('逐字 push 且收尾 finish', [...'甲\n1. 乙\n丙'], { finish: true });
  104. // 逐字 push 与整段 push 的「拼接结果」必须一致(都 finish 掉扣住的尾巴后比)
  105. {
  106. const text = '第一段\n第二段\n1. 甲\n2. 乙\n收尾说明';
  107. const wholeFeed = feed([text]);
  108. wholeFeed.out.push(wholeFeed.normalizer.finish());
  109. const perCharFeed = feed([...text]);
  110. perCharFeed.out.push(perCharFeed.normalizer.finish());
  111. const whole = wholeFeed.out.join('');
  112. const perChar = perCharFeed.out.join('');
  113. check('逐字 push 与整段 push 输出一致', whole === perChar, { whole, perChar });
  114. check('两者的拼接都等于整段规整结果', whole === normalizeChatMarkdown(text));
  115. }
  116. // 种子化伪随机模糊(≥500 轮)
  117. {
  118. let seed = 20260920;
  119. const rand = () => {
  120. seed = (seed * 1103515245 + 12345) % 2147483648;
  121. return seed / 2147483648;
  122. };
  123. const alphabet = ['甲', '乙', '丙', '\n', '\n\n', '- ', '* ', '1', '2', '.', ')', ' ', '。', '12', '-'];
  124. let bad = null;
  125. for (let round = 0; round < 500 && !bad; round++) {
  126. const chunks = [];
  127. const count = 1 + Math.floor(rand() * 20);
  128. for (let i = 0; i < count; i++) chunks.push(alphabet[Math.floor(rand() * alphabet.length)]);
  129. const normalizer = createIncrementalMarkdownNormalizer();
  130. let want = '';
  131. let got = '';
  132. for (const chunk of chunks) {
  133. want += chunk;
  134. got += normalizer.push(chunk);
  135. if (!isPrefixWithHeldLineTail(got, normalizeChatMarkdown(want))) {
  136. bad = { round, chunks, want, got };
  137. break;
  138. }
  139. }
  140. if (!bad) {
  141. got += normalizer.finish();
  142. if (got !== normalizeChatMarkdown(want)) bad = { round, chunks, want, got, finish: true };
  143. }
  144. }
  145. check('种子化模糊 500 轮(含 finish)不变式恒成立', !bad, bad);
  146. }
  147. // 扣住的部分必须「有界」:只是当前行的一小段,不能无限积压
  148. {
  149. const normalizer = createIncrementalMarkdownNormalizer();
  150. let shown = '';
  151. let want = '';
  152. for (const chunk of ['第一段\n', '第二段\n', '1', '.', ' ', '甲', '\n', '2', '.', ' ', '乙']) {
  153. want += chunk;
  154. shown += normalizer.push(chunk);
  155. const held = normalizeChatMarkdown(want).slice(shown.length).replace(/^\n+/, '');
  156. if (held.includes('\n') || held.length > 8) {
  157. check(`扣住有界(本条 ${JSON.stringify(chunk)})`, false, { held, shown });
  158. break;
  159. }
  160. }
  161. check('扣住的只是「可能成为列表项」的短前缀(本例 ≤8 字符且不含换行)', true);
  162. }
  163. console.log('\n【2】流式事件解析(形状不符 → null 走旧路径)');
  164. {
  165. const start = parseAnswerStreamEvent({ stream_id: 's1', target: 'summary', sequence: 0 }, 'answer_start');
  166. check('合法的 answer_start', start?.kind === 'start' && start.streamId === 's1' && start.target === 'summary', start);
  167. check('缺 stream_id → null', parseAnswerStreamEvent({ target: 'summary' }, 'answer_start') === null);
  168. check('空 stream_id → null', parseAnswerStreamEvent({ stream_id: ' ', target: 'summary' }, 'answer_delta') === null);
  169. check('target 非法 → null', parseAnswerStreamEvent({ stream_id: 's1', target: 'other' }, 'answer_delta') === null);
  170. check('非对象 → null', parseAnswerStreamEvent(null, 'answer_delta') === null && parseAnswerStreamEvent([1], 'answer_delta') === null);
  171. const delta = parseAnswerStreamEvent({ stream_id: 's1', target: 'answer', sequence: '3', text: 123 }, 'answer_delta');
  172. check('sequence 宽松(字符串也认)', delta?.sequence === 3, delta);
  173. check('text 非字符串按空串', delta?.text === '', delta);
  174. const end = parseAnswerStreamEvent({ stream_id: 's1', target: 'summary', text: 'x', status: '怪值' }, 'answer_end');
  175. check('status 非法 → fallback', end?.kind === 'end' && end.status === 'fallback', end);
  176. const endOk = parseAnswerStreamEvent({ stream_id: 's1', target: 'summary', text: 'x', status: 'completed' }, 'answer_end');
  177. check('status=completed 保留', endOk?.status === 'completed');
  178. const abort = parseAnswerStreamEvent({ stream_id: 's1', target: 'summary', reason: '怪值' }, 'answer_abort');
  179. check('reason 非法 → generation_failed', abort?.kind === 'abort' && abort.reason === 'generation_failed', abort);
  180. const abortOk = parseAnswerStreamEvent({ stream_id: 's1', target: 'summary', reason: 'not_finalized' }, 'answer_abort');
  181. check('reason=not_finalized 保留', abortOk?.reason === 'not_finalized');
  182. check('未知事件名 → null', parseAnswerStreamEvent({ stream_id: 's1', target: 'summary' }, 'answer_what') === null);
  183. }
  184. console.log('\n【3】流状态机');
  185. {
  186. const run = (events) => {
  187. let state = createAnswerStreamState();
  188. const effects = [];
  189. for (const ev of events) {
  190. const r = applyAnswerStreamEvent(state, ev);
  191. state = r.state;
  192. effects.push(r.effect.kind);
  193. }
  194. return { state, effects };
  195. };
  196. const ev = (kind, extra = {}) => ({ kind, streamId: 's1', target: 'summary', sequence: 0, text: '', ...extra });
  197. // 正常流
  198. {
  199. const { state, effects } = run([
  200. ev('start', { sequence: 0 }),
  201. ev('delta', { sequence: 1, text: '你好' }),
  202. ev('delta', { sequence: 2, text: '世界' }),
  203. ev('end', { sequence: 2, text: '你好,世界' }),
  204. ]);
  205. check('正常流:效果序列 start→append→append→replace', JSON.stringify(effects) === JSON.stringify(['start', 'append', 'append', 'replace']), effects);
  206. check('正常流:end 后 text = 最终全文(整块替换,不拼接)', state.text === '你好,世界' && state.completed === true, state);
  207. }
  208. // 序号缺口
  209. {
  210. const { state, effects } = run([
  211. ev('start'),
  212. ev('delta', { sequence: 1, text: 'a' }),
  213. ev('delta', { sequence: 5, text: 'b' }),
  214. ]);
  215. check('序号缺口:仍拼接、hasGap 置位', state.text === 'ab' && state.hasGap === true, state);
  216. check('缺口不阻断渲染(仍是 append)', effects[2] === 'append');
  217. }
  218. // 重复序号
  219. {
  220. const { state } = run([ev('start'), ev('delta', { sequence: 1, text: 'a' }), ev('delta', { sequence: 1, text: 'XX' })]);
  221. check('重复序号忽略(不重复拼接)', state.text === 'a', state);
  222. }
  223. // 过期流
  224. {
  225. const { state, effects } = run([
  226. ev('start'),
  227. ev('delta', { streamId: 'other', sequence: 1, text: 'X' }),
  228. ev('end', { streamId: 'other', text: 'X' }),
  229. ev('abort', { streamId: 'other' }),
  230. ]);
  231. check('过期 streamId 的 delta/end/abort 全部 none', effects.slice(1).every((k) => k === 'none'), effects);
  232. check('过期事件不改状态', state.text === '' && state.completed === false, state);
  233. }
  234. // end 之后的 delta
  235. {
  236. const { state, effects } = run([ev('start'), ev('end', { text: '定稿' }), ev('delta', { sequence: 9, text: '迟到' })]);
  237. check('end 之后的 delta 忽略', effects[2] === 'none' && state.text === '定稿', state);
  238. }
  239. // abort 复位 + 重新 start
  240. {
  241. const { state, effects } = run([
  242. ev('start'),
  243. ev('delta', { sequence: 1, text: '草稿' }),
  244. ev('abort', { reason: 'generation_failed' }),
  245. ev('start', { streamId: 's2' }),
  246. ev('delta', { streamId: 's2', sequence: 1, text: '新正文' }),
  247. ]);
  248. check('abort 复位(discard 效果 + 状态清空后可再 start)', effects.includes('discard') && state.text === '新正文' && state.streamId === 's2', { effects, state });
  249. }
  250. // end 覆盖草稿(fallback 也替换)
  251. {
  252. const { state } = run([
  253. ev('start'),
  254. ev('delta', { sequence: 1, text: '草稿' }),
  255. ev('end', { sequence: 1, text: '最终', status: 'fallback' }),
  256. ]);
  257. check('fallback 的 end 同样整块替换', state.text === '最终' && state.completed === true, state);
  258. }
  259. check('null 事件 → none 且状态不变', (() => {
  260. const s0 = createAnswerStreamState();
  261. const r = applyAnswerStreamEvent(s0, null);
  262. return r.effect.kind === 'none' && r.state === s0;
  263. })());
  264. }
  265. console.log('\n【3b】应用层的内容对齐(end 替换 / abort 撤稿)');
  266. {
  267. const BASE = '<scope title="思考中">\n</scope>\n';
  268. check(
  269. '正常路径:最终文本是已上屏内容的前缀延伸 → 纯尾缀追加(不加空行、不重建)',
  270. applyAnswerStreamEnd(BASE + '一、说明', BASE, '一、说明。二、补充') === BASE + '一、说明。二、补充'
  271. );
  272. check(
  273. '已上屏一长段、最终只多了尾巴 → 不重复整段',
  274. applyAnswerStreamEnd(BASE + '甲', BASE, '甲乙丙') === BASE + '甲乙丙'
  275. );
  276. check(
  277. '罕见分歧(最终文本与增量拼接不一致)→ 整块重建,且基线里的 <scope> 被剥掉(避免卡片闪现)',
  278. applyAnswerStreamEnd(BASE + '草稿', BASE, '完全不同的最终文本') === '完全不同的最终文本'
  279. );
  280. check(
  281. '重建时**基线(流开始前就有的内容)保留**,最终文本接在其后 —— 只有 scope 标记会被剥掉',
  282. applyAnswerStreamEnd('前言', '前言', '最终') === '前言最终'
  283. );
  284. check('abort 撤稿:回到流开始前的内容', applyAnswerStreamAbort(BASE) === BASE);
  285. check('abort 撤稿:空基线返回空串', applyAnswerStreamAbort('') === '');
  286. // answer 开场概述该不该显示(实测:后端把正文作为 summary 流式下发,
  287. // 而 answer 常是同一段开头的**压缩版**——显示会重复开头;但万一是另一段内容就不能丢)
  288. const body =
  289. '我们青浦目前能对上的小微企业扶持,主要分两类:一类是面向特定对象的创业扶持(如带动就业补贴、贷款贴息、留创企业开办资助),另一类是市级中小企业公共资助项目;此外还有几款“小微贷”金融产品。\n\n### 具体政策\n- 创业扶持:带动就业补贴';
  290. // 压缩版的形态与实测一致:**截断在词中间**(正文接着 '###',概述结尾是半个词 '部分候')
  291. const abridged = body.slice(0, 70).replace(/\n+/g, ' ') + ' 部分候…';
  292. check('压缩版开场(截断在词中间,公共前缀 ≥90%)→ 不显示(避免开头重复)', shouldKeepAnswerOverview(abridged, body) === false, abridged);
  293. check('真正的另一段内容(不是前缀)→ 必须保留', shouldKeepAnswerOverview('这是另一段完全不同的开场说明。', body) === true);
  294. check('概述为空 → 不显示', shouldKeepAnswerOverview('', body) === false && shouldKeepAnswerOverview(null, body) === false);
  295. check('流式正文为空 → 保留概述(不能丢字)', shouldKeepAnswerOverview('开场', '') === true);
  296. // 流式轮的最终组装:基线 + 卡片 + 正文 + 尾部(卡片在正文**之上**)
  297. const SCOPE = '<scope title="思考中">\n</scope>\n';
  298. check(
  299. '组装顺序:基线 → leading(卡片)→ 正文 → trailing(参考资料)',
  300. composeStreamedContent(SCOPE + '前言', '【卡片】', '正文正文', '【参考资料】') === '前言\n\n【卡片】\n\n正文正文\n\n【参考资料】',
  301. composeStreamedContent(SCOPE + '前言', '【卡片】', '正文正文', '【参考资料】')
  302. );
  303. check(
  304. '组装时剥掉基线里的 <scope>(否则卡片会闪现重播)',
  305. !composeStreamedContent(SCOPE + '前言', '', '正文', '').includes('<scope')
  306. );
  307. check('没有卡片时只剩正文(不留空行)', composeStreamedContent('', '', '正文', '') === '正文');
  308. check('空块被跳过、不产生多余空行', composeStreamedContent('', '', '正文', '') === '正文' && composeStreamedContent('', '', '', '尾') === '尾');
  309. }
  310. console.log(`\n===== 纯逻辑部分:通过 ${pass} 项,失败 ${fail} 项 =====`);
  311. /* ================================================================== *
  312. * 第 2 部分:协调器级假流(假 SSE 喂真实协调器)
  313. * ================================================================== */
  314. const sse = (events) =>
  315. events
  316. .map(
  317. ([event, data]) =>
  318. `event: ${event}\ndata: ${JSON.stringify({ thread_id: 't1', request_id: 'r1', data })}\n\n`
  319. )
  320. .join('');
  321. const stubFetchWith = (sseText) => {
  322. globalThis.fetch = async () => {
  323. const stream = new ReadableStream({
  324. start(controller) {
  325. controller.enqueue(new TextEncoder().encode(sseText));
  326. controller.close();
  327. },
  328. });
  329. return new Response(stream, { status: 200, headers: { 'Content-Type': 'text/event-stream' } });
  330. };
  331. };
  332. /** 跑一轮,收集各类事件 */
  333. const runTurn = async (events) => {
  334. stubFetchWith(sse(events));
  335. const coordinator = new ApiChatCoordinator({ baseUrl: 'http://stub', threadId: 't1' });
  336. const seen = { messages: [], total: null, streamStart: [], streamEnd: [], streamAbort: [], compose: [], errors: [], order: [] };
  337. coordinator.addEventListener('message', (m) => {
  338. seen.messages.push(m);
  339. seen.order.push(`message:${m.slice(0, 24)}`);
  340. });
  341. coordinator.addEventListener('totalResponse', (p) => {
  342. seen.total = p;
  343. });
  344. coordinator.addEventListener('answerStreamStart', (p) => {
  345. seen.streamStart.push(p);
  346. seen.order.push('streamStart');
  347. });
  348. coordinator.addEventListener('answerStreamEnd', (p) => {
  349. seen.streamEnd.push(p);
  350. seen.order.push('streamEnd');
  351. });
  352. coordinator.addEventListener('answerStreamAbort', (p) => {
  353. seen.streamAbort.push(p);
  354. seen.order.push('streamAbort');
  355. });
  356. coordinator.addEventListener('answerStreamCompose', (p) => {
  357. seen.compose.push(p);
  358. seen.order.push('streamCompose');
  359. });
  360. coordinator.addEventListener('error', (p) => {
  361. seen.errors.push(p);
  362. seen.order.push('error');
  363. });
  364. await coordinator.generateAnswer('测试问题');
  365. await new Promise((resolve) => setTimeout(resolve, 80)); // flushContent 会等两帧
  366. return seen;
  367. };
  368. // 应用侧就是 `content += msg`(见 appendAiMessageChunk),**没有分隔符** ——
  369. // 用 '\n\n' 拼会把流式增量切成一堆段落,断言就失真了
  370. const messageText = (seen) => seen.messages.join('');
  371. const countOf = (text, needle) => text.split(needle).length - 1;
  372. const DELTA_TEXTS = ['一、', '第一段说明', '。\n\n', '二、', '第二段说明', '。'];
  373. const FINAL_TEXT = '一、第一段说明。\n\n二、第二段说明。';
  374. const MARKER = '第一段说明';
  375. console.log('\n【4】协调器假流:happy path(带卡片轮形态,含流中 heartbeat)');
  376. {
  377. const seen = await runTurn([
  378. ['accepted', { message: '收到' }],
  379. ['progress', { stage: 'retrieve', message: '正在检索……' }],
  380. ['answer_start', { stream_id: 's1', target: 'summary', sequence: 0, provisional: true }],
  381. ...DELTA_TEXTS.map((text, i) => ['answer_delta', { stream_id: 's1', target: 'summary', sequence: i + 1, text }]),
  382. ['heartbeat', { status: 'processing', message: '仍在处理……', elapsed_seconds: 20 }],
  383. ['answer', { text: FINAL_TEXT }],
  384. ['source', { id: 's1', title: '测试政策', type: 'policy', source: { url: 'https://example.com/p1' } }],
  385. // 有 item 才有 POLICY_TABLE 卡片块(buildPolicyTableContent 是按 items 建的)
  386. ['item', { source_id: 's1', title: '测试政策', card: { name: { text: '测试政策' } } }],
  387. ['answer_end', { stream_id: 's1', target: 'summary', sequence: DELTA_TEXTS.length, text: FINAL_TEXT, operation: 'replace', status: 'completed', provisional: false }],
  388. ['summary', { text: FINAL_TEXT, source_ids: ['s1'], notices: [] }],
  389. ['result', { response: FINAL_TEXT, recommendation: {}, errors: [] }],
  390. ['done', { status: 'completed' }],
  391. ]);
  392. const text = messageText(seen);
  393. if (process.env.DBG) console.log('[dbg] text=' + JSON.stringify(text));
  394. check('answerStreamStart 发出一次', seen.streamStart.length === 1, seen.streamStart);
  395. check('answerStreamEnd 发出一次', seen.streamEnd.length === 1, seen.streamEnd);
  396. check('无 abort', seen.streamAbort.length === 0, seen.streamAbort);
  397. check('scope 卡片只发了一次(流中 heartbeat 不再发)', countOf(text, '<scope') === 1, countOf(text, '<scope'));
  398. check('answer_end 的文本是规整后的全文', seen.streamEnd[0]?.text === normalizeChatMarkdown(FINAL_TEXT), seen.streamEnd[0]?.text);
  399. check(
  400. '正文在 message 通道**只出现一次**(完整 answer/summary 事件没有渲染成第二份)',
  401. countOf(text, MARKER) === 1,
  402. countOf(text, MARKER)
  403. );
  404. check(
  405. '流式增量拼接 === 规整后全文(前缀不变式在真链路成立)',
  406. text.includes(normalizeChatMarkdown(FINAL_TEXT).split('\n')[0]),
  407. normalizeChatMarkdown(FINAL_TEXT).slice(0, 30)
  408. );
  409. check('totalResponse.answer 含最终全文', String(seen.total?.answer || '').includes('一、第一段说明。'), String(seen.total?.answer || '').slice(0, 40));
  410. check('totalResponse 里含政策卡片与参考资料', String(seen.total?.answer || '').includes('POLICY_TABLE') && String(seen.total?.answer || '').includes('<ref_links>'));
  411. // 卡片顺序:流式轮**卡片在正文之上**(用户 2026-09-20 要求);正文已上屏,故重排通过
  412. // answerStreamCompose 事件交给应用层(message 通道里只有正文)
  413. const compose = seen.compose[0];
  414. check('发出 answerStreamCompose(流式轮的重排信号)', !!compose, seen.compose);
  415. check('compose.leading 是政策卡片', String(compose?.leading || '').includes('POLICY_TABLE'), String(compose?.leading || '').slice(0, 60));
  416. check('compose.trailing 是参考资料', String(compose?.trailing || '').includes('<ref_links>'), String(compose?.trailing || '').slice(0, 60));
  417. check(
  418. '**totalResponse 里卡片排在正文之前**',
  419. (() => {
  420. const ans = String(seen.total?.answer || '');
  421. return ans.indexOf('POLICY_TABLE') > -1 && ans.indexOf('POLICY_TABLE') < ans.indexOf(MARKER);
  422. })(),
  423. String(seen.total?.answer || '').slice(0, 80)
  424. );
  425. check('正文本身没有在 message 通道里被卡片打断(卡片走 compose)', !text.includes('POLICY_TABLE'));
  426. // ⚠️ 契约:compose 让应用层**整块重写**内容(不是往后追加)——
  427. // 所以渲染层的增量解析必须能识别"非纯追加"并重置,否则会把新内容的中段当正文渲染
  428. // (踩过:参考资料 JSON 的中段变成一坨文本显示在界面上)
  429. const composedLeading = String(compose?.leading || '');
  430. check(
  431. '重排后的内容**不是正文的纯追加**(渲染层必须支持"整块重写")',
  432. composedLeading.length > 0 &&
  433. !`${composedLeading}\n\n${normalizeChatMarkdown(FINAL_TEXT)}`.startsWith(normalizeChatMarkdown(FINAL_TEXT)),
  434. composedLeading.slice(0, 40)
  435. );
  436. }
  437. console.log('\n【5】协调器假流:纯正文轮(无卡片、无 answer 事件)不丢正文');
  438. {
  439. const seen = await runTurn([
  440. ['accepted', { message: '收到' }],
  441. ['answer_start', { stream_id: 's2', target: 'summary', sequence: 0 }],
  442. ...DELTA_TEXTS.map((text, i) => ['answer_delta', { stream_id: 's2', target: 'summary', sequence: i + 1, text }]),
  443. ['answer_end', { stream_id: 's2', target: 'summary', sequence: DELTA_TEXTS.length, text: FINAL_TEXT, status: 'completed' }],
  444. ['summary', { text: FINAL_TEXT }],
  445. ['result', { response: FINAL_TEXT }],
  446. ['done', { status: 'completed' }],
  447. ]);
  448. check('totalResponse.answer === 最终全文(没有因为"没有卡片"而丢正文)', String(seen.total?.answer || '').trim() === normalizeChatMarkdown(FINAL_TEXT).trim(), String(seen.total?.answer || ''));
  449. }
  450. console.log('\n【6】协调器假流:answer_end 与增量拼接不同 → 整块替换生效');
  451. {
  452. const seen = await runTurn([
  453. ['answer_start', { stream_id: 's3', target: 'summary', sequence: 0 }],
  454. ['answer_delta', { stream_id: 's3', target: 'summary', sequence: 1, text: '草稿内容' }],
  455. ['answer_end', { stream_id: 's3', target: 'summary', sequence: 1, text: '定稿内容(后端最终校验改过)', status: 'completed' }],
  456. ['summary', { text: '定稿内容(后端最终校验改过)' }],
  457. ['result', { response: '定稿内容(后端最终校验改过)' }],
  458. ['done', { status: 'completed' }],
  459. ]);
  460. check('End.text = 规整后的最终文本(不是增量拼接)', seen.streamEnd[0]?.text === '定稿内容(后端最终校验改过)', seen.streamEnd[0]?.text);
  461. check('totalResponse.answer 用的是最终文本', String(seen.total?.answer || '').includes('定稿内容'), String(seen.total?.answer || ''));
  462. }
  463. console.log('\n【7】协调器假流:后端 error → 撤草稿(abort 先于 error)');
  464. {
  465. const seen = await runTurn([
  466. ['answer_start', { stream_id: 's4', target: 'summary', sequence: 0 }],
  467. ['answer_delta', { stream_id: 's4', target: 'summary', sequence: 1, text: '半截草稿' }],
  468. ['error', { code: 'processing_failed' }],
  469. ]);
  470. const abortAt = seen.order.indexOf('streamAbort');
  471. const errorAt = seen.order.indexOf('error');
  472. check('abort 在 error **之前**发出(否则 hook 已摘任务、abort 会被守卫丢弃)', abortAt > -1 && abortAt < errorAt, seen.order);
  473. check('abort 的 reason = generation_failed', seen.streamAbort[0]?.reason === 'generation_failed', seen.streamAbort[0]);
  474. check('草稿文本未进入 totalResponse', !String(seen.total?.answer || '').includes('半截草稿'), String(seen.total?.answer || ''));
  475. }
  476. console.log('\n【8】协调器假流:断流(EOF 无 done)→ 撤草稿 + 兼容回退');
  477. {
  478. const seen = await runTurn([
  479. ['answer_start', { stream_id: 's5', target: 'summary', sequence: 0 }],
  480. ['answer_delta', { stream_id: 's5', target: 'summary', sequence: 1, text: '半截草稿' }],
  481. ['answer', { text: '完整事件里的正文(断流前的兼容副本)' }],
  482. ]);
  483. check('断流时发出 abort(reason=not_finalized)', seen.streamAbort[0]?.reason === 'not_finalized', seen.streamAbort);
  484. // ⚠️ 分层:协调器**只发信号**,真正把草稿从界面上撤掉的是应用层(hook 调 applyAnswerStreamAbort)。
  485. // 断流路径也不发 totalResponse(既有行为:内容走 message 通道,hook 的 close 监听负责补写 DMS)。
  486. check('兼容路径的完整 answer 事件仍进了 message 通道', messageText(seen).includes('完整事件里的正文'), messageText(seen).slice(0, 80));
  487. check('报的是连接断开', seen.errors.some((e) => e?.code === 'disconnected'), seen.errors);
  488. }
  489. console.log('\n【9】协调器假流:旧后端(无任何流式事件)行为不变');
  490. {
  491. const seen = await runTurn([
  492. ['accepted', { message: '收到' }],
  493. ['progress', { stage: 'retrieve', message: '正在检索……' }],
  494. ['answer', { text: FINAL_TEXT }],
  495. ['source', { id: 's1', title: '测试政策', type: 'policy', source: { url: 'https://example.com/p1' } }],
  496. ['item', { source_id: 's1', title: '测试政策', card: { name: { text: '测试政策' } } }],
  497. ['summary', { text: FINAL_TEXT }],
  498. ['result', { response: FINAL_TEXT }],
  499. ['done', { status: 'completed' }],
  500. ]);
  501. const text = messageText(seen);
  502. check('没有流事件、没有 abort', seen.streamStart.length === 0 && seen.streamAbort.length === 0);
  503. check('不发 answerStreamCompose(旧路径不重排)', seen.compose.length === 0, seen.compose);
  504. check('scope 卡片正常', countOf(text, '<scope') === 1);
  505. check('正文按原路径整段输出(规整后)', text.includes(normalizeChatMarkdown(FINAL_TEXT)));
  506. const bodyAt = text.indexOf('第一段说明');
  507. const cardAt = text.indexOf('POLICY_TABLE');
  508. const refAt = text.indexOf('<ref_links>');
  509. check(
  510. '**旧路径顺序不变**:正文 → 卡片 → 参考资料',
  511. bodyAt > -1 && cardAt > -1 && refAt > -1 && bodyAt < cardAt && cardAt < refAt,
  512. { bodyAt, cardAt, refAt }
  513. );
  514. }
  515. console.log('\n【10】协调器:流式轮之后跑旧轮,状态不残留');
  516. {
  517. stubFetchWith(
  518. sse([
  519. ['answer_start', { stream_id: 's6', target: 'summary', sequence: 0 }],
  520. ['answer_delta', { stream_id: 's6', target: 'summary', sequence: 1, text: '流式正文' }],
  521. ['answer_end', { stream_id: 's6', target: 'summary', sequence: 1, text: '流式正文', status: 'completed' }],
  522. ['done', { status: 'completed' }],
  523. ])
  524. );
  525. const coordinator = new ApiChatCoordinator({ baseUrl: 'http://stub', threadId: 't1' });
  526. const rounds = [];
  527. coordinator.addEventListener('message', (m) => rounds.push(m));
  528. await coordinator.generateAnswer('第一轮');
  529. await new Promise((r) => setTimeout(r, 80));
  530. const cutoff = rounds.length; // 第一轮结束时的分界
  531. stubFetchWith(
  532. sse([
  533. ['progress', { stage: 'retrieve', message: '正在检索……' }],
  534. ['answer', { text: '第二轮正文' }],
  535. ['done', { status: 'completed' }],
  536. ])
  537. );
  538. await coordinator.generateAnswer('第二轮');
  539. await new Promise((r) => setTimeout(r, 80));
  540. const first = rounds.slice(0, cutoff).join('');
  541. const second = rounds.slice(cutoff).join('');
  542. check('第一轮是流式正文', first.includes('流式正文'), first.slice(0, 60));
  543. check('第二轮回到旧路径:有 scope、正文是整段输出', countOf(second, '<scope') === 1 && second.includes('第二轮正文'), second.slice(0, 80));
  544. }
  545. console.log('\n【11】两个协调器并行流式:互不串台');
  546. {
  547. stubFetchWith(
  548. sse([
  549. ['answer_start', { stream_id: 'A', target: 'summary', sequence: 0 }],
  550. ['answer_delta', { stream_id: 'A', target: 'summary', sequence: 1, text: '甲会话正文' }],
  551. ['answer_end', { stream_id: 'A', target: 'summary', sequence: 1, text: '甲会话正文', status: 'completed' }],
  552. ['done', { status: 'completed' }],
  553. ])
  554. );
  555. const a = new ApiChatCoordinator({ baseUrl: 'http://stub', threadId: 'A' });
  556. const b = new ApiChatCoordinator({ baseUrl: 'http://stub', threadId: 'B' });
  557. const gotA = [];
  558. const gotB = [];
  559. a.addEventListener('message', (m) => gotA.push(m));
  560. b.addEventListener('message', (m) => gotB.push(m));
  561. b.addEventListener('answerStreamAbort', () => gotB.push('ABORT'));
  562. await a.generateAnswer('问 A');
  563. await new Promise((r) => setTimeout(r, 80));
  564. check('B 未收到任何事件(含 abort)', gotB.length === 0, gotB);
  565. check('A 正常收到正文', gotA.join('').includes('甲会话正文'), gotA);
  566. }
  567. console.log(`\n===== 总计:通过 ${pass} 项,失败 ${fail} 项 =====`);
  568. process.exit(fail ? 1 : 0);