Talk to the agent through a Telegram bot

[onechat] / turn_test.go

diff --git a/turn_test.go b/turn_test.go
index f1295e938a54f25fc2b79ff3e8765c7cb360b43c..0de7dd7be85cf86c5498b977b221599fade3d033 100644 (file)
--- a/turn_test.go
+++ b/turn_test.go
@@ -30,7 +30,7 @@ func TestTurnRunsToolsUntilTheFinalAnswer(t *testing.T) {
        }
 
        test := newTestLoop(t, server)
-       _, err := test.loop.Submit("Run echo hi.")
+       _, err := test.loop.Submit(terminalSource, "Run echo hi.")
        require.NoError(t, err)
        test.events.WaitFor(t, "idle", 1)
        require.Equal(t, 2, server.Calls())
@@ -74,7 +74,7 @@ func TestFirstRequestHoldsTheViewTheStateAndTheMessage(t *testing.T) {
        _, err := test.memory.Append("user", "an earlier message")
        require.NoError(t, err)
 
-       _, err = test.loop.Submit("The new message.")
+       _, err = test.loop.Submit(terminalSource, "The new message.")
        require.NoError(t, err)
        test.events.WaitFor(t, "idle", 1)
 
@@ -102,11 +102,11 @@ func TestMessageDuringATurnJoinsTheNextRequest(t *testing.T) {
        }
 
        test := newTestLoop(t, server)
-       _, err := test.loop.Submit("first")
+       _, err := test.loop.Submit(terminalSource, "first")
        require.NoError(t, err)
 
        <-arrived
-       _, err = test.loop.Submit("second, during the turn")
+       _, err = test.loop.Submit(terminalSource, "second, during the turn")
        require.NoError(t, err)
        close(release)
 
@@ -136,11 +136,11 @@ func TestMessageDuringTheFinalRequestStartsANewTurn(t *testing.T) {
        }
 
        test := newTestLoop(t, server)
-       _, err := test.loop.Submit("first")
+       _, err := test.loop.Submit(terminalSource, "first")
        require.NoError(t, err)
 
        <-arrived
-       _, err = test.loop.Submit("second")
+       _, err = test.loop.Submit(terminalSource, "second")
        require.NoError(t, err)
        close(release)
 
@@ -161,7 +161,7 @@ func TestTurnWaitsForTheSummaries(t *testing.T) {
        _, err := test.memory.Append("echo", strings.Repeat("long output ", 100))
        require.NoError(t, err)
 
-       _, err = test.loop.Submit("go on")
+       _, err = test.loop.Submit(terminalSource, "go on")
        require.NoError(t, err)
 
        time.Sleep(50 * time.Millisecond)
@@ -187,7 +187,7 @@ func TestInterruptEndsTheTurn(t *testing.T) {
        }
 
        test := newTestLoop(t, server)
-       _, err := test.loop.Submit("a long task")
+       _, err := test.loop.Submit(terminalSource, "a long task")
        require.NoError(t, err)
 
        <-arrived
@@ -199,7 +199,7 @@ func TestInterruptEndsTheTurn(t *testing.T) {
        require.Equal(t, []string{"user: a long task"}, test.Log(t))
        require.False(t, test.loop.Interrupt())
 
-       _, err = test.loop.Submit("again")
+       _, err = test.loop.Submit(terminalSource, "again")
        require.NoError(t, err)
        test.events.WaitFor(t, "idle", 2)
        require.Equal(t, "reply: answer", test.Log(t)[2])
@@ -212,7 +212,7 @@ func TestFailedCallEndsTheTurnWithoutARetry(t *testing.T) {
        }
 
        test := newTestLoop(t, server)
-       _, err := test.loop.Submit("hello")
+       _, err := test.loop.Submit(terminalSource, "hello")
        require.NoError(t, err)
        test.events.WaitFor(t, "idle", 1)
 
@@ -225,7 +225,7 @@ func TestViewMessagesOnlyGrowBetweenTurns(t *testing.T) {
        server := newFakeResponses(t)
        test := newTestLoop(t, server)
        for n, text := range []string{"one", "two", "three"} {
-               _, err := test.loop.Submit(text)
+               _, err := test.loop.Submit(terminalSource, text)
                require.NoError(t, err)
                test.events.WaitFor(t, "idle", n+1)
        }