@yun520-1/heartflow 0.15.2

package/dist/agent/runtime/PlanExecutor.js

@@ -0,0 +1,322 @@
+/**
+ * PlanExecutor — 自然语言计划解析为任务图
+ * @version v0.1.0
+ *
+ * 功能：
+ *   - 将自然语言计划解析为 TaskGraph（TaskGraph.ts）
+ *   - 集成 CognitionEngine（HeartFlow.think）进行计划分析
+ *   - 支持 DAG 拓扑调度（TaskScheduler.ts）
+ *   - 自动检查点保存
+ *
+ * 使用示例:
+ *   const executor = new PlanExecutor({ cognition });
+ *   const graph = await executor.parse('帮我搜索论文然后总结重点');
+ *   const result = await executor.execute(graph);
+ */
+'use strict';
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.PlanExecutor = void 0;
+exports.createPlanExecutor = createPlanExecutor;
+const TaskGraph_1 = require("./TaskGraph");
+const TaskScheduler_1 = require("./TaskScheduler");
+// ─── PlanExecutor ─────────────────────────────────────────────────────────
+/**
+ * PlanExecutor
+ *
+ * 将自然语言计划转换为可执行的 DAG 任务图，并使用 TaskScheduler 驱动执行。
+ */
+class PlanExecutor {
+    cognition;
+    checkpointEngine;
+    maxConcurrency;
+    checkpointInterval;
+    maxRetries;
+    constructor(options) {
+        this.cognition = options.cognition;
+        this.checkpointEngine = options.checkpointEngine;
+        this.maxConcurrency = options.maxConcurrency ?? Infinity;
+        this.checkpointInterval = options.checkpointInterval ?? 5;
+        this.maxRetries = options.maxRetries ?? 3;
+    }
+    // ─── 公开 API ─────────────────────────────────────────────────────────
+    /**
+     * parse — 将自然语言计划解析为 TaskGraph
+     *
+     * @param planText  自然语言计划描述
+     * @param context   可选上下文（用于 Cognition 分析）
+     * @returns ParseResult
+     */
+    async parse(planText, context) {
+        const start = Date.now();
+        try {
+            // 1. 调用 CognitionEngine 分析计划
+            const cogResult = await this.cognition(planText, context);
+            // 2. 提取关键信息
+            const reasoning = this._extractReasoning(cogResult);
+            // 3. 从 Cognition 结果中推断任务步骤
+            const steps = this._inferSteps(planText, cogResult);
+            if (steps.length === 0) {
+                return {
+                    success: false,
+                    steps: [],
+                    graph: new TaskGraph_1.TaskGraph({ name: 'EmptyPlan' }),
+                    reasoning,
+                    error: '无法从计划中解析出任何可执行步骤',
+                };
+            }
+            // 4. 构建 TaskGraph
+            const graph = this._buildGraph(steps);
+            // 5. 验证拓扑
+            const topo = graph.getTopologicalOrder();
+            if (topo.hasCycle) {
+                return {
+                    success: false,
+                    steps,
+                    graph,
+                    reasoning,
+                    error: `检测到循环依赖: ${(topo.cycleNodes ?? []).join(' → ')}`,
+                };
+            }
+            return {
+                success: true,
+                steps,
+                graph,
+                reasoning,
+            };
+        }
+        catch (err) {
+            const error = err instanceof Error ? err.message : String(err);
+            return {
+                success: false,
+                steps: [],
+                graph: new TaskGraph_1.TaskGraph({ name: 'ErrorPlan' }),
+                error: `解析失败: ${error}`,
+            };
+        }
+    }
+    /**
+     * execute — 执行已解析的 TaskGraph
+     *
+     * @param graph   parse() 返回的 TaskGraph
+     * @param events  可选调度器事件回调
+     * @returns ExecuteResult
+     */
+    async execute(graph, events) {
+        const start = Date.now();
+        const parseMs = 0; // 已由 parse 消耗
+        try {
+            // 构建默认执行器（用户可覆盖）
+            const executor = async (node) => {
+                // 默认：直接返回 payload
+                // 实际应用中这里会调用具体工具
+                return { executed: true, nodeId: node.id, payload: node.payload };
+            };
+            // 创建调度器
+            const scheduler = new TaskScheduler_1.TaskScheduler({
+                graph,
+                executor,
+                maxConcurrency: this.maxConcurrency,
+                checkpointInterval: this.checkpointInterval,
+                events: events ?? {},
+                autoStart: false, // 手动控制 run()
+            });
+            // 执行
+            await scheduler.run();
+            const executeMs = Date.now() - start;
+            return {
+                success: scheduler.getStatus() === 'completed',
+                graph,
+                scheduler,
+                parseMs,
+                executeMs,
+                totalMs: Date.now() - start,
+            };
+        }
+        catch (err) {
+            const error = err instanceof Error ? err.message : String(err);
+            return {
+                success: false,
+                graph,
+                scheduler: null,
+                parseMs,
+                executeMs: Date.now() - start,
+                totalMs: Date.now() - start,
+                error: `执行失败: ${error}`,
+            };
+        }
+    }
+    /**
+     * parseAndExecute — 解析并执行（原子操作）
+     */
+    async parseAndExecute(planText, context, events) {
+        const start = Date.now();
+        const parseResult = await this.parse(planText, context);
+        if (!parseResult.success) {
+            return {
+                success: false,
+                graph: parseResult.graph,
+                scheduler: null,
+                parseMs: Date.now() - start,
+                executeMs: 0,
+                totalMs: Date.now() - start,
+                error: parseResult.error,
+            };
+        }
+        const executeResult = await this.execute(parseResult.graph, events);
+        return {
+            ...executeResult,
+            parseMs: Date.now() - start - executeResult.executeMs,
+        };
+    }
+    // ─── 私有工具 ─────────────────────────────────────────────────────────
+    /**
+     * 从 Cognition 结果中提取分析文本
+     */
+    _extractReasoning(cogResult) {
+        const parts = [];
+        if (cogResult.psychology?.intention) {
+            parts.push(`意图: ${cogResult.psychology.intention}`);
+        }
+        if (cogResult.psychology?.emotion) {
+            parts.push(`情绪: ${cogResult.psychology.emotion}`);
+        }
+        if (cogResult.truthCheck && !cogResult.truthCheck.pass) {
+            parts.push(`真善美问题: ${(cogResult.truthCheck.issues ?? []).join(', ')}`);
+        }
+        return parts.length > 0 ? parts.join(' | ') : '计划分析完成';
+    }
+    /**
+     * 从自然语言计划 + Cognition 结果推断任务步骤
+     */
+    _inferSteps(planText, cogResult) {
+        const steps = [];
+        const text = planText.toLowerCase();
+        const stepIdPrefix = `step-${Date.now()}-`;
+        let stepIndex = 0;
+        // 动作模式识别
+        const patterns = [
+            {
+                regex: /搜索?|查(找|询)|search/i,
+                action: 'search',
+                labelFn: (m) => `搜索: ${m[0] || '相关内容'}`,
+                payloadFn: () => ({}),
+                tags: ['search'],
+            },
+            {
+                regex: /阅读?|读取?|read|看(看|一下)/i,
+                action: 'read',
+                labelFn: (m) => `阅读: ${m[0]}`,
+                payloadFn: () => ({}),
+                tags: ['read'],
+            },
+            {
+                regex: /总结?|概括|提炼|summarize|汇总/i,
+                action: 'summarize',
+                labelFn: (m) => `总结: ${m[0]}`,
+                payloadFn: () => ({}),
+                tags: ['summarize'],
+            },
+            {
+                regex: /分析|研究|analyze/i,
+                action: 'analyze',
+                labelFn: (m) => `分析: ${m[0]}`,
+                payloadFn: () => ({}),
+                tags: ['analyze'],
+            },
+            {
+                regex: /写(入|入)?|保存|存储|save|write/i,
+                action: 'write',
+                labelFn: (m) => `写入: ${m[0]}`,
+                payloadFn: () => ({}),
+                tags: ['write'],
+            },
+            {
+                regex: /执行|运行|run|execute|做(一下)?|搞/i,
+                action: 'execute',
+                labelFn: (m) => `执行: ${m[0]}`,
+                payloadFn: () => ({}),
+                tags: ['execute'],
+            },
+        ];
+        // 简单顺序解析：按顺序匹配动作模式
+        // 实际应用中这里应该用 LLM 来分解计划
+        const sentences = planText.split(/[，,。;；\n]/).filter(Boolean);
+        let lastMatchedIdx = -1;
+        for (const sentence of sentences) {
+            for (const pattern of patterns) {
+                const match = sentence.match(pattern.regex);
+                if (match) {
+                    const stepId = `${stepIdPrefix}${stepIndex++}`;
+                    const prevDeps = lastMatchedIdx >= 0 ? [`${stepIdPrefix}${lastMatchedIdx}`] : [];
+                    steps.push({
+                        id: stepId,
+                        label: pattern.labelFn(match),
+                        action: pattern.action,
+                        intent: sentence.trim(),
+                        payload: pattern.payloadFn(match, sentence),
+                        deps: prevDeps,
+                        tags: pattern.tags,
+                    });
+                    lastMatchedIdx = steps.length - 1;
+                    break;
+                }
+            }
+        }
+        // Fallback: 如果没有匹配到任何动作，创建单个执行步骤
+        if (steps.length === 0) {
+            steps.push({
+                id: `${stepIdPrefix}0`,
+                label: `执行计划`,
+                action: 'execute',
+                intent: planText,
+                payload: { planText },
+                deps: [],
+                tags: ['execute'],
+            });
+        }
+        return steps;
+    }
+    /**
+     * 构建 TaskGraph
+     */
+    _buildGraph(steps) {
+        const graph = new TaskGraph_1.TaskGraph({
+            name: 'PlanExecution',
+            maxRetries: this.maxRetries,
+            checkpointEngine: this.checkpointEngine,
+        });
+        for (const step of steps) {
+            graph.addNode(step.id, step.label, step.payload, step.deps);
+        }
+        return graph;
+    }
+}
+exports.PlanExecutor = PlanExecutor;
+// ─── 工厂函数 ─────────────────────────────────────────────────────────────
+/**
+ * 创建 PlanExecutor（使用内置 Mock Cognition）
+ */
+function createPlanExecutor(options) {
+    // 默认 mock cognition
+    const defaultCognition = async (input) => {
+        return {
+            psychology: {
+                intention: 'request_action',
+                emotion: 'neutral',
+                needs: 'unknown',
+                defense: [],
+            },
+            truthCheck: { pass: true, issues: [] },
+            latency: 0,
+        };
+    };
+    return new PlanExecutor({
+        cognition: options?.cognition ?? defaultCognition,
+        checkpointEngine: options?.checkpointEngine,
+        maxConcurrency: options?.maxConcurrency,
+        checkpointInterval: options?.checkpointInterval,
+        maxRetries: options?.maxRetries,
+    });
+}
+exports.default = PlanExecutor;
+//# sourceMappingURL=PlanExecutor.js.map

package/dist/agent/runtime/PlanExecutor.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"PlanExecutor.js","sourceRoot":"","sources":["../../../src/agent/runtime/PlanExecutor.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,YAAY,CAAC;;;AA+Yb,gDAsBC;AAnaD,2CAA8D;AAC9D,mDAA+E;AAyE/E,6EAA6E;AAE7E;;;;GAIG;AACH,MAAa,YAAY;IACf,SAAS,CAAgE;IACzE,gBAAgB,CAAwE;IACxF,cAAc,CAAS;IACvB,kBAAkB,CAAS;IAC3B,UAAU,CAAS;IAE3B,YAAY,OAA4B;QACtC,IAAI,CAAC,SAAS,GAAG,OAAO,CAAC,SAAS,CAAC;QACnC,IAAI,CAAC,gBAAgB,GAAG,OAAO,CAAC,gBAAgB,CAAC;QACjD,IAAI,CAAC,cAAc,GAAG,OAAO,CAAC,cAAc,IAAI,QAAQ,CAAC;QACzD,IAAI,CAAC,kBAAkB,GAAG,OAAO,CAAC,kBAAkB,IAAI,CAAC,CAAC;QAC1D,IAAI,CAAC,UAAU,GAAG,OAAO,CAAC,UAAU,IAAI,CAAC,CAAC;IAC5C,CAAC;IAED,uEAAuE;IAEvE;;;;;;OAMG;IACH,KAAK,CAAC,KAAK,CAAC,QAAgB,EAAE,OAAgB;QAC5C,MAAM,KAAK,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAEzB,IAAI,CAAC;YACH,6BAA6B;YAC7B,MAAM,SAAS,GAAG,MAAM,IAAI,CAAC,SAAS,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;YAE1D,YAAY;YACZ,MAAM,SAAS,GAAG,IAAI,CAAC,iBAAiB,CAAC,SAAS,CAAC,CAAC;YAEpD,2BAA2B;YAC3B,MAAM,KAAK,GAAG,IAAI,CAAC,WAAW,CAAC,QAAQ,EAAE,SAAS,CAAC,CAAC;YAEpD,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBACvB,OAAO;oBACL,OAAO,EAAE,KAAK;oBACd,KAAK,EAAE,EAAE;oBACT,KAAK,EAAE,IAAI,qBAAS,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,CAAC;oBAC3C,SAAS;oBACT,KAAK,EAAE,kBAAkB;iBAC1B,CAAC;YACJ,CAAC;YAED,kBAAkB;YAClB,MAAM,KAAK,GAAG,IAAI,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC;YAEtC,UAAU;YACV,MAAM,IAAI,GAAG,KAAK,CAAC,mBAAmB,EAAE,CAAC;YACzC,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAC;gBAClB,OAAO;oBACL,OAAO,EAAE,KAAK;oBACd,KAAK;oBACL,KAAK;oBACL,SAAS;oBACT,KAAK,EAAE,YAAY,CAAC,IAAI,CAAC,UAAU,IAAI,EAAE,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE;iBACzD,CAAC;YACJ,CAAC;YAED,OAAO;gBACL,OAAO,EAAE,IAAI;gBACb,KAAK;gBACL,KAAK;gBACL,SAAS;aACV,CAAC;QACJ,CAAC;QAAC,OAAO,GAAY,EAAE,CAAC;YACtB,MAAM,KAAK,GAAG,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;YAC/D,OAAO;gBACL,OAAO,EAAE,KAAK;gBACd,KAAK,EAAE,EAAE;gBACT,KAAK,EAAE,IAAI,qBAAS,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,CAAC;gBAC3C,KAAK,EAAE,SAAS,KAAK,EAAE;aACxB,CAAC;QACJ,CAAC;IACH,CAAC;IAED;;;;;;OAMG;IACH,KAAK,CAAC,OAAO,CACX,KAAgB,EAChB,MAAwB;QAExB,MAAM,KAAK,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QACzB,MAAM,OAAO,GAAG,CAAC,CAAC,CAAC,cAAc;QAEjC,IAAI,CAAC;YACH,iBAAiB;YACjB,MAAM,QAAQ,GAAiB,KAAK,EAAE,IAAc,EAAE,EAAE;gBACtD,kBAAkB;gBAClB,iBAAiB;gBACjB,OAAO,EAAE,QAAQ,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,CAAC,EAAE,EAAE,OAAO,EAAE,IAAI,CAAC,OAAO,EAAE,CAAC;YACpE,CAAC,CAAC;YAEF,QAAQ;YACR,MAAM,SAAS,GAAG,IAAI,6BAAa,CAAC;gBAClC,KAAK;gBACL,QAAQ;gBACR,cAAc,EAAE,IAAI,CAAC,cAAc;gBACnC,kBAAkB,EAAE,IAAI,CAAC,kBAAkB;gBAC3C,MAAM,EAAE,MAAM,IAAI,EAAE;gBACpB,SAAS,EAAE,KAAK,EAAE,aAAa;aAChC,CAAC,CAAC;YAEH,KAAK;YACL,MAAM,SAAS,CAAC,GAAG,EAAE,CAAC;YAEtB,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,KAAK,CAAC;YACrC,OAAO;gBACL,OAAO,EAAE,SAAS,CAAC,SAAS,EAAE,KAAK,WAAW;gBAC9C,KAAK;gBACL,SAAS;gBACT,OAAO;gBACP,SAAS;gBACT,OAAO,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,KAAK;aAC5B,CAAC;QACJ,CAAC;QAAC,OAAO,GAAY,EAAE,CAAC;YACtB,MAAM,KAAK,GAAG,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;YAC/D,OAAO;gBACL,OAAO,EAAE,KAAK;gBACd,KAAK;gBACL,SAAS,EAAE,IAAI;gBACf,OAAO;gBACP,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,KAAK;gBAC7B,OAAO,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,KAAK;gBAC3B,KAAK,EAAE,SAAS,KAAK,EAAE;aACxB,CAAC;QACJ,CAAC;IACH,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,eAAe,CACnB,QAAgB,EAChB,OAAgB,EAChB,MAAwB;QAExB,MAAM,KAAK,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAEzB,MAAM,WAAW,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;QACxD,IAAI,CAAC,WAAW,CAAC,OAAO,EAAE,CAAC;YACzB,OAAO;gBACL,OAAO,EAAE,KAAK;gBACd,KAAK,EAAE,WAAW,CAAC,KAAK;gBACxB,SAAS,EAAE,IAAI;gBACf,OAAO,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,KAAK;gBAC3B,SAAS,EAAE,CAAC;gBACZ,OAAO,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,KAAK;gBAC3B,KAAK,EAAE,WAAW,CAAC,KAAK;aACzB,CAAC;QACJ,CAAC;QAED,MAAM,aAAa,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;QACpE,OAAO;YACL,GAAG,aAAa;YAChB,OAAO,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,KAAK,GAAG,aAAa,CAAC,SAAS;SACtD,CAAC;IACJ,CAAC;IAED,qEAAqE;IAErE;;OAEG;IACK,iBAAiB,CAAC,SAA0B;QAClD,MAAM,KAAK,GAAa,EAAE,CAAC;QAE3B,IAAI,SAAS,CAAC,UAAU,EAAE,SAAS,EAAE,CAAC;YACpC,KAAK,CAAC,IAAI,CAAC,OAAO,SAAS,CAAC,UAAU,CAAC,SAAS,EAAE,CAAC,CAAC;QACtD,CAAC;QACD,IAAI,SAAS,CAAC,UAAU,EAAE,OAAO,EAAE,CAAC;YAClC,KAAK,CAAC,IAAI,CAAC,OAAO,SAAS,CAAC,UAAU,CAAC,OAAO,EAAE,CAAC,CAAC;QACpD,CAAC;QACD,IAAI,SAAS,CAAC,UAAU,IAAI,CAAC,SAAS,CAAC,UAAU,CAAC,IAAI,EAAE,CAAC;YACvD,KAAK,CAAC,IAAI,CAAC,UAAU,CAAC,SAAS,CAAC,UAAU,CAAC,MAAM,IAAI,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QACzE,CAAC;QAED,OAAO,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC;IACzD,CAAC;IAED;;OAEG;IACK,WAAW,CAAC,QAAgB,EAAE,SAA0B;QAC9D,MAAM,KAAK,GAAiB,EAAE,CAAC;QAC/B,MAAM,IAAI,GAAG,QAAQ,CAAC,WAAW,EAAE,CAAC;QACpC,MAAM,YAAY,GAAG,QAAQ,IAAI,CAAC,GAAG,EAAE,GAAG,CAAC;QAC3C,IAAI,SAAS,GAAG,CAAC,CAAC;QAElB,SAAS;QACT,MAAM,QAAQ,GAMT;YACH;gBACE,KAAK,EAAE,oBAAoB;gBAC3B,MAAM,EAAE,QAAQ;gBAChB,OAAO,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,MAAM,EAAE;gBACvC,SAAS,EAAE,GAAG,EAAE,CAAC,CAAC,EAAE,CAAC;gBACrB,IAAI,EAAE,CAAC,QAAQ,CAAC;aACjB;YACD;gBACE,KAAK,EAAE,uBAAuB;gBAC9B,MAAM,EAAE,MAAM;gBACd,OAAO,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,EAAE;gBAC7B,SAAS,EAAE,GAAG,EAAE,CAAC,CAAC,EAAE,CAAC;gBACrB,IAAI,EAAE,CAAC,MAAM,CAAC;aACf;YACD;gBACE,KAAK,EAAE,yBAAyB;gBAChC,MAAM,EAAE,WAAW;gBACnB,OAAO,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,EAAE;gBAC7B,SAAS,EAAE,GAAG,EAAE,CAAC,CAAC,EAAE,CAAC;gBACrB,IAAI,EAAE,CAAC,WAAW,CAAC;aACpB;YACD;gBACE,KAAK,EAAE,gBAAgB;gBACvB,MAAM,EAAE,SAAS;gBACjB,OAAO,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,EAAE;gBAC7B,SAAS,EAAE,GAAG,EAAE,CAAC,CAAC,EAAE,CAAC;gBACrB,IAAI,EAAE,CAAC,SAAS,CAAC;aAClB;YACD;gBACE,KAAK,EAAE,2BAA2B;gBAClC,MAAM,EAAE,OAAO;gBACf,OAAO,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,EAAE;gBAC7B,SAAS,EAAE,GAAG,EAAE,CAAC,CAAC,EAAE,CAAC;gBACrB,IAAI,EAAE,CAAC,OAAO,CAAC;aAChB;YACD;gBACE,KAAK,EAAE,6BAA6B;gBACpC,MAAM,EAAE,SAAS;gBACjB,OAAO,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,EAAE;gBAC7B,SAAS,EAAE,GAAG,EAAE,CAAC,CAAC,EAAE,CAAC;gBACrB,IAAI,EAAE,CAAC,SAAS,CAAC;aAClB;SACF,CAAC;QAEF,mBAAmB;QACnB,uBAAuB;QACvB,MAAM,SAAS,GAAG,QAAQ,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;QAC9D,IAAI,cAAc,GAAG,CAAC,CAAC,CAAC;QAExB,KAAK,MAAM,QAAQ,IAAI,SAAS,EAAE,CAAC;YACjC,KAAK,MAAM,OAAO,IAAI,QAAQ,EAAE,CAAC;gBAC/B,MAAM,KAAK,GAAG,QAAQ,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;gBAC5C,IAAI,KAAK,EAAE,CAAC;oBACV,MAAM,MAAM,GAAG,GAAG,YAAY,GAAG,SAAS,EAAE,EAAE,CAAC;oBAC/C,MAAM,QAAQ,GAAG,cAAc,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,YAAY,GAAG,cAAc,EAAE,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;oBAEjF,KAAK,CAAC,IAAI,CAAC;wBACT,EAAE,EAAE,MAAM;wBACV,KAAK,EAAE,OAAO,CAAC,OAAO,CAAC,KAAK,CAAC;wBAC7B,MAAM,EAAE,OAAO,CAAC,MAAM;wBACtB,MAAM,EAAE,QAAQ,CAAC,IAAI,EAAE;wBACvB,OAAO,EAAE,OAAO,CAAC,SAAS,CAAC,KAAK,EAAE,QAAQ,CAAC;wBAC3C,IAAI,EAAE,QAAQ;wBACd,IAAI,EAAE,OAAO,CAAC,IAAI;qBACnB,CAAC,CAAC;oBAEH,cAAc,GAAG,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC;oBAClC,MAAM;gBACR,CAAC;YACH,CAAC;QACH,CAAC;QAED,iCAAiC;QACjC,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACvB,KAAK,CAAC,IAAI,CAAC;gBACT,EAAE,EAAE,GAAG,YAAY,GAAG;gBACtB,KAAK,EAAE,MAAM;gBACb,MAAM,EAAE,SAAS;gBACjB,MAAM,EAAE,QAAQ;gBAChB,OAAO,EAAE,EAAE,QAAQ,EAAE;gBACrB,IAAI,EAAE,EAAE;gBACR,IAAI,EAAE,CAAC,SAAS,CAAC;aAClB,CAAC,CAAC;QACL,CAAC;QAED,OAAO,KAAK,CAAC;IACf,CAAC;IAED;;OAEG;IACK,WAAW,CAAC,KAAmB;QACrC,MAAM,KAAK,GAAG,IAAI,qBAAS,CAAC;YAC1B,IAAI,EAAE,eAAe;YACrB,UAAU,EAAE,IAAI,CAAC,UAAU;YAC3B,gBAAgB,EAAE,IAAI,CAAC,gBAAgB;SACxC,CAAC,CAAC;QAEH,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;YACzB,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,EAAE,IAAI,CAAC,KAAK,EAAE,IAAI,CAAC,OAAO,EAAE,IAAI,CAAC,IAAI,CAAC,CAAC;QAC9D,CAAC;QAED,OAAO,KAAK,CAAC;IACf,CAAC;CACF;AArTD,oCAqTC;AAED,yEAAyE;AAEzE;;GAEG;AACH,SAAgB,kBAAkB,CAAC,OAAsC;IACvE,oBAAoB;IACpB,MAAM,gBAAgB,GAAkE,KAAK,EAAE,KAAK,EAAE,EAAE;QACtG,OAAO;YACL,UAAU,EAAE;gBACV,SAAS,EAAE,gBAAgB;gBAC3B,OAAO,EAAE,SAAS;gBAClB,KAAK,EAAE,SAAS;gBAChB,OAAO,EAAE,EAAE;aACZ;YACD,UAAU,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,MAAM,EAAE,EAAE,EAAE;YACtC,OAAO,EAAE,CAAC;SACX,CAAC;IACJ,CAAC,CAAC;IAEF,OAAO,IAAI,YAAY,CAAC;QACtB,SAAS,EAAE,OAAO,EAAE,SAAS,IAAI,gBAAgB;QACjD,gBAAgB,EAAE,OAAO,EAAE,gBAAgB;QAC3C,cAAc,EAAE,OAAO,EAAE,cAAc;QACvC,kBAAkB,EAAE,OAAO,EAAE,kBAAkB;QAC/C,UAAU,EAAE,OAAO,EAAE,UAAU;KAChC,CAAC,CAAC;AACL,CAAC;AAED,kBAAe,YAAY,CAAC"}

package/dist/agent/runtime/TaskGraph.d.ts

@@ -0,0 +1,170 @@
+/**
+ * TaskGraph — DAG (Directed Acyclic Graph) 任务图
+ * @version v0.1.0
+ *
+ * 实现：
+ *   - 节点: TaskNode { id, label, payload, status, deps, createdAt }
+ *   - 边: 依赖关系 (must run after)
+ *   - 拓扑排序: Kahn's algorithm
+ *   - 环检测: DFS coloring
+ *   - 状态持久化: toJSON() / fromJSON()
+ */
+import { CheckpointEngine } from '../../storage/checkpoint/CheckpointEngine';
+/** 任务节点状态 */
+export type TaskNodeStatus = 'pending' | 'ready' | 'running' | 'completed' | 'failed' | 'skipped';
+/** 单个任务节点 */
+export interface TaskNode {
+    id: string;
+    label: string;
+    payload: unknown;
+    status: TaskNodeStatus;
+    deps: string[];
+    result?: unknown;
+    error?: string;
+    createdAt: number;
+    startedAt?: number;
+    completedAt?: number;
+    retries: number;
+    maxRetries: number;
+    tags?: string[];
+}
+/** 任务图完整状态 */
+export interface TaskGraphState {
+    id: string;
+    name: string;
+    nodes: Record<string, TaskNode>;
+    createdAt: number;
+    updatedAt: number;
+    status: 'running' | 'completed' | 'failed' | 'aborted';
+}
+/** Kahn's algorithm 拓扑排序结果 */
+export interface TopoResult {
+    order: string[];
+    hasCycle: boolean;
+    cycleNodes?: string[];
+}
+/** TaskGraph 配置 */
+export interface TaskGraphOptions {
+    id?: string;
+    name?: string;
+    maxRetries?: number;
+    checkpointEngine?: CheckpointEngine;
+}
+/**
+ * TaskGraph — 任务有向无环图
+ *
+ * 使用示例:
+ *   const graph = new TaskGraph({ name: 'data-pipeline' });
+ *   graph.addNode('fetch', { label: '获取数据' }, ['clean']);
+ *   graph.addNode('clean', { label: '清洗数据' });
+ *   graph.build();
+ *   const order = graph.getTopologicalOrder(); // ['clean', 'fetch']
+ */
+export declare class TaskGraph {
+    readonly id: string;
+    readonly name: string;
+    readonly maxRetries: number;
+    private checkpointEngine?;
+    private _nodes;
+    private _adjList;
+    private _inDegree;
+    private _status;
+    private _createdAt;
+    private _updatedAt;
+    private _dirty;
+    constructor(options?: TaskGraphOptions);
+    /**
+     * 添加任务节点
+     * @param id 唯一标识
+     * @param label 显示名称
+     * @param payload 任务输入
+     * @param deps 依赖的节点 ID (这些节点必须先完成)
+     */
+    addNode(id: string, label: string, payload: unknown, deps?: string[]): this;
+    /**
+     * 批量添加节点
+     */
+    addNodes(nodes: Array<{
+        id: string;
+        label: string;
+        payload: unknown;
+        deps?: string[];
+    }>): this;
+    /**
+     * 获取节点
+     */
+    getNode(id: string): TaskNode | undefined;
+    /**
+     * 获取所有节点
+     */
+    getNodes(): TaskNode[];
+    /**
+     * 获取所有就绪节点 (所有依赖已完成，状态为 ready)
+     */
+    getReadyNodes(): TaskNode[];
+    /**
+     * Kahn's algorithm 拓扑排序
+     * 返回可以并行执行的层级
+     */
+    getTopologicalOrder(): TopoResult;
+    /**
+     * 按层级分组 (同一层可并行)
+     * 返回数组，每层包含可并行执行的节点 ID
+     */
+    getExecutionLayers(): string[][];
+    /**
+     * 环检测 (DFS 三色标记)
+     */
+    detectCycle(): {
+        hasCycle: boolean;
+        cycleNodes?: string[];
+    };
+    /**
+     * 更新节点状态
+     */
+    updateNodeStatus(id: string, status: TaskNodeStatus, result?: unknown, error?: string): void;
+    /**
+     * 重试节点
+     */
+    retryNode(id: string): boolean;
+    private _propagateFailure;
+    private _markDirty;
+    /**
+     * 保存状态到检查点引擎
+     */
+    checkpoint(): Promise<{
+        success: boolean;
+        id?: string;
+        error?: string;
+    }>;
+    /**
+     * 从检查点恢复
+     */
+    static fromCheckpoint(engine: CheckpointEngine, graphId: string, options?: Partial<TaskGraphOptions>): Promise<TaskGraph | null>;
+    /**
+     * 序列化为 JSON
+     */
+    toJSON(): TaskGraphState;
+    /**
+     * 从 JSON 恢复
+     */
+    static fromJSON(data: TaskGraphState, options?: Partial<TaskGraphOptions>): TaskGraph;
+    isDirty(): boolean;
+    markClean(): void;
+    getStatus(): TaskGraphState['status'];
+    getUpdatedAt(): number;
+    /**
+     * 获取图的大致大小
+     */
+    size(): number;
+    /**
+     * 打印拓扑顺序 (调试用)
+     */
+    printOrder(): string;
+    /**
+     * 打印状态摘要
+     */
+    toSummaryString(): string;
+}
+export default TaskGraph;
+//# sourceMappingURL=TaskGraph.d.ts.map

package/dist/agent/runtime/TaskGraph.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"TaskGraph.d.ts","sourceRoot":"","sources":["../../../src/agent/runtime/TaskGraph.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;GAUG;AAIH,OAAO,EAAE,gBAAgB,EAAE,MAAM,2CAA2C,CAAC;AAE7E,aAAa;AACb,MAAM,MAAM,cAAc,GACtB,SAAS,GACT,OAAO,GACP,SAAS,GACT,WAAW,GACX,QAAQ,GACR,SAAS,CAAC;AAEd,aAAa;AACb,MAAM,WAAW,QAAQ;IACvB,EAAE,EAAE,MAAM,CAAC;IACX,KAAK,EAAE,MAAM,CAAC;IACd,OAAO,EAAE,OAAO,CAAC;IACjB,MAAM,EAAE,cAAc,CAAC;IACvB,IAAI,EAAE,MAAM,EAAE,CAAC;IACf,MAAM,CAAC,EAAE,OAAO,CAAC;IACjB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,SAAS,EAAE,MAAM,CAAC;IAClB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,OAAO,EAAE,MAAM,CAAC;IAChB,UAAU,EAAE,MAAM,CAAC;IACnB,IAAI,CAAC,EAAE,MAAM,EAAE,CAAC;CACjB;AAED,cAAc;AACd,MAAM,WAAW,cAAc;IAC7B,EAAE,EAAE,MAAM,CAAC;IACX,IAAI,EAAE,MAAM,CAAC;IACb,KAAK,EAAE,MAAM,CAAC,MAAM,EAAE,QAAQ,CAAC,CAAC;IAChC,SAAS,EAAE,MAAM,CAAC;IAClB,SAAS,EAAE,MAAM,CAAC;IAClB,MAAM,EAAE,SAAS,GAAG,WAAW,GAAG,QAAQ,GAAG,SAAS,CAAC;CACxD;AAED,8BAA8B;AAC9B,MAAM,WAAW,UAAU;IACzB,KAAK,EAAE,MAAM,EAAE,CAAC;IAChB,QAAQ,EAAE,OAAO,CAAC;IAClB,UAAU,CAAC,EAAE,MAAM,EAAE,CAAC;CACvB;AAED,mBAAmB;AACnB,MAAM,WAAW,gBAAgB;IAC/B,EAAE,CAAC,EAAE,MAAM,CAAC;IACZ,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,gBAAgB,CAAC,EAAE,gBAAgB,CAAC;CACrC;AAED;;;;;;;;;GASG;AACH,qBAAa,SAAS;IACpB,QAAQ,CAAC,EAAE,EAAE,MAAM,CAAC;IACpB,QAAQ,CAAC,IAAI,EAAE,MAAM,CAAC;IACtB,QAAQ,CAAC,UAAU,EAAE,MAAM,CAAC;IAC5B,OAAO,CAAC,gBAAgB,CAAC,CAAmB;IAE5C,OAAO,CAAC,MAAM,CAAoC;IAClD,OAAO,CAAC,QAAQ,CAAuC;IACvD,OAAO,CAAC,SAAS,CAAkC;IAEnD,OAAO,CAAC,OAAO,CAAuC;IACtD,OAAO,CAAC,UAAU,CAAS;IAC3B,OAAO,CAAC,UAAU,CAAS;IAC3B,OAAO,CAAC,MAAM,CAAS;gBAEX,OAAO,GAAE,gBAAqB;IAW1C;;;;;;OAMG;IACH,OAAO,CAAC,EAAE,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,EAAE,IAAI,GAAE,MAAM,EAAO,GAAG,IAAI;IAuC/E;;OAEG;IACH,QAAQ,CAAC,KAAK,EAAE,KAAK,CAAC;QAAE,EAAE,EAAE,MAAM,CAAC;QAAC,KAAK,EAAE,MAAM,CAAC;QAAC,OAAO,EAAE,OAAO,CAAC;QAAC,IAAI,CAAC,EAAE,MAAM,EAAE,CAAA;KAAE,CAAC,GAAG,IAAI;IAO9F;;OAEG;IACH,OAAO,CAAC,EAAE,EAAE,MAAM,GAAG,QAAQ,GAAG,SAAS;IAIzC;;OAEG;IACH,QAAQ,IAAI,QAAQ,EAAE;IAItB;;OAEG;IACH,aAAa,IAAI,QAAQ,EAAE;IAY3B;;;OAGG;IACH,mBAAmB,IAAI,UAAU;IAqCjC;;;OAGG;IACH,kBAAkB,IAAI,MAAM,EAAE,EAAE;IAmChC;;OAEG;IACH,WAAW,IAAI;QAAE,QAAQ,EAAE,OAAO,CAAC;QAAC,UAAU,CAAC,EAAE,MAAM,EAAE,CAAA;KAAE;IAyC3D;;OAEG;IACH,gBAAgB,CAAC,EAAE,EAAE,MAAM,EAAE,MAAM,EAAE,cAAc,EAAE,MAAM,CAAC,EAAE,OAAO,EAAE,KAAK,CAAC,EAAE,MAAM,GAAG,IAAI;IAqB5F;;OAEG;IACH,SAAS,CAAC,EAAE,EAAE,MAAM,GAAG,OAAO;IAY9B,OAAO,CAAC,iBAAiB;IAqBzB,OAAO,CAAC,UAAU;IAMlB;;OAEG;IACG,UAAU,IAAI,OAAO,CAAC;QAAE,OAAO,EAAE,OAAO,CAAC;QAAC,EAAE,CAAC,EAAE,MAAM,CAAC;QAAC,KAAK,CAAC,EAAE,MAAM,CAAA;KAAE,CAAC;IAQ9E;;OAEG;WACU,cAAc,CACzB,MAAM,EAAE,gBAAgB,EACxB,OAAO,EAAE,MAAM,EACf,OAAO,CAAC,EAAE,OAAO,CAAC,gBAAgB,CAAC,GAClC,OAAO,CAAC,SAAS,GAAG,IAAI,CAAC;IAM5B;;OAEG;IACH,MAAM,IAAI,cAAc;IAexB;;OAEG;IACH,MAAM,CAAC,QAAQ,CAAC,IAAI,EAAE,cAAc,EAAE,OAAO,CAAC,EAAE,OAAO,CAAC,gBAAgB,CAAC,GAAG,SAAS;IAgCrF,OAAO,IAAI,OAAO;IAClB,SAAS,IAAI,IAAI;IACjB,SAAS,IAAI,cAAc,CAAC,QAAQ,CAAC;IACrC,YAAY,IAAI,MAAM;IAEtB;;OAEG;IACH,IAAI,IAAI,MAAM;IAEd;;OAEG;IACH,UAAU,IAAI,MAAM;IAMpB;;OAEG;IACH,eAAe,IAAI,MAAM;CAO1B;AAED,eAAe,SAAS,CAAC"}