n8n 2.44.4 → 2.45.0

package/dist/modules/agents/agent-chat-relay.service.js

@@ -0,0 +1,164 @@
+"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.AgentChatRelayService = void 0;
+const api_types_1 = require("@n8n/api-types");
+const di_1 = require("@n8n/di");
+const errors_1 = require("@n8n/errors");
+const sanitize_filename_1 = require("@n8n/utils/files/sanitize-filename");
+const n8n_core_1 = require("n8n-core");
+const promises_1 = require("node:stream/promises");
+const agent_chat_attachment_service_1 = require("./agent-chat-attachment.service");
+const agent_message_queue_service_1 = require("./agent-message-queue.service");
+const agent_queued_preview_stream_service_1 = require("./agent-queued-preview-stream.service");
+const agent_sse_stream_1 = require("./agent-sse-stream");
+const inbound_attachments_1 = require("./utils/inbound-attachments");
+let AgentChatRelayService = class AgentChatRelayService {
+    constructor(messageQueue, queuedPreviewStreams, agentChatAttachmentService) {
+        this.messageQueue = messageQueue;
+        this.queuedPreviewStreams = queuedPreviewStreams;
+        this.agentChatAttachmentService = agentChatAttachmentService;
+    }
+    createChatExecution(res) {
+        const delivery = (0, agent_sse_stream_1.initSseStream)(res);
+        const requestController = new AbortController();
+        const abandon = () => requestController.abort();
+        delivery.abortSignal.addEventListener('abort', abandon, { once: true });
+        if (delivery.abortSignal.aborted)
+            abandon();
+        return {
+            send: delivery.send,
+            abortSignal: requestController.signal,
+            onExecutionStarted: (id, sessionId, inputMessageIds) => {
+                delivery.abortSignal.removeEventListener('abort', abandon);
+                delivery.send({ type: 'execution-started', executionId: id, sessionId, inputMessageIds });
+            },
+            onChunk: delivery.onChunk,
+            close: () => {
+                delivery.abortSignal.removeEventListener('abort', abandon);
+                delivery.close();
+            },
+        };
+    }
+    async storeChatAttachments(params) {
+        const { attachments, agentId, projectId, threadId, resourceId, source = 'chat' } = params;
+        if (!attachments?.length)
+            return undefined;
+        const stored = [];
+        try {
+            for (const attachment of attachments) {
+                const data = Buffer.from(attachment.data, 'base64');
+                if (data.byteLength === 0) {
+                    throw new errors_1.BadRequestError(`Attachment "${attachment.fileName}" is empty`);
+                }
+                if (data.byteLength > api_types_1.MAX_AGENT_CHAT_ATTACHMENT_SIZE_BYTES) {
+                    throw new errors_1.BadRequestError(`Attachment "${attachment.fileName}" exceeds the ${api_types_1.MAX_AGENT_CHAT_ATTACHMENT_SIZE_MB} MB limit`);
+                }
+                const mimeType = await (0, inbound_attachments_1.resolveInboundMimeType)(attachment.mimeType, data);
+                const row = await this.agentChatAttachmentService.storeInbound({
+                    agentId,
+                    projectId,
+                    threadId,
+                    resourceId,
+                    source,
+                    fileName: attachment.fileName,
+                    mimeType,
+                    data,
+                });
+                stored.push({
+                    id: row.id,
+                    fileName: row.fileName,
+                    mimeType: row.mimeType,
+                    sizeBytes: row.fileSizeBytes,
+                });
+            }
+        }
+        catch (error) {
+            await this.agentChatAttachmentService.deleteByIds(stored.map((ref) => ref.id));
+            throw error;
+        }
+        return stored;
+    }
+    async relayQueuedMessage(res, accept) {
+        const delivery = (0, agent_sse_stream_1.initSseStream)(res);
+        const { send, abortSignal } = delivery;
+        let accepted = false;
+        let subscription;
+        let attachments;
+        try {
+            const input = await accept(send, abortSignal);
+            if (!input)
+                return;
+            attachments = input.payload.attachments;
+            abortSignal.throwIfAborted();
+            const result = await this.messageQueue.enqueue(input, (queueId) => {
+                subscription = this.queuedPreviewStreams.subscribe(queueId, delivery);
+            });
+            if (result.status === 'duplicate') {
+                send({ type: 'done' });
+                return;
+            }
+            accepted = true;
+            subscription?.accepted();
+            send({ type: 'message-queued', queueId: result.item.id, sessionId: input.threadId });
+            await subscription?.done;
+        }
+        catch (error) {
+            send((0, agent_sse_stream_1.toChatErrorEvent)(error, 'Chat failed'));
+        }
+        finally {
+            if (!accepted && attachments?.length) {
+                await this.agentChatAttachmentService
+                    .deleteByIds(attachments.map((ref) => ref.id))
+                    .catch(() => { });
+            }
+            subscription?.close();
+            delivery.close();
+        }
+    }
+    async streamAttachment(attachment, res) {
+        const attachmentId = attachment.id;
+        let stream;
+        try {
+            stream = await this.agentChatAttachmentService.getStream(attachment);
+        }
+        catch (error) {
+            if (error instanceof n8n_core_1.FileNotFoundError) {
+                throw new errors_1.NotFoundError(`Attachment "${attachmentId}" is no longer available`);
+            }
+            throw error;
+        }
+        res.setHeader('Content-Type', attachment.mimeType);
+        res.setHeader('Content-Length', attachment.fileSizeBytes);
+        res.setHeader('X-Content-Type-Options', 'nosniff');
+        res.setHeader('Content-Security-Policy', (0, n8n_core_1.getHtmlSandboxCSP)());
+        if (!api_types_1.ViewableMimeTypes.includes(attachment.mimeType.toLowerCase())) {
+            res.setHeader('Content-Disposition', `attachment; filename="${(0, sanitize_filename_1.sanitizeFilename)(attachment.fileName)}"`);
+        }
+        try {
+            await (0, promises_1.pipeline)(stream, res);
+        }
+        catch (error) {
+            if (error instanceof Error &&
+                'code' in error &&
+                error.code === 'ERR_STREAM_PREMATURE_CLOSE') {
+                return;
+            }
+            throw error;
+        }
+    }
+};
+exports.AgentChatRelayService = AgentChatRelayService;
+exports.AgentChatRelayService = AgentChatRelayService = __decorate([
+    (0, di_1.Service)(),
+    __metadata("design:paramtypes", [agent_message_queue_service_1.AgentMessageQueueService, agent_queued_preview_stream_service_1.AgentQueuedPreviewStreamService, agent_chat_attachment_service_1.AgentChatAttachmentService])
+], AgentChatRelayService);
+//# sourceMappingURL=agent-chat-relay.service.js.map

package/dist/modules/agents/agent-chat-relay.service.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"agent-chat-relay.service.js","sourceRoot":"","sources":["../../../src/modules/agents/agent-chat-relay.service.ts"],"names":[],"mappings":";;;;;;;;;;;;AAAA,8CAMwB;AACxB,gCAAkC;AAClC,wCAA6D;AAC7D,0EAAsE;AAEtE,uCAAgE;AAChE,mDAAgD;AAEhD,mFAA6E;AAC7E,+EAAyE;AACzE,+FAAwF;AACxF,yDAA6F;AAG7F,qEAAqE;AAO9D,IAAM,qBAAqB,GAA3B,MAAM,qBAAqB;IACjC,YACkB,YAAsC,EACtC,oBAAqD,EACrD,0BAAsD;4BAFtD,YAAY;oCACZ,oBAAoB;0CACpB,0BAA0B;IACzC,CAAC;IAEJ,mBAAmB,CAAC,GAAsB;QACzC,MAAM,QAAQ,GAAG,IAAA,gCAAa,EAAC,GAAG,CAAC,CAAC;QACpC,MAAM,iBAAiB,GAAG,IAAI,eAAe,EAAE,CAAC;QAChD,MAAM,OAAO,GAAG,GAAG,EAAE,CAAC,iBAAiB,CAAC,KAAK,EAAE,CAAC;QAChD,QAAQ,CAAC,WAAW,CAAC,gBAAgB,CAAC,OAAO,EAAE,OAAO,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC;QACxE,IAAI,QAAQ,CAAC,WAAW,CAAC,OAAO;YAAE,OAAO,EAAE,CAAC;QAC5C,OAAO;YACN,IAAI,EAAE,QAAQ,CAAC,IAAI;YACnB,WAAW,EAAE,iBAAiB,CAAC,MAAM;YACrC,kBAAkB,EAAE,CAAC,EAAU,EAAE,SAAiB,EAAE,eAAyB,EAAE,EAAE;gBAChF,QAAQ,CAAC,WAAW,CAAC,mBAAmB,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;gBAC3D,QAAQ,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,mBAAmB,EAAE,WAAW,EAAE,EAAE,EAAE,SAAS,EAAE,eAAe,EAAE,CAAC,CAAC;YAC3F,CAAC;YACD,OAAO,EAAE,QAAQ,CAAC,OAAO;YACzB,KAAK,EAAE,GAAG,EAAE;gBACX,QAAQ,CAAC,WAAW,CAAC,mBAAmB,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;gBAC3D,QAAQ,CAAC,KAAK,EAAE,CAAC;YAClB,CAAC;SACD,CAAC;IACH,CAAC;IAGD,KAAK,CAAC,oBAAoB,CAAC,MAO1B;QACA,MAAM,EAAE,WAAW,EAAE,OAAO,EAAE,SAAS,EAAE,QAAQ,EAAE,UAAU,EAAE,MAAM,GAAG,MAAM,EAAE,GAAG,MAAM,CAAC;QAC1F,IAAI,CAAC,WAAW,EAAE,MAAM;YAAE,OAAO,SAAS,CAAC;QAE3C,MAAM,MAAM,GAA0B,EAAE,CAAC;QACzC,IAAI,CAAC;YACJ,KAAK,MAAM,UAAU,IAAI,WAAW,EAAE,CAAC;gBACtC,MAAM,IAAI,GAAG,MAAM,CAAC,IAAI,CAAC,UAAU,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC;gBACpD,IAAI,IAAI,CAAC,UAAU,KAAK,CAAC,EAAE,CAAC;oBAC3B,MAAM,IAAI,wBAAe,CAAC,eAAe,UAAU,CAAC,QAAQ,YAAY,CAAC,CAAC;gBAC3E,CAAC;gBACD,IAAI,IAAI,CAAC,UAAU,GAAG,gDAAoC,EAAE,CAAC;oBAC5D,MAAM,IAAI,wBAAe,CACxB,eAAe,UAAU,CAAC,QAAQ,iBAAiB,6CAAiC,WAAW,CAC/F,CAAC;gBACH,CAAC;gBAED,MAAM,QAAQ,GAAG,MAAM,IAAA,4CAAsB,EAAC,UAAU,CAAC,QAAQ,EAAE,IAAI,CAAC,CAAC;gBACzE,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,0BAA0B,CAAC,YAAY,CAAC;oBAC9D,OAAO;oBACP,SAAS;oBACT,QAAQ;oBACR,UAAU;oBACV,MAAM;oBACN,QAAQ,EAAE,UAAU,CAAC,QAAQ;oBAC7B,QAAQ;oBACR,IAAI;iBACJ,CAAC,CAAC;gBACH,MAAM,CAAC,IAAI,CAAC;oBACX,EAAE,EAAE,GAAG,CAAC,EAAE;oBACV,QAAQ,EAAE,GAAG,CAAC,QAAQ;oBACtB,QAAQ,EAAE,GAAG,CAAC,QAAQ;oBACtB,SAAS,EAAE,GAAG,CAAC,aAAa;iBAC5B,CAAC,CAAC;YACJ,CAAC;QACF,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YAEhB,MAAM,IAAI,CAAC,0BAA0B,CAAC,WAAW,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,CAAC;YAC/E,MAAM,KAAK,CAAC;QACb,CAAC;QACD,OAAO,MAAM,CAAC;IACf,CAAC;IAMD,KAAK,CAAC,kBAAkB,CACvB,GAAsB,EACtB,MAG4E;QAE5E,MAAM,QAAQ,GAAG,IAAA,gCAAa,EAAC,GAAG,CAAC,CAAC;QACpC,MAAM,EAAE,IAAI,EAAE,WAAW,EAAE,GAAG,QAAQ,CAAC;QACvC,IAAI,QAAQ,GAAG,KAAK,CAAC;QACrB,IAAI,YAAkF,CAAC;QACvF,IAAI,WAA8C,CAAC;QACnD,IAAI,CAAC;YACJ,MAAM,KAAK,GAAG,MAAM,MAAM,CAAC,IAAI,EAAE,WAAW,CAAC,CAAC;YAC9C,IAAI,CAAC,KAAK;gBAAE,OAAO;YACnB,WAAW,GAAG,KAAK,CAAC,OAAO,CAAC,WAAW,CAAC;YACxC,WAAW,CAAC,cAAc,EAAE,CAAC;YAC7B,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC,OAAO,EAAE,EAAE;gBACjE,YAAY,GAAG,IAAI,CAAC,oBAAoB,CAAC,SAAS,CAAC,OAAO,EAAE,QAAQ,CAAC,CAAC;YACvE,CAAC,CAAC,CAAC;YACH,IAAI,MAAM,CAAC,MAAM,KAAK,WAAW,EAAE,CAAC;gBACnC,IAAI,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC,CAAC;gBACvB,OAAO;YACR,CAAC;YACD,QAAQ,GAAG,IAAI,CAAC;YAChB,YAAY,EAAE,QAAQ,EAAE,CAAC;YACzB,IAAI,CAAC,EAAE,IAAI,EAAE,gBAAgB,EAAE,OAAO,EAAE,MAAM,CAAC,IAAI,CAAC,EAAE,EAAE,SAAS,EAAE,KAAK,CAAC,QAAQ,EAAE,CAAC,CAAC;YACrF,MAAM,YAAY,EAAE,IAAI,CAAC;QAC1B,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YAChB,IAAI,CAAC,IAAA,mCAAgB,EAAC,KAAK,EAAE,aAAa,CAAC,CAAC,CAAC;QAC9C,CAAC;gBAAS,CAAC;YAEV,IAAI,CAAC,QAAQ,IAAI,WAAW,EAAE,MAAM,EAAE,CAAC;gBACtC,MAAM,IAAI,CAAC,0BAA0B;qBACnC,WAAW,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;qBAC7C,KAAK,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;YACnB,CAAC;YACD,YAAY,EAAE,KAAK,EAAE,CAAC;YACtB,QAAQ,CAAC,KAAK,EAAE,CAAC;QAClB,CAAC;IACF,CAAC;IAED,KAAK,CAAC,gBAAgB,CAAC,UAA+B,EAAE,GAAa;QACpE,MAAM,YAAY,GAAG,UAAU,CAAC,EAAE,CAAC;QAInC,IAAI,MAAoE,CAAC;QACzE,IAAI,CAAC;YACJ,MAAM,GAAG,MAAM,IAAI,CAAC,0BAA0B,CAAC,SAAS,CAAC,UAAU,CAAC,CAAC;QACtE,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YAChB,IAAI,KAAK,YAAY,4BAAiB,EAAE,CAAC;gBACxC,MAAM,IAAI,sBAAa,CAAC,eAAe,YAAY,0BAA0B,CAAC,CAAC;YAChF,CAAC;YACD,MAAM,KAAK,CAAC;QACb,CAAC;QAED,GAAG,CAAC,SAAS,CAAC,cAAc,EAAE,UAAU,CAAC,QAAQ,CAAC,CAAC;QACnD,GAAG,CAAC,SAAS,CAAC,gBAAgB,EAAE,UAAU,CAAC,aAAa,CAAC,CAAC;QAC1D,GAAG,CAAC,SAAS,CAAC,wBAAwB,EAAE,SAAS,CAAC,CAAC;QAInD,GAAG,CAAC,SAAS,CAAC,yBAAyB,EAAE,IAAA,4BAAiB,GAAE,CAAC,CAAC;QAE9D,IAAI,CAAC,6BAAiB,CAAC,QAAQ,CAAC,UAAU,CAAC,QAAQ,CAAC,WAAW,EAAE,CAAC,EAAE,CAAC;YACpE,GAAG,CAAC,SAAS,CACZ,qBAAqB,EACrB,yBAAyB,IAAA,oCAAgB,EAAC,UAAU,CAAC,QAAQ,CAAC,GAAG,CACjE,CAAC;QACH,CAAC;QAID,IAAI,CAAC;YACJ,MAAM,IAAA,mBAAQ,EAAC,MAAM,EAAE,GAAG,CAAC,CAAC;QAC7B,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YAChB,IACC,KAAK,YAAY,KAAK;gBACtB,MAAM,IAAI,KAAK;gBACf,KAAK,CAAC,IAAI,KAAK,4BAA4B,EAC1C,CAAC;gBACF,OAAO;YACR,CAAC;YACD,MAAM,KAAK,CAAC;QACb,CAAC;IACF,CAAC;CACD,CAAA;;gCA1KY,qBAAqB;IADjC,IAAA,YAAO,GAAE;qCAGuB,sDAAwB,EAChB,qEAA+B,EACzB,0DAA0B;GAJ5D,qBAAqB,CA0KjC"}

package/dist/modules/agents/agent-chat.controller.d.ts

@@ -4,22 +4,22 @@ import type { AuthenticatedRequest } from '@n8n/db';
 import type { Response } from 'express';
 import { CredentialsService } from '../../credentials/credentials.service';
 import { AgentChatAttachmentService } from './agent-chat-attachment.service';
+import { AgentChatRelayService } from './agent-chat-relay.service';
 import { AgentExecutionOrchestratorService } from './agent-execution-orchestrator.service';
 import { AgentMessageQueueService } from './agent-message-queue.service';
-import { AgentQueuedPreviewStreamService } from './agent-queued-preview-stream.service';
 import { AgentChatExecutionService } from './agent-chat-execution.service';
 import { AgentExecutionService } from './agent-execution.service';
 import { type FlushableResponse } from './agent-sse-stream';
 import { AgentTestChatService } from './agent-test-chat.service';
 import { AgentTestRunService } from './agent-test-run.service';
 import { AgentsService } from './agents.service';
-import { AgentsBuilderService } from './builder/agents-builder.service';
 import { AgentBackgroundJobService } from './background/agent-background-job.service';
+import { N8NCheckpointStorage } from './integrations/n8n-checkpoint-storage';
 export declare class AgentChatController {
     private readonly agentExecutionOrchestratorService;
     private readonly agentTestRunService;
     private readonly agentTestChatService;
-    private readonly agentsBuilderService;
+    private readonly checkpointStorage;
     private readonly credentialsService;
     private readonly agentsService;
     private readonly agentChatAttachmentService;
@@ -27,15 +27,12 @@ export declare class AgentChatController {
     private readonly backgroundJobService;
     private readonly chatExecutionService;
     private readonly messageQueue;
-    private readonly queuedPreviewStreams;
     private readonly agentsConfig;
-    constructor(agentExecutionOrchestratorService: AgentExecutionOrchestratorService, agentTestRunService: AgentTestRunService, agentTestChatService: AgentTestChatService, agentsBuilderService: AgentsBuilderService, credentialsService: CredentialsService, agentsService: AgentsService, agentChatAttachmentService: AgentChatAttachmentService, agentExecutionService: AgentExecutionService, backgroundJobService: AgentBackgroundJobService, chatExecutionService: AgentChatExecutionService, messageQueue: AgentMessageQueueService, queuedPreviewStreams: AgentQueuedPreviewStreamService, agentsConfig: AgentsConfig);
-    private createChatExecution;
-    private storeChatAttachments;
+    private readonly chatRelay;
+    constructor(agentExecutionOrchestratorService: AgentExecutionOrchestratorService, agentTestRunService: AgentTestRunService, agentTestChatService: AgentTestChatService, checkpointStorage: N8NCheckpointStorage, credentialsService: CredentialsService, agentsService: AgentsService, agentChatAttachmentService: AgentChatAttachmentService, agentExecutionService: AgentExecutionService, backgroundJobService: AgentBackgroundJobService, chatExecutionService: AgentChatExecutionService, messageQueue: AgentMessageQueueService, agentsConfig: AgentsConfig, chatRelay: AgentChatRelayService);
     private requireProductionChat;
     private requireProductionThread;
     private requirePreviewThread;
-    private relayQueuedMessage;
     private assertQueueId;
     productionChat(req: AuthenticatedRequest<{
         projectId: string;
@@ -195,7 +192,6 @@ export declare class AgentChatController {
         agentId: string;
         attachmentId: string;
     }>, res: Response): Promise<void>;
-    private streamAttachment;
     clearTestChatMessages(req: AuthenticatedRequest<{
         projectId: string;
         agentId: string;

package/dist/modules/agents/agent-chat.controller.js

@@ -18,18 +18,15 @@ const config_1 = require("@n8n/config");
 const decorators_1 = require("@n8n/decorators");
 const scrub_secrets_1 = require("@n8n/utils/scrub-secrets");
 const redact_text_1 = require("@n8n/utils/redaction/redact-text");
-const sanitize_filename_1 = require("@n8n/utils/files/sanitize-filename");
-const n8n_core_1 = require("n8n-core");
-const promises_1 = require("node:stream/promises");
 const node_crypto_1 = require("node:crypto");
 const credentials_service_1 = require("../../credentials/credentials.service");
 const errors_1 = require("@n8n/errors");
 const agents_credential_provider_1 = require("./adapters/agents-credential-provider");
 const agent_chat_attachment_service_1 = require("./agent-chat-attachment.service");
 const agent_n8n_chat_unavailable_error_1 = require("./agent-n8n-chat-unavailable.error");
+const agent_chat_relay_service_1 = require("./agent-chat-relay.service");
 const agent_execution_orchestrator_service_1 = require("./agent-execution-orchestrator.service");
 const agent_message_queue_service_1 = require("./agent-message-queue.service");
-const agent_queued_preview_stream_service_1 = require("./agent-queued-preview-stream.service");
 const agent_chat_execution_service_1 = require("./agent-chat-execution.service");
 const agent_execution_service_1 = require("./agent-execution.service");
 const agent_thread_access_1 = require("./utils/agent-thread-access");
@@ -38,17 +35,16 @@ const agent_sse_stream_1 = require("./agent-sse-stream");
 const agent_test_chat_service_1 = require("./agent-test-chat.service");
 const agent_test_run_service_1 = require("./agent-test-run.service");
 const agents_service_1 = require("./agents.service");
-const agents_builder_service_1 = require("./builder/agents-builder.service");
 const agent_background_job_service_1 = require("./background/agent-background-job.service");
+const n8n_checkpoint_storage_1 = require("./integrations/n8n-checkpoint-storage");
 const agent_memory_scope_1 = require("./utils/agent-memory-scope");
-const inbound_attachments_1 = require("./utils/inbound-attachments");
 const messages_envelope_1 = require("./utils/messages-envelope");
 let AgentChatController = class AgentChatController {
-    constructor(agentExecutionOrchestratorService, agentTestRunService, agentTestChatService, agentsBuilderService, credentialsService, agentsService, agentChatAttachmentService, agentExecutionService, backgroundJobService, chatExecutionService, messageQueue, queuedPreviewStreams, agentsConfig) {
+    constructor(agentExecutionOrchestratorService, agentTestRunService, agentTestChatService, checkpointStorage, credentialsService, agentsService, agentChatAttachmentService, agentExecutionService, backgroundJobService, chatExecutionService, messageQueue, agentsConfig, chatRelay) {
         this.agentExecutionOrchestratorService = agentExecutionOrchestratorService;
         this.agentTestRunService = agentTestRunService;
         this.agentTestChatService = agentTestChatService;
-        this.agentsBuilderService = agentsBuilderService;
+        this.checkpointStorage = checkpointStorage;
         this.credentialsService = credentialsService;
         this.agentsService = agentsService;
         this.agentChatAttachmentService = agentChatAttachmentService;
@@ -56,68 +52,8 @@ let AgentChatController = class AgentChatController {
         this.backgroundJobService = backgroundJobService;
         this.chatExecutionService = chatExecutionService;
         this.messageQueue = messageQueue;
-        this.queuedPreviewStreams = queuedPreviewStreams;
         this.agentsConfig = agentsConfig;
-    }
-    createChatExecution(res) {
-        const delivery = (0, agent_sse_stream_1.initSseStream)(res);
-        const requestController = new AbortController();
-        const abandon = () => requestController.abort();
-        delivery.abortSignal.addEventListener('abort', abandon, { once: true });
-        if (delivery.abortSignal.aborted)
-            abandon();
-        return {
-            send: delivery.send,
-            abortSignal: requestController.signal,
-            onExecutionStarted: (id, sessionId, inputMessageIds) => {
-                delivery.abortSignal.removeEventListener('abort', abandon);
-                delivery.send({ type: 'execution-started', executionId: id, sessionId, inputMessageIds });
-            },
-            onChunk: delivery.onChunk,
-            close: () => {
-                delivery.abortSignal.removeEventListener('abort', abandon);
-                delivery.close();
-            },
-        };
-    }
-    async storeChatAttachments(params) {
-        const { attachments, agentId, projectId, threadId, resourceId, source = 'chat' } = params;
-        if (!attachments?.length)
-            return undefined;
-        const stored = [];
-        try {
-            for (const attachment of attachments) {
-                const data = Buffer.from(attachment.data, 'base64');
-                if (data.byteLength === 0) {
-                    throw new errors_1.BadRequestError(`Attachment "${attachment.fileName}" is empty`);
-                }
-                if (data.byteLength > api_types_1.MAX_AGENT_CHAT_ATTACHMENT_SIZE_BYTES) {
-                    throw new errors_1.BadRequestError(`Attachment "${attachment.fileName}" exceeds the ${api_types_1.MAX_AGENT_CHAT_ATTACHMENT_SIZE_MB} MB limit`);
-                }
-                const mimeType = await (0, inbound_attachments_1.resolveInboundMimeType)(attachment.mimeType, data);
-                const row = await this.agentChatAttachmentService.storeInbound({
-                    agentId,
-                    projectId,
-                    threadId,
-                    resourceId,
-                    source,
-                    fileName: attachment.fileName,
-                    mimeType,
-                    data,
-                });
-                stored.push({
-                    id: row.id,
-                    fileName: row.fileName,
-                    mimeType: row.mimeType,
-                    sizeBytes: row.fileSizeBytes,
-                });
-            }
-        }
-        catch (error) {
-            await this.agentChatAttachmentService.deleteByIds(stored.map((ref) => ref.id));
-            throw error;
-        }
-        return stored;
+        this.chatRelay = chatRelay;
     }
     async requireProductionChat(agentId, projectId) {
         if (!(await this.agentsService.isN8nChatPublished(agentId, projectId))) {
@@ -135,43 +71,6 @@ let AgentChatController = class AgentChatController {
         })))
             throw new errors_1.NotFoundError(`Thread "${threadId}" not found`);
     }
-    async relayQueuedMessage(res, accept) {
-        const delivery = (0, agent_sse_stream_1.initSseStream)(res);
-        const { send, abortSignal } = delivery;
-        let accepted = false;
-        let subscription;
-        let attachments;
-        try {
-            const input = await accept(send, abortSignal);
-            if (!input)
-                return;
-            attachments = input.payload.attachments;
-            abortSignal.throwIfAborted();
-            const result = await this.messageQueue.enqueue(input, (queueId) => {
-                subscription = this.queuedPreviewStreams.subscribe(queueId, delivery);
-            });
-            if (result.status === 'duplicate') {
-                send({ type: 'done' });
-                return;
-            }
-            accepted = true;
-            subscription?.accepted();
-            send({ type: 'message-queued', queueId: result.item.id, sessionId: input.threadId });
-            await subscription?.done;
-        }
-        catch (error) {
-            send((0, agent_sse_stream_1.toChatErrorEvent)(error, 'Chat failed'));
-        }
-        finally {
-            if (!accepted && attachments?.length) {
-                await this.agentChatAttachmentService
-                    .deleteByIds(attachments.map((ref) => ref.id))
-                    .catch(() => { });
-            }
-            subscription?.close();
-            delivery.close();
-        }
-    }
     assertQueueId(queueId) {
         if (!/^[1-9]\d*$/.test(queueId))
             throw new errors_1.BadRequestError('Invalid queue ID');
@@ -179,14 +78,14 @@ let AgentChatController = class AgentChatController {
     async productionChat(req, res, agentId, payload) {
         const { projectId } = req.params;
         const resourceId = (0, agent_memory_scope_1.productionChatMemoryResourceId)(req.user.id);
-        await this.relayQueuedMessage(res, async () => {
+        await this.chatRelay.relayQueuedMessage(res, async () => {
             if (!(await this.agentsService.isN8nChatPublished(agentId, projectId))) {
                 throw new agent_n8n_chat_unavailable_error_1.AgentN8nChatUnavailableError();
             }
             const sessionMode = payload.sessionId && !payload.newSession ? 'existing' : 'new';
             const threadId = payload.sessionId ?? (0, node_crypto_1.randomUUID)();
             await this.requireProductionThread(threadId, projectId, agentId, req.user.id, sessionMode);
-            const attachments = await this.storeChatAttachments({
+            const attachments = await this.chatRelay.storeChatAttachments({
                 attachments: payload.attachments,
                 agentId,
                 projectId,
@@ -212,7 +111,7 @@ let AgentChatController = class AgentChatController {
         });
     }
     async productionChatResume(req, res, agentId, payload) {
-        const execution = this.createChatExecution(res);
+        const execution = this.chatRelay.createChatExecution(res);
         const { send, onChunk, abortSignal, onExecutionStarted } = execution;
         let executionId;
         try {
@@ -258,7 +157,7 @@ let AgentChatController = class AgentChatController {
         const { projectId } = req.params;
         const { message, sessionId, messageId, newSession, attachments } = payload;
         const credentialProvider = new agents_credential_provider_1.AgentsCredentialProvider(this.credentialsService, projectId, req.user, agentId);
-        await this.relayQueuedMessage(res, async (send, abortSignal) => {
+        await this.chatRelay.relayQueuedMessage(res, async (send, abortSignal) => {
             const prepared = await this.agentTestRunService.prepareDraftRun({
                 agentId,
                 projectId,
@@ -295,7 +194,7 @@ let AgentChatController = class AgentChatController {
                     kind: 'preview',
                     message,
                     messageId,
-                    attachments: await this.storeChatAttachments({
+                    attachments: await this.chatRelay.storeChatAttachments({
                         attachments,
                         agentId,
                         projectId,
@@ -311,7 +210,7 @@ let AgentChatController = class AgentChatController {
     async chatResume(req, res, agentId, payload) {
         const { projectId } = req.params;
         const { runId, toolCallId, resumeData } = payload;
-        const execution = this.createChatExecution(res);
+        const execution = this.chatRelay.createChatExecution(res);
         const { send, onChunk, abortSignal, onExecutionStarted } = execution;
         try {
             abortSignal.throwIfAborted();
@@ -325,7 +224,7 @@ let AgentChatController = class AgentChatController {
                 chatSurface: 'preview',
                 errorMode: 'forward',
                 onChunk,
-                onBudgetNotice: () => send({ type: 'budget-notice', code: 'budget.alert' }),
+                onBudgetNotice: (notice) => send({ type: 'budget-notice', ...notice }),
                 onExecutionStarted,
                 abortSignal,
             });
@@ -402,7 +301,7 @@ let AgentChatController = class AgentChatController {
         });
         if (!history)
             throw new errors_1.NotFoundError('Session not found');
-        const checkpoint = await this.agentsBuilderService.findOpenCheckpointForThread(agentId, threadId);
+        const checkpoint = await this.checkpointStorage.findSuspendedForThread(agentId, threadId);
         return {
             ...(0, messages_envelope_1.withOpenSuspensions)(history.messages, checkpoint?.persistence?.resourceId === (0, agent_memory_scope_1.productionChatMemoryResourceId)(req.user.id)
                 ? checkpoint
@@ -424,7 +323,7 @@ let AgentChatController = class AgentChatController {
             throw new errors_1.NotFoundError(`Attachment "${attachmentId}" not found`);
         }
         await this.requireProductionThread(attachment.threadId, projectId, agentId, req.user.id);
-        await this.streamAttachment(attachment, res);
+        await this.chatRelay.streamAttachment(attachment, res);
     }
     async getProductionQueuedMessages(req) {
         await this.requireProductionChat(req.params.agentId, req.params.projectId);
@@ -664,7 +563,7 @@ let AgentChatController = class AgentChatController {
             agentId,
             userId: req.user.id,
         });
-        const checkpoint = await this.agentsBuilderService.findOpenCheckpointForThread(agentId, threadId);
+        const checkpoint = await this.checkpointStorage.findSuspendedForThread(agentId, threadId);
         if (checkpoint &&
             (thread?.accessScope === 'project'
                 ? (0, agent_memory_scope_1.userIdFromDraftChatMemoryResourceId)(checkpoint.persistence?.resourceId ?? '') !==
@@ -695,7 +594,7 @@ let AgentChatController = class AgentChatController {
             throw new errors_1.NotFoundError('Session not found');
         }
         const messages = await this.agentTestChatService.getTestChatMessages(agentId, req.user.id);
-        const checkpoint = await this.agentsBuilderService.findOpenCheckpointForThread(agentId, (0, agent_test_chat_service_1.chatThreadId)(agentId, req.user.id));
+        const checkpoint = await this.checkpointStorage.findSuspendedForThread(agentId, (0, agent_test_chat_service_1.chatThreadId)(agentId, req.user.id));
         return (0, messages_envelope_1.withOpenSuspensions)((0, agent_message_mapper_1.messagesToDto)(messages), checkpoint?.persistence?.resourceId === (0, agent_memory_scope_1.draftChatMemoryResourceId)(req.user.id)
             ? checkpoint
             : null);
@@ -715,38 +614,7 @@ let AgentChatController = class AgentChatController {
         if (attachment.source === agent_thread_access_1.N8N_CHAT_PRODUCTION_SOURCE) {
             throw new errors_1.NotFoundError(`Attachment "${attachmentId}" not found`);
         }
-        await this.streamAttachment(attachment, res);
-    }
-    async streamAttachment(attachment, res) {
-        const attachmentId = attachment.id;
-        let stream;
-        try {
-            stream = await this.agentChatAttachmentService.getStream(attachment);
-        }
-        catch (error) {
-            if (error instanceof n8n_core_1.FileNotFoundError) {
-                throw new errors_1.NotFoundError(`Attachment "${attachmentId}" is no longer available`);
-            }
-            throw error;
-        }
-        res.setHeader('Content-Type', attachment.mimeType);
-        res.setHeader('Content-Length', attachment.fileSizeBytes);
-        res.setHeader('X-Content-Type-Options', 'nosniff');
-        res.setHeader('Content-Security-Policy', (0, n8n_core_1.getHtmlSandboxCSP)());
-        if (!api_types_1.ViewableMimeTypes.includes(attachment.mimeType.toLowerCase())) {
-            res.setHeader('Content-Disposition', `attachment; filename="${(0, sanitize_filename_1.sanitizeFilename)(attachment.fileName)}"`);
-        }
-        try {
-            await (0, promises_1.pipeline)(stream, res);
-        }
-        catch (error) {
-            if (error instanceof Error &&
-                'code' in error &&
-                error.code === 'ERR_STREAM_PREMATURE_CLOSE') {
-                return;
-            }
-            throw error;
-        }
+        await this.chatRelay.streamAttachment(attachment, res);
     }
     async clearTestChatMessages(req) {
         const { projectId, agentId } = req.params;
@@ -1008,6 +876,6 @@ __decorate([
 ], AgentChatController.prototype, "clearTestChatMessages", null);
 exports.AgentChatController = AgentChatController = __decorate([
     (0, decorators_1.RestController)('/projects/:projectId/agents/v2'),
-    __metadata("design:paramtypes", [agent_execution_orchestrator_service_1.AgentExecutionOrchestratorService, agent_test_run_service_1.AgentTestRunService, agent_test_chat_service_1.AgentTestChatService, agents_builder_service_1.AgentsBuilderService, credentials_service_1.CredentialsService, agents_service_1.AgentsService, agent_chat_attachment_service_1.AgentChatAttachmentService, agent_execution_service_1.AgentExecutionService, agent_background_job_service_1.AgentBackgroundJobService, agent_chat_execution_service_1.AgentChatExecutionService, agent_message_queue_service_1.AgentMessageQueueService, agent_queued_preview_stream_service_1.AgentQueuedPreviewStreamService, config_1.AgentsConfig])
+    __metadata("design:paramtypes", [agent_execution_orchestrator_service_1.AgentExecutionOrchestratorService, agent_test_run_service_1.AgentTestRunService, agent_test_chat_service_1.AgentTestChatService, n8n_checkpoint_storage_1.N8NCheckpointStorage, credentials_service_1.CredentialsService, agents_service_1.AgentsService, agent_chat_attachment_service_1.AgentChatAttachmentService, agent_execution_service_1.AgentExecutionService, agent_background_job_service_1.AgentBackgroundJobService, agent_chat_execution_service_1.AgentChatExecutionService, agent_message_queue_service_1.AgentMessageQueueService, config_1.AgentsConfig, agent_chat_relay_service_1.AgentChatRelayService])
 ], AgentChatController);
 //# sourceMappingURL=agent-chat.controller.js.map