Skip to content

Commit aad6b70

Browse files
committed
Update context-pruner: with no instructionsPrompt added, we don't need to remove it each time
1 parent cd79db8 commit aad6b70

File tree

2 files changed

+2
-14
lines changed

2 files changed

+2
-14
lines changed

.agents/__tests__/context-pruner.test.ts

Lines changed: 2 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -65,12 +65,7 @@ describe('context-pruner handleSteps', () => {
6565
})
6666

6767
const runHandleSteps = (messages: Message[]) => {
68-
// Add instructions prompt that will be removed by context-pruner
69-
const messagesWithInstructions = [
70-
...messages,
71-
createMessage('user', 'Instructions prompt'),
72-
]
73-
mockAgentState.messageHistory = messagesWithInstructions
68+
mockAgentState.messageHistory = messages
7469
const mockLogger = {
7570
debug: () => {},
7671
info: () => {},
@@ -337,12 +332,7 @@ describe('context-pruner edge cases', () => {
337332
})
338333

339334
const runHandleSteps = (messages: Message[]) => {
340-
// Add instructions prompt that will be removed by context-pruner
341-
const messagesWithInstructions = [
342-
...messages,
343-
createMessage('user', 'Instructions prompt'),
344-
]
345-
mockAgentState.messageHistory = messagesWithInstructions
335+
mockAgentState.messageHistory = messages
346336
const mockLogger = {
347337
debug: () => {},
348338
info: () => {},

.agents/context-pruner.ts

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -40,8 +40,6 @@ const definition: AgentDefinition = {
4040
const numTerminalCommandsToKeep = 5
4141

4242
let currentMessages = [...messages]
43-
// Remove injected instructions prompt
44-
currentMessages.pop()
4543

4644
// Initial check - if already under limit, return
4745
const initialTokens = countTokensJson(currentMessages)

0 commit comments

Comments
 (0)