Give the root tools to start and message children

[onechat] / daemon_start_helper_test.go

diff --git a/daemon_start_helper_test.go b/daemon_start_helper_test.go
index 0c25acc01dff4b8fa93cddd9c43f7d799d3a4a6e..81399ce3ca3c3548d815bd52294723e58863d6c3 100644 (file)
--- a/daemon_start_helper_test.go
+++ b/daemon_start_helper_test.go
@@ -2,8 +2,6 @@ package main
 
 import (
        "context"
-       "io"
-       "log"
        "path/filepath"
        "testing"
        "time"
@@ -42,7 +40,7 @@ func daemonStartTestWithSchedules(t *testing.T, server *ModelFake, bot *Telegram
        require.NoError(t, err)
 
        agent.config.Workdir = t.TempDir()
-       memory, err := openMemory(t.Context(), agent.dir, clock, newFakeSummarizer())
+       memory, err := openMemory(t.Context(), agent.dir, clock, newFakeSummarizer(), discardLog())
        require.NoError(t, err)
 
        socket := filepath.Join(dir, socketFileName)
@@ -52,9 +50,10 @@ func daemonStartTestWithSchedules(t *testing.T, server *ModelFake, bot *Telegram
        client := newModelClient(server.URL, ModelAPIKey(modelTestKey), agent.CacheKey(), filepath.Join(agent.dir, "usage.jsonl"))
        ctx, cancel := context.WithCancel(t.Context())
        done := make(chan error)
+       subagents := newTestSubagents(t)
        go func() {
                done <- daemonServe(ctx, listener, bot, whatsapp, schedules, func(notify func(Event)) (*TurnLoop, error) {
-                       return newTurnLoop(agent, memory, newTestSubagents(t), whatsapp, schedules, client, clock, log.New(io.Discard, "", 0), notify)
+                       return newTurnLoop(agent, memory, subagents, whatsapp, schedules, client, clock, discardLog(), notify)
                })
        }()