Import WhatsApp history into a child

[onechat] / rebuild_test.go

diff --git a/rebuild_test.go b/rebuild_test.go
index 8d3c9d4e033b3f076f04a5062731ca76d515291e..fc4ff7e8b09dc3cfd23d46c221cc7e36fbce8fcd 100644 (file)
--- 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()), &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()), &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()), &strings.Builder{})
        require.ErrorContains(t, err, "summary tree has unfinished nodes")
        require.Equal(t, before, rebuildTestSnapshot(t, agent.dir))
 }