X-Git-Url: https://git.giacomolli.uk/onechat/blobdiff_plain/eb05edcf5d575e8a3b1040debfb57423312e8320..786c2b0551517d4befaa35be2c7743f65bd5f1c4:/chain_test.go

diff --git a/chain_test.go b/chain_test.go
index 833f86e..5517620 100644
--- a/chain_test.go
+++ b/chain_test.go
@@ -26,7 +26,7 @@ func TestChainReturnsTheRepliesOfTheFinalAnswer(t *testing.T) {
 		}, 120, 100, 5))
 	}
 
-	memory, err := openMemory(t.Context(), t.TempDir(), newTestClock().Now, newFakeSummarizer())
+	memory, err := openMemory(t.Context(), t.TempDir(), newTestClock().Now, newFakeSummarizer(), discardLog())
 	require.NoError(t, err)
 
 	t.Cleanup(func() {
@@ -36,9 +36,10 @@ func TestChainReturnsTheRepliesOfTheFinalAnswer(t *testing.T) {
 	chain := &Chain{
 		client: newModelClient(server.URL, ModelAPIKey(modelTestKey), "agent-key", filepath.Join(t.TempDir(), "usage.jsonl")),
 		tools: &Toolbox{
-			log:     memory,
-			chat:    memory,
-			workdir: t.TempDir(),
+			log:       memory,
+			chat:      memory,
+			subagents: newTestSubagents(t),
+			workdir:   t.TempDir(),
 		},
 		request: ModelRequest{
 			Model:  "gpt-test",