UNPKG

n8n

Version:

n8n Workflow Automation Tool

164 lines 8.01 kB
"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.AgentsBuilderService = void 0; const agents_1 = require("@n8n/agents"); const backend_common_1 = require("@n8n/backend-common"); const di_1 = require("@n8n/di"); const n8n_workflow_1 = require("n8n-workflow"); 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 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 not_found_error_1 = require("../../../errors/response-errors/not-found.error"); const agents_builder_settings_service_1 = require("./agents-builder-settings.service"); const builder_telemetry_1 = require("../tracing/builder-telemetry"); const BUILDER_MODEL = 'anthropic/claude-sonnet-4-5'; function builderThreadId(agentId) { return `${builder_tool_names_1.AGENT_THREAD_PREFIX.BUILDER}${agentId}`; } let AgentsBuilderService = class AgentsBuilderService { constructor(logger, agentsService, agentsBuilderToolsService, n8nMemory, builderSettings, n8nCheckpointStorage, agentCheckpointRepository) { this.logger = logger; this.agentsService = agentsService; this.agentsBuilderToolsService = agentsBuilderToolsService; this.n8nMemory = n8nMemory; this.builderSettings = builderSettings; this.n8nCheckpointStorage = n8nCheckpointStorage; this.agentCheckpointRepository = agentCheckpointRepository; } async getBuilderMessages(agentId) { const threadId = builderThreadId(agentId); return await this.n8nMemory.getMessages(threadId); } async clearBuilderMessages(agentId) { const threadId = builderThreadId(agentId); await this.n8nMemory.deleteMessagesByThread(threadId); await this.n8nMemory.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`); } 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 instructions = (0, agents_builder_prompts_1.buildBuilderPrompt)({ configJson, configHash: (0, agents_builder_tools_service_1.getAgentConfigHash)(currentConfig), configUpdatedAt: agent.updatedAt.toISOString(), toolList, builderModel: BUILDER_MODEL, }); const tools = this.agentsBuilderToolsService.getTools(agentId, projectId, credentialProvider, user); const builderMemory = new agents_1.Memory().storage(this.n8nMemory).lastMessages(40); const builder = new agents_1.Agent('agent-builder') .model(modelConfig) .instructions(instructions) .memory(builderMemory) .checkpoint(this.n8nCheckpointStorage.getStorage(agentId)); const telemetry = (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, 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]) ], AgentsBuilderService); //# sourceMappingURL=agents-builder.service.js.map