@nodetool-ai/agents 0.7.0-rc.11

package/dist/tools/pdf-tools.js

@@ -0,0 +1,339 @@
+/**
+ * PDF tools for extracting text, tables, and converting documents.
+ *
+ * Port of src/nodetool/agents/tools/pdf_tools.py
+ */
+import { readFile, writeFile, mkdir } from "node:fs/promises";
+import { dirname } from "node:path";
+import { execFile } from "node:child_process";
+import { promisify } from "node:util";
+import { Tool } from "./base-tool.js";
+const execFileAsync = promisify(execFile);
+/**
+ * Extract text from a PDF buffer using @llamaindex/liteparse.
+ * Returns per-page text so callers can slice by page range.
+ */
+async function extractPdfPages(buffer) {
+    const { LiteParse } = await import("@llamaindex/liteparse");
+    const parser = new LiteParse({ ocrEnabled: false });
+    const result = await parser.parse(buffer, true);
+    const pages = result.pages.map((p) => p.text);
+    return { pages, numPages: result.pages.length };
+}
+export class ExtractPDFTextTool extends Tool {
+    name = "extract_pdf_text";
+    description = "Extract plain text from a PDF document";
+    inputSchema = {
+        type: "object",
+        properties: {
+            path: {
+                type: "string",
+                description: "Path to the PDF file"
+            },
+            start_page: {
+                type: "integer",
+                description: "First page to extract (0-based index)",
+                default: 0
+            },
+            end_page: {
+                type: "integer",
+                description: "Last page to extract (-1 for last page)",
+                default: -1
+            }
+        },
+        required: ["path"]
+    };
+    async process(context, params) {
+        try {
+            const filePath = context.resolveWorkspacePath(params["path"]);
+            const buffer = await readFile(filePath);
+            const { pages } = await extractPdfPages(buffer);
+            const startPage = params["start_page"] ?? 0;
+            const endPage = params["end_page"] ?? -1;
+            if (startPage === 0 && endPage === -1) {
+                return { text: pages.join("\n") };
+            }
+            const end = endPage === -1 ? pages.length - 1 : endPage;
+            const selectedPages = pages.slice(startPage, end + 1);
+            return { text: selectedPages.join("\n") };
+        }
+        catch (e) {
+            return { error: String(e instanceof Error ? e.message : e) };
+        }
+    }
+    userMessage(params) {
+        const path = params["path"] ?? "a PDF";
+        const msg = `Extracting text from ${path}...`;
+        return msg.length > 80 ? "Extracting text from PDF..." : msg;
+    }
+}
+export class ExtractPDFTablesTool extends Tool {
+    name = "extract_pdf_tables";
+    description = "Extract tables from a PDF document to a JSON file";
+    inputSchema = {
+        type: "object",
+        properties: {
+            path: {
+                type: "string",
+                description: "Path to the PDF file"
+            },
+            output_file: {
+                type: "string",
+                description: "Path to the output JSON file"
+            },
+            start_page: {
+                type: "integer",
+                description: "First page to extract (0-based index)",
+                default: 0
+            },
+            end_page: {
+                type: "integer",
+                description: "Last page to extract (-1 for last page)",
+                default: -1
+            }
+        },
+        required: ["path", "output_file"]
+    };
+    async process(context, params) {
+        try {
+            const filePath = context.resolveWorkspacePath(params["path"]);
+            const outputFile = context.resolveWorkspacePath(params["output_file"]);
+            const buffer = await readFile(filePath);
+            const { pages } = await extractPdfPages(buffer);
+            const startPage = params["start_page"] ?? 0;
+            const endPage = params["end_page"] ?? -1;
+            const end = endPage === -1 ? pages.length - 1 : endPage;
+            // Best-effort table extraction: look for lines with consistent
+            // column-like separators (multiple spaces or tabs)
+            const allTables = [];
+            for (let pageNum = startPage; pageNum <= end && pageNum < pages.length; pageNum++) {
+                const pageText = pages[pageNum] ?? "";
+                const lines = pageText
+                    .split("\n")
+                    .filter((l) => l.trim().length > 0);
+                // Detect tabular rows: lines with 2+ segments separated by 2+ spaces
+                const tabularLines = [];
+                for (const line of lines) {
+                    const cells = line
+                        .split(/\s{2,}/)
+                        .filter((c) => c.trim().length > 0);
+                    if (cells.length >= 2) {
+                        tabularLines.push(cells);
+                    }
+                }
+                // Group consecutive tabular lines into tables
+                if (tabularLines.length >= 2) {
+                    allTables.push({
+                        page: pageNum,
+                        rows: tabularLines.length,
+                        columns: Math.max(...tabularLines.map((r) => r.length)),
+                        content: tabularLines
+                    });
+                }
+            }
+            const parentDir = dirname(outputFile);
+            await mkdir(parentDir, { recursive: true });
+            await writeFile(outputFile, JSON.stringify(allTables), "utf-8");
+            return { output_file: outputFile };
+        }
+        catch (e) {
+            return { error: String(e instanceof Error ? e.message : e) };
+        }
+    }
+    userMessage(params) {
+        const path = params["path"] ?? "a PDF";
+        const output = params["output_file"] ?? "output";
+        let msg = `Extracting tables from ${path} to ${output}...`;
+        if (msg.length > 80)
+            msg = `Extracting tables from PDF to ${output}...`;
+        if (msg.length > 80)
+            msg = "Extracting tables from PDF...";
+        return msg;
+    }
+}
+export class ConvertPDFToMarkdownTool extends Tool {
+    name = "convert_pdf_to_markdown";
+    description = "Convert PDF to Markdown format";
+    inputSchema = {
+        type: "object",
+        properties: {
+            input_file: {
+                type: "string",
+                description: "Path to the input PDF file"
+            },
+            output_file: {
+                type: "string",
+                description: "Path to the output Markdown file"
+            },
+            start_page: {
+                type: "integer",
+                description: "First page to extract (0-based index)",
+                default: 0
+            },
+            end_page: {
+                type: "integer",
+                description: "Last page to extract (-1 for last page)",
+                default: -1
+            }
+        },
+        required: ["input_file", "output_file"]
+    };
+    async process(context, params) {
+        try {
+            const inputFile = context.resolveWorkspacePath(params["input_file"]);
+            const outputFile = context.resolveWorkspacePath(params["output_file"]);
+            const buffer = await readFile(inputFile);
+            const { pages } = await extractPdfPages(buffer);
+            const startPage = params["start_page"] ?? 0;
+            const endPage = params["end_page"] ?? -1;
+            let mdText;
+            if (startPage !== 0 || endPage !== -1) {
+                const end = endPage === -1 ? pages.length - 1 : endPage;
+                mdText = pages.slice(startPage, end + 1).join("\n");
+            }
+            else {
+                mdText = pages.join("\n");
+            }
+            const parentDir = dirname(outputFile);
+            await mkdir(parentDir, { recursive: true });
+            await writeFile(outputFile, mdText, "utf-8");
+            return { output_file: outputFile };
+        }
+        catch (e) {
+            return { error: String(e instanceof Error ? e.message : e) };
+        }
+    }
+    userMessage(params) {
+        const inputFile = params["input_file"] ?? "a PDF";
+        const outputFile = params["output_file"] ?? "Markdown";
+        let msg = `Converting ${inputFile} to ${outputFile}...`;
+        if (msg.length > 80)
+            msg = `Converting PDF to ${outputFile}...`;
+        if (msg.length > 80)
+            msg = "Converting PDF to Markdown...";
+        return msg;
+    }
+}
+export class ConvertMarkdownToPDFTool extends Tool {
+    name = "convert_markdown_to_pdf";
+    description = "Convert Markdown to PDF using Pandoc.";
+    inputSchema = {
+        type: "object",
+        properties: {
+            input_file: {
+                type: "string",
+                description: "Path to the input Markdown file"
+            },
+            output_file: {
+                type: "string",
+                description: "Path to the output PDF file"
+            }
+        },
+        required: ["input_file", "output_file"]
+    };
+    async process(context, params) {
+        try {
+            const inputFile = context.resolveWorkspacePath(params["input_file"]);
+            const outputFile = context.resolveWorkspacePath(params["output_file"]);
+            const parentDir = dirname(outputFile);
+            await mkdir(parentDir, { recursive: true });
+            await execFileAsync("pandoc", [
+                inputFile,
+                "-f",
+                "markdown",
+                "-t",
+                "pdf",
+                "-o",
+                outputFile
+            ]);
+            return { output_file: outputFile, status: "success" };
+        }
+        catch (e) {
+            return { error: String(e instanceof Error ? e.message : e) };
+        }
+    }
+    userMessage(params) {
+        const inputFile = params["input_file"] ?? "Markdown";
+        const outputFile = params["output_file"] ?? "a PDF";
+        let msg = `Converting ${inputFile} to ${outputFile}...`;
+        if (msg.length > 80)
+            msg = `Converting Markdown to ${outputFile}...`;
+        if (msg.length > 80)
+            msg = "Converting Markdown to PDF...";
+        return msg;
+    }
+}
+export class ConvertDocumentTool extends Tool {
+    name = "convert_document";
+    description = "Convert between document formats using Pandoc, supports markdown, docx, rst, pdf, html, etc.";
+    inputSchema = {
+        type: "object",
+        properties: {
+            input_file: {
+                type: "string",
+                description: "Path to the input file"
+            },
+            output_file: {
+                type: "string",
+                description: "Path to the output file"
+            },
+            from_format: {
+                type: "string",
+                description: "Input format (e.g., markdown, docx, rst)",
+                default: "markdown"
+            },
+            to_format: {
+                type: "string",
+                description: "Output format (e.g., pdf, docx, html)",
+                default: "pdf"
+            },
+            extra_args: {
+                type: "array",
+                description: "Additional Pandoc arguments",
+                items: { type: "string" },
+                default: []
+            }
+        },
+        required: ["input_file", "output_file"]
+    };
+    async process(context, params) {
+        try {
+            const inputFile = context.resolveWorkspacePath(params["input_file"]);
+            const outputFile = context.resolveWorkspacePath(params["output_file"]);
+            const fromFormat = params["from_format"] ?? "markdown";
+            const toFormat = params["to_format"] ?? "pdf";
+            let extraArgs = params["extra_args"] ?? [];
+            if (typeof extraArgs === "string") {
+                extraArgs = [extraArgs];
+            }
+            const parentDir = dirname(outputFile);
+            await mkdir(parentDir, { recursive: true });
+            await execFileAsync("pandoc", [
+                inputFile,
+                "-f",
+                fromFormat,
+                "-t",
+                toFormat,
+                "-o",
+                outputFile,
+                ...extraArgs
+            ]);
+            return { output_file: outputFile, status: "success" };
+        }
+        catch (e) {
+            return { error: String(e instanceof Error ? e.message : e) };
+        }
+    }
+    userMessage(params) {
+        const inputFile = params["input_file"] ?? "input";
+        const outputFile = params["output_file"] ?? "output";
+        const toFormat = params["to_format"] ?? "target format";
+        let msg = `Converting ${inputFile} to ${outputFile} (${toFormat})...`;
+        if (msg.length > 80)
+            msg = `Converting ${inputFile} to ${toFormat}...`;
+        if (msg.length > 80)
+            msg = `Converting document to ${toFormat}...`;
+        return msg;
+    }
+}
+//# sourceMappingURL=pdf-tools.js.map

package/dist/tools/pdf-tools.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"pdf-tools.js","sourceRoot":"","sources":["../../src/tools/pdf-tools.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,QAAQ,EAAE,SAAS,EAAE,KAAK,EAAE,MAAM,kBAAkB,CAAC;AAC9D,OAAO,EAAE,OAAO,EAAE,MAAM,WAAW,CAAC;AACpC,OAAO,EAAE,QAAQ,EAAE,MAAM,oBAAoB,CAAC;AAC9C,OAAO,EAAE,SAAS,EAAE,MAAM,WAAW,CAAC;AAEtC,OAAO,EAAE,IAAI,EAAE,MAAM,gBAAgB,CAAC;AAEtC,MAAM,aAAa,GAAG,SAAS,CAAC,QAAQ,CAAC,CAAC;AAY1C;;;GAGG;AACH,KAAK,UAAU,eAAe,CAAC,MAAc;IAC3C,MAAM,EAAE,SAAS,EAAE,GAAG,MAAM,MAAM,CAAC,uBAAuB,CAAC,CAAC;IAC5D,MAAM,MAAM,GAAG,IAAI,SAAS,CAAC,EAAE,UAAU,EAAE,KAAK,EAAE,CAAC,CAAC;IACpD,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,KAAK,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC;IAChD,MAAM,KAAK,GAAG,MAAM,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;IAC9C,OAAO,EAAE,KAAK,EAAE,QAAQ,EAAE,MAAM,CAAC,KAAK,CAAC,MAAM,EAAE,CAAC;AAClD,CAAC;AAED,MAAM,OAAO,kBAAmB,SAAQ,IAAI;IACjC,IAAI,GAAG,kBAAkB,CAAC;IAC1B,WAAW,GAAG,wCAAwC,CAAC;IACvD,WAAW,GAAG;QACrB,IAAI,EAAE,QAAiB;QACvB,UAAU,EAAE;YACV,IAAI,EAAE;gBACJ,IAAI,EAAE,QAAiB;gBACvB,WAAW,EAAE,sBAAsB;aACpC;YACD,UAAU,EAAE;gBACV,IAAI,EAAE,SAAkB;gBACxB,WAAW,EAAE,uCAAuC;gBACpD,OAAO,EAAE,CAAC;aACX;YACD,QAAQ,EAAE;gBACR,IAAI,EAAE,SAAkB;gBACxB,WAAW,EAAE,yCAAyC;gBACtD,OAAO,EAAE,CAAC,CAAC;aACZ;SACF;QACD,QAAQ,EAAE,CAAC,MAAM,CAAC;KACnB,CAAC;IAEF,KAAK,CAAC,OAAO,CACX,OAA0B,EAC1B,MAA+B;QAE/B,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,OAAO,CAAC,oBAAoB,CAAC,MAAM,CAAC,MAAM,CAAW,CAAC,CAAC;YACxE,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,QAAQ,CAAC,CAAC;YACxC,MAAM,EAAE,KAAK,EAAE,GAAG,MAAM,eAAe,CAAC,MAAM,CAAC,CAAC;YAEhD,MAAM,SAAS,GAAI,MAAM,CAAC,YAAY,CAAY,IAAI,CAAC,CAAC;YACxD,MAAM,OAAO,GAAI,MAAM,CAAC,UAAU,CAAY,IAAI,CAAC,CAAC,CAAC;YAErD,IAAI,SAAS,KAAK,CAAC,IAAI,OAAO,KAAK,CAAC,CAAC,EAAE,CAAC;gBACtC,OAAO,EAAE,IAAI,EAAE,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC;YACpC,CAAC;YAED,MAAM,GAAG,GAAG,OAAO,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC;YACxD,MAAM,aAAa,GAAG,KAAK,CAAC,KAAK,CAAC,SAAS,EAAE,GAAG,GAAG,CAAC,CAAC,CAAC;YACtD,OAAO,EAAE,IAAI,EAAE,aAAa,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC;QAC5C,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACX,OAAO,EAAE,KAAK,EAAE,MAAM,CAAC,CAAC,YAAY,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;QAC/D,CAAC;IACH,CAAC;IAED,WAAW,CAAC,MAA+B;QACzC,MAAM,IAAI,GAAG,MAAM,CAAC,MAAM,CAAC,IAAI,OAAO,CAAC;QACvC,MAAM,GAAG,GAAG,wBAAwB,IAAI,KAAK,CAAC;QAC9C,OAAO,GAAG,CAAC,MAAM,GAAG,EAAE,CAAC,CAAC,CAAC,6BAA6B,CAAC,CAAC,CAAC,GAAG,CAAC;IAC/D,CAAC;CACF;AAED,MAAM,OAAO,oBAAqB,SAAQ,IAAI;IACnC,IAAI,GAAG,oBAAoB,CAAC;IAC5B,WAAW,GAAG,mDAAmD,CAAC;IAClE,WAAW,GAAG;QACrB,IAAI,EAAE,QAAiB;QACvB,UAAU,EAAE;YACV,IAAI,EAAE;gBACJ,IAAI,EAAE,QAAiB;gBACvB,WAAW,EAAE,sBAAsB;aACpC;YACD,WAAW,EAAE;gBACX,IAAI,EAAE,QAAiB;gBACvB,WAAW,EAAE,8BAA8B;aAC5C;YACD,UAAU,EAAE;gBACV,IAAI,EAAE,SAAkB;gBACxB,WAAW,EAAE,uCAAuC;gBACpD,OAAO,EAAE,CAAC;aACX;YACD,QAAQ,EAAE;gBACR,IAAI,EAAE,SAAkB;gBACxB,WAAW,EAAE,yCAAyC;gBACtD,OAAO,EAAE,CAAC,CAAC;aACZ;SACF;QACD,QAAQ,EAAE,CAAC,MAAM,EAAE,aAAa,CAAC;KAClC,CAAC;IAEF,KAAK,CAAC,OAAO,CACX,OAA0B,EAC1B,MAA+B;QAE/B,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,OAAO,CAAC,oBAAoB,CAAC,MAAM,CAAC,MAAM,CAAW,CAAC,CAAC;YACxE,MAAM,UAAU,GAAG,OAAO,CAAC,oBAAoB,CAC7C,MAAM,CAAC,aAAa,CAAW,CAChC,CAAC;YACF,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,QAAQ,CAAC,CAAC;YACxC,MAAM,EAAE,KAAK,EAAE,GAAG,MAAM,eAAe,CAAC,MAAM,CAAC,CAAC;YAEhD,MAAM,SAAS,GAAI,MAAM,CAAC,YAAY,CAAY,IAAI,CAAC,CAAC;YACxD,MAAM,OAAO,GAAI,MAAM,CAAC,UAAU,CAAY,IAAI,CAAC,CAAC,CAAC;YAErD,MAAM,GAAG,GAAG,OAAO,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC;YAExD,+DAA+D;YAC/D,mDAAmD;YACnD,MAAM,SAAS,GAAmC,EAAE,CAAC;YACrD,KACE,IAAI,OAAO,GAAG,SAAS,EACvB,OAAO,IAAI,GAAG,IAAI,OAAO,GAAG,KAAK,CAAC,MAAM,EACxC,OAAO,EAAE,EACT,CAAC;gBACD,MAAM,QAAQ,GAAG,KAAK,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC;gBACtC,MAAM,KAAK,GAAG,QAAQ;qBACnB,KAAK,CAAC,IAAI,CAAC;qBACX,MAAM,CAAC,CAAC,CAAS,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;gBAE9C,qEAAqE;gBACrE,MAAM,YAAY,GAAe,EAAE,CAAC;gBACpC,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;oBACzB,MAAM,KAAK,GAAG,IAAI;yBACf,KAAK,CAAC,QAAQ,CAAC;yBACf,MAAM,CAAC,CAAC,CAAS,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;oBAC9C,IAAI,KAAK,CAAC,MAAM,IAAI,CAAC,EAAE,CAAC;wBACtB,YAAY,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;oBAC3B,CAAC;gBACH,CAAC;gBAED,8CAA8C;gBAC9C,IAAI,YAAY,CAAC,MAAM,IAAI,CAAC,EAAE,CAAC;oBAC7B,SAAS,CAAC,IAAI,CAAC;wBACb,IAAI,EAAE,OAAO;wBACb,IAAI,EAAE,YAAY,CAAC,MAAM;wBACzB,OAAO,EAAE,IAAI,CAAC,GAAG,CAAC,GAAG,YAAY,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC;wBACvD,OAAO,EAAE,YAAY;qBACtB,CAAC,CAAC;gBACL,CAAC;YACH,CAAC;YAED,MAAM,SAAS,GAAG,OAAO,CAAC,UAAU,CAAC,CAAC;YACtC,MAAM,KAAK,CAAC,SAAS,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;YAC5C,MAAM,SAAS,CAAC,UAAU,EAAE,IAAI,CAAC,SAAS,CAAC,SAAS,CAAC,EAAE,OAAO,CAAC,CAAC;YAEhE,OAAO,EAAE,WAAW,EAAE,UAAU,EAAE,CAAC;QACrC,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACX,OAAO,EAAE,KAAK,EAAE,MAAM,CAAC,CAAC,YAAY,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;QAC/D,CAAC;IACH,CAAC;IAED,WAAW,CAAC,MAA+B;QACzC,MAAM,IAAI,GAAG,MAAM,CAAC,MAAM,CAAC,IAAI,OAAO,CAAC;QACvC,MAAM,MAAM,GAAG,MAAM,CAAC,aAAa,CAAC,IAAI,QAAQ,CAAC;QACjD,IAAI,GAAG,GAAG,0BAA0B,IAAI,OAAO,MAAM,KAAK,CAAC;QAC3D,IAAI,GAAG,CAAC,MAAM,GAAG,EAAE;YAAE,GAAG,GAAG,iCAAiC,MAAM,KAAK,CAAC;QACxE,IAAI,GAAG,CAAC,MAAM,GAAG,EAAE;YAAE,GAAG,GAAG,+BAA+B,CAAC;QAC3D,OAAO,GAAG,CAAC;IACb,CAAC;CACF;AAED,MAAM,OAAO,wBAAyB,SAAQ,IAAI;IACvC,IAAI,GAAG,yBAAyB,CAAC;IACjC,WAAW,GAAG,gCAAgC,CAAC;IAC/C,WAAW,GAAG;QACrB,IAAI,EAAE,QAAiB;QACvB,UAAU,EAAE;YACV,UAAU,EAAE;gBACV,IAAI,EAAE,QAAiB;gBACvB,WAAW,EAAE,4BAA4B;aAC1C;YACD,WAAW,EAAE;gBACX,IAAI,EAAE,QAAiB;gBACvB,WAAW,EAAE,kCAAkC;aAChD;YACD,UAAU,EAAE;gBACV,IAAI,EAAE,SAAkB;gBACxB,WAAW,EAAE,uCAAuC;gBACpD,OAAO,EAAE,CAAC;aACX;YACD,QAAQ,EAAE;gBACR,IAAI,EAAE,SAAkB;gBACxB,WAAW,EAAE,yCAAyC;gBACtD,OAAO,EAAE,CAAC,CAAC;aACZ;SACF;QACD,QAAQ,EAAE,CAAC,YAAY,EAAE,aAAa,CAAC;KACxC,CAAC;IAEF,KAAK,CAAC,OAAO,CACX,OAA0B,EAC1B,MAA+B;QAE/B,IAAI,CAAC;YACH,MAAM,SAAS,GAAG,OAAO,CAAC,oBAAoB,CAC5C,MAAM,CAAC,YAAY,CAAW,CAC/B,CAAC;YACF,MAAM,UAAU,GAAG,OAAO,CAAC,oBAAoB,CAC7C,MAAM,CAAC,aAAa,CAAW,CAChC,CAAC;YAEF,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,SAAS,CAAC,CAAC;YACzC,MAAM,EAAE,KAAK,EAAE,GAAG,MAAM,eAAe,CAAC,MAAM,CAAC,CAAC;YAEhD,MAAM,SAAS,GAAI,MAAM,CAAC,YAAY,CAAY,IAAI,CAAC,CAAC;YACxD,MAAM,OAAO,GAAI,MAAM,CAAC,UAAU,CAAY,IAAI,CAAC,CAAC,CAAC;YAErD,IAAI,MAAc,CAAC;YACnB,IAAI,SAAS,KAAK,CAAC,IAAI,OAAO,KAAK,CAAC,CAAC,EAAE,CAAC;gBACtC,MAAM,GAAG,GAAG,OAAO,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC;gBACxD,MAAM,GAAG,KAAK,CAAC,KAAK,CAAC,SAAS,EAAE,GAAG,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACtD,CAAC;iBAAM,CAAC;gBACN,MAAM,GAAG,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAC5B,CAAC;YAED,MAAM,SAAS,GAAG,OAAO,CAAC,UAAU,CAAC,CAAC;YACtC,MAAM,KAAK,CAAC,SAAS,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;YAC5C,MAAM,SAAS,CAAC,UAAU,EAAE,MAAM,EAAE,OAAO,CAAC,CAAC;YAE7C,OAAO,EAAE,WAAW,EAAE,UAAU,EAAE,CAAC;QACrC,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACX,OAAO,EAAE,KAAK,EAAE,MAAM,CAAC,CAAC,YAAY,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;QAC/D,CAAC;IACH,CAAC;IAED,WAAW,CAAC,MAA+B;QACzC,MAAM,SAAS,GAAG,MAAM,CAAC,YAAY,CAAC,IAAI,OAAO,CAAC;QAClD,MAAM,UAAU,GAAG,MAAM,CAAC,aAAa,CAAC,IAAI,UAAU,CAAC;QACvD,IAAI,GAAG,GAAG,cAAc,SAAS,OAAO,UAAU,KAAK,CAAC;QACxD,IAAI,GAAG,CAAC,MAAM,GAAG,EAAE;YAAE,GAAG,GAAG,qBAAqB,UAAU,KAAK,CAAC;QAChE,IAAI,GAAG,CAAC,MAAM,GAAG,EAAE;YAAE,GAAG,GAAG,+BAA+B,CAAC;QAC3D,OAAO,GAAG,CAAC;IACb,CAAC;CACF;AAED,MAAM,OAAO,wBAAyB,SAAQ,IAAI;IACvC,IAAI,GAAG,yBAAyB,CAAC;IACjC,WAAW,GAAG,uCAAuC,CAAC;IACtD,WAAW,GAAG;QACrB,IAAI,EAAE,QAAiB;QACvB,UAAU,EAAE;YACV,UAAU,EAAE;gBACV,IAAI,EAAE,QAAiB;gBACvB,WAAW,EAAE,iCAAiC;aAC/C;YACD,WAAW,EAAE;gBACX,IAAI,EAAE,QAAiB;gBACvB,WAAW,EAAE,6BAA6B;aAC3C;SACF;QACD,QAAQ,EAAE,CAAC,YAAY,EAAE,aAAa,CAAC;KACxC,CAAC;IAEF,KAAK,CAAC,OAAO,CACX,OAA0B,EAC1B,MAA+B;QAE/B,IAAI,CAAC;YACH,MAAM,SAAS,GAAG,OAAO,CAAC,oBAAoB,CAC5C,MAAM,CAAC,YAAY,CAAW,CAC/B,CAAC;YACF,MAAM,UAAU,GAAG,OAAO,CAAC,oBAAoB,CAC7C,MAAM,CAAC,aAAa,CAAW,CAChC,CAAC;YAEF,MAAM,SAAS,GAAG,OAAO,CAAC,UAAU,CAAC,CAAC;YACtC,MAAM,KAAK,CAAC,SAAS,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;YAE5C,MAAM,aAAa,CAAC,QAAQ,EAAE;gBAC5B,SAAS;gBACT,IAAI;gBACJ,UAAU;gBACV,IAAI;gBACJ,KAAK;gBACL,IAAI;gBACJ,UAAU;aACX,CAAC,CAAC;YAEH,OAAO,EAAE,WAAW,EAAE,UAAU,EAAE,MAAM,EAAE,SAAS,EAAE,CAAC;QACxD,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACX,OAAO,EAAE,KAAK,EAAE,MAAM,CAAC,CAAC,YAAY,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;QAC/D,CAAC;IACH,CAAC;IAED,WAAW,CAAC,MAA+B;QACzC,MAAM,SAAS,GAAG,MAAM,CAAC,YAAY,CAAC,IAAI,UAAU,CAAC;QACrD,MAAM,UAAU,GAAG,MAAM,CAAC,aAAa,CAAC,IAAI,OAAO,CAAC;QACpD,IAAI,GAAG,GAAG,cAAc,SAAS,OAAO,UAAU,KAAK,CAAC;QACxD,IAAI,GAAG,CAAC,MAAM,GAAG,EAAE;YAAE,GAAG,GAAG,0BAA0B,UAAU,KAAK,CAAC;QACrE,IAAI,GAAG,CAAC,MAAM,GAAG,EAAE;YAAE,GAAG,GAAG,+BAA+B,CAAC;QAC3D,OAAO,GAAG,CAAC;IACb,CAAC;CACF;AAED,MAAM,OAAO,mBAAoB,SAAQ,IAAI;IAClC,IAAI,GAAG,kBAAkB,CAAC;IAC1B,WAAW,GAClB,8FAA8F,CAAC;IACxF,WAAW,GAAG;QACrB,IAAI,EAAE,QAAiB;QACvB,UAAU,EAAE;YACV,UAAU,EAAE;gBACV,IAAI,EAAE,QAAiB;gBACvB,WAAW,EAAE,wBAAwB;aACtC;YACD,WAAW,EAAE;gBACX,IAAI,EAAE,QAAiB;gBACvB,WAAW,EAAE,yBAAyB;aACvC;YACD,WAAW,EAAE;gBACX,IAAI,EAAE,QAAiB;gBACvB,WAAW,EAAE,0CAA0C;gBACvD,OAAO,EAAE,UAAU;aACpB;YACD,SAAS,EAAE;gBACT,IAAI,EAAE,QAAiB;gBACvB,WAAW,EAAE,uCAAuC;gBACpD,OAAO,EAAE,KAAK;aACf;YACD,UAAU,EAAE;gBACV,IAAI,EAAE,OAAgB;gBACtB,WAAW,EAAE,6BAA6B;gBAC1C,KAAK,EAAE,EAAE,IAAI,EAAE,QAAiB,EAAE;gBAClC,OAAO,EAAE,EAAE;aACZ;SACF;QACD,QAAQ,EAAE,CAAC,YAAY,EAAE,aAAa,CAAC;KACxC,CAAC;IAEF,KAAK,CAAC,OAAO,CACX,OAA0B,EAC1B,MAA+B;QAE/B,IAAI,CAAC;YACH,MAAM,SAAS,GAAG,OAAO,CAAC,oBAAoB,CAC5C,MAAM,CAAC,YAAY,CAAW,CAC/B,CAAC;YACF,MAAM,UAAU,GAAG,OAAO,CAAC,oBAAoB,CAC7C,MAAM,CAAC,aAAa,CAAW,CAChC,CAAC;YACF,MAAM,UAAU,GAAI,MAAM,CAAC,aAAa,CAAY,IAAI,UAAU,CAAC;YACnE,MAAM,QAAQ,GAAI,MAAM,CAAC,WAAW,CAAY,IAAI,KAAK,CAAC;YAC1D,IAAI,SAAS,GAAI,MAAM,CAAC,YAAY,CAAc,IAAI,EAAE,CAAC;YACzD,IAAI,OAAO,SAAS,KAAK,QAAQ,EAAE,CAAC;gBAClC,SAAS,GAAG,CAAC,SAAS,CAAC,CAAC;YAC1B,CAAC;YAED,MAAM,SAAS,GAAG,OAAO,CAAC,UAAU,CAAC,CAAC;YACtC,MAAM,KAAK,CAAC,SAAS,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;YAE5C,MAAM,aAAa,CAAC,QAAQ,EAAE;gBAC5B,SAAS;gBACT,IAAI;gBACJ,UAAU;gBACV,IAAI;gBACJ,QAAQ;gBACR,IAAI;gBACJ,UAAU;gBACV,GAAG,SAAS;aACb,CAAC,CAAC;YAEH,OAAO,EAAE,WAAW,EAAE,UAAU,EAAE,MAAM,EAAE,SAAS,EAAE,CAAC;QACxD,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACX,OAAO,EAAE,KAAK,EAAE,MAAM,CAAC,CAAC,YAAY,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;QAC/D,CAAC;IACH,CAAC;IAED,WAAW,CAAC,MAA+B;QACzC,MAAM,SAAS,GAAG,MAAM,CAAC,YAAY,CAAC,IAAI,OAAO,CAAC;QAClD,MAAM,UAAU,GAAG,MAAM,CAAC,aAAa,CAAC,IAAI,QAAQ,CAAC;QACrD,MAAM,QAAQ,GAAG,MAAM,CAAC,WAAW,CAAC,IAAI,eAAe,CAAC;QACxD,IAAI,GAAG,GAAG,cAAc,SAAS,OAAO,UAAU,KAAK,QAAQ,MAAM,CAAC;QACtE,IAAI,GAAG,CAAC,MAAM,GAAG,EAAE;YAAE,GAAG,GAAG,cAAc,SAAS,OAAO,QAAQ,KAAK,CAAC;QACvE,IAAI,GAAG,CAAC,MAAM,GAAG,EAAE;YAAE,GAAG,GAAG,0BAA0B,QAAQ,KAAK,CAAC;QACnE,OAAO,GAAG,CAAC;IACb,CAAC;CACF"}

package/dist/tools/plan-builder-tools.d.ts

@@ -0,0 +1,72 @@
+/**
+ * Incremental plan construction tools shared via a PlanBuilder state object.
+ *
+ * The LLM calls add_task repeatedly, optionally remove_task to correct a
+ * mistake, and finally finish_plan to commit. Each add_task validates the
+ * task in isolation and against previously added tasks so failures are
+ * caught at the moment they're introduced, pointing to one task rather
+ * than the whole plan.
+ */
+import type { ProcessingContext } from "@nodetool-ai/runtime";
+import { Tool } from "./base-tool.js";
+import type { Task, TaskPlan } from "../types.js";
+export declare class PlanBuilder {
+    private readonly tasks;
+    private readonly inputKeys;
+    private committedPlan;
+    constructor(inputs?: Record<string, unknown>);
+    get currentTasks(): readonly Task[];
+    get taskCount(): number;
+    get plan(): TaskPlan | null;
+    hasTask(id: string): boolean;
+    private allStepIds;
+    addTask(params: Record<string, unknown>): {
+        ok: true;
+        task: Task;
+    } | {
+        ok: false;
+        errors: string[];
+    };
+    removeTask(id: string): {
+        ok: boolean;
+        error?: string;
+    };
+    finish(title: string): {
+        ok: true;
+        plan: TaskPlan;
+    } | {
+        ok: false;
+        errors: string[];
+    };
+    private buildTask;
+    private validateNewTask;
+    private validateFullPlan;
+}
+export declare class AddTaskTool extends Tool {
+    private readonly builder;
+    readonly name = "add_task";
+    readonly description: string;
+    readonly inputSchema: Record<string, unknown>;
+    constructor(builder: PlanBuilder);
+    process(_context: ProcessingContext, params: Record<string, unknown>): Promise<unknown>;
+    userMessage(params: Record<string, unknown>): string;
+}
+export declare class RemoveTaskTool extends Tool {
+    private readonly builder;
+    readonly name = "remove_task";
+    readonly description: string;
+    readonly inputSchema: Record<string, unknown>;
+    constructor(builder: PlanBuilder);
+    process(_context: ProcessingContext, params: Record<string, unknown>): Promise<unknown>;
+    userMessage(params: Record<string, unknown>): string;
+}
+export declare class FinishPlanTool extends Tool {
+    private readonly builder;
+    readonly name = "finish_plan";
+    readonly description: string;
+    readonly inputSchema: Record<string, unknown>;
+    constructor(builder: PlanBuilder);
+    process(_context: ProcessingContext, params: Record<string, unknown>): Promise<unknown>;
+    userMessage(params: Record<string, unknown>): string;
+}
+//# sourceMappingURL=plan-builder-tools.d.ts.map

package/dist/tools/plan-builder-tools.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"plan-builder-tools.d.ts","sourceRoot":"","sources":["../../src/tools/plan-builder-tools.ts"],"names":[],"mappings":"AAAA;;;;;;;;GAQG;AAEH,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,sBAAsB,CAAC;AAC9D,OAAO,EAAE,IAAI,EAAE,MAAM,gBAAgB,CAAC;AACtC,OAAO,KAAK,EAAQ,IAAI,EAAE,QAAQ,EAAE,MAAM,aAAa,CAAC;AAOxD,qBAAa,WAAW;IACtB,OAAO,CAAC,QAAQ,CAAC,KAAK,CAAc;IACpC,OAAO,CAAC,QAAQ,CAAC,SAAS,CAAc;IACxC,OAAO,CAAC,aAAa,CAAyB;gBAElC,MAAM,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC;IAI5C,IAAI,YAAY,IAAI,SAAS,IAAI,EAAE,CAElC;IAED,IAAI,SAAS,IAAI,MAAM,CAEtB;IAED,IAAI,IAAI,IAAI,QAAQ,GAAG,IAAI,CAE1B;IAED,OAAO,CAAC,EAAE,EAAE,MAAM,GAAG,OAAO;IAI5B,OAAO,CAAC,UAAU;IAQlB,OAAO,CACL,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAC9B;QAAE,EAAE,EAAE,IAAI,CAAC;QAAC,IAAI,EAAE,IAAI,CAAA;KAAE,GAAG;QAAE,EAAE,EAAE,KAAK,CAAC;QAAC,MAAM,EAAE,MAAM,EAAE,CAAA;KAAE;IAU7D,UAAU,CAAC,EAAE,EAAE,MAAM,GAAG;QAAE,EAAE,EAAE,OAAO,CAAC;QAAC,KAAK,CAAC,EAAE,MAAM,CAAA;KAAE;IASvD,MAAM,CACJ,KAAK,EAAE,MAAM,GACZ;QAAE,EAAE,EAAE,IAAI,CAAC;QAAC,IAAI,EAAE,QAAQ,CAAA;KAAE,GAAG;QAAE,EAAE,EAAE,KAAK,CAAC;QAAC,MAAM,EAAE,MAAM,EAAE,CAAA;KAAE;IAQjE,OAAO,CAAC,SAAS;IAgDjB,OAAO,CAAC,eAAe;IA+DvB,OAAO,CAAC,gBAAgB;CAWzB;AAwMD,qBAAa,WAAY,SAAQ,IAAI;IAQvB,OAAO,CAAC,QAAQ,CAAC,OAAO;IAPpC,QAAQ,CAAC,IAAI,cAAc;IAC3B,QAAQ,CAAC,WAAW,SAG2B;IAC/C,QAAQ,CAAC,WAAW,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAyB;gBAEzC,OAAO,EAAE,WAAW;IAI3C,OAAO,CACX,QAAQ,EAAE,iBAAiB,EAC3B,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAC9B,OAAO,CAAC,OAAO,CAAC;IAcnB,WAAW,CAAC,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,MAAM;CAIrD;AAUD,qBAAa,cAAe,SAAQ,IAAI;IAO1B,OAAO,CAAC,QAAQ,CAAC,OAAO;IANpC,QAAQ,CAAC,IAAI,iBAAiB;IAC9B,QAAQ,CAAC,WAAW,SAEoC;IACxD,QAAQ,CAAC,WAAW,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAA4B;gBAE5C,OAAO,EAAE,WAAW;IAI3C,OAAO,CACX,QAAQ,EAAE,iBAAiB,EAC3B,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAC9B,OAAO,CAAC,OAAO,CAAC;IASnB,WAAW,CAAC,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,MAAM;CAIrD;AAUD,qBAAa,cAAe,SAAQ,IAAI;IAQ1B,OAAO,CAAC,QAAQ,CAAC,OAAO;IAPpC,QAAQ,CAAC,IAAI,iBAAiB;IAC9B,QAAQ,CAAC,WAAW,SAGQ;IAC5B,QAAQ,CAAC,WAAW,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAA4B;gBAE5C,OAAO,EAAE,WAAW;IAI3C,OAAO,CACX,QAAQ,EAAE,iBAAiB,EAC3B,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAC9B,OAAO,CAAC,OAAO,CAAC;IAenB,WAAW,CAAC,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,MAAM;CAIrD"}