n8n
Version:
n8n Workflow Automation Tool
195 lines • 10.4 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);
};
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 instance_ai_1 = require("@n8n/instance-ai");
const typeorm_1 = require("@n8n/typeorm");
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 instance_ai_credit_service_1 = require("../../instance-ai/instance-ai-credit.service");
const agents_service_1 = require("../agents.service");
const agent_builder_preview_path_1 = require("./agent-builder-preview-path");
const agents_builder_model_recommendations_1 = require("./agents-builder-model-recommendations");
const agents_builder_prompts_1 = require("./agents-builder-prompts");
const agents_builder_tools_service_1 = require("./agents-builder-tools.service");
const errors_1 = require("./errors");
const planner_todos_prompt_1 = require("./prompts/planner-todos.prompt");
const skills_1 = require("./skills");
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_stream_1 = require("../utils/agent-stream");
let AgentsBuilderService = class AgentsBuilderService {
constructor(logger, agentsService, nodeCatalogService, agentsBuilderToolsService, n8nMemory, instanceAiCreditService, n8nCheckpointStorage, agentCheckpointRepository) {
this.logger = logger;
this.agentsService = agentsService;
this.nodeCatalogService = nodeCatalogService;
this.agentsBuilderToolsService = agentsBuilderToolsService;
this.n8nMemory = n8nMemory;
this.instanceAiCreditService = instanceAiCreditService;
this.n8nCheckpointStorage = n8nCheckpointStorage;
this.agentCheckpointRepository = agentCheckpointRepository;
}
async *buildAgent(agentId, projectId, message, credentialProvider, user, session) {
const builder = await this.createBuilderAgent(agentId, projectId, credentialProvider, user, session);
this.logger.debug('Starting builder agent stream', { agentId, projectId });
const resourceId = user.id;
const resultStream = await builder.stream(message, {
persistence: { threadId: session.threadId, resourceId },
});
yield* this.streamFromAgent(resultStream);
}
async *resumeBuild(agentId, projectId, runId, toolCallId, resumeData, credentialProvider, user, session) {
const checkpointStatus = await this.n8nCheckpointStorage.getStatus(runId);
if (checkpointStatus.status === 'expired') {
this.logger.debug('Builder checkpoint unavailable', {
runId,
status: checkpointStatus.status,
});
throw new errors_1.BuilderCheckpointUnavailableError('expired');
}
if (checkpointStatus.status === 'not-found') {
this.logger.debug('Builder checkpoint unavailable', {
runId,
status: checkpointStatus.status,
});
throw new errors_1.BuilderCheckpointUnavailableError('not-found');
}
const builder = await this.createBuilderAgent(agentId, projectId, credentialProvider, user, session);
this.logger.debug('Resuming builder agent', { agentId, runId, toolCallId });
const resultStream = await builder.resume('stream', resumeData, {
runId,
toolCallId,
});
yield* this.streamFromAgent(resultStream);
}
async cancelCheckpoint(agentId, runId) {
await this.n8nCheckpointStorage.delete(runId, agentId);
}
async createBuilderAgent(agentId, projectId, credentialProvider, user, session) {
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 modelConfig = session.modelConfig;
const modelRecommendationsSection = await (0, agents_builder_model_recommendations_1.getModelRecommendationsSection)();
const instructions = (0, agents_builder_prompts_1.buildBuilderPrompt)({
agentPreviewPath: (0, agent_builder_preview_path_1.buildAgentPreviewPath)(projectId, agentId),
modelRecommendationsSection,
});
const finalInstructions = session.instructionsAddendum
? `${instructions}\n\n${session.instructionsAddendum}`
: instructions;
const runtimeSkills = (0, skills_1.getBuilderRuntimeSkills)();
const tools = this.agentsBuilderToolsService.getTools(agentId, projectId, credentialProvider, user, { threadId: session.hostThreadId, runId: session.runId });
const { Agent, Memory, createPlannerTodosTool } = await import('@n8n/agents');
const onMemoryUsage = async (report) => {
try {
const items = (0, instance_ai_1.tokenUsageToBuilderUsageItems)(report.model, report.usage);
if (items.length === 0)
return;
await this.instanceAiCreditService.claimRunUsage(user, session.hostThreadId, `${session.runId}:agent-builder:${agentId}:memory:${report.task}:${report.reportId}`, items, 'completed');
}
catch (error) {
this.logger.warn('Failed to claim agent-builder observational-memory usage', {
agentId,
hostThreadId: session.hostThreadId,
runId: session.runId,
task: report.task,
error: error instanceof Error ? error.message : String(error),
});
}
};
const builderMemory = new Memory()
.storage(this.n8nMemory.getImplementation(agentId))
.observationalMemory({
observe: (0, agents_1.createObservationLogObserveFn)(modelConfig, { onUsage: onMemoryUsage }),
reflect: (0, agents_1.createObservationLogReflectFn)(modelConfig, { onUsage: onMemoryUsage }),
});
const builder = new Agent('agent-builder')
.model(modelConfig)
.promptCaching({ anthropic: { ttl: '5m' } })
.instructions(finalInstructions)
.skills(runtimeSkills)
.memory(builderMemory)
.checkpoint(this.n8nCheckpointStorage.getStorage(agentId))
.configuration({ maxIterations: 30 });
if (session.telemetry)
builder.telemetry(session.telemetry);
if (session.memoryTaskObserver)
builder.memoryTaskObserver(session.memoryTaskObserver);
for (const tool of [...tools.json, ...tools.shared]) {
builder.tool(tool);
}
builder.tool(createPlannerTodosTool({
description: planner_todos_prompt_1.BUILDER_PLANNER_TODOS_DESCRIPTION,
systemInstruction: planner_todos_prompt_1.BUILDER_PLANNER_TODOS_SYSTEM_INSTRUCTION,
}));
(0, instance_ai_1.applyAgentThinking)(builder, modelConfig);
return builder;
}
async *streamFromAgent(resultStream) {
for await (const value of (0, agent_stream_1.streamAgentChunks)(resultStream.stream)) {
yield value;
}
}
async findOpenCheckpoint(agentId) {
return await this.findSuspendedCheckpoint(agentId);
}
async findOpenCheckpointForThread(agentId, threadId, options = {}) {
return await this.findSuspendedCheckpoint(agentId, threadId, options);
}
async findSuspendedCheckpoint(agentId, threadId, options = {}) {
const rows = await this.agentCheckpointRepository.find({
where: options.includeUnscoped
? [
{ agentId, expired: false },
{ agentId: (0, typeorm_1.IsNull)(), expired: false },
]
: { agentId, expired: false },
order: { updatedAt: 'DESC' },
...(threadId === undefined && { 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')
continue;
if (threadId !== undefined && parsed.persistence?.threadId !== threadId)
continue;
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, instance_ai_credit_service_1.InstanceAiCreditService, n8n_checkpoint_storage_1.N8NCheckpointStorage, agent_checkpoint_repository_1.AgentCheckpointRepository])
], AgentsBuilderService);
//# sourceMappingURL=agents-builder.service.js.map