feat: 灵犀 Studio Web UI 定制版
Co-authored-by: Cursor <cursoragent@cursor.com>
This commit is contained in:
@@ -0,0 +1,502 @@
|
||||
import { afterEach, beforeEach, describe, expect, it, vi } from 'vitest'
|
||||
|
||||
const getCompressionSnapshotMock = vi.fn()
|
||||
const saveCompressionSnapshotMock = vi.fn()
|
||||
const deleteCompressionSnapshotMock = vi.fn()
|
||||
const bridgeRequestMock = vi.fn()
|
||||
const bridgeDestroyMock = vi.fn()
|
||||
|
||||
vi.mock('../../packages/server/src/services/logger', () => ({
|
||||
logger: {
|
||||
info: vi.fn(),
|
||||
warn: vi.fn(),
|
||||
error: vi.fn(),
|
||||
debug: vi.fn(),
|
||||
},
|
||||
}))
|
||||
|
||||
vi.mock('../../packages/server/src/db/hermes/compression-snapshot', () => ({
|
||||
getCompressionSnapshot: getCompressionSnapshotMock,
|
||||
saveCompressionSnapshot: saveCompressionSnapshotMock,
|
||||
deleteCompressionSnapshot: deleteCompressionSnapshotMock,
|
||||
}))
|
||||
|
||||
vi.mock('../../packages/server/src/services/hermes/agent-bridge', () => ({
|
||||
AgentBridgeClient: class {
|
||||
request = bridgeRequestMock
|
||||
destroy = bridgeDestroyMock
|
||||
},
|
||||
}))
|
||||
|
||||
describe('ChatContextCompressor', () => {
|
||||
let originalFetch: typeof global.fetch
|
||||
|
||||
beforeEach(() => {
|
||||
originalFetch = global.fetch
|
||||
getCompressionSnapshotMock.mockReset()
|
||||
saveCompressionSnapshotMock.mockReset()
|
||||
deleteCompressionSnapshotMock.mockReset()
|
||||
bridgeRequestMock.mockReset()
|
||||
bridgeDestroyMock.mockReset()
|
||||
bridgeRequestMock.mockRejectedValue(new Error('summarizer failed'))
|
||||
bridgeDestroyMock.mockResolvedValue(undefined)
|
||||
})
|
||||
|
||||
afterEach(() => {
|
||||
global.fetch = originalFetch
|
||||
})
|
||||
|
||||
it('keeps full history when full summarization fails', async () => {
|
||||
const { ChatContextCompressor } = await import('../../packages/server/src/lib/context-compressor')
|
||||
const compressor = new ChatContextCompressor({ config: { tailMessageCount: 3 } })
|
||||
const messages = Array.from({ length: 8 }, (_, i) => ({
|
||||
role: i % 2 === 0 ? 'user' : 'assistant',
|
||||
content: `message ${i}`,
|
||||
}))
|
||||
|
||||
getCompressionSnapshotMock.mockReturnValue(null)
|
||||
|
||||
const result = await compressor.compress(messages, 'http://upstream', undefined, 's1')
|
||||
|
||||
expect(result.messages).toHaveLength(messages.length)
|
||||
expect(result.messages.map(m => m.content)).toEqual(messages.map(m => m.content))
|
||||
expect(result.meta.compressed).toBe(false)
|
||||
expect(result.meta.llmCompressed).toBe(false)
|
||||
expect(saveCompressionSnapshotMock).not.toHaveBeenCalled()
|
||||
})
|
||||
|
||||
it('keeps all new messages when incremental summarization fails', async () => {
|
||||
const { ChatContextCompressor, SUMMARY_PREFIX } = await import('../../packages/server/src/lib/context-compressor')
|
||||
const compressor = new ChatContextCompressor({ config: { tailMessageCount: 3 } })
|
||||
const messages = Array.from({ length: 8 }, (_, i) => ({
|
||||
role: i % 2 === 0 ? 'user' : 'assistant',
|
||||
content: `message ${i}`,
|
||||
}))
|
||||
|
||||
getCompressionSnapshotMock.mockReturnValue({
|
||||
summary: 'previous summary',
|
||||
lastMessageIndex: 1,
|
||||
messageCountAtTime: 2,
|
||||
})
|
||||
|
||||
const result = await compressor.compress(messages, 'http://upstream', undefined, 's1')
|
||||
|
||||
expect(result.messages).toHaveLength(7)
|
||||
expect(result.messages[0]).toEqual({
|
||||
role: 'user',
|
||||
content: `${SUMMARY_PREFIX}\n\nprevious summary`,
|
||||
})
|
||||
expect(result.messages.slice(1).map(m => m.content)).toEqual(messages.slice(2).map(m => m.content))
|
||||
expect(result.meta.compressed).toBe(true)
|
||||
expect(result.meta.llmCompressed).toBe(false)
|
||||
expect(result.meta.compressedStartIndex).toBe(1)
|
||||
expect(result.meta.verbatimCount).toBe(6)
|
||||
expect(saveCompressionSnapshotMock).not.toHaveBeenCalled()
|
||||
})
|
||||
|
||||
it('does not call the summarizer when snapshot has only tail messages after it', async () => {
|
||||
const { ChatContextCompressor, SUMMARY_PREFIX } = await import('../../packages/server/src/lib/context-compressor')
|
||||
const compressor = new ChatContextCompressor({ config: { tailMessageCount: 10 } })
|
||||
const messages = Array.from({ length: 6 }, (_, i) => ({
|
||||
role: i % 2 === 0 ? 'user' : 'assistant',
|
||||
content: `message ${i}`,
|
||||
}))
|
||||
const fetchMock = vi.fn()
|
||||
|
||||
getCompressionSnapshotMock.mockReturnValue({
|
||||
summary: 'previous summary',
|
||||
lastMessageIndex: 3,
|
||||
messageCountAtTime: 4,
|
||||
})
|
||||
global.fetch = fetchMock as any
|
||||
|
||||
const result = await compressor.compress(messages, 'http://upstream', undefined, 's1')
|
||||
|
||||
expect(fetchMock).not.toHaveBeenCalled()
|
||||
expect(result.messages).toHaveLength(3)
|
||||
expect(result.messages[0].content).toBe(`${SUMMARY_PREFIX}\n\nprevious summary`)
|
||||
expect(result.messages.slice(1).map(m => m.content)).toEqual(['message 4', 'message 5'])
|
||||
expect(result.meta.llmCompressed).toBe(false)
|
||||
expect(result.meta.compressedStartIndex).toBe(3)
|
||||
expect(saveCompressionSnapshotMock).not.toHaveBeenCalled()
|
||||
})
|
||||
|
||||
it('keeps configured first and last messages during full compression', async () => {
|
||||
const { ChatContextCompressor, SUMMARY_PREFIX } = await import('../../packages/server/src/lib/context-compressor')
|
||||
const compressor = new ChatContextCompressor({
|
||||
config: { headMessageCount: 2, tailMessageCount: 3, summaryBudget: 1000 },
|
||||
})
|
||||
const messages = Array.from({ length: 10 }, (_, i) => ({
|
||||
role: i % 2 === 0 ? 'user' : 'assistant',
|
||||
content: `message ${i}`,
|
||||
}))
|
||||
|
||||
getCompressionSnapshotMock.mockReturnValue(null)
|
||||
bridgeRequestMock.mockResolvedValue({
|
||||
status: 'completed',
|
||||
result: { final_response: 'compressed summary' },
|
||||
})
|
||||
|
||||
const result = await compressor.compress(messages, 'http://upstream', undefined, 's1')
|
||||
|
||||
expect(result.messages.map(m => m.content)).toEqual([
|
||||
'message 0',
|
||||
'message 1',
|
||||
`${SUMMARY_PREFIX}\n\ncompressed summary`,
|
||||
'message 7',
|
||||
'message 8',
|
||||
'message 9',
|
||||
])
|
||||
expect(result.meta.compressed).toBe(true)
|
||||
expect(result.meta.llmCompressed).toBe(true)
|
||||
expect(result.meta.verbatimCount).toBe(5)
|
||||
expect(saveCompressionSnapshotMock).toHaveBeenCalledWith('s1', 'compressed summary', 6, 10)
|
||||
})
|
||||
|
||||
it('routes summarization through the provided worker key and destroys only the temporary agent session', async () => {
|
||||
const { ChatContextCompressor } = await import('../../packages/server/src/lib/context-compressor')
|
||||
const compressor = new ChatContextCompressor({
|
||||
config: { headMessageCount: 0, tailMessageCount: 1, summaryBudget: 1000 },
|
||||
})
|
||||
const messages = [
|
||||
{ role: 'user', content: 'old context' },
|
||||
{ role: 'assistant', content: 'old response' },
|
||||
{ role: 'user', content: 'tail' },
|
||||
]
|
||||
getCompressionSnapshotMock.mockReturnValue(null)
|
||||
bridgeRequestMock.mockResolvedValue({
|
||||
status: 'completed',
|
||||
result: { final_response: 'compressed summary' },
|
||||
})
|
||||
|
||||
await compressor.compress(messages, 'http://upstream', undefined, 's1', {
|
||||
profile: 'default',
|
||||
workerKey: 'default:compression:s1',
|
||||
})
|
||||
|
||||
expect(bridgeRequestMock).toHaveBeenCalledWith(expect.objectContaining({
|
||||
action: 'chat',
|
||||
profile: 'default',
|
||||
worker_key: 'default:compression:s1',
|
||||
message: 'Generate the context checkpoint summary now.',
|
||||
wait: true,
|
||||
}), expect.any(Object))
|
||||
const request = bridgeRequestMock.mock.calls[0][0]
|
||||
expect(request.conversation_history[0]).toEqual(expect.objectContaining({
|
||||
role: 'user',
|
||||
content: expect.stringContaining('TURNS TO SUMMARIZE:'),
|
||||
}))
|
||||
const compressSessionId = bridgeRequestMock.mock.calls[0][0].session_id
|
||||
expect(String(compressSessionId)).toMatch(/^compress_/)
|
||||
expect(bridgeDestroyMock).toHaveBeenCalledWith(
|
||||
compressSessionId,
|
||||
'default',
|
||||
'default:compression:s1',
|
||||
)
|
||||
})
|
||||
|
||||
it('does not pre-prune tool results before sending them to the summarizer', async () => {
|
||||
const { ChatContextCompressor } = await import('../../packages/server/src/lib/context-compressor')
|
||||
const compressor = new ChatContextCompressor({
|
||||
config: { headMessageCount: 0, tailMessageCount: 1, summaryBudget: 1000 },
|
||||
})
|
||||
const longToolOutput = `${'x'.repeat(180)}KEEP_MARKER${'y'.repeat(180)}`
|
||||
const messages = [
|
||||
{
|
||||
role: 'assistant',
|
||||
content: 'calling terminal',
|
||||
tool_calls: [{ id: 'call_1', type: 'function', function: { name: 'terminal', arguments: '{}' } }],
|
||||
},
|
||||
{ role: 'tool', name: 'terminal', tool_call_id: 'call_1', content: longToolOutput },
|
||||
{ role: 'user', content: 'tail' },
|
||||
]
|
||||
|
||||
getCompressionSnapshotMock.mockReturnValue(null)
|
||||
bridgeRequestMock.mockResolvedValue({
|
||||
status: 'completed',
|
||||
result: { final_response: 'compressed summary' },
|
||||
})
|
||||
|
||||
await compressor.compress(messages, 'http://upstream', undefined, 's1')
|
||||
|
||||
const request = bridgeRequestMock.mock.calls[0][0]
|
||||
const serializedHistory = JSON.stringify(request.conversation_history)
|
||||
expect(serializedHistory).toContain('KEEP_MARKER')
|
||||
expect(serializedHistory).not.toContain('[terminal] ')
|
||||
})
|
||||
|
||||
it('keeps protected head tool results verbatim after successful full compression', async () => {
|
||||
const { ChatContextCompressor, SUMMARY_PREFIX } = await import('../../packages/server/src/lib/context-compressor')
|
||||
const compressor = new ChatContextCompressor({
|
||||
config: { headMessageCount: 2, tailMessageCount: 1, summaryBudget: 1000 },
|
||||
})
|
||||
const longToolOutput = `${'head-tool-output '.repeat(30)}KEEP_HEAD_TOOL`
|
||||
const messages = [
|
||||
{
|
||||
role: 'assistant',
|
||||
content: 'calling terminal',
|
||||
tool_calls: [{ id: 'call_1', type: 'function', function: { name: 'terminal', arguments: '{}' } }],
|
||||
},
|
||||
{ role: 'tool', name: 'terminal', tool_call_id: 'call_1', content: longToolOutput },
|
||||
{ role: 'user', content: 'middle' },
|
||||
{ role: 'assistant', content: 'tail' },
|
||||
]
|
||||
|
||||
getCompressionSnapshotMock.mockReturnValue(null)
|
||||
bridgeRequestMock.mockResolvedValue({
|
||||
status: 'completed',
|
||||
result: { final_response: 'compressed summary' },
|
||||
})
|
||||
|
||||
const result = await compressor.compress(messages, 'http://upstream', undefined, 's1')
|
||||
|
||||
expect(result.messages.map(m => m.content)).toEqual([
|
||||
'calling terminal',
|
||||
longToolOutput,
|
||||
`${SUMMARY_PREFIX}\n\ncompressed summary`,
|
||||
'tail',
|
||||
])
|
||||
})
|
||||
|
||||
it('uses the previous summary plus a safe tail when an existing snapshot index is stale', async () => {
|
||||
const { ChatContextCompressor, SUMMARY_PREFIX } = await import('../../packages/server/src/lib/context-compressor')
|
||||
const compressor = new ChatContextCompressor({
|
||||
config: { headMessageCount: 2, tailMessageCount: 3, summaryBudget: 1000 },
|
||||
})
|
||||
const messages = Array.from({ length: 8 }, (_, i) => ({
|
||||
role: i % 2 === 0 ? 'user' : 'assistant',
|
||||
content: `message ${i}`,
|
||||
}))
|
||||
|
||||
getCompressionSnapshotMock.mockReturnValue({
|
||||
summary: 'stale previous summary',
|
||||
lastMessageIndex: 20,
|
||||
messageCountAtTime: 21,
|
||||
})
|
||||
bridgeRequestMock.mockResolvedValue({
|
||||
status: 'completed',
|
||||
result: { final_response: 'rebuilt summary' },
|
||||
})
|
||||
|
||||
const result = await compressor.compress(messages, 'http://upstream', undefined, 's1')
|
||||
|
||||
expect(deleteCompressionSnapshotMock).not.toHaveBeenCalled()
|
||||
expect(bridgeRequestMock).not.toHaveBeenCalled()
|
||||
expect(result.messages.map(m => m.content)).toEqual([
|
||||
'message 0',
|
||||
'message 1',
|
||||
`${SUMMARY_PREFIX}\n\nstale previous summary`,
|
||||
'message 5',
|
||||
'message 6',
|
||||
'message 7',
|
||||
])
|
||||
expect(saveCompressionSnapshotMock).not.toHaveBeenCalled()
|
||||
})
|
||||
|
||||
it('folds a stale snapshot safe tail into a new summary when it still exceeds budget', async () => {
|
||||
const { ChatContextCompressor, SUMMARY_PREFIX } = await import('../../packages/server/src/lib/context-compressor')
|
||||
const compressor = new ChatContextCompressor({
|
||||
config: { triggerTokens: 800, headMessageCount: 2, tailMessageCount: 3, summaryBudget: 1000 },
|
||||
})
|
||||
const largeTail = 'tail-token '.repeat(200)
|
||||
const messages = [
|
||||
{ role: 'user', content: 'message 0' },
|
||||
{ role: 'assistant', content: 'message 1' },
|
||||
{ role: 'user', content: 'message 2' },
|
||||
{ role: 'assistant', content: 'message 3' },
|
||||
{ role: 'user', content: 'message 4' },
|
||||
{ role: 'assistant', content: largeTail },
|
||||
{ role: 'user', content: largeTail },
|
||||
{ role: 'assistant', content: largeTail },
|
||||
]
|
||||
|
||||
getCompressionSnapshotMock.mockReturnValue({
|
||||
summary: 'stale previous summary',
|
||||
lastMessageIndex: 20,
|
||||
messageCountAtTime: 21,
|
||||
})
|
||||
bridgeRequestMock.mockResolvedValue({
|
||||
status: 'completed',
|
||||
result: { final_response: 'updated stale summary' },
|
||||
})
|
||||
|
||||
const result = await compressor.compress(messages, 'http://upstream', undefined, 's1')
|
||||
|
||||
expect(deleteCompressionSnapshotMock).not.toHaveBeenCalled()
|
||||
expect(bridgeRequestMock).toHaveBeenCalledTimes(1)
|
||||
expect(result.messages.map(m => m.content)).toEqual([
|
||||
'message 0',
|
||||
'message 1',
|
||||
`${SUMMARY_PREFIX}\n\nupdated stale summary`,
|
||||
])
|
||||
expect(saveCompressionSnapshotMock).toHaveBeenCalledWith('s1', 'updated stale summary', 7, 8)
|
||||
})
|
||||
|
||||
it('compresses the full history when protected windows cover all messages', async () => {
|
||||
const { ChatContextCompressor, SUMMARY_PREFIX } = await import('../../packages/server/src/lib/context-compressor')
|
||||
const compressor = new ChatContextCompressor({
|
||||
config: { headMessageCount: 3, tailMessageCount: 20, summaryBudget: 1000 },
|
||||
})
|
||||
const messages = Array.from({ length: 20 }, (_, i) => ({
|
||||
role: i % 2 === 0 ? 'user' : 'assistant',
|
||||
content: `message ${i}`,
|
||||
}))
|
||||
|
||||
getCompressionSnapshotMock.mockReturnValue(null)
|
||||
bridgeRequestMock.mockResolvedValue({
|
||||
status: 'completed',
|
||||
result: { final_response: 'compressed all messages' },
|
||||
})
|
||||
|
||||
const result = await compressor.compress(messages, 'http://upstream', undefined, 's1')
|
||||
|
||||
expect(bridgeRequestMock).toHaveBeenCalledTimes(1)
|
||||
expect(result.messages.map(m => m.content)).toEqual([
|
||||
`${SUMMARY_PREFIX}\n\ncompressed all messages`,
|
||||
])
|
||||
expect(result.meta.compressed).toBe(true)
|
||||
expect(result.meta.llmCompressed).toBe(true)
|
||||
expect(result.meta.verbatimCount).toBe(0)
|
||||
expect(result.meta.compressedStartIndex).toBe(19)
|
||||
expect(saveCompressionSnapshotMock).toHaveBeenCalledWith('s1', 'compressed all messages', 19, 20)
|
||||
})
|
||||
|
||||
it('drops protected messages when compressed output still exceeds the trigger budget', async () => {
|
||||
const { ChatContextCompressor, SUMMARY_PREFIX } = await import('../../packages/server/src/lib/context-compressor')
|
||||
const compressor = new ChatContextCompressor({
|
||||
config: { triggerTokens: 200, headMessageCount: 2, tailMessageCount: 2, summaryBudget: 100 },
|
||||
})
|
||||
const largeText = 'tail-token '.repeat(500)
|
||||
const messages = [
|
||||
{ role: 'user', content: 'head 0' },
|
||||
{ role: 'assistant', content: 'head 1' },
|
||||
{ role: 'user', content: 'middle 2' },
|
||||
{ role: 'assistant', content: 'middle 3' },
|
||||
{ role: 'user', content: largeText },
|
||||
{ role: 'assistant', content: largeText },
|
||||
]
|
||||
|
||||
getCompressionSnapshotMock.mockReturnValue(null)
|
||||
bridgeRequestMock.mockResolvedValue({
|
||||
status: 'completed',
|
||||
result: { final_response: 'short summary' },
|
||||
})
|
||||
|
||||
const result = await compressor.compress(messages, 'http://upstream', undefined, 's1')
|
||||
|
||||
expect(result.messages.map(m => m.content)).toEqual([
|
||||
`${SUMMARY_PREFIX}\n\nshort summary`,
|
||||
])
|
||||
expect(result.meta.compressed).toBe(true)
|
||||
expect(result.meta.llmCompressed).toBe(true)
|
||||
expect(result.meta.verbatimCount).toBe(0)
|
||||
})
|
||||
|
||||
it('truncates the summary when the summary alone exceeds the trigger budget', async () => {
|
||||
const { ChatContextCompressor, SUMMARY_PREFIX, countTokens } = await import('../../packages/server/src/lib/context-compressor')
|
||||
const compressor = new ChatContextCompressor({
|
||||
config: { triggerTokens: 120, headMessageCount: 2, tailMessageCount: 2, summaryBudget: 100 },
|
||||
})
|
||||
const messages = Array.from({ length: 6 }, (_, i) => ({
|
||||
role: i % 2 === 0 ? 'user' : 'assistant',
|
||||
content: `message ${i}`,
|
||||
}))
|
||||
const longSummary = 'summary-token '.repeat(500)
|
||||
|
||||
getCompressionSnapshotMock.mockReturnValue(null)
|
||||
bridgeRequestMock.mockResolvedValue({
|
||||
status: 'completed',
|
||||
result: { final_response: longSummary },
|
||||
})
|
||||
|
||||
const result = await compressor.compress(messages, 'http://upstream', undefined, 's1')
|
||||
|
||||
expect(result.messages).toHaveLength(1)
|
||||
expect(String(result.messages[0].content)).toContain('[Summary truncated to fit context budget]')
|
||||
expect(String(result.messages[0].content).startsWith(SUMMARY_PREFIX)).toBe(true)
|
||||
expect(countTokens(String(result.messages[0].content))).toBeLessThanOrEqual(140)
|
||||
expect(result.meta.verbatimCount).toBe(0)
|
||||
})
|
||||
|
||||
it('keeps configured first messages when incremental compression reuses an existing snapshot', async () => {
|
||||
const { ChatContextCompressor, SUMMARY_PREFIX } = await import('../../packages/server/src/lib/context-compressor')
|
||||
const compressor = new ChatContextCompressor({
|
||||
config: { headMessageCount: 2, tailMessageCount: 10 },
|
||||
})
|
||||
const messages = Array.from({ length: 6 }, (_, i) => ({
|
||||
role: i % 2 === 0 ? 'user' : 'assistant',
|
||||
content: `message ${i}`,
|
||||
}))
|
||||
|
||||
getCompressionSnapshotMock.mockReturnValue({
|
||||
summary: 'previous summary',
|
||||
lastMessageIndex: 3,
|
||||
messageCountAtTime: 4,
|
||||
})
|
||||
|
||||
const result = await compressor.compress(messages, 'http://upstream', undefined, 's1')
|
||||
|
||||
expect(bridgeRequestMock).not.toHaveBeenCalled()
|
||||
expect(result.messages.map(m => m.content)).toEqual([
|
||||
'message 0',
|
||||
'message 1',
|
||||
`${SUMMARY_PREFIX}\n\nprevious summary`,
|
||||
'message 4',
|
||||
'message 5',
|
||||
])
|
||||
expect(result.meta.verbatimCount).toBe(4)
|
||||
expect(saveCompressionSnapshotMock).not.toHaveBeenCalled()
|
||||
})
|
||||
|
||||
it('folds all new messages into the summary when incremental tail protection would exceed budget', async () => {
|
||||
const { ChatContextCompressor, SUMMARY_PREFIX } = await import('../../packages/server/src/lib/context-compressor')
|
||||
const compressor = new ChatContextCompressor({
|
||||
config: { triggerTokens: 1000, headMessageCount: 3, tailMessageCount: 20, summaryBudget: 100 },
|
||||
})
|
||||
const largeText = 'new-token '.repeat(80)
|
||||
const messages = [
|
||||
{ role: 'user', content: 'head 0' },
|
||||
{ role: 'assistant', content: 'head 1' },
|
||||
{ role: 'user', content: 'head 2' },
|
||||
...Array.from({ length: 20 }, (_, i) => ({
|
||||
role: i % 2 === 0 ? 'user' : 'assistant',
|
||||
content: `${largeText}${i}`,
|
||||
})),
|
||||
]
|
||||
|
||||
getCompressionSnapshotMock.mockReturnValue({
|
||||
summary: 'previous summary',
|
||||
lastMessageIndex: 2,
|
||||
messageCountAtTime: 3,
|
||||
})
|
||||
bridgeRequestMock.mockResolvedValue({
|
||||
status: 'completed',
|
||||
result: { final_response: 'updated summary' },
|
||||
})
|
||||
|
||||
const result = await compressor.compress(messages, 'http://upstream', undefined, 's1')
|
||||
|
||||
expect(bridgeRequestMock).toHaveBeenCalledTimes(1)
|
||||
const request = bridgeRequestMock.mock.calls[0][0]
|
||||
expect(request.message).toBe('Generate the context checkpoint summary now.')
|
||||
expect(request.conversation_history.slice(0, 3)).toEqual([
|
||||
{ role: 'user', content: '[Previous summary]\nprevious summary' },
|
||||
{ role: 'assistant', content: 'Understood, I will update the summary.' },
|
||||
expect.objectContaining({
|
||||
role: 'user',
|
||||
content: expect.stringContaining('NEW TURNS TO INCORPORATE:'),
|
||||
}),
|
||||
])
|
||||
expect(result.messages.map(m => m.content)).toEqual([
|
||||
'head 0',
|
||||
'head 1',
|
||||
'head 2',
|
||||
`${SUMMARY_PREFIX}\n\nupdated summary`,
|
||||
])
|
||||
expect(result.meta.compressed).toBe(true)
|
||||
expect(result.meta.llmCompressed).toBe(true)
|
||||
expect(result.meta.verbatimCount).toBe(3)
|
||||
expect(result.meta.compressedStartIndex).toBe(22)
|
||||
expect(saveCompressionSnapshotMock).toHaveBeenCalledWith('s1', 'updated summary', 22, 23)
|
||||
})
|
||||
})
|
||||
Reference in New Issue
Block a user