@n8n/ai-workflow-builder 1.8.1 → 1.9.1

package/dist/code-builder/constants.d.ts

@@ -0,0 +1,62 @@
+import type { BuilderToolBase } from '../utils/stream-processor';
+export declare const MAX_AGENT_ITERATIONS = 50;
+export declare const MAX_VALIDATE_ATTEMPTS = 10;
+export declare const FIX_VALIDATION_ERRORS_INSTRUCTION = "\nAnalyze the errors, then use the editor tool to apply fixes:\n1. Which errors are relevant to the last user request? If NONE, stop \u2014 do not fix unrelated warnings.\n2. Use search_nodes and get_node_types to look up the correct node schema (if not already fetched)\n3. Use batch_str_replace to fix ALL identified issues atomically in one call (preferred), or use individual str_replace/insert calls\n4. Call validate_workflow ONCE after all fixes are applied\nDo NOT output explanations \u2014 just fix the code. Do not add or edit comments.";
+export declare const TEXT_EDITOR_TOOL: {
+    type: "text_editor_20250728";
+    name: "str_replace_based_edit_tool";
+};
+export declare const BATCH_STR_REPLACE_TOOL: {
+    type: "function";
+    function: {
+        name: string;
+        description: string;
+        parameters: {
+            type: "object";
+            properties: {
+                replacements: {
+                    type: "array";
+                    items: {
+                        type: "object";
+                        properties: {
+                            old_str: {
+                                type: "string";
+                                description: string;
+                            };
+                            new_str: {
+                                type: "string";
+                                description: string;
+                            };
+                        };
+                        required: string[];
+                    };
+                    description: string;
+                };
+            };
+            required: string[];
+        };
+    };
+};
+export declare const VALIDATE_TOOL: {
+    type: "function";
+    function: {
+        name: string;
+        description: string;
+        parameters: {
+            type: "object";
+            properties: {
+                path: {
+                    type: "string";
+                    description: string;
+                };
+            };
+            required: string[];
+        };
+    };
+};
+export declare const CODE_BUILDER_TEXT_EDITOR_TOOL: BuilderToolBase;
+export declare const CODE_BUILDER_BATCH_STR_REPLACE_TOOL: BuilderToolBase;
+export declare const CODE_BUILDER_VALIDATE_TOOL: BuilderToolBase;
+export declare const CODE_BUILDER_SEARCH_NODES_TOOL: BuilderToolBase;
+export declare const CODE_BUILDER_GET_NODE_TYPES_TOOL: BuilderToolBase;
+export declare const CODE_BUILDER_GET_SUGGESTED_NODES_TOOL: BuilderToolBase;

package/dist/code-builder/constants.js

@@ -0,0 +1,83 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.CODE_BUILDER_GET_SUGGESTED_NODES_TOOL = exports.CODE_BUILDER_GET_NODE_TYPES_TOOL = exports.CODE_BUILDER_SEARCH_NODES_TOOL = exports.CODE_BUILDER_VALIDATE_TOOL = exports.CODE_BUILDER_BATCH_STR_REPLACE_TOOL = exports.CODE_BUILDER_TEXT_EDITOR_TOOL = exports.VALIDATE_TOOL = exports.BATCH_STR_REPLACE_TOOL = exports.TEXT_EDITOR_TOOL = exports.FIX_VALIDATION_ERRORS_INSTRUCTION = exports.MAX_VALIDATE_ATTEMPTS = exports.MAX_AGENT_ITERATIONS = void 0;
+exports.MAX_AGENT_ITERATIONS = 50;
+exports.MAX_VALIDATE_ATTEMPTS = 10;
+exports.FIX_VALIDATION_ERRORS_INSTRUCTION = `
+Analyze the errors, then use the editor tool to apply fixes:
+1. Which errors are relevant to the last user request? If NONE, stop — do not fix unrelated warnings.
+2. Use search_nodes and get_node_types to look up the correct node schema (if not already fetched)
+3. Use batch_str_replace to fix ALL identified issues atomically in one call (preferred), or use individual str_replace/insert calls
+4. Call validate_workflow ONCE after all fixes are applied
+Do NOT output explanations — just fix the code. Do not add or edit comments.`;
+exports.TEXT_EDITOR_TOOL = {
+    type: 'text_editor_20250728',
+    name: 'str_replace_based_edit_tool',
+};
+exports.BATCH_STR_REPLACE_TOOL = {
+    type: 'function',
+    function: {
+        name: 'batch_str_replace',
+        description: 'Apply multiple str_replace operations to /workflow.js atomically. All replacements are applied in order. If any replacement fails, all changes are rolled back. Example: {"replacements": [{"old_str": "foo", "new_str": "bar"}, {"old_str": "baz", "new_str": "qux"}]}',
+        parameters: {
+            type: 'object',
+            properties: {
+                replacements: {
+                    type: 'array',
+                    items: {
+                        type: 'object',
+                        properties: {
+                            old_str: { type: 'string', description: 'The exact string to find' },
+                            new_str: { type: 'string', description: 'The replacement string' },
+                        },
+                        required: ['old_str', 'new_str'],
+                    },
+                    description: 'Ordered list of replacements to apply',
+                },
+            },
+            required: ['replacements'],
+        },
+    },
+};
+exports.VALIDATE_TOOL = {
+    type: 'function',
+    function: {
+        name: 'validate_workflow',
+        description: 'Validate the current workflow code for errors. Returns validation results - either success or a list of errors to fix.',
+        parameters: {
+            type: 'object',
+            properties: {
+                path: {
+                    type: 'string',
+                    description: 'Path to the workflow file (must be /workflow.js)',
+                },
+            },
+            required: ['path'],
+        },
+    },
+};
+exports.CODE_BUILDER_TEXT_EDITOR_TOOL = {
+    toolName: 'str_replace_based_edit_tool',
+    displayTitle: 'Editing workflow',
+};
+exports.CODE_BUILDER_BATCH_STR_REPLACE_TOOL = {
+    toolName: 'batch_str_replace',
+    displayTitle: 'Editing workflow',
+};
+exports.CODE_BUILDER_VALIDATE_TOOL = {
+    toolName: 'validate_workflow',
+    displayTitle: 'Validating workflow',
+};
+exports.CODE_BUILDER_SEARCH_NODES_TOOL = {
+    toolName: 'search_nodes',
+    displayTitle: 'Searching nodes',
+};
+exports.CODE_BUILDER_GET_NODE_TYPES_TOOL = {
+    toolName: 'get_node_types',
+    displayTitle: 'Getting node definitions',
+};
+exports.CODE_BUILDER_GET_SUGGESTED_NODES_TOOL = {
+    toolName: 'get_suggested_nodes',
+    displayTitle: 'Getting suggested nodes',
+};
+//# sourceMappingURL=constants.js.map

package/dist/code-builder/constants.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"constants.js","sourceRoot":"","sources":["../../src/code-builder/constants.ts"],"names":[],"mappings":";;;AASa,QAAA,oBAAoB,GAAG,EAAE,CAAC;AAG1B,QAAA,qBAAqB,GAAG,EAAE,CAAC;AAG3B,QAAA,iCAAiC,GAAG;;;;;;6EAM4B,CAAC;AAGjE,QAAA,gBAAgB,GAAG;IAC/B,IAAI,EAAE,sBAA+B;IACrC,IAAI,EAAE,6BAAsC;CAC5C,CAAC;AAGW,QAAA,sBAAsB,GAAG;IACrC,IAAI,EAAE,UAAmB;IACzB,QAAQ,EAAE;QACT,IAAI,EAAE,mBAAmB;QACzB,WAAW,EACV,yQAAyQ;QAC1Q,UAAU,EAAE;YACX,IAAI,EAAE,QAAiB;YACvB,UAAU,EAAE;gBACX,YAAY,EAAE;oBACb,IAAI,EAAE,OAAgB;oBACtB,KAAK,EAAE;wBACN,IAAI,EAAE,QAAiB;wBACvB,UAAU,EAAE;4BACX,OAAO,EAAE,EAAE,IAAI,EAAE,QAAiB,EAAE,WAAW,EAAE,0BAA0B,EAAE;4BAC7E,OAAO,EAAE,EAAE,IAAI,EAAE,QAAiB,EAAE,WAAW,EAAE,wBAAwB,EAAE;yBAC3E;wBACD,QAAQ,EAAE,CAAC,SAAS,EAAE,SAAS,CAAC;qBAChC;oBACD,WAAW,EAAE,uCAAuC;iBACpD;aACD;YACD,QAAQ,EAAE,CAAC,cAAc,CAAC;SAC1B;KACD;CACD,CAAC;AAGW,QAAA,aAAa,GAAG;IAC5B,IAAI,EAAE,UAAmB;IACzB,QAAQ,EAAE;QACT,IAAI,EAAE,mBAAmB;QACzB,WAAW,EACV,wHAAwH;QACzH,UAAU,EAAE;YACX,IAAI,EAAE,QAAiB;YACvB,UAAU,EAAE;gBACX,IAAI,EAAE;oBACL,IAAI,EAAE,QAAiB;oBACvB,WAAW,EAAE,kDAAkD;iBAC/D;aACD;YACD,QAAQ,EAAE,CAAC,MAAM,CAAC;SAClB;KACD;CACD,CAAC;AAKW,QAAA,6BAA6B,GAAoB;IAC7D,QAAQ,EAAE,6BAA6B;IACvC,YAAY,EAAE,kBAAkB;CAChC,CAAC;AAEW,QAAA,mCAAmC,GAAoB;IACnE,QAAQ,EAAE,mBAAmB;IAC7B,YAAY,EAAE,kBAAkB;CAChC,CAAC;AAEW,QAAA,0BAA0B,GAAoB;IAC1D,QAAQ,EAAE,mBAAmB;IAC7B,YAAY,EAAE,qBAAqB;CACnC,CAAC;AAEW,QAAA,8BAA8B,GAAoB;IAC9D,QAAQ,EAAE,cAAc;IACxB,YAAY,EAAE,iBAAiB;CAC/B,CAAC;AAEW,QAAA,gCAAgC,GAAoB;IAChE,QAAQ,EAAE,gBAAgB;IAC1B,YAAY,EAAE,0BAA0B;CACxC,CAAC;AAEW,QAAA,qCAAqC,GAAoB;IACrE,QAAQ,EAAE,qBAAqB;IAC/B,YAAY,EAAE,yBAAyB;CACvC,CAAC"}

package/dist/code-builder/engines/code-builder-node-search-engine.d.ts

@@ -0,0 +1,19 @@
+import type { INodeTypeDescription, NodeConnectionType } from 'n8n-workflow';
+import type { CodeBuilderNodeSearchResult } from '../types';
+export declare const SCORE_WEIGHTS: {
+    readonly CONNECTION_EXACT: 100;
+    readonly CONNECTION_IN_EXPRESSION: 50;
+};
+export declare class CodeBuilderNodeSearchEngine {
+    private readonly nodeTypes;
+    constructor(nodeTypes: INodeTypeDescription[]);
+    searchByName(query: string, limit?: number): CodeBuilderNodeSearchResult[];
+    searchByConnectionType(connectionType: NodeConnectionType, limit?: number, nameFilter?: string): CodeBuilderNodeSearchResult[];
+    formatResult(result: CodeBuilderNodeSearchResult): string;
+    getSubnodesForConnectionType(connectionType: string): string[];
+    getRelatedSubnodeIds(nodeIds: string[], excludeNodeIds: Set<string>): Set<string>;
+    getNodeType(nodeId: string): INodeTypeDescription | undefined;
+    private getConnectionScore;
+    static isAiConnectionType(connectionType: string): boolean;
+    static getAiConnectionTypes(): NodeConnectionType[];
+}

package/dist/code-builder/engines/code-builder-node-search-engine.js

@@ -0,0 +1,236 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.CodeBuilderNodeSearchEngine = exports.SCORE_WEIGHTS = void 0;
+const utils_1 = require("@n8n/utils");
+const n8n_workflow_1 = require("n8n-workflow");
+function isNodeConnectionType(value) {
+    return Object.values(n8n_workflow_1.NodeConnectionTypes).includes(value);
+}
+const DEFAULT_SUBNODES = {
+    ai_languageModel: ['@n8n/n8n-nodes-langchain.lmChatOpenAi'],
+    ai_memory: ['@n8n/n8n-nodes-langchain.memoryBufferWindow'],
+    ai_embedding: ['@n8n/n8n-nodes-langchain.embeddingsOpenAi'],
+    ai_vectorStore: ['@n8n/n8n-nodes-langchain.vectorStoreInMemory'],
+};
+const NODE_SEARCH_KEYS = [
+    { key: 'displayName', weight: 1.5 },
+    { key: 'name', weight: 1.3 },
+    { key: 'codex.alias', weight: 1.0 },
+    { key: 'description', weight: 0.7 },
+];
+function getTypeName(nodeName) {
+    if (!nodeName)
+        return '';
+    const lastDotIndex = nodeName.lastIndexOf('.');
+    return lastDotIndex >= 0 ? nodeName.substring(lastDotIndex + 1) : nodeName;
+}
+exports.SCORE_WEIGHTS = {
+    CONNECTION_EXACT: 100,
+    CONNECTION_IN_EXPRESSION: 50,
+};
+function getLatestVersion(version) {
+    return Array.isArray(version) ? Math.max(...version) : version;
+}
+function extractSubnodeRequirements(inputs) {
+    if (!inputs)
+        return [];
+    return Object.entries(inputs).map(([connectionType, config]) => ({
+        connectionType,
+        required: config.required,
+        ...(config.displayOptions && { displayOptions: config.displayOptions }),
+    }));
+}
+function dedupeNodes(nodes) {
+    const dedupeCache = {};
+    nodes.forEach((node) => {
+        const cachedNodeType = dedupeCache[node.name];
+        if (!cachedNodeType) {
+            dedupeCache[node.name] = node;
+            return;
+        }
+        const cachedVersion = getLatestVersion(cachedNodeType.version);
+        const nextVersion = getLatestVersion(node.version);
+        if (nextVersion > cachedVersion) {
+            dedupeCache[node.name] = node;
+        }
+    });
+    return Object.values(dedupeCache);
+}
+class CodeBuilderNodeSearchEngine {
+    nodeTypes;
+    constructor(nodeTypes) {
+        this.nodeTypes = dedupeNodes(nodeTypes);
+    }
+    searchByName(query, limit = 20) {
+        const searchResults = (0, utils_1.sublimeSearch)(query, this.nodeTypes, NODE_SEARCH_KEYS);
+        const queryLower = query.toLowerCase().trim();
+        const fuzzyResultNames = new Set(searchResults.map((r) => r.item.name));
+        const typeNameMatches = this.nodeTypes
+            .filter((node) => {
+            if (fuzzyResultNames.has(node.name))
+                return false;
+            return getTypeName(node.name).toLowerCase() === queryLower;
+        })
+            .map((item) => ({ item, score: 0 }));
+        const allResults = [...searchResults, ...typeNameMatches];
+        allResults.sort((a, b) => {
+            const exactA = getTypeName(a.item.name).toLowerCase() === queryLower;
+            const exactB = getTypeName(b.item.name).toLowerCase() === queryLower;
+            if (exactA && !exactB)
+                return -1;
+            if (!exactA && exactB)
+                return 1;
+            return b.score - a.score;
+        });
+        return allResults
+            .slice(0, limit)
+            .map(({ item, score, }) => {
+            const subnodeRequirements = extractSubnodeRequirements(item.builderHint?.inputs);
+            return {
+                name: item.name,
+                displayName: item.displayName,
+                description: item.description ?? 'No description available',
+                version: getLatestVersion(item.version),
+                inputs: item.inputs,
+                outputs: item.outputs,
+                score,
+                ...(item.builderHint?.message && { builderHintMessage: item.builderHint.message }),
+                ...(subnodeRequirements.length > 0 && { subnodeRequirements }),
+            };
+        });
+    }
+    searchByConnectionType(connectionType, limit = 20, nameFilter) {
+        const nodesWithConnectionType = this.nodeTypes
+            .map((nodeType) => {
+            const connectionScore = this.getConnectionScore(nodeType, connectionType);
+            return connectionScore > 0 ? { nodeType, connectionScore } : null;
+        })
+            .filter((result) => Boolean(result));
+        if (!nameFilter) {
+            return nodesWithConnectionType
+                .sort((a, b) => b.connectionScore - a.connectionScore)
+                .slice(0, limit)
+                .map(({ nodeType, connectionScore }) => {
+                const subnodeRequirements = extractSubnodeRequirements(nodeType.builderHint?.inputs);
+                return {
+                    name: nodeType.name,
+                    displayName: nodeType.displayName,
+                    version: getLatestVersion(nodeType.version),
+                    description: nodeType.description ?? 'No description available',
+                    inputs: nodeType.inputs,
+                    outputs: nodeType.outputs,
+                    score: connectionScore,
+                    ...(nodeType.builderHint?.message && {
+                        builderHintMessage: nodeType.builderHint.message,
+                    }),
+                    ...(subnodeRequirements.length > 0 && { subnodeRequirements }),
+                };
+            });
+        }
+        const nodeTypesOnly = nodesWithConnectionType.map((result) => result.nodeType);
+        const nameFilteredResults = (0, utils_1.sublimeSearch)(nameFilter, nodeTypesOnly, NODE_SEARCH_KEYS);
+        return nameFilteredResults
+            .slice(0, limit)
+            .map(({ item, score: nameScore }) => {
+            const connectionResult = nodesWithConnectionType.find((result) => result.nodeType.name === item.name);
+            const connectionScore = connectionResult?.connectionScore ?? 0;
+            const subnodeRequirements = extractSubnodeRequirements(item.builderHint?.inputs);
+            return {
+                name: item.name,
+                version: getLatestVersion(item.version),
+                displayName: item.displayName,
+                description: item.description ?? 'No description available',
+                inputs: item.inputs,
+                outputs: item.outputs,
+                score: connectionScore + nameScore,
+                ...(item.builderHint?.message && { builderHintMessage: item.builderHint.message }),
+                ...(subnodeRequirements.length > 0 && { subnodeRequirements }),
+            };
+        });
+    }
+    formatResult(result) {
+        const parts = [
+            '		<node>',
+            `			<node_name>${result.name}</node_name>`,
+            `			<node_version>${result.version}</node_version>`,
+            `			<node_description>${result.description}</node_description>`,
+            `			<node_inputs>${typeof result.inputs === 'object' ? JSON.stringify(result.inputs) : result.inputs}</node_inputs>`,
+            `			<node_outputs>${typeof result.outputs === 'object' ? JSON.stringify(result.outputs) : result.outputs}</node_outputs>`,
+        ];
+        if (result.builderHintMessage) {
+            parts.push(`			<builder_hint>${result.builderHintMessage}</builder_hint>`);
+        }
+        if (result.subnodeRequirements && result.subnodeRequirements.length > 0) {
+            parts.push('			<subnode_requirements>');
+            for (const req of result.subnodeRequirements) {
+                const requiredStr = req.required ? 'required' : 'optional';
+                if (req.displayOptions) {
+                    parts.push(`				<requirement type="${req.connectionType}" status="${requiredStr}"><display_options>${JSON.stringify(req.displayOptions)}</display_options></requirement>`);
+                }
+                else {
+                    parts.push('				<requirement type="' +
+                        req.connectionType +
+                        '" status="' +
+                        requiredStr +
+                        '" />');
+                }
+            }
+            parts.push('			</subnode_requirements>');
+        }
+        parts.push('		</node>');
+        return '\n' + parts.join('\n');
+    }
+    getSubnodesForConnectionType(connectionType) {
+        return DEFAULT_SUBNODES[connectionType] ?? [];
+    }
+    getRelatedSubnodeIds(nodeIds, excludeNodeIds) {
+        const allRelated = new Set();
+        const visited = new Set(excludeNodeIds);
+        for (const nodeId of nodeIds) {
+            visited.add(nodeId);
+        }
+        const queue = [...nodeIds];
+        while (queue.length > 0) {
+            const currentNodeId = queue.shift();
+            const nodeType = this.nodeTypes.find((n) => n.name === currentNodeId);
+            if (!nodeType?.builderHint?.inputs)
+                continue;
+            for (const connectionType of Object.keys(nodeType.builderHint.inputs)) {
+                const subnodeIds = this.getSubnodesForConnectionType(connectionType);
+                for (const subnodeId of subnodeIds) {
+                    if (visited.has(subnodeId))
+                        continue;
+                    visited.add(subnodeId);
+                    allRelated.add(subnodeId);
+                    queue.push(subnodeId);
+                }
+            }
+        }
+        return allRelated;
+    }
+    getNodeType(nodeId) {
+        return this.nodeTypes.find((n) => n.name === nodeId);
+    }
+    getConnectionScore(nodeType, connectionType) {
+        const outputs = nodeType.outputs;
+        if (Array.isArray(outputs)) {
+            if (outputs.includes(connectionType)) {
+                return exports.SCORE_WEIGHTS.CONNECTION_EXACT;
+            }
+        }
+        else if (typeof outputs === 'string') {
+            if (outputs.includes(connectionType)) {
+                return exports.SCORE_WEIGHTS.CONNECTION_IN_EXPRESSION;
+            }
+        }
+        return 0;
+    }
+    static isAiConnectionType(connectionType) {
+        return connectionType.startsWith('ai_');
+    }
+    static getAiConnectionTypes() {
+        return Object.values(n8n_workflow_1.NodeConnectionTypes).filter((type) => isNodeConnectionType(type) && CodeBuilderNodeSearchEngine.isAiConnectionType(type));
+    }
+}
+exports.CodeBuilderNodeSearchEngine = CodeBuilderNodeSearchEngine;
+//# sourceMappingURL=code-builder-node-search-engine.js.map

package/dist/code-builder/engines/code-builder-node-search-engine.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"code-builder-node-search-engine.js","sourceRoot":"","sources":["../../../src/code-builder/engines/code-builder-node-search-engine.ts"],"names":[],"mappings":";;;AAWA,sCAA2C;AAE3C,+CAAmD;AAKnD,SAAS,oBAAoB,CAAC,KAAa;IAC1C,OAAO,MAAM,CAAC,MAAM,CAAC,kCAAmB,CAAC,CAAC,QAAQ,CAAC,KAA2B,CAAC,CAAC;AACjF,CAAC;AAMD,MAAM,gBAAgB,GAA6B;IAClD,gBAAgB,EAAE,CAAC,uCAAuC,CAAC;IAC3D,SAAS,EAAE,CAAC,6CAA6C,CAAC;IAC1D,YAAY,EAAE,CAAC,2CAA2C,CAAC;IAC3D,cAAc,EAAE,CAAC,8CAA8C,CAAC;CAEhE,CAAC;AAMF,MAAM,gBAAgB,GAAG;IACxB,EAAE,GAAG,EAAE,aAAa,EAAE,MAAM,EAAE,GAAG,EAAE;IACnC,EAAE,GAAG,EAAE,MAAM,EAAE,MAAM,EAAE,GAAG,EAAE;IAC5B,EAAE,GAAG,EAAE,aAAa,EAAE,MAAM,EAAE,GAAG,EAAE;IACnC,EAAE,GAAG,EAAE,aAAa,EAAE,MAAM,EAAE,GAAG,EAAE;CACnC,CAAC;AAMF,SAAS,WAAW,CAAC,QAAgB;IACpC,IAAI,CAAC,QAAQ;QAAE,OAAO,EAAE,CAAC;IACzB,MAAM,YAAY,GAAG,QAAQ,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC;IAC/C,OAAO,YAAY,IAAI,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,SAAS,CAAC,YAAY,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC;AAC5E,CAAC;AAKY,QAAA,aAAa,GAAG;IAC5B,gBAAgB,EAAE,GAAG;IACrB,wBAAwB,EAAE,EAAE;CACnB,CAAC;AAEX,SAAS,gBAAgB,CAAC,OAA0B;IACnD,OAAO,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,GAAG,OAAO,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC;AAChE,CAAC;AAKD,SAAS,0BAA0B,CAAC,MAA0B;IAC7D,IAAI,CAAC,MAAM;QAAE,OAAO,EAAE,CAAC;IAEvB,OAAO,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,cAAc,EAAE,MAAM,CAAC,EAAE,EAAE,CAAC,CAAC;QAChE,cAAc;QACd,QAAQ,EAAE,MAAM,CAAC,QAAQ;QACzB,GAAG,CAAC,MAAM,CAAC,cAAc,IAAI,EAAE,cAAc,EAAE,MAAM,CAAC,cAAc,EAAE,CAAC;KACvE,CAAC,CAAC,CAAC;AACL,CAAC;AAED,SAAS,WAAW,CAAC,KAA6B;IACjD,MAAM,WAAW,GAAyC,EAAE,CAAC;IAC7D,KAAK,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,EAAE;QACtB,MAAM,cAAc,GAAG,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC9C,IAAI,CAAC,cAAc,EAAE,CAAC;YACrB,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC;YAC9B,OAAO;QACR,CAAC;QAED,MAAM,aAAa,GAAG,gBAAgB,CAAC,cAAc,CAAC,OAAO,CAAC,CAAC;QAC/D,MAAM,WAAW,GAAG,gBAAgB,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAEnD,IAAI,WAAW,GAAG,aAAa,EAAE,CAAC;YACjC,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC;QAC/B,CAAC;IACF,CAAC,CAAC,CAAC;IAEH,OAAO,MAAM,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC;AACnC,CAAC;AAMD,MAAa,2BAA2B;IACtB,SAAS,CAAyB;IACnD,YAAY,SAAiC;QAC5C,IAAI,CAAC,SAAS,GAAG,WAAW,CAAC,SAAS,CAAC,CAAC;IACzC,CAAC;IASD,YAAY,CAAC,KAAa,EAAE,QAAgB,EAAE;QAE7C,MAAM,aAAa,GAAG,IAAA,qBAAa,EAClC,KAAK,EACL,IAAI,CAAC,SAAS,EACd,gBAAgB,CAChB,CAAC;QAEF,MAAM,UAAU,GAAG,KAAK,CAAC,WAAW,EAAE,CAAC,IAAI,EAAE,CAAC;QAC9C,MAAM,gBAAgB,GAAG,IAAI,GAAG,CAAC,aAAa,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC;QAGxE,MAAM,eAAe,GAAG,IAAI,CAAC,SAAS;aACpC,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE;YAChB,IAAI,gBAAgB,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC;gBAAE,OAAO,KAAK,CAAC;YAClD,OAAO,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,WAAW,EAAE,KAAK,UAAU,CAAC;QAC5D,CAAC,CAAC;aACD,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC;QAGtC,MAAM,UAAU,GAAG,CAAC,GAAG,aAAa,EAAE,GAAG,eAAe,CAAC,CAAC;QAC1D,UAAU,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE;YACxB,MAAM,MAAM,GAAG,WAAW,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,WAAW,EAAE,KAAK,UAAU,CAAC;YACrE,MAAM,MAAM,GAAG,WAAW,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,WAAW,EAAE,KAAK,UAAU,CAAC;YACrE,IAAI,MAAM,IAAI,CAAC,MAAM;gBAAE,OAAO,CAAC,CAAC,CAAC;YACjC,IAAI,CAAC,MAAM,IAAI,MAAM;gBAAE,OAAO,CAAC,CAAC;YAChC,OAAO,CAAC,CAAC,KAAK,GAAG,CAAC,CAAC,KAAK,CAAC;QAC1B,CAAC,CAAC,CAAC;QAGH,OAAO,UAAU;aACf,KAAK,CAAC,CAAC,EAAE,KAAK,CAAC;aACf,GAAG,CACH,CAAC,EACA,IAAI,EACJ,KAAK,GAC0C,EAA+B,EAAE;YAChF,MAAM,mBAAmB,GAAG,0BAA0B,CAAC,IAAI,CAAC,WAAW,EAAE,MAAM,CAAC,CAAC;YACjF,OAAO;gBACN,IAAI,EAAE,IAAI,CAAC,IAAI;gBACf,WAAW,EAAE,IAAI,CAAC,WAAW;gBAC7B,WAAW,EAAE,IAAI,CAAC,WAAW,IAAI,0BAA0B;gBAC3D,OAAO,EAAE,gBAAgB,CAAC,IAAI,CAAC,OAAO,CAAC;gBACvC,MAAM,EAAE,IAAI,CAAC,MAAM;gBACnB,OAAO,EAAE,IAAI,CAAC,OAAO;gBACrB,KAAK;gBACL,GAAG,CAAC,IAAI,CAAC,WAAW,EAAE,OAAO,IAAI,EAAE,kBAAkB,EAAE,IAAI,CAAC,WAAW,CAAC,OAAO,EAAE,CAAC;gBAClF,GAAG,CAAC,mBAAmB,CAAC,MAAM,GAAG,CAAC,IAAI,EAAE,mBAAmB,EAAE,CAAC;aAC9D,CAAC;QACH,CAAC,CACD,CAAC;IACJ,CAAC;IAUD,sBAAsB,CACrB,cAAkC,EAClC,QAAgB,EAAE,EAClB,UAAmB;QAGnB,MAAM,uBAAuB,GAAG,IAAI,CAAC,SAAS;aAC5C,GAAG,CAAC,CAAC,QAAQ,EAAE,EAAE;YACjB,MAAM,eAAe,GAAG,IAAI,CAAC,kBAAkB,CAAC,QAAQ,EAAE,cAAc,CAAC,CAAC;YAC1E,OAAO,eAAe,GAAG,CAAC,CAAC,CAAC,CAAC,EAAE,QAAQ,EAAE,eAAe,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC;QACnE,CAAC,CAAC;aACD,MAAM,CAAC,CAAC,MAAM,EAAyE,EAAE,CACzF,OAAO,CAAC,MAAM,CAAC,CACf,CAAC;QAGH,IAAI,CAAC,UAAU,EAAE,CAAC;YACjB,OAAO,uBAAuB;iBAC5B,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,eAAe,GAAG,CAAC,CAAC,eAAe,CAAC;iBACrD,KAAK,CAAC,CAAC,EAAE,KAAK,CAAC;iBACf,GAAG,CAAC,CAAC,EAAE,QAAQ,EAAE,eAAe,EAAE,EAAE,EAAE;gBACtC,MAAM,mBAAmB,GAAG,0BAA0B,CAAC,QAAQ,CAAC,WAAW,EAAE,MAAM,CAAC,CAAC;gBACrF,OAAO;oBACN,IAAI,EAAE,QAAQ,CAAC,IAAI;oBACnB,WAAW,EAAE,QAAQ,CAAC,WAAW;oBACjC,OAAO,EAAE,gBAAgB,CAAC,QAAQ,CAAC,OAAO,CAAC;oBAC3C,WAAW,EAAE,QAAQ,CAAC,WAAW,IAAI,0BAA0B;oBAC/D,MAAM,EAAE,QAAQ,CAAC,MAAM;oBACvB,OAAO,EAAE,QAAQ,CAAC,OAAO;oBACzB,KAAK,EAAE,eAAe;oBACtB,GAAG,CAAC,QAAQ,CAAC,WAAW,EAAE,OAAO,IAAI;wBACpC,kBAAkB,EAAE,QAAQ,CAAC,WAAW,CAAC,OAAO;qBAChD,CAAC;oBACF,GAAG,CAAC,mBAAmB,CAAC,MAAM,GAAG,CAAC,IAAI,EAAE,mBAAmB,EAAE,CAAC;iBAC9D,CAAC;YACH,CAAC,CAAC,CAAC;QACL,CAAC;QAGD,MAAM,aAAa,GAAG,uBAAuB,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;QAC/E,MAAM,mBAAmB,GAAG,IAAA,qBAAa,EAAC,UAAU,EAAE,aAAa,EAAE,gBAAgB,CAAC,CAAC;QAGvF,OAAO,mBAAmB;aACxB,KAAK,CAAC,CAAC,EAAE,KAAK,CAAC;aACf,GAAG,CAAC,CAAC,EAAE,IAAI,EAAE,KAAK,EAAE,SAAS,EAAiD,EAAE,EAAE;YAClF,MAAM,gBAAgB,GAAG,uBAAuB,CAAC,IAAI,CACpD,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,QAAQ,CAAC,IAAI,KAAK,IAAI,CAAC,IAAI,CAC9C,CAAC;YACF,MAAM,eAAe,GAAG,gBAAgB,EAAE,eAAe,IAAI,CAAC,CAAC;YAC/D,MAAM,mBAAmB,GAAG,0BAA0B,CAAC,IAAI,CAAC,WAAW,EAAE,MAAM,CAAC,CAAC;YAEjF,OAAO;gBACN,IAAI,EAAE,IAAI,CAAC,IAAI;gBACf,OAAO,EAAE,gBAAgB,CAAC,IAAI,CAAC,OAAO,CAAC;gBACvC,WAAW,EAAE,IAAI,CAAC,WAAW;gBAC7B,WAAW,EAAE,IAAI,CAAC,WAAW,IAAI,0BAA0B;gBAC3D,MAAM,EAAE,IAAI,CAAC,MAAM;gBACnB,OAAO,EAAE,IAAI,CAAC,OAAO;gBACrB,KAAK,EAAE,eAAe,GAAG,SAAS;gBAClC,GAAG,CAAC,IAAI,CAAC,WAAW,EAAE,OAAO,IAAI,EAAE,kBAAkB,EAAE,IAAI,CAAC,WAAW,CAAC,OAAO,EAAE,CAAC;gBAClF,GAAG,CAAC,mBAAmB,CAAC,MAAM,GAAG,CAAC,IAAI,EAAE,mBAAmB,EAAE,CAAC;aAC9D,CAAC;QACH,CAAC,CAAC,CAAC;IACL,CAAC;IAOD,YAAY,CAAC,MAAmC;QAC/C,MAAM,KAAK,GAAG;YACb,UAAU;YACV,iBAAiB,MAAM,CAAC,IAAI,cAAc;YAC1C,oBAAoB,MAAM,CAAC,OAAO,iBAAiB;YACnD,wBAAwB,MAAM,CAAC,WAAW,qBAAqB;YAC/D,mBAAmB,OAAO,MAAM,CAAC,MAAM,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,MAAM,gBAAgB;YACpH,oBAAoB,OAAO,MAAM,CAAC,OAAO,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,OAAO,iBAAiB;SACzH,CAAC;QAGF,IAAI,MAAM,CAAC,kBAAkB,EAAE,CAAC;YAC/B,KAAK,CAAC,IAAI,CAAC,oBAAoB,MAAM,CAAC,kBAAkB,iBAAiB,CAAC,CAAC;QAC5E,CAAC;QAGD,IAAI,MAAM,CAAC,mBAAmB,IAAI,MAAM,CAAC,mBAAmB,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACzE,KAAK,CAAC,IAAI,CAAC,2BAA2B,CAAC,CAAC;YACxC,KAAK,MAAM,GAAG,IAAI,MAAM,CAAC,mBAAmB,EAAE,CAAC;gBAC9C,MAAM,WAAW,GAAG,GAAG,CAAC,QAAQ,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,UAAU,CAAC;gBAC3D,IAAI,GAAG,CAAC,cAAc,EAAE,CAAC;oBACxB,KAAK,CAAC,IAAI,CACT,0BAA0B,GAAG,CAAC,cAAc,aAAa,WAAW,sBAAsB,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,cAAc,CAAC,kCAAkC,CAC9J,CAAC;gBACH,CAAC;qBAAM,CAAC;oBACP,KAAK,CAAC,IAAI,CACT,yBAAyB;wBACxB,GAAG,CAAC,cAAc;wBAClB,YAAY;wBACZ,WAAW;wBACX,MAAM,CACP,CAAC;gBACH,CAAC;YACF,CAAC;YACD,KAAK,CAAC,IAAI,CAAC,4BAA4B,CAAC,CAAC;QAC1C,CAAC;QAED,KAAK,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;QAExB,OAAO,IAAI,GAAG,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAChC,CAAC;IAQD,4BAA4B,CAAC,cAAsB;QAGlD,OAAO,gBAAgB,CAAC,cAAc,CAAC,IAAI,EAAE,CAAC;IAC/C,CAAC;IASD,oBAAoB,CAAC,OAAiB,EAAE,cAA2B;QAClE,MAAM,UAAU,GAAG,IAAI,GAAG,EAAU,CAAC;QACrC,MAAM,OAAO,GAAG,IAAI,GAAG,CAAS,cAAc,CAAC,CAAC;QAGhD,KAAK,MAAM,MAAM,IAAI,OAAO,EAAE,CAAC;YAC9B,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;QACrB,CAAC;QAGD,MAAM,KAAK,GAAG,CAAC,GAAG,OAAO,CAAC,CAAC;QAE3B,OAAO,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACzB,MAAM,aAAa,GAAG,KAAK,CAAC,KAAK,EAAG,CAAC;YACrC,MAAM,QAAQ,GAAG,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,aAAa,CAAC,CAAC;YACtE,IAAI,CAAC,QAAQ,EAAE,WAAW,EAAE,MAAM;gBAAE,SAAS;YAG7C,KAAK,MAAM,cAAc,IAAI,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,WAAW,CAAC,MAAM,CAAC,EAAE,CAAC;gBACvE,MAAM,UAAU,GAAG,IAAI,CAAC,4BAA4B,CAAC,cAAc,CAAC,CAAC;gBAErE,KAAK,MAAM,SAAS,IAAI,UAAU,EAAE,CAAC;oBACpC,IAAI,OAAO,CAAC,GAAG,CAAC,SAAS,CAAC;wBAAE,SAAS;oBAErC,OAAO,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;oBACvB,UAAU,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;oBAG1B,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;gBACvB,CAAC;YACF,CAAC;QACF,CAAC;QAED,OAAO,UAAU,CAAC;IACnB,CAAC;IAOD,WAAW,CAAC,MAAc;QACzB,OAAO,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,MAAM,CAAC,CAAC;IACtD,CAAC;IAQO,kBAAkB,CACzB,QAA8B,EAC9B,cAAkC;QAElC,MAAM,OAAO,GAAG,QAAQ,CAAC,OAAO,CAAC;QAEjC,IAAI,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC;YAE5B,IAAI,OAAO,CAAC,QAAQ,CAAC,cAAc,CAAC,EAAE,CAAC;gBACtC,OAAO,qBAAa,CAAC,gBAAgB,CAAC;YACvC,CAAC;QACF,CAAC;aAAM,IAAI,OAAO,OAAO,KAAK,QAAQ,EAAE,CAAC;YAExC,IAAI,OAAO,CAAC,QAAQ,CAAC,cAAc,CAAC,EAAE,CAAC;gBACtC,OAAO,qBAAa,CAAC,wBAAwB,CAAC;YAC/C,CAAC;QACF,CAAC;QAED,OAAO,CAAC,CAAC;IACV,CAAC;IAOD,MAAM,CAAC,kBAAkB,CAAC,cAAsB;QAC/C,OAAO,cAAc,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC;IACzC,CAAC;IAMD,MAAM,CAAC,oBAAoB;QAC1B,OAAO,MAAM,CAAC,MAAM,CAAC,kCAAmB,CAAC,CAAC,MAAM,CAC/C,CAAC,IAAI,EAA8B,EAAE,CACpC,oBAAoB,CAAC,IAAI,CAAC,IAAI,2BAA2B,CAAC,kBAAkB,CAAC,IAAI,CAAC,CACnF,CAAC;IACH,CAAC;CACD;AAzSD,kEAySC"}

package/dist/code-builder/handlers/agent-iteration-handler.d.ts

@@ -0,0 +1,36 @@
+import type { Callbacks } from '@langchain/core/callbacks/manager';
+import type { BaseChatModelCallOptions } from '@langchain/core/language_models/chat_models';
+import type { BaseMessage, AIMessage } from '@langchain/core/messages';
+import type { Runnable } from '@langchain/core/runnables';
+import type { StreamOutput } from '../../types/streaming';
+import type { TokenUsage } from '../types';
+export interface AgentIterationHandlerConfig {
+    onTokenUsage?: (usage: TokenUsage) => void;
+    callbacks?: Callbacks;
+    runMetadata?: Record<string, unknown>;
+}
+export interface IterationParams {
+    llmWithTools: Runnable<BaseMessage[], AIMessage, BaseChatModelCallOptions>;
+    messages: BaseMessage[];
+    abortSignal?: AbortSignal;
+    iteration: number;
+    callbacks?: Callbacks;
+}
+export interface LlmInvocationResult {
+    response: AIMessage;
+    inputTokens: number;
+    outputTokens: number;
+    llmDurationMs: number;
+    textContent: string | null;
+    thinkingContent: string | null;
+    hasToolCalls: boolean;
+}
+export declare class AgentIterationHandler {
+    private onTokenUsage?;
+    private callbacks?;
+    private runMetadata?;
+    constructor(config?: AgentIterationHandlerConfig);
+    getCallbacks(): Callbacks | undefined;
+    getRunMetadata(): Record<string, unknown> | undefined;
+    invokeLlm(params: IterationParams): AsyncGenerator<StreamOutput, LlmInvocationResult, unknown>;
+}

package/dist/code-builder/handlers/agent-iteration-handler.js

@@ -0,0 +1,78 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.AgentIterationHandler = void 0;
+const helpers_1 = require("../../utils/cache-control/helpers");
+const content_extractors_1 = require("../utils/content-extractors");
+function hasUsageMetadata(metadata) {
+    return (typeof metadata === 'object' &&
+        metadata !== null &&
+        'usage' in metadata &&
+        typeof metadata.usage === 'object');
+}
+class AgentIterationHandler {
+    onTokenUsage;
+    callbacks;
+    runMetadata;
+    constructor(config = {}) {
+        this.onTokenUsage = config.onTokenUsage;
+        this.callbacks = config.callbacks;
+        this.runMetadata = config.runMetadata;
+    }
+    getCallbacks() {
+        return this.callbacks;
+    }
+    getRunMetadata() {
+        return this.runMetadata;
+    }
+    async *invokeLlm(params) {
+        const { llmWithTools, messages, abortSignal, callbacks: iterationCallbacks } = params;
+        if (abortSignal?.aborted) {
+            throw new Error('Aborted');
+        }
+        (0, helpers_1.applySubgraphCacheMarkers)(messages);
+        const llmStartTime = Date.now();
+        const response = await llmWithTools.invoke(messages, {
+            signal: abortSignal,
+            callbacks: iterationCallbacks ?? this.callbacks,
+            metadata: this.runMetadata,
+        });
+        const llmDurationMs = Date.now() - llmStartTime;
+        const responseMetadata = response.response_metadata;
+        const inputTokens = hasUsageMetadata(responseMetadata)
+            ? (responseMetadata.usage.input_tokens ?? 0)
+            : 0;
+        const outputTokens = hasUsageMetadata(responseMetadata)
+            ? (responseMetadata.usage.output_tokens ?? 0)
+            : 0;
+        const thinkingTokens = hasUsageMetadata(responseMetadata)
+            ? (responseMetadata.usage.thinking_tokens ?? 0)
+            : 0;
+        if (this.onTokenUsage && (inputTokens > 0 || outputTokens > 0)) {
+            this.onTokenUsage({ inputTokens, outputTokens, thinkingTokens });
+        }
+        const thinkingContent = (0, content_extractors_1.extractThinkingContent)(response);
+        const textContent = (0, content_extractors_1.extractTextContent)(response);
+        if (textContent) {
+            const messageChunk = {
+                role: 'assistant',
+                type: 'message',
+                text: textContent,
+            };
+            yield {
+                messages: [messageChunk],
+            };
+        }
+        messages.push(response);
+        return {
+            response,
+            inputTokens,
+            outputTokens,
+            llmDurationMs,
+            textContent,
+            thinkingContent,
+            hasToolCalls: !!(response.tool_calls && response.tool_calls.length > 0),
+        };
+    }
+}
+exports.AgentIterationHandler = AgentIterationHandler;
+//# sourceMappingURL=agent-iteration-handler.js.map

package/dist/code-builder/handlers/agent-iteration-handler.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"agent-iteration-handler.js","sourceRoot":"","sources":["../../../src/code-builder/handlers/agent-iteration-handler.ts"],"names":[],"mappings":";;;AAaA,+DAA8E;AAE9E,oEAAyF;AAOzF,SAAS,gBAAgB,CAAC,QAAiB;IAC1C,OAAO,CACN,OAAO,QAAQ,KAAK,QAAQ;QAC5B,QAAQ,KAAK,IAAI;QACjB,OAAO,IAAI,QAAQ;QACnB,OAAQ,QAAsC,CAAC,KAAK,KAAK,QAAQ,CACjE,CAAC;AACH,CAAC;AA8DD,MAAa,qBAAqB;IACzB,YAAY,CAA+B;IAC3C,SAAS,CAAa;IACtB,WAAW,CAA2B;IAE9C,YAAY,SAAsC,EAAE;QACnD,IAAI,CAAC,YAAY,GAAG,MAAM,CAAC,YAAY,CAAC;QACxC,IAAI,CAAC,SAAS,GAAG,MAAM,CAAC,SAAS,CAAC;QAClC,IAAI,CAAC,WAAW,GAAG,MAAM,CAAC,WAAW,CAAC;IACvC,CAAC;IAGD,YAAY;QACX,OAAO,IAAI,CAAC,SAAS,CAAC;IACvB,CAAC;IAGD,cAAc;QACb,OAAO,IAAI,CAAC,WAAW,CAAC;IACzB,CAAC;IAgBD,KAAK,CAAC,CAAC,SAAS,CACf,MAAuB;QAEvB,MAAM,EAAE,YAAY,EAAE,QAAQ,EAAE,WAAW,EAAE,SAAS,EAAE,kBAAkB,EAAE,GAAG,MAAM,CAAC;QAGtF,IAAI,WAAW,EAAE,OAAO,EAAE,CAAC;YAC1B,MAAM,IAAI,KAAK,CAAC,SAAS,CAAC,CAAC;QAC5B,CAAC;QAGD,IAAA,mCAAyB,EAAC,QAAQ,CAAC,CAAC;QAEpC,MAAM,YAAY,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAChC,MAAM,QAAQ,GAAG,MAAM,YAAY,CAAC,MAAM,CAAC,QAAQ,EAAE;YACpD,MAAM,EAAE,WAAW;YACnB,SAAS,EAAE,kBAAkB,IAAI,IAAI,CAAC,SAAS;YAC/C,QAAQ,EAAE,IAAI,CAAC,WAAW;SAC1B,CAAC,CAAC;QACH,MAAM,aAAa,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,YAAY,CAAC;QAGhD,MAAM,gBAAgB,GAAG,QAAQ,CAAC,iBAAiB,CAAC;QACpD,MAAM,WAAW,GAAG,gBAAgB,CAAC,gBAAgB,CAAC;YACrD,CAAC,CAAC,CAAC,gBAAgB,CAAC,KAAK,CAAC,YAAY,IAAI,CAAC,CAAC;YAC5C,CAAC,CAAC,CAAC,CAAC;QACL,MAAM,YAAY,GAAG,gBAAgB,CAAC,gBAAgB,CAAC;YACtD,CAAC,CAAC,CAAC,gBAAgB,CAAC,KAAK,CAAC,aAAa,IAAI,CAAC,CAAC;YAC7C,CAAC,CAAC,CAAC,CAAC;QACL,MAAM,cAAc,GAAG,gBAAgB,CAAC,gBAAgB,CAAC;YACxD,CAAC,CAAC,CAAC,gBAAgB,CAAC,KAAK,CAAC,eAAe,IAAI,CAAC,CAAC;YAC/C,CAAC,CAAC,CAAC,CAAC;QAGL,IAAI,IAAI,CAAC,YAAY,IAAI,CAAC,WAAW,GAAG,CAAC,IAAI,YAAY,GAAG,CAAC,CAAC,EAAE,CAAC;YAChE,IAAI,CAAC,YAAY,CAAC,EAAE,WAAW,EAAE,YAAY,EAAE,cAAc,EAAE,CAAC,CAAC;QAClE,CAAC;QAED,MAAM,eAAe,GAAG,IAAA,2CAAsB,EAAC,QAAQ,CAAC,CAAC;QAGzD,MAAM,WAAW,GAAG,IAAA,uCAAkB,EAAC,QAAQ,CAAC,CAAC;QACjD,IAAI,WAAW,EAAE,CAAC;YACjB,MAAM,YAAY,GAAsB;gBACvC,IAAI,EAAE,WAAW;gBACjB,IAAI,EAAE,SAAS;gBACf,IAAI,EAAE,WAAW;aACjB,CAAC;YACF,MAAM;gBACL,QAAQ,EAAE,CAAC,YAAY,CAAC;aACxB,CAAC;QACH,CAAC;QAGD,QAAQ,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAExB,OAAO;YACN,QAAQ;YACR,WAAW;YACX,YAAY;YACZ,aAAa;YACb,WAAW;YACX,eAAe;YACf,YAAY,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC,UAAU,IAAI,QAAQ,CAAC,UAAU,CAAC,MAAM,GAAG,CAAC,CAAC;SACvE,CAAC;IACH,CAAC;CACD;AArGD,sDAqGC"}

package/dist/code-builder/handlers/auto-finalize-handler.d.ts

@@ -0,0 +1,29 @@
+import type { BaseMessage } from '@langchain/core/messages';
+import type { WorkflowJSON } from '@n8n/workflow-sdk';
+import type { StreamOutput } from '../../types/streaming';
+import type { WarningTracker } from '../state/warning-tracker';
+import type { ParseAndValidateResult } from '../types';
+type ParseAndValidateFn = (code: string, currentWorkflow?: WorkflowJSON) => Promise<ParseAndValidateResult>;
+type GetErrorContextFn = (code: string, errorMessage: string) => string;
+export interface AutoFinalizeHandlerConfig {
+    parseAndValidate: ParseAndValidateFn;
+    getErrorContext: GetErrorContextFn;
+}
+export interface AutoFinalizeParams {
+    code: string;
+    currentWorkflow: WorkflowJSON | undefined;
+    messages: BaseMessage[];
+    warningTracker?: WarningTracker;
+}
+export interface AutoFinalizeResult {
+    success: boolean;
+    workflow?: WorkflowJSON;
+    parseDuration?: number;
+}
+export declare class AutoFinalizeHandler {
+    private parseAndValidate;
+    private getErrorContext;
+    constructor(config: AutoFinalizeHandlerConfig);
+    execute(params: AutoFinalizeParams): AsyncGenerator<StreamOutput, AutoFinalizeResult, unknown>;
+}
+export {};

package/dist/code-builder/handlers/auto-finalize-handler.js

@@ -0,0 +1,50 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.AutoFinalizeHandler = void 0;
+const constants_1 = require("../constants");
+const content_extractors_1 = require("../utils/content-extractors");
+const format_warnings_1 = require("../utils/format-warnings");
+class AutoFinalizeHandler {
+    parseAndValidate;
+    getErrorContext;
+    constructor(config) {
+        this.parseAndValidate = config.parseAndValidate;
+        this.getErrorContext = config.getErrorContext;
+    }
+    async *execute(params) {
+        const { code, currentWorkflow, messages, warningTracker } = params;
+        const parseStartTime = Date.now();
+        try {
+            const result = await this.parseAndValidate(code, currentWorkflow);
+            const parseDuration = Date.now() - parseStartTime;
+            if (result.warnings.length > 0) {
+                const newWarnings = warningTracker
+                    ? warningTracker.filterNewWarnings(result.warnings)
+                    : result.warnings;
+                if (newWarnings.length > 0) {
+                    if (warningTracker) {
+                        warningTracker.markAsSeen(newWarnings);
+                    }
+                    const warningText = (0, format_warnings_1.formatWarnings)(newWarnings, warningTracker);
+                    const errorContext = this.getErrorContext(code, newWarnings[0].message);
+                    (0, content_extractors_1.pushValidationFeedback)(messages, `Validation warnings:\n${warningText}\n\n${errorContext}\n\n${constants_1.FIX_VALIDATION_ERRORS_INSTRUCTION}`);
+                    return { success: false, parseDuration };
+                }
+            }
+            return {
+                success: true,
+                workflow: result.workflow,
+                parseDuration,
+            };
+        }
+        catch (error) {
+            const parseDuration = Date.now() - parseStartTime;
+            const errorMessage = error instanceof Error ? error.message : String(error);
+            const errorContext = this.getErrorContext(code, errorMessage);
+            (0, content_extractors_1.pushValidationFeedback)(messages, `Parse error: ${errorMessage}\n\n${errorContext}\n\n${constants_1.FIX_VALIDATION_ERRORS_INSTRUCTION}`);
+            return { success: false, parseDuration };
+        }
+    }
+}
+exports.AutoFinalizeHandler = AutoFinalizeHandler;
+//# sourceMappingURL=auto-finalize-handler.js.map

package/dist/code-builder/handlers/auto-finalize-handler.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"auto-finalize-handler.js","sourceRoot":"","sources":["../../../src/code-builder/handlers/auto-finalize-handler.ts"],"names":[],"mappings":";;;AAWA,4CAAiE;AAGjE,oEAAqE;AACrE,8DAA0D;AAwD1D,MAAa,mBAAmB;IACvB,gBAAgB,CAAqB;IACrC,eAAe,CAAoB;IAE3C,YAAY,MAAiC;QAC5C,IAAI,CAAC,gBAAgB,GAAG,MAAM,CAAC,gBAAgB,CAAC;QAChD,IAAI,CAAC,eAAe,GAAG,MAAM,CAAC,eAAe,CAAC;IAC/C,CAAC;IAUD,KAAK,CAAC,CAAC,OAAO,CACb,MAA0B;QAE1B,MAAM,EAAE,IAAI,EAAE,eAAe,EAAE,QAAQ,EAAE,cAAc,EAAE,GAAG,MAAM,CAAC;QAGnE,MAAM,cAAc,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAClC,IAAI,CAAC;YACJ,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,gBAAgB,CAAC,IAAI,EAAE,eAAe,CAAC,CAAC;YAClE,MAAM,aAAa,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,cAAc,CAAC;YAGlD,IAAI,MAAM,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBAChC,MAAM,WAAW,GAAG,cAAc;oBACjC,CAAC,CAAC,cAAc,CAAC,iBAAiB,CAAC,MAAM,CAAC,QAAQ,CAAC;oBACnD,CAAC,CAAC,MAAM,CAAC,QAAQ,CAAC;gBAEnB,IAAI,WAAW,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;oBAC5B,IAAI,cAAc,EAAE,CAAC;wBACpB,cAAc,CAAC,UAAU,CAAC,WAAW,CAAC,CAAC;oBACxC,CAAC;oBAED,MAAM,WAAW,GAAG,IAAA,gCAAc,EAAC,WAAW,EAAE,cAAc,CAAC,CAAC;oBAChE,MAAM,YAAY,GAAG,IAAI,CAAC,eAAe,CAAC,IAAI,EAAE,WAAW,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC;oBAGxE,IAAA,2CAAsB,EACrB,QAAQ,EACR,yBAAyB,WAAW,OAAO,YAAY,OAAO,6CAAiC,EAAE,CACjG,CAAC;oBAEF,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,aAAa,EAAE,CAAC;gBAC1C,CAAC;YACF,CAAC;YAGD,OAAO;gBACN,OAAO,EAAE,IAAI;gBACb,QAAQ,EAAE,MAAM,CAAC,QAAQ;gBACzB,aAAa;aACb,CAAC;QACH,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YAChB,MAAM,aAAa,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,cAAc,CAAC;YAClD,MAAM,YAAY,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;YAC5E,MAAM,YAAY,GAAG,IAAI,CAAC,eAAe,CAAC,IAAI,EAAE,YAAY,CAAC,CAAC;YAG9D,IAAA,2CAAsB,EACrB,QAAQ,EACR,gBAAgB,YAAY,OAAO,YAAY,OAAO,6CAAiC,EAAE,CACzF,CAAC;YAEF,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,aAAa,EAAE,CAAC;QAC1C,CAAC;IACF,CAAC;CACD;AAxED,kDAwEC"}