@@ -3,6 +3,18 @@ import { describe, test, expect, beforeEach } from 'bun:test'
33import contextPruner from '../context-pruner'
44
55import type { Message , ToolMessage } from '../types/util-types'
6+ const createMessage = (
7+ role : 'user' | 'assistant' ,
8+ content : string ,
9+ ) : Message => ( {
10+ role,
11+ content : [
12+ {
13+ type : 'text' ,
14+ text : content ,
15+ } ,
16+ ] ,
17+ } )
618
719describe ( 'context-pruner handleSteps' , ( ) => {
820 let mockAgentState : any
@@ -13,14 +25,6 @@ describe('context-pruner handleSteps', () => {
1325 }
1426 } )
1527
16- const createMessage = (
17- role : 'user' | 'assistant' ,
18- content : string ,
19- ) : Message => ( {
20- role,
21- content,
22- } )
23-
2428 const createTerminalToolMessage = (
2529 command : string ,
2630 output : string ,
@@ -153,8 +157,7 @@ describe('context-pruner handleSteps', () => {
153157 m . role === 'tool' &&
154158 m . toolName === 'run_terminal_command' &&
155159 ( m . content ?. [ 0 ] ?. value ?. command === 'command-7' ||
156- m . content ?. [ 0 ] ?. value ?. message ===
157- '[LARGE_TOOL_RESULT_OMITTED]' ) ,
160+ m . content ?. [ 0 ] ?. value ?. message === '[LARGE_TOOL_RESULT_OMITTED]' ) ,
158161 )
159162 expect ( recentTerminalMessage ) . toBeDefined ( )
160163 } )
@@ -191,9 +194,7 @@ describe('context-pruner handleSteps', () => {
191194 const smallResultMessage = resultMessages . find (
192195 ( m : any ) => m . role === 'tool' && m . toolName === 'code_search' ,
193196 )
194- expect ( smallResultMessage ?. content ?. [ 0 ] ?. value ?. data ) . toBe (
195- 'Small result' ,
196- )
197+ expect ( smallResultMessage ?. content ?. [ 0 ] ?. value ?. data ) . toBe ( 'Small result' )
197198 } )
198199
199200 test ( 'performs message-level pruning when other passes are insufficient' , ( ) => {
@@ -299,14 +300,6 @@ describe('context-pruner edge cases', () => {
299300 }
300301 } )
301302
302- const createMessage = (
303- role : 'user' | 'assistant' ,
304- content : string ,
305- ) : Message => ( {
306- role,
307- content,
308- } )
309-
310303 const createTerminalToolMessage = (
311304 command : string ,
312305 output : string ,
@@ -366,8 +359,7 @@ describe('context-pruner edge cases', () => {
366359
367360 // Valid terminal command should be processed correctly
368361 const validCommand = resultMessages . find (
369- ( m : any ) =>
370- m . role === 'tool' && m . toolName === 'run_terminal_command' ,
362+ ( m : any ) => m . role === 'tool' && m . toolName === 'run_terminal_command' ,
371363 )
372364 expect ( validCommand ) . toBeDefined ( )
373365 } )
@@ -481,8 +473,7 @@ describe('context-pruner edge cases', () => {
481473 const hasLargeToolResultReplacement = resultMessages . some (
482474 ( m : any ) =>
483475 m . role === 'tool' &&
484- m . content ?. [ 0 ] ?. value ?. message ===
485- '[LARGE_TOOL_RESULT_OMITTED]' ,
476+ m . content ?. [ 0 ] ?. value ?. message === '[LARGE_TOOL_RESULT_OMITTED]' ,
486477 )
487478 expect ( hasLargeToolResultReplacement ) . toBe ( true )
488479 } )
0 commit comments