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

feat(go): implement memory extraction task consumer (#17404)

Browse Source
This commit is contained in:
euvre
2026-07-31 11:23:58 +08:00
committed by GitHub
parent 569a29500f
commit ed05f32fee
8 changed files with 592 additions and 104 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
cmd/ragflow_server.go
View File

@@ -560,6 +560,13 @@ func runIngestor(ctx context.Context, cancel context.CancelFunc, args *serverArg
common.Error("Failed to initialize ingestor", err)
}
// Memory extraction consumer: drains task_type="memory" messages
// from the te.0.common Redis stream and runs LLM extraction.
memoryConsumerCtx, stopMemoryConsumer := context.WithCancel(ctx)
defer stopMemoryConsumer()
memoryMessageSvc := service.NewMemoryMessageService(service.NewMemoryService())
go memoryMessageSvc.StartTaskConsumer(memoryConsumerCtx)
common.Info("\n ____ __ _\n" +
" / _/___ ____ ____ _____/ /_(_)___ ____ ________ ______ _____ _____\n" +
" / // __ \\/ __ `/ _ \\/ ___/ __/ / __ \\/ __ \\ / ___/ _ \\/ ___/ | / / _ \\/ ___/\n" +