X-Git-Url: https://git.giacomolli.uk/onechat/blobdiff_plain/0e57dcfc96cd95268d5dd00af7ce0594721d3ded..3cdbd017da9ae7a7045f4272f16d4b262fd01c79:/whatsapp_import_test.go?ds=sidebyside

diff --git a/whatsapp_import_test.go b/whatsapp_import_test.go
index 1f9c9af..4bd0a1f 100644
--- a/whatsapp_import_test.go
+++ b/whatsapp_import_test.go
@@ -563,7 +563,7 @@ func TestWhatsAppImportChildRebuildUsesOwnPromptAndCache(t *testing.T) {
 	}
 
 	client := newModelClient(server.URL, ModelAPIKey(modelTestKey), child.CacheKey(), filepath.Join(t.TempDir(), "usage.jsonl"))
-	backup, err := rebuild(t.Context(), child, plan.added, client, newCompactor(client, child.Prompt, defaultConfig().Compactor, child.Tools()), &strings.Builder{})
+	backup, err := rebuild(t.Context(), child, plan.added, client, newCompactor(client, child.Prompt, defaultConfig().Compactor, child.Tools(), child.Images()), &strings.Builder{})
 	require.NoError(t, err)
 	require.Equal(t, before, rebuildTestSnapshot(t, backup))
 	require.Equal(t, root.dir, filepath.Dir(backup))
@@ -613,7 +613,7 @@ func TestWhatsAppImportChildFailureKeepsConfiguredChildren(t *testing.T) {
 		Kind: "work",
 		Text: strings.Repeat("The plan is ready. ", 80),
 		Date: "2025-01-01T08:00:00Z",
-	}}, client, newCompactor(client, child.Prompt, defaultConfig().Compactor, child.Tools()), &strings.Builder{})
+	}}, client, newCompactor(client, child.Prompt, defaultConfig().Compactor, child.Tools(), child.Images()), &strings.Builder{})
 	require.ErrorContains(t, err, "rebuild")
 	require.Equal(t, before, rebuildTestSnapshot(t, root.dir))