Pass caller contexts through WhatsApp callbacks

[onechat] / build_run_test.go

diff --git a/build_run_test.go b/build_run_test.go
index b45745e7590b7ee8051c2f6f8a4df14208952a2e..ef9c3c0fb48c98582888f440854a32d5577e6463 100644 (file)
--- a/build_run_test.go
+++ b/build_run_test.go
@@ -215,7 +215,7 @@ func newTestBuildRun(t *testing.T, server *ModelFake) *BuildRun {
        root := t.TempDir()
        usagePath := filepath.Join(root, "usage.jsonl")
        client := newModelClient(server.URL, ModelAPIKey(modelTestKey), "onechat-"+probeName, usagePath)
-       memory, err := openMemory(t.Context(), root, newTestClock().Now, newCompactor(client, probePrompt, defaultConfig().Compactor, promptTools()), discardLog())
+       memory, err := openMemory(t.Context(), root, newTestClock().Now, newCompactor(client, probePrompt, defaultConfig().Compactor, promptTools(), &ImageStore{dir: t.TempDir()}), discardLog())
        require.NoError(t, err)
 
        return &BuildRun{