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

diff --git a/whatsapp_import.go b/whatsapp_import.go
index 81eb808..e1cbd02 100644
--- a/whatsapp_import.go
+++ b/whatsapp_import.go
@@ -142,7 +142,7 @@ func whatsappImport(ctx context.Context, args []string, out io.Writer) (err erro
 	}
 
 	client := newModelClient(address, auth, agent.CacheKey(), usagePath)
-	backup, err := rebuild(ctx, agent, plan.added, client, newCompactor(client, agent.Prompt, config.Compactor), out)
+	backup, err := rebuild(ctx, agent, plan.added, client, newCompactor(client, agent.Prompt, config.Compactor, agent.Tools()), out)
 	if err != nil {
 		return fmt.Errorf("rebuild imported memory: %w", err)
 	}
@@ -211,7 +211,7 @@ func (p WhatsAppImportPlan) Write(out io.Writer) error {
 		"owner: " + owner,
 		fmt.Sprintf("bytes: %d (new batch text)", p.bytes),
 		fmt.Sprintf("messages: %d (merged log)", p.messages),
-		fmt.Sprintf("estimated calls: %.0f (about 1.3 per message, limit %d)", estimate, buildCallLimit),
+		fmt.Sprintf("estimated calls: %.0f (about 1.3 per message, limit %d)", estimate, rebuildCallLimit),
 	}
 
 	_, err = io.WriteString(out, strings.Join(lines, "\n")+"\n")