zlei6/ragflow
1
0
Fork 0
mirror of https://github.com/infiniflow/ragflow.git synced 2026-08-15 21:16:03 +08:00
Code Issues Packages Projects Releases Wiki Activity

Go: fix context (#18118)

Browse Source 
Signed-off-by: Jin Hai <haijin.chn@gmail.com>
This commit is contained in:
Jin Hai
2026-08-11 19:19:29 +08:00
committed by GitHub
parent 8bd5768ebc
commit c75edbfbe8
36 changed files with 492 additions and 352 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
internal/agent/component/parallel_test.go
View File

@@ -34,7 +34,6 @@
package component
import (
"context"
"slices"
"testing"
)
@@ -75,7 +74,9 @@ func TestParallel_InvokeIsNoOp(t *testing.T) {
ItemsRef: "sys.arr",
MaxConcurrency: 3,
})
out, err := c.Invoke(context.Background(), nil, map[string]any{"in": 1})
ctx := t.Context()
out, err := c.Invoke(ctx, nil, map[string]any{"in": 1})
if err != nil {
t.Fatalf("Invoke: %v", err)
}
@@ -88,7 +89,9 @@ func TestParallel_InvokeIsNoOp(t *testing.T) {
// empty-map chunk and closes.
func TestParallel_StreamMirrorsInvoke(t *testing.T) {
c := NewParallelComponent(ParallelParam{})
ch, err := c.Stream(context.Background(), nil, nil)
ctx := t.Context()
ch, err := c.Stream(ctx, nil, nil)
if err != nil {
t.Fatalf("Stream: %v", err)
}