wehub-resource-sync
77 lines
3.1 KiB
TypeScript
77 lines
3.1 KiB
TypeScript
/**
|
|
* Tests for the Instructor thread memory compression.
|
|
*/
|
|
import { describe, it, expect } from 'vitest';
|
|
import {
|
|
compressIfNeeded,
|
|
COMPRESS_THRESHOLD,
|
|
MAX_EARLIER_SUMMARY_CHARS,
|
|
} from '@/lib/pbl/v2/agents/instructor-memory';
|
|
import type { PBLAgentThread, PBLChatMessage } from '@/lib/pbl/v2/types';
|
|
|
|
function makeThread(messageCount: number): PBLAgentThread {
|
|
const messages: PBLChatMessage[] = [];
|
|
for (let i = 0; i < messageCount; i++) {
|
|
messages.push({
|
|
id: `msg-${i}`,
|
|
roleType: i % 2 === 0 ? 'user' : 'instructor',
|
|
content: i % 2 === 0 ? `user msg #${i}` : `instructor reply #${i}`,
|
|
ts: new Date(Date.UTC(2026, 4, 26, 12, 0, i)).toISOString(),
|
|
});
|
|
}
|
|
return { agentId: 'role-instructor', messages };
|
|
}
|
|
|
|
describe('PBL v2 — instructor memory compression', () => {
|
|
it('returns the thread unchanged when below threshold', () => {
|
|
const thread = makeThread(COMPRESS_THRESHOLD - 1);
|
|
const out = compressIfNeeded(thread);
|
|
expect(out).toBe(thread); // same reference, no work done
|
|
});
|
|
|
|
it('compresses when above threshold', () => {
|
|
const thread = makeThread(COMPRESS_THRESHOLD + 10);
|
|
const out = compressIfNeeded(thread);
|
|
expect(out.messages.length).toBeLessThan(thread.messages.length);
|
|
expect(out.earlierSummary).toBeDefined();
|
|
expect(out.earlierSummary).toContain('Earlier conversation digest');
|
|
});
|
|
|
|
it('cuts on a user-message boundary so a turn pair is never split', () => {
|
|
const thread = makeThread(COMPRESS_THRESHOLD + 10);
|
|
const out = compressIfNeeded(thread);
|
|
// After cut, the first surviving message should be a user message
|
|
// (digestor invariant).
|
|
if (out.messages.length > 0) {
|
|
expect(out.messages[0].roleType).toBe('user');
|
|
}
|
|
});
|
|
|
|
it('concatenates with prior earlierSummary instead of replacing it', () => {
|
|
const thread = makeThread(COMPRESS_THRESHOLD + 10);
|
|
thread.earlierSummary = 'PRIOR_SUMMARY_MARKER';
|
|
const out = compressIfNeeded(thread);
|
|
expect(out.earlierSummary).toContain('PRIOR_SUMMARY_MARKER');
|
|
expect(out.earlierSummary).toContain('Earlier conversation digest');
|
|
});
|
|
|
|
it('extracts durable learner context from folded turns', () => {
|
|
const thread = makeThread(COMPRESS_THRESHOLD + 10);
|
|
thread.messages[4].content = '我是零基础，想最简单一点。';
|
|
thread.messages[8].content = '我用 VS Code 和终端，Python 已经装好了。';
|
|
thread.messages[12].content = '这里报错了，我看不懂 traceback。';
|
|
const out = compressIfNeeded(thread);
|
|
expect(out.earlierSummary).toContain('Learner context inferred');
|
|
expect(out.earlierSummary).toContain('Level/preference');
|
|
expect(out.earlierSummary).toContain('Environment/tools');
|
|
expect(out.earlierSummary).toContain('Blocker/confusion');
|
|
});
|
|
|
|
it('keeps folded memory bounded across repeated compressions', () => {
|
|
let thread = makeThread(COMPRESS_THRESHOLD + 10);
|
|
thread.earlierSummary = 'x'.repeat(MAX_EARLIER_SUMMARY_CHARS + 2000);
|
|
thread = compressIfNeeded(thread);
|
|
expect(thread.earlierSummary!.length).toBeLessThanOrEqual(MAX_EARLIER_SUMMARY_CHARS + 120);
|
|
});
|
|
});
|