Skip to content

Commit 88c90a2

Browse files
committed
fix tests
1 parent 93ce544 commit 88c90a2

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

src/core/condense/__tests__/index.test.ts

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -36,7 +36,7 @@ describe("getMessagesSinceLastSummary", () => {
3636
expect(result).toEqual(messages)
3737
})
3838

39-
it("should return messages since the last summary", () => {
39+
it("should return messages since the last summary with prepended user message", () => {
4040
const messages: ApiMessage[] = [
4141
{ role: "user", content: "Hello", ts: 1 },
4242
{ role: "assistant", content: "Hi there", ts: 2 },
@@ -47,13 +47,14 @@ describe("getMessagesSinceLastSummary", () => {
4747

4848
const result = getMessagesSinceLastSummary(messages)
4949
expect(result).toEqual([
50+
{ role: "user", content: "Please continue from the following summary:", ts: 0 },
5051
{ role: "assistant", content: "Summary of conversation", ts: 3, isSummary: true },
5152
{ role: "user", content: "How are you?", ts: 4 },
5253
{ role: "assistant", content: "I'm good", ts: 5 },
5354
])
5455
})
5556

56-
it("should handle multiple summary messages and return since the last one", () => {
57+
it("should handle multiple summary messages and return since the last one with prepended user message", () => {
5758
const messages: ApiMessage[] = [
5859
{ role: "user", content: "Hello", ts: 1 },
5960
{ role: "assistant", content: "First summary", ts: 2, isSummary: true },
@@ -64,6 +65,7 @@ describe("getMessagesSinceLastSummary", () => {
6465

6566
const result = getMessagesSinceLastSummary(messages)
6667
expect(result).toEqual([
68+
{ role: "user", content: "Please continue from the following summary:", ts: 0 },
6769
{ role: "assistant", content: "Second summary", ts: 4, isSummary: true },
6870
{ role: "user", content: "What's new?", ts: 5 },
6971
])

0 commit comments

Comments
 (0)