X-Git-Url: https://git.giacomolli.uk/onechat/blobdiff_plain/087d725b5ab41ad34ffd24938482d45a8b99a080..3738533260a8eb914e211b8d909e92a14c0c266c:/build_run_test.go

diff --git a/build_run_test.go b/build_run_test.go
index aef8922..b45745e 100644
--- a/build_run_test.go
+++ b/build_run_test.go
@@ -82,6 +82,7 @@ func TestBuildRunStopsWhenTheQuotaGrows(t *testing.T) {
 	require.Greater(t, len(records), int(buildQuotaLimit))
 
 	var report strings.Builder
+
 	require.NoError(t, buildWriteReport(&report, appended, stop, records, sizes))
 	require.Contains(t, report.String(), "stopped early: ")
 	require.Contains(t, report.String(), "cached (80.0%)")
@@ -111,11 +112,14 @@ func TestBuildRunStopsAtTheCallLimit(t *testing.T) {
 
 func TestBuildRunWaitsOnlyBeforeUserMessages(t *testing.T) {
 	gate := make(chan struct{})
+
 	var mutex sync.Mutex
+
 	running, most := 0, 0
 	server := newModelFake(t)
 	server.reply = func(w http.ResponseWriter, _ ModelFakeCall) {
 		mutex.Lock()
+
 		running++
 		most = max(most, running)
 		mutex.Unlock()
@@ -123,6 +127,7 @@ func TestBuildRunWaitsOnlyBeforeUserMessages(t *testing.T) {
 		<-gate
 
 		mutex.Lock()
+
 		running--
 		mutex.Unlock()
 
@@ -210,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), discardLog())
+	memory, err := openMemory(t.Context(), root, newTestClock().Now, newCompactor(client, probePrompt, defaultConfig().Compactor, promptTools()), discardLog())
 	require.NoError(t, err)
 
 	return &BuildRun{