@nodetool-ai/agents 0.7.0-rc.11

package/dist/tools/filesystem-tools.js

@@ -0,0 +1,334 @@
+/**
+ * Filesystem tools for reading, writing, and listing files.
+ *
+ * Port of src/nodetool/agents/tools/workspace_tools.py
+ */
+import { readFile, writeFile, appendFile, mkdir, readdir, stat, access } from "node:fs/promises";
+import { dirname, join } from "node:path";
+import { Tool } from "./base-tool.js";
+const MAX_READ_CHARS = 100_000;
+const MAX_TOKENS = 25_000;
+const TOKEN_MODEL = "gpt-4";
+async function countTokens(text) {
+    const { encodingForModel } = await import("js-tiktoken");
+    const enc = encodingForModel(TOKEN_MODEL);
+    return enc.encode(text).length;
+}
+function isBinaryBuffer(buf) {
+    // Check first 8KB for null bytes — a strong indicator of binary content
+    const check = buf.subarray(0, 8192);
+    for (let i = 0; i < check.length; i++) {
+        if (check[i] === 0)
+            return true;
+    }
+    return false;
+}
+function resolveSafePath(context, rawPath) {
+    return context.resolveWorkspacePath(rawPath);
+}
+export class ReadFileTool extends Tool {
+    name = "read_file";
+    description = "Read the contents of a text file within the agent workspace. Automatically counts tokens and supports reading specific line ranges. Cannot read binary files.";
+    inputSchema = {
+        type: "object",
+        properties: {
+            path: {
+                type: "string",
+                description: "Path to the file to read, relative to the workspace directory"
+            },
+            start_line: {
+                type: "number",
+                description: "Start line (1-indexed, optional)"
+            },
+            end_line: {
+                type: "number",
+                description: "End line (1-indexed, inclusive, optional)"
+            }
+        },
+        required: ["path"]
+    };
+    async process(context, params) {
+        const rawPath = params["path"];
+        if (typeof rawPath !== "string") {
+            return { error: "path must be a string" };
+        }
+        let filePath;
+        try {
+            filePath = resolveSafePath(context, rawPath);
+        }
+        catch (e) {
+            return {
+                success: false,
+                error: String(e instanceof Error ? e.message : e)
+            };
+        }
+        try {
+            await access(filePath);
+        }
+        catch {
+            return {
+                success: false,
+                error: `Path ${rawPath} does not exist`
+            };
+        }
+        // Binary detection
+        try {
+            const rawBuf = await readFile(filePath);
+            if (isBinaryBuffer(rawBuf)) {
+                return {
+                    success: false,
+                    is_binary: true,
+                    path: rawPath,
+                    full_path: filePath,
+                    error: `The file ${rawPath} contains binary data that cannot be processed`
+                };
+            }
+            const raw = rawBuf.toString("utf-8");
+            const startLine = params["start_line"];
+            const endLine = params["end_line"];
+            let content;
+            let lineInfo;
+            if (typeof startLine === "number" || typeof endLine === "number") {
+                const lines = raw.split("\n");
+                const totalLines = lines.length;
+                const startIdx = Math.max(0, (typeof startLine === "number" ? startLine : 1) - 1);
+                const endIdx = Math.min(totalLines, typeof endLine === "number" ? endLine : totalLines);
+                if (startIdx >= totalLines || startIdx > endIdx) {
+                    return {
+                        success: false,
+                        is_directory: false,
+                        path: rawPath,
+                        full_path: filePath,
+                        error: `Invalid line range: start_line=${startLine}, end_line=${endLine}, total lines=${totalLines}`,
+                        suggested_ranges: [
+                            `1-${Math.min(500, totalLines)}`,
+                            `${Math.max(1, totalLines - 500)}-${totalLines}`
+                        ]
+                    };
+                }
+                content = lines.slice(startIdx, endIdx).join("\n");
+                lineInfo = {
+                    start_line: typeof startLine === "number" ? startLine : 1,
+                    end_line: endIdx,
+                    total_lines: totalLines
+                };
+            }
+            else {
+                content =
+                    raw.length > MAX_READ_CHARS ? raw.slice(0, MAX_READ_CHARS) : raw;
+                lineInfo = { total_lines: (raw.match(/\n/g) || []).length + 1 };
+            }
+            const tokenCount = await countTokens(content);
+            const tokenInfo = { count: tokenCount, model: TOKEN_MODEL };
+            if (tokenCount > MAX_TOKENS) {
+                const totalLines = lineInfo.total_lines ?? 0;
+                const approxLinesPerToken = totalLines / Math.max(1, tokenCount);
+                const suggestedLineCount = Math.floor(approxLinesPerToken * MAX_TOKENS * 0.9);
+                let suggestedRanges;
+                if (typeof startLine === "number") {
+                    const suggestedEnd = Math.min(startLine + suggestedLineCount - 1, totalLines);
+                    suggestedRanges = [`${startLine}-${suggestedEnd}`];
+                }
+                else {
+                    const chunkSize = Math.min(suggestedLineCount, 500);
+                    suggestedRanges = [];
+                    for (let i = 1; i <= totalLines && suggestedRanges.length < 3; i += chunkSize) {
+                        suggestedRanges.push(`${i}-${Math.min(i + chunkSize - 1, totalLines)}`);
+                    }
+                }
+                return {
+                    success: false,
+                    is_directory: false,
+                    path: rawPath,
+                    full_path: filePath,
+                    error: `Token count (${tokenCount}) exceeds maximum (${MAX_TOKENS}). Please read only a portion of the file.`,
+                    token_info: tokenInfo,
+                    line_info: lineInfo,
+                    suggested_ranges: suggestedRanges
+                };
+            }
+            return {
+                success: true,
+                is_directory: false,
+                path: rawPath,
+                full_path: filePath,
+                content,
+                truncated: raw.length > MAX_READ_CHARS,
+                is_binary: false,
+                line_info: lineInfo,
+                token_info: tokenInfo
+            };
+        }
+        catch (e) {
+            return {
+                success: false,
+                error: `Failed to read file: ${String(e)}`
+            };
+        }
+    }
+    userMessage(params) {
+        const path = params["path"];
+        const msg = `Reading content from ${path}...`;
+        return msg.length > 80 ? "Reading content from a path..." : msg;
+    }
+}
+export class WriteFileTool extends Tool {
+    name = "write_file";
+    description = "Write content to a file in the agent workspace, creating it if it doesn't exist.";
+    inputSchema = {
+        type: "object",
+        properties: {
+            path: {
+                type: "string",
+                description: "Path to the file to write, relative to the workspace directory"
+            },
+            content: {
+                type: "string",
+                description: "Content to write to the file"
+            },
+            append: {
+                type: "boolean",
+                description: "Whether to append to the file instead of overwriting",
+                default: false
+            }
+        },
+        required: ["path", "content"]
+    };
+    async process(context, params) {
+        const rawPath = params["path"];
+        const content = params["content"];
+        const appendMode = params["append"] === true;
+        if (typeof rawPath !== "string") {
+            return { error: "path must be a string" };
+        }
+        if (typeof content !== "string") {
+            return { error: "content must be a string" };
+        }
+        let filePath;
+        try {
+            filePath = resolveSafePath(context, rawPath);
+        }
+        catch (e) {
+            return {
+                success: false,
+                error: String(e instanceof Error ? e.message : e)
+            };
+        }
+        try {
+            await mkdir(dirname(filePath), { recursive: true });
+            let created;
+            try {
+                await access(filePath);
+                created = false;
+            }
+            catch {
+                created = true;
+            }
+            if (appendMode) {
+                await appendFile(filePath, content, "utf-8");
+            }
+            else {
+                await writeFile(filePath, content, "utf-8");
+            }
+            return {
+                success: true,
+                path: rawPath,
+                full_path: filePath,
+                append: appendMode,
+                created
+            };
+        }
+        catch (e) {
+            return { success: false, error: `Failed to write file: ${String(e)}` };
+        }
+    }
+    userMessage(params) {
+        const path = params["path"] ?? "";
+        const appendMode = params["append"] === true;
+        const action = appendMode ? "Appending to" : "Writing to";
+        const msg = `${action} file ${path}...`;
+        return msg.length > 160 ? `${action} a file...` : msg;
+    }
+}
+export class ListDirectoryTool extends Tool {
+    name = "list_directory";
+    description = "List the contents (files and subdirectories) of a directory within the agent workspace.";
+    inputSchema = {
+        type: "object",
+        properties: {
+            path: {
+                type: "string",
+                description: "Path to the directory to list, relative to the workspace directory"
+            },
+            recursive: {
+                type: "boolean",
+                description: "Whether to list contents recursively (default false)"
+            }
+        },
+        required: ["path"]
+    };
+    async process(context, params) {
+        const rawPath = params["path"];
+        if (typeof rawPath !== "string") {
+            return { error: "path must be a string" };
+        }
+        let dirPath;
+        try {
+            dirPath = resolveSafePath(context, rawPath);
+        }
+        catch (e) {
+            return {
+                success: false,
+                error: String(e instanceof Error ? e.message : e)
+            };
+        }
+        const recursive = params["recursive"] === true;
+        try {
+            const entries = await readdir(dirPath, { withFileTypes: true });
+            const results = [];
+            for (const entry of entries) {
+                const fullPath = join(dirPath, entry.name);
+                try {
+                    const info = await stat(fullPath);
+                    results.push({
+                        name: entry.name,
+                        size: info.size,
+                        isDirectory: entry.isDirectory()
+                    });
+                }
+                catch {
+                    results.push({
+                        name: entry.name,
+                        size: 0,
+                        isDirectory: entry.isDirectory()
+                    });
+                }
+            }
+            if (recursive) {
+                const subdirs = results.filter((e) => e.isDirectory);
+                for (const sub of subdirs) {
+                    const subResult = (await this.process(context, {
+                        path: join(rawPath, sub.name),
+                        recursive: true
+                    }));
+                    if (subResult.entries) {
+                        for (const child of subResult.entries) {
+                            results.push({
+                                ...child,
+                                name: `${sub.name}/${child.name}`
+                            });
+                        }
+                    }
+                }
+            }
+            return { entries: results };
+        }
+        catch (e) {
+            return { error: `Failed to list directory: ${String(e)}` };
+        }
+    }
+    userMessage(params) {
+        return `Listing directory: ${String(params["path"] ?? "")}`;
+    }
+}
+//# sourceMappingURL=filesystem-tools.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"filesystem-tools.js","sourceRoot":"","sources":["../../src/tools/filesystem-tools.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EACL,QAAQ,EACR,SAAS,EACT,UAAU,EACV,KAAK,EACL,OAAO,EACP,IAAI,EACJ,MAAM,EACP,MAAM,kBAAkB,CAAC;AAC1B,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,WAAW,CAAC;AAE1C,OAAO,EAAE,IAAI,EAAE,MAAM,gBAAgB,CAAC;AAEtC,MAAM,cAAc,GAAG,OAAO,CAAC;AAC/B,MAAM,UAAU,GAAG,MAAM,CAAC;AAC1B,MAAM,WAAW,GAAG,OAAgB,CAAC;AAErC,KAAK,UAAU,WAAW,CAAC,IAAY;IACrC,MAAM,EAAE,gBAAgB,EAAE,GAAG,MAAM,MAAM,CAAC,aAAa,CAAC,CAAC;IACzD,MAAM,GAAG,GAAG,gBAAgB,CAAC,WAAW,CAAC,CAAC;IAC1C,OAAO,GAAG,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,MAAM,CAAC;AACjC,CAAC;AAED,SAAS,cAAc,CAAC,GAAW;IACjC,wEAAwE;IACxE,MAAM,KAAK,GAAG,GAAG,CAAC,QAAQ,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC;IACpC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,KAAK,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;QACtC,IAAI,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC;YAAE,OAAO,IAAI,CAAC;IAClC,CAAC;IACD,OAAO,KAAK,CAAC;AACf,CAAC;AAED,SAAS,eAAe,CAAC,OAA0B,EAAE,OAAe;IAClE,OAAO,OAAO,CAAC,oBAAoB,CAAC,OAAO,CAAC,CAAC;AAC/C,CAAC;AAED,MAAM,OAAO,YAAa,SAAQ,IAAI;IAC3B,IAAI,GAAG,WAAW,CAAC;IACnB,WAAW,GAClB,+JAA+J,CAAC;IACzJ,WAAW,GAAG;QACrB,IAAI,EAAE,QAAiB;QACvB,UAAU,EAAE;YACV,IAAI,EAAE;gBACJ,IAAI,EAAE,QAAiB;gBACvB,WAAW,EACT,+DAA+D;aAClE;YACD,UAAU,EAAE;gBACV,IAAI,EAAE,QAAiB;gBACvB,WAAW,EAAE,kCAAkC;aAChD;YACD,QAAQ,EAAE;gBACR,IAAI,EAAE,QAAiB;gBACvB,WAAW,EAAE,2CAA2C;aACzD;SACF;QACD,QAAQ,EAAE,CAAC,MAAM,CAAC;KACnB,CAAC;IAEF,KAAK,CAAC,OAAO,CACX,OAA0B,EAC1B,MAA+B;QAE/B,MAAM,OAAO,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC;QAC/B,IAAI,OAAO,OAAO,KAAK,QAAQ,EAAE,CAAC;YAChC,OAAO,EAAE,KAAK,EAAE,uBAAuB,EAAE,CAAC;QAC5C,CAAC;QAED,IAAI,QAAgB,CAAC;QACrB,IAAI,CAAC;YACH,QAAQ,GAAG,eAAe,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;QAC/C,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACX,OAAO;gBACL,OAAO,EAAE,KAAK;gBACd,KAAK,EAAE,MAAM,CAAC,CAAC,YAAY,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC;aAClD,CAAC;QACJ,CAAC;QAED,IAAI,CAAC;YACH,MAAM,MAAM,CAAC,QAAQ,CAAC,CAAC;QACzB,CAAC;QAAC,MAAM,CAAC;YACP,OAAO;gBACL,OAAO,EAAE,KAAK;gBACd,KAAK,EAAE,QAAQ,OAAO,iBAAiB;aACxC,CAAC;QACJ,CAAC;QAED,mBAAmB;QACnB,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,QAAQ,CAAC,CAAC;YACxC,IAAI,cAAc,CAAC,MAAM,CAAC,EAAE,CAAC;gBAC3B,OAAO;oBACL,OAAO,EAAE,KAAK;oBACd,SAAS,EAAE,IAAI;oBACf,IAAI,EAAE,OAAO;oBACb,SAAS,EAAE,QAAQ;oBACnB,KAAK,EAAE,YAAY,OAAO,gDAAgD;iBAC3E,CAAC;YACJ,CAAC;YAED,MAAM,GAAG,GAAG,MAAM,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC;YAErC,MAAM,SAAS,GAAG,MAAM,CAAC,YAAY,CAAC,CAAC;YACvC,MAAM,OAAO,GAAG,MAAM,CAAC,UAAU,CAAC,CAAC;YACnC,IAAI,OAAe,CAAC;YACpB,IAAI,QAAgC,CAAC;YAErC,IAAI,OAAO,SAAS,KAAK,QAAQ,IAAI,OAAO,OAAO,KAAK,QAAQ,EAAE,CAAC;gBACjE,MAAM,KAAK,GAAG,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;gBAC9B,MAAM,UAAU,GAAG,KAAK,CAAC,MAAM,CAAC;gBAChC,MAAM,QAAQ,GAAG,IAAI,CAAC,GAAG,CACvB,CAAC,EACD,CAAC,OAAO,SAAS,KAAK,QAAQ,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CACpD,CAAC;gBACF,MAAM,MAAM,GAAG,IAAI,CAAC,GAAG,CACrB,UAAU,EACV,OAAO,OAAO,KAAK,QAAQ,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,UAAU,CACnD,CAAC;gBAEF,IAAI,QAAQ,IAAI,UAAU,IAAI,QAAQ,GAAG,MAAM,EAAE,CAAC;oBAChD,OAAO;wBACL,OAAO,EAAE,KAAK;wBACd,YAAY,EAAE,KAAK;wBACnB,IAAI,EAAE,OAAO;wBACb,SAAS,EAAE,QAAQ;wBACnB,KAAK,EAAE,kCAAkC,SAAS,cAAc,OAAO,iBAAiB,UAAU,EAAE;wBACpG,gBAAgB,EAAE;4BAChB,KAAK,IAAI,CAAC,GAAG,CAAC,GAAG,EAAE,UAAU,CAAC,EAAE;4BAChC,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,UAAU,GAAG,GAAG,CAAC,IAAI,UAAU,EAAE;yBACjD;qBACF,CAAC;gBACJ,CAAC;gBAED,OAAO,GAAG,KAAK,CAAC,KAAK,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;gBACnD,QAAQ,GAAG;oBACT,UAAU,EAAE,OAAO,SAAS,KAAK,QAAQ,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC;oBACzD,QAAQ,EAAE,MAAM;oBAChB,WAAW,EAAE,UAAU;iBACxB,CAAC;YACJ,CAAC;iBAAM,CAAC;gBACN,OAAO;oBACL,GAAG,CAAC,MAAM,GAAG,cAAc,CAAC,CAAC,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC,EAAE,cAAc,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC;gBACnE,QAAQ,GAAG,EAAE,WAAW,EAAE,CAAC,GAAG,CAAC,KAAK,CAAC,KAAK,CAAC,IAAI,EAAE,CAAC,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAClE,CAAC;YAED,MAAM,UAAU,GAAG,MAAM,WAAW,CAAC,OAAO,CAAC,CAAC;YAC9C,MAAM,SAAS,GAAG,EAAE,KAAK,EAAE,UAAU,EAAE,KAAK,EAAE,WAAW,EAAE,CAAC;YAE5D,IAAI,UAAU,GAAG,UAAU,EAAE,CAAC;gBAC5B,MAAM,UAAU,GAAG,QAAQ,CAAC,WAAW,IAAI,CAAC,CAAC;gBAC7C,MAAM,mBAAmB,GAAG,UAAU,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,UAAU,CAAC,CAAC;gBACjE,MAAM,kBAAkB,GAAG,IAAI,CAAC,KAAK,CACnC,mBAAmB,GAAG,UAAU,GAAG,GAAG,CACvC,CAAC;gBAEF,IAAI,eAAyB,CAAC;gBAC9B,IAAI,OAAO,SAAS,KAAK,QAAQ,EAAE,CAAC;oBAClC,MAAM,YAAY,GAAG,IAAI,CAAC,GAAG,CAC3B,SAAS,GAAG,kBAAkB,GAAG,CAAC,EAClC,UAAU,CACX,CAAC;oBACF,eAAe,GAAG,CAAC,GAAG,SAAS,IAAI,YAAY,EAAE,CAAC,CAAC;gBACrD,CAAC;qBAAM,CAAC;oBACN,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,CAAC,kBAAkB,EAAE,GAAG,CAAC,CAAC;oBACpD,eAAe,GAAG,EAAE,CAAC;oBACrB,KACE,IAAI,CAAC,GAAG,CAAC,EACT,CAAC,IAAI,UAAU,IAAI,eAAe,CAAC,MAAM,GAAG,CAAC,EAC7C,CAAC,IAAI,SAAS,EACd,CAAC;wBACD,eAAe,CAAC,IAAI,CAClB,GAAG,CAAC,IAAI,IAAI,CAAC,GAAG,CAAC,CAAC,GAAG,SAAS,GAAG,CAAC,EAAE,UAAU,CAAC,EAAE,CAClD,CAAC;oBACJ,CAAC;gBACH,CAAC;gBAED,OAAO;oBACL,OAAO,EAAE,KAAK;oBACd,YAAY,EAAE,KAAK;oBACnB,IAAI,EAAE,OAAO;oBACb,SAAS,EAAE,QAAQ;oBACnB,KAAK,EAAE,gBAAgB,UAAU,sBAAsB,UAAU,4CAA4C;oBAC7G,UAAU,EAAE,SAAS;oBACrB,SAAS,EAAE,QAAQ;oBACnB,gBAAgB,EAAE,eAAe;iBAClC,CAAC;YACJ,CAAC;YAED,OAAO;gBACL,OAAO,EAAE,IAAI;gBACb,YAAY,EAAE,KAAK;gBACnB,IAAI,EAAE,OAAO;gBACb,SAAS,EAAE,QAAQ;gBACnB,OAAO;gBACP,SAAS,EAAE,GAAG,CAAC,MAAM,GAAG,cAAc;gBACtC,SAAS,EAAE,KAAK;gBAChB,SAAS,EAAE,QAAQ;gBACnB,UAAU,EAAE,SAAS;aACtB,CAAC;QACJ,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACX,OAAO;gBACL,OAAO,EAAE,KAAK;gBACd,KAAK,EAAE,wBAAwB,MAAM,CAAC,CAAC,CAAC,EAAE;aAC3C,CAAC;QACJ,CAAC;IACH,CAAC;IAED,WAAW,CAAC,MAA+B;QACzC,MAAM,IAAI,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC;QAC5B,MAAM,GAAG,GAAG,wBAAwB,IAAI,KAAK,CAAC;QAC9C,OAAO,GAAG,CAAC,MAAM,GAAG,EAAE,CAAC,CAAC,CAAC,gCAAgC,CAAC,CAAC,CAAC,GAAG,CAAC;IAClE,CAAC;CACF;AAED,MAAM,OAAO,aAAc,SAAQ,IAAI;IAC5B,IAAI,GAAG,YAAY,CAAC;IACpB,WAAW,GAClB,kFAAkF,CAAC;IAC5E,WAAW,GAAG;QACrB,IAAI,EAAE,QAAiB;QACvB,UAAU,EAAE;YACV,IAAI,EAAE;gBACJ,IAAI,EAAE,QAAiB;gBACvB,WAAW,EACT,gEAAgE;aACnE;YACD,OAAO,EAAE;gBACP,IAAI,EAAE,QAAiB;gBACvB,WAAW,EAAE,8BAA8B;aAC5C;YACD,MAAM,EAAE;gBACN,IAAI,EAAE,SAAkB;gBACxB,WAAW,EAAE,sDAAsD;gBACnE,OAAO,EAAE,KAAK;aACf;SACF;QACD,QAAQ,EAAE,CAAC,MAAM,EAAE,SAAS,CAAC;KAC9B,CAAC;IAEF,KAAK,CAAC,OAAO,CACX,OAA0B,EAC1B,MAA+B;QAE/B,MAAM,OAAO,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC;QAC/B,MAAM,OAAO,GAAG,MAAM,CAAC,SAAS,CAAC,CAAC;QAClC,MAAM,UAAU,GAAG,MAAM,CAAC,QAAQ,CAAC,KAAK,IAAI,CAAC;QAE7C,IAAI,OAAO,OAAO,KAAK,QAAQ,EAAE,CAAC;YAChC,OAAO,EAAE,KAAK,EAAE,uBAAuB,EAAE,CAAC;QAC5C,CAAC;QACD,IAAI,OAAO,OAAO,KAAK,QAAQ,EAAE,CAAC;YAChC,OAAO,EAAE,KAAK,EAAE,0BAA0B,EAAE,CAAC;QAC/C,CAAC;QAED,IAAI,QAAgB,CAAC;QACrB,IAAI,CAAC;YACH,QAAQ,GAAG,eAAe,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;QAC/C,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACX,OAAO;gBACL,OAAO,EAAE,KAAK;gBACd,KAAK,EAAE,MAAM,CAAC,CAAC,YAAY,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC;aAClD,CAAC;QACJ,CAAC;QAED,IAAI,CAAC;YACH,MAAM,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;YAEpD,IAAI,OAAgB,CAAC;YACrB,IAAI,CAAC;gBACH,MAAM,MAAM,CAAC,QAAQ,CAAC,CAAC;gBACvB,OAAO,GAAG,KAAK,CAAC;YAClB,CAAC;YAAC,MAAM,CAAC;gBACP,OAAO,GAAG,IAAI,CAAC;YACjB,CAAC;YAED,IAAI,UAAU,EAAE,CAAC;gBACf,MAAM,UAAU,CAAC,QAAQ,EAAE,OAAO,EAAE,OAAO,CAAC,CAAC;YAC/C,CAAC;iBAAM,CAAC;gBACN,MAAM,SAAS,CAAC,QAAQ,EAAE,OAAO,EAAE,OAAO,CAAC,CAAC;YAC9C,CAAC;YAED,OAAO;gBACL,OAAO,EAAE,IAAI;gBACb,IAAI,EAAE,OAAO;gBACb,SAAS,EAAE,QAAQ;gBACnB,MAAM,EAAE,UAAU;gBAClB,OAAO;aACR,CAAC;QACJ,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACX,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,yBAAyB,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC;QACzE,CAAC;IACH,CAAC;IAED,WAAW,CAAC,MAA+B;QACzC,MAAM,IAAI,GAAG,MAAM,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC;QAClC,MAAM,UAAU,GAAG,MAAM,CAAC,QAAQ,CAAC,KAAK,IAAI,CAAC;QAC7C,MAAM,MAAM,GAAG,UAAU,CAAC,CAAC,CAAC,cAAc,CAAC,CAAC,CAAC,YAAY,CAAC;QAC1D,MAAM,GAAG,GAAG,GAAG,MAAM,SAAS,IAAI,KAAK,CAAC;QACxC,OAAO,GAAG,CAAC,MAAM,GAAG,GAAG,CAAC,CAAC,CAAC,GAAG,MAAM,YAAY,CAAC,CAAC,CAAC,GAAG,CAAC;IACxD,CAAC;CACF;AAQD,MAAM,OAAO,iBAAkB,SAAQ,IAAI;IAChC,IAAI,GAAG,gBAAgB,CAAC;IACxB,WAAW,GAClB,yFAAyF,CAAC;IACnF,WAAW,GAAG;QACrB,IAAI,EAAE,QAAiB;QACvB,UAAU,EAAE;YACV,IAAI,EAAE;gBACJ,IAAI,EAAE,QAAiB;gBACvB,WAAW,EACT,oEAAoE;aACvE;YACD,SAAS,EAAE;gBACT,IAAI,EAAE,SAAkB;gBACxB,WAAW,EAAE,sDAAsD;aACpE;SACF;QACD,QAAQ,EAAE,CAAC,MAAM,CAAC;KACnB,CAAC;IAEF,KAAK,CAAC,OAAO,CACX,OAA0B,EAC1B,MAA+B;QAE/B,MAAM,OAAO,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC;QAC/B,IAAI,OAAO,OAAO,KAAK,QAAQ,EAAE,CAAC;YAChC,OAAO,EAAE,KAAK,EAAE,uBAAuB,EAAE,CAAC;QAC5C,CAAC;QAED,IAAI,OAAe,CAAC;QACpB,IAAI,CAAC;YACH,OAAO,GAAG,eAAe,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;QAC9C,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACX,OAAO;gBACL,OAAO,EAAE,KAAK;gBACd,KAAK,EAAE,MAAM,CAAC,CAAC,YAAY,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC;aAClD,CAAC;QACJ,CAAC;QAED,MAAM,SAAS,GAAG,MAAM,CAAC,WAAW,CAAC,KAAK,IAAI,CAAC;QAC/C,IAAI,CAAC;YACH,MAAM,OAAO,GAAG,MAAM,OAAO,CAAC,OAAO,EAAE,EAAE,aAAa,EAAE,IAAI,EAAE,CAAC,CAAC;YAChE,MAAM,OAAO,GAAe,EAAE,CAAC;YAC/B,KAAK,MAAM,KAAK,IAAI,OAAO,EAAE,CAAC;gBAC5B,MAAM,QAAQ,GAAG,IAAI,CAAC,OAAO,EAAE,KAAK,CAAC,IAAI,CAAC,CAAC;gBAC3C,IAAI,CAAC;oBACH,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,CAAC;oBAClC,OAAO,CAAC,IAAI,CAAC;wBACX,IAAI,EAAE,KAAK,CAAC,IAAI;wBAChB,IAAI,EAAE,IAAI,CAAC,IAAI;wBACf,WAAW,EAAE,KAAK,CAAC,WAAW,EAAE;qBACjC,CAAC,CAAC;gBACL,CAAC;gBAAC,MAAM,CAAC;oBACP,OAAO,CAAC,IAAI,CAAC;wBACX,IAAI,EAAE,KAAK,CAAC,IAAI;wBAChB,IAAI,EAAE,CAAC;wBACP,WAAW,EAAE,KAAK,CAAC,WAAW,EAAE;qBACjC,CAAC,CAAC;gBACL,CAAC;YACH,CAAC;YACD,IAAI,SAAS,EAAE,CAAC;gBACd,MAAM,OAAO,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC;gBACrD,KAAK,MAAM,GAAG,IAAI,OAAO,EAAE,CAAC;oBAC1B,MAAM,SAAS,GAAG,CAAC,MAAM,IAAI,CAAC,OAAO,CAAC,OAAO,EAAE;wBAC7C,IAAI,EAAE,IAAI,CAAC,OAAO,EAAE,GAAG,CAAC,IAAI,CAAC;wBAC7B,SAAS,EAAE,IAAI;qBAChB,CAAC,CAA6B,CAAC;oBAChC,IAAI,SAAS,CAAC,OAAO,EAAE,CAAC;wBACtB,KAAK,MAAM,KAAK,IAAI,SAAS,CAAC,OAAO,EAAE,CAAC;4BACtC,OAAO,CAAC,IAAI,CAAC;gCACX,GAAG,KAAK;gCACR,IAAI,EAAE,GAAG,GAAG,CAAC,IAAI,IAAI,KAAK,CAAC,IAAI,EAAE;6BAClC,CAAC,CAAC;wBACL,CAAC;oBACH,CAAC;gBACH,CAAC;YACH,CAAC;YACD,OAAO,EAAE,OAAO,EAAE,OAAO,EAAE,CAAC;QAC9B,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACX,OAAO,EAAE,KAAK,EAAE,6BAA6B,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC;QAC7D,CAAC;IACH,CAAC;IAED,WAAW,CAAC,MAA+B;QACzC,OAAO,sBAAsB,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC,EAAE,CAAC;IAC9D,CAAC;CACF"}

package/dist/tools/find-model-tool.d.ts

@@ -0,0 +1,25 @@
+/**
+ * FindModelTool — pick a real {provider, model_id} for a generic AI node.
+ *
+ * The workflow-building agent calls this before adding any of the generic AI
+ * nodes (`nodetool.image.TextToImage`, `nodetool.video.TextToVideo`, etc.).
+ * It walks the configured providers, asks each one for its capability set,
+ * collects models from the providers that support the requested capability,
+ * and returns a small ranked list.
+ *
+ * In-process by design — uses the runtime `BaseProvider` instances directly
+ * rather than going over HTTP, so it works identically from the CLI and the
+ * WebSocket runner.
+ */
+import type { BaseProvider, ProcessingContext } from "@nodetool-ai/runtime";
+import { Tool } from "./base-tool.js";
+export declare class FindModelTool extends Tool {
+    private readonly providers;
+    readonly name = "find_model";
+    readonly description = "Find a real {provider, model_id} for a generic AI node by capability. Returns models from providers the user has configured, ranked by recommended/downloaded/preferences. Call this before adding any generic AI node (TextToImage, TextToVideo, TextToSpeech, etc.).";
+    readonly inputSchema: Record<string, unknown>;
+    constructor(providers: Record<string, BaseProvider>);
+    process(_context: ProcessingContext, params: Record<string, unknown>): Promise<unknown>;
+    userMessage(params: Record<string, unknown>): string;
+}
+//# sourceMappingURL=find-model-tool.d.ts.map

package/dist/tools/find-model-tool.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"find-model-tool.d.ts","sourceRoot":"","sources":["../../src/tools/find-model-tool.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;GAYG;AACH,OAAO,KAAK,EAEV,YAAY,EAIZ,iBAAiB,EAIlB,MAAM,sBAAsB,CAAC;AAM9B,OAAO,EAAE,IAAI,EAAE,MAAM,gBAAgB,CAAC;AAsKtC,qBAAa,aAAc,SAAQ,IAAI;IAMzB,OAAO,CAAC,QAAQ,CAAC,SAAS;IALtC,QAAQ,CAAC,IAAI,gBAAgB;IAC7B,QAAQ,CAAC,WAAW,4QACuP;IAC3Q,QAAQ,CAAC,WAAW,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAA2B;gBAE3C,SAAS,EAAE,MAAM,CAAC,MAAM,EAAE,YAAY,CAAC;IAI9D,OAAO,CACX,QAAQ,EAAE,iBAAiB,EAC3B,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAC9B,OAAO,CAAC,OAAO,CAAC;IAoGnB,WAAW,CAAC,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,MAAM;CAGrD"}

package/dist/tools/find-model-tool.js

@@ -0,0 +1,236 @@
+import { RECOMMENDED_MODELS, providerCapabilities } from "@nodetool-ai/runtime";
+import { Tool } from "./base-tool.js";
+const SUPPORTED_CAPABILITIES = [
+    "text_to_image",
+    "image_to_image",
+    "text_to_video",
+    "image_to_video",
+    "text_to_speech",
+    "automatic_speech_recognition",
+    "generate_embedding",
+    "generate_message"
+];
+const LOCAL_PROVIDER_IDS = new Set([
+    "ollama",
+    "lmstudio",
+    "vllm",
+    "llama_cpp",
+    "huggingface"
+]);
+const FIND_MODEL_INPUT_SCHEMA = {
+    type: "object",
+    properties: {
+        capability: {
+            type: "string",
+            enum: [...SUPPORTED_CAPABILITIES],
+            description: "Provider capability needed by the generic AI node (e.g. text_to_image, generate_embedding)."
+        },
+        task: {
+            type: "string",
+            description: "Optional task hint matched against model.supportedTasks (e.g. 'text_to_image' vs 'image_to_image')."
+        },
+        provider_hint: {
+            type: "string",
+            description: "Optional preferred provider id (e.g. 'openai'). Boosts matching providers."
+        },
+        model_hint: {
+            type: "array",
+            items: { type: "string" },
+            description: "Optional preferred model ids. Strongly boosts matching models in the ranking."
+        },
+        prefer_local: {
+            type: "boolean",
+            description: "If true, ranks local providers (ollama, lmstudio, vllm, llama_cpp, huggingface) above hosted ones.",
+            default: false
+        },
+        limit: {
+            type: "number",
+            description: "Maximum number of results to return (default 5).",
+            default: 5
+        }
+    },
+    required: ["capability"]
+};
+function getRecommendedSet(capability) {
+    const wantedTasks = capabilityToRecommendedTasks(capability);
+    const wantedModalities = capabilityToRecommendedModalities(capability);
+    const ids = new Set();
+    for (const m of RECOMMENDED_MODELS) {
+        const taskMatch = !wantedTasks || (m.task && wantedTasks.has(m.task));
+        const modalityMatch = wantedModalities.has(m.modality);
+        if (modalityMatch && taskMatch && m.provider) {
+            ids.add(`${m.provider}::${m.id}`);
+        }
+    }
+    return ids;
+}
+function capabilityToRecommendedTasks(capability) {
+    switch (capability) {
+        case "text_to_image":
+            return new Set(["text_to_image"]);
+        case "image_to_image":
+            return new Set(["image_to_image"]);
+        case "text_to_video":
+            return new Set(["text_to_video"]);
+        case "image_to_video":
+            return new Set(["image_to_video"]);
+        case "generate_embedding":
+            return new Set(["embedding"]);
+        case "generate_message":
+            return new Set(["text_generation"]);
+        default:
+            // tts/asr have no `task` set on recommended entries — fall back to
+            // modality-only filtering.
+            return null;
+    }
+}
+function capabilityToRecommendedModalities(capability) {
+    switch (capability) {
+        case "text_to_image":
+        case "image_to_image":
+            return new Set(["image"]);
+        case "text_to_video":
+        case "image_to_video":
+            return new Set(["video"]);
+        case "text_to_speech":
+            return new Set(["tts"]);
+        case "automatic_speech_recognition":
+            return new Set(["asr"]);
+        case "generate_embedding":
+        case "generate_message":
+            return new Set(["language"]);
+    }
+}
+async function fetchModelsForCapability(provider, capability) {
+    switch (capability) {
+        case "text_to_image":
+        case "image_to_image":
+            return (await provider.getAvailableImageModels());
+        case "text_to_video":
+        case "image_to_video":
+            return (await provider.getAvailableVideoModels());
+        case "text_to_speech":
+            return (await provider.getAvailableTTSModels());
+        case "automatic_speech_recognition":
+            return (await provider.getAvailableASRModels());
+        case "generate_embedding":
+            return (await provider.getAvailableEmbeddingModels());
+        case "generate_message":
+            return (await provider.getAvailableLanguageModels());
+    }
+}
+function taskMatch(model, task) {
+    if (!task)
+        return true;
+    if (!model.supportedTasks || model.supportedTasks.length === 0)
+        return true;
+    return model.supportedTasks.includes(task);
+}
+export class FindModelTool extends Tool {
+    providers;
+    name = "find_model";
+    description = "Find a real {provider, model_id} for a generic AI node by capability. Returns models from providers the user has configured, ranked by recommended/downloaded/preferences. Call this before adding any generic AI node (TextToImage, TextToVideo, TextToSpeech, etc.).";
+    inputSchema = FIND_MODEL_INPUT_SCHEMA;
+    constructor(providers) {
+        super();
+        this.providers = providers;
+    }
+    async process(_context, params) {
+        const capability = params["capability"];
+        if (!capability || !SUPPORTED_CAPABILITIES.includes(capability)) {
+            return {
+                capability,
+                total: 0,
+                results: [],
+                error: `capability must be one of: ${SUPPORTED_CAPABILITIES.join(", ")}`
+            };
+        }
+        const task = typeof params["task"] === "string" ? params["task"] : undefined;
+        const providerHint = typeof params["provider_hint"] === "string"
+            ? params["provider_hint"]
+            : undefined;
+        const modelHintRaw = params["model_hint"];
+        const modelHints = new Set(typeof modelHintRaw === "string"
+            ? [modelHintRaw]
+            : Array.isArray(modelHintRaw)
+                ? modelHintRaw.filter((x) => typeof x === "string")
+                : []);
+        const preferLocal = params["prefer_local"] === true;
+        const limit = typeof params["limit"] === "number" && params["limit"] > 0
+            ? Math.floor(params["limit"])
+            : 5;
+        const providerEntries = Object.entries(this.providers);
+        if (providerEntries.length === 0) {
+            return {
+                capability,
+                total: 0,
+                results: [],
+                note: "No providers configured. Use nodetool.agents.AgentStep for AI work."
+            };
+        }
+        const recommendedSet = getRecommendedSet(capability);
+        const collected = [];
+        for (const [providerId, instance] of providerEntries) {
+            let supports;
+            try {
+                supports = providerCapabilities(instance).includes(capability);
+            }
+            catch {
+                continue;
+            }
+            if (!supports)
+                continue;
+            let models;
+            try {
+                models = await fetchModelsForCapability(instance, capability);
+            }
+            catch {
+                continue;
+            }
+            for (const m of models) {
+                if (!taskMatch(m, task))
+                    continue;
+                const recommended = recommendedSet.has(`${providerId}::${m.id}`);
+                const downloaded = LOCAL_PROVIDER_IDS.has(providerId);
+                let score = 0;
+                if (recommended)
+                    score += 100;
+                if (downloaded)
+                    score += 30;
+                // Explicit user preferences outrank the default recommended bonus.
+                if (providerHint && providerId === providerHint)
+                    score += 200;
+                if (modelHints.has(m.id))
+                    score += 250;
+                if (preferLocal && LOCAL_PROVIDER_IDS.has(providerId))
+                    score += 150;
+                else if (preferLocal && !LOCAL_PROVIDER_IDS.has(providerId))
+                    score -= 5;
+                collected.push({
+                    provider: providerId,
+                    model_id: m.id,
+                    name: m.name,
+                    downloaded,
+                    recommended,
+                    score
+                });
+            }
+        }
+        collected.sort((a, b) => {
+            if (b.score !== a.score)
+                return b.score - a.score;
+            if (a.provider !== b.provider)
+                return a.provider.localeCompare(b.provider);
+            return a.model_id.localeCompare(b.model_id);
+        });
+        return {
+            capability,
+            total: collected.length,
+            results: collected.slice(0, limit)
+        };
+    }
+    userMessage(params) {
+        return `Looking up models for capability: ${String(params["capability"])}`;
+    }
+}
+//# sourceMappingURL=find-model-tool.js.map

package/dist/tools/find-model-tool.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"find-model-tool.js","sourceRoot":"","sources":["../../src/tools/find-model-tool.ts"],"names":[],"mappings":"AAwBA,OAAO,EACL,kBAAkB,EAClB,oBAAoB,EACrB,MAAM,sBAAsB,CAAC;AAE9B,OAAO,EAAE,IAAI,EAAE,MAAM,gBAAgB,CAAC;AAEtC,MAAM,sBAAsB,GAAG;IAC7B,eAAe;IACf,gBAAgB;IAChB,eAAe;IACf,gBAAgB;IAChB,gBAAgB;IAChB,8BAA8B;IAC9B,oBAAoB;IACpB,kBAAkB;CACV,CAAC;AAIX,MAAM,kBAAkB,GAAG,IAAI,GAAG,CAAC;IACjC,QAAQ;IACR,UAAU;IACV,MAAM;IACN,WAAW;IACX,aAAa;CACd,CAAC,CAAC;AAkBH,MAAM,uBAAuB,GAAG;IAC9B,IAAI,EAAE,QAAiB;IACvB,UAAU,EAAE;QACV,UAAU,EAAE;YACV,IAAI,EAAE,QAAiB;YACvB,IAAI,EAAE,CAAC,GAAG,sBAAsB,CAAC;YACjC,WAAW,EACT,6FAA6F;SAChG;QACD,IAAI,EAAE;YACJ,IAAI,EAAE,QAAiB;YACvB,WAAW,EACT,qGAAqG;SACxG;QACD,aAAa,EAAE;YACb,IAAI,EAAE,QAAiB;YACvB,WAAW,EACT,4EAA4E;SAC/E;QACD,UAAU,EAAE;YACV,IAAI,EAAE,OAAgB;YACtB,KAAK,EAAE,EAAE,IAAI,EAAE,QAAiB,EAAE;YAClC,WAAW,EACT,+EAA+E;SAClF;QACD,YAAY,EAAE;YACZ,IAAI,EAAE,SAAkB;YACxB,WAAW,EACT,oGAAoG;YACtG,OAAO,EAAE,KAAK;SACf;QACD,KAAK,EAAE;YACL,IAAI,EAAE,QAAiB;YACvB,WAAW,EAAE,kDAAkD;YAC/D,OAAO,EAAE,CAAC;SACX;KACF;IACD,QAAQ,EAAE,CAAC,YAAY,CAAa;CACrC,CAAC;AAEF,SAAS,iBAAiB,CACxB,UAA+B;IAE/B,MAAM,WAAW,GAAG,4BAA4B,CAAC,UAAU,CAAC,CAAC;IAC7D,MAAM,gBAAgB,GAAG,iCAAiC,CAAC,UAAU,CAAC,CAAC;IACvE,MAAM,GAAG,GAAG,IAAI,GAAG,EAAU,CAAC;IAC9B,KAAK,MAAM,CAAC,IAAI,kBAA+C,EAAE,CAAC;QAChE,MAAM,SAAS,GAAG,CAAC,WAAW,IAAI,CAAC,CAAC,CAAC,IAAI,IAAI,WAAW,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC;QACtE,MAAM,aAAa,GAAG,gBAAgB,CAAC,GAAG,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC;QACvD,IAAI,aAAa,IAAI,SAAS,IAAI,CAAC,CAAC,QAAQ,EAAE,CAAC;YAC7C,GAAG,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC,QAAQ,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;QACpC,CAAC;IACH,CAAC;IACD,OAAO,GAAG,CAAC;AACb,CAAC;AAED,SAAS,4BAA4B,CACnC,UAA+B;IAE/B,QAAQ,UAAU,EAAE,CAAC;QACnB,KAAK,eAAe;YAClB,OAAO,IAAI,GAAG,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC;QACpC,KAAK,gBAAgB;YACnB,OAAO,IAAI,GAAG,CAAC,CAAC,gBAAgB,CAAC,CAAC,CAAC;QACrC,KAAK,eAAe;YAClB,OAAO,IAAI,GAAG,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC;QACpC,KAAK,gBAAgB;YACnB,OAAO,IAAI,GAAG,CAAC,CAAC,gBAAgB,CAAC,CAAC,CAAC;QACrC,KAAK,oBAAoB;YACvB,OAAO,IAAI,GAAG,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC;QAChC,KAAK,kBAAkB;YACrB,OAAO,IAAI,GAAG,CAAC,CAAC,iBAAiB,CAAC,CAAC,CAAC;QACtC;YACE,mEAAmE;YACnE,2BAA2B;YAC3B,OAAO,IAAI,CAAC;IAChB,CAAC;AACH,CAAC;AAED,SAAS,iCAAiC,CACxC,UAA+B;IAE/B,QAAQ,UAAU,EAAE,CAAC;QACnB,KAAK,eAAe,CAAC;QACrB,KAAK,gBAAgB;YACnB,OAAO,IAAI,GAAG,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC;QAC5B,KAAK,eAAe,CAAC;QACrB,KAAK,gBAAgB;YACnB,OAAO,IAAI,GAAG,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC;QAC5B,KAAK,gBAAgB;YACnB,OAAO,IAAI,GAAG,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC;QAC1B,KAAK,8BAA8B;YACjC,OAAO,IAAI,GAAG,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC;QAC1B,KAAK,oBAAoB,CAAC;QAC1B,KAAK,kBAAkB;YACrB,OAAO,IAAI,GAAG,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC;IACjC,CAAC;AACH,CAAC;AAED,KAAK,UAAU,wBAAwB,CACrC,QAAsB,EACtB,UAA+B;IAE/B,QAAQ,UAAU,EAAE,CAAC;QACnB,KAAK,eAAe,CAAC;QACrB,KAAK,gBAAgB;YACnB,OAAO,CAAC,MAAM,QAAQ,CAAC,uBAAuB,EAAE,CAAiB,CAAC;QACpE,KAAK,eAAe,CAAC;QACrB,KAAK,gBAAgB;YACnB,OAAO,CAAC,MAAM,QAAQ,CAAC,uBAAuB,EAAE,CAAiB,CAAC;QACpE,KAAK,gBAAgB;YACnB,OAAO,CAAC,MAAM,QAAQ,CAAC,qBAAqB,EAAE,CAAe,CAAC;QAChE,KAAK,8BAA8B;YACjC,OAAO,CAAC,MAAM,QAAQ,CAAC,qBAAqB,EAAE,CAAe,CAAC;QAChE,KAAK,oBAAoB;YACvB,OAAO,CAAC,MAAM,QAAQ,CAAC,2BAA2B,EAAE,CAAqB,CAAC;QAC5E,KAAK,kBAAkB;YACrB,OAAO,CAAC,MAAM,QAAQ,CAAC,0BAA0B,EAAE,CAAoB,CAAC;IAC5E,CAAC;AACH,CAAC;AAED,SAAS,SAAS,CAAC,KAAe,EAAE,IAAwB;IAC1D,IAAI,CAAC,IAAI;QAAE,OAAO,IAAI,CAAC;IACvB,IAAI,CAAC,KAAK,CAAC,cAAc,IAAI,KAAK,CAAC,cAAc,CAAC,MAAM,KAAK,CAAC;QAAE,OAAO,IAAI,CAAC;IAC5E,OAAO,KAAK,CAAC,cAAc,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC;AAC7C,CAAC;AAED,MAAM,OAAO,aAAc,SAAQ,IAAI;IAMR;IALpB,IAAI,GAAG,YAAY,CAAC;IACpB,WAAW,GAClB,wQAAwQ,CAAC;IAClQ,WAAW,GAA4B,uBAAuB,CAAC;IAExE,YAA6B,SAAuC;QAClE,KAAK,EAAE,CAAC;QADmB,cAAS,GAAT,SAAS,CAA8B;IAEpE,CAAC;IAED,KAAK,CAAC,OAAO,CACX,QAA2B,EAC3B,MAA+B;QAE/B,MAAM,UAAU,GAAG,MAAM,CAAC,YAAY,CAAoC,CAAC;QAC3E,IAAI,CAAC,UAAU,IAAI,CAAC,sBAAsB,CAAC,QAAQ,CAAC,UAAU,CAAC,EAAE,CAAC;YAChE,OAAO;gBACL,UAAU;gBACV,KAAK,EAAE,CAAC;gBACR,OAAO,EAAE,EAAE;gBACX,KAAK,EAAE,8BAA8B,sBAAsB,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE;aACzE,CAAC;QACJ,CAAC;QAED,MAAM,IAAI,GACR,OAAO,MAAM,CAAC,MAAM,CAAC,KAAK,QAAQ,CAAC,CAAC,CAAE,MAAM,CAAC,MAAM,CAAY,CAAC,CAAC,CAAC,SAAS,CAAC;QAC9E,MAAM,YAAY,GAChB,OAAO,MAAM,CAAC,eAAe,CAAC,KAAK,QAAQ;YACzC,CAAC,CAAE,MAAM,CAAC,eAAe,CAAY;YACrC,CAAC,CAAC,SAAS,CAAC;QAChB,MAAM,YAAY,GAAG,MAAM,CAAC,YAAY,CAAC,CAAC;QAC1C,MAAM,UAAU,GAAgB,IAAI,GAAG,CACrC,OAAO,YAAY,KAAK,QAAQ;YAC9B,CAAC,CAAC,CAAC,YAAY,CAAC;YAChB,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,YAAY,CAAC;gBAC3B,CAAC,CAAE,YAAY,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,OAAO,CAAC,KAAK,QAAQ,CAAc;gBACjE,CAAC,CAAC,EAAE,CACT,CAAC;QACF,MAAM,WAAW,GAAG,MAAM,CAAC,cAAc,CAAC,KAAK,IAAI,CAAC;QACpD,MAAM,KAAK,GACT,OAAO,MAAM,CAAC,OAAO,CAAC,KAAK,QAAQ,IAAI,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC;YACxD,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,OAAO,CAAW,CAAC;YACvC,CAAC,CAAC,CAAC,CAAC;QAER,MAAM,eAAe,GAAG,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QACvD,IAAI,eAAe,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACjC,OAAO;gBACL,UAAU;gBACV,KAAK,EAAE,CAAC;gBACR,OAAO,EAAE,EAAE;gBACX,IAAI,EAAE,qEAAqE;aAC5E,CAAC;QACJ,CAAC;QAED,MAAM,cAAc,GAAG,iBAAiB,CAAC,UAAU,CAAC,CAAC;QACrD,MAAM,SAAS,GAAsB,EAAE,CAAC;QAExC,KAAK,MAAM,CAAC,UAAU,EAAE,QAAQ,CAAC,IAAI,eAAe,EAAE,CAAC;YACrD,IAAI,QAAiB,CAAC;YACtB,IAAI,CAAC;gBACH,QAAQ,GAAG,oBAAoB,CAAC,QAAQ,CAAC,CAAC,QAAQ,CAChD,UAAgC,CACjC,CAAC;YACJ,CAAC;YAAC,MAAM,CAAC;gBACP,SAAS;YACX,CAAC;YACD,IAAI,CAAC,QAAQ;gBAAE,SAAS;YAExB,IAAI,MAAkB,CAAC;YACvB,IAAI,CAAC;gBACH,MAAM,GAAG,MAAM,wBAAwB,CAAC,QAAQ,EAAE,UAAU,CAAC,CAAC;YAChE,CAAC;YAAC,MAAM,CAAC;gBACP,SAAS;YACX,CAAC;YAED,KAAK,MAAM,CAAC,IAAI,MAAM,EAAE,CAAC;gBACvB,IAAI,CAAC,SAAS,CAAC,CAAC,EAAE,IAAI,CAAC;oBAAE,SAAS;gBAClC,MAAM,WAAW,GAAG,cAAc,CAAC,GAAG,CAAC,GAAG,UAAU,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;gBACjE,MAAM,UAAU,GAAG,kBAAkB,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;gBAEtD,IAAI,KAAK,GAAG,CAAC,CAAC;gBACd,IAAI,WAAW;oBAAE,KAAK,IAAI,GAAG,CAAC;gBAC9B,IAAI,UAAU;oBAAE,KAAK,IAAI,EAAE,CAAC;gBAC5B,mEAAmE;gBACnE,IAAI,YAAY,IAAI,UAAU,KAAK,YAAY;oBAAE,KAAK,IAAI,GAAG,CAAC;gBAC9D,IAAI,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC;oBAAE,KAAK,IAAI,GAAG,CAAC;gBACvC,IAAI,WAAW,IAAI,kBAAkB,CAAC,GAAG,CAAC,UAAU,CAAC;oBAAE,KAAK,IAAI,GAAG,CAAC;qBAC/D,IAAI,WAAW,IAAI,CAAC,kBAAkB,CAAC,GAAG,CAAC,UAAU,CAAC;oBAAE,KAAK,IAAI,CAAC,CAAC;gBAExE,SAAS,CAAC,IAAI,CAAC;oBACb,QAAQ,EAAE,UAAU;oBACpB,QAAQ,EAAE,CAAC,CAAC,EAAE;oBACd,IAAI,EAAE,CAAC,CAAC,IAAI;oBACZ,UAAU;oBACV,WAAW;oBACX,KAAK;iBACN,CAAC,CAAC;YACL,CAAC;QACH,CAAC;QAED,SAAS,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE;YACtB,IAAI,CAAC,CAAC,KAAK,KAAK,CAAC,CAAC,KAAK;gBAAE,OAAO,CAAC,CAAC,KAAK,GAAG,CAAC,CAAC,KAAK,CAAC;YAClD,IAAI,CAAC,CAAC,QAAQ,KAAK,CAAC,CAAC,QAAQ;gBAAE,OAAO,CAAC,CAAC,QAAQ,CAAC,aAAa,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC;YAC3E,OAAO,CAAC,CAAC,QAAQ,CAAC,aAAa,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC;QAC9C,CAAC,CAAC,CAAC;QAEH,OAAO;YACL,UAAU;YACV,KAAK,EAAE,SAAS,CAAC,MAAM;YACvB,OAAO,EAAE,SAAS,CAAC,KAAK,CAAC,CAAC,EAAE,KAAK,CAAC;SACnC,CAAC;IACJ,CAAC;IAED,WAAW,CAAC,MAA+B;QACzC,OAAO,qCAAqC,MAAM,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC,EAAE,CAAC;IAC7E,CAAC;CACF"}

package/dist/tools/finish-graph-tool.d.ts

@@ -0,0 +1,23 @@
+/**
+ * FinishGraphTool -- planner tool that validates and finalizes the graph.
+ *
+ * Called by the LLM when it's done building the graph. Runs full validation
+ * (cycle detection, connectivity) and produces the GraphData.
+ */
+import type { ProcessingContext } from "@nodetool-ai/runtime";
+import type { GraphData } from "@nodetool-ai/protocol";
+import { Tool } from "./base-tool.js";
+import type { GraphBuilder } from "../graph-builder.js";
+export declare class FinishGraphTool extends Tool {
+    private readonly builder;
+    readonly name = "finish_graph";
+    readonly description = "Validate and finalize the workflow graph. Call this when you are done adding nodes and edges.";
+    readonly inputSchema: Record<string, unknown>;
+    /** The finalized graph, available after a successful call. */
+    private _graph;
+    get graph(): GraphData | null;
+    constructor(builder: GraphBuilder);
+    process(_context: ProcessingContext, _params: Record<string, unknown>): Promise<unknown>;
+    userMessage(_params: Record<string, unknown>): string;
+}
+//# sourceMappingURL=finish-graph-tool.d.ts.map

package/dist/tools/finish-graph-tool.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"finish-graph-tool.d.ts","sourceRoot":"","sources":["../../src/tools/finish-graph-tool.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,sBAAsB,CAAC;AAC9D,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,uBAAuB,CAAC;AACvD,OAAO,EAAE,IAAI,EAAE,MAAM,gBAAgB,CAAC;AACtC,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,qBAAqB,CAAC;AAQxD,qBAAa,eAAgB,SAAQ,IAAI;IAa3B,OAAO,CAAC,QAAQ,CAAC,OAAO;IAZpC,QAAQ,CAAC,IAAI,kBAAkB;IAC/B,QAAQ,CAAC,WAAW,mGAC8E;IAClG,QAAQ,CAAC,WAAW,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAA6B;IAE1E,8DAA8D;IAC9D,OAAO,CAAC,MAAM,CAA0B;IAExC,IAAI,KAAK,IAAI,SAAS,GAAG,IAAI,CAE5B;gBAE4B,OAAO,EAAE,YAAY;IAI5C,OAAO,CACX,QAAQ,EAAE,iBAAiB,EAC3B,OAAO,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAC/B,OAAO,CAAC,OAAO,CAAC;IAkBnB,WAAW,CAAC,OAAO,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,MAAM;CAGtD"}