n8n 2.23.2 → 2.24.0

package/dist/modules/agents/agent-knowledge.service.js

@@ -0,0 +1,300 @@
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+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 __importStar = (this && this.__importStar) || (function () {
+    var ownKeys = function(o) {
+        ownKeys = Object.getOwnPropertyNames || function (o) {
+            var ar = [];
+            for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
+            return ar;
+        };
+        return ownKeys(o);
+    };
+    return function (mod) {
+        if (mod && mod.__esModule) return mod;
+        var result = {};
+        if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
+        __setModuleDefault(result, mod);
+        return result;
+    };
+})();
+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 };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.AgentKnowledgeService = void 0;
+const di_1 = require("@n8n/di");
+const utils_1 = require("@n8n/utils");
+const n8n_core_1 = require("n8n-core");
+const n8n_workflow_1 = require("n8n-workflow");
+const node_fs_1 = require("node:fs");
+const promises_1 = require("node:fs/promises");
+const node_path_1 = __importDefault(require("node:path"));
+const promises_2 = require("node:stream/promises");
+const bad_request_error_1 = require("../../errors/response-errors/bad-request.error");
+const not_found_error_1 = require("../../errors/response-errors/not-found.error");
+const agent_file_repository_1 = require("./repositories/agent-file.repository");
+const agent_repository_1 = require("./repositories/agent.repository");
+const MAX_AGENT_FILE_METADATA_LENGTH = 255;
+const MAX_WORKSPACE_FILES = 2_000;
+const MAX_WORKSPACE_BYTES = 2 * 1024 * 1024 * 1024;
+let AgentKnowledgeService = class AgentKnowledgeService {
+    constructor(agentRepository, agentFileRepository, binaryDataService) {
+        this.agentRepository = agentRepository;
+        this.agentFileRepository = agentFileRepository;
+        this.binaryDataService = binaryDataService;
+    }
+    async uploadFiles(agentId, projectId, files) {
+        await this.ensureAgentBelongsToProject(agentId, projectId);
+        const storedFiles = [];
+        try {
+            for (const file of files) {
+                storedFiles.push(await this.storeFile(agentId, file));
+            }
+        }
+        catch (error) {
+            await this.cleanupStoredFiles(storedFiles).catch(() => { });
+            throw error;
+        }
+        finally {
+            await this.cleanupUploadTempFiles(files);
+        }
+        return storedFiles.map((file) => this.toDto(file));
+    }
+    async listFiles(agentId, projectId) {
+        await this.ensureAgentBelongsToProject(agentId, projectId);
+        const files = await this.agentFileRepository.findByAgentId(agentId);
+        return files.map((file) => this.toDto(file));
+    }
+    async listWorkspaceFiles(agentId, projectId) {
+        await this.ensureAgentBelongsToProject(agentId, projectId);
+        const files = await this.agentFileRepository.findByAgentId(agentId);
+        return files.map((file) => this.toWorkspaceFile(file));
+    }
+    async deleteFile(agentId, projectId, fileId) {
+        await this.ensureAgentBelongsToProject(agentId, projectId);
+        const file = await this.agentFileRepository.findByIdAndAgentId(fileId, agentId);
+        if (!file) {
+            throw new not_found_error_1.NotFoundError(`Agent file "${fileId}" not found`);
+        }
+        await this.binaryDataService.deleteManyByBinaryDataId([file.binaryDataId]);
+        await this.agentFileRepository.delete({ id: fileId, agentId });
+    }
+    async deleteAllFilesForAgent(agentId) {
+        const files = await this.agentFileRepository.findByAgentId(agentId);
+        if (files.length === 0)
+            return;
+        await this.binaryDataService.deleteManyByBinaryDataId(files.map((file) => file.binaryDataId));
+        await this.agentFileRepository.delete({ agentId });
+    }
+    async resolveWorkspaceFiles(agentId, projectId, fileReferences) {
+        await this.ensureAgentBelongsToProject(agentId, projectId);
+        const files = this.filterFilesForWorkspace(await this.agentFileRepository.findByAgentId(agentId), fileReferences);
+        this.assertWorkspaceWithinLimits(files);
+        return files.map((file) => this.toWorkspaceFile(file));
+    }
+    async materializeWorkspace(agentId, projectId, workspaceRoot, options = {}) {
+        await this.ensureAgentBelongsToProject(agentId, projectId);
+        await (0, promises_1.mkdir)(workspaceRoot, { recursive: true });
+        const files = this.filterFilesForWorkspace(await this.agentFileRepository.findByAgentId(agentId), options.fileReferences);
+        this.assertWorkspaceWithinLimits(files);
+        const materializedFiles = [];
+        for (const file of files) {
+            const relativePath = this.getWorkspaceRelativePath(file);
+            const targetPath = node_path_1.default.join(workspaceRoot, relativePath);
+            const contentStream = await this.binaryDataService.getAsStream(file.binaryDataId);
+            await (0, promises_2.pipeline)(contentStream, (0, node_fs_1.createWriteStream)(targetPath));
+            materializedFiles.push(this.toWorkspaceFile(file));
+        }
+        return materializedFiles;
+    }
+    async ensureAgentBelongsToProject(agentId, projectId) {
+        const agent = await this.agentRepository.findByIdAndProjectId(agentId, projectId);
+        if (!agent) {
+            throw new not_found_error_1.NotFoundError(`Agent "${agentId}" not found`);
+        }
+    }
+    async storeFile(agentId, file) {
+        let storedBinaryDataId;
+        try {
+            const fileId = (0, utils_1.generateNanoId)();
+            const fileName = (0, utils_1.sanitizeFilename)(Buffer.from(file.originalname, 'latin1').toString('utf8'), MAX_AGENT_FILE_METADATA_LENGTH + 1);
+            this.validateMetadataLength('File name', fileName);
+            const buffer = file.buffer ?? (await (0, promises_1.readFile)(file.path));
+            const storedContent = await this.prepareStoredContent(fileName, file.mimetype, buffer);
+            this.validateMetadataLength('MIME type', storedContent.mimeType);
+            const binaryData = {
+                data: '',
+                mimeType: storedContent.mimeType,
+                fileName: storedContent.fileName,
+                fileSize: `${storedContent.buffer.length}`,
+                bytes: storedContent.buffer.length,
+                fileExtension: storedContent.fileExtension,
+            };
+            const storedBinaryData = await this.binaryDataService.store(n8n_core_1.FileLocation.ofCustom({
+                sourceType: 'agent_file',
+                sourceId: fileId,
+                pathSegments: ['agents', agentId, 'files', fileId],
+            }), storedContent.buffer, binaryData);
+            if (!storedBinaryData.id) {
+                throw new n8n_workflow_1.UnexpectedError('Agent file upload requires persisted binary data');
+            }
+            storedBinaryDataId = storedBinaryData.id;
+            const agentFile = this.agentFileRepository.create({
+                id: fileId,
+                agentId,
+                binaryDataId: storedBinaryDataId,
+                fileName,
+                mimeType: storedContent.mimeType,
+                fileSizeBytes: buffer.length,
+            });
+            return await this.agentFileRepository.save(agentFile);
+        }
+        catch (error) {
+            if (storedBinaryDataId) {
+                await this.binaryDataService.deleteManyByBinaryDataId([storedBinaryDataId]);
+            }
+            throw error;
+        }
+        finally {
+            if (file.path) {
+                await (0, promises_1.unlink)(file.path).catch(() => { });
+            }
+        }
+    }
+    toDto(file) {
+        return {
+            id: file.id,
+            agentId: file.agentId,
+            fileName: file.fileName,
+            mimeType: file.mimeType,
+            fileSizeBytes: file.fileSizeBytes,
+            createdAt: file.createdAt.toISOString(),
+        };
+    }
+    toWorkspaceFile(file) {
+        return {
+            id: file.id,
+            fileName: file.fileName,
+            mimeType: file.mimeType,
+            fileSizeBytes: file.fileSizeBytes,
+            relativePath: this.getWorkspaceRelativePath(file),
+        };
+    }
+    assertWorkspaceWithinLimits(files) {
+        if (files.length > MAX_WORKSPACE_FILES) {
+            throw new bad_request_error_1.BadRequestError(`Cannot materialize ${files.length} knowledge files at once (limit ${MAX_WORKSPACE_FILES}). Pass file references to narrow the operation.`);
+        }
+        const totalBytes = files.reduce((total, file) => total + file.fileSizeBytes, 0);
+        if (totalBytes > MAX_WORKSPACE_BYTES) {
+            throw new bad_request_error_1.BadRequestError(`Cannot materialize ${totalBytes} bytes of knowledge files at once (limit ${MAX_WORKSPACE_BYTES}). Pass file references to narrow the operation.`);
+        }
+    }
+    filterFilesForWorkspace(files, fileReferences) {
+        if (!fileReferences)
+            return files;
+        const requested = new Set(fileReferences);
+        return files.filter((file) => requested.has(file.id) ||
+            requested.has(this.getWorkspaceRelativePath(file)) ||
+            requested.has(file.fileName));
+    }
+    getWorkspaceRelativePath(file) {
+        const extension = node_path_1.default.extname(file.fileName).toLowerCase();
+        if (extension === '.pdf' && file.mimeType === 'text/plain') {
+            return `${file.id}.pdf.txt`;
+        }
+        return `${file.id}${node_path_1.default.extname(file.fileName)}`;
+    }
+    async prepareStoredContent(fileName, mimeType, buffer) {
+        if (!this.isPdf(fileName, mimeType)) {
+            return {
+                buffer,
+                mimeType: mimeType || 'application/octet-stream',
+                fileName,
+                fileExtension: fileName.split('.').pop(),
+            };
+        }
+        const extractedText = await this.extractPdfText(fileName, buffer);
+        const extractedBuffer = Buffer.from(extractedText, 'utf8');
+        return {
+            buffer: extractedBuffer,
+            mimeType: 'text/plain',
+            fileName: `${fileName}.txt`,
+            fileExtension: 'txt',
+        };
+    }
+    isPdf(fileName, mimeType) {
+        return node_path_1.default.extname(fileName).toLowerCase() === '.pdf' || mimeType === 'application/pdf';
+    }
+    async extractPdfText(fileName, buffer) {
+        const { PDFParse } = await Promise.resolve().then(() => __importStar(require('pdf-parse')));
+        const parser = new PDFParse({ data: buffer });
+        try {
+            const result = await parser.getText();
+            const text = result.text.trim();
+            if (!text) {
+                throw new bad_request_error_1.BadRequestError(`PDF "${fileName}" contains no extractable text and cannot be added to knowledge`);
+            }
+            return text;
+        }
+        catch (error) {
+            if (error instanceof bad_request_error_1.BadRequestError)
+                throw error;
+            const message = error instanceof Error ? error.message : 'unknown error';
+            throw new bad_request_error_1.BadRequestError(`Failed to extract text from PDF "${fileName}": ${message}`);
+        }
+        finally {
+            await parser.destroy();
+        }
+    }
+    validateMetadataLength(label, value) {
+        if (value.length <= MAX_AGENT_FILE_METADATA_LENGTH)
+            return;
+        throw new bad_request_error_1.BadRequestError(`${label} must be ${MAX_AGENT_FILE_METADATA_LENGTH} characters or less`);
+    }
+    async cleanupStoredFiles(files) {
+        if (files.length === 0)
+            return;
+        await this.agentFileRepository.delete(files.map((file) => file.id));
+        await this.binaryDataService.deleteManyByBinaryDataId(files.map((file) => file.binaryDataId));
+    }
+    async cleanupUploadTempFiles(files) {
+        await Promise.all(files.map(async (file) => await this.cleanupUploadTempFile(file)));
+    }
+    async cleanupUploadTempFile(file) {
+        if (!file.path)
+            return;
+        await (0, promises_1.unlink)(file.path).catch(() => { });
+    }
+};
+exports.AgentKnowledgeService = AgentKnowledgeService;
+exports.AgentKnowledgeService = AgentKnowledgeService = __decorate([
+    (0, di_1.Service)(),
+    __metadata("design:paramtypes", [agent_repository_1.AgentRepository,
+        agent_file_repository_1.AgentFileRepository,
+        n8n_core_1.BinaryDataService])
+], AgentKnowledgeService);
+//# sourceMappingURL=agent-knowledge.service.js.map

package/dist/modules/agents/agent-knowledge.service.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"agent-knowledge.service.js","sourceRoot":"","sources":["../../../src/modules/agents/agent-knowledge.service.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AACA,gCAAkC;AAClC,sCAA8D;AAC9D,uCAA2D;AAC3D,+CAAiE;AACjE,qCAA4C;AAC5C,+CAA2D;AAC3D,0DAA6B;AAC7B,mDAAgD;AAEhD,kFAA6E;AAC7E,8EAAyE;AAGzE,gFAA2E;AAC3E,sEAAkE;AA8BlE,MAAM,8BAA8B,GAAG,GAAG,CAAC;AAO3C,MAAM,mBAAmB,GAAG,KAAK,CAAC;AAClC,MAAM,mBAAmB,GAAG,CAAC,GAAG,IAAI,GAAG,IAAI,GAAG,IAAI,CAAC;AAG5C,IAAM,qBAAqB,GAA3B,MAAM,qBAAqB;IACjC,YACkB,eAAgC,EAChC,mBAAwC,EACxC,iBAAoC;QAFpC,oBAAe,GAAf,eAAe,CAAiB;QAChC,wBAAmB,GAAnB,mBAAmB,CAAqB;QACxC,sBAAiB,GAAjB,iBAAiB,CAAmB;IACnD,CAAC;IAEJ,KAAK,CAAC,WAAW,CAChB,OAAe,EACf,SAAiB,EACjB,KAA4B;QAE5B,MAAM,IAAI,CAAC,2BAA2B,CAAC,OAAO,EAAE,SAAS,CAAC,CAAC;QAE3D,MAAM,WAAW,GAAsB,EAAE,CAAC;QAE1C,IAAI,CAAC;YAIJ,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;gBAC1B,WAAW,CAAC,IAAI,CAAC,MAAM,IAAI,CAAC,SAAS,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC,CAAC;YACvD,CAAC;QACF,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YAChB,MAAM,IAAI,CAAC,kBAAkB,CAAC,WAAW,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;YAC3D,MAAM,KAAK,CAAC;QACb,CAAC;gBAAS,CAAC;YACV,MAAM,IAAI,CAAC,sBAAsB,CAAC,KAAK,CAAC,CAAC;QAC1C,CAAC;QAED,OAAO,WAAW,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC;IACpD,CAAC;IAMD,KAAK,CAAC,SAAS,CAAC,OAAe,EAAE,SAAiB;QACjD,MAAM,IAAI,CAAC,2BAA2B,CAAC,OAAO,EAAE,SAAS,CAAC,CAAC;QAE3D,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,mBAAmB,CAAC,aAAa,CAAC,OAAO,CAAC,CAAC;QACpE,OAAO,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC;IAC9C,CAAC;IAQD,KAAK,CAAC,kBAAkB,CAAC,OAAe,EAAE,SAAiB;QAC1D,MAAM,IAAI,CAAC,2BAA2B,CAAC,OAAO,EAAE,SAAS,CAAC,CAAC;QAE3D,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,mBAAmB,CAAC,aAAa,CAAC,OAAO,CAAC,CAAC;QACpE,OAAO,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC,CAAC;IACxD,CAAC;IAED,KAAK,CAAC,UAAU,CAAC,OAAe,EAAE,SAAiB,EAAE,MAAc;QAClE,MAAM,IAAI,CAAC,2BAA2B,CAAC,OAAO,EAAE,SAAS,CAAC,CAAC;QAE3D,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,mBAAmB,CAAC,kBAAkB,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;QAChF,IAAI,CAAC,IAAI,EAAE,CAAC;YACX,MAAM,IAAI,+BAAa,CAAC,eAAe,MAAM,aAAa,CAAC,CAAC;QAC7D,CAAC;QAED,MAAM,IAAI,CAAC,iBAAiB,CAAC,wBAAwB,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,CAAC;QAC3E,MAAM,IAAI,CAAC,mBAAmB,CAAC,MAAM,CAAC,EAAE,EAAE,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC,CAAC;IAChE,CAAC;IAED,KAAK,CAAC,sBAAsB,CAAC,OAAe;QAC3C,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,mBAAmB,CAAC,aAAa,CAAC,OAAO,CAAC,CAAC;QACpE,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO;QAE/B,MAAM,IAAI,CAAC,iBAAiB,CAAC,wBAAwB,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,CAAC;QAC9F,MAAM,IAAI,CAAC,mBAAmB,CAAC,MAAM,CAAC,EAAE,OAAO,EAAE,CAAC,CAAC;IACpD,CAAC;IAQD,KAAK,CAAC,qBAAqB,CAC1B,OAAe,EACf,SAAiB,EACjB,cAAyB;QAEzB,MAAM,IAAI,CAAC,2BAA2B,CAAC,OAAO,EAAE,SAAS,CAAC,CAAC;QAC3D,MAAM,KAAK,GAAG,IAAI,CAAC,uBAAuB,CACzC,MAAM,IAAI,CAAC,mBAAmB,CAAC,aAAa,CAAC,OAAO,CAAC,EACrD,cAAc,CACd,CAAC;QACF,IAAI,CAAC,2BAA2B,CAAC,KAAK,CAAC,CAAC;QACxC,OAAO,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC,CAAC;IACxD,CAAC;IAED,KAAK,CAAC,oBAAoB,CACzB,OAAe,EACf,SAAiB,EACjB,aAAqB,EACrB,UAAuC,EAAE;QAEzC,MAAM,IAAI,CAAC,2BAA2B,CAAC,OAAO,EAAE,SAAS,CAAC,CAAC;QAC3D,MAAM,IAAA,gBAAK,EAAC,aAAa,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;QAEhD,MAAM,KAAK,GAAG,IAAI,CAAC,uBAAuB,CACzC,MAAM,IAAI,CAAC,mBAAmB,CAAC,aAAa,CAAC,OAAO,CAAC,EACrD,OAAO,CAAC,cAAc,CACtB,CAAC;QACF,IAAI,CAAC,2BAA2B,CAAC,KAAK,CAAC,CAAC;QACxC,MAAM,iBAAiB,GAA6B,EAAE,CAAC;QAEvD,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;YAC1B,MAAM,YAAY,GAAG,IAAI,CAAC,wBAAwB,CAAC,IAAI,CAAC,CAAC;YACzD,MAAM,UAAU,GAAG,mBAAI,CAAC,IAAI,CAAC,aAAa,EAAE,YAAY,CAAC,CAAC;YAK1D,MAAM,aAAa,GAAG,MAAM,IAAI,CAAC,iBAAiB,CAAC,WAAW,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;YAClF,MAAM,IAAA,mBAAQ,EAAC,aAAa,EAAE,IAAA,2BAAiB,EAAC,UAAU,CAAC,CAAC,CAAC;YAE7D,iBAAiB,CAAC,IAAI,CAAC,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC,CAAC;QACpD,CAAC;QAED,OAAO,iBAAiB,CAAC;IAC1B,CAAC;IAEO,KAAK,CAAC,2BAA2B,CAAC,OAAe,EAAE,SAAiB;QAC3E,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,eAAe,CAAC,oBAAoB,CAAC,OAAO,EAAE,SAAS,CAAC,CAAC;QAClF,IAAI,CAAC,KAAK,EAAE,CAAC;YACZ,MAAM,IAAI,+BAAa,CAAC,UAAU,OAAO,aAAa,CAAC,CAAC;QACzD,CAAC;IACF,CAAC;IAEO,KAAK,CAAC,SAAS,CAAC,OAAe,EAAE,IAAyB;QACjE,IAAI,kBAAsC,CAAC;QAC3C,IAAI,CAAC;YACJ,MAAM,MAAM,GAAG,IAAA,sBAAc,GAAE,CAAC;YAChC,MAAM,QAAQ,GAAG,IAAA,wBAAgB,EAChC,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,YAAY,EAAE,QAAQ,CAAC,CAAC,QAAQ,CAAC,MAAM,CAAC,EACzD,8BAA8B,GAAG,CAAC,CAClC,CAAC;YACF,IAAI,CAAC,sBAAsB,CAAC,WAAW,EAAE,QAAQ,CAAC,CAAC;YACnD,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,IAAI,CAAC,MAAM,IAAA,mBAAQ,EAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC;YAC1D,MAAM,aAAa,GAAG,MAAM,IAAI,CAAC,oBAAoB,CAAC,QAAQ,EAAE,IAAI,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC;YACvF,IAAI,CAAC,sBAAsB,CAAC,WAAW,EAAE,aAAa,CAAC,QAAQ,CAAC,CAAC;YACjE,MAAM,UAAU,GAAgB;gBAC/B,IAAI,EAAE,EAAE;gBACR,QAAQ,EAAE,aAAa,CAAC,QAAQ;gBAChC,QAAQ,EAAE,aAAa,CAAC,QAAQ;gBAChC,QAAQ,EAAE,GAAG,aAAa,CAAC,MAAM,CAAC,MAAM,EAAE;gBAC1C,KAAK,EAAE,aAAa,CAAC,MAAM,CAAC,MAAM;gBAClC,aAAa,EAAE,aAAa,CAAC,aAAa;aAC1C,CAAC;YAEF,MAAM,gBAAgB,GAAG,MAAM,IAAI,CAAC,iBAAiB,CAAC,KAAK,CAC1D,uBAAY,CAAC,QAAQ,CAAC;gBACrB,UAAU,EAAE,YAAY;gBACxB,QAAQ,EAAE,MAAM;gBAChB,YAAY,EAAE,CAAC,QAAQ,EAAE,OAAO,EAAE,OAAO,EAAE,MAAM,CAAC;aAClD,CAAC,EACF,aAAa,CAAC,MAAM,EACpB,UAAU,CACV,CAAC;YAEF,IAAI,CAAC,gBAAgB,CAAC,EAAE,EAAE,CAAC;gBAC1B,MAAM,IAAI,8BAAe,CAAC,kDAAkD,CAAC,CAAC;YAC/E,CAAC;YACD,kBAAkB,GAAG,gBAAgB,CAAC,EAAE,CAAC;YAEzC,MAAM,SAAS,GAAG,IAAI,CAAC,mBAAmB,CAAC,MAAM,CAAC;gBACjD,EAAE,EAAE,MAAM;gBACV,OAAO;gBACP,YAAY,EAAE,kBAAkB;gBAChC,QAAQ;gBACR,QAAQ,EAAE,aAAa,CAAC,QAAQ;gBAChC,aAAa,EAAE,MAAM,CAAC,MAAM;aAC5B,CAAC,CAAC;YAEH,OAAO,MAAM,IAAI,CAAC,mBAAmB,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QACvD,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YAChB,IAAI,kBAAkB,EAAE,CAAC;gBACxB,MAAM,IAAI,CAAC,iBAAiB,CAAC,wBAAwB,CAAC,CAAC,kBAAkB,CAAC,CAAC,CAAC;YAC7E,CAAC;YACD,MAAM,KAAK,CAAC;QACb,CAAC;gBAAS,CAAC;YACV,IAAI,IAAI,CAAC,IAAI,EAAE,CAAC;gBACf,MAAM,IAAA,iBAAM,EAAC,IAAI,CAAC,IAAI,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;YACzC,CAAC;QACF,CAAC;IACF,CAAC;IAEO,KAAK,CAAC,IAAe;QAC5B,OAAO;YACN,EAAE,EAAE,IAAI,CAAC,EAAE;YACX,OAAO,EAAE,IAAI,CAAC,OAAO;YACrB,QAAQ,EAAE,IAAI,CAAC,QAAQ;YACvB,QAAQ,EAAE,IAAI,CAAC,QAAQ;YACvB,aAAa,EAAE,IAAI,CAAC,aAAa;YACjC,SAAS,EAAE,IAAI,CAAC,SAAS,CAAC,WAAW,EAAE;SACvC,CAAC;IACH,CAAC;IAEO,eAAe,CAAC,IAAe;QACtC,OAAO;YACN,EAAE,EAAE,IAAI,CAAC,EAAE;YACX,QAAQ,EAAE,IAAI,CAAC,QAAQ;YACvB,QAAQ,EAAE,IAAI,CAAC,QAAQ;YACvB,aAAa,EAAE,IAAI,CAAC,aAAa;YACjC,YAAY,EAAE,IAAI,CAAC,wBAAwB,CAAC,IAAI,CAAC;SACjD,CAAC;IACH,CAAC;IAEO,2BAA2B,CAAC,KAAkB;QACrD,IAAI,KAAK,CAAC,MAAM,GAAG,mBAAmB,EAAE,CAAC;YACxC,MAAM,IAAI,mCAAe,CACxB,sBAAsB,KAAK,CAAC,MAAM,mCAAmC,mBAAmB,kDAAkD,CAC1I,CAAC;QACH,CAAC;QACD,MAAM,UAAU,GAAG,KAAK,CAAC,MAAM,CAAC,CAAC,KAAK,EAAE,IAAI,EAAE,EAAE,CAAC,KAAK,GAAG,IAAI,CAAC,aAAa,EAAE,CAAC,CAAC,CAAC;QAChF,IAAI,UAAU,GAAG,mBAAmB,EAAE,CAAC;YACtC,MAAM,IAAI,mCAAe,CACxB,sBAAsB,UAAU,4CAA4C,mBAAmB,kDAAkD,CACjJ,CAAC;QACH,CAAC;IACF,CAAC;IAEO,uBAAuB,CAAC,KAAkB,EAAE,cAAoC;QACvF,IAAI,CAAC,cAAc;YAAE,OAAO,KAAK,CAAC;QAElC,MAAM,SAAS,GAAG,IAAI,GAAG,CAAC,cAAc,CAAC,CAAC;QAC1C,OAAO,KAAK,CAAC,MAAM,CAClB,CAAC,IAAI,EAAE,EAAE,CACR,SAAS,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC;YACtB,SAAS,CAAC,GAAG,CAAC,IAAI,CAAC,wBAAwB,CAAC,IAAI,CAAC,CAAC;YAClD,SAAS,CAAC,GAAG,CAAC,IAAI,CAAC,QAAQ,CAAC,CAC7B,CAAC;IACH,CAAC;IAEO,wBAAwB,CAAC,IAAe;QAC/C,MAAM,SAAS,GAAG,mBAAI,CAAC,OAAO,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,WAAW,EAAE,CAAC;QAC5D,IAAI,SAAS,KAAK,MAAM,IAAI,IAAI,CAAC,QAAQ,KAAK,YAAY,EAAE,CAAC;YAC5D,OAAO,GAAG,IAAI,CAAC,EAAE,UAAU,CAAC;QAC7B,CAAC;QACD,OAAO,GAAG,IAAI,CAAC,EAAE,GAAG,mBAAI,CAAC,OAAO,CAAC,IAAI,CAAC,QAAQ,CAAC,EAAE,CAAC;IACnD,CAAC;IAEO,KAAK,CAAC,oBAAoB,CACjC,QAAgB,EAChB,QAAgB,EAChB,MAAc;QAEd,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,QAAQ,EAAE,QAAQ,CAAC,EAAE,CAAC;YACrC,OAAO;gBACN,MAAM;gBACN,QAAQ,EAAE,QAAQ,IAAI,0BAA0B;gBAChD,QAAQ;gBACR,aAAa,EAAE,QAAQ,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE;aACxC,CAAC;QACH,CAAC;QAED,MAAM,aAAa,GAAG,MAAM,IAAI,CAAC,cAAc,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC;QAClE,MAAM,eAAe,GAAG,MAAM,CAAC,IAAI,CAAC,aAAa,EAAE,MAAM,CAAC,CAAC;QAE3D,OAAO;YACN,MAAM,EAAE,eAAe;YACvB,QAAQ,EAAE,YAAY;YACtB,QAAQ,EAAE,GAAG,QAAQ,MAAM;YAC3B,aAAa,EAAE,KAAK;SACpB,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,QAAgB,EAAE,QAAgB;QAC/C,OAAO,mBAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC,WAAW,EAAE,KAAK,MAAM,IAAI,QAAQ,KAAK,iBAAiB,CAAC;IAC1F,CAAC;IAEO,KAAK,CAAC,cAAc,CAAC,QAAgB,EAAE,MAAc;QAC5D,MAAM,EAAE,QAAQ,EAAE,GAAG,wDAAa,WAAW,GAAC,CAAC;QAC/C,MAAM,MAAM,GAAG,IAAI,QAAQ,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC,CAAC;QAC9C,IAAI,CAAC;YACJ,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,OAAO,EAAE,CAAC;YACtC,MAAM,IAAI,GAAG,MAAM,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC;YAChC,IAAI,CAAC,IAAI,EAAE,CAAC;gBACX,MAAM,IAAI,mCAAe,CACxB,QAAQ,QAAQ,iEAAiE,CACjF,CAAC;YACH,CAAC;YACD,OAAO,IAAI,CAAC;QACb,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YAChB,IAAI,KAAK,YAAY,mCAAe;gBAAE,MAAM,KAAK,CAAC;YAClD,MAAM,OAAO,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,eAAe,CAAC;YACzE,MAAM,IAAI,mCAAe,CAAC,oCAAoC,QAAQ,MAAM,OAAO,EAAE,CAAC,CAAC;QACxF,CAAC;gBAAS,CAAC;YACV,MAAM,MAAM,CAAC,OAAO,EAAE,CAAC;QACxB,CAAC;IACF,CAAC;IAEO,sBAAsB,CAAC,KAAa,EAAE,KAAa;QAC1D,IAAI,KAAK,CAAC,MAAM,IAAI,8BAA8B;YAAE,OAAO;QAE3D,MAAM,IAAI,mCAAe,CACxB,GAAG,KAAK,YAAY,8BAA8B,qBAAqB,CACvE,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,kBAAkB,CAAC,KAAwB;QACxD,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO;QAE/B,MAAM,IAAI,CAAC,mBAAmB,CAAC,MAAM,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,CAAC;QACpE,MAAM,IAAI,CAAC,iBAAiB,CAAC,wBAAwB,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,CAAC;IAC/F,CAAC;IAEO,KAAK,CAAC,sBAAsB,CAAC,KAA4B;QAChE,MAAM,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,GAAG,CAAC,KAAK,EAAE,IAAI,EAAE,EAAE,CAAC,MAAM,IAAI,CAAC,qBAAqB,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IACtF,CAAC;IAEO,KAAK,CAAC,qBAAqB,CAAC,IAAyB;QAC5D,IAAI,CAAC,IAAI,CAAC,IAAI;YAAE,OAAO;QAEvB,MAAM,IAAA,iBAAM,EAAC,IAAI,CAAC,IAAI,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;IACzC,CAAC;CACD,CAAA;AAnUY,sDAAqB;gCAArB,qBAAqB;IADjC,IAAA,YAAO,GAAE;qCAG0B,kCAAe;QACX,2CAAmB;QACrB,4BAAiB;GAJ1C,qBAAqB,CAmUjC"}

package/dist/modules/agents/agent-task.service.d.ts

@@ -0,0 +1,52 @@
+import type { AgentTaskDto, CreateAgentTaskDto, UpdateAgentTaskDto } from '@n8n/api-types';
+import { Logger } from '@n8n/backend-common';
+import { GlobalConfig } from '@n8n/config';
+import { ProjectRelationRepository } from '@n8n/db';
+import { InstanceSettings } from 'n8n-core';
+import type { PubSubCommandMap } from '../../scaling/pubsub/pubsub.event-map';
+import { Publisher } from '../../scaling/pubsub/publisher.service';
+import { AgentsService } from './agents.service';
+import { AgentRepository } from './repositories/agent.repository';
+import { AgentTaskRunLockRepository } from './repositories/agent-task-run-lock.repository';
+import { AgentTaskSnapshotRepository } from './repositories/agent-task-snapshot.repository';
+import { AgentTaskRepository } from './repositories/agent-task.repository';
+export declare class AgentTaskService {
+    private readonly logger;
+    private readonly globalConfig;
+    private readonly taskRepository;
+    private readonly taskSnapshotRepository;
+    private readonly taskRunLockRepository;
+    private readonly agentRepository;
+    private readonly projectRelationRepository;
+    private readonly agentsService;
+    private readonly instanceSettings;
+    private readonly publisher;
+    private readonly jobs;
+    constructor(logger: Logger, globalConfig: GlobalConfig, taskRepository: AgentTaskRepository, taskSnapshotRepository: AgentTaskSnapshotRepository, taskRunLockRepository: AgentTaskRunLockRepository, agentRepository: AgentRepository, projectRelationRepository: ProjectRelationRepository, agentsService: AgentsService, instanceSettings: InstanceSettings, publisher: Publisher);
+    list(agentId: string): Promise<AgentTaskDto[]>;
+    create(agentId: string, dto: CreateAgentTaskDto): Promise<AgentTaskDto>;
+    update(agentId: string, taskId: string, dto: UpdateAgentTaskDto): Promise<AgentTaskDto>;
+    delete(agentId: string, taskId: string): Promise<void>;
+    private attachTaskRef;
+    requestReconcile(agentId: string): Promise<void>;
+    registerEnabledForAgent(agentId: string): Promise<void>;
+    handleTasksChanged(payload: PubSubCommandMap['agent-tasks-changed']): Promise<void>;
+    private broadcastTasksChanged;
+    deregisterAgentTasks(agentId: string): void;
+    reconnectAll(): Promise<void>;
+    stopAll(): void;
+    private reconcileAgent;
+    private registerOrRefresh;
+    private deregister;
+    private runScheduledTask;
+    private startTaskRunLockRenewal;
+    private runTask;
+    private buildTaskRunMessage;
+    private consumeTaskRun;
+    runNow(agentId: string, taskId: string, userId: string): Promise<void>;
+    private executeNow;
+    private resolveExecutionUserId;
+    private getOrThrow;
+    private assertValidCron;
+    private toDto;
+}