Check operation context on returned errors

[onechat] / chain_test.go

diff --git a/chain_test.go b/chain_test.go
index 833f86e7a73e883a12bc814b43d95a42c4be7437..55176202b66063b1c9838589edb22041777957dc 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",