Say that the agent can change its instructions

[onechat] / turn_test.go

diff --git a/turn_test.go b/turn_test.go
index 3cb15b280687b891043f7f255a5346f935647f82..9bb0e995d5b66393d8c5b46fdeb2480e50e8a233 100644 (file)
--- a/turn_test.go
+++ b/turn_test.go
@@ -222,7 +222,7 @@ func TestFailedCallEndsTheTurnWithoutARetry(t *testing.T) {
        require.Equal(t, 1, server.Calls())
        require.Contains(t, test.events.Last("error"), "HTTP 503")
        require.Equal(t, 1, strings.Count(test.logs.String(), "\n"))
        require.Equal(t, 1, server.Calls())
        require.Contains(t, test.events.Last("error"), "HTTP 503")
        require.Equal(t, 1, strings.Count(test.logs.String(), "\n"))

-       require.Contains(t, test.logs.String(), "the turn stops before its end, and the agent waits for the next message: ")
+       require.Contains(t, test.logs.String(), "turn failed: ")

        require.Contains(t, test.logs.String(), "HTTP 503")
        require.NotContains(t, test.logs.String(), "hello")
 }
        require.Contains(t, test.logs.String(), "HTTP 503")
        require.NotContains(t, test.logs.String(), "hello")
 }


@@ -387,7 +387,7 @@ func newTestTurnLoopWith(t *testing.T, server *ModelFake, summarizer Summarizer)
 
        agent.config.Workdir = t.TempDir()
        clock := newTestClock()
 
        agent.config.Workdir = t.TempDir()
        clock := newTestClock()

-       memory, err := openMemory(t.Context(), t.TempDir(), clock.Now, summarizer)
+       memory, err := openMemory(t.Context(), t.TempDir(), clock.Now, summarizer, discardLog())

        require.NoError(t, err)
 
        test := &TurnTestLoop{
        require.NoError(t, err)
 
        test := &TurnTestLoop{