]> git.giacomolli.uk - onechat/blobdiff - compactor_test.go
Let the agent keep user instructions
[onechat] / compactor_test.go
index 0d591ee9e5602135ef113712e405bddbe3c11bb8..8bdf45c68d7fcd308a8ad7a173e0deb81908641c 100644 (file)
@@ -28,7 +28,7 @@ func TestCompactorSendsThePromptToolsAndTask(t *testing.T) {
        require.Equal(t, 1, server.Calls())
 
        body := server.Call(0).body
        require.Equal(t, 1, server.Calls())
 
        body := server.Call(0).body
-       require.Equal(t, prompt("Ada", ""), body["instructions"])
+       require.Equal(t, prompt("Ada", "/AGENT/instructions.md", ""), 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"])
        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"])
@@ -255,7 +255,7 @@ func TestCompactorBuildsNodesInTheCore(t *testing.T) {
                modelWriteEvents(w, modelCompletedEvent([]any{modelMessageItem("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))
+       memory, err := openMemory(t.Context(), t.TempDir(), newTestClock().Now, newTestCompactor(t, server), discardLog())
        require.NoError(t, err)
 
        defer func() {
        require.NoError(t, err)
 
        defer func() {
@@ -327,5 +327,5 @@ func compactorLeafTask() Task {
 
 // compactorAdaPrompt returns the prompt of the agent Ada, without instructions.
 func compactorAdaPrompt() (string, error) {
 
 // compactorAdaPrompt returns the prompt of the agent Ada, without instructions.
 func compactorAdaPrompt() (string, error) {
-       return prompt("Ada", ""), nil
+       return prompt("Ada", "/AGENT/instructions.md", ""), nil
 }
 }