X-Git-Url: https://git.giacomolli.uk/onechat/blobdiff_plain/522d7fd117285e4e4ce3ea5ec9b5a1722521b681..b2fc2d6a737dd5bab25e57ba822dd07258de93e6:/rebuild_test.go?ds=inline

diff --git a/rebuild_test.go b/rebuild_test.go
index 8d3c9d4..104ec88 100644
--- a/rebuild_test.go
+++ b/rebuild_test.go
@@ -63,7 +63,7 @@ func TestRebuildMergesDatesAndKeepsAgentFiles(t *testing.T) {
 		},
 	}
 
-	backup, err := rebuild(t.Context(), agent, added, client, newCompactor(client, agent.Prompt, defaultConfig().Compactor), &report)
+	backup, err := rebuild(t.Context(), agent, added, client, newCompactor(client, agent.Prompt, defaultConfig().Compactor, agent.Tools(), agent.Images()), &report)
 	require.NoError(t, err)
 	require.Equal(t, before, rebuildTestSnapshot(t, backup))
 
@@ -143,7 +143,7 @@ func TestRebuildFailureAndCancellationKeepOriginal(t *testing.T) {
 				Kind: "work",
 				Text: strings.Repeat("long ", 200),
 				Date: "2025-01-01T00:00:00Z",
-			}}, client, newCompactor(client, agent.Prompt, defaultConfig().Compactor), &strings.Builder{})
+			}}, client, newCompactor(client, agent.Prompt, defaultConfig().Compactor, agent.Tools(), agent.Images()), &strings.Builder{})
 			require.Error(t, err)
 			require.Empty(t, backup)
 			require.Equal(t, before, rebuildTestSnapshot(t, agent.dir))
@@ -179,7 +179,7 @@ func TestRebuildKeepsOriginalWhenAnInvisibleMergeFails(t *testing.T) {
 	}
 
 	client := newModelClient(server.URL, ModelAPIKey(modelTestKey), agent.CacheKey(), filepath.Join(t.TempDir(), "usage.jsonl"))
-	_, err := rebuild(t.Context(), agent, nil, client, newCompactor(client, agent.Prompt, defaultConfig().Compactor), &strings.Builder{})
+	_, err := rebuild(t.Context(), agent, nil, client, newCompactor(client, agent.Prompt, defaultConfig().Compactor, agent.Tools(), agent.Images()), &strings.Builder{})
 	require.ErrorContains(t, err, "summary tree has unfinished nodes")
 	require.Equal(t, before, rebuildTestSnapshot(t, agent.dir))
 }