zlei6/ragflow
1
0
Fork 0
mirror of https://github.com/infiniflow/ragflow.git synced 2026-07-26 18:33:30 +08:00
Code Issues Packages Projects Releases Wiki Activity

Go: add context (#17314)

Browse Source 
### Summary

As title.

---------

Signed-off-by: Jin Hai <haijin.chn@gmail.com>
This commit is contained in:
Jin Hai
2026-07-24 16:47:12 +08:00
committed by GitHub
parent 008fa3e10e
commit bdfc3ada41
78 changed files with 1031 additions and 904 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
internal/handler/mindmap.go
View File

@@ -42,7 +42,7 @@ type mindMapRunConfig struct {
TenantSvc *service.TenantService
}
func runMindMap(config mindMapRunConfig) (mindMapNode, error) {
func runMindMap(ctx context.Context, config mindMapRunConfig) (mindMapNode, error) {
if config.ChunkSvc == nil {
return mindMapNode{}, fmt.Errorf("chunk service not configured")
}
@@ -54,7 +54,7 @@ func runMindMap(config mindMapRunConfig) (mindMapNode, error) {
modelTenantID = config.AuthUserID
}
retrievalReq := mindMapRetrievalRequest(config.Question, config.KbIDs, config.SearchID, config.SearchConfig)
ranks, err := config.ChunkSvc.RetrievalTest(retrievalReq, config.AuthUserID)
ranks, err := config.ChunkSvc.RetrievalTest(ctx, retrievalReq, config.AuthUserID)
if err != nil {
return mindMapNode{}, err
}