klaus-agent 0.3.1 → 0.4.0

package/dist/task-graph/task-graph.js

@@ -0,0 +1,241 @@
+// Task graph — dependency-aware DAG with background execution and auto-unlock
+import { readFileSync, writeFileSync, mkdirSync, existsSync, renameSync } from "fs";
+import { join } from "path";
+import { generateId } from "../utils/id.js";
+export class TaskGraph {
+    _tasks = new Map();
+    _config;
+    _completedQueue = [];
+    _backgroundAborts = new Map();
+    constructor(config = {}) {
+        this._config = config;
+        if (config.persistDir) {
+            mkdirSync(config.persistDir, { recursive: true });
+            this._loadFromDisk();
+        }
+    }
+    get(id) {
+        const task = this._tasks.get(id);
+        return task ? { ...task } : undefined;
+    }
+    listAll() {
+        return [...this._tasks.values()];
+    }
+    /** Tasks that are pending with no unfinished blockers. */
+    listReady() {
+        return this.listAll().filter((t) => t.status === "pending" && t.blockedBy.length === 0);
+    }
+    /** Tasks waiting on unfinished blockers. */
+    listBlocked() {
+        return this.listAll().filter((t) => t.status === "pending" && t.blockedBy.length > 0);
+    }
+    create(subject, description = "") {
+        const max = this._config.maxTasks ?? 100;
+        if (this._tasks.size >= max) {
+            throw new Error(`Task limit reached: ${max}.`);
+        }
+        const node = {
+            id: generateId(),
+            subject,
+            description,
+            status: "pending",
+            blockedBy: [],
+            blocks: [],
+            owner: "",
+            createdAt: Date.now(),
+            updatedAt: Date.now(),
+        };
+        this._tasks.set(node.id, node);
+        this._persist();
+        return { ...node };
+    }
+    /** Add a dependency: `taskId` is blocked by `blockedById`. */
+    addDependency(taskId, blockedById) {
+        const task = this._require(taskId);
+        const blocker = this._require(blockedById);
+        if (task.blockedBy.includes(blockedById))
+            return;
+        // Cycle detection: if blocker is (transitively) blocked by task, adding this edge creates a cycle
+        if (this._isTransitivelyBlockedBy(blockedById, taskId)) {
+            throw new Error(`Adding dependency ${blockedById} → ${taskId} would create a cycle.`);
+        }
+        task.blockedBy.push(blockedById);
+        blocker.blocks.push(taskId);
+        task.updatedAt = Date.now();
+        blocker.updatedAt = Date.now();
+        this._persist();
+    }
+    update(taskId, fields) {
+        const task = this._require(taskId);
+        if (fields.status !== undefined && fields.status !== task.status) {
+            if (fields.status === "in_progress" && task.blockedBy.length > 0) {
+                throw new Error(`Task ${taskId} is blocked by: ${task.blockedBy.join(", ")}`);
+            }
+            task.status = fields.status;
+            if (fields.status === "completed" || fields.status === "failed") {
+                task.result = fields.result ?? task.result;
+                const unblocked = this._clearDependency(taskId);
+                this._completedQueue.push({
+                    taskId,
+                    subject: task.subject,
+                    result: task.result ?? "",
+                    status: fields.status,
+                    unblockedTasks: unblocked,
+                });
+            }
+        }
+        if (fields.owner !== undefined)
+            task.owner = fields.owner;
+        // result is already set inside the completion branch above; only apply here for non-completion updates
+        if (fields.result !== undefined && task.status !== "completed" && task.status !== "failed") {
+            task.result = fields.result;
+        }
+        task.updatedAt = Date.now();
+        this._persist();
+        return { ...task };
+    }
+    /**
+     * Run an async function in the background for a task.
+     * Auto-updates task status to in_progress/completed/failed.
+     */
+    runBackground(taskId, fn) {
+        const task = this._require(taskId);
+        if (task.blockedBy.length > 0) {
+            throw new Error(`Task ${taskId} is blocked by: ${task.blockedBy.join(", ")}`);
+        }
+        const ac = new AbortController();
+        const bgId = generateId();
+        task.status = "in_progress";
+        task.backgroundId = bgId;
+        task.updatedAt = Date.now();
+        this._backgroundAborts.set(bgId, ac);
+        this._persist();
+        fn(ac.signal).then((result) => {
+            this._backgroundAborts.delete(bgId);
+            try {
+                this.update(taskId, { status: "completed", result });
+            }
+            catch { /* task may have been removed */ }
+        }, (err) => {
+            this._backgroundAborts.delete(bgId);
+            try {
+                this.update(taskId, {
+                    status: "failed",
+                    result: err instanceof Error ? err.message : String(err),
+                });
+            }
+            catch { /* task may have been removed */ }
+        });
+    }
+    abortBackground(taskId) {
+        const task = this._tasks.get(taskId);
+        if (!task?.backgroundId)
+            return false;
+        const ac = this._backgroundAborts.get(task.backgroundId);
+        if (!ac)
+            return false;
+        ac.abort();
+        return true;
+    }
+    drainCompleted() {
+        const results = [...this._completedQueue];
+        this._completedQueue = [];
+        return results;
+    }
+    render() {
+        const tasks = this.listAll();
+        if (tasks.length === 0)
+            return "No tasks.";
+        const lines = tasks.map((t) => {
+            const icon = t.status === "completed" ? "[x]"
+                : t.status === "failed" ? "[!]"
+                    : t.status === "in_progress" ? "[>]"
+                        : t.blockedBy.length > 0 ? "[~]"
+                            : "[ ]";
+            const deps = t.blockedBy.length > 0 ? ` (blocked by: ${t.blockedBy.join(", ")})` : "";
+            return `${icon} ${t.id}: ${t.subject}${deps}`;
+        });
+        const done = tasks.filter((t) => t.status === "completed").length;
+        const ready = tasks.filter((t) => t.status === "pending" && t.blockedBy.length === 0).length;
+        return `Tasks: ${done}/${tasks.length} done, ${ready} ready\n${lines.join("\n")}`;
+    }
+    dispose() {
+        for (const ac of this._backgroundAborts.values()) {
+            ac.abort();
+        }
+        this._backgroundAborts.clear();
+    }
+    _require(id) {
+        const task = this._tasks.get(id);
+        if (!task)
+            throw new Error(`Task not found: ${id}`);
+        return task;
+    }
+    /** Remove completedId from all tasks' blockedBy. Returns IDs of newly unblocked tasks. */
+    _clearDependency(completedId) {
+        const completed = this._tasks.get(completedId);
+        if (!completed)
+            return [];
+        const unblocked = [];
+        for (const dependentId of completed.blocks) {
+            const task = this._tasks.get(dependentId);
+            if (!task)
+                continue;
+            const idx = task.blockedBy.indexOf(completedId);
+            if (idx !== -1) {
+                task.blockedBy.splice(idx, 1);
+                if (task.blockedBy.length === 0 && task.status === "pending") {
+                    unblocked.push(task.id);
+                }
+            }
+        }
+        return unblocked;
+    }
+    /** Check if `taskId` is transitively blocked by `targetId`. */
+    _isTransitivelyBlockedBy(taskId, targetId) {
+        const visited = new Set();
+        const stack = [taskId];
+        while (stack.length > 0) {
+            const current = stack.pop();
+            if (current === targetId)
+                return true;
+            if (visited.has(current))
+                continue;
+            visited.add(current);
+            const node = this._tasks.get(current);
+            if (node)
+                stack.push(...node.blockedBy);
+        }
+        return false;
+    }
+    _persist() {
+        if (!this._config.persistDir)
+            return;
+        const data = JSON.stringify([...this._tasks.values()], null, 2);
+        const target = join(this._config.persistDir, "tasks.json");
+        const tmp = target + ".tmp";
+        writeFileSync(tmp, data, "utf-8");
+        renameSync(tmp, target);
+    }
+    _loadFromDisk() {
+        if (!this._config.persistDir)
+            return;
+        const filePath = join(this._config.persistDir, "tasks.json");
+        if (!existsSync(filePath))
+            return;
+        try {
+            const raw = JSON.parse(readFileSync(filePath, "utf-8"));
+            if (!Array.isArray(raw))
+                return;
+            for (const entry of raw) {
+                if (entry && typeof entry === "object" && typeof entry.id === "string" && typeof entry.subject === "string") {
+                    this._tasks.set(entry.id, entry);
+                }
+            }
+        }
+        catch {
+            // Corrupted file — start fresh
+        }
+    }
+}
+//# sourceMappingURL=task-graph.js.map

package/dist/task-graph/task-graph.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"task-graph.js","sourceRoot":"","sources":["../../src/task-graph/task-graph.ts"],"names":[],"mappings":"AAAA,8EAA8E;AAE9E,OAAO,EAAE,YAAY,EAAE,aAAa,EAAE,SAAS,EAAE,UAAU,EAAE,UAAU,EAAE,MAAM,IAAI,CAAC;AACpF,OAAO,EAAE,IAAI,EAAE,MAAM,MAAM,CAAC;AAC5B,OAAO,EAAE,UAAU,EAAE,MAAM,gBAAgB,CAAC;AAG5C,MAAM,OAAO,SAAS;IACZ,MAAM,GAAG,IAAI,GAAG,EAAoB,CAAC;IACrC,OAAO,CAAkB;IACzB,eAAe,GAA0B,EAAE,CAAC;IAC5C,iBAAiB,GAAG,IAAI,GAAG,EAA2B,CAAC;IAE/D,YAAY,SAA0B,EAAE;QACtC,IAAI,CAAC,OAAO,GAAG,MAAM,CAAC;QACtB,IAAI,MAAM,CAAC,UAAU,EAAE,CAAC;YACtB,SAAS,CAAC,MAAM,CAAC,UAAU,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;YAClD,IAAI,CAAC,aAAa,EAAE,CAAC;QACvB,CAAC;IACH,CAAC;IAGD,GAAG,CAAC,EAAU;QACZ,MAAM,IAAI,GAAG,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;QACjC,OAAO,IAAI,CAAC,CAAC,CAAC,EAAE,GAAG,IAAI,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC;IACxC,CAAC;IAED,OAAO;QACL,OAAO,CAAC,GAAG,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC,CAAC;IACnC,CAAC;IAED,0DAA0D;IAC1D,SAAS;QACP,OAAO,IAAI,CAAC,OAAO,EAAE,CAAC,MAAM,CAC1B,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,KAAK,SAAS,IAAI,CAAC,CAAC,SAAS,CAAC,MAAM,KAAK,CAAC,CAC1D,CAAC;IACJ,CAAC;IAED,4CAA4C;IAC5C,WAAW;QACT,OAAO,IAAI,CAAC,OAAO,EAAE,CAAC,MAAM,CAC1B,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,KAAK,SAAS,IAAI,CAAC,CAAC,SAAS,CAAC,MAAM,GAAG,CAAC,CACxD,CAAC;IACJ,CAAC;IAGD,MAAM,CAAC,OAAe,EAAE,WAAW,GAAG,EAAE;QACtC,MAAM,GAAG,GAAG,IAAI,CAAC,OAAO,CAAC,QAAQ,IAAI,GAAG,CAAC;QACzC,IAAI,IAAI,CAAC,MAAM,CAAC,IAAI,IAAI,GAAG,EAAE,CAAC;YAC5B,MAAM,IAAI,KAAK,CAAC,uBAAuB,GAAG,GAAG,CAAC,CAAC;QACjD,CAAC;QAED,MAAM,IAAI,GAAa;YACrB,EAAE,EAAE,UAAU,EAAE;YAChB,OAAO;YACP,WAAW;YACX,MAAM,EAAE,SAAS;YACjB,SAAS,EAAE,EAAE;YACb,MAAM,EAAE,EAAE;YACV,KAAK,EAAE,EAAE;YACT,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;YACrB,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;SACtB,CAAC;QACF,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,EAAE,IAAI,CAAC,CAAC;QAC/B,IAAI,CAAC,QAAQ,EAAE,CAAC;QAChB,OAAO,EAAE,GAAG,IAAI,EAAE,CAAC;IACrB,CAAC;IAED,8DAA8D;IAC9D,aAAa,CAAC,MAAc,EAAE,WAAmB;QAC/C,MAAM,IAAI,GAAG,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC;QACnC,MAAM,OAAO,GAAG,IAAI,CAAC,QAAQ,CAAC,WAAW,CAAC,CAAC;QAE3C,IAAI,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,WAAW,CAAC;YAAE,OAAO;QAEjD,kGAAkG;QAClG,IAAI,IAAI,CAAC,wBAAwB,CAAC,WAAW,EAAE,MAAM,CAAC,EAAE,CAAC;YACvD,MAAM,IAAI,KAAK,CAAC,qBAAqB,WAAW,MAAM,MAAM,wBAAwB,CAAC,CAAC;QACxF,CAAC;QAED,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;QACjC,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAC5B,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAC5B,OAAO,CAAC,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAC/B,IAAI,CAAC,QAAQ,EAAE,CAAC;IAClB,CAAC;IAED,MAAM,CAAC,MAAc,EAAE,MAAgE;QACrF,MAAM,IAAI,GAAG,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC;QAEnC,IAAI,MAAM,CAAC,MAAM,KAAK,SAAS,IAAI,MAAM,CAAC,MAAM,KAAK,IAAI,CAAC,MAAM,EAAE,CAAC;YACjE,IAAI,MAAM,CAAC,MAAM,KAAK,aAAa,IAAI,IAAI,CAAC,SAAS,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBACjE,MAAM,IAAI,KAAK,CAAC,QAAQ,MAAM,mBAAmB,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;YAChF,CAAC;YACD,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC,MAAM,CAAC;YAE5B,IAAI,MAAM,CAAC,MAAM,KAAK,WAAW,IAAI,MAAM,CAAC,MAAM,KAAK,QAAQ,EAAE,CAAC;gBAChE,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC,MAAM,IAAI,IAAI,CAAC,MAAM,CAAC;gBAC3C,MAAM,SAAS,GAAG,IAAI,CAAC,gBAAgB,CAAC,MAAM,CAAC,CAAC;gBAEhD,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC;oBACxB,MAAM;oBACN,OAAO,EAAE,IAAI,CAAC,OAAO;oBACrB,MAAM,EAAE,IAAI,CAAC,MAAM,IAAI,EAAE;oBACzB,MAAM,EAAE,MAAM,CAAC,MAAM;oBACrB,cAAc,EAAE,SAAS;iBAC1B,CAAC,CAAC;YACL,CAAC;QACH,CAAC;QAED,IAAI,MAAM,CAAC,KAAK,KAAK,SAAS;YAAE,IAAI,CAAC,KAAK,GAAG,MAAM,CAAC,KAAK,CAAC;QAC1D,uGAAuG;QACvG,IAAI,MAAM,CAAC,MAAM,KAAK,SAAS,IAAI,IAAI,CAAC,MAAM,KAAK,WAAW,IAAI,IAAI,CAAC,MAAM,KAAK,QAAQ,EAAE,CAAC;YAC3F,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC,MAAM,CAAC;QAC9B,CAAC;QACD,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAC5B,IAAI,CAAC,QAAQ,EAAE,CAAC;QAChB,OAAO,EAAE,GAAG,IAAI,EAAE,CAAC;IACrB,CAAC;IAGD;;;OAGG;IACH,aAAa,CACX,MAAc,EACd,EAA4C;QAE5C,MAAM,IAAI,GAAG,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC;QACnC,IAAI,IAAI,CAAC,SAAS,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC9B,MAAM,IAAI,KAAK,CAAC,QAAQ,MAAM,mBAAmB,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAChF,CAAC;QAED,MAAM,EAAE,GAAG,IAAI,eAAe,EAAE,CAAC;QACjC,MAAM,IAAI,GAAG,UAAU,EAAE,CAAC;QAC1B,IAAI,CAAC,MAAM,GAAG,aAAa,CAAC;QAC5B,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC;QACzB,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAC5B,IAAI,CAAC,iBAAiB,CAAC,GAAG,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;QACrC,IAAI,CAAC,QAAQ,EAAE,CAAC;QAEhB,EAAE,CAAC,EAAE,CAAC,MAAM,CAAC,CAAC,IAAI,CAChB,CAAC,MAAM,EAAE,EAAE;YACT,IAAI,CAAC,iBAAiB,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;YACpC,IAAI,CAAC;gBAAC,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,EAAE,MAAM,EAAE,WAAW,EAAE,MAAM,EAAE,CAAC,CAAC;YAAC,CAAC;YAAC,MAAM,CAAC,CAAC,gCAAgC,CAAC,CAAC;QAC1G,CAAC,EACD,CAAC,GAAG,EAAE,EAAE;YACN,IAAI,CAAC,iBAAiB,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;YACpC,IAAI,CAAC;gBACH,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE;oBAClB,MAAM,EAAE,QAAQ;oBAChB,MAAM,EAAE,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC;iBACzD,CAAC,CAAC;YACL,CAAC;YAAC,MAAM,CAAC,CAAC,gCAAgC,CAAC,CAAC;QAC9C,CAAC,CACF,CAAC;IACJ,CAAC;IAED,eAAe,CAAC,MAAc;QAC5B,MAAM,IAAI,GAAG,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;QACrC,IAAI,CAAC,IAAI,EAAE,YAAY;YAAE,OAAO,KAAK,CAAC;QACtC,MAAM,EAAE,GAAG,IAAI,CAAC,iBAAiB,CAAC,GAAG,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;QACzD,IAAI,CAAC,EAAE;YAAE,OAAO,KAAK,CAAC;QACtB,EAAE,CAAC,KAAK,EAAE,CAAC;QACX,OAAO,IAAI,CAAC;IACd,CAAC;IAGD,cAAc;QACZ,MAAM,OAAO,GAAG,CAAC,GAAG,IAAI,CAAC,eAAe,CAAC,CAAC;QAC1C,IAAI,CAAC,eAAe,GAAG,EAAE,CAAC;QAC1B,OAAO,OAAO,CAAC;IACjB,CAAC;IAGD,MAAM;QACJ,MAAM,KAAK,GAAG,IAAI,CAAC,OAAO,EAAE,CAAC;QAC7B,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO,WAAW,CAAC;QAE3C,MAAM,KAAK,GAAG,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE;YAC5B,MAAM,IAAI,GAAG,CAAC,CAAC,MAAM,KAAK,WAAW,CAAC,CAAC,CAAC,KAAK;gBAC3C,CAAC,CAAC,CAAC,CAAC,MAAM,KAAK,QAAQ,CAAC,CAAC,CAAC,KAAK;oBAC/B,CAAC,CAAC,CAAC,CAAC,MAAM,KAAK,aAAa,CAAC,CAAC,CAAC,KAAK;wBACpC,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,KAAK;4BAChC,CAAC,CAAC,KAAK,CAAC;YACV,MAAM,IAAI,GAAG,CAAC,CAAC,SAAS,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,iBAAiB,CAAC,CAAC,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC;YACtF,OAAO,GAAG,IAAI,IAAI,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC,OAAO,GAAG,IAAI,EAAE,CAAC;QAChD,CAAC,CAAC,CAAC;QAEH,MAAM,IAAI,GAAG,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,KAAK,WAAW,CAAC,CAAC,MAAM,CAAC;QAClE,MAAM,KAAK,GAAG,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,KAAK,SAAS,IAAI,CAAC,CAAC,SAAS,CAAC,MAAM,KAAK,CAAC,CAAC,CAAC,MAAM,CAAC;QAC7F,OAAO,UAAU,IAAI,IAAI,KAAK,CAAC,MAAM,UAAU,KAAK,WAAW,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC;IACpF,CAAC;IAED,OAAO;QACL,KAAK,MAAM,EAAE,IAAI,IAAI,CAAC,iBAAiB,CAAC,MAAM,EAAE,EAAE,CAAC;YACjD,EAAE,CAAC,KAAK,EAAE,CAAC;QACb,CAAC;QACD,IAAI,CAAC,iBAAiB,CAAC,KAAK,EAAE,CAAC;IACjC,CAAC;IAGO,QAAQ,CAAC,EAAU;QACzB,MAAM,IAAI,GAAG,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;QACjC,IAAI,CAAC,IAAI;YAAE,MAAM,IAAI,KAAK,CAAC,mBAAmB,EAAE,EAAE,CAAC,CAAC;QACpD,OAAO,IAAI,CAAC;IACd,CAAC;IAED,0FAA0F;IAClF,gBAAgB,CAAC,WAAmB;QAC1C,MAAM,SAAS,GAAG,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC;QAC/C,IAAI,CAAC,SAAS;YAAE,OAAO,EAAE,CAAC;QAE1B,MAAM,SAAS,GAAa,EAAE,CAAC;QAC/B,KAAK,MAAM,WAAW,IAAI,SAAS,CAAC,MAAM,EAAE,CAAC;YAC3C,MAAM,IAAI,GAAG,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC;YAC1C,IAAI,CAAC,IAAI;gBAAE,SAAS;YACpB,MAAM,GAAG,GAAG,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC;YAChD,IAAI,GAAG,KAAK,CAAC,CAAC,EAAE,CAAC;gBACf,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,GAAG,EAAE,CAAC,CAAC,CAAC;gBAC9B,IAAI,IAAI,CAAC,SAAS,CAAC,MAAM,KAAK,CAAC,IAAI,IAAI,CAAC,MAAM,KAAK,SAAS,EAAE,CAAC;oBAC7D,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;gBAC1B,CAAC;YACH,CAAC;QACH,CAAC;QACD,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,+DAA+D;IACvD,wBAAwB,CAAC,MAAc,EAAE,QAAgB;QAC/D,MAAM,OAAO,GAAG,IAAI,GAAG,EAAU,CAAC;QAClC,MAAM,KAAK,GAAG,CAAC,MAAM,CAAC,CAAC;QACvB,OAAO,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACxB,MAAM,OAAO,GAAG,KAAK,CAAC,GAAG,EAAG,CAAC;YAC7B,IAAI,OAAO,KAAK,QAAQ;gBAAE,OAAO,IAAI,CAAC;YACtC,IAAI,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC;gBAAE,SAAS;YACnC,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;YACrB,MAAM,IAAI,GAAG,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;YACtC,IAAI,IAAI;gBAAE,KAAK,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC,SAAS,CAAC,CAAC;QAC1C,CAAC;QACD,OAAO,KAAK,CAAC;IACf,CAAC;IAEO,QAAQ;QACd,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,UAAU;YAAE,OAAO;QACrC,MAAM,IAAI,GAAG,IAAI,CAAC,SAAS,CAAC,CAAC,GAAG,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC;QAChE,MAAM,MAAM,GAAG,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,UAAU,EAAE,YAAY,CAAC,CAAC;QAC3D,MAAM,GAAG,GAAG,MAAM,GAAG,MAAM,CAAC;QAC5B,aAAa,CAAC,GAAG,EAAE,IAAI,EAAE,OAAO,CAAC,CAAC;QAClC,UAAU,CAAC,GAAG,EAAE,MAAM,CAAC,CAAC;IAC1B,CAAC;IAEO,aAAa;QACnB,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,UAAU;YAAE,OAAO;QACrC,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,UAAU,EAAE,YAAY,CAAC,CAAC;QAC7D,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC;YAAE,OAAO;QAClC,IAAI,CAAC;YACH,MAAM,GAAG,GAAG,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC,CAAC;YACxD,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC;gBAAE,OAAO;YAChC,KAAK,MAAM,KAAK,IAAI,GAAG,EAAE,CAAC;gBACxB,IAAI,KAAK,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,OAAO,KAAK,CAAC,EAAE,KAAK,QAAQ,IAAI,OAAO,KAAK,CAAC,OAAO,KAAK,QAAQ,EAAE,CAAC;oBAC5G,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,EAAE,KAAiB,CAAC,CAAC;gBAC/C,CAAC;YACH,CAAC;QACH,CAAC;QAAC,MAAM,CAAC;YACP,+BAA+B;QACjC,CAAC;IACH,CAAC;CACF"}

package/dist/task-graph/tools.d.ts

@@ -0,0 +1,3 @@
+import type { AgentTool } from "../tools/types.js";
+import type { TaskGraph } from "./task-graph.js";
+export declare function createTaskGraphTools(graph: TaskGraph): AgentTool[];

package/dist/task-graph/tools.js

@@ -0,0 +1,106 @@
+// Task graph tools — CRUD + dependency management + background execution
+import { Type } from "@sinclair/typebox";
+import { TASK_GRAPH_TOOL_NAMES } from "./types.js";
+export function createTaskGraphTools(graph) {
+    return [
+        {
+            name: TASK_GRAPH_TOOL_NAMES.create,
+            label: "Create Task",
+            description: "Create a new task in the task graph. Tasks start as pending. " +
+                "Use task_depend to set up dependency ordering.",
+            parameters: Type.Object({
+                subject: Type.String({ description: "Short title for the task." }),
+                description: Type.Optional(Type.String({ description: "Detailed description." })),
+            }),
+            async execute(_id, params) {
+                const task = graph.create(params.subject, params.description);
+                return text(`Created task ${task.id}: ${task.subject}\n\n${graph.render()}`);
+            },
+        },
+        {
+            name: TASK_GRAPH_TOOL_NAMES.depend,
+            label: "Add Task Dependency",
+            description: "Add a dependency: task_id cannot start until blocked_by_id completes. " +
+                "Rejects if this would create a cycle.",
+            parameters: Type.Object({
+                task_id: Type.String({ description: "Task that is blocked." }),
+                blocked_by_id: Type.String({ description: "Task that must complete first." }),
+            }),
+            async execute(_id, params) {
+                graph.addDependency(params.task_id, params.blocked_by_id);
+                return text(`Dependency added: ${params.task_id} blocked by ${params.blocked_by_id}\n\n${graph.render()}`);
+            },
+        },
+        {
+            name: TASK_GRAPH_TOOL_NAMES.update,
+            label: "Update Task",
+            description: "Update a task's status, owner, or result. " +
+                "Setting status to 'completed' auto-unblocks dependent tasks. " +
+                "Cannot start a task that has unfinished blockers.",
+            parameters: Type.Object({
+                task_id: Type.String({ description: "Task ID." }),
+                status: Type.Optional(Type.Union([Type.Literal("pending"), Type.Literal("in_progress"), Type.Literal("completed"), Type.Literal("failed")], { description: "New status." })),
+                owner: Type.Optional(Type.String({ description: "Assign to an agent or user." })),
+                result: Type.Optional(Type.String({ description: "Result summary." })),
+            }),
+            async execute(_id, params) {
+                const task = graph.update(params.task_id, {
+                    status: params.status,
+                    owner: params.owner,
+                    result: params.result,
+                });
+                return text(`Updated task ${task.id}\n\n${graph.render()}`);
+            },
+        },
+        {
+            name: TASK_GRAPH_TOOL_NAMES.list,
+            label: "List Tasks",
+            description: "List all tasks with their status, dependencies, and progress.",
+            parameters: Type.Object({
+                filter: Type.Optional(Type.Union([Type.Literal("all"), Type.Literal("ready"), Type.Literal("blocked"), Type.Literal("in_progress"), Type.Literal("completed")], { description: "Filter tasks by category. Default: all." })),
+            }),
+            async execute(_id, params) {
+                let tasks;
+                switch (params.filter) {
+                    case "ready":
+                        tasks = graph.listReady();
+                        break;
+                    case "blocked":
+                        tasks = graph.listBlocked();
+                        break;
+                    case "in_progress":
+                        tasks = graph.listAll().filter((t) => t.status === "in_progress");
+                        break;
+                    case "completed":
+                        tasks = graph.listAll().filter((t) => t.status === "completed");
+                        break;
+                    default: tasks = graph.listAll();
+                }
+                if (tasks.length === 0)
+                    return text(`No tasks matching filter: ${params.filter ?? "all"}`);
+                if (!params.filter || params.filter === "all")
+                    return text(graph.render());
+                const lines = tasks.map((t) => `${t.id}: ${t.subject} [${t.status}]`);
+                return text(`${params.filter}: ${tasks.length} task(s)\n${lines.join("\n")}`);
+            },
+        },
+        {
+            name: TASK_GRAPH_TOOL_NAMES.get,
+            label: "Get Task",
+            description: "Get detailed information about a specific task.",
+            parameters: Type.Object({
+                task_id: Type.String({ description: "Task ID." }),
+            }),
+            async execute(_id, params) {
+                const task = graph.get(params.task_id);
+                if (!task)
+                    return text(`Task not found: ${params.task_id}`);
+                return text(JSON.stringify(task, null, 2));
+            },
+        },
+    ];
+}
+function text(t) {
+    return { content: [{ type: "text", text: t }] };
+}
+//# sourceMappingURL=tools.js.map

package/dist/task-graph/tools.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"tools.js","sourceRoot":"","sources":["../../src/task-graph/tools.ts"],"names":[],"mappings":"AAAA,yEAAyE;AAEzE,OAAO,EAAE,IAAI,EAAE,MAAM,mBAAmB,CAAC;AAGzC,OAAO,EAAE,qBAAqB,EAAE,MAAM,YAAY,CAAC;AAGnD,MAAM,UAAU,oBAAoB,CAAC,KAAgB;IACnD,OAAO;QACL;YACE,IAAI,EAAE,qBAAqB,CAAC,MAAM;YAClC,KAAK,EAAE,aAAa;YACpB,WAAW,EACT,+DAA+D;gBAC/D,gDAAgD;YAClD,UAAU,EAAE,IAAI,CAAC,MAAM,CAAC;gBACtB,OAAO,EAAE,IAAI,CAAC,MAAM,CAAC,EAAE,WAAW,EAAE,2BAA2B,EAAE,CAAC;gBAClE,WAAW,EAAE,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,WAAW,EAAE,uBAAuB,EAAE,CAAC,CAAC;aAClF,CAAC;YACF,KAAK,CAAC,OAAO,CAAC,GAAG,EAAE,MAAiD;gBAClE,MAAM,IAAI,GAAG,KAAK,CAAC,MAAM,CAAC,MAAM,CAAC,OAAO,EAAE,MAAM,CAAC,WAAW,CAAC,CAAC;gBAC9D,OAAO,IAAI,CAAC,gBAAgB,IAAI,CAAC,EAAE,KAAK,IAAI,CAAC,OAAO,OAAO,KAAK,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC;YAC/E,CAAC;SACF;QACD;YACE,IAAI,EAAE,qBAAqB,CAAC,MAAM;YAClC,KAAK,EAAE,qBAAqB;YAC5B,WAAW,EACT,wEAAwE;gBACxE,uCAAuC;YACzC,UAAU,EAAE,IAAI,CAAC,MAAM,CAAC;gBACtB,OAAO,EAAE,IAAI,CAAC,MAAM,CAAC,EAAE,WAAW,EAAE,uBAAuB,EAAE,CAAC;gBAC9D,aAAa,EAAE,IAAI,CAAC,MAAM,CAAC,EAAE,WAAW,EAAE,gCAAgC,EAAE,CAAC;aAC9E,CAAC;YACF,KAAK,CAAC,OAAO,CAAC,GAAG,EAAE,MAAkD;gBACnE,KAAK,CAAC,aAAa,CAAC,MAAM,CAAC,OAAO,EAAE,MAAM,CAAC,aAAa,CAAC,CAAC;gBAC1D,OAAO,IAAI,CAAC,qBAAqB,MAAM,CAAC,OAAO,eAAe,MAAM,CAAC,aAAa,OAAO,KAAK,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC;YAC7G,CAAC;SACF;QACD;YACE,IAAI,EAAE,qBAAqB,CAAC,MAAM;YAClC,KAAK,EAAE,aAAa;YACpB,WAAW,EACT,4CAA4C;gBAC5C,+DAA+D;gBAC/D,mDAAmD;YACrD,UAAU,EAAE,IAAI,CAAC,MAAM,CAAC;gBACtB,OAAO,EAAE,IAAI,CAAC,MAAM,CAAC,EAAE,WAAW,EAAE,UAAU,EAAE,CAAC;gBACjD,MAAM,EAAE,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,KAAK,CAC9B,CAAC,IAAI,CAAC,OAAO,CAAC,SAAS,CAAC,EAAE,IAAI,CAAC,OAAO,CAAC,aAAa,CAAC,EAAE,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,EAAE,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC,EACzG,EAAE,WAAW,EAAE,aAAa,EAAE,CAC/B,CAAC;gBACF,KAAK,EAAE,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,WAAW,EAAE,6BAA6B,EAAE,CAAC,CAAC;gBACjF,MAAM,EAAE,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,WAAW,EAAE,iBAAiB,EAAE,CAAC,CAAC;aACvE,CAAC;YACF,KAAK,CAAC,OAAO,CAAC,GAAG,EAAE,MAA6E;gBAC9F,MAAM,IAAI,GAAG,KAAK,CAAC,MAAM,CAAC,MAAM,CAAC,OAAO,EAAE;oBACxC,MAAM,EAAE,MAAM,CAAC,MAAgC;oBAC/C,KAAK,EAAE,MAAM,CAAC,KAAK;oBACnB,MAAM,EAAE,MAAM,CAAC,MAAM;iBACtB,CAAC,CAAC;gBACH,OAAO,IAAI,CAAC,gBAAgB,IAAI,CAAC,EAAE,OAAO,KAAK,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC;YAC9D,CAAC;SACF;QACD;YACE,IAAI,EAAE,qBAAqB,CAAC,IAAI;YAChC,KAAK,EAAE,YAAY;YACnB,WAAW,EAAE,+DAA+D;YAC5E,UAAU,EAAE,IAAI,CAAC,MAAM,CAAC;gBACtB,MAAM,EAAE,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,KAAK,CAC9B,CAAC,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,IAAI,CAAC,OAAO,CAAC,SAAS,CAAC,EAAE,IAAI,CAAC,OAAO,CAAC,aAAa,CAAC,EAAE,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC,EAC7H,EAAE,WAAW,EAAE,yCAAyC,EAAE,CAC3D,CAAC;aACH,CAAC;YACF,KAAK,CAAC,OAAO,CAAC,GAAG,EAAE,MAA2B;gBAC5C,IAAI,KAAiB,CAAC;gBACtB,QAAQ,MAAM,CAAC,MAAM,EAAE,CAAC;oBACtB,KAAK,OAAO;wBAAE,KAAK,GAAG,KAAK,CAAC,SAAS,EAAE,CAAC;wBAAC,MAAM;oBAC/C,KAAK,SAAS;wBAAE,KAAK,GAAG,KAAK,CAAC,WAAW,EAAE,CAAC;wBAAC,MAAM;oBACnD,KAAK,aAAa;wBAAE,KAAK,GAAG,KAAK,CAAC,OAAO,EAAE,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,KAAK,aAAa,CAAC,CAAC;wBAAC,MAAM;oBAC7F,KAAK,WAAW;wBAAE,KAAK,GAAG,KAAK,CAAC,OAAO,EAAE,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,KAAK,WAAW,CAAC,CAAC;wBAAC,MAAM;oBACzF,OAAO,CAAC,CAAC,KAAK,GAAG,KAAK,CAAC,OAAO,EAAE,CAAC;gBACnC,CAAC;gBACD,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC;oBAAE,OAAO,IAAI,CAAC,6BAA6B,MAAM,CAAC,MAAM,IAAI,KAAK,EAAE,CAAC,CAAC;gBAC3F,IAAI,CAAC,MAAM,CAAC,MAAM,IAAI,MAAM,CAAC,MAAM,KAAK,KAAK;oBAAE,OAAO,IAAI,CAAC,KAAK,CAAC,MAAM,EAAE,CAAC,CAAC;gBAC3E,MAAM,KAAK,GAAG,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,GAAG,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC,OAAO,KAAK,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC;gBACtE,OAAO,IAAI,CAAC,GAAG,MAAM,CAAC,MAAM,KAAK,KAAK,CAAC,MAAM,aAAa,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;YAChF,CAAC;SACF;QACD;YACE,IAAI,EAAE,qBAAqB,CAAC,GAAG;YAC/B,KAAK,EAAE,UAAU;YACjB,WAAW,EAAE,iDAAiD;YAC9D,UAAU,EAAE,IAAI,CAAC,MAAM,CAAC;gBACtB,OAAO,EAAE,IAAI,CAAC,MAAM,CAAC,EAAE,WAAW,EAAE,UAAU,EAAE,CAAC;aAClD,CAAC;YACF,KAAK,CAAC,OAAO,CAAC,GAAG,EAAE,MAA2B;gBAC5C,MAAM,IAAI,GAAG,KAAK,CAAC,GAAG,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;gBACvC,IAAI,CAAC,IAAI;oBAAE,OAAO,IAAI,CAAC,mBAAmB,MAAM,CAAC,OAAO,EAAE,CAAC,CAAC;gBAC5D,OAAO,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC;YAC7C,CAAC;SACF;KACF,CAAC;AACJ,CAAC;AAED,SAAS,IAAI,CAAC,CAAS;IACrB,OAAO,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC;AAClD,CAAC"}

package/dist/task-graph/types.d.ts

@@ -0,0 +1,44 @@
+export declare const TASK_GRAPH_TOOL_NAMES: {
+    readonly create: "task_create";
+    readonly depend: "task_depend";
+    readonly update: "task_update";
+    readonly list: "task_list";
+    readonly get: "task_get";
+};
+export type TaskStatus = "pending" | "in_progress" | "completed" | "failed";
+export interface TaskNode {
+    id: string;
+    subject: string;
+    description: string;
+    status: TaskStatus;
+    /** IDs of tasks that must complete before this one can start. */
+    blockedBy: string[];
+    /** IDs of tasks that this task blocks (reverse edges, maintained automatically). */
+    blocks: string[];
+    /** Agent or user assigned to this task. */
+    owner: string;
+    /** Result summary after completion/failure. */
+    result?: string;
+    /** Background execution handle ID, if running in background. */
+    backgroundId?: string;
+    createdAt: number;
+    updatedAt: number;
+}
+export interface TaskGraphConfig {
+    /** Directory for persisting task graph to disk. If omitted, in-memory only. */
+    persistDir?: string;
+    /** Maximum number of tasks. Default: 100. */
+    maxTasks?: number;
+    /**
+     * Auto-inject completed background task results before each LLM call.
+     * Default: true.
+     */
+    autoInjectResults?: boolean;
+}
+export interface CompletedTaskResult {
+    taskId: string;
+    subject: string;
+    result: string;
+    status: "completed" | "failed";
+    unblockedTasks: string[];
+}

package/dist/task-graph/types.js

@@ -0,0 +1,9 @@
+// Task graph types — dependency-aware task DAG with background execution
+export const TASK_GRAPH_TOOL_NAMES = {
+    create: "task_create",
+    depend: "task_depend",
+    update: "task_update",
+    list: "task_list",
+    get: "task_get",
+};
+//# sourceMappingURL=types.js.map

package/dist/task-graph/types.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.js","sourceRoot":"","sources":["../../src/task-graph/types.ts"],"names":[],"mappings":"AAAA,yEAAyE;AAEzE,MAAM,CAAC,MAAM,qBAAqB,GAAG;IACnC,MAAM,EAAE,aAAa;IACrB,MAAM,EAAE,aAAa;IACrB,MAAM,EAAE,aAAa;IACrB,IAAI,EAAE,WAAW;IACjB,GAAG,EAAE,UAAU;CACP,CAAC"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "klaus-agent",
-  "version": "0.3.1",
+  "version": "0.4.0",
   "description": "Universal agent framework",
   "type": "module",
   "main": "dist/index.js",

package/src/core/agent.ts

@@ -22,6 +22,7 @@ import type { SkillSource } from "../skills/types.js";
 import type { MCPServerConfig, MCPClient } from "../tools/mcp-adapter.js";
 import type { TaskFactory } from "../background/types.js";
 import type { PlanningConfig } from "../planning/types.js";
+import type { TaskGraphConfig } from "../task-graph/types.js";
 import { SessionManager } from "../session/session-manager.js";
 import { CheckpointManager } from "../checkpoint/checkpoint-manager.js";
 import { InjectionManager } from "../injection/injection-manager.js";
@@ -39,6 +40,9 @@ import { createBackgroundTaskTools } from "../background/tools.js";
 import { PlanningManager } from "../planning/planning-manager.js";
 import { createPlanningTools } from "../planning/tools.js";
 import { PlanningNagProvider } from "../planning/nag-injection.js";
+import { TaskGraph } from "../task-graph/task-graph.js";
+import { createTaskGraphTools } from "../task-graph/tools.js";
+import { TaskResultInjectionProvider } from "../task-graph/result-injection.js";
 import { runAgentLoop } from "./agent-loop.js";
 
 export interface AgentConfig {
@@ -64,6 +68,7 @@ export interface AgentConfig {
   mcp?: { servers: MCPServerConfig[]; clientFactory: (config: MCPServerConfig) => MCPClient };
   wire?: { bufferSize?: number };
   backgroundTasks?: { factories?: Record<string, TaskFactory> };
+  taskGraph?: TaskGraphConfig;
   planning?: PlanningConfig;
 }
 
@@ -89,6 +94,7 @@ export class Agent {
   private _wire: Wire;
   private _backgroundTaskManager: BackgroundTaskManager | undefined;
   private _planningManager: PlanningManager | undefined;
+  private _taskGraph: TaskGraph;
   private _initialized = false;
 
   constructor(config: AgentConfig) {
@@ -156,6 +162,9 @@ export class Agent {
     if (config.planning) {
       this._planningManager = new PlanningManager(config.planning);
     }
+
+    // Task graph
+    this._taskGraph = new TaskGraph(config.taskGraph ?? {});
   }
 
   // --- Public API ---
@@ -253,6 +262,10 @@ export class Agent {
     return this._planningManager;
   }
 
+  get taskGraph(): TaskGraph {
+    return this._taskGraph;
+  }
+
   setSystemPrompt(prompt: string): void {
     this._state.systemPrompt = prompt;
   }
@@ -282,6 +295,7 @@ export class Agent {
     this._followUpQueue = [];
     await this._mcpAdapter?.dispose();
     this._backgroundTaskManager?.dispose();
+    this._taskGraph.dispose();
     this._wire.dispose();
   }
 
@@ -373,12 +387,21 @@ export class Agent {
       this._state.tools = [...this._state.tools, ...createPlanningTools(this._planningManager)];
 
       // Register nag provider into injection manager (create one if needed)
-      const nagProvider = new PlanningNagProvider(this._planningManager);
-      if (this._injectionManager) {
-        this._injectionManager.addProvider(nagProvider);
-      } else {
-        this._injectionManager = new InjectionManager([nagProvider]);
-      }
+      this._addInjectionProvider(new PlanningNagProvider(this._planningManager));
+    }
+
+    // Task graph tools + result auto-injection
+    this._state.tools = [...this._state.tools, ...createTaskGraphTools(this._taskGraph)];
+    if (this._config.taskGraph?.autoInjectResults !== false) {
+      this._addInjectionProvider(new TaskResultInjectionProvider(this._taskGraph));
+    }
+  }
+
+  private _addInjectionProvider(provider: DynamicInjectionProvider): void {
+    if (this._injectionManager) {
+      this._injectionManager.addProvider(provider);
+    } else {
+      this._injectionManager = new InjectionManager([provider]);
     }
   }
 

package/src/index.ts

@@ -16,6 +16,7 @@ import type { SkillSource } from "./skills/types.js";
 import type { MCPServerConfig, MCPClient } from "./tools/mcp-adapter.js";
 import type { TaskFactory } from "./background/types.js";
 import type { PlanningConfig } from "./planning/types.js";
+import type { TaskGraphConfig } from "./task-graph/types.js";
 
 export interface CreateAgentConfig {
   // Required
@@ -43,6 +44,7 @@ export interface CreateAgentConfig {
   wire?: { bufferSize?: number };
   backgroundTasks?: { factories?: Record<string, TaskFactory> };
   planning?: PlanningConfig;
+  taskGraph?: TaskGraphConfig;
 
   // Advanced: provide your own LLM provider
   provider?: LLMProvider;
@@ -74,6 +76,7 @@ export function createAgent(config: CreateAgentConfig): Agent {
     wire: config.wire,
     backgroundTasks: config.backgroundTasks,
     planning: config.planning,
+    taskGraph: config.taskGraph,
   });
 }
 
@@ -107,6 +110,9 @@ export { createBackgroundTaskTools } from "./background/tools.js";
 export { PlanningManager } from "./planning/planning-manager.js";
 export { createPlanningTools } from "./planning/tools.js";
 export { PlanningNagProvider } from "./planning/nag-injection.js";
+export { TaskGraph } from "./task-graph/task-graph.js";
+export { createTaskGraphTools } from "./task-graph/tools.js";
+export { TaskResultInjectionProvider } from "./task-graph/result-injection.js";
 
 // Core types
 export type {
@@ -243,3 +249,12 @@ export type {
 } from "./planning/types.js";
 
 export { PLANNING_TOOL_NAMES } from "./planning/types.js";
+export { TASK_GRAPH_TOOL_NAMES } from "./task-graph/types.js";
+
+// Task graph types
+export type {
+  TaskGraphConfig,
+  TaskNode,
+  TaskStatus,
+  CompletedTaskResult,
+} from "./task-graph/types.js";

package/src/task-graph/result-injection.ts

@@ -0,0 +1,29 @@
+// Auto-inject completed background task results before each LLM call
+
+import type { DynamicInjectionProvider, DynamicInjection } from "../injection/types.js";
+import type { AgentMessage } from "../types.js";
+import type { TaskGraph } from "./task-graph.js";
+
+export class TaskResultInjectionProvider implements DynamicInjectionProvider {
+  constructor(private _graph: TaskGraph) {}
+
+  async getInjections(_history: AgentMessage[]): Promise<DynamicInjection[]> {
+    const completed = this._graph.drainCompleted();
+    if (completed.length === 0) return [];
+
+    const lines = completed.map((c) => {
+      const status = c.status === "completed" ? "completed" : "FAILED";
+      const unblocked = c.unblockedTasks.length > 0
+        ? ` → unblocked: ${c.unblockedTasks.join(", ")}`
+        : "";
+      return `[task:${c.taskId}] ${c.subject} — ${status}: ${c.result}${unblocked}`;
+    });
+
+    return [
+      {
+        type: "task-results",
+        content: `<background-results>\n${lines.join("\n")}\n</background-results>`,
+      },
+    ];
+  }
+}