@n8n/ai-workflow-builder 1.8.1 → 1.9.0

package/dist/code-builder/handlers/chat-setup-handler.d.ts

@@ -0,0 +1,57 @@
+import type { BaseChatModel, BaseChatModelCallOptions } from '@langchain/core/language_models/chat_models';
+import type { AIMessage, BaseMessage } from '@langchain/core/messages';
+import type { Runnable } from '@langchain/core/runnables';
+import type { StructuredToolInterface } from '@langchain/core/tools';
+import type { Logger } from '@n8n/backend-common';
+import type { WorkflowJSON } from '@n8n/workflow-sdk';
+import { type HistoryContext } from '../prompts';
+import { TextEditorHandler } from './text-editor-handler';
+import { TextEditorToolHandler } from './text-editor-tool-handler';
+import type { PlanOutput } from '../../types/planning';
+import type { ChatPayload } from '../../workflow-builder-agent';
+import type { ParseAndValidateResult } from '../types';
+import type { NodeTypeParser } from '../utils/node-type-parser';
+type ParseAndValidateFn = (code: string, currentWorkflow?: WorkflowJSON) => Promise<ParseAndValidateResult>;
+type GetErrorContextFn = (code: string, errorMessage: string) => string;
+export interface ChatSetupHandlerConfig {
+    llm: BaseChatModel;
+    tools: StructuredToolInterface[];
+    enableTextEditorConfig?: boolean;
+    parseAndValidate: ParseAndValidateFn;
+    getErrorContext: GetErrorContextFn;
+    nodeTypeParser?: NodeTypeParser;
+    logger?: Logger;
+}
+export interface ChatSetupParams {
+    payload: ChatPayload;
+    historyContext?: HistoryContext;
+}
+export type LlmWithTools = Runnable<BaseMessage[], AIMessage, BaseChatModelCallOptions>;
+export interface ChatSetupResult {
+    llmWithTools: LlmWithTools;
+    messages: BaseMessage[];
+    textEditorEnabled: boolean;
+    preGeneratedWorkflowCode?: string;
+    currentWorkflow?: WorkflowJSON;
+    textEditorHandler?: TextEditorHandler;
+    textEditorToolHandler?: TextEditorToolHandler;
+}
+export declare class ChatSetupHandler {
+    private llm;
+    private tools;
+    private enableTextEditorConfig?;
+    private parseAndValidate;
+    private getErrorContext;
+    private nodeTypeParser?;
+    private logger?;
+    constructor(config: ChatSetupHandlerConfig);
+    execute(params: ChatSetupParams): Promise<ChatSetupResult>;
+    private shouldEnableTextEditor;
+    private preGenerateWorkflowCode;
+    private bindToolsToLlm;
+    private formatInitialMessages;
+    private initializeTextEditorHandlers;
+    private prefetchSearchResults;
+}
+export declare function extractNodeNamesFromPlan(plan: PlanOutput): string[];
+export {};

package/dist/code-builder/handlers/chat-setup-handler.js

@@ -0,0 +1,147 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ChatSetupHandler = void 0;
+exports.extractNodeNamesFromPlan = extractNodeNamesFromPlan;
+const workflow_sdk_1 = require("@n8n/workflow-sdk");
+const constants_1 = require("../constants");
+const prompts_1 = require("../prompts");
+const text_editor_handler_1 = require("./text-editor-handler");
+const text_editor_tool_handler_1 = require("./text-editor-tool-handler");
+const code_builder_search_tool_1 = require("../tools/code-builder-search.tool");
+const extract_code_1 = require("../utils/extract-code");
+class ChatSetupHandler {
+    llm;
+    tools;
+    enableTextEditorConfig;
+    parseAndValidate;
+    getErrorContext;
+    nodeTypeParser;
+    logger;
+    constructor(config) {
+        this.llm = config.llm;
+        this.tools = config.tools;
+        this.enableTextEditorConfig = config.enableTextEditorConfig;
+        this.parseAndValidate = config.parseAndValidate;
+        this.getErrorContext = config.getErrorContext;
+        this.nodeTypeParser = config.nodeTypeParser;
+        this.logger = config.logger;
+    }
+    async execute(params) {
+        const { payload, historyContext } = params;
+        const currentWorkflow = payload.workflowContext?.currentWorkflow;
+        const workflowForCodeContext = (currentWorkflow?.nodes?.length ?? 0) > 0 ? currentWorkflow : undefined;
+        const preGeneratedWorkflowCode = this.preGenerateWorkflowCode(payload, workflowForCodeContext);
+        const preSearchResults = payload.planOutput
+            ? this.prefetchSearchResults(payload.planOutput)
+            : undefined;
+        const textEditorEnabled = this.shouldEnableTextEditor();
+        const prompt = (0, prompts_1.buildCodeBuilderPrompt)(workflowForCodeContext, historyContext, {
+            enableTextEditor: textEditorEnabled,
+            executionSchema: payload.workflowContext?.executionSchema,
+            executionData: payload.workflowContext?.executionData,
+            expressionValues: payload.workflowContext?.expressionValues,
+            preGeneratedCode: preGeneratedWorkflowCode,
+            valuesExcluded: payload.workflowContext?.valuesExcluded,
+            pinnedNodes: payload.workflowContext?.pinnedNodes,
+            planOutput: payload.planOutput,
+            preSearchResults,
+        });
+        const llmWithTools = this.bindToolsToLlm(textEditorEnabled, !!payload.planOutput);
+        const messages = await this.formatInitialMessages(prompt, payload.message);
+        const { textEditorHandler, textEditorToolHandler } = this.initializeTextEditorHandlers(textEditorEnabled, preGeneratedWorkflowCode);
+        return {
+            llmWithTools,
+            messages,
+            textEditorEnabled,
+            preGeneratedWorkflowCode,
+            currentWorkflow,
+            textEditorHandler,
+            textEditorToolHandler,
+        };
+    }
+    shouldEnableTextEditor() {
+        if (this.enableTextEditorConfig !== undefined) {
+            return this.enableTextEditorConfig;
+        }
+        const modelName = this.llm.modelId ?? '';
+        return (modelName.includes('claude-4') ||
+            modelName.includes('opus-4') ||
+            modelName.includes('sonnet-4'));
+    }
+    preGenerateWorkflowCode(payload, currentWorkflow) {
+        if (!currentWorkflow) {
+            return undefined;
+        }
+        return (0, workflow_sdk_1.generateWorkflowCode)({
+            workflow: currentWorkflow,
+            executionSchema: payload.workflowContext?.executionSchema,
+            executionData: payload.workflowContext?.executionData,
+            expressionValues: payload.workflowContext?.expressionValues,
+            valuesExcluded: payload.workflowContext?.valuesExcluded,
+            pinnedNodes: payload.workflowContext?.pinnedNodes,
+        });
+    }
+    bindToolsToLlm(textEditorEnabled, hasPlanOutput) {
+        if (!this.llm.bindTools) {
+            throw new Error('LLM does not support bindTools - cannot use tools for node discovery');
+        }
+        let tools = hasPlanOutput ? this.tools.filter((t) => t.name !== 'get_suggested_nodes') : this.tools;
+        if (textEditorEnabled) {
+            tools = [...tools, constants_1.TEXT_EDITOR_TOOL, constants_1.VALIDATE_TOOL, constants_1.BATCH_STR_REPLACE_TOOL];
+        }
+        return this.llm.bindTools(tools);
+    }
+    async formatInitialMessages(prompt, userMessage) {
+        const formattedMessages = await prompt.formatMessages({ userMessage });
+        return [...formattedMessages];
+    }
+    initializeTextEditorHandlers(textEditorEnabled, preGeneratedWorkflowCode) {
+        if (!textEditorEnabled) {
+            return {};
+        }
+        const textEditorHandler = new text_editor_handler_1.TextEditorHandler();
+        const textEditorToolHandler = new text_editor_tool_handler_1.TextEditorToolHandler({
+            textEditorExecute: (args) => textEditorHandler.execute(args),
+            textEditorGetCode: () => textEditorHandler.getWorkflowCode(),
+            parseAndValidate: this.parseAndValidate,
+            getErrorContext: this.getErrorContext,
+        });
+        if (preGeneratedWorkflowCode) {
+            const codeWithImport = `${extract_code_1.SDK_IMPORT_STATEMENT}\n\n${preGeneratedWorkflowCode}`;
+            textEditorHandler.setWorkflowCode(codeWithImport);
+        }
+        return { textEditorHandler, textEditorToolHandler };
+    }
+    prefetchSearchResults(plan) {
+        const nodeNames = extractNodeNamesFromPlan(plan);
+        if (nodeNames.length === 0)
+            return undefined;
+        if (!this.nodeTypeParser) {
+            this.logger?.warn('nodeTypeParser not available, skipping pre-fetch of plan suggestedNodes');
+            return undefined;
+        }
+        const formattedResults = [];
+        for (const nodeName of nodeNames) {
+            const result = (0, code_builder_search_tool_1.formatNodeResult)(this.nodeTypeParser, nodeName);
+            if (result) {
+                formattedResults.push(result);
+            }
+        }
+        if (formattedResults.length === 0)
+            return undefined;
+        return `Found ${formattedResults.length} nodes:\n\n${formattedResults.join('\n\n')}`;
+    }
+}
+exports.ChatSetupHandler = ChatSetupHandler;
+function extractNodeNamesFromPlan(plan) {
+    const nodeSet = new Set();
+    for (const step of plan.steps) {
+        if (step.suggestedNodes) {
+            for (const nodeName of step.suggestedNodes) {
+                nodeSet.add(nodeName);
+            }
+        }
+    }
+    return [...nodeSet];
+}
+//# sourceMappingURL=chat-setup-handler.js.map

package/dist/code-builder/handlers/chat-setup-handler.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"chat-setup-handler.js","sourceRoot":"","sources":["../../../src/code-builder/handlers/chat-setup-handler.ts"],"names":[],"mappings":";;;AAwTA,4DAUC;AAnTD,oDAAyD;AAGzD,4CAAuF;AACvF,wCAAyE;AACzE,+DAA0D;AAC1D,yEAAmE;AAInE,gFAAqE;AAErE,wDAA6D;AAkE7D,MAAa,gBAAgB;IACpB,GAAG,CAAgB;IACnB,KAAK,CAA4B;IACjC,sBAAsB,CAAW;IACjC,gBAAgB,CAAqB;IACrC,eAAe,CAAoB;IACnC,cAAc,CAAkB;IAChC,MAAM,CAAU;IAExB,YAAY,MAA8B;QACzC,IAAI,CAAC,GAAG,GAAG,MAAM,CAAC,GAAG,CAAC;QACtB,IAAI,CAAC,KAAK,GAAG,MAAM,CAAC,KAAK,CAAC;QAC1B,IAAI,CAAC,sBAAsB,GAAG,MAAM,CAAC,sBAAsB,CAAC;QAC5D,IAAI,CAAC,gBAAgB,GAAG,MAAM,CAAC,gBAAgB,CAAC;QAChD,IAAI,CAAC,eAAe,GAAG,MAAM,CAAC,eAAe,CAAC;QAC9C,IAAI,CAAC,cAAc,GAAG,MAAM,CAAC,cAAc,CAAC;QAC5C,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC,MAAM,CAAC;IAC7B,CAAC;IAQD,KAAK,CAAC,OAAO,CAAC,MAAuB;QACpC,MAAM,EAAE,OAAO,EAAE,cAAc,EAAE,GAAG,MAAM,CAAC;QAC3C,MAAM,eAAe,GAAG,OAAO,CAAC,eAAe,EAAE,eAA2C,CAAC;QAI7F,MAAM,sBAAsB,GAC3B,CAAC,eAAe,EAAE,KAAK,EAAE,MAAM,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,SAAS,CAAC;QAGzE,MAAM,wBAAwB,GAAG,IAAI,CAAC,uBAAuB,CAAC,OAAO,EAAE,sBAAsB,CAAC,CAAC;QAG/F,MAAM,gBAAgB,GAAG,OAAO,CAAC,UAAU;YAC1C,CAAC,CAAC,IAAI,CAAC,qBAAqB,CAAC,OAAO,CAAC,UAAU,CAAC;YAChD,CAAC,CAAC,SAAS,CAAC;QAGb,MAAM,iBAAiB,GAAG,IAAI,CAAC,sBAAsB,EAAE,CAAC;QAGxD,MAAM,MAAM,GAAG,IAAA,gCAAsB,EAAC,sBAAsB,EAAE,cAAc,EAAE;YAC7E,gBAAgB,EAAE,iBAAiB;YACnC,eAAe,EAAE,OAAO,CAAC,eAAe,EAAE,eAAe;YACzD,aAAa,EAAE,OAAO,CAAC,eAAe,EAAE,aAAa;YACrD,gBAAgB,EAAE,OAAO,CAAC,eAAe,EAAE,gBAAgB;YAC3D,gBAAgB,EAAE,wBAAwB;YAC1C,cAAc,EAAE,OAAO,CAAC,eAAe,EAAE,cAAc;YACvD,WAAW,EAAE,OAAO,CAAC,eAAe,EAAE,WAAW;YACjD,UAAU,EAAE,OAAO,CAAC,UAAU;YAC9B,gBAAgB;SAChB,CAAC,CAAC;QAGH,MAAM,YAAY,GAAG,IAAI,CAAC,cAAc,CAAC,iBAAiB,EAAE,CAAC,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;QAGlF,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,qBAAqB,CAAC,MAAM,EAAE,OAAO,CAAC,OAAO,CAAC,CAAC;QAG3E,MAAM,EAAE,iBAAiB,EAAE,qBAAqB,EAAE,GAAG,IAAI,CAAC,4BAA4B,CACrF,iBAAiB,EACjB,wBAAwB,CACxB,CAAC;QAEF,OAAO;YACN,YAAY;YACZ,QAAQ;YACR,iBAAiB;YACjB,wBAAwB;YACxB,eAAe;YACf,iBAAiB;YACjB,qBAAqB;SACrB,CAAC;IACH,CAAC;IAMO,sBAAsB;QAC7B,IAAI,IAAI,CAAC,sBAAsB,KAAK,SAAS,EAAE,CAAC;YAC/C,OAAO,IAAI,CAAC,sBAAsB,CAAC;QACpC,CAAC;QAED,MAAM,SAAS,GAAI,IAAI,CAAC,GAA4B,CAAC,OAAO,IAAI,EAAE,CAAC;QACnE,OAAO,CACN,SAAS,CAAC,QAAQ,CAAC,UAAU,CAAC;YAC9B,SAAS,CAAC,QAAQ,CAAC,QAAQ,CAAC;YAC5B,SAAS,CAAC,QAAQ,CAAC,UAAU,CAAC,CAC9B,CAAC;IACH,CAAC;IAKO,uBAAuB,CAC9B,OAAoB,EACpB,eAA8B;QAE9B,IAAI,CAAC,eAAe,EAAE,CAAC;YACtB,OAAO,SAAS,CAAC;QAClB,CAAC;QAED,OAAO,IAAA,mCAAoB,EAAC;YAC3B,QAAQ,EAAE,eAAe;YACzB,eAAe,EAAE,OAAO,CAAC,eAAe,EAAE,eAAe;YACzD,aAAa,EAAE,OAAO,CAAC,eAAe,EAAE,aAAa;YACrD,gBAAgB,EAAE,OAAO,CAAC,eAAe,EAAE,gBAAgB;YAC3D,cAAc,EAAE,OAAO,CAAC,eAAe,EAAE,cAAc;YACvD,WAAW,EAAE,OAAO,CAAC,eAAe,EAAE,WAAW;SACjD,CAAC,CAAC;IACJ,CAAC;IAOO,cAAc,CAAC,iBAA0B,EAAE,aAAsB;QACxE,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,SAAS,EAAE,CAAC;YACzB,MAAM,IAAI,KAAK,CAAC,sEAAsE,CAAC,CAAC;QACzF,CAAC;QAID,IAAI,KAAK,GAKL,aAAa,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,qBAAqB,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC;QAE5F,IAAI,iBAAiB,EAAE,CAAC;YACvB,KAAK,GAAG,CAAC,GAAG,KAAK,EAAE,4BAAgB,EAAE,yBAAa,EAAE,kCAAsB,CAAC,CAAC;QAC7E,CAAC;QAED,OAAO,IAAI,CAAC,GAAG,CAAC,SAAS,CAAC,KAAK,CAAiB,CAAC;IAClD,CAAC;IAKO,KAAK,CAAC,qBAAqB,CAClC,MAAiD,EACjD,WAAmB;QAEnB,MAAM,iBAAiB,GAAG,MAAM,MAAM,CAAC,cAAc,CAAC,EAAE,WAAW,EAAE,CAAC,CAAC;QACvE,OAAO,CAAC,GAAG,iBAAiB,CAAC,CAAC;IAC/B,CAAC;IAKO,4BAA4B,CACnC,iBAA0B,EAC1B,wBAAiC;QAKjC,IAAI,CAAC,iBAAiB,EAAE,CAAC;YACxB,OAAO,EAAE,CAAC;QACX,CAAC;QAED,MAAM,iBAAiB,GAAG,IAAI,uCAAiB,EAAE,CAAC;QAGlD,MAAM,qBAAqB,GAAG,IAAI,gDAAqB,CAAC;YACvD,iBAAiB,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC,iBAAiB,CAAC,OAAO,CAAC,IAAoC,CAAC;YAC5F,iBAAiB,EAAE,GAAG,EAAE,CAAC,iBAAiB,CAAC,eAAe,EAAE;YAC5D,gBAAgB,EAAE,IAAI,CAAC,gBAAgB;YACvC,eAAe,EAAE,IAAI,CAAC,eAAe;SACrC,CAAC,CAAC;QAIH,IAAI,wBAAwB,EAAE,CAAC;YAC9B,MAAM,cAAc,GAAG,GAAG,mCAAoB,OAAO,wBAAwB,EAAE,CAAC;YAChF,iBAAiB,CAAC,eAAe,CAAC,cAAc,CAAC,CAAC;QACnD,CAAC;QAED,OAAO,EAAE,iBAAiB,EAAE,qBAAqB,EAAE,CAAC;IACrD,CAAC;IAMO,qBAAqB,CAAC,IAAgB;QAC7C,MAAM,SAAS,GAAG,wBAAwB,CAAC,IAAI,CAAC,CAAC;QACjD,IAAI,SAAS,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO,SAAS,CAAC;QAC7C,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,CAAC;YAC1B,IAAI,CAAC,MAAM,EAAE,IAAI,CAAC,yEAAyE,CAAC,CAAC;YAC7F,OAAO,SAAS,CAAC;QAClB,CAAC;QAED,MAAM,gBAAgB,GAAa,EAAE,CAAC;QACtC,KAAK,MAAM,QAAQ,IAAI,SAAS,EAAE,CAAC;YAClC,MAAM,MAAM,GAAG,IAAA,2CAAgB,EAAC,IAAI,CAAC,cAAc,EAAE,QAAQ,CAAC,CAAC;YAC/D,IAAI,MAAM,EAAE,CAAC;gBACZ,gBAAgB,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YAC/B,CAAC;QACF,CAAC;QAED,IAAI,gBAAgB,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO,SAAS,CAAC;QAEpD,OAAO,SAAS,gBAAgB,CAAC,MAAM,cAAc,gBAAgB,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC;IACtF,CAAC;CACD;AAtND,4CAsNC;AAKD,SAAgB,wBAAwB,CAAC,IAAgB;IACxD,MAAM,OAAO,GAAG,IAAI,GAAG,EAAU,CAAC;IAClC,KAAK,MAAM,IAAI,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC;QAC/B,IAAI,IAAI,CAAC,cAAc,EAAE,CAAC;YACzB,KAAK,MAAM,QAAQ,IAAI,IAAI,CAAC,cAAc,EAAE,CAAC;gBAC5C,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;YACvB,CAAC;QACF,CAAC;IACF,CAAC;IACD,OAAO,CAAC,GAAG,OAAO,CAAC,CAAC;AACrB,CAAC"}

package/dist/code-builder/handlers/final-response-handler.d.ts

@@ -0,0 +1,29 @@
+import type { AIMessage, BaseMessage } from '@langchain/core/messages';
+import type { WorkflowJSON } from '@n8n/workflow-sdk';
+import type { WarningTracker } from '../state/warning-tracker';
+import type { ParseAndValidateResult } from '../types';
+type ParseAndValidateFn = (code: string, currentWorkflow?: WorkflowJSON) => Promise<ParseAndValidateResult>;
+export interface FinalResponseHandlerConfig {
+    parseAndValidate: ParseAndValidateFn;
+}
+export interface FinalResponseParams {
+    response: AIMessage;
+    currentWorkflow: WorkflowJSON | undefined;
+    messages: BaseMessage[];
+    warningTracker: WarningTracker;
+}
+export interface FinalResponseResult {
+    success: boolean;
+    workflow?: WorkflowJSON;
+    sourceCode?: string;
+    parseDuration?: number;
+    isParseError?: boolean;
+    shouldContinue?: boolean;
+}
+export declare class FinalResponseHandler {
+    private parseAndValidate;
+    constructor(config: FinalResponseHandlerConfig);
+    process(params: FinalResponseParams): Promise<FinalResponseResult>;
+    private parseStructuredOutput;
+}
+export {};

package/dist/code-builder/handlers/final-response-handler.js

@@ -0,0 +1,75 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.FinalResponseHandler = void 0;
+const content_extractors_1 = require("../utils/content-extractors");
+const extract_code_1 = require("../utils/extract-code");
+const format_warnings_1 = require("../utils/format-warnings");
+class FinalResponseHandler {
+    parseAndValidate;
+    constructor(config) {
+        this.parseAndValidate = config.parseAndValidate;
+    }
+    async process(params) {
+        const { response, currentWorkflow, messages, warningTracker } = params;
+        const parseResult = this.parseStructuredOutput(response);
+        if (!parseResult.result) {
+            (0, content_extractors_1.pushValidationFeedback)(messages, `Could not parse your response: ${parseResult.error}\n\nPlease provide your workflow code in a \`\`\`typescript code block.`);
+            return {
+                success: false,
+                isParseError: true,
+                shouldContinue: true,
+            };
+        }
+        const workflowCode = parseResult.result.workflowCode;
+        const parseStartTime = Date.now();
+        try {
+            const result = await this.parseAndValidate(workflowCode, currentWorkflow);
+            const parseDuration = Date.now() - parseStartTime;
+            const newWarnings = warningTracker.filterNewWarnings(result.warnings);
+            if (newWarnings.length > 0) {
+                warningTracker.markAsSeen(newWarnings);
+                const warningMessages = (0, format_warnings_1.formatWarnings)(newWarnings.slice(0, 5), warningTracker);
+                (0, content_extractors_1.pushValidationFeedback)(messages, `The workflow code has validation warnings that should be addressed:\n\n${warningMessages}\n\nPlease fix these issues and provide the corrected version in a \`\`\`typescript code block.`);
+                return {
+                    success: false,
+                    parseDuration,
+                    shouldContinue: true,
+                };
+            }
+            return {
+                success: true,
+                workflow: result.workflow,
+                sourceCode: workflowCode,
+                parseDuration,
+                shouldContinue: false,
+            };
+        }
+        catch (parseError) {
+            const parseDuration = Date.now() - parseStartTime;
+            const errorMessage = parseError instanceof Error ? parseError.message : String(parseError);
+            (0, content_extractors_1.pushValidationFeedback)(messages, `The workflow code you generated has a parsing error:\n\n${errorMessage}\n\nPlease fix the code and provide the corrected version in a \`\`\`typescript code block.`);
+            return {
+                success: false,
+                parseDuration,
+                isParseError: true,
+                shouldContinue: true,
+            };
+        }
+    }
+    parseStructuredOutput(message) {
+        const content = (0, content_extractors_1.extractTextContent)(message);
+        if (!content) {
+            return { result: null, error: 'No text content found in response' };
+        }
+        const workflowCode = (0, extract_code_1.extractWorkflowCode)(content);
+        if (!workflowCode?.includes('workflow')) {
+            return {
+                result: null,
+                error: 'No valid workflow code found in response. Please provide your code in a ```typescript code block.',
+            };
+        }
+        return { result: { workflowCode }, error: null };
+    }
+}
+exports.FinalResponseHandler = FinalResponseHandler;
+//# sourceMappingURL=final-response-handler.js.map

package/dist/code-builder/handlers/final-response-handler.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"final-response-handler.js","sourceRoot":"","sources":["../../../src/code-builder/handlers/final-response-handler.ts"],"names":[],"mappings":";;;AAYA,oEAAyF;AACzF,wDAA4D;AAC5D,8DAA0D;AA0D1D,MAAa,oBAAoB;IACxB,gBAAgB,CAAqB;IAE7C,YAAY,MAAkC;QAC7C,IAAI,CAAC,gBAAgB,GAAG,MAAM,CAAC,gBAAgB,CAAC;IACjD,CAAC;IAQD,KAAK,CAAC,OAAO,CAAC,MAA2B;QACxC,MAAM,EAAE,QAAQ,EAAE,eAAe,EAAE,QAAQ,EAAE,cAAc,EAAE,GAAG,MAAM,CAAC;QAGvE,MAAM,WAAW,GAAG,IAAI,CAAC,qBAAqB,CAAC,QAAQ,CAAC,CAAC;QAEzD,IAAI,CAAC,WAAW,CAAC,MAAM,EAAE,CAAC;YAEzB,IAAA,2CAAsB,EACrB,QAAQ,EACR,kCAAkC,WAAW,CAAC,KAAK,yEAAyE,CAC5H,CAAC;YAEF,OAAO;gBACN,OAAO,EAAE,KAAK;gBACd,YAAY,EAAE,IAAI;gBAClB,cAAc,EAAE,IAAI;aACpB,CAAC;QACH,CAAC;QAED,MAAM,YAAY,GAAG,WAAW,CAAC,MAAM,CAAC,YAAY,CAAC;QAGrD,MAAM,cAAc,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAElC,IAAI,CAAC;YACJ,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,gBAAgB,CAAC,YAAY,EAAE,eAAe,CAAC,CAAC;YAC1E,MAAM,aAAa,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,cAAc,CAAC;YAGlD,MAAM,WAAW,GAAG,cAAc,CAAC,iBAAiB,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;YAEtE,IAAI,WAAW,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBAE5B,cAAc,CAAC,UAAU,CAAC,WAAW,CAAC,CAAC;gBAGvC,MAAM,eAAe,GAAG,IAAA,gCAAc,EAAC,WAAW,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,cAAc,CAAC,CAAC;gBAGhF,IAAA,2CAAsB,EACrB,QAAQ,EACR,0EAA0E,eAAe,iGAAiG,CAC1L,CAAC;gBAEF,OAAO;oBACN,OAAO,EAAE,KAAK;oBACd,aAAa;oBACb,cAAc,EAAE,IAAI;iBACpB,CAAC;YACH,CAAC;YAED,OAAO;gBACN,OAAO,EAAE,IAAI;gBACb,QAAQ,EAAE,MAAM,CAAC,QAAQ;gBACzB,UAAU,EAAE,YAAY;gBACxB,aAAa;gBACb,cAAc,EAAE,KAAK;aACrB,CAAC;QACH,CAAC;QAAC,OAAO,UAAU,EAAE,CAAC;YACrB,MAAM,aAAa,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,cAAc,CAAC;YAClD,MAAM,YAAY,GAAG,UAAU,YAAY,KAAK,CAAC,CAAC,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC;YAG3F,IAAA,2CAAsB,EACrB,QAAQ,EACR,2DAA2D,YAAY,6FAA6F,CACpK,CAAC;YAEF,OAAO;gBACN,OAAO,EAAE,KAAK;gBACd,aAAa;gBACb,YAAY,EAAE,IAAI;gBAClB,cAAc,EAAE,IAAI;aACpB,CAAC;QACH,CAAC;IACF,CAAC;IAMO,qBAAqB,CAAC,OAAkB;QAI/C,MAAM,OAAO,GAAG,IAAA,uCAAkB,EAAC,OAAO,CAAC,CAAC;QAC5C,IAAI,CAAC,OAAO,EAAE,CAAC;YACd,OAAO,EAAE,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,mCAAmC,EAAE,CAAC;QACrE,CAAC;QAGD,MAAM,YAAY,GAAG,IAAA,kCAAmB,EAAC,OAAO,CAAC,CAAC;QAGlD,IAAI,CAAC,YAAY,EAAE,QAAQ,CAAC,UAAU,CAAC,EAAE,CAAC;YACzC,OAAO;gBACN,MAAM,EAAE,IAAI;gBACZ,KAAK,EACJ,mGAAmG;aACpG,CAAC;QACH,CAAC;QAED,OAAO,EAAE,MAAM,EAAE,EAAE,YAAY,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC;IAClD,CAAC;CACD;AAtHD,oDAsHC"}

package/dist/code-builder/handlers/parse-validate-handler.d.ts

@@ -0,0 +1,17 @@
+import type { Logger } from '@n8n/backend-common';
+import type { WorkflowJSON } from '@n8n/workflow-sdk';
+import type { ParseAndValidateResult, ValidationWarning } from '../types';
+export interface ParseValidateHandlerConfig {
+    logger?: Logger;
+    generatePinData?: boolean;
+}
+export declare class ParseValidateHandler {
+    private logger?;
+    private generatePinData;
+    constructor(config?: ParseValidateHandlerConfig);
+    private collectValidationIssues;
+    validateExistingWorkflow(json: WorkflowJSON): ValidationWarning[];
+    parseAndValidate(code: string, currentWorkflow?: WorkflowJSON): Promise<ParseAndValidateResult>;
+    getErrorContext(code: string, errorMessage: string): string;
+    private detectFixHint;
+}

package/dist/code-builder/handlers/parse-validate-handler.js

@@ -0,0 +1,116 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ParseValidateHandler = void 0;
+const workflow_sdk_1 = require("@n8n/workflow-sdk");
+const extract_code_1 = require("../utils/extract-code");
+class ParseValidateHandler {
+    logger;
+    generatePinData;
+    constructor(config = {}) {
+        this.logger = config.logger;
+        this.generatePinData = config.generatePinData ?? true;
+    }
+    collectValidationIssues(issues, allWarnings, context, logLevel) {
+        if (issues.length === 0)
+            return;
+        if (logLevel === 'warn') {
+            this.logger?.warn(`Graph validation ${context.toLowerCase()}`, {
+                [context.toLowerCase()]: issues.map((i) => i.message),
+            });
+        }
+        else {
+            this.logger?.info(`Graph validation ${context.toLowerCase()}`, {
+                [context.toLowerCase()]: issues.map((i) => i.message),
+            });
+        }
+        for (const issue of issues) {
+            allWarnings.push({
+                code: issue.code,
+                message: issue.message,
+                nodeName: issue.nodeName,
+            });
+        }
+    }
+    validateExistingWorkflow(json) {
+        if (json.nodes.length === 0) {
+            return [];
+        }
+        const builder = workflow_sdk_1.workflow.fromJSON(json);
+        const allWarnings = [];
+        const graphValidation = builder.validate();
+        this.collectValidationIssues(graphValidation.errors, allWarnings, 'GRAPH VALIDATION ERRORS', 'warn');
+        this.collectValidationIssues(graphValidation.warnings, allWarnings, 'GRAPH VALIDATION WARNINGS', 'info');
+        return allWarnings;
+    }
+    async parseAndValidate(code, currentWorkflow) {
+        const codeToParse = (0, extract_code_1.stripImportStatements)(code);
+        try {
+            this.logger?.debug('Parsing WorkflowCode', { codeLength: codeToParse.length });
+            const builder = (0, workflow_sdk_1.parseWorkflowCodeToBuilder)(codeToParse);
+            builder.regenerateNodeIds();
+            const allWarnings = [];
+            const graphValidation = builder.validate();
+            this.collectValidationIssues(graphValidation.errors, allWarnings, 'GRAPH VALIDATION ERRORS', 'warn');
+            this.collectValidationIssues(graphValidation.warnings, allWarnings, 'GRAPH VALIDATION WARNINGS', 'info');
+            const json = builder.toJSON();
+            const validationResult = (0, workflow_sdk_1.validateWorkflow)(json);
+            this.collectValidationIssues(validationResult.errors, allWarnings, 'JSON VALIDATION ERRORS', 'warn');
+            this.collectValidationIssues(validationResult.warnings, allWarnings, 'JSON VALIDATION WARNINGS', 'info');
+            if (this.generatePinData) {
+                builder.generatePinData({ beforeWorkflow: currentWorkflow });
+            }
+            const workflowJson = builder.toJSON();
+            this.logger?.debug('Parsed workflow', {
+                id: workflowJson.id,
+                name: workflowJson.name,
+                nodeCount: workflowJson.nodes.length,
+            });
+            return { workflow: workflowJson, warnings: allWarnings };
+        }
+        catch (error) {
+            this.logger?.error('Failed to parse WorkflowCode', {
+                error: error instanceof Error ? error.message : String(error),
+                code: code.substring(0, 500),
+            });
+            throw new Error(`Failed to parse generated workflow code: ${error instanceof Error ? error.message : 'Unknown error'}`);
+        }
+    }
+    getErrorContext(code, errorMessage) {
+        const lineMatch = errorMessage.match(/(?:line|Line)\s*(\d+)/i);
+        if (!lineMatch) {
+            const lines = code.split('\n').slice(0, 10);
+            return `Code context:\n${lines.map((l, i) => `${i + 1}: ${l}`).join('\n')}`;
+        }
+        const errorLine = parseInt(lineMatch[1], 10);
+        const lines = code.split('\n');
+        const start = Math.max(0, errorLine - 4);
+        const end = Math.min(lines.length, errorLine + 3);
+        const context = lines
+            .slice(start, end)
+            .map((l, i) => {
+            const lineNum = start + i + 1;
+            const marker = lineNum === errorLine ? '> ' : '  ';
+            return `${marker}${lineNum}: ${l}`;
+        })
+            .join('\n');
+        const contextText = `Code around line ${errorLine}:\n${context}`;
+        const hint = this.detectFixHint(lines, errorLine);
+        if (hint) {
+            return `${contextText}\n\n${hint}`;
+        }
+        return contextText;
+    }
+    detectFixHint(lines, errorLine) {
+        const searchStart = Math.max(0, errorLine - 5);
+        const searchEnd = Math.min(lines.length, errorLine + 5);
+        const nearbyCode = lines.slice(searchStart, searchEnd).join('\n');
+        if (/expr\s*\(\s*`[^`]*\$\{\{/.test(nearbyCode)) {
+            return ("HINT: The '$' before '{{' inside a backtick template literal is interpreted as JS template interpolation. " +
+                "Use single quotes instead of backticks for expr(), e.g. expr('Amount: ${{ $json.amount }}'). " +
+                'Fix ALL occurrences in the file at once.');
+        }
+        return null;
+    }
+}
+exports.ParseValidateHandler = ParseValidateHandler;
+//# sourceMappingURL=parse-validate-handler.js.map

package/dist/code-builder/handlers/parse-validate-handler.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"parse-validate-handler.js","sourceRoot":"","sources":["../../../src/code-builder/handlers/parse-validate-handler.ts"],"names":[],"mappings":";;;AASA,oDAA2F;AAI3F,wDAA8D;AA0B9D,MAAa,oBAAoB;IACxB,MAAM,CAAU;IAChB,eAAe,CAAU;IAEjC,YAAY,SAAqC,EAAE;QAClD,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC,MAAM,CAAC;QAC5B,IAAI,CAAC,eAAe,GAAG,MAAM,CAAC,eAAe,IAAI,IAAI,CAAC;IACvD,CAAC;IAMO,uBAAuB,CAC9B,MAAyB,EACzB,WAAgC,EAChC,OAAe,EACf,QAAyB;QAEzB,IAAI,MAAM,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO;QAEhC,IAAI,QAAQ,KAAK,MAAM,EAAE,CAAC;YACzB,IAAI,CAAC,MAAM,EAAE,IAAI,CAAC,oBAAoB,OAAO,CAAC,WAAW,EAAE,EAAE,EAAE;gBAC9D,CAAC,OAAO,CAAC,WAAW,EAAE,CAAC,EAAE,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC;aACrD,CAAC,CAAC;QACJ,CAAC;aAAM,CAAC;YACP,IAAI,CAAC,MAAM,EAAE,IAAI,CAAC,oBAAoB,OAAO,CAAC,WAAW,EAAE,EAAE,EAAE;gBAC9D,CAAC,OAAO,CAAC,WAAW,EAAE,CAAC,EAAE,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC;aACrD,CAAC,CAAC;QACJ,CAAC;QAED,KAAK,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;YAC5B,WAAW,CAAC,IAAI,CAAC;gBAChB,IAAI,EAAE,KAAK,CAAC,IAAI;gBAChB,OAAO,EAAE,KAAK,CAAC,OAAO;gBACtB,QAAQ,EAAE,KAAK,CAAC,QAAQ;aACxB,CAAC,CAAC;QACJ,CAAC;IACF,CAAC;IAcD,wBAAwB,CAAC,IAAkB;QAC1C,IAAI,IAAI,CAAC,KAAK,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC7B,OAAO,EAAE,CAAC;QACX,CAAC;QAED,MAAM,OAAO,GAAG,uBAAQ,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC;QACxC,MAAM,WAAW,GAAwB,EAAE,CAAC;QAC5C,MAAM,eAAe,GAAG,OAAO,CAAC,QAAQ,EAAE,CAAC;QAC3C,IAAI,CAAC,uBAAuB,CAC3B,eAAe,CAAC,MAAM,EACtB,WAAW,EACX,yBAAyB,EACzB,MAAM,CACN,CAAC;QACF,IAAI,CAAC,uBAAuB,CAC3B,eAAe,CAAC,QAAQ,EACxB,WAAW,EACX,2BAA2B,EAC3B,MAAM,CACN,CAAC;QACF,OAAO,WAAW,CAAC;IACpB,CAAC;IAUD,KAAK,CAAC,gBAAgB,CACrB,IAAY,EACZ,eAA8B;QAG9B,MAAM,WAAW,GAAG,IAAA,oCAAqB,EAAC,IAAI,CAAC,CAAC;QAEhD,IAAI,CAAC;YAEJ,IAAI,CAAC,MAAM,EAAE,KAAK,CAAC,sBAAsB,EAAE,EAAE,UAAU,EAAE,WAAW,CAAC,MAAM,EAAE,CAAC,CAAC;YAC/E,MAAM,OAAO,GAAG,IAAA,yCAA0B,EAAC,WAAW,CAAC,CAAC;YAGxD,OAAO,CAAC,iBAAiB,EAAE,CAAC;YAG5B,MAAM,WAAW,GAAwB,EAAE,CAAC;YAG5C,MAAM,eAAe,GAAG,OAAO,CAAC,QAAQ,EAAE,CAAC;YAG3C,IAAI,CAAC,uBAAuB,CAC3B,eAAe,CAAC,MAAM,EACtB,WAAW,EACX,yBAAyB,EACzB,MAAM,CACN,CAAC;YAGF,IAAI,CAAC,uBAAuB,CAC3B,eAAe,CAAC,QAAQ,EACxB,WAAW,EACX,2BAA2B,EAC3B,MAAM,CACN,CAAC;YAGF,MAAM,IAAI,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC;YAG9B,MAAM,gBAAgB,GAAG,IAAA,+BAAgB,EAAC,IAAI,CAAC,CAAC;YAGhD,IAAI,CAAC,uBAAuB,CAC3B,gBAAgB,CAAC,MAAM,EACvB,WAAW,EACX,wBAAwB,EACxB,MAAM,CACN,CAAC;YAGF,IAAI,CAAC,uBAAuB,CAC3B,gBAAgB,CAAC,QAAQ,EACzB,WAAW,EACX,0BAA0B,EAC1B,MAAM,CACN,CAAC;YAGF,IAAI,IAAI,CAAC,eAAe,EAAE,CAAC;gBAC1B,OAAO,CAAC,eAAe,CAAC,EAAE,cAAc,EAAE,eAAe,EAAE,CAAC,CAAC;YAC9D,CAAC;YAGD,MAAM,YAAY,GAAiB,OAAO,CAAC,MAAM,EAAE,CAAC;YAEpD,IAAI,CAAC,MAAM,EAAE,KAAK,CAAC,iBAAiB,EAAE;gBACrC,EAAE,EAAE,YAAY,CAAC,EAAE;gBACnB,IAAI,EAAE,YAAY,CAAC,IAAI;gBACvB,SAAS,EAAE,YAAY,CAAC,KAAK,CAAC,MAAM;aACpC,CAAC,CAAC;YAGH,OAAO,EAAE,QAAQ,EAAE,YAAY,EAAE,QAAQ,EAAE,WAAW,EAAE,CAAC;QAC1D,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YAChB,IAAI,CAAC,MAAM,EAAE,KAAK,CAAC,8BAA8B,EAAE;gBAClD,KAAK,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC;gBAC7D,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,CAAC,EAAE,GAAG,CAAC;aAC5B,CAAC,CAAC;YAEH,MAAM,IAAI,KAAK,CACd,4CAA4C,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,eAAe,EAAE,CACtG,CAAC;QACH,CAAC;IACF,CAAC;IASD,eAAe,CAAC,IAAY,EAAE,YAAoB;QAEjD,MAAM,SAAS,GAAG,YAAY,CAAC,KAAK,CAAC,wBAAwB,CAAC,CAAC;QAC/D,IAAI,CAAC,SAAS,EAAE,CAAC;YAEhB,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;YAC5C,OAAO,kBAAkB,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,GAAG,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC;QAC7E,CAAC;QAED,MAAM,SAAS,GAAG,QAAQ,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;QAC7C,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QAG/B,MAAM,KAAK,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,SAAS,GAAG,CAAC,CAAC,CAAC;QACzC,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,MAAM,EAAE,SAAS,GAAG,CAAC,CAAC,CAAC;QAClD,MAAM,OAAO,GAAG,KAAK;aACnB,KAAK,CAAC,KAAK,EAAE,GAAG,CAAC;aACjB,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE;YACb,MAAM,OAAO,GAAG,KAAK,GAAG,CAAC,GAAG,CAAC,CAAC;YAC9B,MAAM,MAAM,GAAG,OAAO,KAAK,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC;YACnD,OAAO,GAAG,MAAM,GAAG,OAAO,KAAK,CAAC,EAAE,CAAC;QACpC,CAAC,CAAC;aACD,IAAI,CAAC,IAAI,CAAC,CAAC;QAEb,MAAM,WAAW,GAAG,oBAAoB,SAAS,MAAM,OAAO,EAAE,CAAC;QAEjE,MAAM,IAAI,GAAG,IAAI,CAAC,aAAa,CAAC,KAAK,EAAE,SAAS,CAAC,CAAC;QAClD,IAAI,IAAI,EAAE,CAAC;YACV,OAAO,GAAG,WAAW,OAAO,IAAI,EAAE,CAAC;QACpC,CAAC;QAED,OAAO,WAAW,CAAC;IACpB,CAAC;IAQO,aAAa,CAAC,KAAe,EAAE,SAAiB;QACvD,MAAM,WAAW,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,SAAS,GAAG,CAAC,CAAC,CAAC;QAC/C,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,MAAM,EAAE,SAAS,GAAG,CAAC,CAAC,CAAC;QACxD,MAAM,UAAU,GAAG,KAAK,CAAC,KAAK,CAAC,WAAW,EAAE,SAAS,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAGlE,IAAI,0BAA0B,CAAC,IAAI,CAAC,UAAU,CAAC,EAAE,CAAC;YACjD,OAAO,CACN,4GAA4G;gBAE5G,+FAA+F;gBAC/F,0CAA0C,CAC1C,CAAC;QACH,CAAC;QAED,OAAO,IAAI,CAAC;IACb,CAAC;CACD;AA1OD,oDA0OC"}

package/dist/code-builder/handlers/session-chat-handler.d.ts

@@ -0,0 +1,30 @@
+import type { BaseChatModel } from '@langchain/core/language_models/chat_models';
+import type { MemorySaver } from '@langchain/langgraph';
+import type { Logger } from '@n8n/backend-common';
+import type { StreamOutput } from '../../types/streaming';
+import type { ChatPayload } from '../../workflow-builder-agent';
+import type { HistoryContext } from '../prompts';
+type AgentChatFn = (payload: ChatPayload, userId: string, abortSignal?: AbortSignal, historyContext?: HistoryContext) => AsyncGenerator<StreamOutput, void, unknown>;
+export interface SessionChatHandlerConfig {
+    checkpointer: MemorySaver;
+    llm: BaseChatModel;
+    logger?: Logger;
+    onGenerationSuccess?: () => Promise<void>;
+}
+export interface SessionChatParams {
+    payload: ChatPayload;
+    userId: string;
+    abortSignal?: AbortSignal;
+    agentChat: AgentChatFn;
+}
+export declare class SessionChatHandler {
+    private checkpointer;
+    private llm;
+    private logger?;
+    private onGenerationSuccess?;
+    constructor(config: SessionChatHandlerConfig);
+    private buildHistoryContext;
+    private processChunk;
+    execute(params: SessionChatParams): AsyncGenerator<StreamOutput, void, unknown>;
+}
+export {};

package/dist/code-builder/handlers/session-chat-handler.js

@@ -0,0 +1,96 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.SessionChatHandler = void 0;
+const code_builder_session_1 = require("../utils/code-builder-session");
+function isSessionMessagesChunk(chunk) {
+    return chunk.type === 'session-messages' && 'messages' in chunk;
+}
+class SessionChatHandler {
+    checkpointer;
+    llm;
+    logger;
+    onGenerationSuccess;
+    constructor(config) {
+        this.checkpointer = config.checkpointer;
+        this.llm = config.llm;
+        this.logger = config.logger;
+        this.onGenerationSuccess = config.onGenerationSuccess;
+    }
+    buildHistoryContext(session) {
+        if (session.conversationEntries.length > 0 || session.previousSummary) {
+            return {
+                conversationEntries: session.conversationEntries,
+                previousSummary: session.previousSummary,
+            };
+        }
+        return undefined;
+    }
+    processChunk(chunk) {
+        let generationSucceeded = false;
+        let sessionMessages;
+        if (chunk.messages?.some((msg) => msg.type === 'workflow-updated')) {
+            generationSucceeded = true;
+        }
+        for (const msg of chunk.messages ?? []) {
+            if (isSessionMessagesChunk(msg)) {
+                sessionMessages = msg.messages;
+            }
+        }
+        const filteredMessages = chunk.messages?.filter((msg) => msg.type !== 'session-messages');
+        return { generationSucceeded, sessionMessages, filteredMessages };
+    }
+    async *execute(params) {
+        const { payload, userId, abortSignal, agentChat } = params;
+        const workflowId = payload.workflowContext?.currentWorkflow?.id;
+        if (!workflowId) {
+            this.logger?.debug('No workflow ID, skipping session management');
+            yield* agentChat(payload, userId, abortSignal);
+            return;
+        }
+        const threadId = (0, code_builder_session_1.generateCodeBuilderThreadId)(workflowId, userId);
+        let session = await (0, code_builder_session_1.loadCodeBuilderSession)(this.checkpointer, threadId);
+        this.logger?.debug('Loaded CodeBuilder session', {
+            threadId,
+            conversationEntriesCount: session.conversationEntries.length,
+            hasPreviousSummary: !!session.previousSummary,
+        });
+        session = await (0, code_builder_session_1.compactSessionIfNeeded)(session, this.llm);
+        const historyContext = this.buildHistoryContext(session);
+        let generationSucceeded = false;
+        let sessionMessages;
+        for await (const chunk of agentChat(payload, userId, abortSignal, historyContext)) {
+            const result = this.processChunk(chunk);
+            if (result.generationSucceeded)
+                generationSucceeded = true;
+            if (result.sessionMessages)
+                sessionMessages = result.sessionMessages;
+            if (result.filteredMessages && result.filteredMessages.length > 0) {
+                yield { messages: result.filteredMessages };
+            }
+        }
+        session.conversationEntries.push({ type: 'build-request', message: payload.message });
+        await (0, code_builder_session_1.saveCodeBuilderSession)(this.checkpointer, threadId, session);
+        if (generationSucceeded && sessionMessages) {
+            await (0, code_builder_session_1.saveSessionMessages)(this.checkpointer, workflowId, userId, sessionMessages, payload.versionId, payload.id);
+            this.logger?.debug('Saved session messages to SessionManager thread', {
+                workflowId,
+                userId,
+                messageCount: sessionMessages.length,
+            });
+        }
+        if (generationSucceeded && this.onGenerationSuccess) {
+            try {
+                await this.onGenerationSuccess();
+            }
+            catch (error) {
+                this.logger?.warn('Failed to execute onGenerationSuccess callback', { error });
+            }
+        }
+        this.logger?.debug('Saved CodeBuilder session', {
+            threadId,
+            newEntryCount: session.conversationEntries.length,
+        });
+    }
+}
+exports.SessionChatHandler = SessionChatHandler;
+//# sourceMappingURL=session-chat-handler.js.map

package/dist/code-builder/handlers/session-chat-handler.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"session-chat-handler.js","sourceRoot":"","sources":["../../../src/code-builder/handlers/session-chat-handler.ts"],"names":[],"mappings":";;;AAeA,wEAMuC;AAGvC,SAAS,sBAAsB,CAAC,KAAkB;IACjD,OAAO,KAAK,CAAC,IAAI,KAAK,kBAAkB,IAAI,UAAU,IAAI,KAAK,CAAC;AACjE,CAAC;AA+CD,MAAa,kBAAkB;IACtB,YAAY,CAAc;IAC1B,GAAG,CAAgB;IACnB,MAAM,CAAU;IAChB,mBAAmB,CAAuB;IAElD,YAAY,MAAgC;QAC3C,IAAI,CAAC,YAAY,GAAG,MAAM,CAAC,YAAY,CAAC;QACxC,IAAI,CAAC,GAAG,GAAG,MAAM,CAAC,GAAG,CAAC;QACtB,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC,MAAM,CAAC;QAC5B,IAAI,CAAC,mBAAmB,GAAG,MAAM,CAAC,mBAAmB,CAAC;IACvD,CAAC;IAKO,mBAAmB,CAAC,OAG3B;QACA,IAAI,OAAO,CAAC,mBAAmB,CAAC,MAAM,GAAG,CAAC,IAAI,OAAO,CAAC,eAAe,EAAE,CAAC;YACvE,OAAO;gBACN,mBAAmB,EAAE,OAAO,CAAC,mBAAmB;gBAChD,eAAe,EAAE,OAAO,CAAC,eAAe;aACxC,CAAC;QACH,CAAC;QACD,OAAO,SAAS,CAAC;IAClB,CAAC;IAKO,YAAY,CAAC,KAAmB;QAKvC,IAAI,mBAAmB,GAAG,KAAK,CAAC;QAChC,IAAI,eAAsC,CAAC;QAE3C,IAAI,KAAK,CAAC,QAAQ,EAAE,IAAI,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,IAAI,KAAK,kBAAkB,CAAC,EAAE,CAAC;YACpE,mBAAmB,GAAG,IAAI,CAAC;QAC5B,CAAC;QAED,KAAK,MAAM,GAAG,IAAI,KAAK,CAAC,QAAQ,IAAI,EAAE,EAAE,CAAC;YACxC,IAAI,sBAAsB,CAAC,GAAG,CAAC,EAAE,CAAC;gBACjC,eAAe,GAAG,GAAG,CAAC,QAAQ,CAAC;YAChC,CAAC;QACF,CAAC;QAED,MAAM,gBAAgB,GAAG,KAAK,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,IAAI,KAAK,kBAAkB,CAAC,CAAC;QAE1F,OAAO,EAAE,mBAAmB,EAAE,eAAe,EAAE,gBAAgB,EAAE,CAAC;IACnE,CAAC;IAQD,KAAK,CAAC,CAAC,OAAO,CAAC,MAAyB;QACvC,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,WAAW,EAAE,SAAS,EAAE,GAAG,MAAM,CAAC;QAG3D,MAAM,UAAU,GAAG,OAAO,CAAC,eAAe,EAAE,eAAe,EAAE,EAAE,CAAC;QAEhE,IAAI,CAAC,UAAU,EAAE,CAAC;YAEjB,IAAI,CAAC,MAAM,EAAE,KAAK,CAAC,6CAA6C,CAAC,CAAC;YAClE,KAAK,CAAC,CAAC,SAAS,CAAC,OAAO,EAAE,MAAM,EAAE,WAAW,CAAC,CAAC;YAC/C,OAAO;QACR,CAAC;QAED,MAAM,QAAQ,GAAG,IAAA,kDAA2B,EAAC,UAAU,EAAE,MAAM,CAAC,CAAC;QAGjE,IAAI,OAAO,GAAG,MAAM,IAAA,6CAAsB,EAAC,IAAI,CAAC,YAAY,EAAE,QAAQ,CAAC,CAAC;QAExE,IAAI,CAAC,MAAM,EAAE,KAAK,CAAC,4BAA4B,EAAE;YAChD,QAAQ;YACR,wBAAwB,EAAE,OAAO,CAAC,mBAAmB,CAAC,MAAM;YAC5D,kBAAkB,EAAE,CAAC,CAAC,OAAO,CAAC,eAAe;SAC7C,CAAC,CAAC;QAGH,OAAO,GAAG,MAAM,IAAA,6CAAsB,EAAC,OAAO,EAAE,IAAI,CAAC,GAAG,CAAC,CAAC;QAC1D,MAAM,cAAc,GAAG,IAAI,CAAC,mBAAmB,CAAC,OAAO,CAAC,CAAC;QAGzD,IAAI,mBAAmB,GAAG,KAAK,CAAC;QAChC,IAAI,eAAsC,CAAC;QAG3C,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,SAAS,CAAC,OAAO,EAAE,MAAM,EAAE,WAAW,EAAE,cAAc,CAAC,EAAE,CAAC;YACnF,MAAM,MAAM,GAAG,IAAI,CAAC,YAAY,CAAC,KAAK,CAAC,CAAC;YACxC,IAAI,MAAM,CAAC,mBAAmB;gBAAE,mBAAmB,GAAG,IAAI,CAAC;YAC3D,IAAI,MAAM,CAAC,eAAe;gBAAE,eAAe,GAAG,MAAM,CAAC,eAAe,CAAC;YAErE,IAAI,MAAM,CAAC,gBAAgB,IAAI,MAAM,CAAC,gBAAgB,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBACnE,MAAM,EAAE,QAAQ,EAAE,MAAM,CAAC,gBAAgB,EAAE,CAAC;YAC7C,CAAC;QACF,CAAC;QAGD,OAAO,CAAC,mBAAmB,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,eAAe,EAAE,OAAO,EAAE,OAAO,CAAC,OAAO,EAAE,CAAC,CAAC;QACtF,MAAM,IAAA,6CAAsB,EAAC,IAAI,CAAC,YAAY,EAAE,QAAQ,EAAE,OAAO,CAAC,CAAC;QAGnE,IAAI,mBAAmB,IAAI,eAAe,EAAE,CAAC;YAC5C,MAAM,IAAA,0CAAmB,EACxB,IAAI,CAAC,YAAY,EACjB,UAAU,EACV,MAAM,EACN,eAAe,EACf,OAAO,CAAC,SAAS,EACjB,OAAO,CAAC,EAAE,CACV,CAAC;YAEF,IAAI,CAAC,MAAM,EAAE,KAAK,CAAC,iDAAiD,EAAE;gBACrE,UAAU;gBACV,MAAM;gBACN,YAAY,EAAE,eAAe,CAAC,MAAM;aACpC,CAAC,CAAC;QACJ,CAAC;QAED,IAAI,mBAAmB,IAAI,IAAI,CAAC,mBAAmB,EAAE,CAAC;YACrD,IAAI,CAAC;gBACJ,MAAM,IAAI,CAAC,mBAAmB,EAAE,CAAC;YAClC,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBAChB,IAAI,CAAC,MAAM,EAAE,IAAI,CAAC,gDAAgD,EAAE,EAAE,KAAK,EAAE,CAAC,CAAC;YAChF,CAAC;QACF,CAAC;QAED,IAAI,CAAC,MAAM,EAAE,KAAK,CAAC,2BAA2B,EAAE;YAC/C,QAAQ;YACR,aAAa,EAAE,OAAO,CAAC,mBAAmB,CAAC,MAAM;SACjD,CAAC,CAAC;IACJ,CAAC;CACD;AA3ID,gDA2IC"}

package/dist/code-builder/handlers/text-editor-handler.d.ts

@@ -0,0 +1,19 @@
+import type { TextEditorCommand, StrReplacement, BatchReplaceResult } from './text-editor.types';
+export declare function formatCodeWithLineNumbers(code: string): string;
+export declare function findDivergenceContext(code: string, searchStr: string): string | undefined;
+export declare class TextEditorHandler {
+    private code;
+    constructor();
+    execute(command: TextEditorCommand): string;
+    private validatePath;
+    private handleView;
+    private handleCreate;
+    private handleStrReplace;
+    private handleInsert;
+    private countOccurrences;
+    executeBatch(replacements: StrReplacement[]): string | BatchReplaceResult[];
+    getWorkflowCode(): string | null;
+    setWorkflowCode(code: string): void;
+    hasWorkflowCode(): boolean;
+    clearWorkflowCode(): void;
+}