X-Git-Url: http://git.giacomolli.uk/onechat/blobdiff_plain/4b9b22619008f0829b7f5a8685e7b1d2a9aeb6bb..6d1d9d9a7cf90b25a2e10af1c5bc56faee7e494c:/compactor_test.go diff --git a/compactor_test.go b/compactor_test.go index 20862c7..8b14c81 100644 --- a/compactor_test.go +++ b/compactor_test.go @@ -14,13 +14,13 @@ import ( ) func TestCompactorSendsThePromptToolsAndTask(t *testing.T) { - server := newFakeResponses(t) + server := newModelFake(t) compactor := newTestCompactor(t, server) line, err := compactor.Summarize(t.Context(), Task{ Level: 0, Index: 7, - Context: "\n0+4|a\n4+1|b\n5+1|c\n6+1|d\n7+1|e\n", + Context: "\n0+4|user: earlier\n", Input: []string{"user: " + strings.Repeat("x", 600)}, }) require.NoError(t, err) @@ -28,7 +28,7 @@ func TestCompactorSendsThePromptToolsAndTask(t *testing.T) { require.Equal(t, 1, server.Calls()) body := server.Call(0).body - require.Equal(t, prompt("Ada"), body["instructions"]) + require.Equal(t, prompt("Ada", ""), body["instructions"]) require.Equal(t, "gpt-test", body["model"]) require.Equal(t, map[string]any{"effort": "high"}, body["reasoning"]) require.Equal(t, "agent-key", body["prompt_cache_key"]) @@ -42,66 +42,14 @@ func TestCompactorSendsThePromptToolsAndTask(t *testing.T) { parts := inputParts(t, server.Call(0)) require.Equal(t, []string{ - "\n", - "0+4|a\n4+1|b\n5+1|c\n6+1|d\n", - "7+1|e\n\n" + - "Compaction: compress message 7 into one line of at most 512 bytes\n" + + "\n0+4|user: earlier\n\n", + "Compaction: compress message 7 into one line of at most 512 bytes\n" + "(about 70 words), the length of this ruler:\n" + strings.Repeat("-", 512) + "\n" + "\nuser: " + strings.Repeat("x", 600) + "\n", }, parts) } -func TestCompactionPartsEndWithTheRestAndTheTask(t *testing.T) { - testCases := []struct { - name string - context string - want []string - }{ - { - name: "empty view has the opening tag and the last part", - context: "\n", - want: []string{"\n", "\nTASK"}, - }, - { - name: "lines short of a block go into the last part", - context: "\n0+1|a\n1+1|b\n2+1|c\n", - want: []string{"\n", "0+1|a\n1+1|b\n2+1|c\n\nTASK"}, - }, - { - name: "one whole block", - context: "\n0+1|a\n1+1|b\n2+1|c\n3+1|d\n", - want: []string{"\n", "0+1|a\n1+1|b\n2+1|c\n3+1|d\n", "\nTASK"}, - }, - { - name: "two whole blocks and a rest", - context: "\n0+2|a\n2+1|b\n3+1|c\n4+1|d\n5+1|e\n6+1|f\n7+1|g\n8+1|h\n9+1|i\n", - want: []string{ - "\n", - "0+2|a\n2+1|b\n3+1|c\n4+1|d\n", - "5+1|e\n6+1|f\n7+1|g\n8+1|h\n", - "9+1|i\n\nTASK", - }, - }, - } - - for _, tc := range testCases { - t.Run(tc.name, func(t *testing.T) { - parts := compactionParts(tc.context, "TASK") - require.Equal(t, tc.want, parts) - require.Equal(t, tc.context+"\nTASK", strings.Join(parts, "")) - }) - } -} - -func TestCompactionPartsOfALongerViewBeginWithTheBlocksOfAShorterOne(t *testing.T) { - shorter := compactionParts("\n0+1|a\n1+1|b\n2+1|c\n3+1|d\n4+1|e\n", "task 5") - longer := compactionParts("\n0+1|a\n1+1|b\n2+1|c\n3+1|d\n4+1|e\n5+1|f\n6+1|g\n7+1|h\n", "task 8") - require.Len(t, shorter, 3) - require.Equal(t, shorter[:2], longer[:2]) - require.NotEqual(t, shorter[2], longer[2]) -} - func TestCompactorWritesTheMergeTask(t *testing.T) { text, err := compactionTask(Task{ Level: 2, @@ -119,9 +67,9 @@ func TestCompactorWritesTheMergeTask(t *testing.T) { func TestCompactorRetriesALongLineWithTheNewestNote(t *testing.T) { long := []string{strings.Repeat("a", 600), strings.Repeat("b", 550), strings.Repeat("c", 400)} - server := newFakeResponses(t) - server.reply = func(w http.ResponseWriter, call fakeCall) { - writeEvents(w, completedEvent([]any{messageItem(long[call.number])}, 100, 0, 5)) + server := newModelFake(t) + server.reply = func(w http.ResponseWriter, call ModelFakeCall) { + modelWriteEvents(w, modelCompletedEvent([]any{modelMessageItem(long[call.number])}, 100, 0, 5)) } compactor := newTestCompactor(t, server) @@ -130,7 +78,7 @@ func TestCompactorRetriesALongLineWithTheNewestNote(t *testing.T) { require.Equal(t, long[2], line) require.Equal(t, 3, server.Calls()) - records := readUsage(t, compactor.client.usage.path) + records := modelReadUsage(t, compactor.client.usage.path) require.Len(t, records, 3) for n, record := range records { require.Equal(t, &[2]int{0, 0}, record.Node) @@ -158,9 +106,9 @@ func TestCompactorRetriesALongLineWithTheNewestNote(t *testing.T) { func TestCompactorKeepsTheShortestOfFiveLines(t *testing.T) { sizes := []int{700, 530, 600, 520, 560, 513} - server := newFakeResponses(t) - server.reply = func(w http.ResponseWriter, call fakeCall) { - writeEvents(w, completedEvent([]any{messageItem(strings.Repeat("z", sizes[call.number]))}, 100, 0, 5)) + server := newModelFake(t) + server.reply = func(w http.ResponseWriter, call ModelFakeCall) { + modelWriteEvents(w, modelCompletedEvent([]any{modelMessageItem(strings.Repeat("z", sizes[call.number]))}, 100, 0, 5)) } line, err := newTestCompactor(t, server).Summarize(t.Context(), leafTask()) @@ -172,12 +120,12 @@ func TestCompactorKeepsTheShortestOfFiveLines(t *testing.T) { func TestCompactorFailsWithoutALine(t *testing.T) { testCases := []struct { name string - reply func(http.ResponseWriter, fakeCall) + reply func(http.ResponseWriter, ModelFakeCall) want string }{ { name: "failed call", - reply: func(w http.ResponseWriter, call fakeCall) { + reply: func(w http.ResponseWriter, call ModelFakeCall) { w.WriteHeader(http.StatusInternalServerError) _, _ = io.WriteString(w, "down") }, @@ -185,8 +133,8 @@ func TestCompactorFailsWithoutALine(t *testing.T) { }, { name: "tool call and no text", - reply: func(w http.ResponseWriter, call fakeCall) { - writeEvents(w, completedEvent([]any{map[string]any{"type": "function_call", "name": "zoom", "arguments": "{}"}}, 100, 0, 5)) + reply: func(w http.ResponseWriter, call ModelFakeCall) { + modelWriteEvents(w, modelCompletedEvent([]any{map[string]any{"type": "function_call", "name": "zoom", "arguments": "{}"}}, 100, 0, 5)) }, want: "no line", }, @@ -194,7 +142,7 @@ func TestCompactorFailsWithoutALine(t *testing.T) { for _, tc := range testCases { t.Run(tc.name, func(t *testing.T) { - server := newFakeResponses(t) + server := newModelFake(t) server.reply = tc.reply _, err := newTestCompactor(t, server).Summarize(t.Context(), leafTask()) @@ -213,14 +161,14 @@ func TestCutLineKeepsWholeCharacters(t *testing.T) { } func TestCompactorBuildsNodesInTheCore(t *testing.T) { - server := newFakeResponses(t) - server.reply = func(w http.ResponseWriter, call fakeCall) { + server := newModelFake(t) + server.reply = func(w http.ResponseWriter, call ModelFakeCall) { if call.number == 0 { w.WriteHeader(http.StatusServiceUnavailable) return } - writeEvents(w, completedEvent([]any{messageItem("summary of the long output")}, 100, 0, 5)) + modelWriteEvents(w, modelCompletedEvent([]any{modelMessageItem("summary of the long output")}, 100, 0, 5)) } memory, err := openMemory(t.Context(), t.TempDir(), newTestClock().Now, newTestCompactor(t, server)) @@ -245,17 +193,22 @@ func TestCompactorBuildsNodesInTheCore(t *testing.T) { require.Equal(t, 2, server.Calls()) } -func newTestCompactor(t *testing.T, server *fakeResponses) *Compactor { +func newTestCompactor(t *testing.T, server *ModelFake) *Compactor { t.Helper() - client := newClient(server.URL, apiKey(testKey), "agent-key", filepath.Join(t.TempDir(), "usage.jsonl")) + client := newModelClient(server.URL, ModelAPIKey(modelTestKey), "agent-key", filepath.Join(t.TempDir(), "usage.jsonl")) - return newCompactor(client, "Ada", ModelConfig{ + return newCompactor(client, adaPrompt, ModelConfig{ Model: "gpt-test", Effort: "high", }) } +// adaPrompt returns the prompt of the agent Ada, without instructions. +func adaPrompt() (string, error) { + return prompt("Ada", ""), nil +} + func leafTask() Task { return Task{ Level: 0, @@ -266,7 +219,7 @@ func leafTask() Task { } // inputParts returns the texts of the one user message of call. -func inputParts(t *testing.T, call fakeCall) []string { +func inputParts(t *testing.T, call ModelFakeCall) []string { t.Helper() var body struct {