verify-answer-stream.mjs 37 KB

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