@n8n/db 1.32.2 → 1.34.0

package/dist/repositories/agent-eval-dataset.repository.ee.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"agent-eval-dataset.repository.ee.js","sourceRoot":"","sources":["../../src/repositories/agent-eval-dataset.repository.ee.ts"],"names":[],"mappings":";;;;;;;;;;;;AACA,gCAAkC;AAClC,0CAAsD;AAGtD,0CAA+C;AAsBxC,IAAM,0BAA0B,GAAhC,MAAM,0BAA2B,SAAQ,oBAA4B;IAC3E,YAAY,UAAsB;QACjC,KAAK,CAAC,2BAAgB,EAAE,UAAU,CAAC,OAAO,CAAC,CAAC;IAC7C,CAAC;IAED,KAAK,CAAC,aAAa,CAAC,KAAkC;QACrD,MAAM,OAAO,GAAG,IAAI,CAAC,MAAM,CAAC;YAC3B,IAAI,EAAE,KAAK,CAAC,IAAI;YAChB,OAAO,EAAE,KAAK,CAAC,OAAO;YACtB,aAAa,EAAE,KAAK,CAAC,aAAa;YAClC,UAAU,EAAE,KAAK,CAAC,UAAU;YAC5B,WAAW,EAAE,KAAK,CAAC,WAAW,IAAI,IAAI;YACtC,aAAa,EAAE,KAAK,CAAC,aAAa,IAAI,IAAI;YAC1C,WAAW,EAAE,KAAK,CAAC,WAAW,IAAI,IAAI;SACtC,CAAC,CAAC;QAEH,OAAO,MAAM,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IACjC,CAAC;IAED,KAAK,CAAC,aAAa,CAAC,OAAe;QAClC,OAAO,MAAM,IAAI,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,EAAE,OAAO,EAAE,EAAE,KAAK,EAAE,EAAE,SAAS,EAAE,MAAM,EAAE,EAAE,CAAC,CAAC;IAC9E,CAAC;IAED,KAAK,CAAC,QAAQ,CAAC,EAAU;QACxB,OAAO,MAAM,IAAI,CAAC,SAAS,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC;IACrC,CAAC;IAID,KAAK,CAAC,kBAAkB,CAAC,EAAU,EAAE,OAAe;QACnD,OAAO,MAAM,IAAI,CAAC,SAAS,CAAC,EAAE,EAAE,EAAE,OAAO,EAAE,CAAC,CAAC;IAC9C,CAAC;IAID,KAAK,CAAC,aAAa,CAClB,EAAU,EACV,OAAe,EACf,KAAkC;QAElC,MAAM,KAAK,GAA6C;YACvD,GAAG,CAAC,KAAK,CAAC,IAAI,KAAK,SAAS,IAAI,EAAE,IAAI,EAAE,KAAK,CAAC,IAAI,EAAE,CAAC;YACrD,GAAG,CAAC,KAAK,CAAC,WAAW,KAAK,SAAS,IAAI,EAAE,WAAW,EAAE,KAAK,CAAC,WAAW,EAAE,CAAC;YAC1E,GAAG,CAAC,KAAK,CAAC,aAAa,KAAK,SAAS,IAAI,EAAE,aAAa,EAAE,KAAK,CAAC,aAAa,EAAE,CAAC;SAChF,CAAC;QAIF,IAAI,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACnC,MAAM,IAAI,CAAC,MAAM,CAAC,EAAE,EAAE,EAAE,OAAO,EAAE,EAAE,KAAK,CAAC,CAAC;QAC3C,CAAC;QAID,OAAO,MAAM,IAAI,CAAC,kBAAkB,CAAC,EAAE,EAAE,OAAO,CAAC,CAAC;IACnD,CAAC;IAGD,KAAK,CAAC,aAAa,CAAC,EAAU,EAAE,OAAe;QAC9C,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,EAAE,EAAE,EAAE,OAAO,EAAE,CAAC,CAAC;QAClD,OAAO,CAAC,MAAM,CAAC,QAAQ,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC;IACnC,CAAC;CACD,CAAA;;qCA9DY,0BAA0B;IADtC,IAAA,YAAO,GAAE;qCAEe,oBAAU;GADtB,0BAA0B,CA8DtC"}

package/dist/repositories/agent-eval-rating.repository.ee.d.ts

@@ -0,0 +1,18 @@
+import { DataSource, Repository } from '@n8n/typeorm';
+import type { JsonObject } from 'n8n-workflow';
+import { AgentEvalRating } from '../entities';
+import type { AgentEvalVote } from '../entities/agent-eval-rating.ee';
+type CreateAgentEvalRatingAttrs = {
+    resultId: string;
+    vote: AgentEvalVote;
+    comment?: string | null;
+    correction?: JsonObject | null;
+    ratedById?: string | null;
+};
+export declare class AgentEvalRatingRepository extends Repository<AgentEvalRating> {
+    constructor(dataSource: DataSource);
+    createRating(attrs: CreateAgentEvalRatingAttrs): Promise<AgentEvalRating>;
+    findByResultId(resultId: string): Promise<AgentEvalRating[]>;
+    findLatestByRunId(runId: string): Promise<AgentEvalRating[]>;
+}
+export {};

package/dist/repositories/agent-eval-rating.repository.ee.js

@@ -0,0 +1,71 @@
+"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.AgentEvalRatingRepository = void 0;
+const di_1 = require("@n8n/di");
+const typeorm_1 = require("@n8n/typeorm");
+const entities_1 = require("../entities");
+const ID_LOOKUP_CHUNK_SIZE = 100;
+let AgentEvalRatingRepository = class AgentEvalRatingRepository extends typeorm_1.Repository {
+    constructor(dataSource) {
+        super(entities_1.AgentEvalRating, dataSource.manager);
+    }
+    async createRating(attrs) {
+        const rating = this.create({
+            resultId: attrs.resultId,
+            vote: attrs.vote,
+            comment: attrs.comment ?? null,
+            correction: attrs.correction ?? null,
+            ratedById: attrs.ratedById ?? null,
+        });
+        return await this.save(rating);
+    }
+    async findByResultId(resultId) {
+        return await this.find({ where: { resultId }, order: { createdAt: 'DESC' } });
+    }
+    async findLatestByRunId(runId) {
+        const rows = await this.createQueryBuilder('rating')
+            .select('rating.id', 'id')
+            .addSelect('rating.resultId', 'resultId')
+            .innerJoin(entities_1.AgentEvalResult, 'result', 'result.id = rating.resultId')
+            .where('result.runId = :runId', { runId })
+            .orderBy('rating.resultId', 'ASC')
+            .addOrderBy('rating.createdAt', 'DESC')
+            .addOrderBy('rating.id', 'DESC')
+            .getRawMany();
+        const seen = new Set();
+        const latestIds = rows
+            .filter((row) => {
+            if (seen.has(row.resultId))
+                return false;
+            seen.add(row.resultId);
+            return true;
+        })
+            .map((row) => row.id);
+        if (latestIds.length === 0)
+            return [];
+        const found = [];
+        for (let i = 0; i < latestIds.length; i += ID_LOOKUP_CHUNK_SIZE) {
+            const chunk = latestIds.slice(i, i + ID_LOOKUP_CHUNK_SIZE);
+            found.push(...(await this.find({ where: { id: (0, typeorm_1.In)(chunk) } })));
+        }
+        const byId = new Map(found.map((rating) => [rating.id, rating]));
+        return latestIds
+            .map((id) => byId.get(id))
+            .filter((rating) => rating !== undefined);
+    }
+};
+exports.AgentEvalRatingRepository = AgentEvalRatingRepository;
+exports.AgentEvalRatingRepository = AgentEvalRatingRepository = __decorate([
+    (0, di_1.Service)(),
+    __metadata("design:paramtypes", [typeorm_1.DataSource])
+], AgentEvalRatingRepository);
+//# sourceMappingURL=agent-eval-rating.repository.ee.js.map

package/dist/repositories/agent-eval-rating.repository.ee.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"agent-eval-rating.repository.ee.js","sourceRoot":"","sources":["../../src/repositories/agent-eval-rating.repository.ee.ts"],"names":[],"mappings":";;;;;;;;;;;;AAAA,gCAAkC;AAClC,0CAA0D;AAG1D,0CAA+D;AAI/D,MAAM,oBAAoB,GAAG,GAAG,CAAC;AAW1B,IAAM,yBAAyB,GAA/B,MAAM,yBAA0B,SAAQ,oBAA2B;IACzE,YAAY,UAAsB;QACjC,KAAK,CAAC,0BAAe,EAAE,UAAU,CAAC,OAAO,CAAC,CAAC;IAC5C,CAAC;IAED,KAAK,CAAC,YAAY,CAAC,KAAiC;QACnD,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC;YAC1B,QAAQ,EAAE,KAAK,CAAC,QAAQ;YACxB,IAAI,EAAE,KAAK,CAAC,IAAI;YAChB,OAAO,EAAE,KAAK,CAAC,OAAO,IAAI,IAAI;YAC9B,UAAU,EAAE,KAAK,CAAC,UAAU,IAAI,IAAI;YACpC,SAAS,EAAE,KAAK,CAAC,SAAS,IAAI,IAAI;SAClC,CAAC,CAAC;QAEH,OAAO,MAAM,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;IAChC,CAAC;IAED,KAAK,CAAC,cAAc,CAAC,QAAgB;QACpC,OAAO,MAAM,IAAI,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,EAAE,QAAQ,EAAE,EAAE,KAAK,EAAE,EAAE,SAAS,EAAE,MAAM,EAAE,EAAE,CAAC,CAAC;IAC/E,CAAC;IAMD,KAAK,CAAC,iBAAiB,CAAC,KAAa;QAEpC,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,kBAAkB,CAAC,QAAQ,CAAC;aAClD,MAAM,CAAC,WAAW,EAAE,IAAI,CAAC;aACzB,SAAS,CAAC,iBAAiB,EAAE,UAAU,CAAC;aACxC,SAAS,CAAC,0BAAe,EAAE,QAAQ,EAAE,6BAA6B,CAAC;aACnE,KAAK,CAAC,uBAAuB,EAAE,EAAE,KAAK,EAAE,CAAC;aACzC,OAAO,CAAC,iBAAiB,EAAE,KAAK,CAAC;aACjC,UAAU,CAAC,kBAAkB,EAAE,MAAM,CAAC;aACtC,UAAU,CAAC,WAAW,EAAE,MAAM,CAAC;aAC/B,UAAU,EAAoC,CAAC;QAEjD,MAAM,IAAI,GAAG,IAAI,GAAG,EAAU,CAAC;QAC/B,MAAM,SAAS,GAAG,IAAI;aACpB,MAAM,CAAC,CAAC,GAAG,EAAE,EAAE;YACf,IAAI,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,QAAQ,CAAC;gBAAE,OAAO,KAAK,CAAC;YACzC,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;YACvB,OAAO,IAAI,CAAC;QACb,CAAC,CAAC;aACD,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;QACvB,IAAI,SAAS,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO,EAAE,CAAC;QAGtC,MAAM,KAAK,GAAsB,EAAE,CAAC;QACpC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,SAAS,CAAC,MAAM,EAAE,CAAC,IAAI,oBAAoB,EAAE,CAAC;YACjE,MAAM,KAAK,GAAG,SAAS,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,GAAG,oBAAoB,CAAC,CAAC;YAC3D,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,MAAM,IAAI,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,EAAE,IAAA,YAAE,EAAC,KAAK,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC;QAChE,CAAC;QAED,MAAM,IAAI,GAAG,IAAI,GAAG,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC,MAAM,CAAC,EAAE,EAAE,MAAM,CAAC,CAAC,CAAC,CAAC;QACjE,OAAO,SAAS;aACd,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;aACzB,MAAM,CAAC,CAAC,MAAM,EAA6B,EAAE,CAAC,MAAM,KAAK,SAAS,CAAC,CAAC;IACvE,CAAC;CACD,CAAA;;oCA3DY,yBAAyB;IADrC,IAAA,YAAO,GAAE;qCAEe,oBAAU;GADtB,yBAAyB,CA2DrC"}

package/dist/repositories/agent-eval-result.repository.ee.d.ts

@@ -0,0 +1,27 @@
+import { DataSource, Repository } from '@n8n/typeorm';
+import type { IDataObject, JsonObject } from 'n8n-workflow';
+import { AgentEvalResult } from '../entities';
+import type { AgentEvalResultStatus } from '../entities/agent-eval-result.ee';
+type CreateAgentEvalResultAttrs = {
+    runId: string;
+    sourceRowId?: string | null;
+    runIndex?: number | null;
+    input?: JsonObject | null;
+};
+export type AgentEvalResultStatusCounts = Record<AgentEvalResultStatus, number>;
+export declare class AgentEvalResultRepository extends Repository<AgentEvalResult> {
+    constructor(dataSource: DataSource);
+    seedResults(cases: CreateAgentEvalResultAttrs[]): Promise<AgentEvalResult[]>;
+    countByStatus(runId: string): Promise<AgentEvalResultStatusCounts>;
+    markAsRunning(id: string): Promise<import("@n8n/typeorm").UpdateResult>;
+    markAsCancelled(id: string): Promise<import("@n8n/typeorm").UpdateResult>;
+    markAsCompleted(id: string, attrs: {
+        output: JsonObject | null;
+        toolCalls?: JsonObject | null;
+        metrics?: IDataObject | null;
+    }): Promise<import("@n8n/typeorm").UpdateResult>;
+    markAsError(id: string, errorCode: string, errorDetails?: IDataObject | null): Promise<import("@n8n/typeorm").UpdateResult>;
+    findByRunId(runId: string): Promise<AgentEvalResult[]>;
+    findById(id: string): Promise<AgentEvalResult | null>;
+}
+export {};

package/dist/repositories/agent-eval-result.repository.ee.js

@@ -0,0 +1,84 @@
+"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.AgentEvalResultRepository = void 0;
+const di_1 = require("@n8n/di");
+const typeorm_1 = require("@n8n/typeorm");
+const entities_1 = require("../entities");
+const SEED_CHUNK_SIZE = 100;
+let AgentEvalResultRepository = class AgentEvalResultRepository extends typeorm_1.Repository {
+    constructor(dataSource) {
+        super(entities_1.AgentEvalResult, dataSource.manager);
+    }
+    async seedResults(cases) {
+        const results = cases.map((c, index) => this.create({
+            status: 'new',
+            runId: c.runId,
+            sourceRowId: c.sourceRowId ?? null,
+            runIndex: c.runIndex ?? index,
+            input: c.input ?? null,
+        }));
+        return await this.save(results, { chunk: SEED_CHUNK_SIZE });
+    }
+    async countByStatus(runId) {
+        const rows = await this.createQueryBuilder('result')
+            .select('result.status', 'status')
+            .addSelect('COUNT(*)', 'count')
+            .where('result.runId = :runId', { runId })
+            .groupBy('result.status')
+            .getRawMany();
+        const counts = {
+            new: 0,
+            running: 0,
+            success: 0,
+            error: 0,
+            cancelled: 0,
+        };
+        for (const row of rows)
+            counts[row.status] = Number(row.count);
+        return counts;
+    }
+    async markAsRunning(id) {
+        return await this.update(id, { status: 'running', runAt: new Date() });
+    }
+    async markAsCancelled(id) {
+        return await this.update(id, { status: 'cancelled', completedAt: new Date() });
+    }
+    async markAsCompleted(id, attrs) {
+        return await this.update(id, {
+            status: 'success',
+            completedAt: new Date(),
+            output: attrs.output,
+            toolCalls: attrs.toolCalls ?? null,
+            metrics: attrs.metrics ?? null,
+        });
+    }
+    async markAsError(id, errorCode, errorDetails) {
+        return await this.update(id, {
+            status: 'error',
+            completedAt: new Date(),
+            errorCode,
+            errorDetails: errorDetails ?? null,
+        });
+    }
+    async findByRunId(runId) {
+        return await this.find({ where: { runId }, order: { runIndex: 'ASC' } });
+    }
+    async findById(id) {
+        return await this.findOneBy({ id });
+    }
+};
+exports.AgentEvalResultRepository = AgentEvalResultRepository;
+exports.AgentEvalResultRepository = AgentEvalResultRepository = __decorate([
+    (0, di_1.Service)(),
+    __metadata("design:paramtypes", [typeorm_1.DataSource])
+], AgentEvalResultRepository);
+//# sourceMappingURL=agent-eval-result.repository.ee.js.map

package/dist/repositories/agent-eval-result.repository.ee.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"agent-eval-result.repository.ee.js","sourceRoot":"","sources":["../../src/repositories/agent-eval-result.repository.ee.ts"],"names":[],"mappings":";;;;;;;;;;;;AAAA,gCAAkC;AAClC,0CAAsD;AAGtD,0CAA8C;AAc9C,MAAM,eAAe,GAAG,GAAG,CAAC;AAGrB,IAAM,yBAAyB,GAA/B,MAAM,yBAA0B,SAAQ,oBAA2B;IACzE,YAAY,UAAsB;QACjC,KAAK,CAAC,0BAAe,EAAE,UAAU,CAAC,OAAO,CAAC,CAAC;IAC5C,CAAC;IAMD,KAAK,CAAC,WAAW,CAAC,KAAmC;QACpD,MAAM,OAAO,GAAG,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,EAAE,CACtC,IAAI,CAAC,MAAM,CAAC;YACX,MAAM,EAAE,KAAK;YACb,KAAK,EAAE,CAAC,CAAC,KAAK;YACd,WAAW,EAAE,CAAC,CAAC,WAAW,IAAI,IAAI;YAIlC,QAAQ,EAAE,CAAC,CAAC,QAAQ,IAAI,KAAK;YAC7B,KAAK,EAAE,CAAC,CAAC,KAAK,IAAI,IAAI;SACtB,CAAC,CACF,CAAC;QAEF,OAAO,MAAM,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,EAAE,KAAK,EAAE,eAAe,EAAE,CAAC,CAAC;IAC7D,CAAC;IAOD,KAAK,CAAC,aAAa,CAAC,KAAa;QAChC,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,kBAAkB,CAAC,QAAQ,CAAC;aAClD,MAAM,CAAC,eAAe,EAAE,QAAQ,CAAC;aACjC,SAAS,CAAC,UAAU,EAAE,OAAO,CAAC;aAC9B,KAAK,CAAC,uBAAuB,EAAE,EAAE,KAAK,EAAE,CAAC;aACzC,OAAO,CAAC,eAAe,CAAC;aACxB,UAAU,EAA6D,CAAC;QAE1E,MAAM,MAAM,GAAgC;YAC3C,GAAG,EAAE,CAAC;YACN,OAAO,EAAE,CAAC;YACV,OAAO,EAAE,CAAC;YACV,KAAK,EAAE,CAAC;YACR,SAAS,EAAE,CAAC;SACZ,CAAC;QACF,KAAK,MAAM,GAAG,IAAI,IAAI;YAAE,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,GAAG,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;QAC/D,OAAO,MAAM,CAAC;IACf,CAAC;IAED,KAAK,CAAC,aAAa,CAAC,EAAU;QAC7B,OAAO,MAAM,IAAI,CAAC,MAAM,CAAC,EAAE,EAAE,EAAE,MAAM,EAAE,SAAS,EAAE,KAAK,EAAE,IAAI,IAAI,EAAE,EAAE,CAAC,CAAC;IACxE,CAAC;IAED,KAAK,CAAC,eAAe,CAAC,EAAU;QAC/B,OAAO,MAAM,IAAI,CAAC,MAAM,CAAC,EAAE,EAAE,EAAE,MAAM,EAAE,WAAW,EAAE,WAAW,EAAE,IAAI,IAAI,EAAE,EAAE,CAAC,CAAC;IAChF,CAAC;IAED,KAAK,CAAC,eAAe,CACpB,EAAU,EACV,KAIC;QAED,OAAO,MAAM,IAAI,CAAC,MAAM,CAAC,EAAE,EAAE;YAC5B,MAAM,EAAE,SAAS;YACjB,WAAW,EAAE,IAAI,IAAI,EAAE;YACvB,MAAM,EAAE,KAAK,CAAC,MAAM;YACpB,SAAS,EAAE,KAAK,CAAC,SAAS,IAAI,IAAI;YAClC,OAAO,EAAE,KAAK,CAAC,OAAO,IAAI,IAAI;SAC9B,CAAC,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,WAAW,CAAC,EAAU,EAAE,SAAiB,EAAE,YAAiC;QACjF,OAAO,MAAM,IAAI,CAAC,MAAM,CAAC,EAAE,EAAE;YAC5B,MAAM,EAAE,OAAO;YACf,WAAW,EAAE,IAAI,IAAI,EAAE;YACvB,SAAS;YACT,YAAY,EAAE,YAAY,IAAI,IAAI;SAClC,CAAC,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,WAAW,CAAC,KAAa;QAC9B,OAAO,MAAM,IAAI,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,EAAE,KAAK,EAAE,EAAE,KAAK,EAAE,EAAE,QAAQ,EAAE,KAAK,EAAE,EAAE,CAAC,CAAC;IAC1E,CAAC;IAED,KAAK,CAAC,QAAQ,CAAC,EAAU;QACxB,OAAO,MAAM,IAAI,CAAC,SAAS,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC;IACrC,CAAC;CACD,CAAA;;oCA3FY,yBAAyB;IADrC,IAAA,YAAO,GAAE;qCAEe,oBAAU;GADtB,yBAAyB,CA2FrC"}

package/dist/repositories/agent-eval-run.repository.ee.d.ts

@@ -0,0 +1,24 @@
+import { DataSource, Repository } from '@n8n/typeorm';
+import type { IDataObject } from 'n8n-workflow';
+import { AgentEvalRun } from '../entities';
+type CreateAgentEvalRunAttrs = {
+    datasetId: string;
+    agentVersionId?: string | null;
+    createdById?: string | null;
+};
+export declare class AgentEvalRunRepository extends Repository<AgentEvalRun> {
+    constructor(dataSource: DataSource);
+    createRun(attrs: CreateAgentEvalRunAttrs): Promise<AgentEvalRun>;
+    markAsRunning(id: string, instanceId?: string): Promise<import("@n8n/typeorm").UpdateResult>;
+    markAsCompleted(id: string, metrics: IDataObject | null): Promise<import("@n8n/typeorm").UpdateResult>;
+    markAsError(id: string, errorCode: string, errorDetails?: IDataObject | null, metrics?: IDataObject | null): Promise<import("@n8n/typeorm").UpdateResult>;
+    markAsCancelled(id: string, metrics?: IDataObject | null): Promise<import("@n8n/typeorm").UpdateResult>;
+    requestCancellation(id: string): Promise<import("@n8n/typeorm").UpdateResult>;
+    findByDatasetId(datasetId: string): Promise<AgentEvalRun[]>;
+    findById(id: string): Promise<AgentEvalRun | null>;
+    findByIdAndAgentId(id: string, agentId: string): Promise<AgentEvalRun | null>;
+    findByDatasetIdAndAgentId(datasetId: string, agentId: string): Promise<AgentEvalRun[]>;
+    markAllIncompleteAsError(): Promise<import("@n8n/typeorm").UpdateResult>;
+    isCancellationRequested(id: string): Promise<boolean>;
+}
+export {};

package/dist/repositories/agent-eval-run.repository.ee.js

@@ -0,0 +1,99 @@
+"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.AgentEvalRunRepository = void 0;
+const di_1 = require("@n8n/di");
+const typeorm_1 = require("@n8n/typeorm");
+const entities_1 = require("../entities");
+let AgentEvalRunRepository = class AgentEvalRunRepository extends typeorm_1.Repository {
+    constructor(dataSource) {
+        super(entities_1.AgentEvalRun, dataSource.manager);
+    }
+    async createRun(attrs) {
+        const run = this.create({
+            status: 'new',
+            datasetId: attrs.datasetId,
+            agentVersionId: attrs.agentVersionId ?? null,
+            createdById: attrs.createdById ?? null,
+            cancelRequested: false,
+        });
+        return await this.save(run);
+    }
+    async markAsRunning(id, instanceId) {
+        return await this.update(id, {
+            status: 'running',
+            runAt: new Date(),
+            runningInstanceId: instanceId ?? null,
+        });
+    }
+    async markAsCompleted(id, metrics) {
+        return await this.update(id, {
+            status: 'completed',
+            completedAt: new Date(),
+            metrics,
+            runningInstanceId: null,
+        });
+    }
+    async markAsError(id, errorCode, errorDetails, metrics) {
+        return await this.update(id, {
+            status: 'error',
+            completedAt: new Date(),
+            errorCode,
+            errorDetails: errorDetails ?? null,
+            metrics: metrics ?? null,
+            runningInstanceId: null,
+        });
+    }
+    async markAsCancelled(id, metrics = null) {
+        return await this.update(id, {
+            status: 'cancelled',
+            completedAt: new Date(),
+            metrics,
+            runningInstanceId: null,
+        });
+    }
+    async requestCancellation(id) {
+        return await this.update(id, { cancelRequested: true });
+    }
+    async findByDatasetId(datasetId) {
+        return await this.find({ where: { datasetId }, order: { createdAt: 'DESC' } });
+    }
+    async findById(id) {
+        return await this.findOneBy({ id });
+    }
+    async findByIdAndAgentId(id, agentId) {
+        return await this.findOne({ where: { id, dataset: { agentId } } });
+    }
+    async findByDatasetIdAndAgentId(datasetId, agentId) {
+        return await this.find({
+            where: { datasetId, dataset: { agentId } },
+            order: { createdAt: 'DESC' },
+        });
+    }
+    async markAllIncompleteAsError() {
+        return await this.update({ status: (0, typeorm_1.In)(['new', 'running']) }, {
+            status: 'error',
+            errorCode: 'interrupted',
+            completedAt: new Date(),
+            runningInstanceId: null,
+        });
+    }
+    async isCancellationRequested(id) {
+        const run = await this.findOne({ where: { id }, select: ['id', 'cancelRequested'] });
+        return run?.cancelRequested ?? false;
+    }
+};
+exports.AgentEvalRunRepository = AgentEvalRunRepository;
+exports.AgentEvalRunRepository = AgentEvalRunRepository = __decorate([
+    (0, di_1.Service)(),
+    __metadata("design:paramtypes", [typeorm_1.DataSource])
+], AgentEvalRunRepository);
+//# sourceMappingURL=agent-eval-run.repository.ee.js.map

package/dist/repositories/agent-eval-run.repository.ee.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"agent-eval-run.repository.ee.js","sourceRoot":"","sources":["../../src/repositories/agent-eval-run.repository.ee.ts"],"names":[],"mappings":";;;;;;;;;;;;AAAA,gCAAkC;AAClC,0CAA0D;AAG1D,0CAA2C;AASpC,IAAM,sBAAsB,GAA5B,MAAM,sBAAuB,SAAQ,oBAAwB;IACnE,YAAY,UAAsB;QACjC,KAAK,CAAC,uBAAY,EAAE,UAAU,CAAC,OAAO,CAAC,CAAC;IACzC,CAAC;IAED,KAAK,CAAC,SAAS,CAAC,KAA8B;QAC7C,MAAM,GAAG,GAAG,IAAI,CAAC,MAAM,CAAC;YACvB,MAAM,EAAE,KAAK;YACb,SAAS,EAAE,KAAK,CAAC,SAAS;YAC1B,cAAc,EAAE,KAAK,CAAC,cAAc,IAAI,IAAI;YAC5C,WAAW,EAAE,KAAK,CAAC,WAAW,IAAI,IAAI;YACtC,eAAe,EAAE,KAAK;SACtB,CAAC,CAAC;QAEH,OAAO,MAAM,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;IAC7B,CAAC;IAED,KAAK,CAAC,aAAa,CAAC,EAAU,EAAE,UAAmB;QAClD,OAAO,MAAM,IAAI,CAAC,MAAM,CAAC,EAAE,EAAE;YAC5B,MAAM,EAAE,SAAS;YACjB,KAAK,EAAE,IAAI,IAAI,EAAE;YACjB,iBAAiB,EAAE,UAAU,IAAI,IAAI;SACrC,CAAC,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,eAAe,CAAC,EAAU,EAAE,OAA2B;QAC5D,OAAO,MAAM,IAAI,CAAC,MAAM,CAAC,EAAE,EAAE;YAC5B,MAAM,EAAE,WAAW;YACnB,WAAW,EAAE,IAAI,IAAI,EAAE;YACvB,OAAO;YACP,iBAAiB,EAAE,IAAI;SACvB,CAAC,CAAC;IACJ,CAAC;IAMD,KAAK,CAAC,WAAW,CAChB,EAAU,EACV,SAAiB,EACjB,YAAiC,EACjC,OAA4B;QAE5B,OAAO,MAAM,IAAI,CAAC,MAAM,CAAC,EAAE,EAAE;YAC5B,MAAM,EAAE,OAAO;YACf,WAAW,EAAE,IAAI,IAAI,EAAE;YACvB,SAAS;YACT,YAAY,EAAE,YAAY,IAAI,IAAI;YAClC,OAAO,EAAE,OAAO,IAAI,IAAI;YACxB,iBAAiB,EAAE,IAAI;SACvB,CAAC,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,eAAe,CAAC,EAAU,EAAE,OAAO,GAAuB,IAAI;QACnE,OAAO,MAAM,IAAI,CAAC,MAAM,CAAC,EAAE,EAAE;YAC5B,MAAM,EAAE,WAAW;YACnB,WAAW,EAAE,IAAI,IAAI,EAAE;YACvB,OAAO;YACP,iBAAiB,EAAE,IAAI;SACvB,CAAC,CAAC;IACJ,CAAC;IAMD,KAAK,CAAC,mBAAmB,CAAC,EAAU;QACnC,OAAO,MAAM,IAAI,CAAC,MAAM,CAAC,EAAE,EAAE,EAAE,eAAe,EAAE,IAAI,EAAE,CAAC,CAAC;IACzD,CAAC;IAED,KAAK,CAAC,eAAe,CAAC,SAAiB;QACtC,OAAO,MAAM,IAAI,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,EAAE,SAAS,EAAE,EAAE,KAAK,EAAE,EAAE,SAAS,EAAE,MAAM,EAAE,EAAE,CAAC,CAAC;IAChF,CAAC;IAED,KAAK,CAAC,QAAQ,CAAC,EAAU;QACxB,OAAO,MAAM,IAAI,CAAC,SAAS,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC;IACrC,CAAC;IAID,KAAK,CAAC,kBAAkB,CAAC,EAAU,EAAE,OAAe;QACnD,OAAO,MAAM,IAAI,CAAC,OAAO,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,EAAE,OAAO,EAAE,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC,CAAC;IACpE,CAAC;IAGD,KAAK,CAAC,yBAAyB,CAAC,SAAiB,EAAE,OAAe;QACjE,OAAO,MAAM,IAAI,CAAC,IAAI,CAAC;YACtB,KAAK,EAAE,EAAE,SAAS,EAAE,OAAO,EAAE,EAAE,OAAO,EAAE,EAAE;YAC1C,KAAK,EAAE,EAAE,SAAS,EAAE,MAAM,EAAE;SAC5B,CAAC,CAAC;IACJ,CAAC;IAQD,KAAK,CAAC,wBAAwB;QAC7B,OAAO,MAAM,IAAI,CAAC,MAAM,CACvB,EAAE,MAAM,EAAE,IAAA,YAAE,EAAC,CAAC,KAAK,EAAE,SAAS,CAAC,CAAC,EAAE,EAClC;YACC,MAAM,EAAE,OAAO;YACf,SAAS,EAAE,aAAa;YACxB,WAAW,EAAE,IAAI,IAAI,EAAE;YACvB,iBAAiB,EAAE,IAAI;SACvB,CACD,CAAC;IACH,CAAC;IAGD,KAAK,CAAC,uBAAuB,CAAC,EAAU;QACvC,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,EAAE,EAAE,MAAM,EAAE,CAAC,IAAI,EAAE,iBAAiB,CAAC,EAAE,CAAC,CAAC;QACrF,OAAO,GAAG,EAAE,eAAe,IAAI,KAAK,CAAC;IACtC,CAAC;CACD,CAAA;;iCApHY,sBAAsB;IADlC,IAAA,YAAO,GAAE;qCAEe,oBAAU;GADtB,sBAAsB,CAoHlC"}

package/dist/repositories/base-repository.d.ts

@@ -0,0 +1,6 @@
+import { Repository } from '@n8n/typeorm';
+import type { EntityManager, ObjectLiteral } from '@n8n/typeorm';
+import type { OperationContext } from '../services/transaction';
+export declare abstract class BaseRepository<Entity extends ObjectLiteral> extends Repository<Entity> {
+    protected managerFor(ctx: OperationContext): EntityManager;
+}

package/dist/repositories/base-repository.js

@@ -0,0 +1,19 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.BaseRepository = void 0;
+const typeorm_1 = require("@n8n/typeorm");
+const n8n_workflow_1 = require("n8n-workflow");
+const typeorm_transaction_1 = require("../services/typeorm-transaction");
+class BaseRepository extends typeorm_1.Repository {
+    managerFor(ctx) {
+        const { trx } = ctx;
+        if (!trx)
+            return this.manager;
+        if (!(trx instanceof typeorm_transaction_1.TypeOrmTransaction)) {
+            throw new n8n_workflow_1.UnexpectedError('Transaction was not created by the TypeORM runner');
+        }
+        return trx.getEntityManager();
+    }
+}
+exports.BaseRepository = BaseRepository;
+//# sourceMappingURL=base-repository.js.map

package/dist/repositories/base-repository.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"base-repository.js","sourceRoot":"","sources":["../../src/repositories/base-repository.ts"],"names":[],"mappings":";;;AAAA,0CAA0C;AAE1C,+CAA+C;AAG/C,yEAAqE;AAUrE,oBAAmE,SAAQ,oBAAkB;IAMlF,UAAU,CAAC,GAAqB;QACzC,MAAM,EAAE,GAAG,EAAE,GAAG,GAAG,CAAC;QACpB,IAAI,CAAC,GAAG;YAAE,OAAO,IAAI,CAAC,OAAO,CAAC;QAE9B,IAAI,CAAC,CAAC,GAAG,YAAY,wCAAkB,CAAC,EAAE,CAAC;YAC1C,MAAM,IAAI,8BAAe,CAAC,mDAAmD,CAAC,CAAC;QAChF,CAAC;QAED,OAAO,GAAG,CAAC,gBAAgB,EAAE,CAAC;IAC/B,CAAC;CACD"}

package/dist/repositories/binary-data.repository.d.ts

@@ -4,6 +4,8 @@ import { BinaryDataFile } from '../entities';
 export declare class BinaryDataRepository extends Repository<BinaryDataFile> {
     private readonly databaseConfig;
     constructor(dataSource: DataSource, databaseConfig: DatabaseConfig);
+    findContentByFileId(fileId: string): Promise<Buffer | null>;
+    deleteByFileIds(fileIds: string[]): Promise<void>;
     copyStoredFile(sourceFileId: string, targetFileId: string, targetSourceType: string, targetSourceId: string): Promise<boolean>;
     private getTableName;
     private getColumnName;

package/dist/repositories/binary-data.repository.js

@@ -20,6 +20,15 @@ let BinaryDataRepository = class BinaryDataRepository extends typeorm_1.Reposito
         super(entities_1.BinaryDataFile, dataSource.manager);
         this.databaseConfig = databaseConfig;
     }
+    async findContentByFileId(fileId) {
+        const file = await this.findOne({ where: { fileId }, select: ['data'] });
+        return file?.data ?? null;
+    }
+    async deleteByFileIds(fileIds) {
+        if (fileIds.length === 0)
+            return;
+        await this.delete({ fileId: (0, typeorm_1.In)(fileIds) });
+    }
     async copyStoredFile(sourceFileId, targetFileId, targetSourceType, targetSourceId) {
         const tableName = this.getTableName('binary_data');
         const fileId = this.getColumnName('fileId');

package/dist/repositories/binary-data.repository.js.map

@@ -1 +1 @@
-{"version":3,"file":"binary-data.repository.js","sourceRoot":"","sources":["../../src/repositories/binary-data.repository.ts"],"names":[],"mappings":";;;;;;;;;;;;AAAA,wCAA6C;AAC7C,gCAAkC;AAClC,0CAAsD;AAEtD,0CAA6C;AAC7C,iEAAqD;AAG9C,IAAM,oBAAoB,GAA1B,MAAM,oBAAqB,SAAQ,oBAA0B;IACnE,YACC,UAAsB,EACL,cAA8B;QAE/C,KAAK,CAAC,yBAAc,EAAE,UAAU,CAAC,OAAO,CAAC,CAAC;8BAFzB,cAAc;IAGhC,CAAC;IAED,KAAK,CAAC,cAAc,CACnB,YAAoB,EACpB,YAAoB,EACpB,gBAAwB,EACxB,cAAsB;QAEtB,MAAM,SAAS,GAAG,IAAI,CAAC,YAAY,CAAC,aAAa,CAAC,CAAC;QACnD,MAAM,MAAM,GAAG,IAAI,CAAC,aAAa,CAAC,QAAQ,CAAC,CAAC;QAC5C,MAAM,UAAU,GAAG,IAAI,CAAC,aAAa,CAAC,YAAY,CAAC,CAAC;QACpD,MAAM,QAAQ,GAAG,IAAI,CAAC,aAAa,CAAC,UAAU,CAAC,CAAC;QAChD,MAAM,IAAI,GAAG,IAAI,CAAC,aAAa,CAAC,MAAM,CAAC,CAAC;QACxC,MAAM,QAAQ,GAAG,IAAI,CAAC,aAAa,CAAC,UAAU,CAAC,CAAC;QAChD,MAAM,QAAQ,GAAG,IAAI,CAAC,aAAa,CAAC,UAAU,CAAC,CAAC;QAChD,MAAM,QAAQ,GAAG,IAAI,CAAC,aAAa,CAAC,UAAU,CAAC,CAAC;QAEhD,MAAM,CAAC,KAAK,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,CAAC,GACnC,wBAAM,KAAK,YAAY,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC,CAAC;QAE3E,MAAM,KAAK,GAAG;iBACC,SAAS,KAAK,MAAM,KAAK,UAAU,KAAK,QAAQ,KAAK,IAAI,KAAK,QAAQ,KAAK,QAAQ,KAAK,QAAQ;YACrG,KAAK,KAAK,MAAM,KAAK,KAAK,KAAK,IAAI,KAAK,QAAQ,KAAK,QAAQ,KAAK,QAAQ;UAC5E,SAAS;WACR,MAAM,MAAM,MAAM;GAC1B,CAAC;QAEF,MAAM,IAAI,GAAG,CAAC,YAAY,EAAE,gBAAgB,EAAE,cAAc,EAAE,YAAY,CAAC,CAAC;QAE5E,MAAM,IAAI,CAAC,KAAK,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC;QAE9B,OAAO,MAAM,IAAI,CAAC,QAAQ,CAAC,EAAE,MAAM,EAAE,YAAY,EAAE,CAAC,CAAC;IACtD,CAAC;IAEO,YAAY,CAAC,IAAY;QAChC,MAAM,EAAE,WAAW,EAAE,GAAG,IAAI,CAAC,cAAc,CAAC;QAC5C,OAAO,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC,MAAM,CAAC,MAAM,CAAC,GAAG,WAAW,GAAG,IAAI,EAAE,CAAC,CAAC;IACvE,CAAC;IAEO,aAAa,CAAC,IAAY;QACjC,OAAO,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;IACpD,CAAC;CACD,CAAA;;+BAhDY,oBAAoB;IADhC,IAAA,YAAO,GAAE;qCAGI,oBAAU,EACW,uBAAc;GAHpC,oBAAoB,CAgDhC"}
+{"version":3,"file":"binary-data.repository.js","sourceRoot":"","sources":["../../src/repositories/binary-data.repository.ts"],"names":[],"mappings":";;;;;;;;;;;;AAAA,wCAA6C;AAC7C,gCAAkC;AAClC,0CAA0D;AAE1D,0CAA6C;AAC7C,iEAAqD;AAG9C,IAAM,oBAAoB,GAA1B,MAAM,oBAAqB,SAAQ,oBAA0B;IACnE,YACC,UAAsB,EACL,cAA8B;QAE/C,KAAK,CAAC,yBAAc,EAAE,UAAU,CAAC,OAAO,CAAC,CAAC;8BAFzB,cAAc;IAGhC,CAAC;IAGD,KAAK,CAAC,mBAAmB,CAAC,MAAc;QACvC,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,EAAE,KAAK,EAAE,EAAE,MAAM,EAAE,EAAE,MAAM,EAAE,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;QAEzE,OAAO,IAAI,EAAE,IAAI,IAAI,IAAI,CAAC;IAC3B,CAAC;IAED,KAAK,CAAC,eAAe,CAAC,OAAiB;QACtC,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO;QAEjC,MAAM,IAAI,CAAC,MAAM,CAAC,EAAE,MAAM,EAAE,IAAA,YAAE,EAAC,OAAO,CAAC,EAAE,CAAC,CAAC;IAC5C,CAAC;IAED,KAAK,CAAC,cAAc,CACnB,YAAoB,EACpB,YAAoB,EACpB,gBAAwB,EACxB,cAAsB;QAEtB,MAAM,SAAS,GAAG,IAAI,CAAC,YAAY,CAAC,aAAa,CAAC,CAAC;QACnD,MAAM,MAAM,GAAG,IAAI,CAAC,aAAa,CAAC,QAAQ,CAAC,CAAC;QAC5C,MAAM,UAAU,GAAG,IAAI,CAAC,aAAa,CAAC,YAAY,CAAC,CAAC;QACpD,MAAM,QAAQ,GAAG,IAAI,CAAC,aAAa,CAAC,UAAU,CAAC,CAAC;QAChD,MAAM,IAAI,GAAG,IAAI,CAAC,aAAa,CAAC,MAAM,CAAC,CAAC;QACxC,MAAM,QAAQ,GAAG,IAAI,CAAC,aAAa,CAAC,UAAU,CAAC,CAAC;QAChD,MAAM,QAAQ,GAAG,IAAI,CAAC,aAAa,CAAC,UAAU,CAAC,CAAC;QAChD,MAAM,QAAQ,GAAG,IAAI,CAAC,aAAa,CAAC,UAAU,CAAC,CAAC;QAEhD,MAAM,CAAC,KAAK,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,CAAC,GACnC,wBAAM,KAAK,YAAY,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC,CAAC;QAE3E,MAAM,KAAK,GAAG;iBACC,SAAS,KAAK,MAAM,KAAK,UAAU,KAAK,QAAQ,KAAK,IAAI,KAAK,QAAQ,KAAK,QAAQ,KAAK,QAAQ;YACrG,KAAK,KAAK,MAAM,KAAK,KAAK,KAAK,IAAI,KAAK,QAAQ,KAAK,QAAQ,KAAK,QAAQ;UAC5E,SAAS;WACR,MAAM,MAAM,MAAM;GAC1B,CAAC;QAEF,MAAM,IAAI,GAAG,CAAC,YAAY,EAAE,gBAAgB,EAAE,cAAc,EAAE,YAAY,CAAC,CAAC;QAE5E,MAAM,IAAI,CAAC,KAAK,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC;QAE9B,OAAO,MAAM,IAAI,CAAC,QAAQ,CAAC,EAAE,MAAM,EAAE,YAAY,EAAE,CAAC,CAAC;IACtD,CAAC;IAEO,YAAY,CAAC,IAAY;QAChC,MAAM,EAAE,WAAW,EAAE,GAAG,IAAI,CAAC,cAAc,CAAC;QAC5C,OAAO,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC,MAAM,CAAC,MAAM,CAAC,GAAG,WAAW,GAAG,IAAI,EAAE,CAAC,CAAC;IACvE,CAAC;IAEO,aAAa,CAAC,IAAY;QACjC,OAAO,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;IACpD,CAAC;CACD,CAAA;;+BA7DY,oBAAoB;IADhC,IAAA,YAAO,GAAE;qCAGI,oBAAU,EACW,uBAAc;GAHpC,oBAAoB,CA6DhC"}

package/dist/repositories/credentials.repository.d.ts

@@ -1,12 +1,29 @@
 import type { Scope } from '@n8n/permissions';
 import type { FindManyOptions } from '@n8n/typeorm';
-import { DataSource, Repository } from '@n8n/typeorm';
+import { DataSource } from '@n8n/typeorm';
 import { CredentialsEntity, type User } from '../entities';
+import { BaseRepository } from './base-repository';
 import { type CredentialDependencyFilter } from './credential-dependency.repository';
-import type { ListQuery } from '../entities/types-db';
-export declare class CredentialsRepository extends Repository<CredentialsEntity> {
-    constructor(dataSource: DataSource);
+import { InstanceCredentialAssignmentRepository } from './instance-credential-assignment.repository';
+import type { ICredentialsDb, ListQuery } from '../entities/types-db';
+import type { OperationContext } from '../services/transaction';
+export declare class CredentialsRepository extends BaseRepository<CredentialsEntity> {
+    private readonly instanceCredentialAssignmentRepository;
+    constructor(dataSource: DataSource, instanceCredentialAssignmentRepository: InstanceCredentialAssignmentRepository);
     findStartingWith(credentialName: string): Promise<CredentialsEntity[]>;
+    findNonProjectCredentialsByIds(ids: string[]): Promise<CredentialsEntity[]>;
+    findDanglingProjectCredentials(): Promise<CredentialsEntity[]>;
+    findInstanceCredentialById(credentialId: string, ctx: OperationContext): Promise<CredentialsEntity | null>;
+    saveInstanceCredential(credential: CredentialsEntity, ctx: OperationContext): Promise<CredentialsEntity>;
+    updateInstanceCredential(credentialId: string, data: Pick<ICredentialsDb, 'id' | 'name' | 'type' | 'data'>, ctx: OperationContext): Promise<CredentialsEntity | null>;
+    deleteInstanceCredentialIfUnassigned(credentialId: string, ctx?: OperationContext): Promise<{
+        status: 'deleted';
+    } | {
+        status: 'notFound';
+    } | {
+        status: 'assigned';
+        credentialUseIds: string[];
+    }>;
     findMany(listQueryOptions?: ListQuery.Options & {
         includeData?: boolean;
         user?: User;
@@ -17,6 +34,7 @@ export declare class CredentialsRepository extends Repository<CredentialsEntity>
         user?: User;
         order?: FindManyOptions<CredentialsEntity>['order'];
     }, credentialIds?: string[]): Promise<[CredentialsEntity[], number]>;
+    private onlyProjectCredentials;
     private toFindManyOptions;
     private handleSharedFilters;
     getManyByIds(ids: string[], { withSharings }?: {

package/dist/repositories/credentials.repository.js

@@ -13,31 +13,96 @@ exports.CredentialsRepository = void 0;
 const di_1 = require("@n8n/di");
 const typeorm_1 = require("@n8n/typeorm");
 const entities_1 = require("../entities");
+const base_repository_1 = require("./base-repository");
 const credential_dependency_repository_1 = require("./credential-dependency.repository");
+const instance_credential_assignment_repository_1 = require("./instance-credential-assignment.repository");
 const shared_credentials_repository_1 = require("./shared-credentials.repository");
-let CredentialsRepository = class CredentialsRepository extends typeorm_1.Repository {
-    constructor(dataSource) {
+let CredentialsRepository = class CredentialsRepository extends base_repository_1.BaseRepository {
+    constructor(dataSource, instanceCredentialAssignmentRepository) {
         super(entities_1.CredentialsEntity, dataSource.manager);
+        this.instanceCredentialAssignmentRepository = instanceCredentialAssignmentRepository;
     }
     async findStartingWith(credentialName) {
         return await this.find({
             select: ['name'],
-            where: { name: (0, typeorm_1.Like)(`${credentialName}%`) },
+            where: { name: (0, typeorm_1.Like)(`${credentialName}%`), usageScope: 'project' },
         });
     }
+    async findNonProjectCredentialsByIds(ids) {
+        return await this.find({
+            where: { id: (0, typeorm_1.In)(ids), usageScope: (0, typeorm_1.Not)('project') },
+            select: ['id'],
+        });
+    }
+    async findDanglingProjectCredentials() {
+        return await this.createQueryBuilder('credentials')
+            .leftJoinAndSelect('credentials.shared', 'shared')
+            .where('shared.credentialsId is null')
+            .andWhere('credentials.usageScope = :usageScope', { usageScope: 'project' })
+            .getMany();
+    }
+    async findInstanceCredentialById(credentialId, ctx) {
+        return await this.managerFor(ctx).findOneBy(entities_1.CredentialsEntity, {
+            id: credentialId,
+            usageScope: 'instance',
+        });
+    }
+    async saveInstanceCredential(credential, ctx) {
+        return await this.managerFor(ctx).save(entities_1.CredentialsEntity, credential);
+    }
+    async updateInstanceCredential(credentialId, data, ctx) {
+        const manager = this.managerFor(ctx);
+        await manager.update(entities_1.CredentialsEntity, { id: credentialId, usageScope: 'instance' }, data);
+        return await manager.findOneBy(entities_1.CredentialsEntity, {
+            id: credentialId,
+            usageScope: 'instance',
+        });
+    }
+    async deleteInstanceCredentialIfUnassigned(credentialId, ctx = {}) {
+        const manager = this.managerFor(ctx);
+        const credential = await manager.findOneBy(entities_1.CredentialsEntity, {
+            id: credentialId,
+            usageScope: 'instance',
+        });
+        if (!credential)
+            return { status: 'notFound' };
+        const credentialUseIds = await this.instanceCredentialAssignmentRepository.findCredentialUseIds(credentialId, ctx);
+        if (credentialUseIds.length > 0)
+            return { status: 'assigned', credentialUseIds };
+        try {
+            const result = await manager.delete(entities_1.CredentialsEntity, {
+                id: credentialId,
+                usageScope: 'instance',
+            });
+            return result.affected === 0 ? { status: 'notFound' } : { status: 'deleted' };
+        }
+        catch (error) {
+            if (error instanceof typeorm_1.QueryFailedError && !ctx.trx) {
+                const concurrentCredentialUseIds = await this.instanceCredentialAssignmentRepository.findCredentialUseIds(credentialId, ctx);
+                if (concurrentCredentialUseIds.length > 0) {
+                    return { status: 'assigned', credentialUseIds: concurrentCredentialUseIds };
+                }
+            }
+            throw error;
+        }
+    }
     async findMany(listQueryOptions, credentialIds) {
         const findManyOptions = this.toFindManyOptions(listQueryOptions);
         if (credentialIds) {
             findManyOptions.where = { ...findManyOptions.where, id: (0, typeorm_1.In)(credentialIds) };
         }
-        return await this.find(findManyOptions);
+        return await this.find(this.onlyProjectCredentials(findManyOptions));
     }
     async findManyAndCount(listQueryOptions, credentialIds) {
         const findManyOptions = this.toFindManyOptions(listQueryOptions);
         if (credentialIds) {
             findManyOptions.where = { ...findManyOptions.where, id: (0, typeorm_1.In)(credentialIds) };
         }
-        return await this.findAndCount(findManyOptions);
+        return await this.findAndCount(this.onlyProjectCredentials(findManyOptions));
+    }
+    onlyProjectCredentials(findManyOptions) {
+        findManyOptions.where = { ...findManyOptions.where, usageScope: 'project' };
+        return findManyOptions;
     }
     toFindManyOptions(listQueryOptions) {
         const findManyOptions = {};
@@ -128,7 +193,9 @@ let CredentialsRepository = class CredentialsRepository extends typeorm_1.Reposi
         }
     }
     async getManyByIds(ids, { withSharings } = { withSharings: false }) {
-        const findManyOptions = { where: { id: (0, typeorm_1.In)(ids) } };
+        const findManyOptions = {
+            where: { id: (0, typeorm_1.In)(ids), usageScope: 'project' },
+        };
         if (withSharings) {
             findManyOptions.relations = {
                 shared: {
@@ -152,6 +219,7 @@ let CredentialsRepository = class CredentialsRepository extends typeorm_1.Reposi
         findManyOptions.where = {
             ...findManyOptions.where,
             isGlobal: true,
+            usageScope: 'project',
             ...(type ? { type: (0, typeorm_1.Like)(`%${type}%`) } : {}),
         };
         return await this.find(findManyOptions);
@@ -160,6 +228,7 @@ let CredentialsRepository = class CredentialsRepository extends typeorm_1.Reposi
         const { includeData, dependencyFilter, type } = options;
         const qb = this.createQueryBuilder('credential');
         qb.where('credential.isGlobal = :isGlobal', { isGlobal: true });
+        qb.andWhere('credential.usageScope = :usageScope', { usageScope: 'project' });
         if (type) {
             qb.andWhere('credential.type LIKE :type', { type: `%${type}%` });
         }
@@ -186,18 +255,22 @@ let CredentialsRepository = class CredentialsRepository extends typeorm_1.Reposi
         return await qb.getMany();
     }
     async findAllPersonalCredentials() {
-        return await this.findBy({ shared: { project: { type: 'personal' } } });
+        return await this.findBy({
+            usageScope: 'project',
+            shared: { project: { type: 'personal' } },
+        });
     }
     async findAllCredentialsForWorkflow(workflowId) {
         return await this.findBy({
+            usageScope: 'project',
             shared: { project: { sharedWorkflows: { workflowId } } },
         });
     }
     async findAllCredentialsForProject(projectId) {
-        return await this.findBy({ shared: { projectId } });
+        return await this.findBy({ usageScope: 'project', shared: { projectId } });
     }
     async findByNameAndTypeInProject(name, type, projectId) {
-        return await this.findBy({ name, type, shared: { projectId } });
+        return await this.findBy({ name, type, usageScope: 'project', shared: { projectId } });
     }
     async getManyAndCountWithSharingSubquery(user, sharingOptions, options = {}) {
         const query = this.getManyQueryWithSharingSubquery(user, sharingOptions, options);
@@ -213,6 +286,7 @@ let CredentialsRepository = class CredentialsRepository extends typeorm_1.Reposi
     }
     getManyQueryWithSharingSubquery(user, sharingOptions, options = {}) {
         const qb = this.createQueryBuilder('credential');
+        qb.andWhere('credential.usageScope = :usageScope', { usageScope: 'project' });
         if (options.filters?.dependency) {
             (0, credential_dependency_repository_1.addCredentialDependencyExistsFilter)(qb, options.filters.dependency);
         }
@@ -282,6 +356,6 @@ let CredentialsRepository = class CredentialsRepository extends typeorm_1.Reposi
 exports.CredentialsRepository = CredentialsRepository;
 exports.CredentialsRepository = CredentialsRepository = __decorate([
     (0, di_1.Service)(),
-    __metadata("design:paramtypes", [typeorm_1.DataSource])
+    __metadata("design:paramtypes", [typeorm_1.DataSource, instance_credential_assignment_repository_1.InstanceCredentialAssignmentRepository])
 ], CredentialsRepository);
 //# sourceMappingURL=credentials.repository.js.map