feat(orchestration): Phase2 —— map 真并行 fan-out + branch 真/假边标签精确选路

Phase1 让引擎按图执行后,本轮补上两块:

1) map 并行 fan-out(dispatcher)
   - map 节点:planItems 把主题拆成 3–6 子项 → 复用 report 的 writeSections
     有界并发(4)逐项撰写 → 结构化 sections 存黑板 + 拼进 answer + 流式呈现。
   - 检索节点记下 owner 作用域库名(b.kb),供 map 各项并行检索复用。
   - render 节点优先用 map 产出的多章 sections 渲染,否则整段成稿当单章。

2) branch 真/假边标签(前端 + DSL + dispatcher)
   - TypedNode:分支节点渲染两个出口手柄 真(绿)/假(红),连线各带 sourceHandle。
   - exportDsl / TaskDsl:边导出携带 sourceHandle。
   - dispatcher dsl.Edge 增 SourceHandle;branchNode 优先按 true/false 标签精确
     选路,无标签的旧图退回"出边顺序"约定,向后兼容。

实测(gateway+dispatcher+DeepSeek 真跑):
- map:input→map→render,DeepSeek 拆出 5 章并行撰写(347–512字),trace 见
  section:0..4 并发 + 有界并发(section3 等槽);render 因 mcp-go 不在优雅降级 ✓
- branch 标签:把 true 边故意列第二位,条件真仍走 true 标签的分支A、假走分支B,
  证明按标签而非边序选路 ✓
- 桌面端:分支节点正确渲染 真/假 两手柄,无 console 报错 ✓

Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
This commit is contained in:
Blizzard
2026-06-15 14:01:51 +08:00
parent fd145b5852
commit 1bd187874d
5 changed files with 140 additions and 31 deletions
+7 -2
View File
@@ -5,7 +5,8 @@ import { NODE_KINDS } from "../studio/nodeCatalog";
export interface TaskDsl { export interface TaskDsl {
version: "1"; version: "1";
nodes: Array<{ id: string; kind: string; label?: string; config: unknown }>; nodes: Array<{ id: string; kind: string; label?: string; config: unknown }>;
edges: Array<{ source: string; target: string }>; // sourceHandle 标记从 branch 节点引出的边走真/假分支("true"/"false")。
edges: Array<{ source: string; target: string; sourceHandle?: string }>;
} }
// exportDsl 把画布的节点/连线导出为类型化 JSON DSL。 // exportDsl 把画布的节点/连线导出为类型化 JSON DSL。
@@ -16,7 +17,11 @@ export function exportDsl(nodes: Node[], edges: Edge[]): TaskDsl {
const d = n.data as { kind: string; label?: string; config?: unknown }; const d = n.data as { kind: string; label?: string; config?: unknown };
return { id: n.id, kind: d.kind, label: d.label, config: d.config ?? {} }; return { id: n.id, kind: d.kind, label: d.label, config: d.config ?? {} };
}), }),
edges: edges.map((e) => ({ source: e.source, target: e.target })), edges: edges.map((e) => ({
source: e.source,
target: e.target,
...(e.sourceHandle ? { sourceHandle: e.sourceHandle } : {}),
})),
}; };
} }
@@ -19,7 +19,7 @@ export function TypedNode({ data, selected }: NodeProps) {
return ( return (
<div <div
className={`min-w-[170px] rounded-lg border border-l-[3px] bg-ink-800 shadow-card ${k.accent} ${ className={`relative min-w-[170px] rounded-lg border border-l-[3px] bg-ink-800 shadow-card ${k.accent} ${
selected ? "ring-2 ring-violet-500/70" : "border-line" selected ? "ring-2 ring-violet-500/70" : "border-line"
}`} }`}
> >
@@ -32,7 +32,17 @@ export function TypedNode({ data, selected }: NodeProps) {
<div className="text-xs font-medium text-slate-200">{d.label || k.desc}</div> <div className="text-xs font-medium text-slate-200">{d.label || k.desc}</div>
{d.summary && <div className="mt-0.5 truncate text-[10px] text-slate-500">{d.summary}</div>} {d.summary && <div className="mt-0.5 truncate text-[10px] text-slate-500">{d.summary}</div>}
</div> </div>
<Handle type="source" position={Position.Right} className="!h-2 !w-2 !border-0 !bg-slate-500" /> {d.kind === "branch" ? (
// 分支节点:两个出口手柄 —— 真(绿)/假(红),连线时各自带 sourceHandle,引擎据此精确选路。
<>
<Handle id="true" type="source" position={Position.Right} style={{ top: "38%" }} className="!h-2.5 !w-2.5 !border-0 !bg-emerald-500" />
<span className="pointer-events-none absolute right-3 top-[32%] text-[9px] font-medium text-emerald-400"></span>
<Handle id="false" type="source" position={Position.Right} style={{ top: "72%" }} className="!h-2.5 !w-2.5 !border-0 !bg-rose-500" />
<span className="pointer-events-none absolute right-3 top-[66%] text-[9px] font-medium text-rose-400"></span>
</>
) : (
<Handle type="source" position={Position.Right} className="!h-2 !w-2 !border-0 !bg-slate-500" />
)}
</div> </div>
); );
} }
+4 -3
View File
@@ -17,10 +17,11 @@ type Node struct {
Config map[string]any `json:"config"` Config map[string]any `json:"config"`
} }
// Edge 是一条连线。 // Edge 是一条连线。SourceHandle 标记从 branch 节点引出的边走真/假分支("true"/"false")。
type Edge struct { type Edge struct {
Source string `json:"source"` Source string `json:"source"`
Target string `json:"target"` Target string `json:"target"`
SourceHandle string `json:"sourceHandle"`
} }
// Flow 是整张图。 // Flow 是整张图。
+88 -24
View File
@@ -22,9 +22,11 @@ type board struct {
query string query string
profile string profile string
history []*schema.Message history []*schema.Message
refs []string // 检索 / 聚合得到的参考资料 kb string // 最近一个检索节点的 owner 作用域库名(供 map 并行各项检索)
toolOut []string // 工具节点产出 refs []string // 检索 / 聚合得到的参考资料
answer string // 终端 agent 的成稿(流式累计) toolOut []string // 工具节点产出
sections []reportSection // map 并行 fan-out 产出的分项成稿(供 render 多章渲染)
answer string // 终端 agent / map 的成稿(流式累计)
} }
// runGraph 按 DSL 图的真实拓扑与连线执行(替代旧的线性拍平 compileFlow)。 // runGraph 按 DSL 图的真实拓扑与连线执行(替代旧的线性拍平 compileFlow)。
@@ -51,9 +53,9 @@ func (o *Orchestrator) runGraph(ctx context.Context, t *contract.Task, tr *execT
return b.answer, nil return b.answer, nil
} }
// 建邻接与入度(只认两端都存在的边)。 // 建邻接与入度(只认两端都存在的边)。保留整条边以便 branch 按 true/false 标签选路。
nodeByID := make(map[string]dsl.Node, len(flow.Nodes)) nodeByID := make(map[string]dsl.Node, len(flow.Nodes))
out := make(map[string][]string) outE := make(map[string][]dsl.Edge)
indeg := make(map[string]int, len(flow.Nodes)) indeg := make(map[string]int, len(flow.Nodes))
for _, n := range flow.Nodes { for _, n := range flow.Nodes {
nodeByID[n.ID] = n nodeByID[n.ID] = n
@@ -66,7 +68,7 @@ func (o *Orchestrator) runGraph(ctx context.Context, t *contract.Task, tr *execT
if _, ok := nodeByID[e.Target]; !ok { if _, ok := nodeByID[e.Target]; !ok {
continue continue
} }
out[e.Source] = append(out[e.Source], e.Target) outE[e.Source] = append(outE[e.Source], e)
indeg[e.Target]++ indeg[e.Target]++
} }
@@ -95,7 +97,7 @@ func (o *Orchestrator) runGraph(ctx context.Context, t *contract.Task, tr *execT
if !active[n.ID] { if !active[n.ID] {
continue // 被 branch 剪掉的下游,不执行 continue // 被 branch 剪掉的下游,不执行
} }
propagate := out[n.ID] // 默认激活全部出边;branch 会改写 propagate := targetsOf(outE[n.ID]) // 默认激活全部出边;branch 会改写
switch n.Kind { switch n.Kind {
case "input": case "input":
if txt := cstr(n.Config, "text"); txt != "" { if txt := cstr(n.Config, "text"); txt != "" {
@@ -124,9 +126,9 @@ func (o *Orchestrator) runGraph(ctx context.Context, t *contract.Task, tr *execT
case "render": case "render":
o.renderNode(ctx, t.ID, n, b, tr) o.renderNode(ctx, t.ID, n, b, tr)
case "branch": case "branch":
propagate = o.branchNode(n, b, out[n.ID], nodeByID, tr) propagate = o.branchNode(n, b, outE[n.ID], nodeByID, tr)
case "map": case "map":
tr.info("map:"+n.ID, "system", labelOf(n, "并行"), "fan-out 暂按串行执行(路线图 Phase 2") o.mapNode(ctx, t.ID, n, b, tr)
case "output": case "output":
tr.info("output:"+n.ID, "system", labelOf(n, "输出"), "目标:"+firstNonEmpty(cstr(n.Config, "target"), "屏幕")) tr.info("output:"+n.ID, "system", labelOf(n, "输出"), "目标:"+firstNonEmpty(cstr(n.Config, "target"), "屏幕"))
default: default:
@@ -151,6 +153,9 @@ func (o *Orchestrator) retrieverNode(ctx context.Context, n dsl.Node, b *board,
if b.uid != "" && kb != "" && !strings.Contains(kb, "/") { if b.uid != "" && kb != "" && !strings.Contains(kb, "/") {
scoped = b.uid + "/" + kb scoped = b.uid + "/" + kb
} }
if scoped != "" {
b.kb = scoped // 记下作用域库名,供后续 map 并行各项检索复用
}
end := tr.span("retriever:"+n.ID, "tool", labelOf(n, "检索")) end := tr.span("retriever:"+n.ID, "tool", labelOf(n, "检索"))
refs := o.retrieve(ctx, scoped, b.query) refs := o.retrieve(ctx, scoped, b.query)
if refs != "" { if refs != "" {
@@ -159,6 +164,24 @@ func (o *Orchestrator) retrieverNode(ctx context.Context, n dsl.Node, b *board,
end(fmt.Sprintf("kb=%s · 命中 %d 段", firstNonEmpty(kb, "(未指定)"), countLines(refs)), nil) end(fmt.Sprintf("kb=%s · 命中 %d 段", firstNonEmpty(kb, "(未指定)"), countLines(refs)), nil)
} }
// mapNode 执行并行 fan-out:把主题拆成若干项,各项有界并发撰写(复用 report 的 writeSections),
// 成稿存入黑板(结构化 sections + 拼进 answer),并流式呈现进度。
func (o *Orchestrator) mapNode(ctx context.Context, taskID string, n dsl.Node, b *board, tr *execTracer) {
end := tr.span("map:"+n.ID, "plan", labelOf(n, "并行 fan-out"))
items := o.planItems(ctx, b.query, cstr(n.Config, "splitBy"))
end(fmt.Sprintf("拆出 %d 项:%s", len(items), strings.Join(items, " / ")), nil)
o.emit(taskID, fmt.Sprintf("\n> 并行处理 %d 项…\n\n", len(items)))
secs := o.writeSections(ctx, b.query, b.kb, items, tr) // 有界并发,trace 出 section:i 各项
b.sections = secs
for _, s := range secs {
chunk := "## " + s.Heading + "\n\n" + s.Body + "\n\n"
o.emit(taskID, chunk)
b.answer += chunk
b.refs = append(b.refs, s.Heading+""+s.Body)
}
}
// execToolNode 执行工具节点:调 MCP 工具,产出累计进黑板;失败降级不阻断。 // execToolNode 执行工具节点:调 MCP 工具,产出累计进黑板;失败降级不阻断。
func (o *Orchestrator) execToolNode(ctx context.Context, taskID string, n dsl.Node, b *board, tr *execTracer) { func (o *Orchestrator) execToolNode(ctx context.Context, taskID string, n dsl.Node, b *board, tr *execTracer) {
tool, args := dsl.ToolBinding(n) tool, args := dsl.ToolBinding(n)
@@ -242,7 +265,10 @@ func (o *Orchestrator) renderNode(ctx context.Context, taskID string, n dsl.Node
format := firstNonEmpty(cstr(n.Config, "format"), "docx") format := firstNonEmpty(cstr(n.Config, "format"), "docx")
end := tr.span("render:"+n.ID, "render", labelOf(n, "渲染 "+format)) end := tr.span("render:"+n.ID, "render", labelOf(n, "渲染 "+format))
title := truncate(b.query, 40) title := truncate(b.query, 40)
secs := []reportSection{{Heading: title, Body: b.answer}} secs := b.sections // map 产出的多章优先;否则把整段成稿当单章
if len(secs) == 0 {
secs = []reportSection{{Heading: title, Body: b.answer}}
}
if path := o.renderReport(ctx, taskID, title, secs); path != "" { if path := o.renderReport(ctx, taskID, title, secs); path != "" {
end("已落盘:"+path, nil) end("已落盘:"+path, nil)
o.emit(taskID, "\n\n---\n✅ 已渲染 "+format+" 文档,可在「下载」获取。\n") o.emit(taskID, "\n\n---\n✅ 已渲染 "+format+" 文档,可在「下载」获取。\n")
@@ -251,32 +277,70 @@ func (o *Orchestrator) renderNode(ctx context.Context, taskID string, n dsl.Node
} }
} }
// branchNode 执行分支节点:求值条件,按边序约定 [true, false] 选出要激活的下游。 // branchNode 执行分支节点:求值条件,按出边的 true/false 标签选出要激活的下游。
// 注:当前 DSL 的边不带 true/false 标签,故以"出边顺序"约定语义(Phase 2 将由前端给边打标) // 边带 sourceHandle("true"/"false") 时按标签精确选路(前端 Phase 2 给 branch 的边打标)
func (o *Orchestrator) branchNode(n dsl.Node, b *board, outs []string, byID map[string]dsl.Node, tr *execTracer) []string { // 无标签的旧图退回"出边顺序 [true,false]"约定,保持向后兼容。
func (o *Orchestrator) branchNode(n dsl.Node, b *board, outs []dsl.Edge, byID map[string]dsl.Node, tr *execTracer) []string {
cond := cstr(n.Config, "condition") cond := cstr(n.Config, "condition")
res := evalCondition(cond, b) res := evalCondition(cond, b)
chosen := outs
switch { var truthy, falsy []string
case len(outs) >= 2: labeled := false
if res { for _, e := range outs {
chosen = outs[:1] switch e.SourceHandle {
} else { case "true":
chosen = outs[1:2] truthy = append(truthy, e.Target)
labeled = true
case "false":
falsy = append(falsy, e.Target)
labeled = true
} }
case len(outs) == 1 && !res:
chosen = nil // 单出边且条件为假 → 不继续
} }
var chosen []string
mode := "边标签"
if labeled {
if res {
chosen = truthy
} else {
chosen = falsy
}
} else {
mode = "边序约定[true,false]"
targets := targetsOf(outs)
switch {
case len(targets) >= 2:
if res {
chosen = targets[:1]
} else {
chosen = targets[1:2]
}
case len(targets) == 1 && !res:
chosen = nil
default:
chosen = targets
}
}
names := make([]string, 0, len(chosen)) names := make([]string, 0, len(chosen))
for _, id := range chosen { for _, id := range chosen {
names = append(names, labelOf(byID[id], id)) names = append(names, labelOf(byID[id], id))
} }
tr.info("branch:"+n.ID, "system", labelOf(n, "分支"), tr.info("branch:"+n.ID, "system", labelOf(n, "分支"),
fmt.Sprintf("条件「%s」→ %v ⇒ 走 [%s]边序约定 [true,false]", fmt.Sprintf("条件「%s」→ %v ⇒ 走 [%s]%s",
firstNonEmpty(cond, "(空=真)"), res, strings.Join(names, ", "))) firstNonEmpty(cond, "(空=真)"), res, strings.Join(names, ", "), mode))
return chosen return chosen
} }
// targetsOf 取一组边的目标节点 ID(保持顺序)。
func targetsOf(edges []dsl.Edge) []string {
out := make([]string, 0, len(edges))
for _, e := range edges {
out = append(out, e.Target)
}
return out
}
// evalCondition 求值 branch 条件。支持: // evalCondition 求值 branch 条件。支持:
// //
// 空 → true;关键字 refs/tools/answer/profile 作左值(取数量/字数); // 空 → true;关键字 refs/tools/answer/profile 作左值(取数量/字数);
@@ -122,6 +122,35 @@ func (o *Orchestrator) planOutline(ctx context.Context, topic string) reportOutl
return out return out
} }
// planItems 为 map 节点把主题拆成一组并行子项(splitBy 为拆分依据提示)。
// 模型不可用/解析失败则用通用兜底分项。
func (o *Orchestrator) planItems(ctx context.Context, topic, splitBy string) []string {
fallback := []string{"背景与现状", "核心分析", "结论与建议"}
if !o.pool.Ready() {
return fallback
}
hint := splitBy
if hint == "" {
hint = "合理的章节"
}
user := fmt.Sprintf("请把主题《%s》拆分为一组「%s」,用于并行撰写。"+
"只输出 JSON 数组:[\"子项1\",\"子项2\", ...]3 到 6 项,不要任何多余文字。", topic, hint)
txt, err := o.pool.Chat(ctx, []llm.ChatMessage{
{Role: "system", Content: "你擅长把一个任务拆解为可并行处理的若干子项。"},
{Role: "user", Content: user},
})
if err != nil {
log.Printf("[map] 拆分子项失败,用兜底: %v", err)
return fallback
}
var items []string
if json.Unmarshal([]byte(stripFence(txt)), &items) != nil || len(items) == 0 {
log.Printf("[map] 子项 JSON 解析失败,用兜底。原文: %s", truncate(txt, 200))
return fallback
}
return items
}
// writeSections 各章节并行撰写(有界并发),结果按原顺序返回。 // writeSections 各章节并行撰写(有界并发),结果按原顺序返回。
func (o *Orchestrator) writeSections(ctx context.Context, topic, kb string, headings []string, tr *execTracer) []reportSection { func (o *Orchestrator) writeSections(ctx context.Context, topic, kb string, headings []string, tr *execTracer) []reportSection {
out := make([]reportSection, len(headings)) out := make([]reportSection, len(headings))