@atom8n/ai-workflow-builder 1.2.0

package/dist/agents/responder.agent.d.ts

@@ -0,0 +1,21 @@
+import type { BaseChatModel } from '@langchain/core/language_models/chat_models';
+import type { AIMessage, BaseMessage } from '@langchain/core/messages';
+import type { CoordinationLogEntry } from '../types/coordination';
+import type { DiscoveryContext } from '../types/discovery-types';
+import type { SimpleWorkflow } from '../types/workflow';
+export interface ResponderAgentConfig {
+    llm: BaseChatModel;
+}
+export interface ResponderContext {
+    messages: BaseMessage[];
+    coordinationLog: CoordinationLogEntry[];
+    discoveryContext?: DiscoveryContext | null;
+    workflowJSON: SimpleWorkflow;
+    previousSummary?: string;
+}
+export declare class ResponderAgent {
+    private llm;
+    constructor(config: ResponderAgentConfig);
+    private buildContextMessage;
+    invoke(context: ResponderContext): Promise<AIMessage>;
+}

package/dist/agents/responder.agent.js

@@ -0,0 +1,71 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ResponderAgent = void 0;
+const messages_1 = require("@langchain/core/messages");
+const prompts_1 = require("@langchain/core/prompts");
+const responder_prompt_1 = require("../prompts/agents/responder.prompt");
+const coordination_log_1 = require("../utils/coordination-log");
+const systemPrompt = prompts_1.ChatPromptTemplate.fromMessages([
+    [
+        'system',
+        [
+            {
+                type: 'text',
+                text: (0, responder_prompt_1.buildResponderPrompt)(),
+                cache_control: { type: 'ephemeral' },
+            },
+        ],
+    ],
+    ['placeholder', '{messages}'],
+]);
+class ResponderAgent {
+    llm;
+    constructor(config) {
+        this.llm = config.llm;
+    }
+    buildContextMessage(context) {
+        const contextParts = [];
+        if (context.previousSummary) {
+            contextParts.push(`**Previous Conversation Summary:**\n${context.previousSummary}`);
+        }
+        const stateManagementEntry = context.coordinationLog.find((e) => e.phase === 'state_management');
+        if (stateManagementEntry) {
+            contextParts.push(`**State Management:** ${stateManagementEntry.summary}`);
+        }
+        const errorEntry = (0, coordination_log_1.getErrorEntry)(context.coordinationLog);
+        if (errorEntry) {
+            contextParts.push(`**Error:** An error occurred in the ${errorEntry.phase} phase: ${errorEntry.summary}`);
+            contextParts.push('Please apologize to the user and explain that something went wrong while building their workflow.');
+        }
+        if (context.discoveryContext?.nodesFound.length) {
+            contextParts.push(`**Discovery:** Found ${context.discoveryContext.nodesFound.length} relevant nodes`);
+        }
+        const builderOutput = (0, coordination_log_1.getBuilderOutput)(context.coordinationLog);
+        if (builderOutput) {
+            contextParts.push(`**Builder:** ${builderOutput}`);
+        }
+        else if (context.workflowJSON.nodes.length) {
+            contextParts.push(`**Workflow:** ${context.workflowJSON.nodes.length} nodes created`);
+        }
+        const configuratorOutput = (0, coordination_log_1.getConfiguratorOutput)(context.coordinationLog);
+        if (configuratorOutput) {
+            contextParts.push(`**Configuration:**\n${configuratorOutput}`);
+        }
+        if (contextParts.length === 0) {
+            return null;
+        }
+        return new messages_1.HumanMessage({
+            content: `[Internal Context - Use this to craft your response]\n${contextParts.join('\n\n')}`,
+        });
+    }
+    async invoke(context) {
+        const agent = systemPrompt.pipe(this.llm);
+        const contextMessage = this.buildContextMessage(context);
+        const messagesToSend = contextMessage
+            ? [...context.messages, contextMessage]
+            : context.messages;
+        return (await agent.invoke({ messages: messagesToSend }));
+    }
+}
+exports.ResponderAgent = ResponderAgent;
+//# sourceMappingURL=responder.agent.js.map

package/dist/agents/responder.agent.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"responder.agent.js","sourceRoot":"","sources":["../../src/agents/responder.agent.ts"],"names":[],"mappings":";;;AAEA,uDAAwD;AACxD,qDAA6D;AAE7D,wEAAyE;AAKzE,gEAAmG;AAEnG,MAAM,YAAY,GAAG,4BAAkB,CAAC,YAAY,CAAC;IACpD;QACC,QAAQ;QACR;YACC;gBACC,IAAI,EAAE,MAAM;gBACZ,IAAI,EAAE,IAAA,uCAAoB,GAAE;gBAC5B,aAAa,EAAE,EAAE,IAAI,EAAE,WAAW,EAAE;aACpC;SACD;KACD;IACD,CAAC,aAAa,EAAE,YAAY,CAAC;CAC7B,CAAC,CAAC;AA4BH,MAAa,cAAc;IAClB,GAAG,CAAgB;IAE3B,YAAY,MAA4B;QACvC,IAAI,CAAC,GAAG,GAAG,MAAM,CAAC,GAAG,CAAC;IACvB,CAAC;IAKO,mBAAmB,CAAC,OAAyB;QACpD,MAAM,YAAY,GAAa,EAAE,CAAC;QAGlC,IAAI,OAAO,CAAC,eAAe,EAAE,CAAC;YAC7B,YAAY,CAAC,IAAI,CAAC,uCAAuC,OAAO,CAAC,eAAe,EAAE,CAAC,CAAC;QACrF,CAAC;QAGD,MAAM,oBAAoB,GAAG,OAAO,CAAC,eAAe,CAAC,IAAI,CACxD,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,KAAK,kBAAkB,CACrC,CAAC;QACF,IAAI,oBAAoB,EAAE,CAAC;YAC1B,YAAY,CAAC,IAAI,CAAC,yBAAyB,oBAAoB,CAAC,OAAO,EAAE,CAAC,CAAC;QAC5E,CAAC;QAGD,MAAM,UAAU,GAAG,IAAA,gCAAa,EAAC,OAAO,CAAC,eAAe,CAAC,CAAC;QAC1D,IAAI,UAAU,EAAE,CAAC;YAChB,YAAY,CAAC,IAAI,CAChB,uCAAuC,UAAU,CAAC,KAAK,WAAW,UAAU,CAAC,OAAO,EAAE,CACtF,CAAC;YACF,YAAY,CAAC,IAAI,CAChB,mGAAmG,CACnG,CAAC;QACH,CAAC;QAGD,IAAI,OAAO,CAAC,gBAAgB,EAAE,UAAU,CAAC,MAAM,EAAE,CAAC;YACjD,YAAY,CAAC,IAAI,CAChB,wBAAwB,OAAO,CAAC,gBAAgB,CAAC,UAAU,CAAC,MAAM,iBAAiB,CACnF,CAAC;QACH,CAAC;QAGD,MAAM,aAAa,GAAG,IAAA,mCAAgB,EAAC,OAAO,CAAC,eAAe,CAAC,CAAC;QAChE,IAAI,aAAa,EAAE,CAAC;YACnB,YAAY,CAAC,IAAI,CAAC,gBAAgB,aAAa,EAAE,CAAC,CAAC;QACpD,CAAC;aAAM,IAAI,OAAO,CAAC,YAAY,CAAC,KAAK,CAAC,MAAM,EAAE,CAAC;YAC9C,YAAY,CAAC,IAAI,CAAC,iBAAiB,OAAO,CAAC,YAAY,CAAC,KAAK,CAAC,MAAM,gBAAgB,CAAC,CAAC;QACvF,CAAC;QAGD,MAAM,kBAAkB,GAAG,IAAA,wCAAqB,EAAC,OAAO,CAAC,eAAe,CAAC,CAAC;QAC1E,IAAI,kBAAkB,EAAE,CAAC;YACxB,YAAY,CAAC,IAAI,CAAC,uBAAuB,kBAAkB,EAAE,CAAC,CAAC;QAChE,CAAC;QAED,IAAI,YAAY,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC/B,OAAO,IAAI,CAAC;QACb,CAAC;QAED,OAAO,IAAI,uBAAY,CAAC;YACvB,OAAO,EAAE,yDAAyD,YAAY,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE;SAC7F,CAAC,CAAC;IACJ,CAAC;IAKD,KAAK,CAAC,MAAM,CAAC,OAAyB;QACrC,MAAM,KAAK,GAAG,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QAE1C,MAAM,cAAc,GAAG,IAAI,CAAC,mBAAmB,CAAC,OAAO,CAAC,CAAC;QACzD,MAAM,cAAc,GAAG,cAAc;YACpC,CAAC,CAAC,CAAC,GAAG,OAAO,CAAC,QAAQ,EAAE,cAAc,CAAC;YACvC,CAAC,CAAC,OAAO,CAAC,QAAQ,CAAC;QAEpB,OAAO,CAAC,MAAM,KAAK,CAAC,MAAM,CAAC,EAAE,QAAQ,EAAE,cAAc,EAAE,CAAC,CAAc,CAAC;IACxE,CAAC;CACD;AAhFD,wCAgFC"}

package/dist/agents/supervisor.agent.d.ts

@@ -0,0 +1,31 @@
+import type { BaseChatModel } from '@langchain/core/language_models/chat_models';
+import type { BaseMessage } from '@langchain/core/messages';
+import { z } from 'zod';
+import type { CoordinationLogEntry } from '../types/coordination';
+import type { SimpleWorkflow } from '../types/workflow';
+export declare const supervisorRoutingSchema: z.ZodObject<{
+    reasoning: z.ZodString;
+    next: z.ZodEnum<["responder", "discovery", "builder", "configurator"]>;
+}, "strip", z.ZodTypeAny, {
+    reasoning: string;
+    next: "responder" | "discovery" | "builder" | "configurator";
+}, {
+    reasoning: string;
+    next: "responder" | "discovery" | "builder" | "configurator";
+}>;
+export type SupervisorRouting = z.infer<typeof supervisorRoutingSchema>;
+export interface SupervisorAgentConfig {
+    llm: BaseChatModel;
+}
+export interface SupervisorContext {
+    messages: BaseMessage[];
+    workflowJSON: SimpleWorkflow;
+    coordinationLog: CoordinationLogEntry[];
+    previousSummary?: string;
+}
+export declare class SupervisorAgent {
+    private llm;
+    constructor(config: SupervisorAgentConfig);
+    private buildContextMessage;
+    invoke(context: SupervisorContext): Promise<SupervisorRouting>;
+}

package/dist/agents/supervisor.agent.js

@@ -0,0 +1,68 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.SupervisorAgent = exports.supervisorRoutingSchema = void 0;
+const messages_1 = require("@langchain/core/messages");
+const prompts_1 = require("@langchain/core/prompts");
+const zod_1 = require("zod");
+const supervisor_prompt_1 = require("../prompts/agents/supervisor.prompt");
+const context_builders_1 = require("../utils/context-builders");
+const coordination_log_1 = require("../utils/coordination-log");
+const systemPrompt = prompts_1.ChatPromptTemplate.fromMessages([
+    [
+        'system',
+        [
+            {
+                type: 'text',
+                text: (0, supervisor_prompt_1.buildSupervisorPrompt)(),
+                cache_control: { type: 'ephemeral' },
+            },
+        ],
+    ],
+    ['placeholder', '{messages}'],
+]);
+exports.supervisorRoutingSchema = zod_1.z.object({
+    reasoning: zod_1.z.string().describe('One sentence explaining why this agent should act next'),
+    next: zod_1.z
+        .enum(['responder', 'discovery', 'builder', 'configurator'])
+        .describe('The next agent to call'),
+});
+class SupervisorAgent {
+    llm;
+    constructor(config) {
+        this.llm = config.llm;
+    }
+    buildContextMessage(context) {
+        const contextParts = [];
+        if (context.previousSummary) {
+            contextParts.push('<previous_conversation_summary>');
+            contextParts.push(context.previousSummary);
+            contextParts.push('</previous_conversation_summary>');
+        }
+        if (context.workflowJSON.nodes.length > 0) {
+            contextParts.push('<workflow_summary>');
+            contextParts.push((0, context_builders_1.buildWorkflowSummary)(context.workflowJSON));
+            contextParts.push('</workflow_summary>');
+        }
+        if (context.coordinationLog.length > 0) {
+            contextParts.push('<completed_phases>');
+            contextParts.push((0, coordination_log_1.summarizeCoordinationLog)(context.coordinationLog));
+            contextParts.push('</completed_phases>');
+        }
+        if (contextParts.length === 0) {
+            return null;
+        }
+        return new messages_1.HumanMessage({ content: contextParts.join('\n\n') });
+    }
+    async invoke(context) {
+        const agent = systemPrompt.pipe(this.llm.withStructuredOutput(exports.supervisorRoutingSchema, {
+            name: 'routing_decision',
+        }));
+        const contextMessage = this.buildContextMessage(context);
+        const messagesToSend = contextMessage
+            ? [...context.messages, contextMessage]
+            : context.messages;
+        return await agent.invoke({ messages: messagesToSend });
+    }
+}
+exports.SupervisorAgent = SupervisorAgent;
+//# sourceMappingURL=supervisor.agent.js.map

package/dist/agents/supervisor.agent.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"supervisor.agent.js","sourceRoot":"","sources":["../../src/agents/supervisor.agent.ts"],"names":[],"mappings":";;;AAEA,uDAAwD;AACxD,qDAA6D;AAC7D,6BAAwB;AAExB,0EAA2E;AAI3E,gEAAiE;AACjE,gEAAqE;AAErE,MAAM,YAAY,GAAG,4BAAkB,CAAC,YAAY,CAAC;IACpD;QACC,QAAQ;QACR;YACC;gBACC,IAAI,EAAE,MAAM;gBACZ,IAAI,EAAE,IAAA,yCAAqB,GAAE;gBAC7B,aAAa,EAAE,EAAE,IAAI,EAAE,WAAW,EAAE;aACpC;SACD;KACD;IACD,CAAC,aAAa,EAAE,YAAY,CAAC;CAC7B,CAAC,CAAC;AAKU,QAAA,uBAAuB,GAAG,OAAC,CAAC,MAAM,CAAC;IAC/C,SAAS,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,wDAAwD,CAAC;IACxF,IAAI,EAAE,OAAC;SACL,IAAI,CAAC,CAAC,WAAW,EAAE,WAAW,EAAE,SAAS,EAAE,cAAc,CAAC,CAAC;SAC3D,QAAQ,CAAC,wBAAwB,CAAC;CACpC,CAAC,CAAC;AA4BH,MAAa,eAAe;IACnB,GAAG,CAAgB;IAE3B,YAAY,MAA6B;QACxC,IAAI,CAAC,GAAG,GAAG,MAAM,CAAC,GAAG,CAAC;IACvB,CAAC;IAKO,mBAAmB,CAAC,OAA0B;QACrD,MAAM,YAAY,GAAa,EAAE,CAAC;QAGlC,IAAI,OAAO,CAAC,eAAe,EAAE,CAAC;YAC7B,YAAY,CAAC,IAAI,CAAC,iCAAiC,CAAC,CAAC;YACrD,YAAY,CAAC,IAAI,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC;YAC3C,YAAY,CAAC,IAAI,CAAC,kCAAkC,CAAC,CAAC;QACvD,CAAC;QAGD,IAAI,OAAO,CAAC,YAAY,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC3C,YAAY,CAAC,IAAI,CAAC,oBAAoB,CAAC,CAAC;YACxC,YAAY,CAAC,IAAI,CAAC,IAAA,uCAAoB,EAAC,OAAO,CAAC,YAAY,CAAC,CAAC,CAAC;YAC9D,YAAY,CAAC,IAAI,CAAC,qBAAqB,CAAC,CAAC;QAC1C,CAAC;QAGD,IAAI,OAAO,CAAC,eAAe,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACxC,YAAY,CAAC,IAAI,CAAC,oBAAoB,CAAC,CAAC;YACxC,YAAY,CAAC,IAAI,CAAC,IAAA,2CAAwB,EAAC,OAAO,CAAC,eAAe,CAAC,CAAC,CAAC;YACrE,YAAY,CAAC,IAAI,CAAC,qBAAqB,CAAC,CAAC;QAC1C,CAAC;QAED,IAAI,YAAY,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC/B,OAAO,IAAI,CAAC;QACb,CAAC;QAED,OAAO,IAAI,uBAAY,CAAC,EAAE,OAAO,EAAE,YAAY,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;IACjE,CAAC;IAKD,KAAK,CAAC,MAAM,CAAC,OAA0B;QACtC,MAAM,KAAK,GAAG,YAAY,CAAC,IAAI,CAC9B,IAAI,CAAC,GAAG,CAAC,oBAAoB,CAAC,+BAAuB,EAAE;YACtD,IAAI,EAAE,kBAAkB;SACxB,CAAC,CACF,CAAC;QAEF,MAAM,cAAc,GAAG,IAAI,CAAC,mBAAmB,CAAC,OAAO,CAAC,CAAC;QACzD,MAAM,cAAc,GAAG,cAAc;YACpC,CAAC,CAAC,CAAC,GAAG,OAAO,CAAC,QAAQ,EAAE,cAAc,CAAC;YACvC,CAAC,CAAC,OAAO,CAAC,QAAQ,CAAC;QAEpB,OAAO,MAAM,KAAK,CAAC,MAAM,CAAC,EAAE,QAAQ,EAAE,cAAc,EAAE,CAAC,CAAC;IACzD,CAAC;CACD;AA1DD,0CA0DC"}

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

@@ -0,0 +1,32 @@
+import { Logger } from '@n8n/backend-common';
+import { AiAssistantClient, AiAssistantSDK } from '@n8n_io/ai-assistant-sdk';
+import type { IUser, INodeTypeDescription, ITelemetryTrackProperties } from 'n8n-workflow';
+import { type ChatPayload } from './workflow-builder-agent';
+type OnCreditsUpdated = (userId: string, creditsQuota: number, creditsClaimed: number) => void;
+type OnTelemetryEvent = (event: string, properties: ITelemetryTrackProperties) => void;
+export declare class AiWorkflowBuilderService {
+    private readonly client?;
+    private readonly logger?;
+    private readonly instanceId?;
+    private readonly instanceUrl?;
+    private readonly n8nVersion?;
+    private readonly onCreditsUpdated?;
+    private readonly onTelemetryEvent?;
+    private readonly parsedNodeTypes;
+    private sessionManager;
+    constructor(parsedNodeTypes: INodeTypeDescription[], client?: AiAssistantClient | undefined, logger?: Logger | undefined, instanceId?: string | undefined, instanceUrl?: string | undefined, n8nVersion?: string | undefined, onCreditsUpdated?: OnCreditsUpdated | undefined, onTelemetryEvent?: OnTelemetryEvent | undefined);
+    private static getAnthropicClaudeModel;
+    private getApiProxyAuthHeaders;
+    private setupModels;
+    private filterNodeTypes;
+    private getAgent;
+    private onGenerationSuccess;
+    chat(payload: ChatPayload, user: IUser, abortSignal?: AbortSignal): AsyncGenerator<import("./types").StreamOutput, void, unknown>;
+    private trackBuilderReplyTelemetry;
+    getSessions(workflowId: string | undefined, user?: IUser): Promise<{
+        sessions: import("./types/sessions").Session[];
+    }>;
+    getBuilderInstanceCredits(user: IUser): Promise<AiAssistantSDK.BuilderInstanceCreditsResponse>;
+    truncateMessagesAfter(workflowId: string, user: IUser, messageId: string): Promise<boolean>;
+}
+export {};

package/dist/ai-workflow-builder-agent.service.js

@@ -0,0 +1,236 @@
+"use strict";
+var __decorate = (this && this.__decorate) || function (decorators, target, key, desc) {
+    var c = arguments.length, r = c < 3 ? target : desc === null ? desc = Object.getOwnPropertyDescriptor(target, key) : desc, d;
+    if (typeof Reflect === "object" && typeof Reflect.decorate === "function") r = Reflect.decorate(decorators, target, key, desc);
+    else for (var i = decorators.length - 1; i >= 0; i--) if (d = decorators[i]) r = (c < 3 ? d(r) : c > 3 ? d(target, key, r) : d(target, key)) || r;
+    return c > 3 && r && Object.defineProperty(target, key, r), r;
+};
+var __metadata = (this && this.__metadata) || function (k, v) {
+    if (typeof Reflect === "object" && typeof Reflect.metadata === "function") return Reflect.metadata(k, v);
+};
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+var AiWorkflowBuilderService_1;
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.AiWorkflowBuilderService = void 0;
+const messages_1 = require("@langchain/core/messages");
+const tracer_langchain_1 = require("@langchain/core/tracers/tracer_langchain");
+const backend_common_1 = require("@n8n/backend-common");
+const di_1 = require("@n8n/di");
+const ai_assistant_sdk_1 = require("@n8n_io/ai-assistant-sdk");
+const assert_1 = __importDefault(require("assert"));
+const langsmith_1 = require("langsmith");
+const errors_1 = require("./errors");
+const llm_config_1 = require("./llm-config");
+const session_manager_service_1 = require("./session-manager.service");
+const workflow_builder_agent_1 = require("./workflow-builder-agent");
+let AiWorkflowBuilderService = AiWorkflowBuilderService_1 = class AiWorkflowBuilderService {
+    client;
+    logger;
+    instanceId;
+    instanceUrl;
+    n8nVersion;
+    onCreditsUpdated;
+    onTelemetryEvent;
+    parsedNodeTypes;
+    sessionManager;
+    constructor(parsedNodeTypes, client, logger, instanceId, instanceUrl, n8nVersion, onCreditsUpdated, onTelemetryEvent) {
+        this.client = client;
+        this.logger = logger;
+        this.instanceId = instanceId;
+        this.instanceUrl = instanceUrl;
+        this.n8nVersion = n8nVersion;
+        this.onCreditsUpdated = onCreditsUpdated;
+        this.onTelemetryEvent = onTelemetryEvent;
+        this.parsedNodeTypes = this.filterNodeTypes(parsedNodeTypes);
+        this.sessionManager = new session_manager_service_1.SessionManagerService(this.parsedNodeTypes, logger);
+    }
+    static async getAnthropicClaudeModel({ baseUrl, authHeaders = {}, apiKey = '-', } = {}) {
+        return await (0, llm_config_1.anthropicClaudeSonnet45)({
+            baseUrl,
+            apiKey,
+            headers: {
+                ...authHeaders,
+                'anthropic-beta': 'prompt-caching-2024-07-31',
+            },
+        });
+    }
+    async getApiProxyAuthHeaders(user, userMessageId) {
+        (0, assert_1.default)(this.client);
+        const authResponse = await this.client.getBuilderApiProxyToken(user, { userMessageId });
+        const authHeaders = {
+            Authorization: `${authResponse.tokenType} ${authResponse.accessToken}`,
+        };
+        return authHeaders;
+    }
+    async setupModels(user, userMessageId) {
+        try {
+            if (this.client) {
+                const authHeaders = await this.getApiProxyAuthHeaders(user, userMessageId);
+                const baseUrl = this.client.getApiProxyBaseUrl();
+                const anthropicClaude = await AiWorkflowBuilderService_1.getAnthropicClaudeModel({
+                    baseUrl: baseUrl + '/anthropic',
+                    authHeaders,
+                });
+                const tracingClient = new langsmith_1.Client({
+                    apiKey: '-',
+                    apiUrl: baseUrl + '/langsmith',
+                    autoBatchTracing: false,
+                    traceBatchConcurrency: 1,
+                    fetchOptions: {
+                        headers: {
+                            ...authHeaders,
+                        },
+                    },
+                });
+                return { tracingClient, anthropicClaude, authHeaders };
+            }
+            const anthropicClaude = await AiWorkflowBuilderService_1.getAnthropicClaudeModel({
+                apiKey: process.env.N8N_AI_ANTHROPIC_KEY ?? '',
+            });
+            return { anthropicClaude };
+        }
+        catch (error) {
+            const errorMessage = error instanceof Error ? `: ${error.message}` : '';
+            const llmError = new errors_1.LLMServiceError(`Failed to connect to LLM Provider${errorMessage}`, {
+                cause: error,
+                tags: {
+                    hasClient: !!this.client,
+                    hasUser: !!user,
+                },
+            });
+            throw llmError;
+        }
+    }
+    filterNodeTypes(nodeTypes) {
+        const ignoredTypes = new Set([
+            '@n8n/n8n-nodes-langchain.toolVectorStore',
+            '@n8n/n8n-nodes-langchain.documentGithubLoader',
+            '@n8n/n8n-nodes-langchain.code',
+        ]);
+        const visibleNodeTypes = nodeTypes.filter((nodeType) => !ignoredTypes.has(nodeType.name) &&
+            (nodeType.hidden !== true || nodeType.name === 'n8n-nodes-base.dataTable'));
+        return visibleNodeTypes.map((nodeType) => {
+            const isTool = nodeType.name.endsWith('Tool');
+            if (!isTool)
+                return nodeType;
+            const nonToolNode = nodeTypes.find((nt) => nt.name === nodeType.name.replace('Tool', ''));
+            if (!nonToolNode)
+                return nodeType;
+            return {
+                ...nonToolNode,
+                ...nodeType,
+            };
+        });
+    }
+    async getAgent(user, userMessageId, featureFlags) {
+        const { anthropicClaude, tracingClient, authHeaders } = await this.setupModels(user, userMessageId);
+        const agent = new workflow_builder_agent_1.WorkflowBuilderAgent({
+            parsedNodeTypes: this.parsedNodeTypes,
+            llmSimpleTask: anthropicClaude,
+            llmComplexTask: anthropicClaude,
+            logger: this.logger,
+            checkpointer: this.sessionManager.getCheckpointer(),
+            tracer: tracingClient
+                ? new tracer_langchain_1.LangChainTracer({ client: tracingClient, projectName: 'n8n-workflow-builder' })
+                : undefined,
+            instanceUrl: this.instanceUrl,
+            onGenerationSuccess: async () => {
+                await this.onGenerationSuccess(user, authHeaders);
+            },
+            runMetadata: {
+                n8nVersion: this.n8nVersion,
+                featureFlags: featureFlags ?? {},
+            },
+        });
+        return agent;
+    }
+    async onGenerationSuccess(user, authHeaders) {
+        try {
+            if (this.client) {
+                (0, assert_1.default)(authHeaders, 'Auth headers must be set when AI Assistant Service client is used');
+                (0, assert_1.default)(user);
+                const creditsInfo = await this.client.markBuilderSuccess(user, authHeaders);
+                if (this.onCreditsUpdated && user.id && creditsInfo) {
+                    this.onCreditsUpdated(user.id, creditsInfo.creditsQuota, creditsInfo.creditsClaimed);
+                }
+            }
+        }
+        catch (error) {
+            if (error instanceof Error) {
+                this.logger?.error(`Unable to mark generation success ${error.message}`, { error });
+            }
+        }
+    }
+    async *chat(payload, user, abortSignal) {
+        const agent = await this.getAgent(user, payload.id, payload.featureFlags);
+        const userId = user?.id?.toString();
+        const workflowId = payload.workflowContext?.currentWorkflow?.id;
+        for await (const output of agent.chat(payload, userId, abortSignal)) {
+            yield output;
+        }
+        if (this.onTelemetryEvent && userId) {
+            try {
+                await this.trackBuilderReplyTelemetry(agent, workflowId, userId, payload.id);
+            }
+            catch (error) {
+                this.logger?.error('Failed to track builder reply telemetry', { error });
+            }
+        }
+    }
+    async trackBuilderReplyTelemetry(agent, workflowId, userId, userMessageId) {
+        if (!this.onTelemetryEvent)
+            return;
+        const state = await agent.getState(workflowId, userId);
+        const threadId = session_manager_service_1.SessionManagerService.generateThreadId(workflowId, userId);
+        const lastAiMessage = state.values.messages.findLast((m) => m instanceof messages_1.AIMessage);
+        const messageAi = typeof lastAiMessage?.content === 'string'
+            ? lastAiMessage.content
+            : JSON.stringify(lastAiMessage?.content ?? '');
+        const toolMessages = state.values.messages.filter((m) => m instanceof messages_1.ToolMessage);
+        const toolsCalled = [
+            ...new Set(toolMessages
+                .map((m) => m.name)
+                .filter((name) => name !== undefined)),
+        ];
+        const properties = {
+            user_id: userId,
+            instance_id: this.instanceId,
+            workflow_id: workflowId,
+            sequence_id: threadId,
+            message_ai: messageAi,
+            tools_called: toolsCalled,
+            techniques_categories: state.values.techniqueCategories,
+            validations: state.values.validationHistory,
+            ...(state.values.templateIds.length > 0 && {
+                templates_selected: state.values.templateIds,
+            }),
+            user_message_id: userMessageId,
+        };
+        this.onTelemetryEvent('Builder replied to user message', properties);
+    }
+    async getSessions(workflowId, user) {
+        const userId = user?.id?.toString();
+        return await this.sessionManager.getSessions(workflowId, userId);
+    }
+    async getBuilderInstanceCredits(user) {
+        if (this.client) {
+            return await this.client.getBuilderInstanceCredits(user);
+        }
+        return {
+            creditsQuota: -1,
+            creditsClaimed: 0,
+        };
+    }
+    async truncateMessagesAfter(workflowId, user, messageId) {
+        return await this.sessionManager.truncateMessagesAfter(workflowId, user.id, messageId);
+    }
+};
+exports.AiWorkflowBuilderService = AiWorkflowBuilderService;
+exports.AiWorkflowBuilderService = AiWorkflowBuilderService = AiWorkflowBuilderService_1 = __decorate([
+    (0, di_1.Service)(),
+    __metadata("design:paramtypes", [Array, ai_assistant_sdk_1.AiAssistantClient,
+        backend_common_1.Logger, String, String, String, Function, Function])
+], AiWorkflowBuilderService);
+//# sourceMappingURL=ai-workflow-builder-agent.service.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"ai-workflow-builder-agent.service.js","sourceRoot":"","sources":["../src/ai-workflow-builder-agent.service.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;AACA,uDAAkE;AAElE,+EAA2E;AAC3E,wDAA6C;AAC7C,gCAAkC;AAClC,+DAA6E;AAC7E,oDAA4B;AAC5B,yCAAoD;AAGpD,qCAA2C;AAC3C,6CAAuD;AACvD,uEAAkE;AAClE,qEAIkC;AAO3B,IAAM,wBAAwB,gCAA9B,MAAM,wBAAwB;IAMlB;IACA;IACA;IACA;IACA;IACA;IACA;IAXD,eAAe,CAAyB;IACjD,cAAc,CAAwB;IAE9C,YACC,eAAuC,EACtB,MAA0B,EAC1B,MAAe,EACf,UAAmB,EACnB,WAAoB,EACpB,UAAmB,EACnB,gBAAmC,EACnC,gBAAmC;QANnC,WAAM,GAAN,MAAM,CAAoB;QAC1B,WAAM,GAAN,MAAM,CAAS;QACf,eAAU,GAAV,UAAU,CAAS;QACnB,gBAAW,GAAX,WAAW,CAAS;QACpB,eAAU,GAAV,UAAU,CAAS;QACnB,qBAAgB,GAAhB,gBAAgB,CAAmB;QACnC,qBAAgB,GAAhB,gBAAgB,CAAmB;QAEpD,IAAI,CAAC,eAAe,GAAG,IAAI,CAAC,eAAe,CAAC,eAAe,CAAC,CAAC;QAC7D,IAAI,CAAC,cAAc,GAAG,IAAI,+CAAqB,CAAC,IAAI,CAAC,eAAe,EAAE,MAAM,CAAC,CAAC;IAC/E,CAAC;IAEO,MAAM,CAAC,KAAK,CAAC,uBAAuB,CAAC,EAC5C,OAAO,EACP,WAAW,GAAG,EAAE,EAChB,MAAM,GAAG,GAAG,MAKT,EAAE;QACL,OAAO,MAAM,IAAA,oCAAuB,EAAC;YACpC,OAAO;YACP,MAAM;YACN,OAAO,EAAE;gBACR,GAAG,WAAW;gBAEd,gBAAgB,EAAE,2BAA2B;aAC7C;SACD,CAAC,CAAC;IACJ,CAAC;IAEO,KAAK,CAAC,sBAAsB,CAAC,IAAW,EAAE,aAAqB;QACtE,IAAA,gBAAM,EAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAEpB,MAAM,YAAY,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,uBAAuB,CAAC,IAAI,EAAE,EAAE,aAAa,EAAE,CAAC,CAAC;QACxF,MAAM,WAAW,GAAG;YAEnB,aAAa,EAAE,GAAG,YAAY,CAAC,SAAS,IAAI,YAAY,CAAC,WAAW,EAAE;SACtE,CAAC;QAEF,OAAO,WAAW,CAAC;IACpB,CAAC;IAEO,KAAK,CAAC,WAAW,CACxB,IAAW,EACX,aAAqB;QAOrB,IAAI,CAAC;YAEJ,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;gBACjB,MAAM,WAAW,GAAG,MAAM,IAAI,CAAC,sBAAsB,CAAC,IAAI,EAAE,aAAa,CAAC,CAAC;gBAG3E,MAAM,OAAO,GAAG,IAAI,CAAC,MAAM,CAAC,kBAAkB,EAAE,CAAC;gBAEjD,MAAM,eAAe,GAAG,MAAM,0BAAwB,CAAC,uBAAuB,CAAC;oBAC9E,OAAO,EAAE,OAAO,GAAG,YAAY;oBAC/B,WAAW;iBACX,CAAC,CAAC;gBAEH,MAAM,aAAa,GAAG,IAAI,kBAAa,CAAC;oBACvC,MAAM,EAAE,GAAG;oBACX,MAAM,EAAE,OAAO,GAAG,YAAY;oBAC9B,gBAAgB,EAAE,KAAK;oBACvB,qBAAqB,EAAE,CAAC;oBACxB,YAAY,EAAE;wBACb,OAAO,EAAE;4BACR,GAAG,WAAW;yBACd;qBACD;iBACD,CAAC,CAAC;gBAEH,OAAO,EAAE,aAAa,EAAE,eAAe,EAAE,WAAW,EAAE,CAAC;YACxD,CAAC;YAGD,MAAM,eAAe,GAAG,MAAM,0BAAwB,CAAC,uBAAuB,CAAC;gBAC9E,MAAM,EAAE,OAAO,CAAC,GAAG,CAAC,oBAAoB,IAAI,EAAE;aAC9C,CAAC,CAAC;YAEH,OAAO,EAAE,eAAe,EAAE,CAAC;QAC5B,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YAChB,MAAM,YAAY,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YACxE,MAAM,QAAQ,GAAG,IAAI,wBAAe,CAAC,oCAAoC,YAAY,EAAE,EAAE;gBACxF,KAAK,EAAE,KAAK;gBACZ,IAAI,EAAE;oBACL,SAAS,EAAE,CAAC,CAAC,IAAI,CAAC,MAAM;oBACxB,OAAO,EAAE,CAAC,CAAC,IAAI;iBACf;aACD,CAAC,CAAC;YACH,MAAM,QAAQ,CAAC;QAChB,CAAC;IACF,CAAC;IAEO,eAAe,CAAC,SAAiC;QAExD,MAAM,YAAY,GAAG,IAAI,GAAG,CAAC;YAC5B,0CAA0C;YAC1C,+CAA+C;YAC/C,+BAA+B;SAC/B,CAAC,CAAC;QAEH,MAAM,gBAAgB,GAAG,SAAS,CAAC,MAAM,CACxC,CAAC,QAAQ,EAAE,EAAE,CAGZ,CAAC,YAAY,CAAC,GAAG,CAAC,QAAQ,CAAC,IAAI,CAAC;YAChC,CAAC,QAAQ,CAAC,MAAM,KAAK,IAAI,IAAI,QAAQ,CAAC,IAAI,KAAK,0BAA0B,CAAC,CAC3E,CAAC;QAEF,OAAO,gBAAgB,CAAC,GAAG,CAAC,CAAC,QAAQ,EAAE,EAAE;YAGxC,MAAM,MAAM,GAAG,QAAQ,CAAC,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC;YAC9C,IAAI,CAAC,MAAM;gBAAE,OAAO,QAAQ,CAAC;YAE7B,MAAM,WAAW,GAAG,SAAS,CAAC,IAAI,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,IAAI,KAAK,QAAQ,CAAC,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC,CAAC;YAC1F,IAAI,CAAC,WAAW;gBAAE,OAAO,QAAQ,CAAC;YAElC,OAAO;gBACN,GAAG,WAAW;gBACd,GAAG,QAAQ;aACX,CAAC;QACH,CAAC,CAAC,CAAC;IACJ,CAAC;IAEO,KAAK,CAAC,QAAQ,CAAC,IAAW,EAAE,aAAqB,EAAE,YAAkC;QAC5F,MAAM,EAAE,eAAe,EAAE,aAAa,EAAE,WAAW,EAAE,GAAG,MAAM,IAAI,CAAC,WAAW,CAC7E,IAAI,EACJ,aAAa,CACb,CAAC;QAEF,MAAM,KAAK,GAAG,IAAI,6CAAoB,CAAC;YACtC,eAAe,EAAE,IAAI,CAAC,eAAe;YAErC,aAAa,EAAE,eAAe;YAC9B,cAAc,EAAE,eAAe;YAC/B,MAAM,EAAE,IAAI,CAAC,MAAM;YACnB,YAAY,EAAE,IAAI,CAAC,cAAc,CAAC,eAAe,EAAE;YACnD,MAAM,EAAE,aAAa;gBACpB,CAAC,CAAC,IAAI,kCAAe,CAAC,EAAE,MAAM,EAAE,aAAa,EAAE,WAAW,EAAE,sBAAsB,EAAE,CAAC;gBACrF,CAAC,CAAC,SAAS;YACZ,WAAW,EAAE,IAAI,CAAC,WAAW;YAC7B,mBAAmB,EAAE,KAAK,IAAI,EAAE;gBAC/B,MAAM,IAAI,CAAC,mBAAmB,CAAC,IAAI,EAAE,WAAW,CAAC,CAAC;YACnD,CAAC;YACD,WAAW,EAAE;gBACZ,UAAU,EAAE,IAAI,CAAC,UAAU;gBAC3B,YAAY,EAAE,YAAY,IAAI,EAAE;aAChC;SACD,CAAC,CAAC;QAEH,OAAO,KAAK,CAAC;IACd,CAAC;IAEO,KAAK,CAAC,mBAAmB,CAChC,IAAY,EAEZ,WAAuC;QAEvC,IAAI,CAAC;YACJ,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;gBACjB,IAAA,gBAAM,EAAC,WAAW,EAAE,mEAAmE,CAAC,CAAC;gBACzF,IAAA,gBAAM,EAAC,IAAI,CAAC,CAAC;gBACb,MAAM,WAAW,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,kBAAkB,CAAC,IAAI,EAAE,WAAW,CAAC,CAAC;gBAG5E,IAAI,IAAI,CAAC,gBAAgB,IAAI,IAAI,CAAC,EAAE,IAAI,WAAW,EAAE,CAAC;oBACrD,IAAI,CAAC,gBAAgB,CAAC,IAAI,CAAC,EAAE,EAAE,WAAW,CAAC,YAAY,EAAE,WAAW,CAAC,cAAc,CAAC,CAAC;gBACtF,CAAC;YACF,CAAC;QACF,CAAC;QAAC,OAAO,KAAc,EAAE,CAAC;YACzB,IAAI,KAAK,YAAY,KAAK,EAAE,CAAC;gBAC5B,IAAI,CAAC,MAAM,EAAE,KAAK,CAAC,qCAAqC,KAAK,CAAC,OAAO,EAAE,EAAE,EAAE,KAAK,EAAE,CAAC,CAAC;YACrF,CAAC;QACF,CAAC;IACF,CAAC;IAED,KAAK,CAAC,CAAC,IAAI,CAAC,OAAoB,EAAE,IAAW,EAAE,WAAyB;QACvE,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,IAAI,EAAE,OAAO,CAAC,EAAE,EAAE,OAAO,CAAC,YAAY,CAAC,CAAC;QAC1E,MAAM,MAAM,GAAG,IAAI,EAAE,EAAE,EAAE,QAAQ,EAAE,CAAC;QACpC,MAAM,UAAU,GAAG,OAAO,CAAC,eAAe,EAAE,eAAe,EAAE,EAAE,CAAC;QAEhE,IAAI,KAAK,EAAE,MAAM,MAAM,IAAI,KAAK,CAAC,IAAI,CAAC,OAAO,EAAE,MAAM,EAAE,WAAW,CAAC,EAAE,CAAC;YACrE,MAAM,MAAM,CAAC;QACd,CAAC;QAGD,IAAI,IAAI,CAAC,gBAAgB,IAAI,MAAM,EAAE,CAAC;YACrC,IAAI,CAAC;gBACJ,MAAM,IAAI,CAAC,0BAA0B,CAAC,KAAK,EAAE,UAAU,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,CAAC,CAAC;YAC9E,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBAChB,IAAI,CAAC,MAAM,EAAE,KAAK,CAAC,yCAAyC,EAAE,EAAE,KAAK,EAAE,CAAC,CAAC;YAC1E,CAAC;QACF,CAAC;IACF,CAAC;IAEO,KAAK,CAAC,0BAA0B,CACvC,KAA2B,EAC3B,UAA8B,EAC9B,MAAc,EACd,aAAqB;QAErB,IAAI,CAAC,IAAI,CAAC,gBAAgB;YAAE,OAAO;QAEnC,MAAM,KAAK,GAAG,MAAM,KAAK,CAAC,QAAQ,CAAC,UAAU,EAAE,MAAM,CAAC,CAAC;QACvD,MAAM,QAAQ,GAAG,+CAAqB,CAAC,gBAAgB,CAAC,UAAU,EAAE,MAAM,CAAC,CAAC;QAG5E,MAAM,aAAa,GAAG,KAAK,CAAC,MAAM,CAAC,QAAQ,CAAC,QAAQ,CACnD,CAAC,CAAc,EAAkB,EAAE,CAAC,CAAC,YAAY,oBAAS,CAC1D,CAAC;QACF,MAAM,SAAS,GACd,OAAO,aAAa,EAAE,OAAO,KAAK,QAAQ;YACzC,CAAC,CAAC,aAAa,CAAC,OAAO;YACvB,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,aAAa,EAAE,OAAO,IAAI,EAAE,CAAC,CAAC;QAEjD,MAAM,YAAY,GAAG,KAAK,CAAC,MAAM,CAAC,QAAQ,CAAC,MAAM,CAChD,CAAC,CAAc,EAAoB,EAAE,CAAC,CAAC,YAAY,sBAAW,CAC9D,CAAC;QACF,MAAM,WAAW,GAAG;YACnB,GAAG,IAAI,GAAG,CACT,YAAY;iBACV,GAAG,CAAC,CAAC,CAAc,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC;iBAC/B,MAAM,CAAC,CAAC,IAAwB,EAAkB,EAAE,CAAC,IAAI,KAAK,SAAS,CAAC,CAC1E;SACD,CAAC;QAGF,MAAM,UAAU,GAA8B;YAC7C,OAAO,EAAE,MAAM;YACf,WAAW,EAAE,IAAI,CAAC,UAAU;YAC5B,WAAW,EAAE,UAAU;YACvB,WAAW,EAAE,QAAQ;YACrB,UAAU,EAAE,SAAS;YACrB,YAAY,EAAE,WAAW;YACzB,qBAAqB,EAAE,KAAK,CAAC,MAAM,CAAC,mBAAmB;YACvD,WAAW,EAAE,KAAK,CAAC,MAAM,CAAC,iBAAiB;YAE3C,GAAG,CAAC,KAAK,CAAC,MAAM,CAAC,WAAW,CAAC,MAAM,GAAG,CAAC,IAAI;gBAC1C,kBAAkB,EAAE,KAAK,CAAC,MAAM,CAAC,WAAW;aAC5C,CAAC;YACF,eAAe,EAAE,aAAa;SAC9B,CAAC;QAEF,IAAI,CAAC,gBAAgB,CAAC,iCAAiC,EAAE,UAAU,CAAC,CAAC;IACtE,CAAC;IAED,KAAK,CAAC,WAAW,CAAC,UAA8B,EAAE,IAAY;QAC7D,MAAM,MAAM,GAAG,IAAI,EAAE,EAAE,EAAE,QAAQ,EAAE,CAAC;QACpC,OAAO,MAAM,IAAI,CAAC,cAAc,CAAC,WAAW,CAAC,UAAU,EAAE,MAAM,CAAC,CAAC;IAClE,CAAC;IAED,KAAK,CAAC,yBAAyB,CAC9B,IAAW;QAEX,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;YACjB,OAAO,MAAM,IAAI,CAAC,MAAM,CAAC,yBAAyB,CAAC,IAAI,CAAC,CAAC;QAC1D,CAAC;QAGD,OAAO;YACN,YAAY,EAAE,CAAC,CAAC;YAChB,cAAc,EAAE,CAAC;SACjB,CAAC;IACH,CAAC;IAMD,KAAK,CAAC,qBAAqB,CAC1B,UAAkB,EAClB,IAAW,EACX,SAAiB;QAEjB,OAAO,MAAM,IAAI,CAAC,cAAc,CAAC,qBAAqB,CAAC,UAAU,EAAE,IAAI,CAAC,EAAE,EAAE,SAAS,CAAC,CAAC;IACxF,CAAC;CACD,CAAA;AAlSY,4DAAwB;mCAAxB,wBAAwB;IADpC,IAAA,YAAO,GAAE;4CAOkB,oCAAiB;QACjB,uBAAM;GAPrB,wBAAwB,CAkSpC"}