@n8n/ai-workflow-builder 0.12.1 → 0.14.0

package/dist/utils/operations-processor.js

@@ -0,0 +1,114 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.applyOperations = applyOperations;
+exports.processOperations = processOperations;
+function applyOperations(workflow, operations) {
+    let result = {
+        nodes: [...workflow.nodes],
+        connections: { ...workflow.connections },
+    };
+    for (const operation of operations) {
+        switch (operation.type) {
+            case 'clear':
+                result = { nodes: [], connections: {} };
+                break;
+            case 'removeNode': {
+                const nodesToRemove = new Set(operation.nodeIds);
+                result.nodes = result.nodes.filter((node) => !nodesToRemove.has(node.id));
+                const cleanedConnections = {};
+                for (const [sourceId, nodeConnections] of Object.entries(result.connections)) {
+                    if (!nodesToRemove.has(sourceId)) {
+                        cleanedConnections[sourceId] = {};
+                        for (const [connectionType, outputs] of Object.entries(nodeConnections)) {
+                            if (Array.isArray(outputs)) {
+                                cleanedConnections[sourceId][connectionType] = outputs.map((outputConnections) => {
+                                    if (Array.isArray(outputConnections)) {
+                                        return outputConnections.filter((conn) => !nodesToRemove.has(conn.node));
+                                    }
+                                    return outputConnections;
+                                });
+                            }
+                        }
+                    }
+                }
+                result.connections = cleanedConnections;
+                break;
+            }
+            case 'addNodes': {
+                const nodeMap = new Map();
+                result.nodes.forEach((node) => nodeMap.set(node.id, node));
+                operation.nodes.forEach((node) => {
+                    nodeMap.set(node.id, node);
+                });
+                result.nodes = Array.from(nodeMap.values());
+                break;
+            }
+            case 'updateNode': {
+                result.nodes = result.nodes.map((node) => {
+                    if (node.id === operation.nodeId) {
+                        return { ...node, ...operation.updates };
+                    }
+                    return node;
+                });
+                break;
+            }
+            case 'setConnections': {
+                result.connections = operation.connections;
+                break;
+            }
+            case 'mergeConnections': {
+                for (const [sourceId, nodeConnections] of Object.entries(operation.connections)) {
+                    if (!result.connections[sourceId]) {
+                        result.connections[sourceId] = nodeConnections;
+                    }
+                    else {
+                        for (const [connectionType, newOutputs] of Object.entries(nodeConnections)) {
+                            if (!result.connections[sourceId][connectionType]) {
+                                result.connections[sourceId][connectionType] = newOutputs;
+                            }
+                            else {
+                                const existingOutputs = result.connections[sourceId][connectionType];
+                                if (Array.isArray(newOutputs) && Array.isArray(existingOutputs)) {
+                                    for (let i = 0; i < Math.max(newOutputs.length, existingOutputs.length); i++) {
+                                        if (!newOutputs[i])
+                                            continue;
+                                        if (!existingOutputs[i]) {
+                                            existingOutputs[i] = newOutputs[i];
+                                        }
+                                        else if (Array.isArray(newOutputs[i]) && Array.isArray(existingOutputs[i])) {
+                                            const existingSet = new Set(existingOutputs[i].map((conn) => JSON.stringify({ node: conn.node, type: conn.type, index: conn.index })));
+                                            newOutputs[i].forEach((conn) => {
+                                                const connStr = JSON.stringify({
+                                                    node: conn.node,
+                                                    type: conn.type,
+                                                    index: conn.index,
+                                                });
+                                                if (!existingSet.has(connStr)) {
+                                                    existingOutputs[i].push(conn);
+                                                }
+                                            });
+                                        }
+                                    }
+                                }
+                            }
+                        }
+                    }
+                }
+                break;
+            }
+        }
+    }
+    return result;
+}
+function processOperations(state) {
+    const { workflowJSON, workflowOperations } = state;
+    if (!workflowOperations || workflowOperations.length === 0) {
+        return {};
+    }
+    const newWorkflow = applyOperations(workflowJSON, workflowOperations);
+    return {
+        workflowJSON: newWorkflow,
+        workflowOperations: null,
+    };
+}
+//# sourceMappingURL=operations-processor.js.map

package/dist/utils/operations-processor.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"operations-processor.js","sourceRoot":"","sources":["../../src/utils/operations-processor.ts"],"names":[],"mappings":";;AASA,0CAkIC;AAMD,8CAgBC;AAxJD,SAAgB,eAAe,CAC9B,QAAwB,EACxB,UAA+B;IAG/B,IAAI,MAAM,GAAmB;QAC5B,KAAK,EAAE,CAAC,GAAG,QAAQ,CAAC,KAAK,CAAC;QAC1B,WAAW,EAAE,EAAE,GAAG,QAAQ,CAAC,WAAW,EAAE;KACxC,CAAC;IAGF,KAAK,MAAM,SAAS,IAAI,UAAU,EAAE,CAAC;QACpC,QAAQ,SAAS,CAAC,IAAI,EAAE,CAAC;YACxB,KAAK,OAAO;gBACX,MAAM,GAAG,EAAE,KAAK,EAAE,EAAE,EAAE,WAAW,EAAE,EAAE,EAAE,CAAC;gBACxC,MAAM;YAEP,KAAK,YAAY,CAAC,CAAC,CAAC;gBACnB,MAAM,aAAa,GAAG,IAAI,GAAG,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;gBAGjD,MAAM,CAAC,KAAK,GAAG,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,aAAa,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,CAAC;gBAG1E,MAAM,kBAAkB,GAAiB,EAAE,CAAC;gBAG5C,KAAK,MAAM,CAAC,QAAQ,EAAE,eAAe,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,WAAW,CAAC,EAAE,CAAC;oBAC9E,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,QAAQ,CAAC,EAAE,CAAC;wBAClC,kBAAkB,CAAC,QAAQ,CAAC,GAAG,EAAE,CAAC;wBAElC,KAAK,MAAM,CAAC,cAAc,EAAE,OAAO,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,eAAe,CAAC,EAAE,CAAC;4BACzE,IAAI,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC;gCAC5B,kBAAkB,CAAC,QAAQ,CAAC,CAAC,cAAc,CAAC,GAAG,OAAO,CAAC,GAAG,CAAC,CAAC,iBAAiB,EAAE,EAAE;oCAChF,IAAI,KAAK,CAAC,OAAO,CAAC,iBAAiB,CAAC,EAAE,CAAC;wCACtC,OAAO,iBAAiB,CAAC,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,aAAa,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC;oCAC1E,CAAC;oCACD,OAAO,iBAAiB,CAAC;gCAC1B,CAAC,CAAC,CAAC;4BACJ,CAAC;wBACF,CAAC;oBACF,CAAC;gBACF,CAAC;gBAED,MAAM,CAAC,WAAW,GAAG,kBAAkB,CAAC;gBACxC,MAAM;YACP,CAAC;YAED,KAAK,UAAU,CAAC,CAAC,CAAC;gBAEjB,MAAM,OAAO,GAAG,IAAI,GAAG,EAAiB,CAAC;gBACzC,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,EAAE,IAAI,CAAC,CAAC,CAAC;gBAG3D,SAAS,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,EAAE;oBAChC,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,EAAE,IAAI,CAAC,CAAC;gBAC5B,CAAC,CAAC,CAAC;gBAEH,MAAM,CAAC,KAAK,GAAG,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC;gBAC5C,MAAM;YACP,CAAC;YAED,KAAK,YAAY,CAAC,CAAC,CAAC;gBACnB,MAAM,CAAC,KAAK,GAAG,MAAM,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE;oBACxC,IAAI,IAAI,CAAC,EAAE,KAAK,SAAS,CAAC,MAAM,EAAE,CAAC;wBAClC,OAAO,EAAE,GAAG,IAAI,EAAE,GAAG,SAAS,CAAC,OAAO,EAAE,CAAC;oBAC1C,CAAC;oBACD,OAAO,IAAI,CAAC;gBACb,CAAC,CAAC,CAAC;gBACH,MAAM;YACP,CAAC;YAED,KAAK,gBAAgB,CAAC,CAAC,CAAC;gBAEvB,MAAM,CAAC,WAAW,GAAG,SAAS,CAAC,WAAW,CAAC;gBAC3C,MAAM;YACP,CAAC;YAED,KAAK,kBAAkB,CAAC,CAAC,CAAC;gBAEzB,KAAK,MAAM,CAAC,QAAQ,EAAE,eAAe,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,SAAS,CAAC,WAAW,CAAC,EAAE,CAAC;oBACjF,IAAI,CAAC,MAAM,CAAC,WAAW,CAAC,QAAQ,CAAC,EAAE,CAAC;wBACnC,MAAM,CAAC,WAAW,CAAC,QAAQ,CAAC,GAAG,eAAe,CAAC;oBAChD,CAAC;yBAAM,CAAC;wBAEP,KAAK,MAAM,CAAC,cAAc,EAAE,UAAU,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,eAAe,CAAC,EAAE,CAAC;4BAC5E,IAAI,CAAC,MAAM,CAAC,WAAW,CAAC,QAAQ,CAAC,CAAC,cAAc,CAAC,EAAE,CAAC;gCACnD,MAAM,CAAC,WAAW,CAAC,QAAQ,CAAC,CAAC,cAAc,CAAC,GAAG,UAAU,CAAC;4BAC3D,CAAC;iCAAM,CAAC;gCAEP,MAAM,eAAe,GAAG,MAAM,CAAC,WAAW,CAAC,QAAQ,CAAC,CAAC,cAAc,CAAC,CAAC;gCAErE,IAAI,KAAK,CAAC,OAAO,CAAC,UAAU,CAAC,IAAI,KAAK,CAAC,OAAO,CAAC,eAAe,CAAC,EAAE,CAAC;oCAEjE,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,GAAG,CAAC,UAAU,CAAC,MAAM,EAAE,eAAe,CAAC,MAAM,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;wCAC9E,IAAI,CAAC,UAAU,CAAC,CAAC,CAAC;4CAAE,SAAS;wCAE7B,IAAI,CAAC,eAAe,CAAC,CAAC,CAAC,EAAE,CAAC;4CACzB,eAAe,CAAC,CAAC,CAAC,GAAG,UAAU,CAAC,CAAC,CAAC,CAAC;wCACpC,CAAC;6CAAM,IAAI,KAAK,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;4CAE9E,MAAM,WAAW,GAAG,IAAI,GAAG,CAC1B,eAAe,CAAC,CAAC,CAAE,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAChC,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,KAAK,EAAE,IAAI,CAAC,KAAK,EAAE,CAAC,CACvE,CACD,CAAC;4CAEF,UAAU,CAAC,CAAC,CAAE,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,EAAE;gDAC/B,MAAM,OAAO,GAAG,IAAI,CAAC,SAAS,CAAC;oDAC9B,IAAI,EAAE,IAAI,CAAC,IAAI;oDACf,IAAI,EAAE,IAAI,CAAC,IAAI;oDACf,KAAK,EAAE,IAAI,CAAC,KAAK;iDACjB,CAAC,CAAC;gDACH,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC;oDAC/B,eAAe,CAAC,CAAC,CAAE,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;gDAChC,CAAC;4CACF,CAAC,CAAC,CAAC;wCACJ,CAAC;oCACF,CAAC;gCACF,CAAC;4BACF,CAAC;wBACF,CAAC;oBACF,CAAC;gBACF,CAAC;gBACD,MAAM;YACP,CAAC;QACF,CAAC;IACF,CAAC;IAED,OAAO,MAAM,CAAC;AACf,CAAC;AAMD,SAAgB,iBAAiB,CAAC,KAAiC;IAClE,MAAM,EAAE,YAAY,EAAE,kBAAkB,EAAE,GAAG,KAAK,CAAC;IAGnD,IAAI,CAAC,kBAAkB,IAAI,kBAAkB,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QAC5D,OAAO,EAAE,CAAC;IACX,CAAC;IAGD,MAAM,WAAW,GAAG,eAAe,CAAC,YAAY,EAAE,kBAAkB,CAAC,CAAC;IAGtE,OAAO;QACN,YAAY,EAAE,WAAW;QACzB,kBAAkB,EAAE,IAAI;KACxB,CAAC;AACH,CAAC"}

package/dist/utils/stream-processor.d.ts

@@ -0,0 +1,6 @@
+import { AIMessage, HumanMessage, ToolMessage } from '@langchain/core/messages';
+import type { StreamOutput } from '../types/streaming';
+export declare const DEFAULT_WORKFLOW_UPDATE_TOOLS: string[];
+export declare function processStreamChunk(streamMode: string, chunk: unknown): StreamOutput | null;
+export declare function createStreamProcessor(stream: AsyncGenerator<[string, unknown], void, unknown>): AsyncGenerator<StreamOutput>;
+export declare function formatMessages(messages: Array<AIMessage | HumanMessage | ToolMessage>): Array<Record<string, unknown>>;

package/dist/utils/stream-processor.js

@@ -0,0 +1,153 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.DEFAULT_WORKFLOW_UPDATE_TOOLS = void 0;
+exports.processStreamChunk = processStreamChunk;
+exports.createStreamProcessor = createStreamProcessor;
+exports.formatMessages = formatMessages;
+const messages_1 = require("@langchain/core/messages");
+exports.DEFAULT_WORKFLOW_UPDATE_TOOLS = [
+    'add_nodes',
+    'connect_nodes',
+    'update_node_parameters',
+    'remove_node',
+];
+function processStreamChunk(streamMode, chunk) {
+    if (streamMode === 'updates') {
+        const agentChunk = chunk;
+        if ((agentChunk?.delete_messages?.messages ?? []).length > 0) {
+            const messageChunk = {
+                role: 'assistant',
+                type: 'message',
+                text: 'Deleted, refresh?',
+            };
+            return { messages: [messageChunk] };
+        }
+        if ((agentChunk?.compact_messages?.messages ?? []).length > 0) {
+            const lastMessage = agentChunk.compact_messages.messages[agentChunk.compact_messages.messages.length - 1];
+            const messageChunk = {
+                role: 'assistant',
+                type: 'message',
+                text: lastMessage.content,
+            };
+            return { messages: [messageChunk] };
+        }
+        if ((agentChunk?.agent?.messages ?? []).length > 0) {
+            const lastMessage = agentChunk.agent.messages[agentChunk.agent.messages.length - 1];
+            if (lastMessage.content) {
+                let content;
+                if (Array.isArray(lastMessage.content)) {
+                    content = lastMessage.content
+                        .filter((c) => c.type === 'text')
+                        .map((b) => b.text)
+                        .join('\n');
+                }
+                else {
+                    content = lastMessage.content;
+                }
+                const messageChunk = {
+                    role: 'assistant',
+                    type: 'message',
+                    text: content,
+                };
+                return { messages: [messageChunk] };
+            }
+        }
+        if (agentChunk?.process_operations) {
+            const update = agentChunk.process_operations;
+            if (update.workflowJSON && update.workflowOperations !== undefined) {
+                const workflowUpdateChunk = {
+                    role: 'assistant',
+                    type: 'workflow-updated',
+                    codeSnippet: JSON.stringify(update.workflowJSON, null, 2),
+                };
+                return { messages: [workflowUpdateChunk] };
+            }
+        }
+    }
+    else if (streamMode === 'custom') {
+        const toolChunk = chunk;
+        if (toolChunk?.type === 'tool') {
+            const output = { messages: [toolChunk] };
+            return output;
+        }
+    }
+    return null;
+}
+async function* createStreamProcessor(stream) {
+    for await (const [streamMode, chunk] of stream) {
+        const output = processStreamChunk(streamMode, chunk);
+        if (output) {
+            yield output;
+        }
+    }
+}
+function formatMessages(messages) {
+    const formattedMessages = [];
+    for (const msg of messages) {
+        if (msg instanceof messages_1.HumanMessage) {
+            formattedMessages.push({
+                role: 'user',
+                type: 'message',
+                text: msg.content,
+            });
+        }
+        else if (msg instanceof messages_1.AIMessage) {
+            if (msg.content) {
+                if (Array.isArray(msg.content)) {
+                    const textMessages = msg.content.filter((c) => c.type === 'text');
+                    textMessages.forEach((textMessage) => {
+                        if (textMessage.type !== 'text') {
+                            return;
+                        }
+                        formattedMessages.push({
+                            role: 'assistant',
+                            type: 'message',
+                            text: textMessage.text,
+                        });
+                    });
+                }
+                else {
+                    formattedMessages.push({
+                        role: 'assistant',
+                        type: 'message',
+                        text: msg.content,
+                    });
+                }
+            }
+            if (msg.tool_calls && msg.tool_calls.length > 0) {
+                for (const toolCall of msg.tool_calls) {
+                    formattedMessages.push({
+                        id: toolCall.id,
+                        toolCallId: toolCall.id,
+                        role: 'assistant',
+                        type: 'tool',
+                        toolName: toolCall.name,
+                        status: 'completed',
+                        updates: [
+                            {
+                                type: 'input',
+                                data: toolCall.args || {},
+                            },
+                        ],
+                    });
+                }
+            }
+        }
+        else if (msg instanceof messages_1.ToolMessage) {
+            const toolCallId = msg.tool_call_id;
+            for (let i = formattedMessages.length - 1; i >= 0; i--) {
+                const m = formattedMessages[i];
+                if (m.type === 'tool' && m.id === toolCallId) {
+                    m.updates ??= [];
+                    m.updates.push({
+                        type: 'output',
+                        data: typeof msg.content === 'string' ? { result: msg.content } : msg.content,
+                    });
+                    break;
+                }
+            }
+        }
+    }
+    return formattedMessages;
+}
+//# sourceMappingURL=stream-processor.js.map

package/dist/utils/stream-processor.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"stream-processor.js","sourceRoot":"","sources":["../../src/utils/stream-processor.ts"],"names":[],"mappings":";;;AAuBA,gDA2FC;AAKD,sDAUC;AAED,wCA4EC;AA/MD,uDAAgF;AAYnE,QAAA,6BAA6B,GAAG;IAC5C,WAAW;IACX,eAAe;IACf,wBAAwB;IACxB,aAAa;CACb,CAAC;AAMF,SAAgB,kBAAkB,CAAC,UAAkB,EAAE,KAAc;IACpE,IAAI,UAAU,KAAK,SAAS,EAAE,CAAC;QAE9B,MAAM,UAAU,GAAG,KAYlB,CAAC;QAEF,IAAI,CAAC,UAAU,EAAE,eAAe,EAAE,QAAQ,IAAI,EAAE,CAAC,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC9D,MAAM,YAAY,GAAsB;gBACvC,IAAI,EAAE,WAAW;gBACjB,IAAI,EAAE,SAAS;gBACf,IAAI,EAAE,mBAAmB;aACzB,CAAC;YAEF,OAAO,EAAE,QAAQ,EAAE,CAAC,YAAY,CAAC,EAAE,CAAC;QACrC,CAAC;QAED,IAAI,CAAC,UAAU,EAAE,gBAAgB,EAAE,QAAQ,IAAI,EAAE,CAAC,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC/D,MAAM,WAAW,GAChB,UAAU,CAAC,gBAAiB,CAAC,QAAS,CAAC,UAAU,CAAC,gBAAiB,CAAC,QAAS,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;YAC3F,MAAM,YAAY,GAAsB;gBACvC,IAAI,EAAE,WAAW;gBACjB,IAAI,EAAE,SAAS;gBACf,IAAI,EAAE,WAAW,CAAC,OAAiB;aACnC,CAAC;YAEF,OAAO,EAAE,QAAQ,EAAE,CAAC,YAAY,CAAC,EAAE,CAAC;QACrC,CAAC;QAED,IAAI,CAAC,UAAU,EAAE,KAAK,EAAE,QAAQ,IAAI,EAAE,CAAC,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACpD,MAAM,WAAW,GAAG,UAAU,CAAC,KAAM,CAAC,QAAS,CAAC,UAAU,CAAC,KAAM,CAAC,QAAS,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;YACxF,IAAI,WAAW,CAAC,OAAO,EAAE,CAAC;gBACzB,IAAI,OAAe,CAAC;gBAGpB,IAAI,KAAK,CAAC,OAAO,CAAC,WAAW,CAAC,OAAO,CAAC,EAAE,CAAC;oBACxC,OAAO,GAAG,WAAW,CAAC,OAAO;yBAC3B,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,MAAM,CAAC;yBAChC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC;yBAClB,IAAI,CAAC,IAAI,CAAC,CAAC;gBACd,CAAC;qBAAM,CAAC;oBACP,OAAO,GAAG,WAAW,CAAC,OAAO,CAAC;gBAC/B,CAAC;gBAED,MAAM,YAAY,GAAsB;oBACvC,IAAI,EAAE,WAAW;oBACjB,IAAI,EAAE,SAAS;oBACf,IAAI,EAAE,OAAO;iBACb,CAAC;gBAEF,OAAO,EAAE,QAAQ,EAAE,CAAC,YAAY,CAAC,EAAE,CAAC;YACrC,CAAC;QACF,CAAC;QAGD,IAAI,UAAU,EAAE,kBAAkB,EAAE,CAAC;YAEpC,MAAM,MAAM,GAAG,UAAU,CAAC,kBAAkB,CAAC;YAC7C,IAAI,MAAM,CAAC,YAAY,IAAI,MAAM,CAAC,kBAAkB,KAAK,SAAS,EAAE,CAAC;gBAEpE,MAAM,mBAAmB,GAAwB;oBAChD,IAAI,EAAE,WAAW;oBACjB,IAAI,EAAE,kBAAkB;oBACxB,WAAW,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,YAAY,EAAE,IAAI,EAAE,CAAC,CAAC;iBACzD,CAAC;gBAEF,OAAO,EAAE,QAAQ,EAAE,CAAC,mBAAmB,CAAC,EAAE,CAAC;YAC5C,CAAC;QACF,CAAC;IACF,CAAC;SAAM,IAAI,UAAU,KAAK,QAAQ,EAAE,CAAC;QAEpC,MAAM,SAAS,GAAG,KAA0B,CAAC;QAE7C,IAAI,SAAS,EAAE,IAAI,KAAK,MAAM,EAAE,CAAC;YAChC,MAAM,MAAM,GAAiB,EAAE,QAAQ,EAAE,CAAC,SAAS,CAAC,EAAE,CAAC;YAEvD,OAAO,MAAM,CAAC;QACf,CAAC;IACF,CAAC;IAED,OAAO,IAAI,CAAC;AACb,CAAC;AAKM,KAAK,SAAS,CAAC,CAAC,qBAAqB,CAC3C,MAAwD;IAExD,IAAI,KAAK,EAAE,MAAM,CAAC,UAAU,EAAE,KAAK,CAAC,IAAI,MAAM,EAAE,CAAC;QAChD,MAAM,MAAM,GAAG,kBAAkB,CAAC,UAAU,EAAE,KAAK,CAAC,CAAC;QAErD,IAAI,MAAM,EAAE,CAAC;YACZ,MAAM,MAAM,CAAC;QACd,CAAC;IACF,CAAC;AACF,CAAC;AAED,SAAgB,cAAc,CAC7B,QAAuD;IAEvD,MAAM,iBAAiB,GAAmC,EAAE,CAAC;IAE7D,KAAK,MAAM,GAAG,IAAI,QAAQ,EAAE,CAAC;QAC5B,IAAI,GAAG,YAAY,uBAAY,EAAE,CAAC;YACjC,iBAAiB,CAAC,IAAI,CAAC;gBACtB,IAAI,EAAE,MAAM;gBACZ,IAAI,EAAE,SAAS;gBACf,IAAI,EAAE,GAAG,CAAC,OAAO;aACjB,CAAC,CAAC;QACJ,CAAC;aAAM,IAAI,GAAG,YAAY,oBAAS,EAAE,CAAC;YAErC,IAAI,GAAG,CAAC,OAAO,EAAE,CAAC;gBACjB,IAAI,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC;oBAEhC,MAAM,YAAY,GAAG,GAAG,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,MAAM,CAAC,CAAC;oBAElE,YAAY,CAAC,OAAO,CAAC,CAAC,WAAW,EAAE,EAAE;wBACpC,IAAI,WAAW,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;4BACjC,OAAO;wBACR,CAAC;wBACD,iBAAiB,CAAC,IAAI,CAAC;4BACtB,IAAI,EAAE,WAAW;4BACjB,IAAI,EAAE,SAAS;4BACf,IAAI,EAAE,WAAW,CAAC,IAAI;yBACtB,CAAC,CAAC;oBACJ,CAAC,CAAC,CAAC;gBACJ,CAAC;qBAAM,CAAC;oBACP,iBAAiB,CAAC,IAAI,CAAC;wBACtB,IAAI,EAAE,WAAW;wBACjB,IAAI,EAAE,SAAS;wBACf,IAAI,EAAE,GAAG,CAAC,OAAO;qBACjB,CAAC,CAAC;gBACJ,CAAC;YACF,CAAC;YAED,IAAI,GAAG,CAAC,UAAU,IAAI,GAAG,CAAC,UAAU,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBAEjD,KAAK,MAAM,QAAQ,IAAI,GAAG,CAAC,UAAU,EAAE,CAAC;oBACvC,iBAAiB,CAAC,IAAI,CAAC;wBACtB,EAAE,EAAE,QAAQ,CAAC,EAAE;wBACf,UAAU,EAAE,QAAQ,CAAC,EAAE;wBACvB,IAAI,EAAE,WAAW;wBACjB,IAAI,EAAE,MAAM;wBACZ,QAAQ,EAAE,QAAQ,CAAC,IAAI;wBACvB,MAAM,EAAE,WAAW;wBACnB,OAAO,EAAE;4BACR;gCACC,IAAI,EAAE,OAAO;gCACb,IAAI,EAAE,QAAQ,CAAC,IAAI,IAAI,EAAE;6BACzB;yBACD;qBACD,CAAC,CAAC;gBACJ,CAAC;YACF,CAAC;QACF,CAAC;aAAM,IAAI,GAAG,YAAY,sBAAW,EAAE,CAAC;YAEvC,MAAM,UAAU,GAAG,GAAG,CAAC,YAAY,CAAC;YACpC,KAAK,IAAI,CAAC,GAAG,iBAAiB,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;gBACxD,MAAM,CAAC,GAAG,iBAAiB,CAAC,CAAC,CAAC,CAAC;gBAC/B,IAAI,CAAC,CAAC,IAAI,KAAK,MAAM,IAAI,CAAC,CAAC,EAAE,KAAK,UAAU,EAAE,CAAC;oBAE9C,CAAC,CAAC,OAAO,KAAK,EAAE,CAAC;oBAChB,CAAC,CAAC,OAA0C,CAAC,IAAI,CAAC;wBAClD,IAAI,EAAE,QAAQ;wBACd,IAAI,EAAE,OAAO,GAAG,CAAC,OAAO,KAAK,QAAQ,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,GAAG,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO;qBAC7E,CAAC,CAAC;oBACH,MAAM;gBACP,CAAC;YACF,CAAC;QACF,CAAC;IACF,CAAC;IAED,OAAO,iBAAiB,CAAC;AAC1B,CAAC"}

package/dist/utils/tool-executor.d.ts

@@ -0,0 +1,3 @@
+import type { ToolExecutorOptions } from '../types/config';
+import type { WorkflowState } from '../workflow-state';
+export declare function executeToolsInParallel(options: ToolExecutorOptions): Promise<Partial<typeof WorkflowState.State>>;

package/dist/utils/tool-executor.js

@@ -0,0 +1,86 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.executeToolsInParallel = executeToolsInParallel;
+const messages_1 = require("@langchain/core/messages");
+const tools_1 = require("@langchain/core/tools");
+const langgraph_1 = require("@langchain/langgraph");
+const errors_1 = require("../errors");
+async function executeToolsInParallel(options) {
+    const { state, toolMap } = options;
+    const lastMessage = state.messages.at(-1);
+    if (!lastMessage || !(0, messages_1.isAIMessage)(lastMessage)) {
+        const error = new errors_1.WorkflowStateError('Most recent message must be an AIMessage with tool calls');
+        throw error;
+    }
+    const aiMessage = lastMessage;
+    if (!aiMessage.tool_calls?.length) {
+        const error = new errors_1.WorkflowStateError('AIMessage must have tool calls');
+        throw error;
+    }
+    const toolResults = await Promise.all(aiMessage.tool_calls.map(async (toolCall) => {
+        try {
+            const tool = toolMap.get(toolCall.name);
+            if (!tool) {
+                throw new errors_1.ToolExecutionError(`Tool ${toolCall.name} not found`, {
+                    toolName: toolCall.name,
+                });
+            }
+            const result = await tool.invoke(toolCall.args ?? {}, {
+                toolCall: {
+                    id: toolCall.id,
+                    name: toolCall.name,
+                    args: toolCall.args ?? {},
+                },
+            });
+            return result;
+        }
+        catch (error) {
+            const errorMessage = error instanceof Error ? error.message : 'Unknown error occurred';
+            let errorContent;
+            if (error instanceof tools_1.ToolInputParsingException ||
+                errorMessage.includes('expected schema')) {
+                errorContent = `Invalid input for tool ${toolCall.name}: ${errorMessage}`;
+            }
+            else {
+                errorContent = `Tool ${toolCall.name} failed: ${errorMessage}`;
+            }
+            return new messages_1.ToolMessage({
+                content: errorContent,
+                tool_call_id: toolCall.id ?? '',
+                additional_kwargs: { error: true },
+            });
+        }
+    }));
+    const allMessages = [];
+    const stateUpdates = [];
+    toolResults.forEach((result) => {
+        if ((0, langgraph_1.isCommand)(result)) {
+            const update = result.update;
+            if (update) {
+                stateUpdates.push(update);
+            }
+        }
+        else {
+            allMessages.push(result);
+        }
+    });
+    stateUpdates.forEach((update) => {
+        if (update.messages && Array.isArray(update.messages)) {
+            allMessages.push(...update.messages);
+        }
+    });
+    const allOperations = [];
+    for (const update of stateUpdates) {
+        if (update.workflowOperations && Array.isArray(update.workflowOperations)) {
+            allOperations.push(...update.workflowOperations);
+        }
+    }
+    const finalUpdate = {
+        messages: allMessages,
+    };
+    if (allOperations.length > 0) {
+        finalUpdate.workflowOperations = allOperations;
+    }
+    return finalUpdate;
+}
+//# sourceMappingURL=tool-executor.js.map

package/dist/utils/tool-executor.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"tool-executor.js","sourceRoot":"","sources":["../../src/utils/tool-executor.ts"],"names":[],"mappings":";;AAgCA,wDA4GC;AA3ID,uDAAoE;AACpE,iDAAkE;AAClE,oDAAiD;AAEjD,sCAAmE;AA2B5D,KAAK,UAAU,sBAAsB,CAC3C,OAA4B;IAE5B,MAAM,EAAE,KAAK,EAAE,OAAO,EAAE,GAAG,OAAO,CAAC;IACnC,MAAM,WAAW,GAAG,KAAK,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;IAE1C,IAAI,CAAC,WAAW,IAAI,CAAC,IAAA,sBAAW,EAAC,WAAW,CAAC,EAAE,CAAC;QAC/C,MAAM,KAAK,GAAG,IAAI,2BAAkB,CACnC,0DAA0D,CAC1D,CAAC;QACF,MAAM,KAAK,CAAC;IACb,CAAC;IAED,MAAM,SAAS,GAAG,WAAW,CAAC;IAC9B,IAAI,CAAC,SAAS,CAAC,UAAU,EAAE,MAAM,EAAE,CAAC;QACnC,MAAM,KAAK,GAAG,IAAI,2BAAkB,CAAC,gCAAgC,CAAC,CAAC;QACvE,MAAM,KAAK,CAAC;IACb,CAAC;IAGD,MAAM,WAAW,GAAG,MAAM,OAAO,CAAC,GAAG,CACpC,SAAS,CAAC,UAAU,CAAC,GAAG,CAAC,KAAK,EAAE,QAAQ,EAAE,EAAE;QAC3C,IAAI,CAAC;YACJ,MAAM,IAAI,GAAG,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC;YACxC,IAAI,CAAC,IAAI,EAAE,CAAC;gBACX,MAAM,IAAI,2BAAkB,CAAC,QAAQ,QAAQ,CAAC,IAAI,YAAY,EAAE;oBAC/D,QAAQ,EAAE,QAAQ,CAAC,IAAI;iBACvB,CAAC,CAAC;YACJ,CAAC;YACD,MAAM,MAAM,GAAY,MAAM,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,IAAI,IAAI,EAAE,EAAE;gBAC9D,QAAQ,EAAE;oBACT,EAAE,EAAE,QAAQ,CAAC,EAAE;oBACf,IAAI,EAAE,QAAQ,CAAC,IAAI;oBACnB,IAAI,EAAE,QAAQ,CAAC,IAAI,IAAI,EAAE;iBACzB;aACD,CAAC,CAAC;YAEH,OAAO,MAAM,CAAC;QACf,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YAGhB,MAAM,YAAY,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,wBAAwB,CAAC;YAGvF,IAAI,YAAoB,CAAC;YACzB,IACC,KAAK,YAAY,iCAAyB;gBAC1C,YAAY,CAAC,QAAQ,CAAC,iBAAiB,CAAC,EACvC,CAAC;gBACF,YAAY,GAAG,0BAA0B,QAAQ,CAAC,IAAI,KAAK,YAAY,EAAE,CAAC;YAC3E,CAAC;iBAAM,CAAC;gBACP,YAAY,GAAG,QAAQ,QAAQ,CAAC,IAAI,YAAY,YAAY,EAAE,CAAC;YAChE,CAAC;YAGD,OAAO,IAAI,sBAAW,CAAC;gBACtB,OAAO,EAAE,YAAY;gBACrB,YAAY,EAAE,QAAQ,CAAC,EAAE,IAAI,EAAE;gBAE/B,iBAAiB,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE;aAClC,CAAC,CAAC;QACJ,CAAC;IACF,CAAC,CAAC,CACF,CAAC;IAGF,MAAM,WAAW,GAAkB,EAAE,CAAC;IACtC,MAAM,YAAY,GAA+C,EAAE,CAAC;IAEpE,WAAW,CAAC,OAAO,CAAC,CAAC,MAAM,EAAE,EAAE;QAC9B,IAAI,IAAA,qBAAS,EAAC,MAAM,CAAC,EAAE,CAAC;YAEvB,MAAM,MAAM,GAAG,MAAM,CAAC,MAA6C,CAAC;YACpE,IAAI,MAAM,EAAE,CAAC;gBACZ,YAAY,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YAC3B,CAAC;QACF,CAAC;aAAM,CAAC;YAEP,WAAW,CAAC,IAAI,CAAC,MAAqB,CAAC,CAAC;QACzC,CAAC;IACF,CAAC,CAAC,CAAC;IAGH,YAAY,CAAC,OAAO,CAAC,CAAC,MAAM,EAAE,EAAE;QAC/B,IAAI,MAAM,CAAC,QAAQ,IAAI,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,EAAE,CAAC;YACvD,WAAW,CAAC,IAAI,CAAC,GAAG,MAAM,CAAC,QAAQ,CAAC,CAAC;QACtC,CAAC;IACF,CAAC,CAAC,CAAC;IAGH,MAAM,aAAa,GAAwB,EAAE,CAAC;IAE9C,KAAK,MAAM,MAAM,IAAI,YAAY,EAAE,CAAC;QACnC,IAAI,MAAM,CAAC,kBAAkB,IAAI,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,kBAAkB,CAAC,EAAE,CAAC;YAC3E,aAAa,CAAC,IAAI,CAAC,GAAG,MAAM,CAAC,kBAAkB,CAAC,CAAC;QAClD,CAAC;IACF,CAAC;IAGD,MAAM,WAAW,GAAwC;QACxD,QAAQ,EAAE,WAAW;KACrB,CAAC;IAEF,IAAI,aAAa,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QAC9B,WAAW,CAAC,kBAAkB,GAAG,aAAa,CAAC;IAChD,CAAC;IAED,OAAO,WAAW,CAAC;AACpB,CAAC"}

package/dist/workflow-builder-agent.d.ts

@@ -0,0 +1,41 @@
+import type { BaseChatModel } from '@langchain/core/language_models/chat_models';
+import type { LangChainTracer } from '@langchain/core/tracers/tracer_langchain';
+import { MemorySaver } from '@langchain/langgraph';
+import type { Logger } from '@n8n/backend-common';
+import type { INodeTypeDescription, IRunExecutionData, IWorkflowBase, NodeExecutionSchema } from 'n8n-workflow';
+export interface WorkflowBuilderAgentConfig {
+    parsedNodeTypes: INodeTypeDescription[];
+    llmSimpleTask: BaseChatModel;
+    llmComplexTask: BaseChatModel;
+    logger?: Logger;
+    checkpointer?: MemorySaver;
+    tracer?: LangChainTracer;
+}
+export interface ChatPayload {
+    message: string;
+    workflowContext?: {
+        executionSchema?: NodeExecutionSchema[];
+        currentWorkflow?: Partial<IWorkflowBase>;
+        executionData?: IRunExecutionData['resultData'];
+    };
+}
+export declare class WorkflowBuilderAgent {
+    private checkpointer;
+    private parsedNodeTypes;
+    private llmSimpleTask;
+    private llmComplexTask;
+    private logger?;
+    private tracer?;
+    constructor(config: WorkflowBuilderAgentConfig);
+    private createWorkflow;
+    getState(workflowId: string, userId?: string): Promise<import("@langchain/langgraph").StateSnapshot>;
+    static generateThreadId(workflowId?: string, userId?: string): string;
+    chat(payload: ChatPayload, userId?: string): AsyncGenerator<import("./types").StreamOutput, void, unknown>;
+    getSessions(workflowId: string | undefined, userId?: string): Promise<{
+        sessions: {
+            sessionId: string;
+            messages: Record<string, unknown>[];
+            lastUpdated: string;
+        }[];
+    }>;
+}

package/dist/workflow-builder-agent.js

@@ -0,0 +1,205 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.WorkflowBuilderAgent = void 0;
+const messages_1 = require("@langchain/core/messages");
+const langgraph_1 = require("@langchain/langgraph");
+const conversation_compact_1 = require("./chains/conversation-compact");
+const errors_1 = require("./errors");
+const add_node_tool_1 = require("./tools/add-node.tool");
+const connect_nodes_tool_1 = require("./tools/connect-nodes.tool");
+const node_details_tool_1 = require("./tools/node-details.tool");
+const node_search_tool_1 = require("./tools/node-search.tool");
+const main_agent_prompt_1 = require("./tools/prompts/main-agent.prompt");
+const remove_node_tool_1 = require("./tools/remove-node.tool");
+const update_node_parameters_tool_1 = require("./tools/update-node-parameters.tool");
+const operations_processor_1 = require("./utils/operations-processor");
+const stream_processor_1 = require("./utils/stream-processor");
+const tool_executor_1 = require("./utils/tool-executor");
+const workflow_state_1 = require("./workflow-state");
+class WorkflowBuilderAgent {
+    constructor(config) {
+        this.parsedNodeTypes = config.parsedNodeTypes;
+        this.llmSimpleTask = config.llmSimpleTask;
+        this.llmComplexTask = config.llmComplexTask;
+        this.logger = config.logger;
+        this.checkpointer = config.checkpointer ?? new langgraph_1.MemorySaver();
+        this.tracer = config.tracer;
+    }
+    createWorkflow() {
+        const tools = [
+            (0, node_search_tool_1.createNodeSearchTool)(this.parsedNodeTypes),
+            (0, node_details_tool_1.createNodeDetailsTool)(this.parsedNodeTypes),
+            (0, add_node_tool_1.createAddNodeTool)(this.parsedNodeTypes),
+            (0, connect_nodes_tool_1.createConnectNodesTool)(this.parsedNodeTypes, this.logger),
+            (0, remove_node_tool_1.createRemoveNodeTool)(this.logger),
+            (0, update_node_parameters_tool_1.createUpdateNodeParametersTool)(this.parsedNodeTypes, this.llmComplexTask, this.logger),
+        ];
+        const toolMap = new Map(tools.map((tool) => [tool.name, tool]));
+        const callModel = async (state) => {
+            if (!this.llmSimpleTask) {
+                throw new errors_1.LLMServiceError('LLM not setup');
+            }
+            if (typeof this.llmSimpleTask.bindTools !== 'function') {
+                throw new errors_1.LLMServiceError('LLM does not support tools', {
+                    llmModel: this.llmSimpleTask._llmType(),
+                });
+            }
+            const prompt = await main_agent_prompt_1.mainAgentPrompt.invoke({
+                ...state,
+                executionData: state.workflowContext?.executionData ?? {},
+                executionSchema: state.workflowContext?.executionSchema ?? [],
+            });
+            const response = await this.llmSimpleTask.bindTools(tools).invoke(prompt);
+            return { messages: [response] };
+        };
+        const shouldModifyState = ({ messages }) => {
+            const lastMessage = messages[messages.length - 1];
+            if (lastMessage.content === '/compact') {
+                return 'compact_messages';
+            }
+            if (lastMessage.content === '/clear') {
+                return 'delete_messages';
+            }
+            return 'agent';
+        };
+        const shouldContinue = ({ messages }) => {
+            const lastMessage = messages[messages.length - 1];
+            if (lastMessage.tool_calls?.length) {
+                return 'tools';
+            }
+            return langgraph_1.END;
+        };
+        const customToolExecutor = async (state) => {
+            return await (0, tool_executor_1.executeToolsInParallel)({ state, toolMap });
+        };
+        function deleteMessages(state) {
+            const messages = state.messages;
+            const stateUpdate = {
+                workflowOperations: null,
+                workflowContext: {},
+                messages: messages.map((m) => new messages_1.RemoveMessage({ id: m.id })) ?? [],
+                workflowJSON: {
+                    nodes: [],
+                    connections: {},
+                },
+            };
+            return stateUpdate;
+        }
+        const compactSession = async (state) => {
+            if (!this.llmSimpleTask) {
+                throw new errors_1.LLMServiceError('LLM not setup');
+            }
+            const messages = state.messages;
+            const compactedMessages = await (0, conversation_compact_1.conversationCompactChain)(this.llmSimpleTask, messages);
+            return {
+                messages: [
+                    ...messages.map((m) => new messages_1.RemoveMessage({ id: m.id })),
+                    ...compactedMessages.newMessages,
+                ],
+            };
+        };
+        const workflow = new langgraph_1.StateGraph(workflow_state_1.WorkflowState)
+            .addNode('agent', callModel)
+            .addNode('tools', customToolExecutor)
+            .addNode('process_operations', operations_processor_1.processOperations)
+            .addNode('delete_messages', deleteMessages)
+            .addNode('compact_messages', compactSession)
+            .addConditionalEdges('__start__', shouldModifyState)
+            .addEdge('tools', 'process_operations')
+            .addEdge('process_operations', 'agent')
+            .addEdge('delete_messages', langgraph_1.END)
+            .addEdge('compact_messages', langgraph_1.END)
+            .addConditionalEdges('agent', shouldContinue);
+        return workflow;
+    }
+    async getState(workflowId, userId) {
+        const workflow = this.createWorkflow();
+        const agent = workflow.compile({ checkpointer: this.checkpointer });
+        return await agent.getState({
+            configurable: { thread_id: `workflow-${workflowId}-user-${userId ?? new Date().getTime()}` },
+        });
+    }
+    static generateThreadId(workflowId, userId) {
+        return workflowId
+            ? `workflow-${workflowId}-user-${userId ?? new Date().getTime()}`
+            : crypto.randomUUID();
+    }
+    async *chat(payload, userId) {
+        const agent = this.createWorkflow().compile({ checkpointer: this.checkpointer });
+        const workflowId = payload.workflowContext?.currentWorkflow?.id;
+        const threadId = WorkflowBuilderAgent.generateThreadId(workflowId, userId);
+        const threadConfig = {
+            configurable: {
+                thread_id: threadId,
+            },
+        };
+        const existingCheckpoint = await this.checkpointer.getTuple(threadConfig);
+        let stream;
+        if (!existingCheckpoint?.checkpoint) {
+            const initialState = {
+                messages: [new messages_1.HumanMessage({ content: payload.message })],
+                workflowJSON: payload.workflowContext?.currentWorkflow ?? {
+                    nodes: [],
+                    connections: {},
+                },
+                workflowOperations: [],
+                workflowContext: payload.workflowContext,
+            };
+            stream = await agent.stream(initialState, {
+                ...threadConfig,
+                streamMode: ['updates', 'custom'],
+                recursionLimit: 30,
+                callbacks: this.tracer ? [this.tracer] : undefined,
+            });
+        }
+        else {
+            const stateUpdate = {
+                messages: [new messages_1.HumanMessage({ content: payload.message })],
+                workflowOperations: [],
+                workflowContext: payload.workflowContext,
+                workflowJSON: { nodes: [], connections: {} },
+            };
+            if (payload.workflowContext?.currentWorkflow) {
+                stateUpdate.workflowJSON = payload.workflowContext?.currentWorkflow;
+            }
+            stream = await agent.stream(stateUpdate, {
+                ...threadConfig,
+                streamMode: ['updates', 'custom'],
+                recursionLimit: 80,
+                callbacks: this.tracer ? [this.tracer] : undefined,
+            });
+        }
+        const streamProcessor = (0, stream_processor_1.createStreamProcessor)(stream);
+        for await (const output of streamProcessor) {
+            yield output;
+        }
+    }
+    async getSessions(workflowId, userId) {
+        const sessions = [];
+        if (workflowId) {
+            const threadId = WorkflowBuilderAgent.generateThreadId(workflowId, userId);
+            const threadConfig = {
+                configurable: {
+                    thread_id: threadId,
+                },
+            };
+            try {
+                const checkpoint = await this.checkpointer.getTuple(threadConfig);
+                if (checkpoint?.checkpoint) {
+                    const messages = checkpoint.checkpoint.channel_values?.messages ?? [];
+                    sessions.push({
+                        sessionId: threadId,
+                        messages: (0, stream_processor_1.formatMessages)(messages),
+                        lastUpdated: checkpoint.checkpoint.ts,
+                    });
+                }
+            }
+            catch (error) {
+                this.logger?.debug('No session found for workflow:', { workflowId, error });
+            }
+        }
+        return { sessions };
+    }
+}
+exports.WorkflowBuilderAgent = WorkflowBuilderAgent;
+//# sourceMappingURL=workflow-builder-agent.js.map

package/dist/workflow-builder-agent.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"workflow-builder-agent.js","sourceRoot":"","sources":["../src/workflow-builder-agent.ts"],"names":[],"mappings":";;;AAEA,uDAAuE;AAEvE,oDAAoE;AASpE,wEAAyE;AACzE,qCAA2C;AAC3C,yDAA0D;AAC1D,mEAAoE;AACpE,iEAAkE;AAClE,+DAAgE;AAChE,yEAAoE;AACpE,+DAAgE;AAChE,qFAAqF;AAErF,uEAAiE;AACjE,+DAAiF;AACjF,yDAA+D;AAC/D,qDAAiD;AAoBjD,MAAa,oBAAoB;IAQhC,YAAY,MAAkC;QAC7C,IAAI,CAAC,eAAe,GAAG,MAAM,CAAC,eAAe,CAAC;QAC9C,IAAI,CAAC,aAAa,GAAG,MAAM,CAAC,aAAa,CAAC;QAC1C,IAAI,CAAC,cAAc,GAAG,MAAM,CAAC,cAAc,CAAC;QAC5C,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC,MAAM,CAAC;QAC5B,IAAI,CAAC,YAAY,GAAG,MAAM,CAAC,YAAY,IAAI,IAAI,uBAAW,EAAE,CAAC;QAC7D,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC,MAAM,CAAC;IAC7B,CAAC;IAEO,cAAc;QACrB,MAAM,KAAK,GAAG;YACb,IAAA,uCAAoB,EAAC,IAAI,CAAC,eAAe,CAAC;YAC1C,IAAA,yCAAqB,EAAC,IAAI,CAAC,eAAe,CAAC;YAC3C,IAAA,iCAAiB,EAAC,IAAI,CAAC,eAAe,CAAC;YACvC,IAAA,2CAAsB,EAAC,IAAI,CAAC,eAAe,EAAE,IAAI,CAAC,MAAM,CAAC;YACzD,IAAA,uCAAoB,EAAC,IAAI,CAAC,MAAM,CAAC;YACjC,IAAA,4DAA8B,EAAC,IAAI,CAAC,eAAe,EAAE,IAAI,CAAC,cAAc,EAAE,IAAI,CAAC,MAAM,CAAC;SACtF,CAAC;QAGF,MAAM,OAAO,GAAG,IAAI,GAAG,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC,CAAC,CAAC;QAEhE,MAAM,SAAS,GAAG,KAAK,EAAE,KAAiC,EAAE,EAAE;YAC7D,IAAI,CAAC,IAAI,CAAC,aAAa,EAAE,CAAC;gBACzB,MAAM,IAAI,wBAAe,CAAC,eAAe,CAAC,CAAC;YAC5C,CAAC;YACD,IAAI,OAAO,IAAI,CAAC,aAAa,CAAC,SAAS,KAAK,UAAU,EAAE,CAAC;gBACxD,MAAM,IAAI,wBAAe,CAAC,4BAA4B,EAAE;oBACvD,QAAQ,EAAE,IAAI,CAAC,aAAa,CAAC,QAAQ,EAAE;iBACvC,CAAC,CAAC;YACJ,CAAC;YAED,MAAM,MAAM,GAAG,MAAM,mCAAe,CAAC,MAAM,CAAC;gBAC3C,GAAG,KAAK;gBACR,aAAa,EAAE,KAAK,CAAC,eAAe,EAAE,aAAa,IAAI,EAAE;gBACzD,eAAe,EAAE,KAAK,CAAC,eAAe,EAAE,eAAe,IAAI,EAAE;aAC7D,CAAC,CAAC;YACH,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,aAAa,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;YAE1E,OAAO,EAAE,QAAQ,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC;QACjC,CAAC,CAAC;QAEF,MAAM,iBAAiB,GAAG,CAAC,EAAE,QAAQ,EAA8B,EAAE,EAAE;YACtE,MAAM,WAAW,GAAG,QAAQ,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC,CAAiB,CAAC;YAElE,IAAI,WAAW,CAAC,OAAO,KAAK,UAAU,EAAE,CAAC;gBACxC,OAAO,kBAAkB,CAAC;YAC3B,CAAC;YAED,IAAI,WAAW,CAAC,OAAO,KAAK,QAAQ,EAAE,CAAC;gBACtC,OAAO,iBAAiB,CAAC;YAC1B,CAAC;YAED,OAAO,OAAO,CAAC;QAChB,CAAC,CAAC;QAEF,MAAM,cAAc,GAAG,CAAC,EAAE,QAAQ,EAA8B,EAAE,EAAE;YACnE,MAAM,WAAW,GAAG,QAAQ,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC,CAAc,CAAC;YAE/D,IAAI,WAAW,CAAC,UAAU,EAAE,MAAM,EAAE,CAAC;gBACpC,OAAO,OAAO,CAAC;YAChB,CAAC;YACD,OAAO,eAAG,CAAC;QACZ,CAAC,CAAC;QAEF,MAAM,kBAAkB,GAAG,KAAK,EAAE,KAAiC,EAAE,EAAE;YACtE,OAAO,MAAM,IAAA,sCAAsB,EAAC,EAAE,KAAK,EAAE,OAAO,EAAE,CAAC,CAAC;QACzD,CAAC,CAAC;QAEF,SAAS,cAAc,CAAC,KAAiC;YACxD,MAAM,QAAQ,GAAG,KAAK,CAAC,QAAQ,CAAC;YAChC,MAAM,WAAW,GAAwC;gBACxD,kBAAkB,EAAE,IAAI;gBACxB,eAAe,EAAE,EAAE;gBACnB,QAAQ,EAAE,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,wBAAa,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC,EAAG,EAAE,CAAC,CAAC,IAAI,EAAE;gBACrE,YAAY,EAAE;oBACb,KAAK,EAAE,EAAE;oBACT,WAAW,EAAE,EAAE;iBACf;aACD,CAAC;YAEF,OAAO,WAAW,CAAC;QACpB,CAAC;QAED,MAAM,cAAc,GAAG,KAAK,EAAE,KAAiC,EAAE,EAAE;YAClE,IAAI,CAAC,IAAI,CAAC,aAAa,EAAE,CAAC;gBACzB,MAAM,IAAI,wBAAe,CAAC,eAAe,CAAC,CAAC;YAC5C,CAAC;YACD,MAAM,QAAQ,GAAG,KAAK,CAAC,QAAQ,CAAC;YAChC,MAAM,iBAAiB,GAAG,MAAM,IAAA,+CAAwB,EAAC,IAAI,CAAC,aAAa,EAAE,QAAQ,CAAC,CAAC;YAEvF,OAAO;gBACN,QAAQ,EAAE;oBACT,GAAG,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,wBAAa,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC,EAAG,EAAE,CAAC,CAAC;oBACxD,GAAG,iBAAiB,CAAC,WAAW;iBAChC;aACD,CAAC;QACH,CAAC,CAAC;QAEF,MAAM,QAAQ,GAAG,IAAI,sBAAU,CAAC,8BAAa,CAAC;aAC5C,OAAO,CAAC,OAAO,EAAE,SAAS,CAAC;aAC3B,OAAO,CAAC,OAAO,EAAE,kBAAkB,CAAC;aACpC,OAAO,CAAC,oBAAoB,EAAE,wCAAiB,CAAC;aAChD,OAAO,CAAC,iBAAiB,EAAE,cAAc,CAAC;aAC1C,OAAO,CAAC,kBAAkB,EAAE,cAAc,CAAC;aAC3C,mBAAmB,CAAC,WAAW,EAAE,iBAAiB,CAAC;aACnD,OAAO,CAAC,OAAO,EAAE,oBAAoB,CAAC;aACtC,OAAO,CAAC,oBAAoB,EAAE,OAAO,CAAC;aACtC,OAAO,CAAC,iBAAiB,EAAE,eAAG,CAAC;aAC/B,OAAO,CAAC,kBAAkB,EAAE,eAAG,CAAC;aAChC,mBAAmB,CAAC,OAAO,EAAE,cAAc,CAAC,CAAC;QAE/C,OAAO,QAAQ,CAAC;IACjB,CAAC;IACD,KAAK,CAAC,QAAQ,CAAC,UAAkB,EAAE,MAAe;QACjD,MAAM,QAAQ,GAAG,IAAI,CAAC,cAAc,EAAE,CAAC;QACvC,MAAM,KAAK,GAAG,QAAQ,CAAC,OAAO,CAAC,EAAE,YAAY,EAAE,IAAI,CAAC,YAAY,EAAE,CAAC,CAAC;QACpE,OAAO,MAAM,KAAK,CAAC,QAAQ,CAAC;YAC3B,YAAY,EAAE,EAAE,SAAS,EAAE,YAAY,UAAU,SAAS,MAAM,IAAI,IAAI,IAAI,EAAE,CAAC,OAAO,EAAE,EAAE,EAAE;SAC5F,CAAC,CAAC;IACJ,CAAC;IAED,MAAM,CAAC,gBAAgB,CAAC,UAAmB,EAAE,MAAe;QAC3D,OAAO,UAAU;YAChB,CAAC,CAAC,YAAY,UAAU,SAAS,MAAM,IAAI,IAAI,IAAI,EAAE,CAAC,OAAO,EAAE,EAAE;YACjE,CAAC,CAAC,MAAM,CAAC,UAAU,EAAE,CAAC;IACxB,CAAC;IAED,KAAK,CAAC,CAAC,IAAI,CAAC,OAAoB,EAAE,MAAe;QAChD,MAAM,KAAK,GAAG,IAAI,CAAC,cAAc,EAAE,CAAC,OAAO,CAAC,EAAE,YAAY,EAAE,IAAI,CAAC,YAAY,EAAE,CAAC,CAAC;QACjF,MAAM,UAAU,GAAG,OAAO,CAAC,eAAe,EAAE,eAAe,EAAE,EAAE,CAAC;QAGhE,MAAM,QAAQ,GAAG,oBAAoB,CAAC,gBAAgB,CAAC,UAAU,EAAE,MAAM,CAAC,CAAC;QAG3E,MAAM,YAAY,GAAG;YACpB,YAAY,EAAE;gBACb,SAAS,EAAE,QAAQ;aACnB;SACD,CAAC;QAIF,MAAM,kBAAkB,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC;QAE1E,IAAI,MAAM,CAAC;QAEX,IAAI,CAAC,kBAAkB,EAAE,UAAU,EAAE,CAAC;YAErC,MAAM,YAAY,GAA+B;gBAChD,QAAQ,EAAE,CAAC,IAAI,uBAAY,CAAC,EAAE,OAAO,EAAE,OAAO,CAAC,OAAO,EAAE,CAAC,CAAC;gBAC1D,YAAY,EAAG,OAAO,CAAC,eAAe,EAAE,eAAkC,IAAI;oBAC7E,KAAK,EAAE,EAAE;oBACT,WAAW,EAAE,EAAE;iBACf;gBACD,kBAAkB,EAAE,EAAE;gBACtB,eAAe,EAAE,OAAO,CAAC,eAAe;aACxC,CAAC;YAEF,MAAM,GAAG,MAAM,KAAK,CAAC,MAAM,CAAC,YAAY,EAAE;gBACzC,GAAG,YAAY;gBACf,UAAU,EAAE,CAAC,SAAS,EAAE,QAAQ,CAAC;gBACjC,cAAc,EAAE,EAAE;gBAClB,SAAS,EAAE,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,SAAS;aAClD,CAAC,CAAC;QACJ,CAAC;aAAM,CAAC;YAEP,MAAM,WAAW,GAAwC;gBACxD,QAAQ,EAAE,CAAC,IAAI,uBAAY,CAAC,EAAE,OAAO,EAAE,OAAO,CAAC,OAAO,EAAE,CAAC,CAAC;gBAC1D,kBAAkB,EAAE,EAAE;gBACtB,eAAe,EAAE,OAAO,CAAC,eAAe;gBACxC,YAAY,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE,WAAW,EAAE,EAAE,EAAE;aAC5C,CAAC;YAEF,IAAI,OAAO,CAAC,eAAe,EAAE,eAAe,EAAE,CAAC;gBAC9C,WAAW,CAAC,YAAY,GAAG,OAAO,CAAC,eAAe,EAAE,eAAiC,CAAC;YACvF,CAAC;YAGD,MAAM,GAAG,MAAM,KAAK,CAAC,MAAM,CAAC,WAAW,EAAE;gBACxC,GAAG,YAAY;gBACf,UAAU,EAAE,CAAC,SAAS,EAAE,QAAQ,CAAC;gBACjC,cAAc,EAAE,EAAE;gBAClB,SAAS,EAAE,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,SAAS;aAClD,CAAC,CAAC;QACJ,CAAC;QAGD,MAAM,eAAe,GAAG,IAAA,wCAAqB,EAAC,MAAM,CAAC,CAAC;QAEtD,IAAI,KAAK,EAAE,MAAM,MAAM,IAAI,eAAe,EAAE,CAAC;YAC5C,MAAM,MAAM,CAAC;QACd,CAAC;IACF,CAAC;IAED,KAAK,CAAC,WAAW,CAAC,UAA8B,EAAE,MAAe;QAIhE,MAAM,QAAQ,GAAG,EAAE,CAAC;QAEpB,IAAI,UAAU,EAAE,CAAC;YAChB,MAAM,QAAQ,GAAG,oBAAoB,CAAC,gBAAgB,CAAC,UAAU,EAAE,MAAM,CAAC,CAAC;YAC3E,MAAM,YAAY,GAAG;gBACpB,YAAY,EAAE;oBACb,SAAS,EAAE,QAAQ;iBACnB;aACD,CAAC;YAEF,IAAI,CAAC;gBAEJ,MAAM,UAAU,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC;gBAElE,IAAI,UAAU,EAAE,UAAU,EAAE,CAAC;oBAC5B,MAAM,QAAQ,GACZ,UAAU,CAAC,UAAU,CAAC,cAAc,EAAE,QAErC,IAAI,EAAE,CAAC;oBAEV,QAAQ,CAAC,IAAI,CAAC;wBACb,SAAS,EAAE,QAAQ;wBACnB,QAAQ,EAAE,IAAA,iCAAc,EAAC,QAAQ,CAAC;wBAClC,WAAW,EAAE,UAAU,CAAC,UAAU,CAAC,EAAE;qBACrC,CAAC,CAAC;gBACJ,CAAC;YACF,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBAEhB,IAAI,CAAC,MAAM,EAAE,KAAK,CAAC,gCAAgC,EAAE,EAAE,UAAU,EAAE,KAAK,EAAE,CAAC,CAAC;YAC7E,CAAC;QACF,CAAC;QAED,OAAO,EAAE,QAAQ,EAAE,CAAC;IACrB,CAAC;CACD;AAlPD,oDAkPC"}