n8n 2.23.2 → 2.25.1

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

@@ -0,0 +1,221 @@
+"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 __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.AgentKnowledgeCommandService = exports.AGENT_KNOWLEDGE_COMMANDS = void 0;
+const di_1 = require("@n8n/di");
+const node_child_process_1 = require("node:child_process");
+const promises_1 = require("node:fs/promises");
+const node_os_1 = require("node:os");
+const node_path_1 = __importDefault(require("node:path"));
+const p_limit_1 = __importDefault(require("p-limit"));
+const MAX_OUTPUT_BYTES = 64 * 1024;
+const COMMAND_TIMEOUT_MS = 5_000;
+const MAX_CONCURRENT_WORKSPACES = 4;
+const WORKSPACE_CACHE_TTL_MS = 10 * 60_000;
+const MAX_CACHED_WORKSPACES = 25;
+exports.AGENT_KNOWLEDGE_COMMANDS = ['git_grep', 'cat', 'sed'];
+const workspaceLimit = (0, p_limit_1.default)(MAX_CONCURRENT_WORKSPACES);
+let AgentKnowledgeCommandService = class AgentKnowledgeCommandService {
+    constructor() {
+        this.cachedWorkspaces = new Map();
+        this.workspaceLocks = new Map();
+    }
+    async run(workspaceRoot, request) {
+        const root = await (0, promises_1.realpath)(workspaceRoot);
+        const { executable, args } = await this.toSpawnArgs(root, request);
+        return await this.spawnCommand(root, executable, args, request.command);
+    }
+    async withCachedWorkspace(cacheKey, materialize, operation) {
+        return await this.serializeByKey(cacheKey, async () => await workspaceLimit(async () => {
+            const workspaceRoot = await this.ensureCachedWorkspace(cacheKey, materialize);
+            return await operation(workspaceRoot);
+        }));
+    }
+    async serializeByKey(key, fn) {
+        const previous = this.workspaceLocks.get(key) ?? Promise.resolve();
+        const run = previous.then(fn, fn);
+        const tail = run.then(() => undefined, () => undefined);
+        this.workspaceLocks.set(key, tail);
+        try {
+            return await run;
+        }
+        finally {
+            if (this.workspaceLocks.get(key) === tail)
+                this.workspaceLocks.delete(key);
+        }
+    }
+    async ensureCachedWorkspace(cacheKey, materialize) {
+        const existing = this.cachedWorkspaces.get(cacheKey);
+        if (existing && (await this.directoryExists(existing.root))) {
+            existing.lastUsedAt = Date.now();
+            return existing.root;
+        }
+        if (existing)
+            this.cachedWorkspaces.delete(cacheKey);
+        const workspaceRoot = await (0, promises_1.mkdtemp)(node_path_1.default.join((0, node_os_1.tmpdir)(), 'n8n-agent-knowledge-'));
+        try {
+            await materialize(workspaceRoot);
+        }
+        catch (error) {
+            await (0, promises_1.rm)(workspaceRoot, { recursive: true, force: true }).catch(() => { });
+            throw error;
+        }
+        this.cachedWorkspaces.set(cacheKey, { root: workspaceRoot, lastUsedAt: Date.now() });
+        await this.evictStaleWorkspaces();
+        return workspaceRoot;
+    }
+    async evictStaleWorkspaces() {
+        const now = Date.now();
+        const evictable = [];
+        const fresh = [];
+        for (const entry of this.cachedWorkspaces) {
+            (now - entry[1].lastUsedAt > WORKSPACE_CACHE_TTL_MS ? evictable : fresh).push(entry);
+        }
+        if (fresh.length > MAX_CACHED_WORKSPACES) {
+            fresh.sort((left, right) => left[1].lastUsedAt - right[1].lastUsedAt);
+            evictable.push(...fresh.slice(0, fresh.length - MAX_CACHED_WORKSPACES));
+        }
+        for (const [key, workspace] of evictable) {
+            this.cachedWorkspaces.delete(key);
+            await (0, promises_1.rm)(workspace.root, { recursive: true, force: true }).catch(() => { });
+        }
+    }
+    async directoryExists(directory) {
+        try {
+            await (0, promises_1.realpath)(directory);
+            return true;
+        }
+        catch {
+            return false;
+        }
+    }
+    async toSpawnArgs(root, request) {
+        switch (request.command) {
+            case 'git_grep': {
+                if (request.pattern.trim() === '')
+                    throw new Error('Search pattern is required');
+                const args = ['grep', '--no-index', '-n', '-I'];
+                if (request.caseInsensitive)
+                    args.push('-i');
+                if (request.fixedStrings)
+                    args.push('-F');
+                if (request.fixedStrings === false)
+                    args.push('-E');
+                if (request.outputMode === 'count')
+                    args.push('-c');
+                if (request.context !== undefined) {
+                    args.push('-C', String(Math.min(Math.max(request.context, 0), 5)));
+                }
+                args.push('--', request.pattern);
+                const files = await Promise.all((request.files ?? ['.']).map(async (file) => await this.safePath(root, file, { allowRoot: true })));
+                args.push(...files.map((file) => node_path_1.default.relative(root, file) || '.'));
+                return { executable: 'git', args };
+            }
+            case 'cat': {
+                const file = await this.safePath(root, request.file);
+                return { executable: 'cat', args: [node_path_1.default.relative(root, file)] };
+            }
+            case 'sed': {
+                const file = await this.safePath(root, request.file);
+                const startLine = Math.max(1, request.startLine);
+                const endLine = Math.max(startLine, request.endLine);
+                return {
+                    executable: 'sed',
+                    args: [
+                        '-n',
+                        `${startLine},${Math.min(endLine, startLine + 500)}p`,
+                        node_path_1.default.relative(root, file),
+                    ],
+                };
+            }
+        }
+    }
+    async safePath(root, requestedPath, options = {}) {
+        if (this.hasControlCharacters(requestedPath))
+            throw new Error('Invalid path');
+        if (node_path_1.default.isAbsolute(requestedPath))
+            throw new Error('Absolute paths are not allowed');
+        if (requestedPath.split(/[\\/]/).includes('..')) {
+            throw new Error('Parent path segments are not allowed');
+        }
+        const resolved = node_path_1.default.resolve(root, requestedPath);
+        const actual = await (0, promises_1.realpath)(resolved);
+        const relative = node_path_1.default.relative(root, actual);
+        if ((!options.allowRoot && relative === '') ||
+            relative.startsWith('..') ||
+            node_path_1.default.isAbsolute(relative)) {
+            throw new Error('Path escapes the knowledge workspace');
+        }
+        return actual;
+    }
+    hasControlCharacters(value) {
+        for (const character of value) {
+            const code = character.charCodeAt(0);
+            if (code <= 0x1f || code === 0x7f)
+                return true;
+        }
+        return false;
+    }
+    async spawnCommand(cwd, executable, args, command) {
+        return await new Promise((resolve, reject) => {
+            const child = (0, node_child_process_1.spawn)(executable, args, {
+                cwd,
+                shell: false,
+                env: {
+                    PATH: process.env.PATH,
+                    HOME: cwd,
+                    GIT_CONFIG_NOSYSTEM: '1',
+                    GIT_CONFIG_GLOBAL: '/dev/null',
+                    GIT_TERMINAL_PROMPT: '0',
+                },
+            });
+            let stdout = '';
+            let stderr = '';
+            let truncated = false;
+            const timer = setTimeout(() => {
+                child.kill('SIGKILL');
+                truncated = true;
+            }, COMMAND_TIMEOUT_MS);
+            const append = (current, chunk) => {
+                const next = Buffer.concat([Buffer.from(current, 'utf8'), chunk]);
+                if (next.length > MAX_OUTPUT_BYTES) {
+                    truncated = true;
+                    return truncateBufferToUtf8String(next, MAX_OUTPUT_BYTES);
+                }
+                return next.toString('utf8');
+            };
+            child.stdout.on('data', (chunk) => {
+                stdout = append(stdout, chunk);
+            });
+            child.stderr.on('data', (chunk) => {
+                stderr = append(stderr, chunk);
+            });
+            child.on('error', reject);
+            child.on('close', (exitCode) => {
+                clearTimeout(timer);
+                resolve({ command, exitCode, stdout, stderr, truncated });
+            });
+        });
+    }
+};
+exports.AgentKnowledgeCommandService = AgentKnowledgeCommandService;
+exports.AgentKnowledgeCommandService = AgentKnowledgeCommandService = __decorate([
+    (0, di_1.Service)()
+], AgentKnowledgeCommandService);
+function truncateBufferToUtf8String(buffer, maxBytes) {
+    for (let end = maxBytes; end >= 0; end--) {
+        const output = buffer.subarray(0, end).toString('utf8');
+        if (Buffer.byteLength(output) <= maxBytes)
+            return output;
+    }
+    return '';
+}
+//# sourceMappingURL=agent-knowledge-command.service.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"agent-knowledge-command.service.js","sourceRoot":"","sources":["../../../src/modules/agents/agent-knowledge-command.service.ts"],"names":[],"mappings":";;;;;;;;;;;;AAAA,gCAAkC;AAClC,2DAA2C;AAC3C,+CAAyD;AACzD,qCAAiC;AACjC,0DAA6B;AAC7B,sDAA6B;AAE7B,MAAM,gBAAgB,GAAG,EAAE,GAAG,IAAI,CAAC;AACnC,MAAM,kBAAkB,GAAG,KAAK,CAAC;AAMjC,MAAM,yBAAyB,GAAG,CAAC,CAAC;AAEpC,MAAM,sBAAsB,GAAG,EAAE,GAAG,MAAM,CAAC;AAE3C,MAAM,qBAAqB,GAAG,EAAE,CAAC;AACpB,QAAA,wBAAwB,GAAG,CAAC,UAAU,EAAE,KAAK,EAAE,KAAK,CAAU,CAAC;AAG5E,MAAM,cAAc,GAAG,IAAA,iBAAM,EAAC,yBAAyB,CAAC,CAAC;AAyClD,IAAM,4BAA4B,GAAlC,MAAM,4BAA4B;IAAlC;QACW,qBAAgB,GAAG,IAAI,GAAG,EAA2B,CAAC;QACtD,mBAAc,GAAG,IAAI,GAAG,EAA4B,CAAC;IA0NvE,CAAC;IAxNA,KAAK,CAAC,GAAG,CAAC,aAAqB,EAAE,OAAqC;QACrE,MAAM,IAAI,GAAG,MAAM,IAAA,mBAAQ,EAAC,aAAa,CAAC,CAAC;QAC3C,MAAM,EAAE,UAAU,EAAE,IAAI,EAAE,GAAG,MAAM,IAAI,CAAC,WAAW,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC;QACnE,OAAO,MAAM,IAAI,CAAC,YAAY,CAAC,IAAI,EAAE,UAAU,EAAE,IAAI,EAAE,OAAO,CAAC,OAAO,CAAC,CAAC;IACzE,CAAC;IAUD,KAAK,CAAC,mBAAmB,CACxB,QAAgB,EAChB,WAAqD,EACrD,SAAgD;QAEhD,OAAO,MAAM,IAAI,CAAC,cAAc,CAC/B,QAAQ,EACR,KAAK,IAAI,EAAE,CACV,MAAM,cAAc,CAAC,KAAK,IAAI,EAAE;YAC/B,MAAM,aAAa,GAAG,MAAM,IAAI,CAAC,qBAAqB,CAAC,QAAQ,EAAE,WAAW,CAAC,CAAC;YAC9E,OAAO,MAAM,SAAS,CAAC,aAAa,CAAC,CAAC;QACvC,CAAC,CAAC,CACH,CAAC;IACH,CAAC;IAGO,KAAK,CAAC,cAAc,CAAI,GAAW,EAAE,EAAoB;QAChE,MAAM,QAAQ,GAAG,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,GAAG,CAAC,IAAI,OAAO,CAAC,OAAO,EAAE,CAAC;QACnE,MAAM,GAAG,GAAG,QAAQ,CAAC,IAAI,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC;QAClC,MAAM,IAAI,GAAG,GAAG,CAAC,IAAI,CACpB,GAAG,EAAE,CAAC,SAAS,EACf,GAAG,EAAE,CAAC,SAAS,CACf,CAAC;QACF,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,GAAG,EAAE,IAAI,CAAC,CAAC;QACnC,IAAI,CAAC;YACJ,OAAO,MAAM,GAAG,CAAC;QAClB,CAAC;gBAAS,CAAC;YACV,IAAI,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,GAAG,CAAC,KAAK,IAAI;gBAAE,IAAI,CAAC,cAAc,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;QAC5E,CAAC;IACF,CAAC;IAEO,KAAK,CAAC,qBAAqB,CAClC,QAAgB,EAChB,WAAqD;QAErD,MAAM,QAAQ,GAAG,IAAI,CAAC,gBAAgB,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QACrD,IAAI,QAAQ,IAAI,CAAC,MAAM,IAAI,CAAC,eAAe,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,EAAE,CAAC;YAC7D,QAAQ,CAAC,UAAU,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;YACjC,OAAO,QAAQ,CAAC,IAAI,CAAC;QACtB,CAAC;QACD,IAAI,QAAQ;YAAE,IAAI,CAAC,gBAAgB,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;QAErD,MAAM,aAAa,GAAG,MAAM,IAAA,kBAAO,EAAC,mBAAI,CAAC,IAAI,CAAC,IAAA,gBAAM,GAAE,EAAE,sBAAsB,CAAC,CAAC,CAAC;QACjF,IAAI,CAAC;YACJ,MAAM,WAAW,CAAC,aAAa,CAAC,CAAC;QAClC,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YAChB,MAAM,IAAA,aAAE,EAAC,aAAa,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;YAC1E,MAAM,KAAK,CAAC;QACb,CAAC;QACD,IAAI,CAAC,gBAAgB,CAAC,GAAG,CAAC,QAAQ,EAAE,EAAE,IAAI,EAAE,aAAa,EAAE,UAAU,EAAE,IAAI,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC;QACrF,MAAM,IAAI,CAAC,oBAAoB,EAAE,CAAC;QAClC,OAAO,aAAa,CAAC;IACtB,CAAC;IAEO,KAAK,CAAC,oBAAoB;QACjC,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QACvB,MAAM,SAAS,GAAqC,EAAE,CAAC;QACvD,MAAM,KAAK,GAAqC,EAAE,CAAC;QACnD,KAAK,MAAM,KAAK,IAAI,IAAI,CAAC,gBAAgB,EAAE,CAAC;YAC3C,CAAC,GAAG,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC,UAAU,GAAG,sBAAsB,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACtF,CAAC;QACD,IAAI,KAAK,CAAC,MAAM,GAAG,qBAAqB,EAAE,CAAC;YAC1C,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,KAAK,EAAE,EAAE,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,UAAU,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC;YACtE,SAAS,CAAC,IAAI,CAAC,GAAG,KAAK,CAAC,KAAK,CAAC,CAAC,EAAE,KAAK,CAAC,MAAM,GAAG,qBAAqB,CAAC,CAAC,CAAC;QACzE,CAAC;QACD,KAAK,MAAM,CAAC,GAAG,EAAE,SAAS,CAAC,IAAI,SAAS,EAAE,CAAC;YAC1C,IAAI,CAAC,gBAAgB,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;YAClC,MAAM,IAAA,aAAE,EAAC,SAAS,CAAC,IAAI,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;QAC5E,CAAC;IACF,CAAC;IAEO,KAAK,CAAC,eAAe,CAAC,SAAiB;QAC9C,IAAI,CAAC;YACJ,MAAM,IAAA,mBAAQ,EAAC,SAAS,CAAC,CAAC;YAC1B,OAAO,IAAI,CAAC;QACb,CAAC;QAAC,MAAM,CAAC;YACR,OAAO,KAAK,CAAC;QACd,CAAC;IACF,CAAC;IAEO,KAAK,CAAC,WAAW,CACxB,IAAY,EACZ,OAAqC;QAErC,QAAQ,OAAO,CAAC,OAAO,EAAE,CAAC;YACzB,KAAK,UAAU,CAAC,CAAC,CAAC;gBACjB,IAAI,OAAO,CAAC,OAAO,CAAC,IAAI,EAAE,KAAK,EAAE;oBAAE,MAAM,IAAI,KAAK,CAAC,4BAA4B,CAAC,CAAC;gBACjF,MAAM,IAAI,GAAG,CAAC,MAAM,EAAE,YAAY,EAAE,IAAI,EAAE,IAAI,CAAC,CAAC;gBAChD,IAAI,OAAO,CAAC,eAAe;oBAAE,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;gBAC7C,IAAI,OAAO,CAAC,YAAY;oBAAE,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;gBAC1C,IAAI,OAAO,CAAC,YAAY,KAAK,KAAK;oBAAE,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;gBACpD,IAAI,OAAO,CAAC,UAAU,KAAK,OAAO;oBAAE,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;gBACpD,IAAI,OAAO,CAAC,OAAO,KAAK,SAAS,EAAE,CAAC;oBACnC,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,GAAG,CAAC,OAAO,CAAC,OAAO,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;gBACpE,CAAC;gBACD,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE,OAAO,CAAC,OAAO,CAAC,CAAC;gBACjC,MAAM,KAAK,GAAG,MAAM,OAAO,CAAC,GAAG,CAC9B,CAAC,OAAO,CAAC,KAAK,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,GAAG,CAC3B,KAAK,EAAE,IAAI,EAAE,EAAE,CAAC,MAAM,IAAI,CAAC,QAAQ,CAAC,IAAI,EAAE,IAAI,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CACpE,CACD,CAAC;gBACF,IAAI,CAAC,IAAI,CAAC,GAAG,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,mBAAI,CAAC,QAAQ,CAAC,IAAI,EAAE,IAAI,CAAC,IAAI,GAAG,CAAC,CAAC,CAAC;gBACpE,OAAO,EAAE,UAAU,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC;YACpC,CAAC;YACD,KAAK,KAAK,CAAC,CAAC,CAAC;gBACZ,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,IAAI,EAAE,OAAO,CAAC,IAAI,CAAC,CAAC;gBACrD,OAAO,EAAE,UAAU,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,mBAAI,CAAC,QAAQ,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC,EAAE,CAAC;YACjE,CAAC;YACD,KAAK,KAAK,CAAC,CAAC,CAAC;gBACZ,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,IAAI,EAAE,OAAO,CAAC,IAAI,CAAC,CAAC;gBACrD,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,OAAO,CAAC,SAAS,CAAC,CAAC;gBACjD,MAAM,OAAO,GAAG,IAAI,CAAC,GAAG,CAAC,SAAS,EAAE,OAAO,CAAC,OAAO,CAAC,CAAC;gBACrD,OAAO;oBACN,UAAU,EAAE,KAAK;oBACjB,IAAI,EAAE;wBACL,IAAI;wBACJ,GAAG,SAAS,IAAI,IAAI,CAAC,GAAG,CAAC,OAAO,EAAE,SAAS,GAAG,GAAG,CAAC,GAAG;wBACrD,mBAAI,CAAC,QAAQ,CAAC,IAAI,EAAE,IAAI,CAAC;qBACzB;iBACD,CAAC;YACH,CAAC;QACF,CAAC;IACF,CAAC;IAEO,KAAK,CAAC,QAAQ,CAAC,IAAY,EAAE,aAAqB,EAAE,UAA2B,EAAE;QACxF,IAAI,IAAI,CAAC,oBAAoB,CAAC,aAAa,CAAC;YAAE,MAAM,IAAI,KAAK,CAAC,cAAc,CAAC,CAAC;QAC9E,IAAI,mBAAI,CAAC,UAAU,CAAC,aAAa,CAAC;YAAE,MAAM,IAAI,KAAK,CAAC,gCAAgC,CAAC,CAAC;QACtF,IAAI,aAAa,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE,CAAC;YACjD,MAAM,IAAI,KAAK,CAAC,sCAAsC,CAAC,CAAC;QACzD,CAAC;QACD,MAAM,QAAQ,GAAG,mBAAI,CAAC,OAAO,CAAC,IAAI,EAAE,aAAa,CAAC,CAAC;QACnD,MAAM,MAAM,GAAG,MAAM,IAAA,mBAAQ,EAAC,QAAQ,CAAC,CAAC;QACxC,MAAM,QAAQ,GAAG,mBAAI,CAAC,QAAQ,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;QAC7C,IACC,CAAC,CAAC,OAAO,CAAC,SAAS,IAAI,QAAQ,KAAK,EAAE,CAAC;YACvC,QAAQ,CAAC,UAAU,CAAC,IAAI,CAAC;YACzB,mBAAI,CAAC,UAAU,CAAC,QAAQ,CAAC,EACxB,CAAC;YACF,MAAM,IAAI,KAAK,CAAC,sCAAsC,CAAC,CAAC;QACzD,CAAC;QACD,OAAO,MAAM,CAAC;IACf,CAAC;IAEO,oBAAoB,CAAC,KAAa;QACzC,KAAK,MAAM,SAAS,IAAI,KAAK,EAAE,CAAC;YAC/B,MAAM,IAAI,GAAG,SAAS,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC;YACrC,IAAI,IAAI,IAAI,IAAI,IAAI,IAAI,KAAK,IAAI;gBAAE,OAAO,IAAI,CAAC;QAChD,CAAC;QACD,OAAO,KAAK,CAAC;IACd,CAAC;IAEO,KAAK,CAAC,YAAY,CACzB,GAAW,EACX,UAAkB,EAClB,IAAc,EACd,OAA8B;QAE9B,OAAO,MAAM,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;YAC5C,MAAM,KAAK,GAAG,IAAA,0BAAK,EAAC,UAAU,EAAE,IAAI,EAAE;gBACrC,GAAG;gBACH,KAAK,EAAE,KAAK;gBAIZ,GAAG,EAAE;oBACJ,IAAI,EAAE,OAAO,CAAC,GAAG,CAAC,IAAI;oBACtB,IAAI,EAAE,GAAG;oBACT,mBAAmB,EAAE,GAAG;oBACxB,iBAAiB,EAAE,WAAW;oBAC9B,mBAAmB,EAAE,GAAG;iBACxB;aACD,CAAC,CAAC;YACH,IAAI,MAAM,GAAG,EAAE,CAAC;YAChB,IAAI,MAAM,GAAG,EAAE,CAAC;YAChB,IAAI,SAAS,GAAG,KAAK,CAAC;YACtB,MAAM,KAAK,GAAG,UAAU,CAAC,GAAG,EAAE;gBAC7B,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;gBACtB,SAAS,GAAG,IAAI,CAAC;YAClB,CAAC,EAAE,kBAAkB,CAAC,CAAC;YAEvB,MAAM,MAAM,GAAG,CAAC,OAAe,EAAE,KAAa,EAAE,EAAE;gBACjD,MAAM,IAAI,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,OAAO,EAAE,MAAM,CAAC,EAAE,KAAK,CAAC,CAAC,CAAC;gBAClE,IAAI,IAAI,CAAC,MAAM,GAAG,gBAAgB,EAAE,CAAC;oBACpC,SAAS,GAAG,IAAI,CAAC;oBACjB,OAAO,0BAA0B,CAAC,IAAI,EAAE,gBAAgB,CAAC,CAAC;gBAC3D,CAAC;gBACD,OAAO,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC;YAC9B,CAAC,CAAC;YAEF,KAAK,CAAC,MAAM,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,KAAa,EAAE,EAAE;gBACzC,MAAM,GAAG,MAAM,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC;YAChC,CAAC,CAAC,CAAC;YACH,KAAK,CAAC,MAAM,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,KAAa,EAAE,EAAE;gBACzC,MAAM,GAAG,MAAM,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC;YAChC,CAAC,CAAC,CAAC;YACH,KAAK,CAAC,EAAE,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;YAC1B,KAAK,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,QAAQ,EAAE,EAAE;gBAC9B,YAAY,CAAC,KAAK,CAAC,CAAC;gBACpB,OAAO,CAAC,EAAE,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,MAAM,EAAE,SAAS,EAAE,CAAC,CAAC;YAC3D,CAAC,CAAC,CAAC;QACJ,CAAC,CAAC,CAAC;IACJ,CAAC;CACD,CAAA;AA5NY,oEAA4B;uCAA5B,4BAA4B;IADxC,IAAA,YAAO,GAAE;GACG,4BAA4B,CA4NxC;AAED,SAAS,0BAA0B,CAAC,MAAc,EAAE,QAAgB;IACnE,KAAK,IAAI,GAAG,GAAG,QAAQ,EAAE,GAAG,IAAI,CAAC,EAAE,GAAG,EAAE,EAAE,CAAC;QAC1C,MAAM,MAAM,GAAG,MAAM,CAAC,QAAQ,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC;QACxD,IAAI,MAAM,CAAC,UAAU,CAAC,MAAM,CAAC,IAAI,QAAQ;YAAE,OAAO,MAAM,CAAC;IAC1D,CAAC;IAED,OAAO,EAAE,CAAC;AACX,CAAC"}

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

@@ -0,0 +1,42 @@
+import type { AgentFileDto } from '@n8n/api-types';
+import { BinaryDataService } from 'n8n-core';
+import { AgentFileRepository } from './repositories/agent-file.repository';
+import { AgentRepository } from './repositories/agent.repository';
+export interface KnowledgeWorkspaceFile {
+    id: string;
+    fileName: string;
+    mimeType: string;
+    fileSizeBytes: number;
+    relativePath: string;
+}
+interface MaterializeWorkspaceOptions {
+    fileReferences?: string[];
+}
+export declare class AgentKnowledgeService {
+    private readonly agentRepository;
+    private readonly agentFileRepository;
+    private readonly binaryDataService;
+    constructor(agentRepository: AgentRepository, agentFileRepository: AgentFileRepository, binaryDataService: BinaryDataService);
+    uploadFiles(agentId: string, projectId: string, files: Express.Multer.File[]): Promise<AgentFileDto[]>;
+    listFiles(agentId: string, projectId: string): Promise<AgentFileDto[]>;
+    listWorkspaceFiles(agentId: string, projectId: string): Promise<KnowledgeWorkspaceFile[]>;
+    deleteFile(agentId: string, projectId: string, fileId: string): Promise<void>;
+    deleteAllFilesForAgent(agentId: string): Promise<void>;
+    resolveWorkspaceFiles(agentId: string, projectId: string, fileReferences?: string[]): Promise<KnowledgeWorkspaceFile[]>;
+    materializeWorkspace(agentId: string, projectId: string, workspaceRoot: string, options?: MaterializeWorkspaceOptions): Promise<KnowledgeWorkspaceFile[]>;
+    private ensureAgentBelongsToProject;
+    private storeFile;
+    private toDto;
+    private toWorkspaceFile;
+    private assertWorkspaceWithinLimits;
+    private filterFilesForWorkspace;
+    private getWorkspaceRelativePath;
+    private prepareStoredContent;
+    private isPdf;
+    private extractPdfText;
+    private validateMetadataLength;
+    private cleanupStoredFiles;
+    private cleanupUploadTempFiles;
+    private cleanupUploadTempFile;
+}
+export {};

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;
+}