X-Git-Url: https://git.giacomolli.uk/onechat/blobdiff_plain/eb05edcf5d575e8a3b1040debfb57423312e8320..016d46008fc9726d2617e4e3ea098113f34c7636:/chain_test.go?ds=sidebyside

diff --git a/chain_test.go b/chain_test.go
index 833f86e..e329c27 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",
@@ -55,6 +56,7 @@ func TestChainReturnsTheRepliesOfTheFinalAnswer(t *testing.T) {
 	require.Equal(t, 2, server.Calls())
 
 	var kinds []string
+
 	for id := range memory.Len() {
 		message, err := memory.Message(id)
 		require.NoError(t, err)