fix: update comments for clarity in splitMessagesForPreservation function

This commit is contained in:
Charlie 2026-01-27 04:04:40 +00:00
parent 89080816b2
commit f19971b34a
2 changed files with 4 additions and 4 deletions

View File

@ -220,8 +220,8 @@ describe("splitMessagesForPreservation", () => {
const result = splitMessagesForPreservation({ messages, keepLastMessages: 2 }); const result = splitMessagesForPreservation({ messages, keepLastMessages: 2 });
// Should preserve both complete pairs (u1+a1, u2+a2), and include the trailing u3 // Should preserve the last 2 complete pairs (u1+a1, u2+a2), plus the trailing incomplete u3
// because it comes after the preserved pairs // because trailing incomplete messages are always included with preserved pairs
expect(result.preservedPairs).toBe(2); expect(result.preservedPairs).toBe(2);
expect(result.toPreserve.length).toBe(5); // u1, a1, u2, a2, u3 expect(result.toPreserve.length).toBe(5); // u1, a1, u2, a2, u3
expect(result.toSummarize.length).toBe(0); expect(result.toSummarize.length).toBe(0);

View File

@ -353,7 +353,7 @@ export function splitMessagesForPreservation(params: {
preservedPairs: number; preservedPairs: number;
} { } {
const { messages, keepLastMessages } = params; const { messages, keepLastMessages } = params;
if (keepLastMessages <= 0 || messages.length === 0) { if (keepLastMessages <= 0 || messages.length === 0) {
return { return {
toSummarize: messages, toSummarize: messages,
@ -390,9 +390,9 @@ export function splitMessagesForPreservation(params: {
// If we found a user message, we have a complete pair // If we found a user message, we have a complete pair
if (userIndex !== -1) { if (userIndex !== -1) {
pairsFound++; pairsFound++;
splitIndex = userIndex; // Update split index to start of this pair
if (pairsFound >= keepLastMessages) { if (pairsFound >= keepLastMessages) {
splitIndex = userIndex; // Update split index to start of this pair
// We've found enough pairs // We've found enough pairs
break; break;
} }