n8n
Version:
n8n Workflow Automation Tool
199 lines • 9.48 kB
JavaScript
"use strict";
var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
if (k2 === undefined) k2 = k;
var desc = Object.getOwnPropertyDescriptor(m, k);
if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
desc = { enumerable: true, get: function() { return m[k]; } };
}
Object.defineProperty(o, k2, desc);
}) : (function(o, m, k, k2) {
if (k2 === undefined) k2 = k;
o[k2] = m[k];
}));
var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
Object.defineProperty(o, "default", { enumerable: true, value: v });
}) : function(o, v) {
o["default"] = v;
});
var __decorate = (this && this.__decorate) || function (decorators, target, key, desc) {
var c = arguments.length, r = c < 3 ? target : desc === null ? desc = Object.getOwnPropertyDescriptor(target, key) : desc, d;
if (typeof Reflect === "object" && typeof Reflect.decorate === "function") r = Reflect.decorate(decorators, target, key, desc);
else for (var i = decorators.length - 1; i >= 0; i--) if (d = decorators[i]) r = (c < 3 ? d(r) : c > 3 ? d(target, key, r) : d(target, key)) || r;
return c > 3 && r && Object.defineProperty(target, key, r), r;
};
var __importStar = (this && this.__importStar) || (function () {
var ownKeys = function(o) {
ownKeys = Object.getOwnPropertyNames || function (o) {
var ar = [];
for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
return ar;
};
return ownKeys(o);
};
return function (mod) {
if (mod && mod.__esModule) return mod;
var result = {};
if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
__setModuleDefault(result, mod);
return result;
};
})();
var __metadata = (this && this.__metadata) || function (k, v) {
if (typeof Reflect === "object" && typeof Reflect.metadata === "function") return Reflect.metadata(k, v);
};
Object.defineProperty(exports, "__esModule", { value: true });
exports.AgentsBuilderSettingsService = void 0;
const ai_utilities_1 = require("@n8n/ai-utilities");
const api_types_1 = require("@n8n/api-types");
const backend_common_1 = require("@n8n/backend-common");
const db_1 = require("@n8n/db");
const di_1 = require("@n8n/di");
const n8n_workflow_1 = require("n8n-workflow");
const nanoid_1 = require("nanoid");
const credentials_finder_service_1 = require("../../../credentials/credentials-finder.service");
const credentials_service_1 = require("../../../credentials/credentials.service");
const unprocessable_error_1 = require("../../../errors/response-errors/unprocessable.error");
const ai_service_1 = require("../../../services/ai.service");
const proxy_token_manager_1 = require("../../../services/proxy-token-manager");
const credential_field_mapping_1 = require("../json-config/credential-field-mapping");
const errors_1 = require("./errors");
const SETTINGS_KEY = 'agentBuilder.settings';
const DEFAULT_SETTINGS = { mode: 'default' };
const PROXY_HEADERS = {
'x-n8n-feature': 'agent-builder',
};
function readEnvAnthropicKey() {
const key = process.env.N8N_AI_ANTHROPIC_KEY ?? process.env.ANTHROPIC_API_KEY;
return key && key.length > 0 ? key : null;
}
let AgentsBuilderSettingsService = class AgentsBuilderSettingsService {
constructor(logger, settingsRepository, aiService, credentialsService, credentialsFinderService) {
this.logger = logger;
this.settingsRepository = settingsRepository;
this.aiService = aiService;
this.credentialsService = credentialsService;
this.credentialsFinderService = credentialsFinderService;
this.cached = null;
}
async loadSettings() {
if (this.cached)
return this.cached;
const row = await this.settingsRepository.findByKey(SETTINGS_KEY);
if (!row) {
this.cached = DEFAULT_SETTINGS;
return this.cached;
}
const raw = (0, n8n_workflow_1.jsonParse)(row.value, { fallbackValue: undefined });
const parseResult = api_types_1.agentBuilderAdminSettingsSchema.safeParse(raw);
this.cached = parseResult.success ? parseResult.data : DEFAULT_SETTINGS;
return this.cached;
}
async persist(settings) {
await this.settingsRepository.upsert({
key: SETTINGS_KEY,
value: JSON.stringify(settings),
loadOnStartup: true,
}, ['key']);
this.cached = settings;
}
async getAdminSettings() {
const settings = await this.loadSettings();
const isConfigured = await this.computeIsConfigured(settings);
return { settings, isConfigured };
}
async getStatus() {
const settings = await this.loadSettings();
const isConfigured = await this.computeIsConfigured(settings);
return { isConfigured };
}
async computeIsConfigured(settings) {
if (settings.mode === 'custom') {
const credential = await this.credentialsFinderService.findCredentialById(settings.credentialId);
return !!credential;
}
return this.aiService.isProxyEnabled() || !!readEnvAnthropicKey();
}
async updateAdminSettings(payload) {
if (payload.mode === 'custom' && !(0, credential_field_mapping_1.isSupportedAgentProvider)(payload.provider)) {
throw new unprocessable_error_1.UnprocessableRequestError(`Unsupported provider "${payload.provider}". Supported: ${credential_field_mapping_1.SUPPORTED_AGENT_PROVIDERS.join(', ')}`);
}
await this.persist(payload);
}
async resolveModelConfig(user) {
const settings = await this.loadSettings();
if (settings.mode === 'custom') {
const fromCredential = await this.tryResolveCustomCredential(settings);
if (fromCredential)
return { config: fromCredential, isProxied: false };
this.logger.warn('Agent builder custom credential could not be resolved; falling back to default', { credentialId: settings.credentialId });
}
if (this.aiService.isProxyEnabled()) {
return { config: await this.resolveProxyModel(user), isProxied: true };
}
const envKey = readEnvAnthropicKey();
if (envKey) {
return {
config: {
id: `anthropic/${api_types_1.AGENT_BUILDER_DEFAULT_MODEL}`,
apiKey: envKey,
},
isProxied: false,
};
}
throw new errors_1.BuilderNotConfiguredError();
}
async tryResolveCustomCredential(settings) {
if (!(0, credential_field_mapping_1.isSupportedAgentProvider)(settings.provider)) {
this.logger.warn('Agent builder provider is not supported by the runtime', {
provider: settings.provider,
credentialId: settings.credentialId,
});
return null;
}
const credential = await this.credentialsFinderService.findCredentialById(settings.credentialId);
if (!credential)
return null;
const data = await this.credentialsService.decrypt(credential, true);
const mapped = (0, credential_field_mapping_1.mapCredentialForProvider)(settings.provider, data);
const id = `${settings.provider}/${settings.modelName}`;
return { id, ...mapped };
}
async resolveProxyModel(user) {
const client = await this.aiService.getClient();
const baseURL = client.getApiProxyBaseUrl().replace(/\/$/, '') + '/anthropic/v1';
const tokenManager = new proxy_token_manager_1.ProxyTokenManager(async () => {
return await client.getBuilderApiProxyToken({ id: user.id }, { userMessageId: (0, nanoid_1.nanoid)() });
});
const { createAnthropic } = await Promise.resolve().then(() => __importStar(require('@ai-sdk/anthropic')));
const provider = createAnthropic({
baseURL,
apiKey: 'proxy-managed',
fetch: async (input, init) => {
const headers = new Headers(init?.headers);
const auth = await tokenManager.getAuthHeaders();
for (const [k, v] of Object.entries(auth)) {
headers.set(k, v);
}
for (const [k, v] of Object.entries(PROXY_HEADERS)) {
headers.set(k, v);
}
return await (0, ai_utilities_1.proxyFetch)(input, { ...init, headers });
},
});
const model = provider(api_types_1.AGENT_BUILDER_DEFAULT_MODEL);
if (!model) {
throw new n8n_workflow_1.UnexpectedError('Failed to instantiate Anthropic proxy model');
}
return model;
}
};
exports.AgentsBuilderSettingsService = AgentsBuilderSettingsService;
exports.AgentsBuilderSettingsService = AgentsBuilderSettingsService = __decorate([
(0, di_1.Service)(),
__metadata("design:paramtypes", [backend_common_1.Logger,
db_1.SettingsRepository,
ai_service_1.AiService,
credentials_service_1.CredentialsService,
credentials_finder_service_1.CredentialsFinderService])
], AgentsBuilderSettingsService);
//# sourceMappingURL=agents-builder-settings.service.js.map