UNPKG

n8n

Version:

n8n Workflow Automation Tool

221 lines • 10.8 kB
"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.AgentsBuilderService = void 0; const backend_common_1 = require("@n8n/backend-common"); const config_1 = require("@n8n/config"); const di_1 = require("@n8n/di"); const n8n_workflow_1 = require("n8n-workflow"); const not_found_error_1 = require("../../../errors/response-errors/not-found.error"); const node_catalog_1 = require("../../../node-catalog"); const agents_service_1 = require("../agents.service"); const agent_config_composition_1 = require("../json-config/agent-config-composition"); const n8n_checkpoint_storage_1 = require("../integrations/n8n-checkpoint-storage"); const n8n_memory_1 = require("../integrations/n8n-memory"); const agent_checkpoint_repository_1 = require("../repositories/agent-checkpoint.repository"); const agent_builder_preview_path_1 = require("./agent-builder-preview-path"); const agents_builder_prompts_1 = require("./agents-builder-prompts"); const agents_builder_tools_service_1 = require("./agents-builder-tools.service"); const builder_tool_names_1 = require("./builder-tool-names"); const agents_builder_settings_service_1 = require("./agents-builder-settings.service"); const builder_telemetry_1 = require("../tracing/builder-telemetry"); const agents_builder_model_recommendations_1 = require("./agents-builder-model-recommendations"); const skills_1 = require("./skills"); function builderThreadId(agentId) { return `${builder_tool_names_1.AGENT_THREAD_PREFIX.BUILDER}${agentId}`; } let AgentsBuilderService = class AgentsBuilderService { constructor(logger, agentsService, nodeCatalogService, agentsBuilderToolsService, n8nMemory, builderSettings, n8nCheckpointStorage, agentCheckpointRepository, agentsConfig) { this.logger = logger; this.agentsService = agentsService; this.nodeCatalogService = nodeCatalogService; this.agentsBuilderToolsService = agentsBuilderToolsService; this.n8nMemory = n8nMemory; this.builderSettings = builderSettings; this.n8nCheckpointStorage = n8nCheckpointStorage; this.agentCheckpointRepository = agentCheckpointRepository; this.agentsConfig = agentsConfig; } async getBuilderMessages(agentId) { const threadId = builderThreadId(agentId); return await this.n8nMemory.getImplementation(agentId).getMessages(threadId); } async clearBuilderMessages(agentId) { const threadId = builderThreadId(agentId); const memory = this.n8nMemory.getImplementation(agentId); await memory.deleteMessagesByThread(threadId); await memory.deleteThread(threadId); } async *buildAgent(agentId, projectId, message, credentialProvider, user) { const builder = await this.createBuilderAgent(agentId, projectId, credentialProvider, user); this.logger.debug('Starting builder agent stream', { agentId, projectId }); const resourceId = user.id; const resultStream = await builder.stream(message, { persistence: { threadId: builderThreadId(agentId), resourceId }, }); yield* this.streamFromAgent(resultStream); } async *resumeBuild(agentId, projectId, runId, toolCallId, resumeData, credentialProvider, user) { const checkpointStatus = await this.n8nCheckpointStorage.getStatus(runId); if (checkpointStatus.status === 'expired') { throw new n8n_workflow_1.UserError(`Builder checkpoint ${runId} has expired and cannot be resumed`); } if (checkpointStatus.status === 'not-found') { throw new n8n_workflow_1.UserError(`Builder checkpoint ${runId} not found`); } const builder = await this.createBuilderAgent(agentId, projectId, credentialProvider, user); this.logger.debug('Resuming builder agent', { agentId, runId, toolCallId }); const resultStream = await builder.resume('stream', resumeData, { runId, toolCallId, }); yield* this.streamFromAgent(resultStream); } async createBuilderAgent(agentId, projectId, credentialProvider, user) { const agent = await this.agentsService.findById(agentId, projectId); if (!agent) { throw new not_found_error_1.NotFoundError(`Agent "${agentId}" not found`); } void this.nodeCatalogService.initialize().catch((error) => { this.logger.warn('Failed to initialize node catalog in builder warmup', { error: error instanceof Error ? error.message : String(error), agentId, }); }); const { config: modelConfig } = await this.builderSettings.resolveModelConfig(user); const currentConfig = (0, agent_config_composition_1.composeJsonConfig)(agent); const currentToolsMap = agent.tools ?? {}; const toolList = Object.entries(currentToolsMap) .map(([id, t]) => `- ${id}: ${t.descriptor.name} -- ${t.descriptor.description}`) .join('\n') || '(none)'; const configJson = currentConfig ? JSON.stringify(currentConfig, null, 2) : '(no config yet)'; const modelRecommendationsSection = await (0, agents_builder_model_recommendations_1.getModelRecommendationsSection)(); const enabledModules = this.agentsConfig.modules; const instructions = (0, agents_builder_prompts_1.buildBuilderPrompt)({ configJson, configHash: (0, agents_builder_tools_service_1.getAgentConfigHash)(currentConfig), configUpdatedAt: agent.updatedAt.toISOString(), toolList, agentPreviewPath: (0, agent_builder_preview_path_1.buildAgentPreviewPath)(projectId, agentId), modelRecommendationsSection, enabledModules, }); const runtimeSkills = (0, skills_1.getBuilderRuntimeSkills)(); const tools = this.agentsBuilderToolsService.getTools(agentId, projectId, credentialProvider, user); const { Agent, Memory } = await Promise.resolve().then(() => __importStar(require('@n8n/agents'))); const builderMemory = new Memory() .storage(this.n8nMemory.getImplementation(agentId)) .observationalMemory(); const builder = new Agent('agent-builder') .model(modelConfig) .instructions(instructions) .skills(runtimeSkills) .memory(builderMemory) .checkpoint(this.n8nCheckpointStorage.getStorage(agentId)) .configuration({ maxIterations: 30 }); const telemetry = await (0, builder_telemetry_1.buildBuilderTelemetry)({ agentId, projectId, userId: user.id, threadId: builderThreadId(agentId), model: modelConfig, }); if (telemetry) builder.telemetry(telemetry); for (const tool of [...tools.json, ...tools.shared]) { builder.tool(tool); } return builder; } async *streamFromAgent(resultStream) { const reader = resultStream.stream.getReader(); try { while (true) { const { done, value } = await reader.read(); if (done) break; yield value; } } finally { reader.releaseLock(); } } async findOpenCheckpoint(agentId) { const rows = await this.agentCheckpointRepository.find({ where: { agentId, expired: false }, order: { updatedAt: 'DESC' }, take: 5, }); for (const row of rows) { if (!row.state) continue; let parsed; try { parsed = (0, n8n_workflow_1.jsonParse)(row.state); } catch { continue; } if (parsed.status === 'suspended') { return parsed; } } return null; } }; exports.AgentsBuilderService = AgentsBuilderService; exports.AgentsBuilderService = AgentsBuilderService = __decorate([ (0, di_1.Service)(), __metadata("design:paramtypes", [backend_common_1.Logger, agents_service_1.AgentsService, node_catalog_1.NodeCatalogService, agents_builder_tools_service_1.AgentsBuilderToolsService, n8n_memory_1.N8nMemory, agents_builder_settings_service_1.AgentsBuilderSettingsService, n8n_checkpoint_storage_1.N8NCheckpointStorage, agent_checkpoint_repository_1.AgentCheckpointRepository, config_1.AgentsConfig]) ], AgentsBuilderService); //# sourceMappingURL=agents-builder.service.js.map