compaction-basic.spec.ts 75 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613161416151616161716181619162016211622162316241625162616271628162916301631163216331634163516361637163816391640164116421643164416451646164716481649165016511652165316541655165616571658165916601661166216631664166516661667166816691670167116721673167416751676167716781679168016811682168316841685168616871688168916901691169216931694169516961697169816991700170117021703170417051706170717081709171017111712171317141715171617171718171917201721172217231724172517261727172817291730173117321733173417351736173717381739174017411742174317441745174617471748174917501751175217531754175517561757175817591760176117621763176417651766176717681769177017711772177317741775177617771778177917801781178217831784178517861787178817891790179117921793179417951796179717981799180018011802180318041805180618071808180918101811181218131814181518161817181818191820182118221823182418251826182718281829183018311832183318341835183618371838183918401841184218431844184518461847184818491850185118521853185418551856185718581859186018611862186318641865186618671868186918701871187218731874187518761877187818791880188118821883188418851886188718881889189018911892189318941895189618971898189919001901190219031904190519061907190819091910191119121913191419151916191719181919192019211922192319241925192619271928192919301931193219331934193519361937193819391940194119421943194419451946194719481949195019511952195319541955195619571958195919601961196219631964196519661967196819691970197119721973197419751976197719781979198019811982198319841985198619871988198919901991199219931994199519961997199819992000200120022003200420052006200720082009201020112012201320142015201620172018201920202021202220232024
  1. import { describe, expect, expectTypeOf, it, vi } from 'vitest'
  2. import { Context } from '@deepseek-ai/cordis'
  3. import { AttachmentId } from '@deepseek-ai/dsh-attachment'
  4. import BasicCompactionEngine from '@deepseek-ai/dsh-compaction-basic'
  5. import type { BasicCompactionConfig } from '@deepseek-ai/dsh-compaction-basic'
  6. import { selectCompactableRange } from '@deepseek-ai/dsh-compaction-basic/src/region.ts'
  7. import { frameSummary } from '@deepseek-ai/dsh-compaction-basic/src/summarizer.ts'
  8. import type { SummarizationInput, SummaryResult } from '@deepseek-ai/dsh-compaction-basic/src/summarizer.ts'
  9. import { CompactionId, toolPairingBalancedAfter, toolPairingBalancedBefore } from '@deepseek-ai/dsh-compaction'
  10. import {
  11. resolveCompactSpec,
  12. resolveConfig,
  13. resolveTargetPolicy,
  14. } from '@deepseek-ai/dsh-compaction-basic/src/config.ts'
  15. import type { CompactionResult } from '@deepseek-ai/dsh-compaction'
  16. import LlmRuntime, { createUserMessage, ToolCallId, CONTEXT_WINDOW_EXCEEDED_CODE, createToolResultMessage, LlmAdapter , createMessage } from '@deepseek-ai/dsh-llm'
  17. import type {
  18. ContentBlock,
  19. GenerateOptions,
  20. LlmFailure,
  21. LlmResolvedModelInfo,
  22. Message,
  23. StreamChunk,
  24. TokenUsage,
  25. } from '@deepseek-ai/dsh-llm'
  26. import SessionStore, { Session, SessionId } from '@deepseek-ai/dsh-session'
  27. import TokenMeter from '@deepseek-ai/dsh-token-meter'
  28. import { agentEvents, type Agent, type RequestErrorAction } from '@deepseek-ai/dsh-agent'
  29. import ToolResultPruner from '@deepseek-ai/dsh-compaction-tool-result-pruner'
  30. const SIGNAL = new AbortController().signal
  31. const MODEL = 'test-model'
  32. class ContextAdapter extends LlmAdapter {
  33. constructor(private readonly contextWindow: number) {
  34. super()
  35. }
  36. override resolveModel(provider: string, model: string): Promise<LlmResolvedModelInfo> {
  37. return Promise.resolve({
  38. provider,
  39. id: model,
  40. name: model,
  41. context: { contextWindow: this.contextWindow },
  42. })
  43. }
  44. override async * stream(): AsyncIterable<StreamChunk> {
  45. yield { type: 'finish', reason: { kind: 'stop' } }
  46. }
  47. }
  48. class RoutedContextAdapter extends LlmAdapter {
  49. constructor(private readonly windows: Readonly<Record<string, number>>) {
  50. super()
  51. }
  52. override resolveModel(provider: string, model: string): Promise<LlmResolvedModelInfo> {
  53. const contextWindow = this.windows[provider]
  54. return Promise.resolve({
  55. provider,
  56. id: model,
  57. name: model,
  58. ...contextWindow === undefined ? {} : { context: { contextWindow } },
  59. })
  60. }
  61. override async * stream(): AsyncIterable<StreamChunk> {
  62. yield { type: 'finish', reason: { kind: 'stop' } }
  63. }
  64. }
  65. function createContext(contextWindow = 1_000): Context {
  66. const ctx = new Context()
  67. void new LlmRuntime(ctx)
  68. void new TokenMeter(ctx)
  69. ctx.llm.registerAdapter([MODEL, 'actual', 'unlisted-provider'], new ContextAdapter(contextWindow))
  70. return ctx
  71. }
  72. function agent(session: Session, model?: string): Agent {
  73. return {
  74. session,
  75. options: model === undefined ? {} : { provider: model, model },
  76. } as Agent
  77. }
  78. /** Flatten every text fragment the summarizer received, recursing tool-result blocks. */
  79. function summarizedText(input: SummarizationInput): string {
  80. const collect = (blocks: readonly ContentBlock[]): string =>
  81. blocks.map(block =>
  82. block.type === 'text' ? block.text
  83. : block.type === 'tool-result' ? collect(block.content)
  84. : '').join('\n')
  85. return input.messages.map(message => collect(message.content)).join('\n')
  86. }
  87. /** A minimal replayed prefix carrying one user message of the given text. */
  88. function promptInput(text: string): SummarizationInput {
  89. return { messages: [createUserMessage({
  90. content: [{ type: 'text', text }],
  91. source: { kind: 'plugin', plugin: 'test' },
  92. })] }
  93. }
  94. /** Closed two-message turns followed by one open turn for durable compaction events. */
  95. function conversation(turns = 4, text = 'fixture '.repeat(40).trim()): Session {
  96. const session = Session.create(SessionId(`conversation-${turns}`))
  97. for (let turn = 1; turn <= turns; turn += 1) {
  98. session.append('turn/start', { turn })
  99. session.append('user/message', createUserMessage({
  100. content: [{ type: 'text', text: `${text} user ${turn}` }],
  101. source: { kind: 'user' },
  102. }), { surfaceOp: 'append' })
  103. session.append('step/start', { turn, step: 1 })
  104. if (turn === 1) {
  105. session.append('request/header', {
  106. header: { config: { provider: MODEL, model: MODEL } },
  107. reason: 'initial',
  108. })
  109. }
  110. session.append('assistant/message', {
  111. turn,
  112. step: 1,
  113. message: createMessage({
  114. role: 'assistant',
  115. content: [{ type: 'text', text: `${text} assistant ${turn}` }],
  116. source: {
  117. kind: 'model',
  118. ...{ provider: MODEL, model: MODEL },
  119. },
  120. }),
  121. }, { surfaceOp: 'append' })
  122. session.append('step/end', { turn, step: 1 })
  123. session.append('turn/end', { turn, reason: { kind: 'completed' } })
  124. }
  125. session.append('turn/start', {
  126. turn: turns + 1,
  127. })
  128. return session
  129. }
  130. function toolConversation(): Session {
  131. const session = Session.create(SessionId('tools'))
  132. for (let turn = 1; turn <= 3; turn += 1) {
  133. const callId = ToolCallId(`call-${turn}`)
  134. session.append('turn/start', { turn })
  135. session.append('user/message', createUserMessage({
  136. content: [{ type: 'text', text: `request ${turn} `.repeat(300) }],
  137. source: { kind: 'user' },
  138. }), { surfaceOp: 'append' })
  139. session.append('step/start', { turn, step: 1 })
  140. if (turn === 1) {
  141. session.append('request/header', {
  142. header: { config: { provider: MODEL, model: MODEL } },
  143. reason: 'initial',
  144. })
  145. }
  146. session.append('assistant/message', {
  147. turn,
  148. step: 1,
  149. message: createMessage({
  150. role: 'assistant',
  151. content: [
  152. { type: 'text', text: `calling ${turn} `.repeat(300) },
  153. { type: 'tool-call', id: callId, name: 'read', arguments: '{}' },
  154. ],
  155. source: {
  156. kind: 'model',
  157. ...{ provider: MODEL, model: MODEL },
  158. },
  159. }),
  160. }, { surfaceOp: 'append' })
  161. session.append('tool/call', { turn, step: 1, callId, name: 'read', arguments: '{}' })
  162. session.append('tool/result', {
  163. turn,
  164. step: 1,
  165. message: createToolResultMessage({
  166. callId,
  167. content: [{ type: 'text', text: `result ${turn} `.repeat(300) }],
  168. isError: false,
  169. }),
  170. }, { surfaceOp: 'append' })
  171. session.append('step/end', { turn, step: 1 })
  172. session.append('turn/end', { turn, reason: { kind: 'completed' } })
  173. }
  174. session.append('turn/start', { turn: 4 })
  175. return session
  176. }
  177. /** One closed routed tool step followed by an open turn for rewrite events. */
  178. function oversizedToolResult(chars = 3_000, withCompactablePrompt = false): Session {
  179. const session = Session.create(SessionId(`oversized-tool-${chars}`))
  180. const callId = ToolCallId('oversized')
  181. session.append('turn/start', { turn: 1 })
  182. if (withCompactablePrompt) {
  183. session.append('user/message', createUserMessage({
  184. content: [{ type: 'text', text: 'older history '.repeat(200) }],
  185. source: { kind: 'user' },
  186. }), { surfaceOp: 'append' })
  187. }
  188. session.append('step/start', { turn: 1, step: 1 })
  189. session.append('request/header', {
  190. header: { config: { provider: MODEL, model: MODEL } },
  191. reason: 'initial',
  192. })
  193. session.append('assistant/message', {
  194. turn: 1,
  195. step: 1,
  196. message: createMessage({
  197. role: 'assistant',
  198. content: [{ type: 'tool-call', id: callId, name: 'bash', arguments: '{}' }],
  199. source: {
  200. kind: 'model',
  201. ...{ provider: MODEL, model: MODEL },
  202. },
  203. }),
  204. }, { surfaceOp: 'append' })
  205. session.append('tool/call', { turn: 1, step: 1, callId, name: 'bash', arguments: '{}' })
  206. session.append('tool/result', {
  207. turn: 1,
  208. step: 1,
  209. message: createToolResultMessage({
  210. callId,
  211. content: [{ type: 'text', text: 'X'.repeat(chars) }],
  212. isError: false,
  213. }),
  214. meta: { presentation: 'preserved' },
  215. }, { surfaceOp: 'append' })
  216. session.append('step/end', { turn: 1, step: 1 })
  217. session.append('turn/end', { turn: 1, reason: { kind: 'completed' } })
  218. session.append('turn/start', { turn: 2 })
  219. return session
  220. }
  221. class TestCompactionEngine extends BasicCompactionEngine {
  222. summary: ContentBlock[] = [{ type: 'text', text: 'small checkpoint' }]
  223. rawOutput: ContentBlock[] | undefined
  224. usage: TokenUsage | undefined
  225. summaryProvider = 'summary-provider'
  226. summaryModel = 'summary-model'
  227. error: unknown
  228. mutateDuringSummary: (() => void) | undefined
  229. calls: Array<{ input: SummarizationInput; signal: AbortSignal | undefined }> = []
  230. override async summarize(
  231. input: SummarizationInput,
  232. _agent: Agent,
  233. signal?: AbortSignal,
  234. ): Promise<{
  235. summary: ContentBlock[]
  236. rawOutput?: ContentBlock[]
  237. provider: string
  238. model: string
  239. maxTokens?: number
  240. usage?: TokenUsage
  241. }> {
  242. this.calls.push({ input, signal })
  243. this.mutateDuringSummary?.()
  244. if (this.error !== undefined) throw this.error
  245. return {
  246. summary: this.summary,
  247. ...this.rawOutput === undefined ? {} : { rawOutput: this.rawOutput },
  248. provider: this.summaryProvider,
  249. model: this.summaryModel,
  250. maxTokens: 123,
  251. ...this.usage === undefined ? {} : { usage: this.usage },
  252. }
  253. }
  254. }
  255. function service(
  256. config: BasicCompactionConfig = { auto: false },
  257. ctx = createContext(),
  258. ): TestCompactionEngine {
  259. return new TestCompactionEngine(ctx, config)
  260. }
  261. async function compactIfNeeded(
  262. compact: BasicCompactionEngine,
  263. session: Session,
  264. trigger: 'pressure' | 'context-overflow' = 'pressure',
  265. model: string | undefined = MODEL,
  266. ): Promise<CompactionResult | null> {
  267. return compact.compactIfNeeded(agent(session, model), trigger, SIGNAL)
  268. }
  269. describe('compact configuration and defaults', () => {
  270. it('uses low-friction service-wide defaults', () => {
  271. const resolved = resolveConfig({})
  272. expect(resolved).toEqual({
  273. thresholdRatio: 0.8,
  274. retainRatio: 0.16,
  275. summarizationProvider: '',
  276. summarizationModel: '',
  277. maxTokens: 8192,
  278. compactionRetries: 1,
  279. maxOverflowRetries: 1,
  280. modelPolicies: [],
  281. auto: true,
  282. })
  283. expect(Object.isFrozen(resolved)).toBe(true)
  284. })
  285. it('resolves threshold and retention overrides independently', () => {
  286. const thresholdOnly = resolveConfig({
  287. thresholdRatio: 0.5,
  288. })
  289. expect(thresholdOnly).toMatchObject({
  290. thresholdRatio: 0.5,
  291. retainRatio: 0.16,
  292. })
  293. const retentionOnly = resolveConfig({
  294. retainTokens: 70,
  295. })
  296. expect(retentionOnly).toMatchObject({
  297. thresholdRatio: 0.8,
  298. retainTokens: 70,
  299. })
  300. expect(retentionOnly).not.toHaveProperty('retainRatio')
  301. })
  302. it('merges exact provider/model policy overrides and scales ratios per model', () => {
  303. const config = resolveConfig({
  304. thresholdRatio: 0.8,
  305. retainRatio: 0.1,
  306. modelPolicies: [{
  307. provider: 'small-provider',
  308. model: 'shared-id',
  309. thresholdRatio: 0.5,
  310. retainTokens: 120,
  311. }],
  312. })
  313. const small = resolveTargetPolicy(config, {
  314. provider: 'small-provider',
  315. model: 'shared-id',
  316. })
  317. const otherProvider = resolveTargetPolicy(config, {
  318. provider: 'large-provider',
  319. model: 'shared-id',
  320. })
  321. expect(resolveCompactSpec(small, 1_000)).toMatchObject({
  322. thresholdTokens: 500,
  323. retainTokens: 120,
  324. })
  325. expect(resolveCompactSpec(otherProvider, 2_000)).toMatchObject({
  326. thresholdTokens: 1_600,
  327. retainTokens: 200,
  328. })
  329. const ratioOverride = resolveTargetPolicy(resolveConfig({
  330. retainTokens: 200,
  331. modelPolicies: [{
  332. provider: 'ratio-provider',
  333. model: 'ratio-model',
  334. thresholdRatio: 0.6,
  335. retainRatio: 0.2,
  336. summarizationProvider: 'summary-provider',
  337. summarizationModel: 'summary-model',
  338. maxTokens: 512,
  339. compactionRetries: 2,
  340. maxOverflowRetries: 3,
  341. }],
  342. }), { provider: 'ratio-provider', model: 'ratio-model' })
  343. expect(resolveCompactSpec(ratioOverride, 2_000)).toMatchObject({
  344. thresholdTokens: 1_200,
  345. retainTokens: 400,
  346. summarizationProvider: 'summary-provider',
  347. summarizationModel: 'summary-model',
  348. maxTokens: 512,
  349. compactionRetries: 2,
  350. maxOverflowRetries: 3,
  351. })
  352. })
  353. it('inherits, clears, and replaces the summarization target as a pair', () => {
  354. const config = resolveConfig({
  355. summarizationProvider: 'default-provider',
  356. summarizationModel: 'default-model',
  357. modelPolicies: [
  358. { provider: 'inherit-provider', model: MODEL },
  359. {
  360. provider: 'clear-provider',
  361. model: MODEL,
  362. summarizationProvider: '',
  363. summarizationModel: '',
  364. },
  365. {
  366. provider: 'replace-provider',
  367. model: MODEL,
  368. summarizationProvider: 'replacement-provider',
  369. summarizationModel: 'replacement-model',
  370. },
  371. ],
  372. })
  373. expect(resolveTargetPolicy(config, { provider: 'inherit-provider', model: MODEL }))
  374. .toMatchObject({
  375. summarizationProvider: 'default-provider',
  376. summarizationModel: 'default-model',
  377. })
  378. expect(resolveTargetPolicy(config, { provider: 'clear-provider', model: MODEL }))
  379. .toMatchObject({ summarizationProvider: '', summarizationModel: '' })
  380. expect(resolveTargetPolicy(config, { provider: 'replace-provider', model: MODEL }))
  381. .toMatchObject({
  382. summarizationProvider: 'replacement-provider',
  383. summarizationModel: 'replacement-model',
  384. })
  385. })
  386. it('validates common values and pressure-policy invariants', () => {
  387. const bad = [
  388. [{ maxTokens: 0 }, /maxTokens/],
  389. [{ compactionRetries: -1 }, /compactionRetries/],
  390. [{ maxOverflowRetries: -1 }, /maxOverflowRetries/],
  391. [{ auto: 'yes' }, /auto must be a boolean/],
  392. [{ summarizationProvider: 1 }, /summarizationProvider must be a string/],
  393. [{ summarizationModel: 1 }, /summarizationModel must be a string/],
  394. [{ summarizationProvider: MODEL }, /must be set together/],
  395. [{ summarizationModel: MODEL }, /must be set together/],
  396. [{ summarizationProvider: '' }, /must be set together/],
  397. [{ summarizationModel: '' }, /must be set together/],
  398. [{ thresholdRatio: 0 }, /number in \(0, 1\]/],
  399. [{ thresholdRatio: 1.1 }, /number in \(0, 1\]/],
  400. [{ retainRatio: 0.9 }, /retainRatio \(0.9\) must be less than the resolved thresholdRatio \(0.8\)/],
  401. [{ thresholdRatio: 0.1 }, /retainRatio \(0.16\) must be less than the resolved thresholdRatio \(0.1\)/],
  402. [{ retainTokens: -1 }, /non-negative integer/],
  403. [{ retainRatio: 0.2, retainTokens: 100 }, /mutually exclusive/],
  404. [{ modelPolicies: {} }, /modelPolicies must be an array/],
  405. [{ modelPolicies: [1] }, /modelPolicies\[0\] must be an object/],
  406. [{ modelPolicies: [null] }, /modelPolicies\[0\] must be an object/],
  407. [{ modelPolicies: [[]] }, /modelPolicies\[0\] must be an object/],
  408. [{ modelPolicies: [{ provider: 1, model: MODEL }] }, /provider must be a non-empty string/],
  409. [{ modelPolicies: [{ provider: '', model: MODEL }] }, /provider must be a non-empty string/],
  410. [{ modelPolicies: [{ provider: MODEL, model: 1 }] }, /model must be a non-empty string/],
  411. [{ modelPolicies: [{ provider: MODEL, model: '' }] }, /model must be a non-empty string/],
  412. [{ modelPolicies: [{ provider: MODEL, model: MODEL, summarizationProvider: 1 }] }, /summarizationProvider must be a string/],
  413. [{
  414. summarizationProvider: 'default-provider',
  415. summarizationModel: 'default-model',
  416. modelPolicies: [{ provider: MODEL, model: MODEL, summarizationModel: '' }],
  417. }, /modelPolicies\[0\].*must be set together/],
  418. [{
  419. summarizationProvider: 'default-provider',
  420. summarizationModel: 'default-model',
  421. modelPolicies: [{ provider: MODEL, model: MODEL, summarizationProvider: '' }],
  422. }, /modelPolicies\[0\].*must be set together/],
  423. [{ modelPolicies: [{ provider: MODEL, model: MODEL, retainRatio: 0.2, retainTokens: 100 }] }, /mutually exclusive/],
  424. [
  425. { modelPolicies: [{ provider: MODEL, model: MODEL, thresholdRatio: 0.1 }] },
  426. /modelPolicies\[0\]: retainRatio \(0.16\).*thresholdRatio \(0.1\)/,
  427. ],
  428. [
  429. { modelPolicies: [{ provider: MODEL, model: MODEL, retainRatio: 0.9 }] },
  430. /modelPolicies\[0\]: retainRatio \(0.9\).*thresholdRatio \(0.8\)/,
  431. ],
  432. [{ modelPolicies: [{ provider: MODEL, model: MODEL }, { provider: MODEL, model: MODEL }] }, /duplicate model policy/],
  433. [{ models: { [MODEL]: { retainTokens: 10 } } }, /BasicCompactionConfig: unknown key "models"/],
  434. [{ thresholdRato: 0.5 }, /BasicCompactionConfig: unknown key "thresholdRato"/],
  435. ] as Array<[unknown, RegExp]>
  436. for (const [config, pattern] of bad) {
  437. expect(() => resolveConfig(config as BasicCompactionConfig)).toThrow(pattern)
  438. }
  439. const invalidPressure = resolveTargetPolicy(resolveConfig({
  440. thresholdRatio: 0.5,
  441. retainTokens: 500,
  442. }), { provider: MODEL, model: MODEL })
  443. expect(() => resolveCompactSpec(invalidPressure, 1_000)).toThrow(/less than threshold/)
  444. expect(() => resolveCompactSpec(invalidPressure, 1.5)).toThrow(/positive integer/)
  445. expect(() => resolveCompactSpec(invalidPressure, 0)).toThrow(/positive integer/)
  446. })
  447. })
  448. describe('pressure measurement and retention', () => {
  449. const compactConfig: BasicCompactionConfig = {
  450. auto: false,
  451. thresholdRatio: 0.5,
  452. retainTokens: 180,
  453. }
  454. it('skips when no durable routed model exists instead of using AgentOptions fallback', async () => {
  455. const compact = service(compactConfig)
  456. const session = Session.create(SessionId('headerless'))
  457. session.append('turn/start', { turn: 1 })
  458. await expect(compact.compactIfNeeded(agent(session, MODEL), 'pressure', SIGNAL))
  459. .resolves.toBeNull()
  460. expect(compact.calls).toHaveLength(0)
  461. })
  462. it('meters an unlisted model when its provider adapter supplies context metadata', async () => {
  463. const compact = service(compactConfig)
  464. const session = conversation()
  465. session.append('request/header', {
  466. header: { config: { provider: 'unlisted-provider', model: 'unlisted-model' } },
  467. reason: 'resume',
  468. })
  469. await expect(compactIfNeeded(compact, session))
  470. .resolves.not.toBeNull()
  471. })
  472. it('forwards turn cancellation to proactive model metadata resolution', async () => {
  473. const ctx = createContext()
  474. const resolveModelInfo = vi.spyOn(ctx.llm, 'resolveModelInfo')
  475. const compact = service(compactConfig, ctx)
  476. const session = conversation()
  477. const signal = new AbortController().signal
  478. await expect(compact.compactIfNeeded(agent(session, MODEL), 'pressure', signal))
  479. .resolves.not.toBeNull()
  480. expect(resolveModelInfo).toHaveBeenCalledWith(MODEL, MODEL, signal)
  481. })
  482. it('re-resolves capacity after a same-model-id provider switch in one session', async () => {
  483. const ctx = new Context()
  484. void new LlmRuntime(ctx)
  485. void new TokenMeter(ctx)
  486. ctx.llm.registerAdapter(['large', 'small'], new RoutedContextAdapter({
  487. large: 10_000,
  488. small: 1_000,
  489. }))
  490. const compact = service({
  491. auto: false,
  492. thresholdRatio: 0.5,
  493. retainRatio: 0.1,
  494. }, ctx)
  495. const session = conversation(4)
  496. session.append('request/header', {
  497. header: { config: { provider: 'large', model: 'shared-id' } },
  498. reason: 'resume',
  499. })
  500. await expect(compactIfNeeded(compact, session)).resolves.toBeNull()
  501. session.append('request/header', {
  502. header: { config: { provider: 'small', model: 'shared-id' } },
  503. reason: 'change',
  504. })
  505. await expect(compactIfNeeded(compact, session)).resolves.not.toBeNull()
  506. })
  507. it('requires capacity only for proactive pressure, not provider-confirmed overflow', async () => {
  508. const ctx = new Context()
  509. void new LlmRuntime(ctx)
  510. void new TokenMeter(ctx)
  511. ctx.llm.registerAdapter(['unknown-context'], new ContextAdapter(1_000))
  512. vi.spyOn(ctx.llm, 'resolveModelInfo').mockImplementation((provider, model) => Promise.resolve({
  513. provider,
  514. id: model,
  515. name: model,
  516. }))
  517. const compact = service(compactConfig, ctx)
  518. const session = conversation(4)
  519. session.append('request/header', {
  520. header: { config: { provider: 'unknown-context', model: 'model' } },
  521. reason: 'resume',
  522. })
  523. await expect(compactIfNeeded(compact, session, 'pressure'))
  524. .rejects.toThrow(/no context capacity for unknown-context\/model/)
  525. await expect(compactIfNeeded(compact, session, 'context-overflow'))
  526. .resolves.not.toBeNull()
  527. })
  528. it('declines forced overflow when the whole surface is one indivisible tool pair', async () => {
  529. const compact = service(compactConfig)
  530. const session = Session.create(SessionId('single-tool-pair'))
  531. const callId = ToolCallId('single-call')
  532. session.append('turn/start', { turn: 1 })
  533. session.append('step/start', { turn: 1, step: 1 })
  534. session.append('request/header', {
  535. header: { config: { provider: MODEL, model: MODEL } },
  536. reason: 'initial',
  537. })
  538. session.append('assistant/message', {
  539. turn: 1,
  540. step: 1,
  541. message: createMessage({
  542. role: 'assistant',
  543. content: [{ type: 'tool-call', id: callId, name: 'read', arguments: '{}' }],
  544. source: {
  545. kind: 'model',
  546. ...{ provider: MODEL, model: MODEL },
  547. },
  548. }),
  549. }, { surfaceOp: 'append' })
  550. session.append('tool/call', { turn: 1, step: 1, callId, name: 'read', arguments: '{}' })
  551. session.append('tool/result', {
  552. turn: 1,
  553. step: 1,
  554. message: createToolResultMessage({
  555. callId,
  556. content: [{ type: 'text', text: 'result' }],
  557. isError: false,
  558. }),
  559. }, { surfaceOp: 'append' })
  560. session.append('step/end', { turn: 1, step: 1 })
  561. const generation = session.surface.replaceGeneration
  562. await expect(compactIfNeeded(compact, session, 'context-overflow')).resolves.toBeNull()
  563. expect(session.surface.replaceGeneration).toBe(generation)
  564. expect(session.events.some(event => event.type === 'compaction/start')).toBe(false)
  565. })
  566. it('does nothing below threshold and compacts a priced head above threshold', async () => {
  567. const compact = service(compactConfig)
  568. expect(await compactIfNeeded(compact, conversation(2))).toBeNull()
  569. const session = conversation(4)
  570. const result = await compactIfNeeded(compact, session)
  571. expect(result).not.toBeNull()
  572. expect(result?.shadowedSeqs.length).toBeGreaterThan(2)
  573. expect(session.surface.nodes.length).toBeLessThan(8)
  574. })
  575. it('counts the durable routed request envelope without putting it on the surface', async () => {
  576. const compact = service({
  577. auto: false,
  578. thresholdRatio: 0.9,
  579. retainTokens: 50,
  580. })
  581. const session = conversation(2, 'x'.repeat(600))
  582. expect(await compactIfNeeded(compact, session)).toBeNull()
  583. session.append('request/header', {
  584. header: {
  585. config: { provider: MODEL, model: MODEL },
  586. system: 's'.repeat(2_000),
  587. },
  588. reason: 'resume',
  589. })
  590. const result = await compactIfNeeded(compact, session)
  591. expect(result).not.toBeNull()
  592. })
  593. it('uses the latest logged request envelope without an AgentOptions override', async () => {
  594. const ctx = createContext()
  595. const compact = service({
  596. auto: false,
  597. thresholdRatio: 0.5,
  598. retainTokens: 180,
  599. }, ctx)
  600. const session = conversation(4)
  601. session.append('request/header', {
  602. header: { config: { provider: 'actual', model: 'actual' } },
  603. reason: 'initial',
  604. })
  605. const measure = vi.spyOn(ctx.tokenMeter, 'measure')
  606. const result = await compactIfNeeded(compact, session, 'pressure', 'fallback')
  607. expect(result).not.toBeNull()
  608. expect(session.requestHeader()?.config.model).toBe('actual')
  609. expect(measure.mock.calls[0]).toEqual([session])
  610. })
  611. it('declines when envelope pressure is high but the surface has no compactable range', async () => {
  612. const compact = service(compactConfig)
  613. const empty = Session.create(SessionId('empty'))
  614. empty.append('turn/start', { turn: 1 })
  615. empty.append('request/header', {
  616. header: { config: { provider: MODEL, model: MODEL }, system: 'x'.repeat(100_000) },
  617. reason: 'initial',
  618. })
  619. expect(await compactIfNeeded(compact, empty)).toBeNull()
  620. const retained = conversation(1)
  621. retained.append('request/header', {
  622. header: { config: { provider: MODEL, model: MODEL }, system: 'x'.repeat(100_000) },
  623. reason: 'resume',
  624. })
  625. expect(await compactIfNeeded(compact, retained)).toBeNull()
  626. })
  627. it('uses one unified measurement for each pressure-and-retention decision', async () => {
  628. const ctx = createContext()
  629. const compact = service(compactConfig, ctx)
  630. const measure = vi.spyOn(ctx.tokenMeter, 'measure')
  631. const stop = new Error('stop after first decision')
  632. vi.spyOn(compact, 'compactRegion').mockRejectedValueOnce(stop)
  633. await expect(compactIfNeeded(compact, conversation(4))).rejects.toBe(stop)
  634. expect(measure).toHaveBeenCalledTimes(1)
  635. })
  636. it('bounds retries when a shrinking checkpoint remains above threshold', async () => {
  637. const compact = service({
  638. auto: false,
  639. compactionRetries: 0,
  640. thresholdRatio: 0.3,
  641. retainTokens: 180,
  642. })
  643. compact.summary = Array.from({ length: 7 }, (_, index) => ({
  644. type: 'text',
  645. text: `summary ${index}`,
  646. }))
  647. await expect(compactIfNeeded(compact, conversation(4)))
  648. .rejects.toThrow(/still above threshold after 1 compaction attempts/)
  649. })
  650. it('rounds a retention cut head-ward to preserve tool-call/result pairing', async () => {
  651. const compact = service({
  652. auto: false,
  653. thresholdRatio: 0.8,
  654. retainTokens: 80,
  655. }, createContext(4_000))
  656. const session = toolConversation()
  657. const result = await compactIfNeeded(compact, session)
  658. expect(result).not.toBeNull()
  659. const messages = session.deriveMessages()
  660. const calls = new Set<string>()
  661. for (const message of messages) {
  662. for (const block of message.content) {
  663. if (block.type === 'tool-call') calls.add(block.id)
  664. if (block.type === 'tool-result') expect(calls.has(block.toolCallId)).toBe(true)
  665. }
  666. }
  667. })
  668. it('rejects a priced surface that is not the current positional surface', () => {
  669. const ctx = createContext()
  670. const session = conversation(2)
  671. const priced = ctx.tokenMeter.measure(session)
  672. expect(() => selectCompactableRange(session, {
  673. ...priced,
  674. nodes: priced.nodes.slice(1),
  675. }, 1)).toThrow(/does not match/)
  676. })
  677. it('declines when rounding a cut would consume the only tool pair', () => {
  678. const ctx = createContext()
  679. const session = Session.create(SessionId('one-tool-pair'))
  680. const callId = ToolCallId('only')
  681. session.append('turn/start', { turn: 1 })
  682. session.append('step/start', { turn: 1, step: 1 })
  683. session.append('assistant/message', {
  684. turn: 1,
  685. step: 1,
  686. message: createMessage({
  687. role: 'assistant',
  688. content: [{ type: 'tool-call', id: callId, name: 'read', arguments: '{}' }],
  689. source: {
  690. kind: 'model',
  691. ...{ provider: MODEL, model: MODEL },
  692. },
  693. }),
  694. }, { surfaceOp: 'append' })
  695. session.append('tool/call', { turn: 1, step: 1, callId, name: 'read', arguments: '{}' })
  696. session.append('tool/result', {
  697. turn: 1,
  698. step: 1,
  699. message: createToolResultMessage({
  700. callId,
  701. content: [{ type: 'text', text: 'result' }],
  702. isError: false,
  703. }),
  704. }, { surfaceOp: 'append' })
  705. session.append('step/end', { turn: 1, step: 1 })
  706. const priced = ctx.tokenMeter.measure(session)
  707. expect(selectCompactableRange(session, priced, 1)).toBeNull()
  708. })
  709. })
  710. describe('optional model-free tool-result pruning', () => {
  711. const pruneConfig = { thresholdChars: 100, headChars: 20, tailChars: 10 }
  712. it('does not prune a below-pressure session opportunistically', async () => {
  713. const ctx = createContext(10_000)
  714. const prune = new ToolResultPruner(ctx, pruneConfig)
  715. const compact = new TestCompactionEngine(ctx, {
  716. auto: false,
  717. thresholdRatio: 0.8,
  718. retainTokens: 100,
  719. })
  720. const session = oversizedToolResult()
  721. const pruneSession = vi.spyOn(prune, 'pruneSession')
  722. expect(await compactIfNeeded(compact, session)).toBeNull()
  723. expect(pruneSession).not.toHaveBeenCalled()
  724. expect(compact.calls).toHaveLength(0)
  725. expect(session.surface.replaceGeneration).toBe(0)
  726. })
  727. it('skips LLM summarization when pruning alone clears pressure', async () => {
  728. const ctx = createContext(1_000)
  729. void new ToolResultPruner(ctx, pruneConfig)
  730. const compact = new TestCompactionEngine(ctx, {
  731. auto: false,
  732. thresholdRatio: 0.5,
  733. retainTokens: 50,
  734. })
  735. const session = oversizedToolResult()
  736. expect(ctx.tokenMeter.measure(session).totalTokens).toBeGreaterThanOrEqual(500)
  737. expect(await compactIfNeeded(compact, session)).toBeNull()
  738. expect(ctx.tokenMeter.measure(session).totalTokens).toBeLessThan(500)
  739. expect(compact.calls).toHaveLength(0)
  740. expect(session.surface.replaceGeneration).toBe(1)
  741. })
  742. it('summarizes the pruned surface when pruning is insufficient', async () => {
  743. const ctx = createContext(2_000)
  744. void new ToolResultPruner(ctx, pruneConfig)
  745. const compact = new TestCompactionEngine(ctx, {
  746. auto: false,
  747. thresholdRatio: 0.5,
  748. retainTokens: 50,
  749. })
  750. const session = toolConversation()
  751. expect(await compactIfNeeded(compact, session)).not.toBeNull()
  752. expect(compact.calls).toHaveLength(1)
  753. expect(summarizedText(compact.calls[0]!.input)).toContain('tool result middle pruned')
  754. expect(summarizedText(compact.calls[0]!.input)).not.toContain('result 1 '.repeat(300))
  755. })
  756. it('retains the original compaction-basic behavior without the optional plugin', async () => {
  757. const ctx = createContext(2_000)
  758. const compact = new TestCompactionEngine(ctx, {
  759. auto: false,
  760. thresholdRatio: 0.5,
  761. retainTokens: 50,
  762. })
  763. const session = oversizedToolResult(3_000, true)
  764. expect(await compactIfNeeded(compact, session)).not.toBeNull()
  765. expect(compact.calls).toHaveLength(1)
  766. const original = session.events.find(event => event.type === 'tool/result')
  767. expect(original?.type === 'tool/result' && original.data.message.content[0].content[0])
  768. .toEqual({ type: 'text', text: 'X'.repeat(3_000) })
  769. expect(session.events.filter(event =>
  770. event.type === 'tool/result' && event.surfaceOp !== 'append')).toHaveLength(0)
  771. })
  772. })
  773. describe('compaction region transaction', () => {
  774. it('lands a framed, replayable checkpoint with exact source seqs and token price', async () => {
  775. const compact = service()
  776. compact.rawOutput = [
  777. { type: 'reasoning', text: 'private compact thought' },
  778. ...compact.summary,
  779. ]
  780. compact.usage = { inputTokens: 40, outputTokens: 5 }
  781. const session = conversation(3)
  782. const before = [...session.surface.nodes]
  783. const result = await compact.compactRegion(
  784. before[0]!,
  785. before[3]!,
  786. agent(session, MODEL),
  787. SIGNAL,
  788. )
  789. expect(result.shadowedSeqs).toEqual(before.slice(0, 4))
  790. expect(result.shadowedTokenCount).toBeGreaterThan(0)
  791. expect(compact.calls[0]).toMatchObject({ signal: SIGNAL })
  792. expect(summarizedText(compact.calls[0]!.input)).toContain('fixture user 1')
  793. const summary = session.events.findLast(event => event.type === 'compaction/summary')
  794. expect(summary?.data).toMatchObject({
  795. shadowedSeqs: result.shadowedSeqs,
  796. shadowedTokenCount: result.shadowedTokenCount,
  797. provider: 'summary-provider',
  798. model: 'summary-model',
  799. maxTokens: 123,
  800. rawOutput: compact.rawOutput,
  801. usage: compact.usage,
  802. })
  803. expect(summary?.data).not.toHaveProperty('llmStreamCall')
  804. const head = session.deriveMessages()[0]!
  805. expect(head.content[0]?.type).toBe('text')
  806. expect(head.content[0]?.type === 'text' ? head.content[0].text : '').toContain('<compacted-summary>')
  807. expect(head.content.at(-1)).toEqual({ type: 'text', text: '</compacted-summary>' })
  808. const replay = Session.create(SessionId('replay'), [...session.events])
  809. expect(replay.deriveMessages()).toEqual(session.deriveMessages())
  810. })
  811. it('replays the latest routed header so the summarizer reuses the cache', async () => {
  812. const compact = service()
  813. const session = conversation(3)
  814. const tools = [{ name: 'do_thing', description: 'd', parameters: { type: 'object' } }]
  815. session.append('request/header', {
  816. header: { config: { provider: MODEL, model: MODEL }, system: 'CONVERSATION SYSTEM', tools },
  817. reason: 'resume',
  818. })
  819. const nodes = session.surface.nodes
  820. await compact.compactRegion(nodes[0]!, nodes[1]!, agent(session, MODEL), SIGNAL)
  821. const { input } = compact.calls[0]!
  822. expect(input.system).toBe('CONVERSATION SYSTEM')
  823. expect(input.tools).toEqual(tools)
  824. expect(summarizedText(input)).toContain('fixture user 1')
  825. })
  826. it.each([
  827. ['start missing', 9_001, undefined, /start seq 9001 not found/],
  828. ['end missing', undefined, 9_002, /end seq 9002 not found/],
  829. ])('rejects %s', async (_label, startOverride, endOverride, pattern) => {
  830. const compact = service()
  831. const session = conversation(2)
  832. const nodes = session.surface.nodes
  833. await expect(compact.compactRegion(
  834. startOverride ?? nodes[0]!,
  835. endOverride ?? nodes[1]!,
  836. agent(session, MODEL),
  837. )).rejects.toThrow(pattern)
  838. })
  839. it('rejects reversed and tool-unbalanced positional boundaries', async () => {
  840. const compact = service()
  841. const plain = conversation(2)
  842. const nodes = plain.surface.nodes
  843. await expect(compact.compactRegion(
  844. nodes[2]!,
  845. nodes[1]!,
  846. agent(plain, MODEL),
  847. )).rejects.toThrow(/is after end/)
  848. const tools = toolConversation()
  849. const toolNodes = tools.surface.nodes
  850. await expect(compact.compactRegion(
  851. toolNodes[2]!,
  852. toolNodes[4]!,
  853. agent(tools, MODEL),
  854. )).rejects.toThrow(/start seq .* not a balanced boundary/)
  855. await expect(compact.compactRegion(
  856. toolNodes[0]!,
  857. toolNodes[1]!,
  858. agent(tools, MODEL),
  859. )).rejects.toThrow(/end seq .* not a balanced boundary/)
  860. })
  861. it('requires an open turn and an idle compaction bracket', async () => {
  862. const compact = service()
  863. const closed = conversation(1)
  864. closed.append('turn/end', { turn: 2, reason: { kind: 'completed' } })
  865. const nodes = closed.surface.nodes
  866. await expect(compact.compactRegion(
  867. nodes[0]!,
  868. nodes[1]!,
  869. agent(closed, MODEL),
  870. )).rejects.toThrow(/no open turn/)
  871. const locked = conversation(1)
  872. locked.append('compaction/start', {
  873. compactionId: CompactionId('locked-compaction'),
  874. turn: 2,
  875. })
  876. const lockedNodes = locked.surface.nodes
  877. await expect(compact.compactRegion(
  878. lockedNodes[0]!,
  879. lockedNodes[1]!,
  880. agent(locked, MODEL),
  881. )).rejects.toThrow(/already in progress/)
  882. })
  883. it('rejects a session with no turn boundary at all', async () => {
  884. const compact = service()
  885. const session = Session.create(SessionId('turnless'))
  886. session.append('user/message', createUserMessage({
  887. content: [{ type: 'text', text: 'orphan' }],
  888. source: { kind: 'user' },
  889. }), { surfaceOp: 'append' })
  890. const node = session.surface.nodes[0]!
  891. await expect(compact.compactRegion(
  892. node,
  893. node,
  894. agent(session, MODEL),
  895. )).rejects.toThrow(/no open turn/)
  896. })
  897. it('rejects a meter snapshot that changed before summarization began', async () => {
  898. const ctx = createContext()
  899. const meter = ctx.tokenMeter
  900. const original = meter.measure.bind(meter)
  901. vi.spyOn(meter, 'measure').mockImplementationOnce((session) => {
  902. const measurement = original(session)
  903. return { ...measurement, nodes: measurement.nodes.slice(1) }
  904. })
  905. const compact = service({ auto: false }, ctx)
  906. const session = conversation(2)
  907. const nodes = session.surface.nodes
  908. await expect(compact.compactRegion(
  909. nodes[0]!,
  910. nodes[2]!,
  911. agent(session, MODEL),
  912. )).rejects.toThrow(/selected surface changed/)
  913. })
  914. it('records summarizer failures without mutating the surface', async () => {
  915. const compact = service()
  916. compact.error = new Error('summary unavailable')
  917. const session = conversation(2)
  918. const before = session.surface.nodes
  919. await expect(compact.compactRegion(
  920. before[0]!,
  921. before[2]!,
  922. agent(session, MODEL),
  923. )).rejects.toThrow('summary unavailable')
  924. expect(session.surface.nodes).toEqual(before)
  925. expect(session.events.findLast(event => event.type === 'compaction/end')?.data)
  926. .toMatchObject({ error: 'summary unavailable' })
  927. })
  928. it('stringifies non-Error failures in the durable end bracket', async () => {
  929. const compact = service()
  930. compact.error = 'plain failure'
  931. const session = conversation(2)
  932. const nodes = session.surface.nodes
  933. await expect(compact.compactRegion(
  934. nodes[0]!,
  935. nodes[2]!,
  936. agent(session, MODEL),
  937. )).rejects.toBe('plain failure')
  938. expect(session.events.findLast(event => event.type === 'compaction/end')?.data)
  939. .toMatchObject({ error: 'plain failure' })
  940. })
  941. it('tolerates concurrent log-only appends while the selected surface is stable', async () => {
  942. const compact = service()
  943. const session = conversation(2)
  944. compact.mutateDuringSummary = () => {
  945. session.append('request/header', {
  946. header: { config: { provider: MODEL, model: MODEL } },
  947. reason: 'change',
  948. })
  949. }
  950. const nodes = session.surface.nodes
  951. await expect(compact.compactRegion(
  952. nodes[0]!,
  953. nodes[2]!,
  954. agent(session, MODEL),
  955. )).resolves.toMatchObject({ shadowedSeqs: nodes.slice(0, 3) })
  956. expect(session.events.some(event => event.type === 'compaction/summary')).toBe(true)
  957. })
  958. it('rejects concurrent surface appends before committing the replacement', async () => {
  959. const compact = service()
  960. const session = conversation(2)
  961. compact.mutateDuringSummary = () => {
  962. session.append('user/message', createUserMessage({
  963. content: [{ type: 'text', text: 'concurrent surface mutation' }],
  964. source: { kind: 'plugin', plugin: 'test' },
  965. }), { surfaceOp: 'append' })
  966. }
  967. const nodes = session.surface.nodes
  968. await expect(compact.compactRegion(
  969. nodes[0]!,
  970. nodes[2]!,
  971. agent(session, MODEL),
  972. )).rejects.toThrow(/session surface changed/)
  973. expect(session.events.some(event => event.type === 'compaction/summary')).toBe(false)
  974. })
  975. it('rejects a non-shrinking framed summary under the conversation meter', async () => {
  976. const compact = service()
  977. compact.summary = Array.from({ length: 100 }, (_, index) => ({
  978. type: 'text',
  979. text: `verbose ${index}`,
  980. }))
  981. const session = conversation(2)
  982. const nodes = session.surface.nodes
  983. await expect(compact.compactRegion(
  984. nodes[0]!,
  985. nodes[2]!,
  986. agent(session, MODEL),
  987. )).rejects.toThrow(/summary is not smaller/)
  988. expect(session.events.some(event => event.type === 'compaction/summary')).toBe(false)
  989. })
  990. it('lets a model-independent custom summarizer compact without a conversation model', async () => {
  991. const compact = service()
  992. const session = Session.create(SessionId('model-less-region'))
  993. session.append('turn/start', { turn: 1 })
  994. session.append('user/message', createUserMessage({
  995. content: [{ type: 'text', text: 'history '.repeat(100) }],
  996. source: { kind: 'user' },
  997. }), { surfaceOp: 'append' })
  998. session.append('step/start', { turn: 1, step: 1 })
  999. session.append('assistant/message', {
  1000. turn: 1,
  1001. step: 1,
  1002. message: createMessage({
  1003. role: 'assistant',
  1004. content: [{ type: 'text', text: 'answer '.repeat(100) }],
  1005. source: {
  1006. kind: 'model',
  1007. ...{ provider: 'historical', model: 'historical' },
  1008. },
  1009. }),
  1010. }, { surfaceOp: 'append' })
  1011. session.append('step/end', { turn: 1, step: 1 })
  1012. const nodes = session.surface.nodes
  1013. await expect(compact.compactRegion(
  1014. nodes[0]!,
  1015. nodes[1]!,
  1016. agent(session),
  1017. )).resolves.toMatchObject({ shadowedSeqs: [nodes[0]!, nodes[1]!] })
  1018. })
  1019. })
  1020. class ScriptedAdapter extends LlmAdapter {
  1021. lastOptions: GenerateOptions | undefined
  1022. usage: TokenUsage | undefined
  1023. constructor(
  1024. private readonly blocks: readonly ContentBlock[],
  1025. private readonly finish: (StreamChunk & { type: 'finish' })['reason'] = { kind: 'stop' },
  1026. ) {
  1027. super()
  1028. }
  1029. override async * stream(options: GenerateOptions): AsyncIterable<StreamChunk> {
  1030. this.lastOptions = options
  1031. for (const [index, block] of this.blocks.entries()) {
  1032. yield { type: 'block-start', index, blockType: block.type }
  1033. if (block.type === 'text') {
  1034. yield { type: 'text-delta', index, text: block.text }
  1035. } else if (block.type === 'reasoning') {
  1036. yield { type: 'reasoning-delta', index, text: block.text }
  1037. } else {
  1038. yield { type: 'block-end', index, block }
  1039. }
  1040. }
  1041. if (this.usage !== undefined) yield { type: 'usage', usage: this.usage }
  1042. yield { type: 'finish', reason: this.finish }
  1043. }
  1044. }
  1045. class ExposedCompactionEngine extends BasicCompactionEngine {
  1046. runSummarize(
  1047. input: SummarizationInput,
  1048. owner: Agent,
  1049. signal?: AbortSignal,
  1050. ): Promise<{
  1051. summary: ContentBlock[]
  1052. rawOutput?: ContentBlock[]
  1053. provider: string
  1054. model: string
  1055. maxTokens?: number
  1056. usage?: TokenUsage
  1057. }> {
  1058. return this.summarize(input, owner, signal)
  1059. }
  1060. }
  1061. async function summarizerHarness(
  1062. blocks: readonly ContentBlock[],
  1063. finish?: (StreamChunk & { type: 'finish' })['reason'],
  1064. model = MODEL,
  1065. config: BasicCompactionConfig = { auto: false },
  1066. ): Promise<{ ctx: Context; adapter: ScriptedAdapter; compact: ExposedCompactionEngine }> {
  1067. const ctx = new Context()
  1068. await ctx.plugin(LlmRuntime)
  1069. void new TokenMeter(ctx)
  1070. const adapter = new ScriptedAdapter(blocks, finish)
  1071. ctx.llm.registerAdapter([model], adapter)
  1072. const compact = new ExposedCompactionEngine(ctx, config)
  1073. return { ctx, adapter, compact }
  1074. }
  1075. describe('default one-shot summarizer', () => {
  1076. it('requires complete raw output when a subclass marks one local LLM stream call', () => {
  1077. expectTypeOf<{
  1078. summary: ContentBlock[]
  1079. llmStreamCall: true
  1080. provider: string
  1081. model: string
  1082. }>().not.toExtend<SummaryResult>()
  1083. })
  1084. it('uses configured model/default cap, forwards cancellation, and keeps only safe text', async () => {
  1085. const { adapter, compact } = await summarizerHarness([
  1086. { type: 'reasoning', text: 'private' },
  1087. { type: 'text', text: 'public summary' },
  1088. { type: 'tool-call', id: ToolCallId('unexpected'), name: 'x', arguments: '{}' },
  1089. ], undefined, MODEL, {
  1090. auto: false,
  1091. summarizationProvider: MODEL,
  1092. summarizationModel: MODEL,
  1093. maxTokens: 321,
  1094. })
  1095. const session = conversation(1)
  1096. adapter.usage = { inputTokens: 12, outputTokens: 3 }
  1097. const output = await compact.runSummarize(promptInput('transcript'), agent(session, 'fallback'), SIGNAL)
  1098. expect(output).toEqual({
  1099. summary: [{ type: 'text', text: 'public summary' }],
  1100. rawOutput: [
  1101. { type: 'reasoning', text: 'private' },
  1102. { type: 'text', text: 'public summary' },
  1103. { type: 'tool-call', id: ToolCallId('unexpected'), name: 'x', arguments: '{}' },
  1104. ],
  1105. llmStreamCall: true,
  1106. provider: MODEL,
  1107. model: MODEL,
  1108. maxTokens: 321,
  1109. usage: adapter.usage,
  1110. })
  1111. expect(adapter.lastOptions).toMatchObject({
  1112. provider: MODEL,
  1113. model: MODEL,
  1114. maxTokens: 321,
  1115. signal: SIGNAL,
  1116. sessionId: session.id,
  1117. purpose: 'compaction',
  1118. })
  1119. const instruction = adapter.lastOptions?.messages.at(-1)?.content[0]
  1120. expect(instruction?.type === 'text' ? instruction.text : '').toContain('## Primary Request and Intent')
  1121. })
  1122. it('replays the conversation prefix and appends the instruction as the final message', async () => {
  1123. const { adapter, compact } = await summarizerHarness([{ type: 'text', text: 'summary' }])
  1124. const tools = [{ name: 'do_thing', description: 'd', parameters: { type: 'object' } }]
  1125. const prefix: Message = createUserMessage({
  1126. content: [
  1127. { type: 'text', text: 'earlier turn' },
  1128. {
  1129. type: 'image',
  1130. attachment: {
  1131. attachmentId: AttachmentId(`sha256:${'a'.repeat(64)}`),
  1132. mediaType: 'image/png',
  1133. bytes: 1,
  1134. width: 1,
  1135. height: 1,
  1136. },
  1137. },
  1138. ],
  1139. source: { kind: 'plugin', plugin: 'test' },
  1140. })
  1141. await compact.runSummarize({
  1142. system: 'REPLAYED SYSTEM',
  1143. tools,
  1144. messages: [prefix],
  1145. }, agent(conversation(1), MODEL))
  1146. expect(adapter.lastOptions?.system).toBe('REPLAYED SYSTEM')
  1147. expect(adapter.lastOptions?.tools).toEqual(tools)
  1148. const messages = adapter.lastOptions?.messages ?? []
  1149. expect(messages[0]).toEqual(prefix)
  1150. const last = messages.at(-1)?.content[0]
  1151. const lastText = last?.type === 'text' ? last.text : ''
  1152. expect(lastText).toContain('Write concise English engineering prose.')
  1153. expect(lastText).toContain('numeric values, function signatures, and syntax fragments.')
  1154. expect(lastText).toContain('## Primary Request and Intent')
  1155. })
  1156. it('applies the routed model policy without changing the replayed prefix', async () => {
  1157. const { ctx, compact } = await summarizerHarness(
  1158. [{ type: 'text', text: 'unused default summary' }],
  1159. undefined,
  1160. MODEL,
  1161. {
  1162. auto: false,
  1163. maxTokens: 111,
  1164. modelPolicies: [{
  1165. provider: MODEL,
  1166. model: MODEL,
  1167. summarizationProvider: 'policy-summary',
  1168. summarizationModel: 'policy-summary',
  1169. maxTokens: 222,
  1170. }],
  1171. },
  1172. )
  1173. const policyAdapter = new ScriptedAdapter([{ type: 'text', text: 'policy summary' }])
  1174. ctx.llm.registerAdapter(['policy-summary'], policyAdapter)
  1175. const prefix: Message = createUserMessage({
  1176. content: [{ type: 'text', text: 'warm prefix' }],
  1177. source: { kind: 'plugin', plugin: 'test' },
  1178. })
  1179. const output = await compact.runSummarize({
  1180. system: 'WARM SYSTEM',
  1181. messages: [prefix],
  1182. }, agent(conversation(1), 'fallback'))
  1183. expect(output).toMatchObject({
  1184. provider: 'policy-summary',
  1185. model: 'policy-summary',
  1186. maxTokens: 222,
  1187. })
  1188. expect(policyAdapter.lastOptions).toMatchObject({
  1189. provider: 'policy-summary',
  1190. model: 'policy-summary',
  1191. maxTokens: 222,
  1192. system: 'WARM SYSTEM',
  1193. })
  1194. expect(policyAdapter.lastOptions?.messages[0]).toEqual(prefix)
  1195. })
  1196. it('resolves the latest routed provider/model before the AgentOptions pair', async () => {
  1197. const { adapter, compact } = await summarizerHarness([{ type: 'text', text: 'summary' }], undefined, 'routed')
  1198. const session = conversation(1)
  1199. session.append('request/header', {
  1200. header: { config: { provider: 'routed', model: 'routed' } },
  1201. reason: 'initial',
  1202. })
  1203. const output = await compact.runSummarize(promptInput('history'), agent(session, 'fallback'))
  1204. expect(output.provider).toBe('routed')
  1205. expect(output.model).toBe('routed')
  1206. expect(adapter.lastOptions?.provider).toBe('routed')
  1207. expect(adapter.lastOptions?.model).toBe('routed')
  1208. })
  1209. it('records the model actually dispatched after one-shot stream routing', async () => {
  1210. const { ctx, compact } = await summarizerHarness([{ type: 'text', text: 'unused' }])
  1211. const routedAdapter = new ScriptedAdapter([{ type: 'text', text: 'routed summary' }])
  1212. ctx.llm.registerAdapter(['routed-summary-provider'], routedAdapter)
  1213. ctx.on('llm/stream', (options, next) => {
  1214. options.provider = 'routed-summary-provider'
  1215. options.model = 'routed-summary-model'
  1216. return next()
  1217. })
  1218. const session = conversation(3, 'large history '.repeat(500))
  1219. const nodes = session.surface.nodes
  1220. await compact.compactRegion(nodes[0]!, nodes[3]!, agent(session, MODEL), SIGNAL)
  1221. expect(session.events.findLast(event => event.type === 'compaction/summary')?.data).toMatchObject({
  1222. summary: [{ type: 'text', text: 'routed summary' }],
  1223. llmStreamCall: true,
  1224. provider: 'routed-summary-provider',
  1225. model: 'routed-summary-model',
  1226. })
  1227. expect(routedAdapter.lastOptions?.provider).toBe('routed-summary-provider')
  1228. expect(routedAdapter.lastOptions?.model).toBe('routed-summary-model')
  1229. })
  1230. it('fails clearly when no complete summarization target can be resolved', async () => {
  1231. const ctx = new Context()
  1232. await ctx.plugin(LlmRuntime)
  1233. void new TokenMeter(ctx)
  1234. const compact = new ExposedCompactionEngine(ctx, { auto: false })
  1235. await expect(compact.runSummarize(promptInput('history'), agent(Session.create(SessionId('model-less')))))
  1236. .rejects.toThrow(/no provider\/model available for summarization/)
  1237. })
  1238. it('uses a complete AgentOptions target when no durable route exists', async () => {
  1239. const { adapter, compact } = await summarizerHarness([{ type: 'text', text: 'summary' }])
  1240. const session = Session.create(SessionId('headerless-summary'))
  1241. await expect(compact.runSummarize(promptInput('history'), agent(session, MODEL))).resolves.toMatchObject({
  1242. provider: MODEL,
  1243. model: MODEL,
  1244. })
  1245. expect(adapter.lastOptions).toMatchObject({ provider: MODEL, model: MODEL })
  1246. })
  1247. it.each([
  1248. { provider: '', model: MODEL },
  1249. { provider: MODEL },
  1250. { provider: MODEL, model: '' },
  1251. ])('rejects incomplete AgentOptions target %#', async (options) => {
  1252. const { compact } = await summarizerHarness([{ type: 'text', text: 'unused' }])
  1253. const owner = {
  1254. session: Session.create(SessionId(`incomplete-${String(options.model)}`)),
  1255. options,
  1256. } as Agent
  1257. await expect(compact.runSummarize(promptInput('history'), owner))
  1258. .rejects.toThrow(/no provider\/model available for summarization/)
  1259. })
  1260. it.each([
  1261. [{ kind: 'error', failure: { message: 'provider failed', code: 'PROVIDER' } }, 'PROVIDER', /provider failed/],
  1262. [{ kind: 'error', failure: { message: 'opaque', code: 'UNKNOWN' } }, 'UNKNOWN', /opaque/],
  1263. [{ kind: 'aborted', failure: { message: 'summarization aborted', code: 'ABORTED' } }, 'ABORTED', /aborted/],
  1264. [{ kind: 'max-tokens' }, 'MAX_TOKENS', /token cap/],
  1265. ] as Array<[(StreamChunk & { type: 'finish' })['reason'], string | undefined, RegExp]>) (
  1266. 'rejects terminal finish %#',
  1267. async (finish, code, pattern) => {
  1268. const { compact } = await summarizerHarness([], finish)
  1269. let thrown: unknown
  1270. try {
  1271. await compact.runSummarize(promptInput('history'), agent(conversation(1), MODEL))
  1272. } catch (error: unknown) {
  1273. thrown = error
  1274. }
  1275. expect(thrown).toBeInstanceOf(Error)
  1276. expect((thrown as Error).message).toMatch(pattern)
  1277. expect((thrown as Error & { code?: string }).code).toBe(code)
  1278. },
  1279. )
  1280. it('rejects empty or reasoning-only successful output', async () => {
  1281. const { compact } = await summarizerHarness([{ type: 'reasoning', text: 'private' }])
  1282. await expect(compact.runSummarize(promptInput('history'), agent(conversation(1), MODEL)))
  1283. .rejects.toThrow(/no text summary content/)
  1284. })
  1285. it('rejects image summary output instead of silently dropping it', async () => {
  1286. const { compact } = await summarizerHarness([
  1287. {
  1288. type: 'image',
  1289. attachment: {
  1290. attachmentId: AttachmentId(`sha256:${'b'.repeat(64)}`),
  1291. mediaType: 'image/png',
  1292. bytes: 1,
  1293. width: 1,
  1294. height: 1,
  1295. },
  1296. },
  1297. { type: 'text', text: 'partial summary' },
  1298. ])
  1299. await expect(compact.runSummarize(promptInput('history'), agent(conversation(1), MODEL)))
  1300. .rejects.toMatchObject({ code: 'UNSUPPORTED_CONTENT' })
  1301. })
  1302. it('rejects image summary output nested in a tool result', async () => {
  1303. const { compact } = await summarizerHarness([{
  1304. type: 'tool-result',
  1305. toolCallId: ToolCallId('summary-tool'),
  1306. content: [{
  1307. type: 'image',
  1308. attachment: {
  1309. attachmentId: AttachmentId(`sha256:${'c'.repeat(64)}`),
  1310. mediaType: 'image/png',
  1311. bytes: 1,
  1312. width: 1,
  1313. height: 1,
  1314. },
  1315. }],
  1316. }])
  1317. await expect(compact.runSummarize(promptInput('history'), agent(conversation(1), MODEL)))
  1318. .rejects.toMatchObject({ code: 'UNSUPPORTED_CONTENT' })
  1319. })
  1320. })
  1321. describe('automatic listener and loader composition', () => {
  1322. function preStep(ctx: Context, owner: Agent, signal = SIGNAL) {
  1323. return agentEvents(ctx, owner).waterfall(
  1324. 'agent/pre-step', { messages: [], turn: 1, step: 1, signal },
  1325. () => Promise.resolve({ kind: 'enter' as const, messages: [] }),
  1326. )
  1327. }
  1328. function recover(
  1329. ctx: Context,
  1330. owner: Agent,
  1331. error: Error & { code?: string },
  1332. signal = SIGNAL,
  1333. next: () => Promise<RequestErrorAction> = () => Promise.resolve(undefined),
  1334. ): Promise<boolean> {
  1335. const failure: LlmFailure = { message: error.message, code: error.code ?? 'UNKNOWN' }
  1336. const turn = owner.session.events.findLast(event => event.type === 'turn/start')?.data.turn ?? 1
  1337. return agentEvents(ctx, owner).waterfall(
  1338. 'agent/request-error',
  1339. { turn, step: 1, provider: 'test', failure, retryPolicy: undefined, signal },
  1340. next,
  1341. ).then(action => action?.kind === 'retry')
  1342. }
  1343. function overflow(message = 'provider overflow'): Error & { code: string } {
  1344. return Object.assign(new Error(message), { code: CONTEXT_WINDOW_EXCEEDED_CODE })
  1345. }
  1346. it('compacts before a step above threshold using the durable routed model and remains idle below it', async () => {
  1347. const ctx = createContext()
  1348. const compact = new TestCompactionEngine(ctx, {
  1349. thresholdRatio: 0.5,
  1350. retainTokens: 180,
  1351. })
  1352. const pressured = conversation(4)
  1353. await preStep(ctx, agent(pressured, 'unconfigured-agent-fallback'))
  1354. expect(pressured.events.some(event => event.type === 'compaction/summary')).toBe(true)
  1355. const small = conversation(1)
  1356. await preStep(ctx, agent(small, MODEL))
  1357. expect(small.events.some(event => event.type === 'compaction/start')).toBe(false)
  1358. expect(compact.calls).toHaveLength(1)
  1359. })
  1360. it('skips pre-step pressure when the step signal is already aborted', async () => {
  1361. const ctx = createContext()
  1362. const compact = new TestCompactionEngine(ctx, {
  1363. thresholdRatio: 0.5,
  1364. retainTokens: 180,
  1365. })
  1366. const pressured = conversation(4)
  1367. const compactIfNeeded = vi.spyOn(compact, 'compactIfNeeded')
  1368. await expect(preStep(ctx, agent(pressured, MODEL), AbortSignal.abort('step aborted')))
  1369. .resolves.toEqual({ kind: 'enter', messages: [] })
  1370. expect(compactIfNeeded).not.toHaveBeenCalled()
  1371. expect(pressured.events.some(event => event.type === 'compaction/start')).toBe(false)
  1372. })
  1373. it('warns and continues after operational failures, including non-Errors', async () => {
  1374. const ctx = createContext()
  1375. const warnings: string[] = []
  1376. ctx.logger.warn = ((message: string) => void warnings.push(message)) as typeof ctx.logger.warn
  1377. const compact = new TestCompactionEngine(ctx, {
  1378. thresholdRatio: 0.5,
  1379. retainTokens: 180,
  1380. })
  1381. compact.error = 'temporary failure'
  1382. const session = conversation(4)
  1383. await expect(preStep(ctx, agent(session, MODEL))).resolves.toEqual({ kind: 'enter', messages: [] })
  1384. expect(warnings).toContainEqual(expect.stringContaining('temporary failure'))
  1385. expect(session.events.some(event => event.type === 'compaction/summary')).toBe(false)
  1386. })
  1387. it('warns once per routed target when proactive pressure has no context metadata', async () => {
  1388. const ctx = createContext()
  1389. const warnings: string[] = []
  1390. ctx.logger.warn = ((message: string) => void warnings.push(message)) as typeof ctx.logger.warn
  1391. vi.spyOn(ctx.llm, 'resolveModelInfo').mockImplementation((provider, model) => Promise.resolve({
  1392. provider,
  1393. id: model,
  1394. name: model,
  1395. }))
  1396. void new TestCompactionEngine(ctx, {
  1397. thresholdRatio: 0.5,
  1398. retainTokens: 180,
  1399. })
  1400. const session = conversation(4)
  1401. await preStep(ctx, agent(session, MODEL))
  1402. await preStep(ctx, agent(session, MODEL))
  1403. expect(warnings).toEqual([
  1404. expect.stringContaining(`no context capacity for ${MODEL}/${MODEL}`),
  1405. ])
  1406. })
  1407. it('warns once per routed target when absolute retention exceeds its resolved threshold', async () => {
  1408. const ctx = createContext()
  1409. const warnings: string[] = []
  1410. ctx.logger.warn = ((message: string) => void warnings.push(message)) as typeof ctx.logger.warn
  1411. void new TestCompactionEngine(ctx, {
  1412. thresholdRatio: 0.5,
  1413. retainTokens: 500,
  1414. })
  1415. const session = conversation(4)
  1416. await preStep(ctx, agent(session, MODEL))
  1417. await preStep(ctx, agent(session, MODEL))
  1418. expect(warnings).toEqual([
  1419. expect.stringContaining('retainTokens (500) must be less than threshold tokens 500'),
  1420. ])
  1421. })
  1422. it('force-compacts below normal pressure for canonical overflow and retries only after replacement', async () => {
  1423. const ctx = createContext(10_000)
  1424. void new TestCompactionEngine(ctx, {
  1425. thresholdRatio: 1,
  1426. retainTokens: 900,
  1427. })
  1428. const session = conversation(3)
  1429. const beforeGeneration = session.surface.replaceGeneration
  1430. const retainedSeq = session.surface.nodes.at(-1)!
  1431. const threshold = 10_000
  1432. expect(ctx.tokenMeter.measure(session).totalTokens).toBeLessThan(threshold)
  1433. const decision = await recover(ctx, agent(session, 'unconfigured-agent-fallback'), overflow())
  1434. expect(decision).toBe(true)
  1435. expect(session.surface.replaceGeneration).toBe(beforeGeneration + 1)
  1436. expect(session.events.some(event => event.type === 'compaction/summary')).toBe(true)
  1437. expect(session.surface.nodes).toContain(retainedSeq)
  1438. })
  1439. it('authorizes overflow retry when pruning alone advances an indivisible surface', async () => {
  1440. const ctx = createContext(10_000)
  1441. void new ToolResultPruner(ctx, {
  1442. thresholdChars: 100,
  1443. headChars: 20,
  1444. tailChars: 10,
  1445. })
  1446. const compact = new TestCompactionEngine(ctx, {
  1447. thresholdRatio: 1,
  1448. retainTokens: 900,
  1449. })
  1450. const session = oversizedToolResult()
  1451. expect(await recover(ctx, agent(session, MODEL), overflow())).toBe(true)
  1452. expect(session.surface.replaceGeneration).toBe(1)
  1453. expect(session.events.some(event => event.type === 'compaction/summary')).toBe(false)
  1454. expect(compact.calls).toHaveLength(0)
  1455. })
  1456. it('continues overflow recovery with summarization on the pruned surface', async () => {
  1457. const ctx = createContext(10_000)
  1458. void new ToolResultPruner(ctx, {
  1459. thresholdChars: 100,
  1460. headChars: 20,
  1461. tailChars: 10,
  1462. })
  1463. const compact = new TestCompactionEngine(ctx, {
  1464. thresholdRatio: 1,
  1465. retainTokens: 900,
  1466. })
  1467. const session = toolConversation()
  1468. expect(await recover(ctx, agent(session, MODEL), overflow())).toBe(true)
  1469. expect(session.events.some(event => event.type === 'compaction/summary')).toBe(true)
  1470. expect(compact.calls).toHaveLength(1)
  1471. expect(summarizedText(compact.calls[0]!.input)).toContain('tool result middle pruned')
  1472. })
  1473. it('retries from a durable prune when later overflow summarization throws', async () => {
  1474. const ctx = createContext(10_000)
  1475. const warnings: string[] = []
  1476. ctx.logger.warn = ((message: string) => void warnings.push(message)) as typeof ctx.logger.warn
  1477. void new ToolResultPruner(ctx, {
  1478. thresholdChars: 100,
  1479. headChars: 20,
  1480. tailChars: 10,
  1481. })
  1482. const compact = new TestCompactionEngine(ctx, {
  1483. thresholdRatio: 1,
  1484. retainTokens: 900,
  1485. })
  1486. compact.error = new Error('summary unavailable after prune')
  1487. const session = oversizedToolResult(3_000, true)
  1488. expect(await recover(ctx, agent(session, MODEL), overflow())).toBe(true)
  1489. expect(session.surface.replaceGeneration).toBe(1)
  1490. expect(session.events.filter(event => event.type === 'tool/result')).toHaveLength(2)
  1491. expect(session.events.findLast(event => event.type === 'compaction/end')?.data)
  1492. .toMatchObject({ error: 'summary unavailable after prune' })
  1493. expect(warnings).toContainEqual(expect.stringContaining('retrying from the replacement surface'))
  1494. })
  1495. it('lets cancellation win when summary throws after a durable prune', async () => {
  1496. const ctx = createContext(10_000)
  1497. const controller = new AbortController()
  1498. void new ToolResultPruner(ctx, {
  1499. thresholdChars: 100,
  1500. headChars: 20,
  1501. tailChars: 10,
  1502. })
  1503. const compact = new TestCompactionEngine(ctx, {
  1504. thresholdRatio: 1,
  1505. retainTokens: 900,
  1506. })
  1507. compact.mutateDuringSummary = () => { controller.abort('cancelled during summary') }
  1508. compact.error = new Error('summary cancelled after prune')
  1509. const session = oversizedToolResult(3_000, true)
  1510. expect(await recover(ctx, agent(session, MODEL), overflow(), controller.signal)).toBe(false)
  1511. expect(session.surface.replaceGeneration).toBe(1)
  1512. })
  1513. it('preserves the newest whole tool-call/result pair during forced overflow compaction', async () => {
  1514. const ctx = createContext()
  1515. void new TestCompactionEngine(ctx, {
  1516. thresholdRatio: 1,
  1517. retainTokens: 90,
  1518. })
  1519. const session = toolConversation()
  1520. const newestAssistant = session.surface.nodes.at(-2)!
  1521. const newestResult = session.surface.nodes.at(-1)!
  1522. expect(await recover(ctx, agent(session, MODEL), overflow())).toBe(true)
  1523. const currentAssistant = session.surface.nodes.find(node => node === newestAssistant)
  1524. const currentResult = session.surface.nodes.find(node => node === newestResult)
  1525. expect(currentAssistant).toBeDefined()
  1526. expect(currentResult).toBeDefined()
  1527. expect(toolPairingBalancedBefore(session, currentAssistant!)).toBe(true)
  1528. expect(toolPairingBalancedAfter(session, currentResult!)).toBe(true)
  1529. })
  1530. it('does not retry when a backend reports success without replacing the surface', async () => {
  1531. const ctx = createContext()
  1532. const compact = new TestCompactionEngine(ctx)
  1533. const session = conversation(2)
  1534. const fakeResult: CompactionResult = {
  1535. compactionId: CompactionId('fake-compaction'),
  1536. startSeq: 1,
  1537. summarySeq: 2,
  1538. endSeq: 3,
  1539. summary: [{ type: 'text', text: 'fake' }],
  1540. shadowedRange: { start: 1, end: 2 },
  1541. shadowedSeqs: [1, 2],
  1542. shadowedTokenCount: 10,
  1543. }
  1544. vi.spyOn(compact, 'compactIfNeeded').mockResolvedValue(fakeResult)
  1545. expect(await recover(ctx, agent(session, MODEL), overflow())).toBe(false)
  1546. expect(session.surface.replaceGeneration).toBe(0)
  1547. })
  1548. it('delegates downstream exactly once when no replacement is available', async () => {
  1549. const ctx = createContext()
  1550. const compact = new TestCompactionEngine(ctx)
  1551. vi.spyOn(compact, 'compactIfNeeded').mockResolvedValue(null)
  1552. const downstream = new Error('downstream recovery failed')
  1553. let calls = 0
  1554. await expect(recover(
  1555. ctx,
  1556. agent(conversation(2), MODEL),
  1557. overflow(),
  1558. SIGNAL,
  1559. () => {
  1560. calls += 1
  1561. return Promise.reject(downstream)
  1562. },
  1563. )).rejects.toBe(downstream)
  1564. expect(calls).toBe(1)
  1565. })
  1566. it('preserves the original provider error when recovery throws', async () => {
  1567. const ctx = createContext()
  1568. const warnings: string[] = []
  1569. ctx.logger.warn = ((message: string) => void warnings.push(message)) as typeof ctx.logger.warn
  1570. const compact = new TestCompactionEngine(ctx)
  1571. compact.error = new Error('summary unavailable')
  1572. const original = overflow('original provider overflow')
  1573. expect(await recover(ctx, agent(conversation(3), MODEL), original)).toBe(false)
  1574. expect(original).toMatchObject({
  1575. message: 'original provider overflow',
  1576. code: CONTEXT_WINDOW_EXCEEDED_CODE,
  1577. })
  1578. expect(warnings).toContainEqual(expect.stringContaining('preserving the original request error'))
  1579. })
  1580. it('delegates once when overflow recovery throws a non-Error value', async () => {
  1581. const ctx = createContext()
  1582. const warnings: string[] = []
  1583. ctx.logger.warn = ((message: string) => void warnings.push(message)) as typeof ctx.logger.warn
  1584. const compact = new TestCompactionEngine(ctx)
  1585. compact.error = 'non-error recovery failure'
  1586. const session = conversation(3)
  1587. const generation = session.surface.replaceGeneration
  1588. const original = overflow('original provider failure')
  1589. let delegations = 0
  1590. const decision = await recover(ctx, agent(session, MODEL), original, SIGNAL, () => {
  1591. delegations += 1
  1592. return Promise.resolve(undefined)
  1593. })
  1594. expect(decision).toBe(false)
  1595. expect(delegations).toBe(1)
  1596. expect(session.surface.replaceGeneration).toBe(generation)
  1597. expect(original).toMatchObject({
  1598. message: 'original provider failure',
  1599. code: CONTEXT_WINDOW_EXCEEDED_CODE,
  1600. })
  1601. expect(warnings).toContainEqual(expect.stringContaining('non-error recovery failure'))
  1602. })
  1603. it('recovers an overflow for an unlisted routed model', async () => {
  1604. const ctx = createContext()
  1605. void new TestCompactionEngine(ctx)
  1606. const session = conversation(2)
  1607. session.append('request/header', {
  1608. header: { config: { provider: 'unknown-routed-provider', model: 'unknown-routed-model' } },
  1609. reason: 'resume',
  1610. })
  1611. expect(await recover(ctx, agent(session, MODEL), overflow('unlisted-model overflow')))
  1612. .toBe(true)
  1613. })
  1614. it('delegates canonical overflow when no durable routed target exists', async () => {
  1615. const ctx = createContext()
  1616. void new TestCompactionEngine(ctx)
  1617. const session = Session.create(SessionId('headerless-overflow'))
  1618. session.append('turn/start', {
  1619. turn: 1,
  1620. })
  1621. await expect(recover(ctx, agent(session, MODEL), overflow())).resolves.toBe(false)
  1622. })
  1623. it('honors retry caps and ignores non-context failures', async () => {
  1624. const ctx = createContext()
  1625. const compact = new TestCompactionEngine(ctx, { maxOverflowRetries: 1 })
  1626. const compactSpy = vi.spyOn(compact, 'compactIfNeeded')
  1627. const owner = agent(conversation(3), MODEL)
  1628. expect(await recover(ctx, owner, Object.assign(new Error('rate limit'), { code: 'RATE_LIMIT' })))
  1629. .toBe(false)
  1630. expect(await recover(ctx, owner, overflow())).toBe(true)
  1631. compactSpy.mockClear()
  1632. expect(await recover(ctx, owner, overflow())).toBe(false)
  1633. expect(compactSpy).not.toHaveBeenCalled()
  1634. })
  1635. it('applies the routed model override to the overflow retry cap', async () => {
  1636. const ctx = createContext()
  1637. const compact = new TestCompactionEngine(ctx, {
  1638. maxOverflowRetries: 2,
  1639. modelPolicies: [{
  1640. provider: MODEL,
  1641. model: MODEL,
  1642. maxOverflowRetries: 1,
  1643. }],
  1644. })
  1645. const compactSpy = vi.spyOn(compact, 'compactIfNeeded')
  1646. const owner = agent(conversation(3), MODEL)
  1647. expect(await recover(ctx, owner, overflow())).toBe(true)
  1648. compactSpy.mockClear()
  1649. expect(await recover(ctx, owner, overflow())).toBe(false)
  1650. expect(compactSpy).not.toHaveBeenCalled()
  1651. })
  1652. it('does not retry when cancellation lands during an awaited compaction', async () => {
  1653. const ctx = createContext()
  1654. const compact = new TestCompactionEngine(ctx)
  1655. const controller = new AbortController()
  1656. compact.mutateDuringSummary = () => { controller.abort('cancelled during summary') }
  1657. const session = conversation(3)
  1658. const generation = session.surface.replaceGeneration
  1659. expect(await recover(ctx, agent(session, MODEL), overflow(), controller.signal)).toBe(false)
  1660. expect(session.surface.replaceGeneration).toBe(generation + 1)
  1661. })
  1662. it('maxOverflowRetries:0 disables recovery without disabling post-step pressure', async () => {
  1663. const ctx = createContext()
  1664. void new TestCompactionEngine(ctx, {
  1665. maxOverflowRetries: 0,
  1666. thresholdRatio: 0.5,
  1667. retainTokens: 180,
  1668. })
  1669. const session = conversation(4)
  1670. await preStep(ctx, agent(session, MODEL))
  1671. const summaries = session.events.filter(event => event.type === 'compaction/summary').length
  1672. expect(summaries).toBe(1)
  1673. expect(await recover(ctx, agent(session, MODEL), overflow())).toBe(false)
  1674. expect(session.events.filter(event => event.type === 'compaction/summary')).toHaveLength(summaries)
  1675. })
  1676. it('auto:false installs neither automatic listener', async () => {
  1677. const ctx = createContext()
  1678. void new TestCompactionEngine(ctx, {
  1679. auto: false,
  1680. thresholdRatio: 0.5,
  1681. retainTokens: 180,
  1682. })
  1683. const session = conversation(4)
  1684. await preStep(ctx, agent(session, MODEL))
  1685. expect(session.events.some(event => event.type === 'compaction/start')).toBe(false)
  1686. expect(await recover(ctx, agent(session, MODEL), overflow())).toBe(false)
  1687. })
  1688. it('loads and disposes the real zero-config service stack', async () => {
  1689. const ctx = new Context()
  1690. await ctx.plugin(LlmRuntime)
  1691. await ctx.plugin(SessionStore)
  1692. const meterFiber = await ctx.plugin(TokenMeter)
  1693. const compactFiber = await ctx.plugin(BasicCompactionEngine, { auto: false })
  1694. expect(ctx.get('compaction')).toBeInstanceOf(BasicCompactionEngine)
  1695. await compactFiber.dispose()
  1696. expect(ctx.get('compaction')).toBeUndefined()
  1697. await meterFiber.dispose()
  1698. expect(ctx.get('tokenMeter')).toBeUndefined()
  1699. })
  1700. it('removes its automatic listener with the plugin fiber', async () => {
  1701. const ctx = new Context()
  1702. await ctx.plugin(LlmRuntime)
  1703. await ctx.plugin(TokenMeter)
  1704. const fiber = await ctx.plugin(TestCompactionEngine, {
  1705. thresholdRatio: 0.5,
  1706. retainTokens: 180,
  1707. })
  1708. await fiber.dispose()
  1709. const session = conversation(4)
  1710. await preStep(ctx, agent(session, MODEL))
  1711. expect(session.events.some(event => event.type === 'compaction/start')).toBe(false)
  1712. expect(await recover(ctx, agent(session, MODEL), overflow())).toBe(false)
  1713. })
  1714. })
  1715. describe('route-priced image pressure', () => {
  1716. const IMAGE_VISUAL_TOKENS = 300
  1717. const IMAGE_HANDLE_TEXT = 'request preview'
  1718. class PricedContextAdapter extends ContextAdapter {
  1719. override imageRequestPricing(): { priceImages: (images: readonly unknown[]) => Array<{ visualTokens: number; text: string }> } {
  1720. return {
  1721. priceImages: images => images.map(() => ({
  1722. visualTokens: IMAGE_VISUAL_TOKENS,
  1723. text: IMAGE_HANDLE_TEXT,
  1724. })),
  1725. }
  1726. }
  1727. }
  1728. function pricedContext(contextWindow = 1_000): Context {
  1729. const ctx = new Context()
  1730. void new LlmRuntime(ctx)
  1731. void new TokenMeter(ctx)
  1732. ctx.llm.registerAdapter([MODEL], new PricedContextAdapter(contextWindow))
  1733. return ctx
  1734. }
  1735. /** Closed short-text turns whose user messages each carry one image. */
  1736. function imageConversation(turns = 4): Session {
  1737. const session = Session.create(SessionId(`image-dense-${turns}`))
  1738. for (let turn = 1; turn <= turns; turn += 1) {
  1739. session.append('turn/start', { turn })
  1740. session.append('user/message', createUserMessage({
  1741. content: [
  1742. { type: 'text', text: `image turn ${turn}` },
  1743. {
  1744. type: 'image',
  1745. attachment: {
  1746. attachmentId: AttachmentId(`sha256:${String(turn).repeat(8)}`),
  1747. mediaType: 'image/png',
  1748. bytes: 2048,
  1749. width: 800,
  1750. height: 800,
  1751. name: `shot-${turn}`,
  1752. },
  1753. },
  1754. ],
  1755. source: { kind: 'user' },
  1756. }), { surfaceOp: 'append' })
  1757. session.append('step/start', { turn, step: 1 })
  1758. if (turn === 1) {
  1759. session.append('request/header', {
  1760. header: { config: { provider: MODEL, model: MODEL } },
  1761. reason: 'initial',
  1762. })
  1763. }
  1764. session.append('assistant/message', {
  1765. turn,
  1766. step: 1,
  1767. message: createMessage({
  1768. role: 'assistant',
  1769. content: [{ type: 'text', text: `ok ${turn}` }],
  1770. source: {
  1771. kind: 'model',
  1772. ...{ provider: MODEL, model: MODEL },
  1773. },
  1774. }),
  1775. }, { surfaceOp: 'append' })
  1776. session.append('step/end', { turn, step: 1 })
  1777. session.append('turn/end', { turn, reason: { kind: 'completed' } })
  1778. }
  1779. session.append('turn/start', { turn: turns + 1 })
  1780. return session
  1781. }
  1782. it('selects an image-dense range only when the routed price counts visual tokens', () => {
  1783. const session = imageConversation()
  1784. const routed = pricedContext().tokenMeter.measure(session)
  1785. const neutral = createContext().tokenMeter.measure(session)
  1786. expect(routed.surfaceTokens).toBeGreaterThan(neutral.surfaceTokens + 4 * IMAGE_VISUAL_TOKENS - 200)
  1787. expect(routed.nodes.map(node => node.seq)).toEqual(neutral.nodes.map(node => node.seq))
  1788. expect(routed.nodes.map(node => node.heuristicTokens)).toEqual(neutral.nodes.map(node => node.tokens))
  1789. // The same verbatim tail budget retains almost everything under the
  1790. // neutral heuristic but forces a cut once visual tokens are counted.
  1791. expect(selectCompactableRange(session, neutral, 350)).toBeNull()
  1792. const range = selectCompactableRange(session, routed, 350)
  1793. expect(range).not.toBeNull()
  1794. })
  1795. it('accepts a summary larger than the span heuristic when the route price shrinks', async () => {
  1796. // A single short image message prices below a framed summary under the
  1797. // fixed heuristic but far above it under the route: the shrink comparison
  1798. // must ask whether the replacement lowers route pressure.
  1799. const ctx = pricedContext(1_000)
  1800. const session = imageConversation(1)
  1801. const before = ctx.tokenMeter.measure(session)
  1802. const imageNode = before.nodes[0]!
  1803. const compact = new TestCompactionEngine(ctx, { auto: false })
  1804. compact.summary = [{
  1805. type: 'text',
  1806. text: 'summary text sized between the heuristic and route prices of the shadowed image message, '
  1807. + 'long enough that the fixed heuristic alone would reject it as not smaller '
  1808. + 'while the route-priced comparison accepts the pressure reduction.',
  1809. }]
  1810. const framed = ctx.tokenMeter.estimateMessage(createUserMessage({
  1811. content: frameSummary(compact.summary),
  1812. source: { kind: 'plugin', plugin: 'test' },
  1813. }))
  1814. expect(framed).toBeGreaterThan(imageNode.heuristicTokens)
  1815. expect(framed).toBeLessThan(imageNode.tokens)
  1816. const result = await compact.compactRegion(imageNode.seq, imageNode.seq, agent(session), SIGNAL)
  1817. expect(result.shadowedSeqs).toEqual([imageNode.seq])
  1818. expect(result.shadowedTokenCount).toBe(imageNode.heuristicTokens)
  1819. })
  1820. it('triggers pressure compaction from routed visual tokens and logs heuristic shadow prices', async () => {
  1821. const ctx = pricedContext(1_000)
  1822. const session = imageConversation()
  1823. const before = ctx.tokenMeter.measure(session)
  1824. const compact = new TestCompactionEngine(ctx, {
  1825. auto: false,
  1826. thresholdRatio: 0.8,
  1827. retainTokens: 350,
  1828. })
  1829. // The same history stays below the 800-token threshold without pricing.
  1830. const neutralResult = await compactIfNeeded(service({
  1831. auto: false,
  1832. thresholdRatio: 0.8,
  1833. retainTokens: 350,
  1834. }), session)
  1835. expect(neutralResult).toBeNull()
  1836. const result = await compact.compactIfNeeded(agent(session), 'pressure', SIGNAL)
  1837. expect(result).not.toBeNull()
  1838. const summaryEvent = session.events.find(event => event.type === 'compaction/summary')
  1839. expect(summaryEvent).toBeDefined()
  1840. const shadowedHeuristic = before.nodes
  1841. .filter(node => result?.shadowedSeqs.includes(node.seq))
  1842. .reduce((total, node) => total + node.heuristicTokens, 0)
  1843. expect(summaryEvent?.data.shadowedTokenCount).toBe(shadowedHeuristic)
  1844. })
  1845. })