feat: 实时入库监控 + 向量拆分可视化(异步入库 + 进度 SSE)
入库从同步改为异步流水线 + 进度回流(复用 token 流 NATS streaming)。 UI 实时看到 解析→切块→向量化(分批)→写入 各阶段 + 拆分块预览。 - shared: contract.IngestEvent(stage/done/total/chunks/error) - mcp-go: rag.Ingest 加 onProgress + 分批向量化(10/批)逐批回报;kb_ingest 带 job_id 把进度发到 sundynix.streams.<job_id> + CompleteStream - gateway: 入库异步返回 job_id,后台 runIngest 发进度;GET /kb/ingest/:id/stream SSE - frontend: streamIngest(EventSource);KbView 实时进度面板(阶段徽标+进度条+拆分列表) - 验证: build✓+e2e PASS; 浏览器 12 行→6 阶段点亮+进度条 12/12+拆分 12 块逐条 Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
This commit is contained in:
@@ -160,14 +160,32 @@ func (g *Gateway) kbSearch(ctx context.Context, call *contract.ToolCall) *contra
|
||||
return &contract.ToolResult{OK: true, Content: string(data)}
|
||||
}
|
||||
|
||||
// kbIngest 把文本入库(切块→embedding→Milvus)。
|
||||
// kbIngest 把文本入库(切块→embedding→Milvus+Bleve)。
|
||||
// 带 job_id 时逐阶段把进度发到 sundynix.streams.<job_id>,供 UI 实时入库监控。
|
||||
func (g *Gateway) kbIngest(ctx context.Context, call *contract.ToolCall) *contract.ToolResult {
|
||||
kb, _ := call.Args["kb"].(string)
|
||||
text, _ := call.Args["text"].(string)
|
||||
jobID, _ := call.Args["job_id"].(string)
|
||||
if text == "" {
|
||||
return &contract.ToolResult{OK: false, Error: "kb_ingest: text 必填"}
|
||||
}
|
||||
n, err := g.rag.Ingest(ctx, kb, text)
|
||||
var onProgress func(contract.IngestEvent)
|
||||
if jobID != "" {
|
||||
onProgress = func(ev contract.IngestEvent) {
|
||||
if data, err := json.Marshal(ev); err == nil {
|
||||
_ = g.bus.PublishToken(jobID, data)
|
||||
}
|
||||
}
|
||||
}
|
||||
n, err := g.rag.Ingest(ctx, kb, text, onProgress)
|
||||
if jobID != "" {
|
||||
if err != nil {
|
||||
onProgress(contract.IngestEvent{Stage: "失败", Error: err.Error()})
|
||||
} else {
|
||||
onProgress(contract.IngestEvent{Stage: "完成", Done: n, Total: n, Msg: fmt.Sprintf("已入库 %d 块", n)})
|
||||
}
|
||||
_ = g.bus.CompleteStream(jobID)
|
||||
}
|
||||
if err != nil {
|
||||
return &contract.ToolResult{OK: false, Error: "kb_ingest: " + err.Error()}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user