@nodetool-ai/agents 0.7.0-rc.11

package/dist/task-executor.js

@@ -0,0 +1,359 @@
+/**
+ * TaskExecutor -- orchestrates execution of a complete Task plan.
+ *
+ * Port of src/nodetool/agents/task_executor.py
+ *
+ * Iteratively finds steps whose dependencies are satisfied, runs
+ * StepExecutor for each, and collects results until all steps complete
+ * or the safety limit is reached.
+ *
+ * Process-mode steps automatically fan out over list inputs produced by
+ * a preceding discover step. Each item is rendered into the step's
+ * `perItemInstructions` template and executed as an ephemeral step.
+ * Results are aggregated into a list that downstream aggregate steps consume.
+ */
+import { createHash } from "node:crypto";
+import { createLogger } from "@nodetool-ai/config";
+const log = createLogger("nodetool.agents.task-executor");
+import { StepExecutor } from "./step-executor.js";
+import { DEFAULT_TOKEN_LIMIT } from "./constants.js";
+const DEFAULT_MAX_STEPS = 50;
+const DEFAULT_MAX_STEP_ITERATIONS = 10;
+export class TaskExecutor {
+    provider;
+    model;
+    tools;
+    task;
+    context;
+    inputs;
+    systemPrompt;
+    maxSteps;
+    maxStepIterations;
+    maxTokenLimit;
+    finalStepId;
+    parallelExecution;
+    _finishStepId;
+    constructor(opts) {
+        this.provider = opts.provider;
+        this.model = opts.model;
+        this.tools = opts.tools;
+        this.task = opts.task;
+        this.context = opts.context;
+        this.inputs = opts.inputs ?? {};
+        this.systemPrompt = opts.systemPrompt;
+        this.maxSteps = opts.maxSteps ?? DEFAULT_MAX_STEPS;
+        this.maxStepIterations =
+            opts.maxStepIterations ?? DEFAULT_MAX_STEP_ITERATIONS;
+        this.maxTokenLimit = opts.maxTokenLimit ?? DEFAULT_TOKEN_LIMIT;
+        this.finalStepId = opts.finalStepId;
+        this.parallelExecution = opts.parallelExecution ?? false;
+    }
+    /**
+     * Execute all steps in the task plan, respecting dependency order.
+     * Supports both sequential and parallel execution modes.
+     */
+    async *executeTasks() {
+        // Seed inputs into context
+        for (const [key, value] of Object.entries(this.inputs)) {
+            this.context.set(key, value);
+        }
+        // Auto-detect finish step (last step) like Python does
+        this._finishStepId =
+            this.finalStepId ??
+                (this.task.steps.length > 0
+                    ? this.task.steps[this.task.steps.length - 1].id
+                    : undefined);
+        log.info("Task execution started", {
+            title: this.task.title,
+            steps: this.task.steps.length
+        });
+        let stepsTaken = 0;
+        while (!this.allTasksComplete() && stepsTaken < this.maxSteps) {
+            stepsTaken++;
+            let executableSteps = this.getExecutableSteps();
+            executableSteps = this.maybeDeferFinishStep(executableSteps);
+            if (executableSteps.length === 0) {
+                if (!this.allTasksComplete()) {
+                    yield {
+                        type: "chunk",
+                        content: "\nNo executable steps but not all complete. Possible dependency issues.\n",
+                        done: false
+                    };
+                }
+                break;
+            }
+            log.debug("Dispatching steps", {
+                stepIds: executableSteps.map((s) => s.id)
+            });
+            // Separate process-mode steps from normal steps
+            const processSteps = executableSteps.filter((s) => s.mode === "process");
+            const normalSteps = executableSteps.filter((s) => s.mode !== "process");
+            // Handle process-mode steps with fan-out
+            for (const pStep of processSteps) {
+                yield* this.handleProcessStep(pStep);
+            }
+            // Create step executors for normal steps
+            const stepGenerators = normalSteps.map((step) => {
+                const executor = new StepExecutor({
+                    task: this.task,
+                    step,
+                    context: this.context,
+                    provider: this.provider,
+                    model: this.model,
+                    tools: [...this.tools],
+                    systemPrompt: this.systemPrompt,
+                    maxTokenLimit: this.maxTokenLimit,
+                    maxIterations: this.maxStepIterations,
+                    useFinishTask: this.isFinishStep(step)
+                });
+                return executor.execute();
+            });
+            if (this.parallelExecution && stepGenerators.length > 1) {
+                // Execute all steps concurrently, merging yielded messages
+                yield* mergeAsyncGenerators(stepGenerators);
+            }
+            else {
+                // Execute steps sequentially
+                for (const generator of stepGenerators) {
+                    for await (const message of generator) {
+                        yield message;
+                    }
+                }
+            }
+        }
+    }
+    /**
+     * Produce a short deterministic hash for a value (used in ephemeral step IDs).
+     */
+    shortHash(value) {
+        const data = JSON.stringify(value, (_key, val) => {
+            if (val != null && typeof val === "object" && !Array.isArray(val)) {
+                const sorted = {};
+                for (const k of Object.keys(val).sort()) {
+                    sorted[k] = val[k];
+                }
+                return sorted;
+            }
+            return val;
+        });
+        return createHash("sha1").update(data).digest("hex").slice(0, 12);
+    }
+    /**
+     * Handle a process-mode step by fanning out over list inputs.
+     * Creates ephemeral steps for each item in the discover step's result
+     * and aggregates the outputs into a list stored in context.
+     */
+    async *handleProcessStep(step) {
+        const discoverStepId = step.dependsOn[0];
+        if (!discoverStepId) {
+            log.warn("Process step has no dependencies, skipping fan-out", {
+                stepId: step.id
+            });
+            step.completed = true;
+            return;
+        }
+        let discoverResult = this.context.get(discoverStepId);
+        if (discoverResult === undefined || discoverResult === null) {
+            log.warn("Discover step result is null/undefined, skipping fan-out", {
+                stepId: step.id
+            });
+            step.completed = true;
+            this.context.set(step.id, []);
+            step.endTime = Date.now();
+            return;
+        }
+        if (!Array.isArray(discoverResult)) {
+            log.warn("Discover step result is not an array, wrapping as single-item list", {
+                stepId: step.id,
+                resultType: typeof discoverResult
+            });
+            discoverResult = [discoverResult];
+        }
+        const items = discoverResult;
+        const template = step.perItemInstructions ?? step.instructions;
+        const perItemSchema = step.perItemSchema;
+        log.info("Fan-out processing", {
+            stepId: step.id,
+            itemCount: items.length
+        });
+        // Create ephemeral steps for each item
+        const ephemeralSteps = items.map((item) => {
+            let instructions = template;
+            if (typeof item === "object" && item !== null) {
+                for (const [key, value] of Object.entries(item)) {
+                    const escapedKey = key.replace(/[.*+?^${}()|[\]\\]/g, "\\$&");
+                    const strValue = String(value);
+                    instructions = instructions.replace(new RegExp(`\\{${escapedKey}\\}`, "g"), () => strValue);
+                }
+            }
+            else {
+                const strItem = String(item);
+                instructions = instructions.replace(/\{item\}/g, () => strItem);
+            }
+            const hash = this.shortHash(item);
+            return {
+                id: `${step.id}_item_${hash}`,
+                instructions,
+                completed: false,
+                dependsOn: [],
+                logs: [],
+                outputSchema: perItemSchema ?? step.outputSchema
+            };
+        });
+        // Create step executors for each ephemeral step
+        const generators = ephemeralSteps.map((ephStep) => {
+            const executor = new StepExecutor({
+                task: this.task,
+                step: ephStep,
+                context: this.context,
+                provider: this.provider,
+                model: this.model,
+                tools: [...this.tools],
+                systemPrompt: this.systemPrompt,
+                maxTokenLimit: this.maxTokenLimit,
+                maxIterations: this.maxStepIterations,
+                useFinishTask: false
+            });
+            return executor.execute();
+        });
+        // Execute and collect results
+        const results = [];
+        if (this.parallelExecution && generators.length > 1) {
+            for await (const msg of mergeAsyncGenerators(generators)) {
+                if (msg.type === "step_result") {
+                    results.push(msg.result);
+                }
+                yield msg;
+            }
+        }
+        else {
+            for (const gen of generators) {
+                for await (const msg of gen) {
+                    if (msg.type === "step_result") {
+                        results.push(msg.result);
+                    }
+                    yield msg;
+                }
+            }
+        }
+        // Store aggregated results and mark complete
+        this.context.set(step.id, results);
+        step.completed = true;
+        step.endTime = Date.now();
+        log.info("Fan-out complete", {
+            stepId: step.id,
+            resultCount: results.length
+        });
+    }
+    /**
+     * Check if all steps in the task are completed.
+     */
+    allTasksComplete() {
+        return this.task.steps.every((step) => step.completed);
+    }
+    /**
+     * Find steps whose dependencies are all satisfied (completed).
+     */
+    getExecutableSteps() {
+        const completedIds = new Set(this.task.steps.filter((s) => s.completed).map((s) => s.id));
+        // Also count inputs as satisfied dependencies
+        for (const key of Object.keys(this.inputs)) {
+            completedIds.add(key);
+        }
+        return this.task.steps.filter((step) => !step.completed &&
+            !this.isStepRunning(step) &&
+            step.dependsOn.every((dep) => completedIds.has(dep)));
+    }
+    /**
+     * Check if a step is currently running (started but not finished).
+     * Mirrors Python's Step.is_running().
+     */
+    isStepRunning(step) {
+        return step.startTime != null && step.endTime == null;
+    }
+    /**
+     * Check if a step is the designated finish/aggregation step.
+     */
+    isFinishStep(step) {
+        if (this._finishStepId) {
+            return step.id === this._finishStepId;
+        }
+        return (this.task.steps.length > 0 &&
+            step === this.task.steps[this.task.steps.length - 1]);
+    }
+    /**
+     * Defer the finish step until all other steps are complete.
+     * This ensures the final aggregation step runs last.
+     */
+    maybeDeferFinishStep(executableSteps) {
+        if (!this._finishStepId)
+            return executableSteps;
+        const finishReady = executableSteps.some((s) => s.id === this._finishStepId);
+        if (!finishReady)
+            return executableSteps;
+        const otherPending = this.task.steps.some((s) => !s.completed && s.id !== this._finishStepId);
+        if (!otherPending)
+            return executableSteps;
+        return executableSteps.filter((s) => s.id !== this._finishStepId);
+    }
+}
+// ---------------------------------------------------------------------------
+// Async generator merge utility (TS equivalent of wrap_generators_parallel)
+// ---------------------------------------------------------------------------
+async function* mergeAsyncGenerators(generators) {
+    // Channel: a queue of resolved values with a promise-based pull mechanism
+    const queue = [];
+    let activeCount = generators.length;
+    let resolve = null;
+    let firstError = undefined;
+    function notify() {
+        if (resolve) {
+            const r = resolve;
+            resolve = null;
+            r();
+        }
+    }
+    // Start a consumer task for each generator
+    const tasks = generators.map(async (gen) => {
+        try {
+            for await (const item of gen) {
+                queue.push(item);
+                notify();
+            }
+        }
+        catch (e) {
+            if (firstError === undefined)
+                firstError = e;
+        }
+        finally {
+            activeCount--;
+            notify();
+        }
+    });
+    // Yield items as they arrive
+    while (activeCount > 0 || queue.length > 0) {
+        if (queue.length > 0) {
+            yield queue.shift();
+        }
+        else if (activeCount > 0) {
+            // Set resolve BEFORE checking queue again to avoid race condition
+            // where an item is pushed between the check and the await.
+            const waitPromise = new Promise((r) => {
+                resolve = r;
+            });
+            // Re-check after setting resolve — item may have arrived between
+            // the outer check and setting resolve.
+            if (queue.length > 0) {
+                resolve = null;
+                continue;
+            }
+            await waitPromise;
+        }
+    }
+    // Wait for all producer promises to settle
+    await Promise.allSettled(tasks);
+    if (firstError !== undefined) {
+        throw firstError;
+    }
+}
+//# sourceMappingURL=task-executor.js.map

package/dist/task-executor.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"task-executor.js","sourceRoot":"","sources":["../src/task-executor.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;GAaG;AAEH,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AAGzC,OAAO,EAAE,YAAY,EAAE,MAAM,qBAAqB,CAAC;AAGnD,MAAM,GAAG,GAAG,YAAY,CAAC,+BAA+B,CAAC,CAAC;AAC1D,OAAO,EAAE,YAAY,EAAE,MAAM,oBAAoB,CAAC;AAGlD,OAAO,EAAE,mBAAmB,EAAE,MAAM,gBAAgB,CAAC;AAErD,MAAM,iBAAiB,GAAG,EAAE,CAAC;AAC7B,MAAM,2BAA2B,GAAG,EAAE,CAAC;AAmBvC,MAAM,OAAO,YAAY;IACf,QAAQ,CAAe;IACvB,KAAK,CAAS;IACd,KAAK,CAAS;IACd,IAAI,CAAO;IACX,OAAO,CAAoB;IAC3B,MAAM,CAA0B;IAChC,YAAY,CAAqB;IACjC,QAAQ,CAAS;IACjB,iBAAiB,CAAS;IAC1B,aAAa,CAAS;IACtB,WAAW,CAAqB;IAChC,iBAAiB,CAAU;IAC3B,aAAa,CAAqB;IAE1C,YAAY,IAAyB;QACnC,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAC,QAAQ,CAAC;QAC9B,IAAI,CAAC,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC;QACxB,IAAI,CAAC,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC;QACxB,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC,IAAI,CAAC;QACtB,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC;QAC5B,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC,MAAM,IAAI,EAAE,CAAC;QAChC,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC,YAAY,CAAC;QACtC,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAC,QAAQ,IAAI,iBAAiB,CAAC;QACnD,IAAI,CAAC,iBAAiB;YACpB,IAAI,CAAC,iBAAiB,IAAI,2BAA2B,CAAC;QACxD,IAAI,CAAC,aAAa,GAAG,IAAI,CAAC,aAAa,IAAI,mBAAmB,CAAC;QAC/D,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC,WAAW,CAAC;QACpC,IAAI,CAAC,iBAAiB,GAAG,IAAI,CAAC,iBAAiB,IAAI,KAAK,CAAC;IAC3D,CAAC;IAED;;;OAGG;IACH,KAAK,CAAC,CAAC,YAAY;QACjB,2BAA2B;QAC3B,KAAK,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC;YACvD,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,GAAG,EAAE,KAAK,CAAC,CAAC;QAC/B,CAAC;QAED,uDAAuD;QACvD,IAAI,CAAC,aAAa;YAChB,IAAI,CAAC,WAAW;gBAChB,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC;oBACzB,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,EAAE;oBAChD,CAAC,CAAC,SAAS,CAAC,CAAC;QAEjB,GAAG,CAAC,IAAI,CAAC,wBAAwB,EAAE;YACjC,KAAK,EAAE,IAAI,CAAC,IAAI,CAAC,KAAK;YACtB,KAAK,EAAE,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM;SAC9B,CAAC,CAAC;QAEH,IAAI,UAAU,GAAG,CAAC,CAAC;QAEnB,OAAO,CAAC,IAAI,CAAC,gBAAgB,EAAE,IAAI,UAAU,GAAG,IAAI,CAAC,QAAQ,EAAE,CAAC;YAC9D,UAAU,EAAE,CAAC;YAEb,IAAI,eAAe,GAAG,IAAI,CAAC,kBAAkB,EAAE,CAAC;YAChD,eAAe,GAAG,IAAI,CAAC,oBAAoB,CAAC,eAAe,CAAC,CAAC;YAE7D,IAAI,eAAe,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBACjC,IAAI,CAAC,IAAI,CAAC,gBAAgB,EAAE,EAAE,CAAC;oBAC7B,MAAM;wBACJ,IAAI,EAAE,OAAO;wBACb,OAAO,EACL,2EAA2E;wBAC7E,IAAI,EAAE,KAAK;qBACI,CAAC;gBACpB,CAAC;gBACD,MAAM;YACR,CAAC;YAED,GAAG,CAAC,KAAK,CAAC,mBAAmB,EAAE;gBAC7B,OAAO,EAAE,eAAe,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;aAC1C,CAAC,CAAC;YAEH,gDAAgD;YAChD,MAAM,YAAY,GAAG,eAAe,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,SAAS,CAAC,CAAC;YACzE,MAAM,WAAW,GAAG,eAAe,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,SAAS,CAAC,CAAC;YAExE,yCAAyC;YACzC,KAAK,MAAM,KAAK,IAAI,YAAY,EAAE,CAAC;gBACjC,KAAK,CAAC,CAAC,IAAI,CAAC,iBAAiB,CAAC,KAAK,CAAC,CAAC;YACvC,CAAC;YAED,yCAAyC;YACzC,MAAM,cAAc,GAAG,WAAW,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE;gBAC9C,MAAM,QAAQ,GAAG,IAAI,YAAY,CAAC;oBAChC,IAAI,EAAE,IAAI,CAAC,IAAI;oBACf,IAAI;oBACJ,OAAO,EAAE,IAAI,CAAC,OAAO;oBACrB,QAAQ,EAAE,IAAI,CAAC,QAAQ;oBACvB,KAAK,EAAE,IAAI,CAAC,KAAK;oBACjB,KAAK,EAAE,CAAC,GAAG,IAAI,CAAC,KAAK,CAAC;oBACtB,YAAY,EAAE,IAAI,CAAC,YAAY;oBAC/B,aAAa,EAAE,IAAI,CAAC,aAAa;oBACjC,aAAa,EAAE,IAAI,CAAC,iBAAiB;oBACrC,aAAa,EAAE,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC;iBACvC,CAAC,CAAC;gBACH,OAAO,QAAQ,CAAC,OAAO,EAAE,CAAC;YAC5B,CAAC,CAAC,CAAC;YAEH,IAAI,IAAI,CAAC,iBAAiB,IAAI,cAAc,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBACxD,2DAA2D;gBAC3D,KAAK,CAAC,CAAC,oBAAoB,CAAC,cAAc,CAAC,CAAC;YAC9C,CAAC;iBAAM,CAAC;gBACN,6BAA6B;gBAC7B,KAAK,MAAM,SAAS,IAAI,cAAc,EAAE,CAAC;oBACvC,IAAI,KAAK,EAAE,MAAM,OAAO,IAAI,SAAS,EAAE,CAAC;wBACtC,MAAM,OAAO,CAAC;oBAChB,CAAC;gBACH,CAAC;YACH,CAAC;QACH,CAAC;IACH,CAAC;IAED;;OAEG;IACK,SAAS,CAAC,KAAc;QAC9B,MAAM,IAAI,GAAG,IAAI,CAAC,SAAS,CAAC,KAAK,EAAE,CAAC,IAAI,EAAE,GAAG,EAAE,EAAE;YAC/C,IAAI,GAAG,IAAI,IAAI,IAAI,OAAO,GAAG,KAAK,QAAQ,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC,EAAE,CAAC;gBAClE,MAAM,MAAM,GAA4B,EAAE,CAAC;gBAC3C,KAAK,MAAM,CAAC,IAAI,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC;oBACxC,MAAM,CAAC,CAAC,CAAC,GAAG,GAAG,CAAC,CAAC,CAAC,CAAC;gBACrB,CAAC;gBACD,OAAO,MAAM,CAAC;YAChB,CAAC;YACD,OAAO,GAAG,CAAC;QACb,CAAC,CAAC,CAAC;QACH,OAAO,UAAU,CAAC,MAAM,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;IACpE,CAAC;IAED;;;;OAIG;IACK,KAAK,CAAC,CAAC,iBAAiB,CAC9B,IAAU;QAEV,MAAM,cAAc,GAAG,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC;QACzC,IAAI,CAAC,cAAc,EAAE,CAAC;YACpB,GAAG,CAAC,IAAI,CAAC,oDAAoD,EAAE;gBAC7D,MAAM,EAAE,IAAI,CAAC,EAAE;aAChB,CAAC,CAAC;YACH,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC;YACtB,OAAO;QACT,CAAC;QAED,IAAI,cAAc,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,cAAc,CAAC,CAAC;QACtD,IAAI,cAAc,KAAK,SAAS,IAAI,cAAc,KAAK,IAAI,EAAE,CAAC;YAC5D,GAAG,CAAC,IAAI,CAAC,0DAA0D,EAAE;gBACnE,MAAM,EAAE,IAAI,CAAC,EAAE;aAChB,CAAC,CAAC;YACH,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC;YACtB,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC;YAC9B,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;YAC1B,OAAO;QACT,CAAC;QACD,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,cAAc,CAAC,EAAE,CAAC;YACnC,GAAG,CAAC,IAAI,CACN,oEAAoE,EACpE;gBACE,MAAM,EAAE,IAAI,CAAC,EAAE;gBACf,UAAU,EAAE,OAAO,cAAc;aAClC,CACF,CAAC;YACF,cAAc,GAAG,CAAC,cAAc,CAAC,CAAC;QACpC,CAAC;QAED,MAAM,KAAK,GAAG,cAA2B,CAAC;QAC1C,MAAM,QAAQ,GAAG,IAAI,CAAC,mBAAmB,IAAI,IAAI,CAAC,YAAY,CAAC;QAC/D,MAAM,aAAa,GAAG,IAAI,CAAC,aAAa,CAAC;QAEzC,GAAG,CAAC,IAAI,CAAC,oBAAoB,EAAE;YAC7B,MAAM,EAAE,IAAI,CAAC,EAAE;YACf,SAAS,EAAE,KAAK,CAAC,MAAM;SACxB,CAAC,CAAC;QAEH,uCAAuC;QACvC,MAAM,cAAc,GAAW,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE;YAChD,IAAI,YAAY,GAAG,QAAQ,CAAC;YAC5B,IAAI,OAAO,IAAI,KAAK,QAAQ,IAAI,IAAI,KAAK,IAAI,EAAE,CAAC;gBAC9C,KAAK,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CACvC,IAA+B,CAChC,EAAE,CAAC;oBACF,MAAM,UAAU,GAAG,GAAG,CAAC,OAAO,CAAC,qBAAqB,EAAE,MAAM,CAAC,CAAC;oBAC9D,MAAM,QAAQ,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC;oBAC/B,YAAY,GAAG,YAAY,CAAC,OAAO,CACjC,IAAI,MAAM,CAAC,MAAM,UAAU,KAAK,EAAE,GAAG,CAAC,EACtC,GAAG,EAAE,CAAC,QAAQ,CACf,CAAC;gBACJ,CAAC;YACH,CAAC;iBAAM,CAAC;gBACN,MAAM,OAAO,GAAG,MAAM,CAAC,IAAI,CAAC,CAAC;gBAC7B,YAAY,GAAG,YAAY,CAAC,OAAO,CAAC,WAAW,EAAE,GAAG,EAAE,CAAC,OAAO,CAAC,CAAC;YAClE,CAAC;YAED,MAAM,IAAI,GAAG,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC;YAClC,OAAO;gBACL,EAAE,EAAE,GAAG,IAAI,CAAC,EAAE,SAAS,IAAI,EAAE;gBAC7B,YAAY;gBACZ,SAAS,EAAE,KAAK;gBAChB,SAAS,EAAE,EAAE;gBACb,IAAI,EAAE,EAAE;gBACR,YAAY,EAAE,aAAa,IAAI,IAAI,CAAC,YAAY;aACzC,CAAC;QACZ,CAAC,CAAC,CAAC;QAEH,gDAAgD;QAChD,MAAM,UAAU,GAAG,cAAc,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE;YAChD,MAAM,QAAQ,GAAG,IAAI,YAAY,CAAC;gBAChC,IAAI,EAAE,IAAI,CAAC,IAAI;gBACf,IAAI,EAAE,OAAO;gBACb,OAAO,EAAE,IAAI,CAAC,OAAO;gBACrB,QAAQ,EAAE,IAAI,CAAC,QAAQ;gBACvB,KAAK,EAAE,IAAI,CAAC,KAAK;gBACjB,KAAK,EAAE,CAAC,GAAG,IAAI,CAAC,KAAK,CAAC;gBACtB,YAAY,EAAE,IAAI,CAAC,YAAY;gBAC/B,aAAa,EAAE,IAAI,CAAC,aAAa;gBACjC,aAAa,EAAE,IAAI,CAAC,iBAAiB;gBACrC,aAAa,EAAE,KAAK;aACrB,CAAC,CAAC;YACH,OAAO,QAAQ,CAAC,OAAO,EAAE,CAAC;QAC5B,CAAC,CAAC,CAAC;QAEH,8BAA8B;QAC9B,MAAM,OAAO,GAAc,EAAE,CAAC;QAE9B,IAAI,IAAI,CAAC,iBAAiB,IAAI,UAAU,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACpD,IAAI,KAAK,EAAE,MAAM,GAAG,IAAI,oBAAoB,CAAC,UAAU,CAAC,EAAE,CAAC;gBACzD,IAAK,GAAkB,CAAC,IAAI,KAAK,aAAa,EAAE,CAAC;oBAC/C,OAAO,CAAC,IAAI,CAAE,GAAkB,CAAC,MAAM,CAAC,CAAC;gBAC3C,CAAC;gBACD,MAAM,GAAG,CAAC;YACZ,CAAC;QACH,CAAC;aAAM,CAAC;YACN,KAAK,MAAM,GAAG,IAAI,UAAU,EAAE,CAAC;gBAC7B,IAAI,KAAK,EAAE,MAAM,GAAG,IAAI,GAAG,EAAE,CAAC;oBAC5B,IAAK,GAAkB,CAAC,IAAI,KAAK,aAAa,EAAE,CAAC;wBAC/C,OAAO,CAAC,IAAI,CAAE,GAAkB,CAAC,MAAM,CAAC,CAAC;oBAC3C,CAAC;oBACD,MAAM,GAAG,CAAC;gBACZ,CAAC;YACH,CAAC;QACH,CAAC;QAED,6CAA6C;QAC7C,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,EAAE,OAAO,CAAC,CAAC;QACnC,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC;QACtB,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAE1B,GAAG,CAAC,IAAI,CAAC,kBAAkB,EAAE;YAC3B,MAAM,EAAE,IAAI,CAAC,EAAE;YACf,WAAW,EAAE,OAAO,CAAC,MAAM;SAC5B,CAAC,CAAC;IACL,CAAC;IAED;;OAEG;IACK,gBAAgB;QACtB,OAAO,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;IACzD,CAAC;IAED;;OAEG;IACK,kBAAkB;QACxB,MAAM,YAAY,GAAG,IAAI,GAAG,CAC1B,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAC5D,CAAC;QACF,8CAA8C;QAC9C,KAAK,MAAM,GAAG,IAAI,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC;YAC3C,YAAY,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;QACxB,CAAC;QAED,OAAO,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,CAC3B,CAAC,IAAI,EAAE,EAAE,CACP,CAAC,IAAI,CAAC,SAAS;YACf,CAAC,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC;YACzB,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,YAAY,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC,CACvD,CAAC;IACJ,CAAC;IAED;;;OAGG;IACK,aAAa,CAAC,IAAU;QAC9B,OAAO,IAAI,CAAC,SAAS,IAAI,IAAI,IAAI,IAAI,CAAC,OAAO,IAAI,IAAI,CAAC;IACxD,CAAC;IAED;;OAEG;IACK,YAAY,CAAC,IAAU;QAC7B,IAAI,IAAI,CAAC,aAAa,EAAE,CAAC;YACvB,OAAO,IAAI,CAAC,EAAE,KAAK,IAAI,CAAC,aAAa,CAAC;QACxC,CAAC;QACD,OAAO,CACL,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC;YAC1B,IAAI,KAAK,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC,CACrD,CAAC;IACJ,CAAC;IAED;;;OAGG;IACK,oBAAoB,CAAC,eAAuB;QAClD,IAAI,CAAC,IAAI,CAAC,aAAa;YAAE,OAAO,eAAe,CAAC;QAEhD,MAAM,WAAW,GAAG,eAAe,CAAC,IAAI,CACtC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,IAAI,CAAC,aAAa,CACnC,CAAC;QACF,IAAI,CAAC,WAAW;YAAE,OAAO,eAAe,CAAC;QAEzC,MAAM,YAAY,GAAG,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CACvC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,SAAS,IAAI,CAAC,CAAC,EAAE,KAAK,IAAI,CAAC,aAAa,CACnD,CAAC;QACF,IAAI,CAAC,YAAY;YAAE,OAAO,eAAe,CAAC;QAE1C,OAAO,eAAe,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,IAAI,CAAC,aAAa,CAAC,CAAC;IACpE,CAAC;CACF;AAED,8EAA8E;AAC9E,4EAA4E;AAC5E,8EAA8E;AAE9E,KAAK,SAAS,CAAC,CAAC,oBAAoB,CAClC,UAA+B;IAE/B,0EAA0E;IAC1E,MAAM,KAAK,GAAQ,EAAE,CAAC;IACtB,IAAI,WAAW,GAAG,UAAU,CAAC,MAAM,CAAC;IACpC,IAAI,OAAO,GAAwB,IAAI,CAAC;IACxC,IAAI,UAAU,GAAY,SAAS,CAAC;IAEpC,SAAS,MAAM;QACb,IAAI,OAAO,EAAE,CAAC;YACZ,MAAM,CAAC,GAAG,OAAO,CAAC;YAClB,OAAO,GAAG,IAAI,CAAC;YACf,CAAC,EAAE,CAAC;QACN,CAAC;IACH,CAAC;IAED,2CAA2C;IAC3C,MAAM,KAAK,GAAG,UAAU,CAAC,GAAG,CAAC,KAAK,EAAE,GAAG,EAAE,EAAE;QACzC,IAAI,CAAC;YACH,IAAI,KAAK,EAAE,MAAM,IAAI,IAAI,GAAG,EAAE,CAAC;gBAC7B,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;gBACjB,MAAM,EAAE,CAAC;YACX,CAAC;QACH,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACX,IAAI,UAAU,KAAK,SAAS;gBAAE,UAAU,GAAG,CAAC,CAAC;QAC/C,CAAC;gBAAS,CAAC;YACT,WAAW,EAAE,CAAC;YACd,MAAM,EAAE,CAAC;QACX,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,6BAA6B;IAC7B,OAAO,WAAW,GAAG,CAAC,IAAI,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QAC3C,IAAI,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACrB,MAAM,KAAK,CAAC,KAAK,EAAG,CAAC;QACvB,CAAC;aAAM,IAAI,WAAW,GAAG,CAAC,EAAE,CAAC;YAC3B,kEAAkE;YAClE,2DAA2D;YAC3D,MAAM,WAAW,GAAG,IAAI,OAAO,CAAO,CAAC,CAAC,EAAE,EAAE;gBAC1C,OAAO,GAAG,CAAC,CAAC;YACd,CAAC,CAAC,CAAC;YACH,iEAAiE;YACjE,uCAAuC;YACvC,IAAI,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBACrB,OAAO,GAAG,IAAI,CAAC;gBACf,SAAS;YACX,CAAC;YACD,MAAM,WAAW,CAAC;QACpB,CAAC;IACH,CAAC;IAED,2CAA2C;IAC3C,MAAM,OAAO,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC;IAEhC,IAAI,UAAU,KAAK,SAAS,EAAE,CAAC;QAC7B,MAAM,UAAU,CAAC;IACnB,CAAC;AACH,CAAC"}

package/dist/task-planner.d.ts

@@ -0,0 +1,44 @@
+/**
+ * TaskPlanner -- decomposes an objective into a Task or TaskPlan via tool calls.
+ *
+ * planMultiTask() builds the plan incrementally: the LLM calls add_task one
+ * task at a time, with per-task validation. Progress is yielded as task_update
+ * events (TaskPlanned) so clients can watch the plan take shape in real time.
+ *
+ * plan() (single-task) keeps the original one-shot create_task tool.
+ */
+import type { BaseProvider, ProcessingContext } from "@nodetool-ai/runtime";
+import { type ProcessingMessage } from "@nodetool-ai/protocol";
+import type { Task, TaskPlan } from "./types.js";
+import type { Tool } from "./tools/base-tool.js";
+export interface TaskPlannerOptions {
+    provider: BaseProvider;
+    model: string;
+    reasoningModel?: string;
+    tools?: Tool[];
+    systemPrompt?: string;
+    outputSchema?: Record<string, unknown>;
+    inputs?: Record<string, unknown>;
+    maxRetries?: number;
+    threadId?: string;
+}
+export declare class TaskPlanner {
+    private provider;
+    private model;
+    readonly reasoningModel: string;
+    private tools;
+    private systemPrompt;
+    private outputSchema;
+    private inputs;
+    private maxRetries;
+    private threadId?;
+    constructor(opts: TaskPlannerOptions);
+    plan(objective: string, context: ProcessingContext): AsyncGenerator<ProcessingMessage, Task | null>;
+    private _planImpl;
+    planMultiTask(objective: string, context: ProcessingContext): AsyncGenerator<ProcessingMessage, TaskPlan | null>;
+    private _planMultiTaskImpl;
+    private callSingleTaskTool;
+    private taskPlannedEvent;
+    private formatToolsInfo;
+}
+//# sourceMappingURL=task-planner.d.ts.map

package/dist/task-planner.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"task-planner.d.ts","sourceRoot":"","sources":["../src/task-planner.ts"],"names":[],"mappings":"AAAA;;;;;;;;GAQG;AAEH,OAAO,KAAK,EACV,YAAY,EACZ,iBAAiB,EAGlB,MAAM,sBAAsB,CAAC;AAG9B,OAAO,EAEL,KAAK,iBAAiB,EAKvB,MAAM,uBAAuB,CAAC;AAG/B,OAAO,KAAK,EAAE,IAAI,EAAE,QAAQ,EAAQ,MAAM,YAAY,CAAC;AACvD,OAAO,KAAK,EAAE,IAAI,EAAE,MAAM,sBAAsB,CAAC;AA2GjD,MAAM,WAAW,kBAAkB;IACjC,QAAQ,EAAE,YAAY,CAAC;IACvB,KAAK,EAAE,MAAM,CAAC;IACd,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB,KAAK,CAAC,EAAE,IAAI,EAAE,CAAC;IACf,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,YAAY,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IACvC,MAAM,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IACjC,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,QAAQ,CAAC,EAAE,MAAM,CAAC;CACnB;AAED,qBAAa,WAAW;IACtB,OAAO,CAAC,QAAQ,CAAe;IAC/B,OAAO,CAAC,KAAK,CAAS;IACtB,QAAQ,CAAC,cAAc,EAAE,MAAM,CAAC;IAChC,OAAO,CAAC,KAAK,CAAS;IACtB,OAAO,CAAC,YAAY,CAAS;IAC7B,OAAO,CAAC,YAAY,CAAsC;IAC1D,OAAO,CAAC,MAAM,CAA0B;IACxC,OAAO,CAAC,UAAU,CAAS;IAC3B,OAAO,CAAC,QAAQ,CAAC,CAAS;gBAEd,IAAI,EAAE,kBAAkB;IAgB7B,IAAI,CACT,SAAS,EAAE,MAAM,EACjB,OAAO,EAAE,iBAAiB,GACzB,cAAc,CAAC,iBAAiB,EAAE,IAAI,GAAG,IAAI,CAAC;YAclC,SAAS;IAyGjB,aAAa,CAClB,SAAS,EAAE,MAAM,EACjB,OAAO,EAAE,iBAAiB,GACzB,cAAc,CAAC,iBAAiB,EAAE,QAAQ,GAAG,IAAI,CAAC;YActC,kBAAkB;YA6OlB,kBAAkB;IAmFjC,OAAO,CAAC,gBAAgB;IAkBxB,OAAO,CAAC,eAAe;CA0BxB"}