transcript-adapter.spec.ts 25 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556
  1. /**
  2. * TranscriptAdapter over the raw append-only window: log-ordered projection of
  3. * append-origin events, one marker per landed compaction, replacement copies
  4. * hidden, command-lifecycle folding, node/array identity, call pairing, and
  5. * host-provided wire views.
  6. */
  7. import { createUserMessage, CallId, createMessage, createToolResultMessage } from '@deepseek-ai/dsh-llm'
  8. import { describe, expect, it } from 'vitest'
  9. import type { SessionEvent } from '@deepseek-ai/dsh-session/types'
  10. import { TranscriptAdapter } from '../src/client/sessions/transcript-adapter.ts'
  11. import { ev, plainTurn } from './event-script.ts'
  12. const at = (seq: number, e: Record<string, unknown>): SessionEvent =>
  13. ({ seq, time: 1_700_000_000_000 + seq, ...e }) as unknown as SessionEvent
  14. /** A `compact/summary` provenance event (log-only, no surfaceOp). */
  15. function compactSummary(seq: number, summary: unknown = [{ type: 'text', text: '# 摘要\n\n保留事实' }]): SessionEvent {
  16. return at(seq, {
  17. type: 'compact/summary',
  18. data: {
  19. summary,
  20. shadowedRange: { start: 1, end: 3 },
  21. shadowedSeqs: [1, 3],
  22. shadowedTokenCount: 100,
  23. provider: 'fake',
  24. model: 'compact-1',
  25. },
  26. })
  27. }
  28. /** The replacement user message a compaction backend lands (the checkpoint). */
  29. function checkpoint(
  30. seq: number,
  31. summarySeq: number,
  32. { start = 1, end = 3, sourceEventSeqs = [summarySeq, start, end] }: {
  33. start?: number
  34. end?: number
  35. sourceEventSeqs?: number[]
  36. } = {},
  37. ): SessionEvent {
  38. return at(seq, {
  39. type: 'user/message',
  40. surfaceOp: { op: 'replace', start, end },
  41. sourceEventSeqs,
  42. data: createUserMessage({
  43. content: [{ type: 'text', text: '<context_checkpoint>model only</context_checkpoint>' }],
  44. source: { kind: 'plugin', plugin: 'compact' },
  45. }),
  46. })
  47. }
  48. describe('TranscriptAdapter', () => {
  49. it('projects a window starting past seq 0 at its own log positions', () => {
  50. const adapter = new TranscriptAdapter()
  51. adapter.reset(plainTurn(100, 5, '偏移问', '偏移答'))
  52. expect(adapter.nodes().map(n => [n.kind, n.seq])).toEqual([['user', 101], ['assistant', 103]])
  53. })
  54. it('appends incrementally keeping old node references (materialize-once identity)', () => {
  55. const adapter = new TranscriptAdapter()
  56. adapter.reset(plainTurn(0, 0, 'a', 'b'))
  57. const first = adapter.nodes()
  58. adapter.append(ev.user(6, '追加'))
  59. const second = adapter.nodes()
  60. expect(second).toHaveLength(3)
  61. expect(second[0]).toBe(first[0])
  62. expect(second[1]).toBe(first[1])
  63. expect(second).not.toBe(first) // a real change swaps the array
  64. })
  65. it('keeps the array reference across a chunk storm and swaps it when a node lands', () => {
  66. const adapter = new TranscriptAdapter()
  67. adapter.reset(plainTurn(0, 0, 'a', 'b'))
  68. const settled = adapter.nodes()
  69. adapter.append(ev.chunkStart(6, 1))
  70. expect(adapter.nodes()).toBe(settled)
  71. adapter.append(ev.chunkText(7, 1, '流式'))
  72. expect(adapter.nodes()).toBe(settled)
  73. adapter.append(ev.assistant(8, 1, '流式完成'))
  74. const finalized = adapter.nodes()
  75. expect(finalized).not.toBe(settled)
  76. expect(finalized.at(-1)).toMatchObject({ kind: 'assistant', seq: 8 })
  77. })
  78. it('materializes every append-origin variant with field mapping', () => {
  79. const adapter = new TranscriptAdapter()
  80. const steering = createUserMessage({
  81. content: [{ type: 'text', text: '插话' }],
  82. source: { kind: 'user' },
  83. })
  84. adapter.reset([
  85. ev.user(0, '用户'),
  86. ev.assistant(1, 0, '助手'),
  87. at(2, { type: 'agent/inbox/spliced', data: {
  88. target: 'next-step', start: 0, inserted: [steering],
  89. } }),
  90. at(3, { type: 'agent/inbox/spliced', data: {
  91. target: 'next-step', start: 0, removedCount: 1, inserted: [],
  92. } }),
  93. at(4, { type: 'user/message', surfaceOp: 'append', data: steering }),
  94. at(5, { type: 'user/message', surfaceOp: 'append', data: createUserMessage({
  95. content: [{ type: 'text', text: '上下文' }], source: { kind: 'plugin', plugin: 'p' },
  96. }) }),
  97. ev.toolCall(6, 0, 'c1', 'echo', '{"x":1}'),
  98. ev.toolResult(7, 0, 'c1', '结果'),
  99. ])
  100. const nodes = adapter.nodes()
  101. expect(nodes.map(n => n.kind)).toEqual(['user', 'assistant', 'steering', 'context', 'tool-result'])
  102. expect(nodes.find(n => n.kind === 'steering')).toMatchObject({ messageId: steering.id })
  103. expect(nodes.find(n => n.kind === 'tool-result')).toMatchObject({
  104. callId: 'c1', call: { name: 'echo', argsRaw: '{"x":1}' }, isError: false,
  105. })
  106. })
  107. it('identifies steering on the live append path', () => {
  108. const adapter = new TranscriptAdapter()
  109. const steering = createUserMessage({
  110. content: [{ type: 'text', text: 'live steer' }],
  111. source: { kind: 'user' },
  112. })
  113. adapter.reset([])
  114. adapter.append(at(0, { type: 'agent/inbox/spliced', data: {
  115. target: 'next-step', start: 0, inserted: [steering],
  116. } }))
  117. adapter.append(at(1, { type: 'agent/inbox/spliced', data: {
  118. target: 'next-step', start: 0, removedCount: 1, inserted: [],
  119. } }))
  120. adapter.append(at(2, { type: 'user/message', surfaceOp: 'append', data: steering }))
  121. expect(adapter.nodes()).toMatchObject([{ kind: 'steering', messageId: steering.id }])
  122. })
  123. it('does not mark queued, canceled, or non-user next-step messages as steering', () => {
  124. const adapter = new TranscriptAdapter()
  125. const queued = createUserMessage({ content: [{ type: 'text', text: 'queued' }], source: { kind: 'user' } })
  126. const canceled = createUserMessage({ content: [{ type: 'text', text: 'canceled' }], source: { kind: 'user' } })
  127. const context = createUserMessage({
  128. content: [{ type: 'text', text: 'context' }],
  129. source: { kind: 'plugin', plugin: 'test' },
  130. })
  131. adapter.reset([
  132. at(0, { type: 'agent/inbox/spliced', data: {
  133. target: 'next-turn', start: 0, inserted: [queued],
  134. } }),
  135. at(1, { type: 'agent/inbox/spliced', data: {
  136. target: 'next-turn', start: 0, removedCount: 1, inserted: [],
  137. } }),
  138. at(2, { type: 'user/message', surfaceOp: 'append', data: queued }),
  139. at(3, { type: 'agent/inbox/spliced', data: {
  140. target: 'next-step', start: 0, inserted: [canceled],
  141. } }),
  142. at(4, { type: 'agent/inbox/spliced', data: {
  143. target: 'next-step', start: 0, removedCount: 1, inserted: [], outcome: 'canceled',
  144. } }),
  145. at(5, { type: 'user/message', surfaceOp: 'append', data: canceled }),
  146. at(6, { type: 'agent/inbox/spliced', data: {
  147. target: 'next-step', start: 0, inserted: [context],
  148. } }),
  149. at(7, { type: 'agent/inbox/spliced', data: {
  150. target: 'next-step', start: 0, removedCount: 1, inserted: [],
  151. } }),
  152. at(8, { type: 'user/message', surfaceOp: 'append', data: context }),
  153. ])
  154. expect(adapter.nodes().map(node => node.kind)).toEqual(['user', 'user', 'context'])
  155. })
  156. it('materializes a skill-invocation source as its dedicated node', () => {
  157. const adapter = new TranscriptAdapter()
  158. adapter.reset([
  159. at(0, { type: 'user/message', surfaceOp: 'append', data: createUserMessage({
  160. content: [{ type: 'text', text: '<skill_content name="hidden-demo">body</skill_content>\n\ncheck the fixture' }],
  161. source: { kind: 'skill-invocation', name: 'hidden-demo', args: 'check the fixture' } as never,
  162. }) }),
  163. at(1, { type: 'user/message', surfaceOp: 'append', data: createUserMessage({
  164. content: [{ type: 'text', text: '<skill_content name="bare-skill">body</skill_content>' }],
  165. source: { kind: 'skill-invocation', name: 'bare-skill' } as never,
  166. }) }),
  167. ])
  168. const nodes = adapter.nodes()
  169. expect(nodes.map(node => node.kind)).toEqual(['skill-invocation', 'skill-invocation'])
  170. expect(nodes[0]).toMatchObject({ name: 'hidden-demo', args: 'check the fixture' })
  171. expect(nodes[1]).toMatchObject({ name: 'bare-skill' })
  172. expect((nodes[1] as { args?: string }).args).toBeUndefined()
  173. // A malformed record (no readable name) degrades to injected context, not a crash.
  174. adapter.append(at(2, { type: 'user/message', surfaceOp: 'append', data: createUserMessage({
  175. content: [{ type: 'text', text: 'odd' }],
  176. source: { kind: 'skill-invocation' } as never,
  177. }) }))
  178. expect(adapter.nodes().at(-1)?.kind).toBe('context')
  179. })
  180. it('skips events core does not call surface-eligible, marker or not', () => {
  181. // The transcript is the append-origin surface, so log-only events (a chunk,
  182. // a turn boundary, a compact/* provenance record) and a future type core
  183. // has not admitted contribute no node.
  184. const adapter = new TranscriptAdapter()
  185. adapter.reset([
  186. ev.turnStart(0, 1),
  187. at(1, { type: 'notice/message', surfaceOp: 'append', data: { note: 1 } }),
  188. compactSummary(2),
  189. ev.user(3, '唯一的一条'),
  190. ev.turnEnd(4, 1),
  191. ])
  192. expect(adapter.nodes().map(n => [n.kind, n.seq])).toEqual([['user', 3]])
  193. })
  194. describe('compaction markers', () => {
  195. it('keeps the original messages and full tool output, hiding replacement copies', () => {
  196. const adapter = new TranscriptAdapter()
  197. adapter.reset([
  198. ev.user(0, '原始问题'),
  199. ev.assistant(1, 0, '原始回答'),
  200. ev.toolCall(4, 0, 'c1', 'echo', '{}'),
  201. ev.toolResult(5, 0, 'c1', '完整工具输出'),
  202. // A pruned tool/result copy: rewrites one node for the model, marks nothing.
  203. at(6, { type: 'tool/result', surfaceOp: { op: 'replace', start: 5, end: 5 }, sourceEventSeqs: [5], data: {
  204. turn: 0, step: 0,
  205. message: createToolResultMessage({ callId: CallId('c1'), content: [{ type: 'text', text: '已裁剪' }], isError: false }),
  206. } }),
  207. compactSummary(7),
  208. checkpoint(8, 7, { start: 1, end: 5, sourceEventSeqs: [7, 1, 5] }),
  209. // A regenerated assistant/message: also a silent model-only rewrite.
  210. at(9, { type: 'assistant/message', surfaceOp: { op: 'replace', start: 8, end: 8 }, sourceEventSeqs: [8], data: {
  211. turn: 0, step: 0,
  212. message: createMessage({
  213. role: 'assistant',
  214. content: [{ type: 'text', text: '通用 replacement 副本' }],
  215. source: { kind: 'model', ...{ provider: 'x', model: 'copy' } },
  216. }),
  217. } }),
  218. ])
  219. const nodes = adapter.nodes()
  220. expect(nodes.map(n => [n.kind, n.seq])).toEqual([
  221. ['user', 0], ['assistant', 1], ['tool-result', 5], ['compaction', 8],
  222. ])
  223. expect(nodes[2]).toMatchObject({ kind: 'tool-result', content: [{ type: 'text', text: '完整工具输出' }] })
  224. expect(nodes[3]).toMatchObject({ kind: 'compaction', summary: '# 摘要\n\n保留事实' })
  225. })
  226. it('adds one marker per landed compaction, in log order', () => {
  227. const adapter = new TranscriptAdapter()
  228. adapter.reset([
  229. ev.user(0, 'a'),
  230. compactSummary(1, [{ type: 'text', text: 'first' }]),
  231. checkpoint(2, 1, { start: 0, end: 0, sourceEventSeqs: [1, 0] }),
  232. ev.user(3, 'b'),
  233. compactSummary(4, [{ type: 'text', text: 'second' }]),
  234. checkpoint(5, 4, { start: 2, end: 3, sourceEventSeqs: [4, 2, 3] }),
  235. ])
  236. expect(adapter.nodes().filter(n => n.kind === 'compaction')).toEqual([
  237. { kind: 'compaction', seq: 2, time: 1_700_000_000_002, summary: 'first' },
  238. { kind: 'compaction', seq: 5, time: 1_700_000_000_005, summary: 'second' },
  239. ])
  240. })
  241. it('renders the marker when the shadowed range is outside the window and logs nothing', () => {
  242. // The pagination hole A1 left open: quota is no longer spent on
  243. // replacement copies, so a page can carry a checkpoint whose
  244. // surfaceOp.start lies below the window head. The old surface fold threw
  245. // on the missing range and degraded with a console error; a log-ordered
  246. // projection has no range to resolve.
  247. const adapter = new TranscriptAdapter()
  248. const noise = { error: console.error, warn: console.warn }
  249. const logged: unknown[] = []
  250. console.error = (...args: unknown[]) => logged.push(args)
  251. console.warn = (...args: unknown[]) => logged.push(args)
  252. try {
  253. adapter.reset([
  254. compactSummary(80, [{ type: 'text', text: '窗外范围' }]),
  255. checkpoint(81, 80, { start: 3, end: 40, sourceEventSeqs: [80, 3, 40] }),
  256. ev.user(82, '压缩后的新问题'),
  257. ])
  258. expect(adapter.nodes().map(n => [n.kind, n.seq])).toEqual([['compaction', 81], ['user', 82]])
  259. expect(adapter.nodes()[0]).toMatchObject({ summary: '窗外范围' })
  260. } finally {
  261. console.error = noise.error
  262. console.warn = noise.warn
  263. }
  264. expect(logged).toEqual([])
  265. })
  266. it('treats an APPENDING plugin-sourced user/message as injected context, not a compaction', () => {
  267. // A session-reference card carries the same plugin source shape; only the
  268. // replacement marker makes an event a checkpoint.
  269. const adapter = new TranscriptAdapter()
  270. adapter.reset([
  271. at(0, { type: 'user/message', surfaceOp: 'append', data: createUserMessage({
  272. content: [{ type: 'text', text: '注入的上下文' }],
  273. source: { kind: 'plugin', plugin: 'compact', form: 'instructions' },
  274. }) }),
  275. ])
  276. expect(adapter.nodes()).toMatchObject([{
  277. kind: 'context',
  278. seq: 0,
  279. provenance: { role: 'inject', label: 'compact' },
  280. form: 'instructions',
  281. }])
  282. })
  283. it('ignores a foreign plugin s replacement user/message', () => {
  284. const adapter = new TranscriptAdapter()
  285. adapter.reset([
  286. ev.user(0, '保留'),
  287. at(1, { type: 'user/message', surfaceOp: { op: 'replace', start: 0, end: 0 }, sourceEventSeqs: [0], data: createUserMessage({
  288. content: [{ type: 'text', text: '别的插件重写' }],
  289. source: { kind: 'plugin', plugin: 'not-compact' },
  290. }) }),
  291. ])
  292. expect(adapter.nodes().map(n => [n.kind, n.seq])).toEqual([['user', 0]])
  293. })
  294. it.each([
  295. ['absent provenance', undefined],
  296. ['text-less summary blocks', compactSummary(1, [{ type: 'image', data: 'nope' }])],
  297. ['a whitespace-only summary', compactSummary(1, [{ type: 'text', text: ' ' }])],
  298. ['an empty summary array', compactSummary(1, [])],
  299. ['a non-array summary', compactSummary(1, 'plain string')],
  300. ])('degrades %s to a non-expandable marker', (_label, summary) => {
  301. const adapter = new TranscriptAdapter()
  302. adapter.reset([
  303. ...(summary === undefined ? [] : [summary]),
  304. checkpoint(2, 1, { start: 0, end: 0, sourceEventSeqs: [1, 0] }),
  305. ])
  306. expect(adapter.nodes()).toEqual([
  307. { kind: 'compaction', seq: 2, time: 1_700_000_000_002, summary: null },
  308. ])
  309. })
  310. it('keeps the text of a mixed-block summary, skipping the blocks it cannot render', () => {
  311. // ContentBlock is merge-extensible and the payload type is ContentBlock[],
  312. // so a non-text block must not discard recoverable text beside it.
  313. const adapter = new TranscriptAdapter()
  314. adapter.reset([
  315. compactSummary(1, [{ type: 'text', text: '可用摘要' }, { type: 'image', data: 'nope' }]),
  316. checkpoint(2, 1, { start: 0, end: 0, sourceEventSeqs: [1, 0] }),
  317. ])
  318. expect(adapter.nodes()).toEqual([
  319. { kind: 'compaction', seq: 2, time: 1_700_000_000_002, summary: '可用摘要' },
  320. ])
  321. })
  322. it('leaves the summary null when the checkpoint records no provenance at all', () => {
  323. const adapter = new TranscriptAdapter()
  324. adapter.reset([at(2, {
  325. type: 'user/message',
  326. surfaceOp: { op: 'replace', start: 0, end: 0 },
  327. data: createUserMessage({
  328. content: [{ type: 'text', text: '<context_checkpoint>x</context_checkpoint>' }],
  329. source: { kind: 'plugin', plugin: 'compact' },
  330. }),
  331. })])
  332. expect(adapter.nodes()).toEqual([{ kind: 'compaction', seq: 2, time: 1_700_000_000_002, summary: null }])
  333. })
  334. it('skips a non-summary provenance seq before reaching the real one', () => {
  335. const adapter = new TranscriptAdapter()
  336. adapter.reset([
  337. ev.user(0, '被压缩的问题'),
  338. at(1, { type: 'compact/start', data: { turn: 0 } }),
  339. compactSummary(2, [{ type: 'text', text: '第三个来源才是摘要' }]),
  340. checkpoint(3, 2, { start: 0, end: 0, sourceEventSeqs: [1, 2, 0] }),
  341. ])
  342. expect(adapter.nodes().at(-1)).toMatchObject({ kind: 'compaction', summary: '第三个来源才是摘要' })
  343. })
  344. it('resolves the summary once an older page supplies the provenance', () => {
  345. const adapter = new TranscriptAdapter()
  346. const landed = checkpoint(8, 7, { start: 0, end: 0, sourceEventSeqs: [7, 0] })
  347. adapter.reset([landed])
  348. expect(adapter.nodes()[0]).toMatchObject({ kind: 'compaction', summary: null })
  349. adapter.reset([compactSummary(7, [{ type: 'text', text: '分页补齐的摘要' }]), landed])
  350. expect(adapter.nodes()[0]).toMatchObject({ kind: 'compaction', summary: '分页补齐的摘要' })
  351. })
  352. it('creates the marker on the live append path', () => {
  353. const adapter = new TranscriptAdapter()
  354. adapter.reset(plainTurn(0, 0, 'a', 'b'))
  355. adapter.append(compactSummary(6, [{ type: 'text', text: '直播摘要' }]))
  356. adapter.append(checkpoint(7, 6, { start: 1, end: 3, sourceEventSeqs: [6, 1, 3] }))
  357. const nodes = adapter.nodes()
  358. // The compacted history is still there; the marker is one more row after it.
  359. expect(nodes.map(n => [n.kind, n.seq])).toEqual([['user', 1], ['assistant', 3], ['compaction', 7]])
  360. expect(nodes.at(-1)).toMatchObject({ kind: 'compaction', seq: 7, summary: '直播摘要' })
  361. })
  362. })
  363. it('returns call:null for a tool-result whose call fell outside the window', () => {
  364. const adapter = new TranscriptAdapter()
  365. adapter.reset([ev.toolResult(50, 3, 'outside-call', '孤儿结果')])
  366. expect(adapter.nodes()[0]).toMatchObject({ kind: 'tool-result', callId: 'outside-call', call: null })
  367. })
  368. it('materializes a tool-result error field when present', () => {
  369. const adapter = new TranscriptAdapter()
  370. adapter.reset([
  371. at(0, { type: 'tool/result', surfaceOp: 'append', data: {
  372. turn: 0, step: 0,
  373. message: createToolResultMessage({ callId: CallId('c1'), content: [], isError: true }),
  374. error: { name: 'Boom', code: 'boom' },
  375. } }),
  376. ])
  377. expect(adapter.nodes()[0]).toMatchObject({ kind: 'tool-result', isError: true, error: { code: 'boom' } })
  378. })
  379. it('attaches wire views to the materialized result node', () => {
  380. const adapter = new TranscriptAdapter()
  381. const callView = { for: 'call' as const, view: { card: 'terminal' as const, command: 'ls' } }
  382. const resultView = { for: 'result' as const, view: { card: 'generic' as const, title: '完成' } }
  383. adapter.reset([
  384. ev.toolCall(0, 1, 'c1', 'bash', '{"cmd":"ls"}'),
  385. ev.toolResult(1, 1, 'c1', 'listing'),
  386. ], [callView, resultView] as never)
  387. expect(adapter.nodes().find(n => n.kind === 'tool-result')).toMatchObject({
  388. callView: { card: 'terminal' }, resultView: { card: 'generic', title: '完成' },
  389. })
  390. })
  391. it('attaches views on the live append path and defaults to null without views', () => {
  392. const adapter = new TranscriptAdapter()
  393. adapter.reset(plainTurn(0, 0, 'a', 'b')) // no views argument
  394. adapter.append(ev.toolCall(6, 1, 'c2', 'echo', '{}'), { for: 'call', view: { card: 'generic', title: '回声' } } as never)
  395. adapter.append(ev.toolResult(7, 1, 'c2', 'ok')) // no view on the result
  396. expect(adapter.nodes().find(n => n.kind === 'tool-result')).toMatchObject({
  397. callView: { title: '回声' }, resultView: null,
  398. })
  399. })
  400. it('leaves callView null when the paired call fell outside the window (cross-page break)', () => {
  401. const adapter = new TranscriptAdapter()
  402. const resultView = { for: 'result' as const, view: { card: 'generic' as const, title: '孤儿' } }
  403. adapter.reset([ev.toolResult(50, 3, 'outside', '窗外配对')], [resultView] as never)
  404. expect(adapter.nodes()[0]).toMatchObject({
  405. kind: 'tool-result', call: null, callView: null, resultView: { title: '孤儿' },
  406. })
  407. })
  408. describe('command lifecycle nodes', () => {
  409. it('folds a run/done pair into one settled node merged into flow order by seq', () => {
  410. const adapter = new TranscriptAdapter()
  411. adapter.reset([
  412. ev.user(0, '先说话'),
  413. ev.commandRun(1, 'cmd-1', 'plan'),
  414. ev.commandDone(2, 'cmd-1', 'success', '已进入 plan mode'),
  415. ev.assistant(3, 0, '然后回答'),
  416. ])
  417. const nodes = adapter.nodes()
  418. expect(nodes.map(n => [n.kind, n.seq])).toEqual([['user', 0], ['command', 1], ['assistant', 3]])
  419. expect(nodes[1]).toMatchObject({
  420. kind: 'command', commandId: 'cmd-1', name: 'plan', args: '',
  421. outcome: { kind: 'success', text: '已进入 plan mode' },
  422. })
  423. })
  424. it('renders a run with no done as still executing (outcome null)', () => {
  425. const adapter = new TranscriptAdapter()
  426. adapter.reset([ev.commandRun(0, 'cmd-2', 'goal', ' ship it')])
  427. expect(adapter.nodes()[0]).toMatchObject({ kind: 'command', name: 'goal', args: ' ship it', outcome: null })
  428. })
  429. it('represents command input omitted by the host as null', () => {
  430. const adapter = new TranscriptAdapter()
  431. adapter.reset([ev.commandRunWithoutInput(0, 'cmd-private', 'feedback')])
  432. expect(adapter.nodes()[0]).toMatchObject({
  433. kind: 'command', name: 'feedback', args: null, outcome: null,
  434. })
  435. })
  436. it('soft-falls a done-only window into a node built from the done (cross-window cut)', () => {
  437. const adapter = new TranscriptAdapter()
  438. adapter.reset([ev.commandDone(80, 'cmd-3', 'error', '失败了')])
  439. expect(adapter.nodes()[0]).toMatchObject({
  440. kind: 'command', seq: 80, commandId: 'cmd-3', name: null, args: null,
  441. outcome: { kind: 'error', text: '失败了' },
  442. })
  443. })
  444. it('settles a live-appended done in place, keeping the node at the run seq', () => {
  445. const adapter = new TranscriptAdapter()
  446. adapter.reset(plainTurn(0, 0, 'q', 'a'))
  447. adapter.append(ev.commandRun(6, 'cmd-4', 'clear'))
  448. const running = adapter.nodes().find(n => n.kind === 'command')
  449. expect(running).toMatchObject({ outcome: null })
  450. adapter.append(ev.commandDone(7, 'cmd-4'))
  451. const settled = adapter.nodes().find(n => n.kind === 'command')
  452. expect(settled).toMatchObject({ seq: 6, outcome: { kind: 'success' } })
  453. // Settlement replaced the node object rather than mutating the published one.
  454. expect(settled).not.toBe(running)
  455. })
  456. it('tails command nodes whose seq is past every transcript node', () => {
  457. const adapter = new TranscriptAdapter()
  458. adapter.reset([ev.user(0, '问'), ev.commandRun(1, 'cmd-tail', 'plan')])
  459. expect(adapter.nodes().map(n => n.kind)).toEqual(['user', 'command'])
  460. })
  461. it('renders the /compact row alongside the marker its own command produced', () => {
  462. // The row that reports the compaction is a command node; dropping command
  463. // folding would delete it together with every other slash-command row.
  464. const adapter = new TranscriptAdapter()
  465. adapter.reset([
  466. ev.user(0, '压缩前的问题'),
  467. ev.commandRun(1, 'cmd-compact', 'compact'),
  468. compactSummary(2, [{ type: 'text', text: '手动压缩摘要' }]),
  469. checkpoint(3, 2, { start: 0, end: 0, sourceEventSeqs: [2, 0] }),
  470. ev.commandDone(4, 'cmd-compact', 'success', '已压缩'),
  471. ])
  472. const nodes = adapter.nodes()
  473. expect(nodes.map(n => [n.kind, n.seq])).toEqual([['user', 0], ['command', 1], ['compaction', 3]])
  474. expect(nodes[1]).toMatchObject({ name: 'compact', outcome: { kind: 'success', text: '已压缩' } })
  475. })
  476. })
  477. describe('assistant timing', () => {
  478. const base = 1_700_000_000_000
  479. it('derives step timing across a window rebuild (start + first token + completion)', () => {
  480. const adapter = new TranscriptAdapter()
  481. adapter.reset([
  482. ev.turnStart(0, 0),
  483. ev.user(1, '问'),
  484. ev.stepStart(2, 0),
  485. ev.chunkStart(3, 0),
  486. ev.chunkText(4, 0, '答'),
  487. ev.chunkText(5, 0, '案'),
  488. ev.assistant(6, 0, '答案'),
  489. ev.turnEnd(7, 0),
  490. ])
  491. const assistant = adapter.nodes().find(n => n.kind === 'assistant')
  492. expect(assistant).toMatchObject({
  493. timing: { stepStartTime: base + 2, firstTokenTime: base + 4, completedTime: base + 6 },
  494. })
  495. })
  496. it('derives the same timing on the live append path, first token winning once', () => {
  497. const adapter = new TranscriptAdapter()
  498. adapter.reset([ev.user(0, '问')])
  499. adapter.append(ev.stepStart(1, 0))
  500. adapter.append(ev.chunkText(2, 0, '首'))
  501. adapter.append(ev.chunkText(3, 0, '次'))
  502. adapter.append(ev.assistant(4, 0, '首次'))
  503. const assistant = adapter.nodes().find(n => n.kind === 'assistant')
  504. expect(assistant).toMatchObject({
  505. timing: { stepStartTime: base + 1, firstTokenTime: base + 2, completedTime: base + 4 },
  506. })
  507. })
  508. it('soft-falls to null boundaries when the step opening fell outside the window', () => {
  509. const adapter = new TranscriptAdapter()
  510. adapter.reset([ev.assistant(100, 0, '被切窗的答案')])
  511. const assistant = adapter.nodes().find(n => n.kind === 'assistant')
  512. expect(assistant).toMatchObject({
  513. timing: { stepStartTime: null, firstTokenTime: null, completedTime: base + 100 },
  514. })
  515. })
  516. })
  517. })