n8n 2.34.3 → 2.35.0

package/dist/modules/agents/json-config/node-tool-ai-gateway.service.js

@@ -0,0 +1,34 @@
+"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.NodeToolAiGatewayService = void 0;
+const di_1 = require("@n8n/di");
+const node_types_1 = require("../../../node-types");
+const ai_gateway_service_1 = require("../../../services/ai-gateway.service");
+const reconcile_node_tool_gateway_credentials_1 = require("./reconcile-node-tool-gateway-credentials");
+let NodeToolAiGatewayService = class NodeToolAiGatewayService {
+    constructor(nodeTypes, aiGatewayService) {
+        this.nodeTypes = nodeTypes;
+        this.aiGatewayService = aiGatewayService;
+    }
+    async assignManagedCredentials(tools, ownedCredentialTypes) {
+        if (tools === undefined)
+            return;
+        const availability = await this.aiGatewayService.isAvailable();
+        (0, reconcile_node_tool_gateway_credentials_1.reconcileNodeToolGatewayCredentials)(tools, this.nodeTypes, availability.available ? availability.config : undefined, ownedCredentialTypes);
+    }
+};
+exports.NodeToolAiGatewayService = NodeToolAiGatewayService;
+exports.NodeToolAiGatewayService = NodeToolAiGatewayService = __decorate([
+    (0, di_1.Service)(),
+    __metadata("design:paramtypes", [node_types_1.NodeTypes, ai_gateway_service_1.AiGatewayService])
+], NodeToolAiGatewayService);
+//# sourceMappingURL=node-tool-ai-gateway.service.js.map

package/dist/modules/agents/json-config/node-tool-ai-gateway.service.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"node-tool-ai-gateway.service.js","sourceRoot":"","sources":["../../../../src/modules/agents/json-config/node-tool-ai-gateway.service.ts"],"names":[],"mappings":";;;;;;;;;;;;AACA,gCAAkC;AAElC,6CAAyC;AACzC,sEAAiE;AAEjE,uGAAgG;AAQzF,IAAM,wBAAwB,GAA9B,MAAM,wBAAwB;IACpC,YACkB,SAAoB,EACpB,gBAAkC;yBADlC,SAAS;gCACT,gBAAgB;IAC/B,CAAC;IAEJ,KAAK,CAAC,wBAAwB,CAC7B,KAAwC,EACxC,oBAAyC;QAEzC,IAAI,KAAK,KAAK,SAAS;YAAE,OAAO;QAChC,MAAM,YAAY,GAAG,MAAM,IAAI,CAAC,gBAAgB,CAAC,WAAW,EAAE,CAAC;QAC/D,IAAA,6EAAmC,EAClC,KAAK,EACL,IAAI,CAAC,SAAS,EACd,YAAY,CAAC,SAAS,CAAC,CAAC,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC,CAAC,SAAS,EACxD,oBAAoB,CACpB,CAAC;IACH,CAAC;CACD,CAAA;;mCAnBY,wBAAwB;IADpC,IAAA,YAAO,GAAE;qCAGoB,sBAAS,EACF,qCAAgB;GAHxC,wBAAwB,CAmBpC"}

package/dist/modules/agents/json-config/reconcile-node-tool-gateway-credentials.d.ts

@@ -0,0 +1,4 @@
+import type { AgentJsonToolConfig, AiGatewayConfigDto } from '@n8n/api-types';
+import type { NodeTypes } from '../../../node-types';
+export declare function reconcileNodeToolGatewayCredentials(tools: AgentJsonToolConfig[] | undefined, nodeTypes: NodeTypes, aiGatewayConfig: AiGatewayConfigDto | undefined, ownedCredentialTypes: ReadonlySet<string>): void;
+export declare function listAiGatewayManagedCredentialTypes(tools: AgentJsonToolConfig[] | undefined, nodeTypes: NodeTypes): string[];

package/dist/modules/agents/json-config/reconcile-node-tool-gateway-credentials.js

@@ -0,0 +1,152 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.reconcileNodeToolGatewayCredentials = reconcileNodeToolGatewayCredentials;
+exports.listAiGatewayManagedCredentialTypes = listAiGatewayManagedCredentialTypes;
+const node_catalog_1 = require("@n8n/ai-utilities/node-catalog");
+const n8n_workflow_1 = require("n8n-workflow");
+const ai_gateway_eligibility_1 = require("../../../services/ai-gateway-eligibility");
+const N8N_CONNECT_CREDENTIAL_NAME = 'n8n credits';
+const AI_GATEWAY_MANAGED_CREDENTIAL_FLAG = '__aiGatewayManaged';
+function aiGatewayManagedCredential() {
+    return {
+        id: null,
+        name: N8N_CONNECT_CREDENTIAL_NAME,
+        [AI_GATEWAY_MANAGED_CREDENTIAL_FLAG]: true,
+    };
+}
+function setAiGatewayManagedCredential(node, credentialType, activationParameters) {
+    node.credentials = node.credentials ?? {};
+    node.credentials[credentialType] = aiGatewayManagedCredential();
+    if (activationParameters && Object.keys(activationParameters).length > 0) {
+        node.nodeParameters = { ...(node.nodeParameters ?? {}), ...activationParameters };
+    }
+}
+function resolveNodeParameters(node, description) {
+    const rawParameters = (node.nodeParameters ?? {});
+    return (n8n_workflow_1.NodeHelpers.getNodeParameters(description.properties, rawParameters, true, false, { typeVersion: node.nodeTypeVersion }, description) ?? rawParameters);
+}
+function isCredentialDisplayed(node, description, credentialType, resolvedParameters) {
+    const credentialDefinition = description.credentials?.find((c) => c.name === credentialType);
+    if (!credentialDefinition)
+        return true;
+    return n8n_workflow_1.NodeHelpers.displayParameter(resolvedParameters, credentialDefinition, { typeVersion: node.nodeTypeVersion }, description);
+}
+function reconcileAiGatewayManagedMarkers(node, description, isEligible) {
+    if (!node.credentials)
+        return false;
+    let parametersChanged = false;
+    for (const [slot, ref] of Object.entries(node.credentials)) {
+        if (!(AI_GATEWAY_MANAGED_CREDENTIAL_FLAG in ref))
+            continue;
+        if (!isEligible(slot)) {
+            delete node.credentials[slot];
+            continue;
+        }
+        const activation = (0, n8n_workflow_1.resolveSupportedCredentialActivation)(description, { typeVersion: node.nodeTypeVersion, parameters: resolveNodeParameters(node, description) }, (candidateType) => candidateType === slot);
+        setAiGatewayManagedCredential(node, slot, activation?.parameters);
+        if (activation && Object.keys(activation.parameters).length > 0)
+            parametersChanged = true;
+    }
+    return parametersChanged;
+}
+function reconcileNodeToolGatewayCredentials(tools, nodeTypes, aiGatewayConfig, ownedCredentialTypes) {
+    for (const tool of tools ?? []) {
+        if (tool.type === 'node') {
+            reconcileNode(tool.node, nodeTypes, aiGatewayConfig, ownedCredentialTypes);
+        }
+    }
+}
+function listAiGatewayManagedCredentialTypes(tools, nodeTypes) {
+    const managed = new Set();
+    const unsatisfied = new Set();
+    for (const tool of tools ?? []) {
+        if (tool.type !== 'node')
+            continue;
+        const node = tool.node;
+        let description;
+        try {
+            description = nodeTypes.getByNameAndVersion(node.nodeType, node.nodeTypeVersion).description;
+        }
+        catch {
+            continue;
+        }
+        for (const [credentialType, ref] of Object.entries(node.credentials ?? {})) {
+            if (AI_GATEWAY_MANAGED_CREDENTIAL_FLAG in ref)
+                managed.add(credentialType);
+        }
+        const resolvedParameters = resolveNodeParameters(node, description);
+        for (const slot of (0, node_catalog_1.getRequiredNodeCredentialSlots)(description)) {
+            const credentialType = slot.credentialType;
+            const ref = node.credentials?.[credentialType];
+            if (ref && (ref.id || AI_GATEWAY_MANAGED_CREDENTIAL_FLAG in ref))
+                continue;
+            if (isCredentialDisplayed(node, description, credentialType, resolvedParameters)) {
+                unsatisfied.add(credentialType);
+            }
+        }
+    }
+    return [...managed].filter((credentialType) => !unsatisfied.has(credentialType));
+}
+function reconcileNode(node, nodeTypes, aiGatewayConfig, ownedCredentialTypes) {
+    let description;
+    try {
+        description = nodeTypes.getByNameAndVersion(node.nodeType, node.nodeTypeVersion).description;
+    }
+    catch {
+        dropAiGatewayMarkers(node);
+        return;
+    }
+    let resolvedParameters = resolveNodeParameters(node, description);
+    const eligibility = (credentialType) => aiGatewayConfig === undefined
+        ? { eligible: false, reason: 'nodeNotCovered' }
+        : (0, ai_gateway_eligibility_1.checkAiGatewayEligibility)({
+            type: node.nodeType,
+            typeVersion: node.nodeTypeVersion,
+            parameters: (node.nodeParameters ?? {}),
+        }, credentialType, aiGatewayConfig, resolvedParameters);
+    const isEligible = (credentialType) => eligibility(credentialType).eligible;
+    const assignSupportedSiblingCredential = (credentialType) => {
+        if (aiGatewayConfig === undefined)
+            return;
+        const activation = (0, n8n_workflow_1.resolveSupportedCredentialActivation)(description, { typeVersion: node.nodeTypeVersion, parameters: resolvedParameters }, (candidateType) => candidateType !== credentialType &&
+            !node.credentials?.[candidateType]?.id &&
+            !ownedCredentialTypes.has(candidateType) &&
+            isEligible(candidateType));
+        if (!activation)
+            return;
+        setAiGatewayManagedCredential(node, activation.credentialType, activation.parameters);
+        resolvedParameters = resolveNodeParameters(node, description);
+    };
+    if (reconcileAiGatewayManagedMarkers(node, description, isEligible)) {
+        resolvedParameters = resolveNodeParameters(node, description);
+    }
+    if (aiGatewayConfig === undefined)
+        return;
+    if (ai_gateway_eligibility_1.HTTP_NODE_TYPES.has(node.nodeType))
+        return;
+    for (const slot of (0, node_catalog_1.getRequiredNodeCredentialSlots)(description)) {
+        const credentialType = slot.credentialType;
+        const existing = node.credentials?.[credentialType];
+        if (existing && (existing.id || AI_GATEWAY_MANAGED_CREDENTIAL_FLAG in existing))
+            continue;
+        if (ownedCredentialTypes.has(credentialType))
+            continue;
+        if (!isCredentialDisplayed(node, description, credentialType, resolvedParameters))
+            continue;
+        const slotEligibility = eligibility(credentialType);
+        if (slotEligibility.eligible) {
+            setAiGatewayManagedCredential(node, credentialType);
+            continue;
+        }
+        assignSupportedSiblingCredential(credentialType);
+    }
+}
+function dropAiGatewayMarkers(node) {
+    if (!node.credentials)
+        return;
+    for (const [slot, ref] of Object.entries(node.credentials)) {
+        if (AI_GATEWAY_MANAGED_CREDENTIAL_FLAG in ref)
+            delete node.credentials[slot];
+    }
+}
+//# sourceMappingURL=reconcile-node-tool-gateway-credentials.js.map

package/dist/modules/agents/json-config/reconcile-node-tool-gateway-credentials.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"reconcile-node-tool-gateway-credentials.js","sourceRoot":"","sources":["../../../../src/modules/agents/json-config/reconcile-node-tool-gateway-credentials.ts"],"names":[],"mappings":";;;;AACA,iEAAgF;AAEhF,+CAAiF;AAGjF,8EAA+F;AAE/F,MAAM,2BAA2B,GAAG,aAAa,CAAC;AAClD,MAAM,kCAAkC,GAAG,oBAAoB,CAAC;AAIhE,SAAS,0BAA0B;IAClC,OAAO;QACN,EAAE,EAAE,IAAI;QACR,IAAI,EAAE,2BAA2B;QACjC,CAAC,kCAAkC,CAAC,EAAE,IAAI;KAC1C,CAAC;AACH,CAAC;AAED,SAAS,6BAA6B,CACrC,IAAoB,EACpB,cAAsB,EACtB,oBAAsC;IAEtC,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC,WAAW,IAAI,EAAE,CAAC;IAC1C,IAAI,CAAC,WAAW,CAAC,cAAc,CAAC,GAAG,0BAA0B,EAAE,CAAC;IAEhE,IAAI,oBAAoB,IAAI,MAAM,CAAC,IAAI,CAAC,oBAAoB,CAAC,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QAC1E,IAAI,CAAC,cAAc,GAAG,EAAE,GAAG,CAAC,IAAI,CAAC,cAAc,IAAI,EAAE,CAAC,EAAE,GAAG,oBAAoB,EAAE,CAAC;IACnF,CAAC;AACF,CAAC;AAED,SAAS,qBAAqB,CAC7B,IAAoB,EACpB,WAAiC;IAEjC,MAAM,aAAa,GAAG,CAAC,IAAI,CAAC,cAAc,IAAI,EAAE,CAAoB,CAAC;IAErE,OAAO,CACN,0BAAW,CAAC,iBAAiB,CAC5B,WAAW,CAAC,UAAU,EACtB,aAAa,EACb,IAAI,EACJ,KAAK,EACL,EAAE,WAAW,EAAE,IAAI,CAAC,eAAe,EAAE,EACrC,WAAW,CACX,IAAI,aAAa,CAClB,CAAC;AACH,CAAC;AAED,SAAS,qBAAqB,CAC7B,IAAoB,EACpB,WAAiC,EACjC,cAAsB,EACtB,kBAAmC;IAEnC,MAAM,oBAAoB,GAAG,WAAW,CAAC,WAAW,EAAE,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,cAAc,CAAC,CAAC;IAC7F,IAAI,CAAC,oBAAoB;QAAE,OAAO,IAAI,CAAC;IAEvC,OAAO,0BAAW,CAAC,gBAAgB,CAClC,kBAAkB,EAClB,oBAAoB,EACpB,EAAE,WAAW,EAAE,IAAI,CAAC,eAAe,EAAE,EACrC,WAAW,CACX,CAAC;AACH,CAAC;AAGD,SAAS,gCAAgC,CACxC,IAAoB,EACpB,WAAiC,EACjC,UAA+C;IAE/C,IAAI,CAAC,IAAI,CAAC,WAAW;QAAE,OAAO,KAAK,CAAC;IAEpC,IAAI,iBAAiB,GAAG,KAAK,CAAC;IAC9B,KAAK,MAAM,CAAC,IAAI,EAAE,GAAG,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,WAAW,CAAC,EAAE,CAAC;QAC5D,IAAI,CAAC,CAAC,kCAAkC,IAAI,GAAG,CAAC;YAAE,SAAS;QAC3D,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,EAAE,CAAC;YACvB,OAAO,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC;YAC9B,SAAS;QACV,CAAC;QAMD,MAAM,UAAU,GAAG,IAAA,mDAAoC,EACtD,WAAW,EACX,EAAE,WAAW,EAAE,IAAI,CAAC,eAAe,EAAE,UAAU,EAAE,qBAAqB,CAAC,IAAI,EAAE,WAAW,CAAC,EAAE,EAC3F,CAAC,aAAa,EAAE,EAAE,CAAC,aAAa,KAAK,IAAI,CACzC,CAAC;QACF,6BAA6B,CAAC,IAAI,EAAE,IAAI,EAAE,UAAU,EAAE,UAAU,CAAC,CAAC;QAClE,IAAI,UAAU,IAAI,MAAM,CAAC,IAAI,CAAC,UAAU,CAAC,UAAU,CAAC,CAAC,MAAM,GAAG,CAAC;YAAE,iBAAiB,GAAG,IAAI,CAAC;IAC3F,CAAC;IACD,OAAO,iBAAiB,CAAC;AAC1B,CAAC;AAWD,6CACC,KAAwC,EACxC,SAAoB,EACpB,eAA+C,EAC/C,oBAAyC;IAEzC,KAAK,MAAM,IAAI,IAAI,KAAK,IAAI,EAAE,EAAE,CAAC;QAChC,IAAI,IAAI,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;YAC1B,aAAa,CAAC,IAAI,CAAC,IAAI,EAAE,SAAS,EAAE,eAAe,EAAE,oBAAoB,CAAC,CAAC;QAC5E,CAAC;IACF,CAAC;AACF,CAAC;AAYD,6CACC,KAAwC,EACxC,SAAoB;IAEpB,MAAM,OAAO,GAAG,IAAI,GAAG,EAAU,CAAC;IAClC,MAAM,WAAW,GAAG,IAAI,GAAG,EAAU,CAAC;IAEtC,KAAK,MAAM,IAAI,IAAI,KAAK,IAAI,EAAE,EAAE,CAAC;QAChC,IAAI,IAAI,CAAC,IAAI,KAAK,MAAM;YAAE,SAAS;QACnC,MAAM,IAAI,GAAG,IAAI,CAAC,IAAI,CAAC;QAEvB,IAAI,WAAiC,CAAC;QACtC,IAAI,CAAC;YACJ,WAAW,GAAG,SAAS,CAAC,mBAAmB,CAAC,IAAI,CAAC,QAAQ,EAAE,IAAI,CAAC,eAAe,CAAC,CAAC,WAAW,CAAC;QAC9F,CAAC;QAAC,MAAM,CAAC;YACR,SAAS;QACV,CAAC;QAED,KAAK,MAAM,CAAC,cAAc,EAAE,GAAG,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,WAAW,IAAI,EAAE,CAAC,EAAE,CAAC;YAC5E,IAAI,kCAAkC,IAAI,GAAG;gBAAE,OAAO,CAAC,GAAG,CAAC,cAAc,CAAC,CAAC;QAC5E,CAAC;QAED,MAAM,kBAAkB,GAAG,qBAAqB,CAAC,IAAI,EAAE,WAAW,CAAC,CAAC;QACpE,KAAK,MAAM,IAAI,IAAI,IAAA,6CAA8B,EAAC,WAAW,CAAC,EAAE,CAAC;YAChE,MAAM,cAAc,GAAG,IAAI,CAAC,cAAc,CAAC;YAC3C,MAAM,GAAG,GAAG,IAAI,CAAC,WAAW,EAAE,CAAC,cAAc,CAAC,CAAC;YAC/C,IAAI,GAAG,IAAI,CAAC,GAAG,CAAC,EAAE,IAAI,kCAAkC,IAAI,GAAG,CAAC;gBAAE,SAAS;YAC3E,IAAI,qBAAqB,CAAC,IAAI,EAAE,WAAW,EAAE,cAAc,EAAE,kBAAkB,CAAC,EAAE,CAAC;gBAClF,WAAW,CAAC,GAAG,CAAC,cAAc,CAAC,CAAC;YACjC,CAAC;QACF,CAAC;IACF,CAAC;IAED,OAAO,CAAC,GAAG,OAAO,CAAC,CAAC,MAAM,CAAC,CAAC,cAAc,EAAE,EAAE,CAAC,CAAC,WAAW,CAAC,GAAG,CAAC,cAAc,CAAC,CAAC,CAAC;AAClF,CAAC;AAED,SAAS,aAAa,CACrB,IAAoB,EACpB,SAAoB,EACpB,eAA+C,EAC/C,oBAAyC;IAEzC,IAAI,WAAiC,CAAC;IACtC,IAAI,CAAC;QACJ,WAAW,GAAG,SAAS,CAAC,mBAAmB,CAAC,IAAI,CAAC,QAAQ,EAAE,IAAI,CAAC,eAAe,CAAC,CAAC,WAAW,CAAC;IAC9F,CAAC;IAAC,MAAM,CAAC;QACR,oBAAoB,CAAC,IAAI,CAAC,CAAC;QAC3B,OAAO;IACR,CAAC;IAED,IAAI,kBAAkB,GAAG,qBAAqB,CAAC,IAAI,EAAE,WAAW,CAAC,CAAC;IAElE,MAAM,WAAW,GAAG,CAAC,cAAsB,EAAE,EAAE,CAC9C,eAAe,KAAK,SAAS;QAC5B,CAAC,CAAE,EAAE,QAAQ,EAAE,KAAK,EAAE,MAAM,EAAE,gBAAgB,EAAY;QAC1D,CAAC,CAAC,IAAA,kDAAyB,EACzB;YACC,IAAI,EAAE,IAAI,CAAC,QAAQ;YACnB,WAAW,EAAE,IAAI,CAAC,eAAe;YACjC,UAAU,EAAE,CAAC,IAAI,CAAC,cAAc,IAAI,EAAE,CAAoB;SAC1D,EACD,cAAc,EACd,eAAe,EACf,kBAAkB,CAClB,CAAC;IAEL,MAAM,UAAU,GAAG,CAAC,cAAsB,EAAW,EAAE,CAAC,WAAW,CAAC,cAAc,CAAC,CAAC,QAAQ,CAAC;IAE7F,MAAM,gCAAgC,GAAG,CAAC,cAAsB,EAAQ,EAAE;QACzE,IAAI,eAAe,KAAK,SAAS;YAAE,OAAO;QAE1C,MAAM,UAAU,GAAG,IAAA,mDAAoC,EACtD,WAAW,EACX,EAAE,WAAW,EAAE,IAAI,CAAC,eAAe,EAAE,UAAU,EAAE,kBAAkB,EAAE,EACrE,CAAC,aAAa,EAAE,EAAE,CACjB,aAAa,KAAK,cAAc;YAChC,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC,aAAa,CAAC,EAAE,EAAE;YACtC,CAAC,oBAAoB,CAAC,GAAG,CAAC,aAAa,CAAC;YACxC,UAAU,CAAC,aAAa,CAAC,CAC1B,CAAC;QAEF,IAAI,CAAC,UAAU;YAAE,OAAO;QAExB,6BAA6B,CAAC,IAAI,EAAE,UAAU,CAAC,cAAc,EAAE,UAAU,CAAC,UAAU,CAAC,CAAC;QACtF,kBAAkB,GAAG,qBAAqB,CAAC,IAAI,EAAE,WAAW,CAAC,CAAC;IAC/D,CAAC,CAAC;IAEF,IAAI,gCAAgC,CAAC,IAAI,EAAE,WAAW,EAAE,UAAU,CAAC,EAAE,CAAC;QACrE,kBAAkB,GAAG,qBAAqB,CAAC,IAAI,EAAE,WAAW,CAAC,CAAC;IAC/D,CAAC;IAED,IAAI,eAAe,KAAK,SAAS;QAAE,OAAO;IAC1C,IAAI,wCAAe,CAAC,GAAG,CAAC,IAAI,CAAC,QAAQ,CAAC;QAAE,OAAO;IAE/C,KAAK,MAAM,IAAI,IAAI,IAAA,6CAA8B,EAAC,WAAW,CAAC,EAAE,CAAC;QAChE,MAAM,cAAc,GAAG,IAAI,CAAC,cAAc,CAAC;QAG3C,MAAM,QAAQ,GAAG,IAAI,CAAC,WAAW,EAAE,CAAC,cAAc,CAAC,CAAC;QACpD,IAAI,QAAQ,IAAI,CAAC,QAAQ,CAAC,EAAE,IAAI,kCAAkC,IAAI,QAAQ,CAAC;YAAE,SAAS;QAI1F,IAAI,oBAAoB,CAAC,GAAG,CAAC,cAAc,CAAC;YAAE,SAAS;QAEvD,IAAI,CAAC,qBAAqB,CAAC,IAAI,EAAE,WAAW,EAAE,cAAc,EAAE,kBAAkB,CAAC;YAAE,SAAS;QAE5F,MAAM,eAAe,GAAG,WAAW,CAAC,cAAc,CAAC,CAAC;QACpD,IAAI,eAAe,CAAC,QAAQ,EAAE,CAAC;YAC9B,6BAA6B,CAAC,IAAI,EAAE,cAAc,CAAC,CAAC;YACpD,SAAS;QACV,CAAC;QAED,gCAAgC,CAAC,cAAc,CAAC,CAAC;IAClD,CAAC;AACF,CAAC;AAED,SAAS,oBAAoB,CAAC,IAAoB;IACjD,IAAI,CAAC,IAAI,CAAC,WAAW;QAAE,OAAO;IAC9B,KAAK,MAAM,CAAC,IAAI,EAAE,GAAG,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,WAAW,CAAC,EAAE,CAAC;QAC5D,IAAI,kCAAkC,IAAI,GAAG;YAAE,OAAO,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC;IAC9E,CAAC;AACF,CAAC"}

package/dist/modules/agents/repositories/agent-credential-dependency.repository.d.ts

@@ -0,0 +1,12 @@
+import { BaseRepository, TransactionRunner } from '@n8n/db';
+import { DataSource } from '@n8n/typeorm';
+import { AgentCredentialDependency } from '../entities/agent-credential-dependency.entity';
+export type AgentCredentialDependencyReference = Pick<AgentCredentialDependency, 'agentId' | 'credentialId'>;
+export declare class AgentCredentialDependencyRepository extends BaseRepository<AgentCredentialDependency> {
+    private readonly txRunner;
+    constructor(dataSource: DataSource, txRunner: TransactionRunner);
+    refreshForAgent(agentId: string): Promise<void>;
+    removeForAgent(agentId: string): Promise<void>;
+    findByCredentialIds(credentialIds: string[]): Promise<AgentCredentialDependencyReference[]>;
+    private findExistingCredentialIds;
+}

package/dist/modules/agents/repositories/agent-credential-dependency.repository.js

@@ -0,0 +1,92 @@
+"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.AgentCredentialDependencyRepository = void 0;
+const db_1 = require("@n8n/db");
+const di_1 = require("@n8n/di");
+const typeorm_1 = require("@n8n/typeorm");
+const agent_credential_dependency_entity_1 = require("../entities/agent-credential-dependency.entity");
+const agent_history_entity_1 = require("../entities/agent-history.entity");
+const agent_entity_1 = require("../entities/agent.entity");
+const extract_agent_credential_ids_1 = require("../utils/extract-agent-credential-ids");
+let AgentCredentialDependencyRepository = class AgentCredentialDependencyRepository extends db_1.BaseRepository {
+    constructor(dataSource, txRunner) {
+        super(agent_credential_dependency_entity_1.AgentCredentialDependency, dataSource.manager);
+        this.txRunner = txRunner;
+    }
+    async refreshForAgent(agentId) {
+        await this.txRunner.run({}, async (ctx) => {
+            const manager = this.managerFor(ctx);
+            const isPostgres = manager.connection.options.type === 'postgres';
+            if (!isPostgres) {
+                await manager.delete(agent_credential_dependency_entity_1.AgentCredentialDependency, { agentId });
+            }
+            const agent = isPostgres
+                ? await manager.findOne(agent_entity_1.Agent, {
+                    where: { id: agentId },
+                    lock: { mode: 'pessimistic_write' },
+                })
+                : await manager.findOne(agent_entity_1.Agent, { where: { id: agentId } });
+            if (agent === null) {
+                if (isPostgres)
+                    await manager.delete(agent_credential_dependency_entity_1.AgentCredentialDependency, { agentId });
+                return;
+            }
+            const publishedVersion = agent.activeVersionId === null
+                ? null
+                : await manager.findOne(agent_history_entity_1.AgentHistory, {
+                    where: { versionId: agent.activeVersionId, agentId },
+                });
+            const draftCredentialIds = (0, extract_agent_credential_ids_1.extractAgentCredentialIds)([agent.schema, agent.integrations]);
+            const publishedCredentialIds = (0, extract_agent_credential_ids_1.extractAgentCredentialIds)(publishedVersion?.schema);
+            const referencedCredentialIds = new Set([...draftCredentialIds, ...publishedCredentialIds]);
+            const existingCredentialIds = await this.findExistingCredentialIds(manager, referencedCredentialIds, isPostgres);
+            if (isPostgres) {
+                await manager.delete(agent_credential_dependency_entity_1.AgentCredentialDependency, { agentId });
+            }
+            const rows = [...referencedCredentialIds]
+                .filter((credentialId) => existingCredentialIds.has(credentialId))
+                .map((credentialId) => ({ agentId, credentialId }));
+            if (rows.length > 0) {
+                await manager.insert(agent_credential_dependency_entity_1.AgentCredentialDependency, rows);
+            }
+        });
+    }
+    async removeForAgent(agentId) {
+        await this.managerFor({}).delete(agent_credential_dependency_entity_1.AgentCredentialDependency, { agentId });
+    }
+    async findByCredentialIds(credentialIds) {
+        if (credentialIds.length === 0)
+            return [];
+        return await this.find({
+            select: ['agentId', 'credentialId'],
+            where: { credentialId: (0, typeorm_1.In)(credentialIds) },
+        });
+    }
+    async findExistingCredentialIds(manager, credentialIds, lockForUpdate) {
+        if (credentialIds.size === 0)
+            return new Set();
+        const query = manager
+            .createQueryBuilder(db_1.CredentialsEntity, 'credential')
+            .select(['credential.id'])
+            .where('credential.id IN (:...credentialIds)', { credentialIds: [...credentialIds] });
+        query.orderBy('credential.id', 'ASC');
+        if (lockForUpdate)
+            query.setLock('pessimistic_write');
+        return new Set((await query.getMany()).map(({ id }) => id));
+    }
+};
+exports.AgentCredentialDependencyRepository = AgentCredentialDependencyRepository;
+exports.AgentCredentialDependencyRepository = AgentCredentialDependencyRepository = __decorate([
+    (0, di_1.Service)(),
+    __metadata("design:paramtypes", [typeorm_1.DataSource, db_1.TransactionRunner])
+], AgentCredentialDependencyRepository);
+//# sourceMappingURL=agent-credential-dependency.repository.js.map

package/dist/modules/agents/repositories/agent-credential-dependency.repository.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"agent-credential-dependency.repository.js","sourceRoot":"","sources":["../../../../src/modules/agents/repositories/agent-credential-dependency.repository.ts"],"names":[],"mappings":";;;;;;;;;;;;AAAA,gCAA+E;AAC/E,gCAAkC;AAClC,0CAAkE;AAElE,uGAA2F;AAC3F,2EAAgE;AAChE,2DAAiD;AACjD,wFAAkF;AAQ3E,IAAM,mCAAmC,GAAzC,MAAM,mCAAoC,SAAQ,mBAAyC;IACjG,YACC,UAAsB,EACL,QAA2B;QAE5C,KAAK,CAAC,8DAAyB,EAAE,UAAU,CAAC,OAAO,CAAC,CAAC;wBAFpC,QAAQ;IAG1B,CAAC;IAED,KAAK,CAAC,eAAe,CAAC,OAAe;QACpC,MAAM,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,EAAE,KAAK,EAAE,GAAG,EAAE,EAAE;YACzC,MAAM,OAAO,GAAG,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC;YACrC,MAAM,UAAU,GAAG,OAAO,CAAC,UAAU,CAAC,OAAO,CAAC,IAAI,KAAK,UAAU,CAAC;YAGlE,IAAI,CAAC,UAAU,EAAE,CAAC;gBACjB,MAAM,OAAO,CAAC,MAAM,CAAC,8DAAyB,EAAE,EAAE,OAAO,EAAE,CAAC,CAAC;YAC9D,CAAC;YAED,MAAM,KAAK,GAAG,UAAU;gBACvB,CAAC,CAAC,MAAM,OAAO,CAAC,OAAO,CAAC,oBAAK,EAAE;oBAC7B,KAAK,EAAE,EAAE,EAAE,EAAE,OAAO,EAAE;oBACtB,IAAI,EAAE,EAAE,IAAI,EAAE,mBAAmB,EAAE;iBACnC,CAAC;gBACH,CAAC,CAAC,MAAM,OAAO,CAAC,OAAO,CAAC,oBAAK,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE,EAAE,OAAO,EAAE,EAAE,CAAC,CAAC;YAE5D,IAAI,KAAK,KAAK,IAAI,EAAE,CAAC;gBACpB,IAAI,UAAU;oBAAE,MAAM,OAAO,CAAC,MAAM,CAAC,8DAAyB,EAAE,EAAE,OAAO,EAAE,CAAC,CAAC;gBAC7E,OAAO;YACR,CAAC;YAED,MAAM,gBAAgB,GACrB,KAAK,CAAC,eAAe,KAAK,IAAI;gBAC7B,CAAC,CAAC,IAAI;gBACN,CAAC,CAAC,MAAM,OAAO,CAAC,OAAO,CAAC,mCAAY,EAAE;oBACpC,KAAK,EAAE,EAAE,SAAS,EAAE,KAAK,CAAC,eAAe,EAAE,OAAO,EAAE;iBACpD,CAAC,CAAC;YACN,MAAM,kBAAkB,GAAG,IAAA,wDAAyB,EAAC,CAAC,KAAK,CAAC,MAAM,EAAE,KAAK,CAAC,YAAY,CAAC,CAAC,CAAC;YACzF,MAAM,sBAAsB,GAAG,IAAA,wDAAyB,EAAC,gBAAgB,EAAE,MAAM,CAAC,CAAC;YACnF,MAAM,uBAAuB,GAAG,IAAI,GAAG,CAAC,CAAC,GAAG,kBAAkB,EAAE,GAAG,sBAAsB,CAAC,CAAC,CAAC;YAE5F,MAAM,qBAAqB,GAAG,MAAM,IAAI,CAAC,yBAAyB,CACjE,OAAO,EACP,uBAAuB,EACvB,UAAU,CACV,CAAC;YAEF,IAAI,UAAU,EAAE,CAAC;gBAChB,MAAM,OAAO,CAAC,MAAM,CAAC,8DAAyB,EAAE,EAAE,OAAO,EAAE,CAAC,CAAC;YAC9D,CAAC;YAED,MAAM,IAAI,GAAG,CAAC,GAAG,uBAAuB,CAAC;iBACvC,MAAM,CAAC,CAAC,YAAY,EAAE,EAAE,CAAC,qBAAqB,CAAC,GAAG,CAAC,YAAY,CAAC,CAAC;iBACjE,GAAG,CAAC,CAAC,YAAY,EAAE,EAAE,CAAC,CAAC,EAAE,OAAO,EAAE,YAAY,EAAE,CAAC,CAAC,CAAC;YAErD,IAAI,IAAI,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBACrB,MAAM,OAAO,CAAC,MAAM,CAAC,8DAAyB,EAAE,IAAI,CAAC,CAAC;YACvD,CAAC;QACF,CAAC,CAAC,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,cAAc,CAAC,OAAe;QACnC,MAAM,IAAI,CAAC,UAAU,CAAC,EAAE,CAAC,CAAC,MAAM,CAAC,8DAAyB,EAAE,EAAE,OAAO,EAAE,CAAC,CAAC;IAC1E,CAAC;IAED,KAAK,CAAC,mBAAmB,CACxB,aAAuB;QAEvB,IAAI,aAAa,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO,EAAE,CAAC;QAE1C,OAAO,MAAM,IAAI,CAAC,IAAI,CAAC;YACtB,MAAM,EAAE,CAAC,SAAS,EAAE,cAAc,CAAC;YACnC,KAAK,EAAE,EAAE,YAAY,EAAE,IAAA,YAAE,EAAC,aAAa,CAAC,EAAE;SAC1C,CAAC,CAAC;IACJ,CAAC;IAEO,KAAK,CAAC,yBAAyB,CACtC,OAAsB,EACtB,aAAkC,EAClC,aAAsB;QAEtB,IAAI,aAAa,CAAC,IAAI,KAAK,CAAC;YAAE,OAAO,IAAI,GAAG,EAAE,CAAC;QAE/C,MAAM,KAAK,GAAG,OAAO;aACnB,kBAAkB,CAAC,sBAAiB,EAAE,YAAY,CAAC;aACnD,MAAM,CAAC,CAAC,eAAe,CAAC,CAAC;aACzB,KAAK,CAAC,sCAAsC,EAAE,EAAE,aAAa,EAAE,CAAC,GAAG,aAAa,CAAC,EAAE,CAAC,CAAC;QACvF,KAAK,CAAC,OAAO,CAAC,eAAe,EAAE,KAAK,CAAC,CAAC;QACtC,IAAI,aAAa;YAAE,KAAK,CAAC,OAAO,CAAC,mBAAmB,CAAC,CAAC;QAEtD,OAAO,IAAI,GAAG,CAAC,CAAC,MAAM,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC;IAC7D,CAAC;CACD,CAAA;;8CA3FY,mCAAmC;IAD/C,IAAA,YAAO,GAAE;qCAGI,oBAAU,EACK,sBAAiB;GAHjC,mCAAmC,CA2F/C"}

package/dist/modules/agents/repositories/agent-execution.repository.d.ts

@@ -1,8 +1,14 @@
 import { DataSource, Repository } from '@n8n/typeorm';
 import { AgentExecution } from '../entities/agent-execution.entity';
+export type RunningAgentExecution = Pick<AgentExecution, 'id' | 'threadId' | 'startedAt' | 'updatedAt' | 'timeline'>;
+type AgentExecutionFinalizationValues = Pick<AgentExecution, 'status' | 'stoppedAt' | 'duration' | 'timeline' | 'storedAt' | 'error'> & Partial<Pick<AgentExecution, 'model' | 'promptTokens' | 'completionTokens' | 'totalTokens' | 'cost' | 'hitlStatus'>>;
 export declare class AgentExecutionRepository extends Repository<AgentExecution> {
     constructor(dataSource: DataSource);
     findByThreadIdOrdered(threadId: string): Promise<AgentExecution[]>;
+    findRunning(): Promise<RunningAgentExecution[]>;
+    touchRunning(executionId: string): Promise<void>;
+    updateTimelineIfRunning(executionId: string, timeline: AgentExecution['timeline']): Promise<boolean>;
+    updateIfRunning(executionId: string, values: AgentExecutionFinalizationValues): Promise<boolean>;
     findFirstUserMessageByThreadIds(threadIds: string[]): Promise<Map<string, string>>;
     findFirstSourceByThreadIds(threadIds: string[]): Promise<Map<string, string>>;
     findSuspendedWithoutModel(threadId: string): Promise<AgentExecution[]>;
@@ -12,3 +18,4 @@ export declare class AgentExecutionRepository extends Repository<AgentExecution>
     findBlobRefsByThreadId(threadId: string): Promise<Array<Pick<AgentExecution, 'id' | 'storedAt'>>>;
     findBlobRefsByAgentId(agentId: string): Promise<Array<Pick<AgentExecution, 'id' | 'threadId' | 'storedAt'>>>;
 }
+export {};

package/dist/modules/agents/repositories/agent-execution.repository.js

@@ -20,6 +20,26 @@ let AgentExecutionRepository = class AgentExecutionRepository extends typeorm_1.
     async findByThreadIdOrdered(threadId) {
         return await this.find({ where: { threadId }, order: { createdAt: 'ASC' } });
     }
+    async findRunning() {
+        return await this.find({
+            select: ['id', 'threadId', 'startedAt', 'updatedAt', 'timeline'],
+            where: { status: 'running' },
+        });
+    }
+    async touchRunning(executionId) {
+        await this.update({ id: executionId, status: 'running' }, { updatedAt: new Date() });
+    }
+    async updateTimelineIfRunning(executionId, timeline) {
+        const result = await this.update({ id: executionId, status: 'running' }, {
+            timeline,
+            updatedAt: new Date(),
+        });
+        return result.affected === 1;
+    }
+    async updateIfRunning(executionId, values) {
+        const result = await this.update({ id: executionId, status: 'running' }, values);
+        return result.affected === 1;
+    }
     async findFirstUserMessageByThreadIds(threadIds) {
         if (threadIds.length === 0)
             return new Map();

package/dist/modules/agents/repositories/agent-execution.repository.js.map

@@ -1 +1 @@
-{"version":3,"file":"agent-execution.repository.js","sourceRoot":"","sources":["../../../../src/modules/agents/repositories/agent-execution.repository.ts"],"names":[],"mappings":";;;;;;;;;;;;AAAA,gCAAkC;AAClC,0CAAmE;AAEnE,+EAAoE;AAG7D,IAAM,wBAAwB,GAA9B,MAAM,wBAAyB,SAAQ,oBAA0B;IACvE,YAAY,UAAsB;QACjC,KAAK,CAAC,uCAAc,EAAE,UAAU,CAAC,OAAO,CAAC,CAAC;IAC3C,CAAC;IAGD,KAAK,CAAC,qBAAqB,CAAC,QAAgB;QAC3C,OAAO,MAAM,IAAI,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,EAAE,QAAQ,EAAE,EAAE,KAAK,EAAE,EAAE,SAAS,EAAE,KAAK,EAAE,EAAE,CAAC,CAAC;IAC9E,CAAC;IAUD,KAAK,CAAC,+BAA+B,CAAC,SAAmB;QACxD,IAAI,SAAS,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO,IAAI,GAAG,EAAE,CAAC;QAM7C,MAAM,SAAS,GAAG,IAAI,CAAC,QAAQ,CAAC,SAAS,CAAC;QAC1C,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,kBAAkB,CAAC,GAAG,CAAC;aAC7C,MAAM,CAAC,CAAC,4BAA4B,EAAE,kCAAkC,CAAC,CAAC;aAC1E,KAAK,CAAC,iCAAiC,EAAE,EAAE,SAAS,EAAE,CAAC;aACvD,QAAQ,CAAC,6BAA6B,CAAC;aACvC,QAAQ,CAAC,yBAAyB,CAAC;aACnC,QAAQ,CACR,oDAAoD,SAAS,MAAM;YAClE,sEAAsE;YACtE,+BAA+B,CAChC;aACA,UAAU,EAA6C,CAAC;QAE1D,OAAO,IAAI,GAAG,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,QAAQ,EAAE,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC;IAC9D,CAAC;IAQD,KAAK,CAAC,0BAA0B,CAAC,SAAmB;QACnD,IAAI,SAAS,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO,IAAI,GAAG,EAAE,CAAC;QAM7C,MAAM,SAAS,GAAG,IAAI,CAAC,QAAQ,CAAC,SAAS,CAAC;QAC1C,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,kBAAkB,CAAC,GAAG,CAAC;aAC7C,MAAM,CAAC,CAAC,4BAA4B,EAAE,wBAAwB,CAAC,CAAC;aAChE,KAAK,CAAC,iCAAiC,EAAE,EAAE,SAAS,EAAE,CAAC;aACvD,QAAQ,CAAC,wBAAwB,CAAC;aAClC,QAAQ,CACR,oDAAoD,SAAS,MAAM;YAClE,iEAAiE,CAClE;aACA,UAAU,EAAwC,CAAC;QAErD,OAAO,IAAI,GAAG,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,QAAQ,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;IACzD,CAAC;IAOD,KAAK,CAAC,yBAAyB,CAAC,QAAgB;QAC/C,OAAO,MAAM,IAAI,CAAC,IAAI,CAAC;YACtB,KAAK,EAAE,EAAE,QAAQ,EAAE,UAAU,EAAE,WAAW,EAAE,KAAK,EAAE,IAAA,gBAAM,GAAE,EAAE;SAC7D,CAAC,CAAC;IACJ,CAAC;IAQD,KAAK,CAAC,6BAA6B,CAAC,QAAgB;QACnD,OAAO,MAAM,IAAI,CAAC,OAAO,CAAC;YACzB,KAAK,EAAE,EAAE,QAAQ,EAAE,UAAU,EAAE,WAAW,EAAE;YAC5C,KAAK,EAAE,EAAE,SAAS,EAAE,MAAM,EAAE;SAC5B,CAAC,CAAC;IACJ,CAAC;IAGD,KAAK,CAAC,aAAa,CAAC,YAAsB,EAAE,KAAa;QACxD,IAAI,YAAY,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO;QACtC,MAAM,IAAI,CAAC,kBAAkB,EAAE;aAC7B,MAAM,CAAC,uCAAc,CAAC;aACtB,GAAG,CAAC,EAAE,KAAK,EAAE,CAAC;aACd,UAAU,CAAC,YAAY,CAAC;aACxB,OAAO,EAAE,CAAC;IACb,CAAC;IAGD,KAAK,CAAC,gBAAgB,CAAC,QAAgB;QACtC,MAAM,IAAI,CAAC,MAAM,CAAC,EAAE,QAAQ,EAAE,CAAC,CAAC;IACjC,CAAC;IAGD,KAAK,CAAC,sBAAsB,CAC3B,QAAgB;QAEhB,OAAO,MAAM,IAAI,CAAC,IAAI,CAAC;YACtB,MAAM,EAAE,CAAC,IAAI,EAAE,UAAU,CAAC;YAC1B,KAAK,EAAE,EAAE,QAAQ,EAAE,QAAQ,EAAE,IAAA,aAAG,EAAC,IAAI,CAAC,EAAE;SACxC,CAAC,CAAC;IACJ,CAAC;IAGD,KAAK,CAAC,qBAAqB,CAC1B,OAAe;QAEf,OAAO,MAAM,IAAI,CAAC,IAAI,CAAC;YACtB,MAAM,EAAE,CAAC,IAAI,EAAE,UAAU,EAAE,UAAU,CAAC;YACtC,KAAK,EAAE,EAAE,MAAM,EAAE,EAAE,OAAO,EAAE,EAAE,QAAQ,EAAE,IAAA,aAAG,EAAC,IAAI,CAAC,EAAE;SACnD,CAAC,CAAC;IACJ,CAAC;CACD,CAAA;;mCA9HY,wBAAwB;IADpC,IAAA,YAAO,GAAE;qCAEe,oBAAU;GADtB,wBAAwB,CA8HpC"}
+{"version":3,"file":"agent-execution.repository.js","sourceRoot":"","sources":["../../../../src/modules/agents/repositories/agent-execution.repository.ts"],"names":[],"mappings":";;;;;;;;;;;;AAAA,gCAAkC;AAClC,0CAAmE;AAGnE,+EAAoE;AAmB7D,IAAM,wBAAwB,GAA9B,MAAM,wBAAyB,SAAQ,oBAA0B;IACvE,YAAY,UAAsB;QACjC,KAAK,CAAC,uCAAc,EAAE,UAAU,CAAC,OAAO,CAAC,CAAC;IAC3C,CAAC;IAGD,KAAK,CAAC,qBAAqB,CAAC,QAAgB;QAC3C,OAAO,MAAM,IAAI,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,EAAE,QAAQ,EAAE,EAAE,KAAK,EAAE,EAAE,SAAS,EAAE,KAAK,EAAE,EAAE,CAAC,CAAC;IAC9E,CAAC;IAED,KAAK,CAAC,WAAW;QAChB,OAAO,MAAM,IAAI,CAAC,IAAI,CAAC;YACtB,MAAM,EAAE,CAAC,IAAI,EAAE,UAAU,EAAE,WAAW,EAAE,WAAW,EAAE,UAAU,CAAC;YAChE,KAAK,EAAE,EAAE,MAAM,EAAE,SAAS,EAAE;SAC5B,CAAC,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,YAAY,CAAC,WAAmB;QACrC,MAAM,IAAI,CAAC,MAAM,CAAC,EAAE,EAAE,EAAE,WAAW,EAAE,MAAM,EAAE,SAAS,EAAE,EAAE,EAAE,SAAS,EAAE,IAAI,IAAI,EAAE,EAAE,CAAC,CAAC;IACtF,CAAC;IAED,KAAK,CAAC,uBAAuB,CAC5B,WAAmB,EACnB,QAAoC;QAEpC,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,EAAE,EAAE,EAAE,WAAW,EAAE,MAAM,EAAE,SAAS,EAAE,EAAE;YACxE,QAAQ;YACR,SAAS,EAAE,IAAI,IAAI,EAAE;SACqB,CAAC,CAAC;QAC7C,OAAO,MAAM,CAAC,QAAQ,KAAK,CAAC,CAAC;IAC9B,CAAC;IAED,KAAK,CAAC,eAAe,CACpB,WAAmB,EACnB,MAAwC;QAExC,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,MAAM,CAC/B,EAAE,EAAE,EAAE,WAAW,EAAE,MAAM,EAAE,SAAS,EAAE,EACtC,MAAgD,CAChD,CAAC;QACF,OAAO,MAAM,CAAC,QAAQ,KAAK,CAAC,CAAC;IAC9B,CAAC;IAUD,KAAK,CAAC,+BAA+B,CAAC,SAAmB;QACxD,IAAI,SAAS,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO,IAAI,GAAG,EAAE,CAAC;QAM7C,MAAM,SAAS,GAAG,IAAI,CAAC,QAAQ,CAAC,SAAS,CAAC;QAC1C,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,kBAAkB,CAAC,GAAG,CAAC;aAC7C,MAAM,CAAC,CAAC,4BAA4B,EAAE,kCAAkC,CAAC,CAAC;aAC1E,KAAK,CAAC,iCAAiC,EAAE,EAAE,SAAS,EAAE,CAAC;aACvD,QAAQ,CAAC,6BAA6B,CAAC;aACvC,QAAQ,CAAC,yBAAyB,CAAC;aACnC,QAAQ,CACR,oDAAoD,SAAS,MAAM;YAClE,sEAAsE;YACtE,+BAA+B,CAChC;aACA,UAAU,EAA6C,CAAC;QAE1D,OAAO,IAAI,GAAG,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,QAAQ,EAAE,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC;IAC9D,CAAC;IAQD,KAAK,CAAC,0BAA0B,CAAC,SAAmB;QACnD,IAAI,SAAS,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO,IAAI,GAAG,EAAE,CAAC;QAM7C,MAAM,SAAS,GAAG,IAAI,CAAC,QAAQ,CAAC,SAAS,CAAC;QAC1C,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,kBAAkB,CAAC,GAAG,CAAC;aAC7C,MAAM,CAAC,CAAC,4BAA4B,EAAE,wBAAwB,CAAC,CAAC;aAChE,KAAK,CAAC,iCAAiC,EAAE,EAAE,SAAS,EAAE,CAAC;aACvD,QAAQ,CAAC,wBAAwB,CAAC;aAClC,QAAQ,CACR,oDAAoD,SAAS,MAAM;YAClE,iEAAiE,CAClE;aACA,UAAU,EAAwC,CAAC;QAErD,OAAO,IAAI,GAAG,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,QAAQ,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;IACzD,CAAC;IAOD,KAAK,CAAC,yBAAyB,CAAC,QAAgB;QAC/C,OAAO,MAAM,IAAI,CAAC,IAAI,CAAC;YACtB,KAAK,EAAE,EAAE,QAAQ,EAAE,UAAU,EAAE,WAAW,EAAE,KAAK,EAAE,IAAA,gBAAM,GAAE,EAAE;SAC7D,CAAC,CAAC;IACJ,CAAC;IAQD,KAAK,CAAC,6BAA6B,CAAC,QAAgB;QACnD,OAAO,MAAM,IAAI,CAAC,OAAO,CAAC;YACzB,KAAK,EAAE,EAAE,QAAQ,EAAE,UAAU,EAAE,WAAW,EAAE;YAC5C,KAAK,EAAE,EAAE,SAAS,EAAE,MAAM,EAAE;SAC5B,CAAC,CAAC;IACJ,CAAC;IAGD,KAAK,CAAC,aAAa,CAAC,YAAsB,EAAE,KAAa;QACxD,IAAI,YAAY,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO;QACtC,MAAM,IAAI,CAAC,kBAAkB,EAAE;aAC7B,MAAM,CAAC,uCAAc,CAAC;aACtB,GAAG,CAAC,EAAE,KAAK,EAAE,CAAC;aACd,UAAU,CAAC,YAAY,CAAC;aACxB,OAAO,EAAE,CAAC;IACb,CAAC;IAGD,KAAK,CAAC,gBAAgB,CAAC,QAAgB;QACtC,MAAM,IAAI,CAAC,MAAM,CAAC,EAAE,QAAQ,EAAE,CAAC,CAAC;IACjC,CAAC;IAGD,KAAK,CAAC,sBAAsB,CAC3B,QAAgB;QAEhB,OAAO,MAAM,IAAI,CAAC,IAAI,CAAC;YACtB,MAAM,EAAE,CAAC,IAAI,EAAE,UAAU,CAAC;YAC1B,KAAK,EAAE,EAAE,QAAQ,EAAE,QAAQ,EAAE,IAAA,aAAG,EAAC,IAAI,CAAC,EAAE;SACxC,CAAC,CAAC;IACJ,CAAC;IAGD,KAAK,CAAC,qBAAqB,CAC1B,OAAe;QAEf,OAAO,MAAM,IAAI,CAAC,IAAI,CAAC;YACtB,MAAM,EAAE,CAAC,IAAI,EAAE,UAAU,EAAE,UAAU,CAAC;YACtC,KAAK,EAAE,EAAE,MAAM,EAAE,EAAE,OAAO,EAAE,EAAE,QAAQ,EAAE,IAAA,aAAG,EAAC,IAAI,CAAC,EAAE;SACnD,CAAC,CAAC;IACJ,CAAC;CACD,CAAA;;mCA/JY,wBAAwB;IADpC,IAAA,YAAO,GAAE;qCAEe,oBAAU;GADtB,wBAAwB,CA+JpC"}

package/dist/modules/agents/repositories/agent.repository.d.ts

@@ -22,6 +22,8 @@ export declare class AgentRepository extends Repository<Agent> {
     private applySorting;
     findByIdAndProjectId(id: string, projectId: string): Promise<Agent | null>;
     findById(id: string): Promise<Agent | null>;
+    findCredentialIndexAgentIdsBatch(afterId: string | null, batchSize: number): Promise<Array<Pick<Agent, 'id'>>>;
+    findSummariesByIds(ids: string[]): Promise<Array<Pick<Agent, 'id' | 'name' | 'projectId'>>>;
     findByIdInProjects(id: string, projectIds: string[]): Promise<Agent | null>;
     existsByIdAndProjectId(id: string, projectId: string): Promise<boolean>;
     findByIdsAndProjectId(ids: string[], projectId: string): Promise<Array<Pick<Agent, 'id' | 'activeVersionId'>>>;

package/dist/modules/agents/repositories/agent.repository.js

@@ -103,6 +103,24 @@ let AgentRepository = class AgentRepository extends typeorm_1.Repository {
             relations: { activeVersion: true },
         });
     }
+    async findCredentialIndexAgentIdsBatch(afterId, batchSize) {
+        const query = this.createQueryBuilder('agent')
+            .select(['agent.id'])
+            .orderBy('agent.id', 'ASC')
+            .take(batchSize);
+        if (afterId !== null) {
+            query.where('agent.id > :afterId', { afterId });
+        }
+        return await query.getMany();
+    }
+    async findSummariesByIds(ids) {
+        if (ids.length === 0)
+            return [];
+        return await this.find({
+            select: ['id', 'name', 'projectId'],
+            where: { id: (0, typeorm_1.In)(ids) },
+        });
+    }
     async findByIdInProjects(id, projectIds) {
         if (projectIds.length === 0)
             return null;

package/dist/modules/agents/repositories/agent.repository.js.map

@@ -1 +1 @@
-{"version":3,"file":"agent.repository.js","sourceRoot":"","sources":["../../../../src/modules/agents/repositories/agent.repository.ts"],"names":[],"mappings":";;;;;;;;;;;;AACA,gCAAkC;AAClC,0CAA2F;AAE3F,2DAAiD;AAe1C,IAAM,eAAe,GAArB,MAAM,eAAgB,SAAQ,oBAAiB;IACrD,YAAY,UAAsB;QACjC,KAAK,CAAC,oBAAK,EAAE,UAAU,CAAC,OAAO,CAAC,CAAC;IAClC,CAAC;IAED,KAAK,CAAC,eAAe,CAAC,SAAiB;QACtC,OAAO,MAAM,IAAI,CAAC,IAAI,CAAC;YACtB,KAAK,EAAE,EAAE,SAAS,EAAE;YACpB,SAAS,EAAE,EAAE,aAAa,EAAE,IAAI,EAAE;YAClC,KAAK,EAAE,EAAE,SAAS,EAAE,MAAM,EAAE;SAC5B,CAAC,CAAC;IACJ,CAAC;IAOD,KAAK,CAAC,yBAAyB,CAC9B,UAA2B,EAC3B,OAAO,GAAwB,EAAE;QAEjC,IAAI,UAAU,EAAE,MAAM,KAAK,CAAC;YAAE,OAAO,EAAE,CAAC;QAExC,MAAM,KAAK,GAAG,IAAI,CAAC,kBAAkB,CAAC,OAAO,CAAC;aAC5C,MAAM,CAAC;YACP,UAAU;YACV,YAAY;YACZ,iBAAiB;YACjB,uBAAuB;YACvB,sBAAsB;YACtB,iBAAiB;SACjB,CAAC;aACD,OAAO,CAAC,iBAAiB,EAAE,MAAM,CAAC,CAAC;QAErC,IAAI,UAAU,KAAK,IAAI,EAAE,CAAC;YACzB,KAAK,CAAC,KAAK,CAAC,qCAAqC,EAAE,EAAE,UAAU,EAAE,CAAC,CAAC;QACpE,CAAC;QACD,IAAI,OAAO,CAAC,KAAK,EAAE,CAAC;YACnB,KAAK,CAAC,QAAQ,CAAC,sCAAsC,EAAE,EAAE,KAAK,EAAE,IAAI,OAAO,CAAC,KAAK,GAAG,EAAE,CAAC,CAAC;QACzF,CAAC;QACD,IAAI,OAAO,CAAC,aAAa,EAAE,CAAC;YAC3B,KAAK,CAAC,QAAQ,CAAC,mCAAmC,CAAC,CAAC;QACrD,CAAC;QACD,IAAI,OAAO,CAAC,cAAc,EAAE,CAAC;YAC5B,KAAK,CAAC,QAAQ,CAAC,6BAA6B,EAAE,EAAE,cAAc,EAAE,OAAO,CAAC,cAAc,EAAE,CAAC,CAAC;QAC3F,CAAC;QACD,IAAI,OAAO,CAAC,KAAK,KAAK,SAAS,EAAE,CAAC;YACjC,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;QAC3B,CAAC;QAED,OAAO,MAAM,KAAK,CAAC,OAAO,EAAE,CAAC;IAC9B,CAAC;IAED,KAAK,CAAC,yBAAyB,CAC9B,UAA2B,EAC3B,OAA2B,EAC3B,EAAE,WAAW,GAAG,KAAK,EAAE,GAA8B,EAAE;QAEvD,IAAI,UAAU,EAAE,MAAM,KAAK,CAAC;YAAE,OAAO,EAAE,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC;QAE5D,MAAM,KAAK,GAAG,IAAI,CAAC,kBAAkB,CAAC,OAAO,CAAC,CAAC,iBAAiB,CAC/D,qBAAqB,EACrB,eAAe,CACf,CAAC;QAIF,IAAI,WAAW,EAAE,CAAC;YACjB,KAAK,CAAC,iBAAiB,CAAC,eAAe,EAAE,SAAS,CAAC,CAAC;QACrD,CAAC;QAED,IAAI,UAAU,KAAK,IAAI,EAAE,CAAC;YACzB,KAAK,CAAC,KAAK,CAAC,qCAAqC,EAAE,EAAE,UAAU,EAAE,CAAC,CAAC;QACpE,CAAC;QACD,IAAI,CAAC,YAAY,CAAC,KAAK,EAAE,OAAO,CAAC,MAAM,CAAC,CAAC;QACzC,IAAI,CAAC,YAAY,CAAC,KAAK,EAAE,OAAO,CAAC,MAAM,CAAC,CAAC;QACzC,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;QAE5C,MAAM,CAAC,IAAI,EAAE,KAAK,CAAC,GAAG,MAAM,KAAK,CAAC,eAAe,EAAE,CAAC;QACpD,OAAO,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC;IACxB,CAAC;IAEO,YAAY,CACnB,KAAgC,EAChC,MAAoC;QAEpC,IAAI,MAAM,EAAE,KAAK,EAAE,CAAC;YACnB,KAAK,CAAC,QAAQ,CAAC,sCAAsC,EAAE,EAAE,KAAK,EAAE,IAAI,MAAM,CAAC,KAAK,GAAG,EAAE,CAAC,CAAC;QACxF,CAAC;QACD,IAAI,MAAM,EAAE,cAAc,KAAK,SAAS,EAAE,CAAC;YAC1C,KAAK,CAAC,QAAQ,CAAC,wCAAwC,EAAE;gBACxD,cAAc,EAAE,MAAM,CAAC,cAAc;aACrC,CAAC,CAAC;QACJ,CAAC;IACF,CAAC;IAEO,YAAY,CACnB,KAAgC,EAChC,MAAqC;QAErC,MAAM,CAAC,KAAK,GAAG,WAAW,EAAE,SAAS,GAAG,MAAM,CAAC,GAAG,MAAM,EAAE,KAAK,CAAC,GAAG,CAAC,IAAI,EAAE,CAAC;QAC3E,MAAM,aAAa,GAAG,SAAS,CAAC,WAAW,EAAE,KAAK,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,CAAC;QAEzE,IAAI,KAAK,KAAK,MAAM,EAAE,CAAC;YACtB,KAAK;iBACH,SAAS,CAAC,mBAAmB,EAAE,kBAAkB,CAAC;iBAClD,OAAO,CAAC,kBAAkB,EAAE,aAAa,CAAC,CAAC;YAC7C,OAAO;QACR,CAAC;QAED,KAAK,CAAC,OAAO,CAAC,SAAS,KAAK,EAAE,EAAE,aAAa,CAAC,CAAC;IAChD,CAAC;IAWD,KAAK,CAAC,oBAAoB,CAAC,EAAU,EAAE,SAAiB;QACvD,OAAO,MAAM,IAAI,CAAC,OAAO,CAAC;YACzB,KAAK,EAAE,EAAE,EAAE,EAAE,SAAS,EAAE;YACxB,SAAS,EAAE,EAAE,aAAa,EAAE,IAAI,EAAE;SAClC,CAAC,CAAC;IACJ,CAAC;IAOD,KAAK,CAAC,QAAQ,CAAC,EAAU;QACxB,OAAO,MAAM,IAAI,CAAC,OAAO,CAAC;YACzB,KAAK,EAAE,EAAE,EAAE,EAAE;YACb,SAAS,EAAE,EAAE,aAAa,EAAE,IAAI,EAAE;SAClC,CAAC,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,kBAAkB,CAAC,EAAU,EAAE,UAAoB;QACxD,IAAI,UAAU,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO,IAAI,CAAC;QACzC,OAAO,MAAM,IAAI,CAAC,OAAO,CAAC;YACzB,KAAK,EAAE,EAAE,EAAE,EAAE,SAAS,EAAE,IAAA,YAAE,EAAC,UAAU,CAAC,EAAE;YACxC,SAAS,EAAE,EAAE,aAAa,EAAE,IAAI,EAAE;SAClC,CAAC,CAAC;IACJ,CAAC;IAGD,KAAK,CAAC,sBAAsB,CAAC,EAAU,EAAE,SAAiB;QACzD,OAAO,MAAM,IAAI,CAAC,MAAM,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,EAAE,SAAS,EAAE,EAAE,CAAC,CAAC;IACxD,CAAC;IAED,KAAK,CAAC,qBAAqB,CAC1B,GAAa,EACb,SAAiB;QAEjB,IAAI,GAAG,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO,EAAE,CAAC;QAChC,OAAO,MAAM,IAAI,CAAC,IAAI,CAAC;YACtB,MAAM,EAAE,CAAC,IAAI,EAAE,iBAAiB,CAAC;YACjC,KAAK,EAAE,EAAE,EAAE,EAAE,IAAA,YAAE,EAAC,GAAG,CAAC,EAAE,SAAS,EAAE;SACjC,CAAC,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,6BAA6B,CAClC,KAAmE;QAEnE,IAAI,KAAK,IAAI,KAAK,IAAI,KAAK,CAAC,GAAG,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO,EAAE,CAAC;QACxD,IAAI,YAAY,IAAI,KAAK,IAAI,KAAK,CAAC,UAAU,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO,EAAE,CAAC;QAEtE,MAAM,QAAQ,GACb,KAAK,IAAI,KAAK;YACb,CAAC,CAAC,EAAE,EAAE,EAAE,IAAA,YAAE,EAAC,KAAK,CAAC,GAAG,CAAC,EAAE;YACvB,CAAC,CAAC,YAAY,IAAI,KAAK;gBACtB,CAAC,CAAC,EAAE,SAAS,EAAE,IAAA,YAAE,EAAC,KAAK,CAAC,UAAU,CAAC,EAAE;gBACrC,CAAC,CAAC,SAAS,CAAC;QAEf,OAAO,MAAM,IAAI,CAAC,IAAI,CAAC;YACtB,MAAM,EAAE,CAAC,IAAI,EAAE,WAAW,EAAE,gBAAgB,CAAC;YAC7C,KAAK,EAAE,QAAQ;SACf,CAAC,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,iBAAiB,CAAC,QAAkB,EAAE,cAAuB;QAClE,IAAI,QAAQ,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO;QAClC,MAAM,IAAI,CAAC,MAAM,CAAC,EAAE,EAAE,EAAE,IAAA,YAAE,EAAC,QAAQ,CAAC,EAAE,EAAE,EAAE,cAAc,EAAE,CAAC,CAAC;IAC7D,CAAC;IAOD,KAAK,CAAC,mBAAmB,CAAC,EAAU,EAAE,WAAiB;QACtD,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,MAAM,CAC/B,EAAE,EAAE,EAAE,gBAAgB,EAAE,IAAA,gBAAM,GAAE,EAAE,EAClC,EAAE,gBAAgB,EAAE,WAAW,EAAE,CACjC,CAAC;QAEF,OAAO,CAAC,MAAM,CAAC,QAAQ,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC;IACnC,CAAC;IAED,KAAK,CAAC,aAAa;QAClB,OAAO,MAAM,IAAI,CAAC,kBAAkB,CAAC,OAAO,CAAC;aAC3C,kBAAkB,CAAC,qBAAqB,EAAE,eAAe,CAAC;aAC1D,OAAO,EAAE,CAAC;IACb,CAAC;IAcD,KAAK,CAAC,2BAA2B,CAChC,IAAY,EACZ,YAAoB,EACpB,SAAiB,EACjB,cAAsB;QAEtB,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,EAAE,SAAS,EAAE,EAAE,CAAC,CAAC;QACzD,OAAO,MAAM,CAAC,MAAM,CACnB,CAAC,KAAK,EAAE,EAAE,CACT,KAAK,CAAC,EAAE,KAAK,cAAc;YAC3B,CAAC,KAAK,CAAC,YAAY,IAAI,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,IAAI,IAAI,CAAC,CAAC,YAAY,KAAK,YAAY,CAAC,CAC3F,CAAC;IACH,CAAC;CACD,CAAA;;0BA3OY,eAAe;IAD3B,IAAA,YAAO,GAAE;qCAEe,oBAAU;GADtB,eAAe,CA2O3B"}
+{"version":3,"file":"agent.repository.js","sourceRoot":"","sources":["../../../../src/modules/agents/repositories/agent.repository.ts"],"names":[],"mappings":";;;;;;;;;;;;AACA,gCAAkC;AAClC,0CAA2F;AAE3F,2DAAiD;AAe1C,IAAM,eAAe,GAArB,MAAM,eAAgB,SAAQ,oBAAiB;IACrD,YAAY,UAAsB;QACjC,KAAK,CAAC,oBAAK,EAAE,UAAU,CAAC,OAAO,CAAC,CAAC;IAClC,CAAC;IAED,KAAK,CAAC,eAAe,CAAC,SAAiB;QACtC,OAAO,MAAM,IAAI,CAAC,IAAI,CAAC;YACtB,KAAK,EAAE,EAAE,SAAS,EAAE;YACpB,SAAS,EAAE,EAAE,aAAa,EAAE,IAAI,EAAE;YAClC,KAAK,EAAE,EAAE,SAAS,EAAE,MAAM,EAAE;SAC5B,CAAC,CAAC;IACJ,CAAC;IAOD,KAAK,CAAC,yBAAyB,CAC9B,UAA2B,EAC3B,OAAO,GAAwB,EAAE;QAEjC,IAAI,UAAU,EAAE,MAAM,KAAK,CAAC;YAAE,OAAO,EAAE,CAAC;QAExC,MAAM,KAAK,GAAG,IAAI,CAAC,kBAAkB,CAAC,OAAO,CAAC;aAC5C,MAAM,CAAC;YACP,UAAU;YACV,YAAY;YACZ,iBAAiB;YACjB,uBAAuB;YACvB,sBAAsB;YACtB,iBAAiB;SACjB,CAAC;aACD,OAAO,CAAC,iBAAiB,EAAE,MAAM,CAAC,CAAC;QAErC,IAAI,UAAU,KAAK,IAAI,EAAE,CAAC;YACzB,KAAK,CAAC,KAAK,CAAC,qCAAqC,EAAE,EAAE,UAAU,EAAE,CAAC,CAAC;QACpE,CAAC;QACD,IAAI,OAAO,CAAC,KAAK,EAAE,CAAC;YACnB,KAAK,CAAC,QAAQ,CAAC,sCAAsC,EAAE,EAAE,KAAK,EAAE,IAAI,OAAO,CAAC,KAAK,GAAG,EAAE,CAAC,CAAC;QACzF,CAAC;QACD,IAAI,OAAO,CAAC,aAAa,EAAE,CAAC;YAC3B,KAAK,CAAC,QAAQ,CAAC,mCAAmC,CAAC,CAAC;QACrD,CAAC;QACD,IAAI,OAAO,CAAC,cAAc,EAAE,CAAC;YAC5B,KAAK,CAAC,QAAQ,CAAC,6BAA6B,EAAE,EAAE,cAAc,EAAE,OAAO,CAAC,cAAc,EAAE,CAAC,CAAC;QAC3F,CAAC;QACD,IAAI,OAAO,CAAC,KAAK,KAAK,SAAS,EAAE,CAAC;YACjC,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;QAC3B,CAAC;QAED,OAAO,MAAM,KAAK,CAAC,OAAO,EAAE,CAAC;IAC9B,CAAC;IAED,KAAK,CAAC,yBAAyB,CAC9B,UAA2B,EAC3B,OAA2B,EAC3B,EAAE,WAAW,GAAG,KAAK,EAAE,GAA8B,EAAE;QAEvD,IAAI,UAAU,EAAE,MAAM,KAAK,CAAC;YAAE,OAAO,EAAE,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC;QAE5D,MAAM,KAAK,GAAG,IAAI,CAAC,kBAAkB,CAAC,OAAO,CAAC,CAAC,iBAAiB,CAC/D,qBAAqB,EACrB,eAAe,CACf,CAAC;QAIF,IAAI,WAAW,EAAE,CAAC;YACjB,KAAK,CAAC,iBAAiB,CAAC,eAAe,EAAE,SAAS,CAAC,CAAC;QACrD,CAAC;QAED,IAAI,UAAU,KAAK,IAAI,EAAE,CAAC;YACzB,KAAK,CAAC,KAAK,CAAC,qCAAqC,EAAE,EAAE,UAAU,EAAE,CAAC,CAAC;QACpE,CAAC;QACD,IAAI,CAAC,YAAY,CAAC,KAAK,EAAE,OAAO,CAAC,MAAM,CAAC,CAAC;QACzC,IAAI,CAAC,YAAY,CAAC,KAAK,EAAE,OAAO,CAAC,MAAM,CAAC,CAAC;QACzC,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;QAE5C,MAAM,CAAC,IAAI,EAAE,KAAK,CAAC,GAAG,MAAM,KAAK,CAAC,eAAe,EAAE,CAAC;QACpD,OAAO,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC;IACxB,CAAC;IAEO,YAAY,CACnB,KAAgC,EAChC,MAAoC;QAEpC,IAAI,MAAM,EAAE,KAAK,EAAE,CAAC;YACnB,KAAK,CAAC,QAAQ,CAAC,sCAAsC,EAAE,EAAE,KAAK,EAAE,IAAI,MAAM,CAAC,KAAK,GAAG,EAAE,CAAC,CAAC;QACxF,CAAC;QACD,IAAI,MAAM,EAAE,cAAc,KAAK,SAAS,EAAE,CAAC;YAC1C,KAAK,CAAC,QAAQ,CAAC,wCAAwC,EAAE;gBACxD,cAAc,EAAE,MAAM,CAAC,cAAc;aACrC,CAAC,CAAC;QACJ,CAAC;IACF,CAAC;IAEO,YAAY,CACnB,KAAgC,EAChC,MAAqC;QAErC,MAAM,CAAC,KAAK,GAAG,WAAW,EAAE,SAAS,GAAG,MAAM,CAAC,GAAG,MAAM,EAAE,KAAK,CAAC,GAAG,CAAC,IAAI,EAAE,CAAC;QAC3E,MAAM,aAAa,GAAG,SAAS,CAAC,WAAW,EAAE,KAAK,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,CAAC;QAEzE,IAAI,KAAK,KAAK,MAAM,EAAE,CAAC;YACtB,KAAK;iBACH,SAAS,CAAC,mBAAmB,EAAE,kBAAkB,CAAC;iBAClD,OAAO,CAAC,kBAAkB,EAAE,aAAa,CAAC,CAAC;YAC7C,OAAO;QACR,CAAC;QAED,KAAK,CAAC,OAAO,CAAC,SAAS,KAAK,EAAE,EAAE,aAAa,CAAC,CAAC;IAChD,CAAC;IAWD,KAAK,CAAC,oBAAoB,CAAC,EAAU,EAAE,SAAiB;QACvD,OAAO,MAAM,IAAI,CAAC,OAAO,CAAC;YACzB,KAAK,EAAE,EAAE,EAAE,EAAE,SAAS,EAAE;YACxB,SAAS,EAAE,EAAE,aAAa,EAAE,IAAI,EAAE;SAClC,CAAC,CAAC;IACJ,CAAC;IAOD,KAAK,CAAC,QAAQ,CAAC,EAAU;QACxB,OAAO,MAAM,IAAI,CAAC,OAAO,CAAC;YACzB,KAAK,EAAE,EAAE,EAAE,EAAE;YACb,SAAS,EAAE,EAAE,aAAa,EAAE,IAAI,EAAE;SAClC,CAAC,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,gCAAgC,CACrC,OAAsB,EACtB,SAAiB;QAEjB,MAAM,KAAK,GAAG,IAAI,CAAC,kBAAkB,CAAC,OAAO,CAAC;aAC5C,MAAM,CAAC,CAAC,UAAU,CAAC,CAAC;aACpB,OAAO,CAAC,UAAU,EAAE,KAAK,CAAC;aAC1B,IAAI,CAAC,SAAS,CAAC,CAAC;QAElB,IAAI,OAAO,KAAK,IAAI,EAAE,CAAC;YACtB,KAAK,CAAC,KAAK,CAAC,qBAAqB,EAAE,EAAE,OAAO,EAAE,CAAC,CAAC;QACjD,CAAC;QAED,OAAO,MAAM,KAAK,CAAC,OAAO,EAAE,CAAC;IAC9B,CAAC;IAED,KAAK,CAAC,kBAAkB,CACvB,GAAa;QAEb,IAAI,GAAG,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO,EAAE,CAAC;QAEhC,OAAO,MAAM,IAAI,CAAC,IAAI,CAAC;YACtB,MAAM,EAAE,CAAC,IAAI,EAAE,MAAM,EAAE,WAAW,CAAC;YACnC,KAAK,EAAE,EAAE,EAAE,EAAE,IAAA,YAAE,EAAC,GAAG,CAAC,EAAE;SACtB,CAAC,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,kBAAkB,CAAC,EAAU,EAAE,UAAoB;QACxD,IAAI,UAAU,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO,IAAI,CAAC;QACzC,OAAO,MAAM,IAAI,CAAC,OAAO,CAAC;YACzB,KAAK,EAAE,EAAE,EAAE,EAAE,SAAS,EAAE,IAAA,YAAE,EAAC,UAAU,CAAC,EAAE;YACxC,SAAS,EAAE,EAAE,aAAa,EAAE,IAAI,EAAE;SAClC,CAAC,CAAC;IACJ,CAAC;IAGD,KAAK,CAAC,sBAAsB,CAAC,EAAU,EAAE,SAAiB;QACzD,OAAO,MAAM,IAAI,CAAC,MAAM,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,EAAE,SAAS,EAAE,EAAE,CAAC,CAAC;IACxD,CAAC;IAED,KAAK,CAAC,qBAAqB,CAC1B,GAAa,EACb,SAAiB;QAEjB,IAAI,GAAG,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO,EAAE,CAAC;QAChC,OAAO,MAAM,IAAI,CAAC,IAAI,CAAC;YACtB,MAAM,EAAE,CAAC,IAAI,EAAE,iBAAiB,CAAC;YACjC,KAAK,EAAE,EAAE,EAAE,EAAE,IAAA,YAAE,EAAC,GAAG,CAAC,EAAE,SAAS,EAAE;SACjC,CAAC,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,6BAA6B,CAClC,KAAmE;QAEnE,IAAI,KAAK,IAAI,KAAK,IAAI,KAAK,CAAC,GAAG,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO,EAAE,CAAC;QACxD,IAAI,YAAY,IAAI,KAAK,IAAI,KAAK,CAAC,UAAU,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO,EAAE,CAAC;QAEtE,MAAM,QAAQ,GACb,KAAK,IAAI,KAAK;YACb,CAAC,CAAC,EAAE,EAAE,EAAE,IAAA,YAAE,EAAC,KAAK,CAAC,GAAG,CAAC,EAAE;YACvB,CAAC,CAAC,YAAY,IAAI,KAAK;gBACtB,CAAC,CAAC,EAAE,SAAS,EAAE,IAAA,YAAE,EAAC,KAAK,CAAC,UAAU,CAAC,EAAE;gBACrC,CAAC,CAAC,SAAS,CAAC;QAEf,OAAO,MAAM,IAAI,CAAC,IAAI,CAAC;YACtB,MAAM,EAAE,CAAC,IAAI,EAAE,WAAW,EAAE,gBAAgB,CAAC;YAC7C,KAAK,EAAE,QAAQ;SACf,CAAC,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,iBAAiB,CAAC,QAAkB,EAAE,cAAuB;QAClE,IAAI,QAAQ,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO;QAClC,MAAM,IAAI,CAAC,MAAM,CAAC,EAAE,EAAE,EAAE,IAAA,YAAE,EAAC,QAAQ,CAAC,EAAE,EAAE,EAAE,cAAc,EAAE,CAAC,CAAC;IAC7D,CAAC;IAOD,KAAK,CAAC,mBAAmB,CAAC,EAAU,EAAE,WAAiB;QACtD,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,MAAM,CAC/B,EAAE,EAAE,EAAE,gBAAgB,EAAE,IAAA,gBAAM,GAAE,EAAE,EAClC,EAAE,gBAAgB,EAAE,WAAW,EAAE,CACjC,CAAC;QAEF,OAAO,CAAC,MAAM,CAAC,QAAQ,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC;IACnC,CAAC;IAED,KAAK,CAAC,aAAa;QAClB,OAAO,MAAM,IAAI,CAAC,kBAAkB,CAAC,OAAO,CAAC;aAC3C,kBAAkB,CAAC,qBAAqB,EAAE,eAAe,CAAC;aAC1D,OAAO,EAAE,CAAC;IACb,CAAC;IAcD,KAAK,CAAC,2BAA2B,CAChC,IAAY,EACZ,YAAoB,EACpB,SAAiB,EACjB,cAAsB;QAEtB,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,EAAE,SAAS,EAAE,EAAE,CAAC,CAAC;QACzD,OAAO,MAAM,CAAC,MAAM,CACnB,CAAC,KAAK,EAAE,EAAE,CACT,KAAK,CAAC,EAAE,KAAK,cAAc;YAC3B,CAAC,KAAK,CAAC,YAAY,IAAI,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,IAAI,IAAI,CAAC,CAAC,YAAY,KAAK,YAAY,CAAC,CAC3F,CAAC;IACH,CAAC;CACD,CAAA;;0BAtQY,eAAe;IAD3B,IAAA,YAAO,GAAE;qCAEe,oBAAU;GADtB,eAAe,CAsQ3B"}

package/dist/modules/agents/sub-agents/sub-agent-foreground-runner.d.ts

@@ -6,12 +6,14 @@ import type { AgentRunTelemetryType } from '../../../interfaces';
 import { AgentExecutionService } from '../agent-execution.service';
 import type { AgentRuntimeInstrumentation } from '../agent-runtime-instrumentation';
 import { N8NCheckpointStorage } from '../integrations/n8n-checkpoint-storage';
+import type { WorkflowToolExecutionMode } from '../tools/workflow-tool-factory';
 import { SubAgentSourceResolver } from './sub-agent-source-resolver';
 export interface SubAgentForegroundRunContext {
     projectId: string;
     parentAgentId?: string;
     credentialProvider: CredentialProvider;
     runType: AgentRunTelemetryType;
+    workflowToolExecutionMode?: WorkflowToolExecutionMode;
     executionCounter?: AgentExecutionCounter;
     abortSignal?: AbortSignal;
     telemetry?: BuiltTelemetry;

package/dist/modules/agents/sub-agents/sub-agent-foreground-runner.js

@@ -77,12 +77,27 @@ let SubAgentForegroundRunner = class SubAgentForegroundRunner {
             skills: runtimeSource.skills,
             runtimeProfile: 'sub-agent',
             runType: context.runType,
+            workflowToolExecutionMode: context.workflowToolExecutionMode,
             parentAgentIdForDelegation: context.parentAgentId,
             user: context.user,
             instrumentation: context.instrumentation,
         });
         const telemetry = (0, agents_1.deriveSubAgentTelemetry)(context.telemetry);
         const userMessage = operation.type === 'run' ? (0, agents_1.renderDelegateSubAgentPrompt)(operation.request) : null;
+        let executionId;
+        const recorder = new execution_recorder_1.ExecutionRecorder(undefined, (timeline) => {
+            if (executionId) {
+                this.agentExecutionService.recordTimelineSnapshot({
+                    projectId: context.projectId,
+                    agentId: runtimeSource.source.sourceId,
+                    threadId,
+                    executionId,
+                    timeline,
+                });
+            }
+        });
+        const startedAt = recorder.startedAt;
+        let recorded = false;
         try {
             const executionOptions = {
                 ...(context.abortSignal !== undefined ? { abortSignal: context.abortSignal } : {}),
@@ -103,7 +118,33 @@ let SubAgentForegroundRunner = class SubAgentForegroundRunner {
                     runId: operation.request.childRunId,
                     toolCallId: operation.request.childToolCallId,
                 });
-            const { messageRecord, result } = await consumeAgentStream(resultStream, context.onChunk);
+            try {
+                const currentExecutionId = await this.agentExecutionService.startExecutionRecording({
+                    threadId,
+                    agentId: runtimeSource.source.sourceId,
+                    agentName: runtimeSource.source.config.name,
+                    projectId: context.projectId,
+                    userMessage,
+                    source: 'subagent',
+                    threadMetadata: {
+                        parentThreadId: operation.request.parentThreadId,
+                        parentAgentId: context.parentAgentId,
+                    },
+                    telemetry: {
+                        runType: context.runType,
+                        configuration: (0, agent_telemetry_1.buildAgentConfigurationTelemetryFromConfig)(runtimeSource.source.config),
+                    },
+                }, startedAt);
+                executionId = currentExecutionId;
+            }
+            catch (error) {
+                this.logger.warn('Failed to start subagent execution recording', {
+                    agentId: runtimeSource.source.sourceId,
+                    taskPath: operation.taskPath,
+                    error: error instanceof Error ? error.message : String(error),
+                });
+            }
+            const { messageRecord, result } = await consumeAgentStream(resultStream, recorder, context.onChunk);
             const suspended = result.pendingSuspend !== undefined && result.pendingSuspend.length > 0;
             const hitlStatus = suspended
                 ? 'suspended'
@@ -120,8 +161,10 @@ let SubAgentForegroundRunner = class SubAgentForegroundRunner {
                 taskPath: operation.taskPath,
                 userMessage,
                 record: messageRecord,
+                executionId,
                 ...(hitlStatus !== undefined ? { hitlStatus } : {}),
             });
+            recorded = true;
             return {
                 taskPath: operation.taskPath,
                 threadId,
@@ -134,6 +177,26 @@ let SubAgentForegroundRunner = class SubAgentForegroundRunner {
                 ...(suspended ? { resumeContext: createResumeContext(runtimeSource.source) } : {}),
             };
         }
+        catch (error) {
+            if (!recorded) {
+                recorder.record({ type: 'error', error });
+                recorder.record({ type: 'finish', finishReason: 'error' });
+                await this.recordSubAgentExecution({
+                    runtimeSource: runtimeSource.source,
+                    projectId: context.projectId,
+                    threadId,
+                    parentThreadId: operation.request.parentThreadId,
+                    parentAgentId: context.parentAgentId,
+                    runType: context.runType,
+                    taskPath: operation.taskPath,
+                    userMessage,
+                    record: recorder.getMessageRecord(),
+                    executionId,
+                    ...(operation.type === 'resume' ? { hitlStatus: 'resumed' } : {}),
+                });
+            }
+            throw error;
+        }
         finally {
             await agent.close().catch((error) => {
                 this.logger.warn(`Failed to close subagent after ${operation.type}`, {
@@ -144,9 +207,11 @@ let SubAgentForegroundRunner = class SubAgentForegroundRunner {
         }
     }
     async recordSubAgentExecution(params) {
-        const { runtimeSource, projectId, threadId, parentThreadId, parentAgentId, runType, taskPath, userMessage, record, hitlStatus, } = params;
+        const { runtimeSource, projectId, threadId, parentThreadId, parentAgentId, runType, taskPath, userMessage, record, executionId, hitlStatus, } = params;
+        if (!executionId)
+            return;
         try {
-            await this.agentExecutionService.recordMessage({
+            await this.agentExecutionService.finalizeExecution(executionId, {
                 threadId,
                 agentId: runtimeSource.sourceId,
                 agentName: runtimeSource.config.name,
@@ -183,8 +248,7 @@ async function getReconstructionService() {
     const { AgentRuntimeReconstructionService } = await import('../agent-runtime-reconstruction.service.js');
     return di_1.Container.get(AgentRuntimeReconstructionService);
 }
-async function consumeAgentStream(resultStream, onChunk) {
-    const recorder = new execution_recorder_1.ExecutionRecorder();
+async function consumeAgentStream(resultStream, recorder, onChunk) {
     const pendingSuspend = [];
     let structuredOutput;
     for await (const value of (0, agent_stream_1.streamAgentChunks)(resultStream.stream)) {