n8n
Version:
n8n Workflow Automation Tool
371 lines • 16.3 kB
JavaScript
;
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);
};
var __param = (this && this.__param) || function (paramIndex, decorator) {
return function (target, key) { decorator(target, key, paramIndex); }
};
Object.defineProperty(exports, "__esModule", { value: true });
exports.AiController = void 0;
const constants_1 = require("../constants");
const api_types_1 = require("@n8n/api-types");
const decorators_1 = require("@n8n/decorators");
const ai_assistant_sdk_1 = require("@n8n_io/ai-assistant-sdk");
const n8n_workflow_1 = require("n8n-workflow");
const node_assert_1 = require("node:assert");
const web_1 = require("node:stream/web");
const credentials_service_1 = require("../credentials/credentials.service");
const bad_request_error_1 = require("../errors/response-errors/bad-request.error");
const content_too_large_error_1 = require("../errors/response-errors/content-too-large.error");
const internal_server_error_1 = require("../errors/response-errors/internal-server.error");
const too_many_requests_error_1 = require("../errors/response-errors/too-many-requests.error");
const ai_gateway_service_1 = require("../services/ai-gateway.service");
const ai_usage_service_1 = require("../services/ai-usage.service");
const ai_workflow_builder_service_1 = require("../services/ai-workflow-builder.service");
const ai_service_1 = require("../services/ai.service");
const user_service_1 = require("../services/user.service");
let AiController = class AiController {
constructor(aiService, workflowBuilderService, credentialsService, userService, aiUsageService, aiGatewayService) {
this.aiService = aiService;
this.workflowBuilderService = workflowBuilderService;
this.credentialsService = credentialsService;
this.userService = userService;
this.aiUsageService = aiUsageService;
this.aiGatewayService = aiGatewayService;
}
async build(req, res, payload) {
try {
const abortController = new AbortController();
const { signal } = abortController;
const handleClose = () => abortController.abort();
res.on('close', handleClose);
const { id, text, workflowContext, featureFlags, versionId } = payload.payload;
const aiResponse = this.workflowBuilderService.chat({
id,
message: text,
workflowContext: {
currentWorkflow: workflowContext.currentWorkflow,
executionData: workflowContext.executionData,
executionSchema: workflowContext.executionSchema,
expressionValues: workflowContext.expressionValues,
valuesExcluded: workflowContext.valuesExcluded,
pinnedNodes: workflowContext.pinnedNodes,
selectedNodes: workflowContext.selectedNodes,
},
featureFlags,
versionId,
mode: payload.payload.mode,
resumeData: payload.payload.resumeData,
}, req.user, signal);
res.header('Content-type', 'application/json-lines').flush();
try {
for await (const chunk of aiResponse) {
res.flush();
res.write(JSON.stringify(chunk) + constants_1.STREAM_SEPARATOR);
}
}
catch (streamError) {
(0, node_assert_1.strict)(streamError instanceof Error);
const errorChunk = {
messages: [
{
role: 'assistant',
type: 'error',
content: streamError.message,
},
],
};
res.write(JSON.stringify(errorChunk) + constants_1.STREAM_SEPARATOR);
}
finally {
res.off('close', handleClose);
}
res.end();
}
catch (e) {
(0, node_assert_1.strict)(e instanceof Error);
if (!res.headersSent) {
res.status(500).json({
code: 500,
message: e.message,
});
}
else {
res.end();
}
}
}
async chat(req, res, payload) {
try {
const abortController = new AbortController();
const { signal } = abortController;
const handleClose = () => abortController.abort();
res.on('close', handleClose);
const aiResponse = await this.aiService.chat(payload, req.user);
if (aiResponse.body) {
res.header('Content-type', 'application/json-lines').flush();
try {
await aiResponse.body.pipeTo(new web_1.WritableStream({
write(chunk) {
res.write(chunk);
res.flush();
},
}), { signal });
}
finally {
res.off('close', handleClose);
}
res.end();
}
}
catch (e) {
if (e instanceof DOMException && e.name === 'AbortError') {
return;
}
(0, node_assert_1.strict)(e instanceof Error);
throw new internal_server_error_1.InternalServerError(e.message, e);
}
}
async applySuggestion(req, _, payload) {
try {
return await this.aiService.applySuggestion(payload, req.user);
}
catch (e) {
(0, node_assert_1.strict)(e instanceof Error);
throw new internal_server_error_1.InternalServerError(e.message, e);
}
}
async askAi(req, _, payload) {
try {
return await this.aiService.askAi(payload, req.user);
}
catch (e) {
if (e instanceof ai_assistant_sdk_1.APIResponseError) {
switch (e.statusCode) {
case 413:
throw new content_too_large_error_1.ContentTooLargeError(e.message);
case 429:
throw new too_many_requests_error_1.TooManyRequestsError(e.message);
case 400:
throw new bad_request_error_1.BadRequestError(e.message);
default:
throw new internal_server_error_1.InternalServerError(e.message, e);
}
}
(0, node_assert_1.strict)(e instanceof Error);
throw new internal_server_error_1.InternalServerError(e.message, e);
}
}
async aiCredits(req, _, payload) {
try {
const aiCredits = await this.aiService.createFreeAiCredits(req.user);
const credentialProperties = {
name: constants_1.FREE_AI_CREDITS_CREDENTIAL_NAME,
type: n8n_workflow_1.OPEN_AI_API_CREDENTIAL_TYPE,
data: {
apiKey: aiCredits.apiKey,
url: aiCredits.url,
},
projectId: payload?.projectId,
};
const newCredential = await this.credentialsService.createManagedCredential(credentialProperties, req.user);
await this.userService.updateSettings(req.user.id, {
userClaimedAiCredits: true,
});
return newCredential;
}
catch (e) {
(0, node_assert_1.strict)(e instanceof Error);
throw new internal_server_error_1.InternalServerError(e.message, e);
}
}
async getSessions(req, _, payload) {
try {
const sessions = await this.workflowBuilderService.getSessions(payload.workflowId, req.user);
return sessions;
}
catch (e) {
(0, node_assert_1.strict)(e instanceof Error);
throw new internal_server_error_1.InternalServerError(e.message, e);
}
}
async getGatewayConfig() {
try {
return await this.aiGatewayService.getGatewayConfig();
}
catch (e) {
(0, node_assert_1.strict)(e instanceof Error);
throw new internal_server_error_1.InternalServerError(e.message, e);
}
}
async getGatewayWallet(req) {
try {
return await this.aiGatewayService.getWallet(req.user.id);
}
catch (e) {
(0, node_assert_1.strict)(e instanceof Error);
throw new internal_server_error_1.InternalServerError(e.message, e);
}
}
async getGatewayUsage(req, _, query) {
try {
return await this.aiGatewayService.getUsage(req.user.id, query.offset, query.limit);
}
catch (e) {
(0, node_assert_1.strict)(e instanceof Error);
throw new internal_server_error_1.InternalServerError(e.message, e);
}
}
async getBuilderCredits(req, _) {
try {
return await this.workflowBuilderService.getBuilderInstanceCredits(req.user);
}
catch (e) {
(0, node_assert_1.strict)(e instanceof Error);
throw new internal_server_error_1.InternalServerError(e.message, e);
}
}
async truncateMessages(req, _, payload) {
try {
const success = await this.workflowBuilderService.truncateMessagesAfter(payload.workflowId, req.user, payload.messageId, payload.versionCardId);
return { success };
}
catch (e) {
(0, node_assert_1.strict)(e instanceof Error);
throw new internal_server_error_1.InternalServerError(e.message, e);
}
}
async clearSession(req, _, payload) {
try {
await this.workflowBuilderService.clearSession(payload.workflowId, req.user);
return { success: true };
}
catch (e) {
(0, node_assert_1.strict)(e instanceof Error);
throw new internal_server_error_1.InternalServerError(e.message, e);
}
}
async updateUsageSettings(_req, _res, payload) {
try {
await this.aiUsageService.updateAiUsageSettings(payload.allowSendingParameterValues);
}
catch (e) {
(0, node_assert_1.strict)(e instanceof Error);
throw new internal_server_error_1.InternalServerError(e.message, e);
}
}
};
exports.AiController = AiController;
__decorate([
(0, decorators_1.Licensed)('feat:aiBuilder'),
(0, decorators_1.Post)('/build', { ipRateLimit: { limit: 100 }, usesTemplates: true }),
__param(2, decorators_1.Body),
__metadata("design:type", Function),
__metadata("design:paramtypes", [Object, Object, api_types_1.AiBuilderChatRequestDto]),
__metadata("design:returntype", Promise)
], AiController.prototype, "build", null);
__decorate([
(0, decorators_1.Post)('/chat', { ipRateLimit: { limit: 100 } }),
__param(2, decorators_1.Body),
__metadata("design:type", Function),
__metadata("design:paramtypes", [Object, Object, api_types_1.AiChatRequestDto]),
__metadata("design:returntype", Promise)
], AiController.prototype, "chat", null);
__decorate([
(0, decorators_1.Post)('/chat/apply-suggestion'),
__param(2, decorators_1.Body),
__metadata("design:type", Function),
__metadata("design:paramtypes", [Object, Object, api_types_1.AiApplySuggestionRequestDto]),
__metadata("design:returntype", Promise)
], AiController.prototype, "applySuggestion", null);
__decorate([
(0, decorators_1.Post)('/ask-ai'),
__param(2, decorators_1.Body),
__metadata("design:type", Function),
__metadata("design:paramtypes", [Object, Object, api_types_1.AiAskRequestDto]),
__metadata("design:returntype", Promise)
], AiController.prototype, "askAi", null);
__decorate([
(0, decorators_1.Post)('/free-credits'),
__param(2, decorators_1.Body),
__metadata("design:type", Function),
__metadata("design:paramtypes", [Object, Object, api_types_1.AiFreeCreditsRequestDto]),
__metadata("design:returntype", Promise)
], AiController.prototype, "aiCredits", null);
__decorate([
(0, decorators_1.Licensed)('feat:aiBuilder'),
(0, decorators_1.Post)('/sessions', { ipRateLimit: { limit: 100 } }),
__param(2, decorators_1.Body),
__metadata("design:type", Function),
__metadata("design:paramtypes", [Object, Object, api_types_1.AiSessionRetrievalRequestDto]),
__metadata("design:returntype", Promise)
], AiController.prototype, "getSessions", null);
__decorate([
(0, decorators_1.Licensed)('feat:aiGateway'),
(0, decorators_1.Get)('/gateway/config'),
__metadata("design:type", Function),
__metadata("design:paramtypes", []),
__metadata("design:returntype", Promise)
], AiController.prototype, "getGatewayConfig", null);
__decorate([
(0, decorators_1.Licensed)('feat:aiGateway'),
(0, decorators_1.Get)('/gateway/wallet'),
__metadata("design:type", Function),
__metadata("design:paramtypes", [Object]),
__metadata("design:returntype", Promise)
], AiController.prototype, "getGatewayWallet", null);
__decorate([
(0, decorators_1.Licensed)('feat:aiGateway'),
(0, decorators_1.Get)('/gateway/usage'),
__param(2, decorators_1.Query),
__metadata("design:type", Function),
__metadata("design:paramtypes", [Object, Object, api_types_1.AiGatewayUsageQueryDto]),
__metadata("design:returntype", Promise)
], AiController.prototype, "getGatewayUsage", null);
__decorate([
(0, decorators_1.Licensed)('feat:aiBuilder'),
(0, decorators_1.Get)('/build/credits'),
__metadata("design:type", Function),
__metadata("design:paramtypes", [Object, Object]),
__metadata("design:returntype", Promise)
], AiController.prototype, "getBuilderCredits", null);
__decorate([
(0, decorators_1.Licensed)('feat:aiBuilder'),
(0, decorators_1.Post)('/build/truncate-messages', { ipRateLimit: { limit: 100 } }),
__param(2, decorators_1.Body),
__metadata("design:type", Function),
__metadata("design:paramtypes", [Object, Object, api_types_1.AiTruncateMessagesRequestDto]),
__metadata("design:returntype", Promise)
], AiController.prototype, "truncateMessages", null);
__decorate([
(0, decorators_1.Licensed)('feat:aiBuilder'),
(0, decorators_1.Post)('/build/clear-session', { ipRateLimit: { limit: 100 } }),
__param(2, decorators_1.Body),
__metadata("design:type", Function),
__metadata("design:paramtypes", [Object, Object, api_types_1.AiClearSessionRequestDto]),
__metadata("design:returntype", Promise)
], AiController.prototype, "clearSession", null);
__decorate([
(0, decorators_1.Post)('/usage-settings'),
(0, decorators_1.GlobalScope)('aiAssistant:manage'),
__param(2, decorators_1.Body),
__metadata("design:type", Function),
__metadata("design:paramtypes", [Object, Object, api_types_1.AiUsageSettingsRequestDto]),
__metadata("design:returntype", Promise)
], AiController.prototype, "updateUsageSettings", null);
exports.AiController = AiController = __decorate([
(0, decorators_1.RestController)('/ai'),
__metadata("design:paramtypes", [ai_service_1.AiService,
ai_workflow_builder_service_1.WorkflowBuilderService,
credentials_service_1.CredentialsService,
user_service_1.UserService,
ai_usage_service_1.AiUsageService,
ai_gateway_service_1.AiGatewayService])
], AiController);
//# sourceMappingURL=ai.controller.js.map