Carry new images through model request chains

[onechat] / chain_test.go

diff --git a/chain_test.go b/chain_test.go
index 833f86e7a73e883a12bc814b43d95a42c4be7437..e329c27eeb9f8d3a39caebd31d07b27c2c826485 100644 (file)
--- a/chain_test.go
+++ b/chain_test.go
@@ -26,7 +26,7 @@ func TestChainReturnsTheRepliesOfTheFinalAnswer(t *testing.T) {
                }, 120, 100, 5))
        }
 
                }, 120, 100, 5))
        }
 

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

        require.NoError(t, err)
 
        t.Cleanup(func() {
        require.NoError(t, err)
 
        t.Cleanup(func() {


@@ -36,9 +36,10 @@ func TestChainReturnsTheRepliesOfTheFinalAnswer(t *testing.T) {
        chain := &Chain{
                client: newModelClient(server.URL, ModelAPIKey(modelTestKey), "agent-key", filepath.Join(t.TempDir(), "usage.jsonl")),
                tools: &Toolbox{
        chain := &Chain{
                client: newModelClient(server.URL, ModelAPIKey(modelTestKey), "agent-key", filepath.Join(t.TempDir(), "usage.jsonl")),
                tools: &Toolbox{

-                       log:     memory,
-                       chat:    memory,
-                       workdir: t.TempDir(),
+                       log:       memory,
+                       chat:      memory,
+                       subagents: newTestSubagents(t),
+                       workdir:   t.TempDir(),

                },
                request: ModelRequest{
                        Model:  "gpt-test",
                },
                request: ModelRequest{
                        Model:  "gpt-test",


@@ -55,6 +56,7 @@ func TestChainReturnsTheRepliesOfTheFinalAnswer(t *testing.T) {
        require.Equal(t, 2, server.Calls())
 
        var kinds []string
        require.Equal(t, 2, server.Calls())
 
        var kinds []string

+

        for id := range memory.Len() {
                message, err := memory.Message(id)
                require.NoError(t, err)
        for id := range memory.Len() {
                message, err := memory.Message(id)
                require.NoError(t, err)