@n8n/ai-workflow-builder 0.12.1 → 0.14.0

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

@@ -0,0 +1,28 @@
+import { Logger } from '@n8n/backend-common';
+import { AiAssistantClient } from '@n8n_io/ai-assistant-sdk';
+import { INodeTypes } from 'n8n-workflow';
+import type { IUser } from 'n8n-workflow';
+import { type ChatPayload } from './workflow-builder-agent';
+export declare class AiWorkflowBuilderService {
+    private readonly nodeTypes;
+    private readonly client?;
+    private readonly logger?;
+    private parsedNodeTypes;
+    private llmSimpleTask;
+    private llmComplexTask;
+    private tracingClient;
+    private checkpointer;
+    private agent;
+    constructor(nodeTypes: INodeTypes, client?: AiAssistantClient | undefined, logger?: Logger | undefined);
+    private setupModels;
+    private getNodeTypes;
+    private getAgent;
+    chat(payload: ChatPayload, user?: IUser): AsyncGenerator<import("./types").StreamOutput, void, unknown>;
+    getSessions(workflowId: string | undefined, user?: IUser): Promise<{
+        sessions: {
+            sessionId: string;
+            messages: Record<string, unknown>[];
+            lastUpdated: string;
+        }[];
+    }>;
+}

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

@@ -0,0 +1,160 @@
+"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);
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.AiWorkflowBuilderService = void 0;
+const tracer_langchain_1 = require("@langchain/core/tracers/tracer_langchain");
+const langgraph_1 = require("@langchain/langgraph");
+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 langsmith_1 = require("langsmith");
+const errors_1 = require("./errors");
+const llm_config_1 = require("./llm-config");
+const workflow_builder_agent_1 = require("./workflow-builder-agent");
+let AiWorkflowBuilderService = class AiWorkflowBuilderService {
+    constructor(nodeTypes, client, logger) {
+        this.nodeTypes = nodeTypes;
+        this.client = client;
+        this.logger = logger;
+        this.parsedNodeTypes = [];
+        this.checkpointer = new langgraph_1.MemorySaver();
+        this.parsedNodeTypes = this.getNodeTypes();
+    }
+    async setupModels(user) {
+        try {
+            if (this.llmSimpleTask && this.llmComplexTask) {
+                return;
+            }
+            if (this.client && user) {
+                const authHeaders = await this.client.generateApiProxyCredentials(user);
+                const baseUrl = this.client.getApiProxyBaseUrl();
+                this.llmSimpleTask = await (0, llm_config_1.gpt41mini)({
+                    baseUrl: baseUrl + '/openai',
+                    apiKey: '-',
+                    headers: {
+                        Authorization: authHeaders.apiKey,
+                    },
+                });
+                this.llmComplexTask = await (0, llm_config_1.anthropicClaudeSonnet4)({
+                    baseUrl: baseUrl + '/anthropic',
+                    apiKey: '-',
+                    headers: {
+                        Authorization: authHeaders.apiKey,
+                        'anthropic-beta': 'prompt-caching-2024-07-31',
+                    },
+                });
+                this.tracingClient = new langsmith_1.Client({
+                    apiKey: '-',
+                    apiUrl: baseUrl + '/langsmith',
+                    autoBatchTracing: false,
+                    traceBatchConcurrency: 1,
+                    fetchOptions: {
+                        headers: {
+                            Authorization: authHeaders.apiKey,
+                        },
+                    },
+                });
+                return;
+            }
+            this.llmSimpleTask = await (0, llm_config_1.gpt41mini)({
+                apiKey: process.env.N8N_AI_OPENAI_API_KEY ?? '',
+            });
+            this.llmComplexTask = await (0, llm_config_1.anthropicClaudeSonnet4)({
+                apiKey: process.env.N8N_AI_ANTHROPIC_KEY ?? '',
+                headers: {
+                    'anthropic-beta': 'prompt-caching-2024-07-31',
+                },
+            });
+        }
+        catch (error) {
+            const llmError = new errors_1.LLMServiceError('Failed to setup LLM models', {
+                cause: error,
+                tags: {
+                    hasClient: !!this.client,
+                    hasUser: !!user,
+                },
+            });
+            throw llmError;
+        }
+    }
+    getNodeTypes() {
+        const ignoredTypes = [
+            '@n8n/n8n-nodes-langchain.toolVectorStore',
+            '@n8n/n8n-nodes-langchain.documentGithubLoader',
+            '@n8n/n8n-nodes-langchain.code',
+        ];
+        const nodeTypesKeys = Object.keys(this.nodeTypes.getKnownTypes());
+        const nodeTypes = nodeTypesKeys
+            .filter((nodeType) => !ignoredTypes.includes(nodeType))
+            .map((nodeName) => {
+            try {
+                return { ...this.nodeTypes.getByNameAndVersion(nodeName).description, name: nodeName };
+            }
+            catch (error) {
+                this.logger?.error('Error getting node type', {
+                    nodeName,
+                    error: error instanceof Error ? error.message : 'Unknown error',
+                });
+                return undefined;
+            }
+        })
+            .filter((nodeType) => nodeType !== undefined && nodeType.hidden !== true)
+            .map((nodeType, _index, nodeTypes) => {
+            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,
+            };
+        });
+        return nodeTypes;
+    }
+    async getAgent(user) {
+        if (!this.llmComplexTask || !this.llmSimpleTask) {
+            await this.setupModels(user);
+        }
+        if (!this.llmComplexTask || !this.llmSimpleTask) {
+            throw new errors_1.LLMServiceError('Failed to initialize LLM models');
+        }
+        this.agent ??= new workflow_builder_agent_1.WorkflowBuilderAgent({
+            parsedNodeTypes: this.parsedNodeTypes,
+            llmSimpleTask: this.llmComplexTask,
+            llmComplexTask: this.llmComplexTask,
+            logger: this.logger,
+            checkpointer: this.checkpointer,
+            tracer: this.tracingClient
+                ? new tracer_langchain_1.LangChainTracer({ client: this.tracingClient, projectName: 'n8n-workflow-builder' })
+                : undefined,
+        });
+        return this.agent;
+    }
+    async *chat(payload, user) {
+        const agent = await this.getAgent(user);
+        for await (const output of agent.chat(payload, user?.id?.toString())) {
+            yield output;
+        }
+    }
+    async getSessions(workflowId, user) {
+        const agent = await this.getAgent(user);
+        return await agent.getSessions(workflowId, user?.id?.toString());
+    }
+};
+exports.AiWorkflowBuilderService = AiWorkflowBuilderService;
+exports.AiWorkflowBuilderService = AiWorkflowBuilderService = __decorate([
+    (0, di_1.Service)(),
+    __metadata("design:paramtypes", [Object, ai_assistant_sdk_1.AiAssistantClient,
+        backend_common_1.Logger])
+], 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,+EAA2E;AAC3E,oDAAmD;AACnD,wDAA6C;AAC7C,gCAAkC;AAClC,+DAA6D;AAC7D,yCAAmC;AAInC,qCAA2C;AAC3C,6CAAiE;AACjE,qEAAkF;AAG3E,IAAM,wBAAwB,GAA9B,MAAM,wBAAwB;IAapC,YACkB,SAAqB,EACrB,MAA0B,EAC1B,MAAe;QAFf,cAAS,GAAT,SAAS,CAAY;QACrB,WAAM,GAAN,MAAM,CAAoB;QAC1B,WAAM,GAAN,MAAM,CAAS;QAfzB,oBAAe,GAA2B,EAAE,CAAC;QAQ7C,iBAAY,GAAG,IAAI,uBAAW,EAAE,CAAC;QASxC,IAAI,CAAC,eAAe,GAAG,IAAI,CAAC,YAAY,EAAE,CAAC;IAC5C,CAAC;IAEO,KAAK,CAAC,WAAW,CAAC,IAAY;QACrC,IAAI,CAAC;YACJ,IAAI,IAAI,CAAC,aAAa,IAAI,IAAI,CAAC,cAAc,EAAE,CAAC;gBAC/C,OAAO;YACR,CAAC;YAGD,IAAI,IAAI,CAAC,MAAM,IAAI,IAAI,EAAE,CAAC;gBACzB,MAAM,WAAW,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,2BAA2B,CAAC,IAAI,CAAC,CAAC;gBAExE,MAAM,OAAO,GAAG,IAAI,CAAC,MAAM,CAAC,kBAAkB,EAAE,CAAC;gBAEjD,IAAI,CAAC,aAAa,GAAG,MAAM,IAAA,sBAAS,EAAC;oBACpC,OAAO,EAAE,OAAO,GAAG,SAAS;oBAE5B,MAAM,EAAE,GAAG;oBACX,OAAO,EAAE;wBACR,aAAa,EAAE,WAAW,CAAC,MAAM;qBACjC;iBACD,CAAC,CAAC;gBACH,IAAI,CAAC,cAAc,GAAG,MAAM,IAAA,mCAAsB,EAAC;oBAClD,OAAO,EAAE,OAAO,GAAG,YAAY;oBAC/B,MAAM,EAAE,GAAG;oBACX,OAAO,EAAE;wBACR,aAAa,EAAE,WAAW,CAAC,MAAM;wBACjC,gBAAgB,EAAE,2BAA2B;qBAC7C;iBACD,CAAC,CAAC;gBAEH,IAAI,CAAC,aAAa,GAAG,IAAI,kBAAM,CAAC;oBAC/B,MAAM,EAAE,GAAG;oBACX,MAAM,EAAE,OAAO,GAAG,YAAY;oBAC9B,gBAAgB,EAAE,KAAK;oBACvB,qBAAqB,EAAE,CAAC;oBACxB,YAAY,EAAE;wBACb,OAAO,EAAE;4BACR,aAAa,EAAE,WAAW,CAAC,MAAM;yBACjC;qBACD;iBACD,CAAC,CAAC;gBACH,OAAO;YACR,CAAC;YAED,IAAI,CAAC,aAAa,GAAG,MAAM,IAAA,sBAAS,EAAC;gBACpC,MAAM,EAAE,OAAO,CAAC,GAAG,CAAC,qBAAqB,IAAI,EAAE;aAC/C,CAAC,CAAC;YAEH,IAAI,CAAC,cAAc,GAAG,MAAM,IAAA,mCAAsB,EAAC;gBAClD,MAAM,EAAE,OAAO,CAAC,GAAG,CAAC,oBAAoB,IAAI,EAAE;gBAC9C,OAAO,EAAE;oBACR,gBAAgB,EAAE,2BAA2B;iBAC7C;aACD,CAAC,CAAC;QACJ,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YAChB,MAAM,QAAQ,GAAG,IAAI,wBAAe,CAAC,4BAA4B,EAAE;gBAClE,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,YAAY;QAEnB,MAAM,YAAY,GAAG;YACpB,0CAA0C;YAC1C,+CAA+C;YAC/C,+BAA+B;SAC/B,CAAC;QACF,MAAM,aAAa,GAAG,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,aAAa,EAAE,CAAC,CAAC;QAElE,MAAM,SAAS,GAAG,aAAa;aAC7B,MAAM,CAAC,CAAC,QAAQ,EAAE,EAAE,CAAC,CAAC,YAAY,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;aACtD,GAAG,CAAC,CAAC,QAAQ,EAAE,EAAE;YACjB,IAAI,CAAC;gBACJ,OAAO,EAAE,GAAG,IAAI,CAAC,SAAS,CAAC,mBAAmB,CAAC,QAAQ,CAAC,CAAC,WAAW,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC;YACxF,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBAChB,IAAI,CAAC,MAAM,EAAE,KAAK,CAAC,yBAAyB,EAAE;oBAC7C,QAAQ;oBACR,KAAK,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,eAAe;iBAC/D,CAAC,CAAC;gBACH,OAAO,SAAS,CAAC;YAClB,CAAC;QACF,CAAC,CAAC;aACD,MAAM,CACN,CAAC,QAAQ,EAAoC,EAAE,CAC9C,QAAQ,KAAK,SAAS,IAAI,QAAQ,CAAC,MAAM,KAAK,IAAI,CACnD;aACA,GAAG,CAAC,CAAC,QAAQ,EAAE,MAAM,EAAE,SAAiC,EAAE,EAAE;YAG5D,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;QAEJ,OAAO,SAAS,CAAC;IAClB,CAAC;IAEO,KAAK,CAAC,QAAQ,CAAC,IAAY;QAClC,IAAI,CAAC,IAAI,CAAC,cAAc,IAAI,CAAC,IAAI,CAAC,aAAa,EAAE,CAAC;YACjD,MAAM,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC;QAC9B,CAAC;QAED,IAAI,CAAC,IAAI,CAAC,cAAc,IAAI,CAAC,IAAI,CAAC,aAAa,EAAE,CAAC;YACjD,MAAM,IAAI,wBAAe,CAAC,iCAAiC,CAAC,CAAC;QAC9D,CAAC;QAED,IAAI,CAAC,KAAK,KAAK,IAAI,6CAAoB,CAAC;YACvC,eAAe,EAAE,IAAI,CAAC,eAAe;YAErC,aAAa,EAAE,IAAI,CAAC,cAAc;YAClC,cAAc,EAAE,IAAI,CAAC,cAAc;YACnC,MAAM,EAAE,IAAI,CAAC,MAAM;YACnB,YAAY,EAAE,IAAI,CAAC,YAAY;YAC/B,MAAM,EAAE,IAAI,CAAC,aAAa;gBACzB,CAAC,CAAC,IAAI,kCAAe,CAAC,EAAE,MAAM,EAAE,IAAI,CAAC,aAAa,EAAE,WAAW,EAAE,sBAAsB,EAAE,CAAC;gBAC1F,CAAC,CAAC,SAAS;SACZ,CAAC,CAAC;QAEH,OAAO,IAAI,CAAC,KAAK,CAAC;IACnB,CAAC;IAED,KAAK,CAAC,CAAC,IAAI,CAAC,OAAoB,EAAE,IAAY;QAC7C,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC;QAExC,IAAI,KAAK,EAAE,MAAM,MAAM,IAAI,KAAK,CAAC,IAAI,CAAC,OAAO,EAAE,IAAI,EAAE,EAAE,EAAE,QAAQ,EAAE,CAAC,EAAE,CAAC;YACtE,MAAM,MAAM,CAAC;QACd,CAAC;IACF,CAAC;IAED,KAAK,CAAC,WAAW,CAAC,UAA8B,EAAE,IAAY;QAC7D,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC;QACxC,OAAO,MAAM,KAAK,CAAC,WAAW,CAAC,UAAU,EAAE,IAAI,EAAE,EAAE,EAAE,QAAQ,EAAE,CAAC,CAAC;IAClE,CAAC;CACD,CAAA;AAtKY,4DAAwB;mCAAxB,wBAAwB;IADpC,IAAA,YAAO,GAAE;6CAgBkB,oCAAiB;QACjB,uBAAM;GAhBrB,wBAAwB,CAsKpC"}