n8n 2.22.4 → 2.23.0

package/dist/modules/instance-ai/storage/typeorm-agent-memory.js

@@ -0,0 +1,324 @@
+"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.TypeORMAgentMemory = void 0;
+const backend_common_1 = require("@n8n/backend-common");
+const di_1 = require("@n8n/di");
+const typeorm_1 = require("@n8n/typeorm");
+const typeorm_observation_log_store_1 = require("./typeorm-observation-log-store");
+const instance_ai_message_repository_1 = require("../repositories/instance-ai-message.repository");
+const instance_ai_observation_cursor_repository_1 = require("../repositories/instance-ai-observation-cursor.repository");
+const instance_ai_observation_lock_repository_1 = require("../repositories/instance-ai-observation-lock.repository");
+const instance_ai_observation_repository_1 = require("../repositories/instance-ai-observation.repository");
+const instance_ai_resource_repository_1 = require("../repositories/instance-ai-resource.repository");
+const instance_ai_thread_repository_1 = require("../repositories/instance-ai-thread.repository");
+function parseJsonSafe(text) {
+    try {
+        return JSON.parse(text);
+    }
+    catch {
+        return undefined;
+    }
+}
+function isRecord(value) {
+    return typeof value === 'object' && value !== null;
+}
+function isAgentMessage(value) {
+    if (!isRecord(value))
+        return false;
+    if (value.type === 'custom') {
+        return 'data' in value;
+    }
+    return typeof value.role === 'string' && Array.isArray(value.content);
+}
+function toThread(entity) {
+    return {
+        id: entity.id,
+        resourceId: entity.resourceId,
+        title: entity.title || undefined,
+        createdAt: entity.createdAt,
+        updatedAt: entity.updatedAt,
+        metadata: entity.metadata ?? undefined,
+    };
+}
+function getMessageRole(message) {
+    return 'role' in message ? message.role : 'custom';
+}
+function getMessageType(message) {
+    if (message.type === 'custom')
+        return 'custom';
+    if (message.type === 'llm')
+        return 'llm';
+    return null;
+}
+function toAgentMessage(entity) {
+    const parsed = parseJsonSafe(entity.content);
+    if (!isAgentMessage(parsed))
+        return undefined;
+    return { ...parsed, id: entity.id, createdAt: entity.createdAt };
+}
+function workingMemoryKey(params) {
+    return params.scope === 'thread' ? `thread:${params.threadId}` : params.resourceId;
+}
+const PATCH_ONLY_METADATA_KEYS = new Set([
+    'instanceAiIterationLog',
+    'instanceAiPlannedTasks',
+    'instanceAiTasks',
+    'instanceAiTerminalOutcomes',
+    'instanceAiWorkflowLoop',
+]);
+function cloneThreadForPatch(thread) {
+    return {
+        ...thread,
+        metadata: { ...(thread.metadata ?? {}) },
+    };
+}
+function mergeSaveThreadMetadata(current, incoming) {
+    const safeIncoming = { ...incoming };
+    for (const key of PATCH_ONLY_METADATA_KEYS) {
+        delete safeIncoming[key];
+    }
+    return {
+        ...(current ?? {}),
+        ...safeIncoming,
+    };
+}
+let TypeORMAgentMemory = class TypeORMAgentMemory {
+    constructor(threadRepo, messageRepo, resourceRepo, observationRepo, observationCursorRepo, observationLockRepo, logger) {
+        this.threadRepo = threadRepo;
+        this.messageRepo = messageRepo;
+        this.resourceRepo = resourceRepo;
+        this.threadMutationQueues = new Map();
+        this.logger = logger.scoped('instance-ai');
+        this.observationLog = new typeorm_observation_log_store_1.TypeORMObservationLogStore(observationRepo, observationCursorRepo, observationLockRepo, this.messageRepo, (entity) => this.toAgentMessage(entity));
+    }
+    describe() {
+        return {
+            name: 'typeorm-agent-memory',
+            constructorName: this.constructor.name,
+            connectionParams: null,
+        };
+    }
+    async getThread(threadId) {
+        const thread = await this.threadRepo.findOneBy({ id: threadId });
+        return thread ? toThread(thread) : null;
+    }
+    async listThreads(args) {
+        const perPage = args.perPage ?? 100;
+        const page = args.page ?? 0;
+        const field = args.orderBy?.field ?? 'updatedAt';
+        const direction = args.orderBy?.direction ?? 'DESC';
+        const [threads, total] = await this.threadRepo.findAndCount({
+            where: args.filter?.resourceId ? { resourceId: args.filter.resourceId } : {},
+            order: { [field]: direction, id: direction },
+            take: perPage,
+            skip: page * perPage,
+        });
+        return {
+            threads: threads.map(toThread),
+            total,
+            page,
+            hasMore: (page + 1) * perPage < total,
+        };
+    }
+    async saveThread(thread) {
+        return await this.serializeThreadMutation(thread.id, async () => {
+            const existing = await this.threadRepo.findOneBy({ id: thread.id });
+            if (existing) {
+                existing.resourceId = thread.resourceId;
+                if (thread.title !== undefined)
+                    existing.title = thread.title;
+                if (thread.metadata !== undefined) {
+                    existing.metadata = mergeSaveThreadMetadata(existing.metadata, thread.metadata);
+                }
+                return toThread(await this.threadRepo.save(existing));
+            }
+            const saved = await this.threadRepo.save(this.threadRepo.create({
+                id: thread.id,
+                resourceId: thread.resourceId,
+                title: thread.title ?? '',
+                metadata: thread.metadata ?? null,
+            }));
+            return toThread(saved);
+        });
+    }
+    async patchThread(args) {
+        return await this.serializeThreadMutation(args.threadId, async () => {
+            const existing = await this.threadRepo.findOneBy({ id: args.threadId });
+            if (!existing)
+                return null;
+            const current = toThread(existing);
+            const patch = args.update(cloneThreadForPatch(current));
+            if (!patch)
+                return current;
+            if (patch.title !== undefined)
+                existing.title = patch.title;
+            if (patch.metadata !== undefined)
+                existing.metadata = patch.metadata;
+            return toThread(await this.threadRepo.save(existing));
+        });
+    }
+    async deleteThread(threadId) {
+        await this.threadRepo.delete({ id: threadId });
+    }
+    async deleteThreadsByResourceIdPrefix(resourceIdPrefix) {
+        const threads = await this.threadRepo.find({
+            where: { resourceId: (0, typeorm_1.Like)(`${resourceIdPrefix}%`) },
+        });
+        if (threads.length === 0)
+            return;
+        const threadIds = threads.map((thread) => thread.id);
+        const resourceIds = Array.from(new Set(threads.map((thread) => thread.resourceId)));
+        await this.resourceRepo.delete({ id: (0, typeorm_1.In)(resourceIds) });
+        await this.resourceRepo.delete({
+            id: (0, typeorm_1.In)(threadIds.map((threadId) => `thread:${threadId}`)),
+        });
+        await this.threadRepo.delete({ id: (0, typeorm_1.In)(threadIds) });
+    }
+    async getMessages(threadId, opts) {
+        const where = opts?.before ? { threadId, createdAt: (0, typeorm_1.LessThan)(opts.before) } : { threadId };
+        const entities = await this.messageRepo.find({
+            where,
+            order: opts?.limit ? { createdAt: 'DESC', id: 'DESC' } : { createdAt: 'ASC', id: 'ASC' },
+            ...(opts?.limit ? { take: opts.limit } : {}),
+        });
+        const ordered = opts?.limit ? entities.reverse() : entities;
+        return ordered.flatMap((entity) => {
+            const message = this.toAgentMessage(entity);
+            return message ? [message] : [];
+        });
+    }
+    async listMessages(args) {
+        const limit = args.limit ?? 50;
+        const page = args.page ?? 0;
+        const entities = await this.messageRepo.find({
+            where: { threadId: args.threadId },
+            order: { createdAt: 'DESC', id: 'DESC' },
+            take: limit,
+            skip: page * limit,
+        });
+        return {
+            messages: entities.reverse().flatMap((entity) => {
+                const message = this.toAgentMessage(entity);
+                return message ? [message] : [];
+            }),
+        };
+    }
+    async saveMessages(args) {
+        if (args.messages.length === 0)
+            return;
+        const entities = args.messages.map((message) => this.messageRepo.create({
+            id: message.id,
+            threadId: args.threadId,
+            content: JSON.stringify(message),
+            role: getMessageRole(message),
+            type: getMessageType(message),
+            resourceId: args.resourceId,
+            createdAt: message.createdAt,
+            updatedAt: message.createdAt,
+        }));
+        await this.messageRepo.save(entities);
+    }
+    async deleteMessages(messageIds) {
+        if (messageIds.length === 0)
+            return;
+        await this.messageRepo.delete({ id: (0, typeorm_1.In)(messageIds) });
+    }
+    async getWorkingMemory(params) {
+        const resource = await this.resourceRepo.findOneBy({ id: workingMemoryKey(params) });
+        return resource?.workingMemory ?? null;
+    }
+    async saveWorkingMemory(params, content) {
+        const id = workingMemoryKey(params);
+        const existing = await this.resourceRepo.findOneBy({ id });
+        if (existing) {
+            existing.workingMemory = content;
+            await this.resourceRepo.save(existing);
+            return;
+        }
+        await this.resourceRepo.save(this.resourceRepo.create({
+            id,
+            workingMemory: content,
+            metadata: { scope: params.scope },
+        }));
+    }
+    async appendObservationLogEntries(rows) {
+        return await this.observationLog.appendObservationLogEntries(rows);
+    }
+    async getActiveObservationLog(scope) {
+        return await this.observationLog.getActiveObservationLog(scope);
+    }
+    async getObservationLog(opts) {
+        return await this.observationLog.getObservationLog(opts);
+    }
+    async getMessagesForObservationScope(observationScopeId, opts) {
+        return await this.observationLog.getMessagesForObservationScope(observationScopeId, opts);
+    }
+    async dropObservationLogEntries(ids) {
+        return await this.observationLog.dropObservationLogEntries(ids);
+    }
+    async supersedeObservationLogEntries(ids, supersededBy) {
+        return await this.observationLog.supersedeObservationLogEntries(ids, supersededBy);
+    }
+    async applyObservationLogReflection(scope, reflection) {
+        return await this.observationLog.applyObservationLogReflection(scope, reflection);
+    }
+    async getCursor(observationScopeId) {
+        return await this.observationLog.getCursor(observationScopeId);
+    }
+    async setCursor(cursor) {
+        return await this.observationLog.setCursor(cursor);
+    }
+    async acquireObservationLogTaskLock(observationScopeId, taskKind, opts) {
+        return await this.observationLog.acquireObservationLogTaskLock(observationScopeId, taskKind, opts);
+    }
+    async releaseObservationLogTaskLock(handle) {
+        return await this.observationLog.releaseObservationLogTaskLock(handle);
+    }
+    async serializeThreadMutation(threadId, mutation) {
+        const previous = this.threadMutationQueues.get(threadId) ?? Promise.resolve();
+        const next = previous.catch(() => undefined).then(mutation);
+        this.threadMutationQueues.set(threadId, next);
+        try {
+            return await next;
+        }
+        finally {
+            if (this.threadMutationQueues.get(threadId) === next) {
+                this.threadMutationQueues.delete(threadId);
+            }
+        }
+    }
+    toAgentMessage(entity) {
+        const message = toAgentMessage(entity);
+        if (!message) {
+            this.logger.warn('Skipping invalid Instance AI message row', {
+                messageId: entity.id,
+                threadId: entity.threadId,
+                resourceId: entity.resourceId,
+                role: entity.role,
+                type: entity.type,
+            });
+        }
+        return message;
+    }
+};
+exports.TypeORMAgentMemory = TypeORMAgentMemory;
+exports.TypeORMAgentMemory = TypeORMAgentMemory = __decorate([
+    (0, di_1.Service)(),
+    __metadata("design:paramtypes", [instance_ai_thread_repository_1.InstanceAiThreadRepository,
+        instance_ai_message_repository_1.InstanceAiMessageRepository,
+        instance_ai_resource_repository_1.InstanceAiResourceRepository,
+        instance_ai_observation_repository_1.InstanceAiObservationRepository,
+        instance_ai_observation_cursor_repository_1.InstanceAiObservationCursorRepository,
+        instance_ai_observation_lock_repository_1.InstanceAiObservationLockRepository,
+        backend_common_1.Logger])
+], TypeORMAgentMemory);
+//# sourceMappingURL=typeorm-agent-memory.js.map

package/dist/modules/instance-ai/storage/typeorm-agent-memory.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"typeorm-agent-memory.js","sourceRoot":"","sources":["../../../../src/modules/instance-ai/storage/typeorm-agent-memory.ts"],"names":[],"mappings":";;;;;;;;;;;;AAcA,wDAA6C;AAC7C,gCAAkC;AAQlC,0CAAkD;AAElD,mFAA6E;AAG7E,mGAA6F;AAC7F,yHAAkH;AAClH,qHAA8G;AAC9G,2GAAqG;AACrG,qGAA+F;AAC/F,iGAA2F;AAE3F,SAAS,aAAa,CAAC,IAAY;IAClC,IAAI,CAAC;QACJ,OAAO,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;IACzB,CAAC;IAAC,MAAM,CAAC;QACR,OAAO,SAAS,CAAC;IAClB,CAAC;AACF,CAAC;AAED,SAAS,QAAQ,CAAC,KAAc;IAC/B,OAAO,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,KAAK,IAAI,CAAC;AACpD,CAAC;AAED,SAAS,cAAc,CAAC,KAAc;IACrC,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC;QAAE,OAAO,KAAK,CAAC;IAEnC,IAAI,KAAK,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;QAC7B,OAAO,MAAM,IAAI,KAAK,CAAC;IACxB,CAAC;IAED,OAAO,OAAO,KAAK,CAAC,IAAI,KAAK,QAAQ,IAAI,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;AACvE,CAAC;AAED,SAAS,QAAQ,CAAC,MAAwB;IACzC,OAAO;QACN,EAAE,EAAE,MAAM,CAAC,EAAE;QACb,UAAU,EAAE,MAAM,CAAC,UAAU;QAC7B,KAAK,EAAE,MAAM,CAAC,KAAK,IAAI,SAAS;QAChC,SAAS,EAAE,MAAM,CAAC,SAAS;QAC3B,SAAS,EAAE,MAAM,CAAC,SAAS;QAC3B,QAAQ,EAAE,MAAM,CAAC,QAAQ,IAAI,SAAS;KACtC,CAAC;AACH,CAAC;AAED,SAAS,cAAc,CAAC,OAAuB;IAC9C,OAAO,MAAM,IAAI,OAAO,CAAC,CAAC,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,QAAQ,CAAC;AACpD,CAAC;AAED,SAAS,cAAc,CAAC,OAAuB;IAC9C,IAAI,OAAO,CAAC,IAAI,KAAK,QAAQ;QAAE,OAAO,QAAQ,CAAC;IAC/C,IAAI,OAAO,CAAC,IAAI,KAAK,KAAK;QAAE,OAAO,KAAK,CAAC;IACzC,OAAO,IAAI,CAAC;AACb,CAAC;AAED,SAAS,cAAc,CAAC,MAAyB;IAChD,MAAM,MAAM,GAAG,aAAa,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;IAC7C,IAAI,CAAC,cAAc,CAAC,MAAM,CAAC;QAAE,OAAO,SAAS,CAAC;IAC9C,OAAO,EAAE,GAAG,MAAM,EAAE,EAAE,EAAE,MAAM,CAAC,EAAE,EAAE,SAAS,EAAE,MAAM,CAAC,SAAS,EAAE,CAAC;AAClE,CAAC;AAED,SAAS,gBAAgB,CAAC,MAIzB;IACA,OAAO,MAAM,CAAC,KAAK,KAAK,QAAQ,CAAC,CAAC,CAAC,UAAU,MAAM,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAC,MAAM,CAAC,UAAU,CAAC;AACpF,CAAC;AAED,MAAM,wBAAwB,GAAG,IAAI,GAAG,CAAC;IACxC,wBAAwB;IACxB,wBAAwB;IACxB,iBAAiB;IACjB,4BAA4B;IAC5B,wBAAwB;CACxB,CAAC,CAAC;AAEH,SAAS,mBAAmB,CAAC,MAAc;IAC1C,OAAO;QACN,GAAG,MAAM;QACT,QAAQ,EAAE,EAAE,GAAG,CAAC,MAAM,CAAC,QAAQ,IAAI,EAAE,CAAC,EAAE;KACxC,CAAC;AACH,CAAC;AAED,SAAS,uBAAuB,CAC/B,OAAmD,EACnD,QAAiC;IAEjC,MAAM,YAAY,GAAG,EAAE,GAAG,QAAQ,EAAE,CAAC;IACrC,KAAK,MAAM,GAAG,IAAI,wBAAwB,EAAE,CAAC;QAC5C,OAAO,YAAY,CAAC,GAAG,CAAC,CAAC;IAC1B,CAAC;IACD,OAAO;QACN,GAAG,CAAC,OAAO,IAAI,EAAE,CAAC;QAClB,GAAG,YAAY;KACf,CAAC;AACH,CAAC;AAGM,IAAM,kBAAkB,GAAxB,MAAM,kBAAkB;IAM9B,YACkB,UAAsC,EACtC,WAAwC,EACxC,YAA0C,EAC3D,eAAgD,EAChD,qBAA4D,EAC5D,mBAAwD,EACxD,MAAc;QANG,eAAU,GAAV,UAAU,CAA4B;QACtC,gBAAW,GAAX,WAAW,CAA6B;QACxC,iBAAY,GAAZ,YAAY,CAA8B;QAN3C,yBAAoB,GAAG,IAAI,GAAG,EAA4B,CAAC;QAY3E,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC;QAC3C,IAAI,CAAC,cAAc,GAAG,IAAI,0DAA0B,CACnD,eAAe,EACf,qBAAqB,EACrB,mBAAmB,EACnB,IAAI,CAAC,WAAW,EAChB,CAAC,MAAM,EAAE,EAAE,CAAC,IAAI,CAAC,cAAc,CAAC,MAAM,CAAC,CACvC,CAAC;IACH,CAAC;IAID,QAAQ;QACP,OAAO;YACN,IAAI,EAAE,sBAAsB;YAC5B,eAAe,EAAE,IAAI,CAAC,WAAW,CAAC,IAAI;YACtC,gBAAgB,EAAE,IAAI;SACtB,CAAC;IACH,CAAC;IAED,KAAK,CAAC,SAAS,CAAC,QAAgB;QAC/B,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,SAAS,CAAC,EAAE,EAAE,EAAE,QAAQ,EAAE,CAAC,CAAC;QACjE,OAAO,MAAM,CAAC,CAAC,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC;IACzC,CAAC;IAED,KAAK,CAAC,WAAW,CAAC,IAKjB;QACA,MAAM,OAAO,GAAG,IAAI,CAAC,OAAO,IAAI,GAAG,CAAC;QACpC,MAAM,IAAI,GAAG,IAAI,CAAC,IAAI,IAAI,CAAC,CAAC;QAC5B,MAAM,KAAK,GAAG,IAAI,CAAC,OAAO,EAAE,KAAK,IAAI,WAAW,CAAC;QACjD,MAAM,SAAS,GAAG,IAAI,CAAC,OAAO,EAAE,SAAS,IAAI,MAAM,CAAC;QACpD,MAAM,CAAC,OAAO,EAAE,KAAK,CAAC,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,YAAY,CAAC;YAC3D,KAAK,EAAE,IAAI,CAAC,MAAM,EAAE,UAAU,CAAC,CAAC,CAAC,EAAE,UAAU,EAAE,IAAI,CAAC,MAAM,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC,EAAE;YAC5E,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,EAAE,SAAS,EAAE,EAAE,EAAE,SAAS,EAAE;YAC5C,IAAI,EAAE,OAAO;YACb,IAAI,EAAE,IAAI,GAAG,OAAO;SACpB,CAAC,CAAC;QAEH,OAAO;YACN,OAAO,EAAE,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC;YAC9B,KAAK;YACL,IAAI;YACJ,OAAO,EAAE,CAAC,IAAI,GAAG,CAAC,CAAC,GAAG,OAAO,GAAG,KAAK;SACrC,CAAC;IACH,CAAC;IAED,KAAK,CAAC,UAAU,CAAC,MAA+C;QAC/D,OAAO,MAAM,IAAI,CAAC,uBAAuB,CAAC,MAAM,CAAC,EAAE,EAAE,KAAK,IAAI,EAAE;YAC/D,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,SAAS,CAAC,EAAE,EAAE,EAAE,MAAM,CAAC,EAAE,EAAE,CAAC,CAAC;YACpE,IAAI,QAAQ,EAAE,CAAC;gBACd,QAAQ,CAAC,UAAU,GAAG,MAAM,CAAC,UAAU,CAAC;gBACxC,IAAI,MAAM,CAAC,KAAK,KAAK,SAAS;oBAAE,QAAQ,CAAC,KAAK,GAAG,MAAM,CAAC,KAAK,CAAC;gBAC9D,IAAI,MAAM,CAAC,QAAQ,KAAK,SAAS,EAAE,CAAC;oBACnC,QAAQ,CAAC,QAAQ,GAAG,uBAAuB,CAAC,QAAQ,CAAC,QAAQ,EAAE,MAAM,CAAC,QAAQ,CAAC,CAAC;gBACjF,CAAC;gBACD,OAAO,QAAQ,CAAC,MAAM,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC;YACvD,CAAC;YAED,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,IAAI,CACvC,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC;gBACtB,EAAE,EAAE,MAAM,CAAC,EAAE;gBACb,UAAU,EAAE,MAAM,CAAC,UAAU;gBAC7B,KAAK,EAAE,MAAM,CAAC,KAAK,IAAI,EAAE;gBACzB,QAAQ,EAAE,MAAM,CAAC,QAAQ,IAAI,IAAI;aACjC,CAAC,CACF,CAAC;YACF,OAAO,QAAQ,CAAC,KAAK,CAAC,CAAC;QACxB,CAAC,CAAC,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,WAAW,CAAC,IAGjB;QACA,OAAO,MAAM,IAAI,CAAC,uBAAuB,CAAC,IAAI,CAAC,QAAQ,EAAE,KAAK,IAAI,EAAE;YACnE,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,SAAS,CAAC,EAAE,EAAE,EAAE,IAAI,CAAC,QAAQ,EAAE,CAAC,CAAC;YACxE,IAAI,CAAC,QAAQ;gBAAE,OAAO,IAAI,CAAC;YAE3B,MAAM,OAAO,GAAG,QAAQ,CAAC,QAAQ,CAAC,CAAC;YACnC,MAAM,KAAK,GAAG,IAAI,CAAC,MAAM,CAAC,mBAAmB,CAAC,OAAO,CAAC,CAAC,CAAC;YACxD,IAAI,CAAC,KAAK;gBAAE,OAAO,OAAO,CAAC;YAE3B,IAAI,KAAK,CAAC,KAAK,KAAK,SAAS;gBAAE,QAAQ,CAAC,KAAK,GAAG,KAAK,CAAC,KAAK,CAAC;YAC5D,IAAI,KAAK,CAAC,QAAQ,KAAK,SAAS;gBAAE,QAAQ,CAAC,QAAQ,GAAG,KAAK,CAAC,QAAQ,CAAC;YACrE,OAAO,QAAQ,CAAC,MAAM,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC;QACvD,CAAC,CAAC,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,YAAY,CAAC,QAAgB;QAClC,MAAM,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,EAAE,EAAE,EAAE,QAAQ,EAAE,CAAC,CAAC;IAChD,CAAC;IAED,KAAK,CAAC,+BAA+B,CAAC,gBAAwB;QAC7D,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC;YAC1C,KAAK,EAAE,EAAE,UAAU,EAAE,IAAA,cAAI,EAAC,GAAG,gBAAgB,GAAG,CAAC,EAAE;SACnD,CAAC,CAAC;QACH,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO;QAEjC,MAAM,SAAS,GAAG,OAAO,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;QACrD,MAAM,WAAW,GAAG,KAAK,CAAC,IAAI,CAAC,IAAI,GAAG,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC;QAEpF,MAAM,IAAI,CAAC,YAAY,CAAC,MAAM,CAAC,EAAE,EAAE,EAAE,IAAA,YAAE,EAAC,WAAW,CAAC,EAAE,CAAC,CAAC;QACxD,MAAM,IAAI,CAAC,YAAY,CAAC,MAAM,CAAC;YAC9B,EAAE,EAAE,IAAA,YAAE,EAAC,SAAS,CAAC,GAAG,CAAC,CAAC,QAAQ,EAAE,EAAE,CAAC,UAAU,QAAQ,EAAE,CAAC,CAAC;SACzD,CAAC,CAAC;QACH,MAAM,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,EAAE,EAAE,EAAE,IAAA,YAAE,EAAC,SAAS,CAAC,EAAE,CAAC,CAAC;IACrD,CAAC;IAED,KAAK,CAAC,WAAW,CAChB,QAAgB,EAChB,IAAwC;QAExC,MAAM,KAAK,GAAG,IAAI,EAAE,MAAM,CAAC,CAAC,CAAC,EAAE,QAAQ,EAAE,SAAS,EAAE,IAAA,kBAAQ,EAAC,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,QAAQ,EAAE,CAAC;QAE3F,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC;YAC5C,KAAK;YACL,KAAK,EAAE,IAAI,EAAE,KAAK,CAAC,CAAC,CAAC,EAAE,SAAS,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,EAAE,CAAC,CAAC,CAAC,EAAE,SAAS,EAAE,KAAK,EAAE,EAAE,EAAE,KAAK,EAAE;YACxF,GAAG,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,IAAI,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;SAC5C,CAAC,CAAC;QAEH,MAAM,OAAO,GAAG,IAAI,EAAE,KAAK,CAAC,CAAC,CAAC,QAAQ,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC;QAC5D,OAAO,OAAO,CAAC,OAAO,CAAC,CAAC,MAAM,EAAE,EAAE;YACjC,MAAM,OAAO,GAAG,IAAI,CAAC,cAAc,CAAC,MAAM,CAAC,CAAC;YAC5C,OAAO,OAAO,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;QACjC,CAAC,CAAC,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,YAAY,CAAC,IAIlB;QACA,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,IAAI,EAAE,CAAC;QAC/B,MAAM,IAAI,GAAG,IAAI,CAAC,IAAI,IAAI,CAAC,CAAC;QAC5B,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC;YAC5C,KAAK,EAAE,EAAE,QAAQ,EAAE,IAAI,CAAC,QAAQ,EAAE;YAClC,KAAK,EAAE,EAAE,SAAS,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,EAAE;YACxC,IAAI,EAAE,KAAK;YACX,IAAI,EAAE,IAAI,GAAG,KAAK;SAClB,CAAC,CAAC;QAEH,OAAO;YACN,QAAQ,EAAE,QAAQ,CAAC,OAAO,EAAE,CAAC,OAAO,CAAC,CAAC,MAAM,EAAE,EAAE;gBAC/C,MAAM,OAAO,GAAG,IAAI,CAAC,cAAc,CAAC,MAAM,CAAC,CAAC;gBAC5C,OAAO,OAAO,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;YACjC,CAAC,CAAC;SACF,CAAC;IACH,CAAC;IAED,KAAK,CAAC,YAAY,CAAC,IAIlB;QACA,IAAI,IAAI,CAAC,QAAQ,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO;QAEvC,MAAM,QAAQ,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE,CAC9C,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC;YACvB,EAAE,EAAE,OAAO,CAAC,EAAE;YACd,QAAQ,EAAE,IAAI,CAAC,QAAQ;YACvB,OAAO,EAAE,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC;YAChC,IAAI,EAAE,cAAc,CAAC,OAAO,CAAC;YAC7B,IAAI,EAAE,cAAc,CAAC,OAAO,CAAC;YAC7B,UAAU,EAAE,IAAI,CAAC,UAAU;YAC3B,SAAS,EAAE,OAAO,CAAC,SAAS;YAC5B,SAAS,EAAE,OAAO,CAAC,SAAS;SAC5B,CAAC,CACF,CAAC;QAEF,MAAM,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;IACvC,CAAC;IAED,KAAK,CAAC,cAAc,CAAC,UAAoB;QACxC,IAAI,UAAU,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO;QACpC,MAAM,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,EAAE,EAAE,EAAE,IAAA,YAAE,EAAC,UAAU,CAAC,EAAE,CAAC,CAAC;IACvD,CAAC;IAED,KAAK,CAAC,gBAAgB,CAAC,MAItB;QACA,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,SAAS,CAAC,EAAE,EAAE,EAAE,gBAAgB,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;QACrF,OAAO,QAAQ,EAAE,aAAa,IAAI,IAAI,CAAC;IACxC,CAAC;IAED,KAAK,CAAC,iBAAiB,CACtB,MAA8E,EAC9E,OAAe;QAEf,MAAM,EAAE,GAAG,gBAAgB,CAAC,MAAM,CAAC,CAAC;QACpC,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,SAAS,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC;QAC3D,IAAI,QAAQ,EAAE,CAAC;YACd,QAAQ,CAAC,aAAa,GAAG,OAAO,CAAC;YACjC,MAAM,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;YACvC,OAAO;QACR,CAAC;QAED,MAAM,IAAI,CAAC,YAAY,CAAC,IAAI,CAC3B,IAAI,CAAC,YAAY,CAAC,MAAM,CAAC;YACxB,EAAE;YACF,aAAa,EAAE,OAAO;YACtB,QAAQ,EAAE,EAAE,KAAK,EAAE,MAAM,CAAC,KAAK,EAAE;SACjC,CAAC,CACF,CAAC;IACH,CAAC;IAED,KAAK,CAAC,2BAA2B,CAChC,IAA8B;QAE9B,OAAO,MAAM,IAAI,CAAC,cAAc,CAAC,2BAA2B,CAAC,IAAI,CAAC,CAAC;IACpE,CAAC;IAED,KAAK,CAAC,uBAAuB,CAC5B,KAAuE;QAEvE,OAAO,MAAM,IAAI,CAAC,cAAc,CAAC,uBAAuB,CAAC,KAAK,CAAC,CAAC;IACjE,CAAC;IAED,KAAK,CAAC,iBAAiB,CAAC,IAA+B;QACtD,OAAO,MAAM,IAAI,CAAC,cAAc,CAAC,iBAAiB,CAAC,IAAI,CAAC,CAAC;IAC1D,CAAC;IAED,KAAK,CAAC,8BAA8B,CACnC,kBAA0B,EAC1B,IAAmE;QAEnE,OAAO,MAAM,IAAI,CAAC,cAAc,CAAC,8BAA8B,CAAC,kBAAkB,EAAE,IAAI,CAAC,CAAC;IAC3F,CAAC;IAED,KAAK,CAAC,yBAAyB,CAAC,GAAa;QAC5C,OAAO,MAAM,IAAI,CAAC,cAAc,CAAC,yBAAyB,CAAC,GAAG,CAAC,CAAC;IACjE,CAAC;IAED,KAAK,CAAC,8BAA8B,CAAC,GAAa,EAAE,YAAoB;QACvE,OAAO,MAAM,IAAI,CAAC,cAAc,CAAC,8BAA8B,CAAC,GAAG,EAAE,YAAY,CAAC,CAAC;IACpF,CAAC;IAED,KAAK,CAAC,6BAA6B,CAClC,KAA0B,EAC1B,UAAoC;QAEpC,OAAO,MAAM,IAAI,CAAC,cAAc,CAAC,6BAA6B,CAAC,KAAK,EAAE,UAAU,CAAC,CAAC;IACnF,CAAC;IAED,KAAK,CAAC,SAAS,CAAC,kBAA0B;QACzC,OAAO,MAAM,IAAI,CAAC,cAAc,CAAC,SAAS,CAAC,kBAAkB,CAAC,CAAC;IAChE,CAAC;IAED,KAAK,CAAC,SAAS,CAAC,MAAyB;QACxC,OAAO,MAAM,IAAI,CAAC,cAAc,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;IACpD,CAAC;IAED,KAAK,CAAC,6BAA6B,CAClC,kBAA0B,EAC1B,QAAgC,EAChC,IAAyC;QAEzC,OAAO,MAAM,IAAI,CAAC,cAAc,CAAC,6BAA6B,CAC7D,kBAAkB,EAClB,QAAQ,EACR,IAAI,CACJ,CAAC;IACH,CAAC;IAED,KAAK,CAAC,6BAA6B,CAAC,MAAoC;QACvE,OAAO,MAAM,IAAI,CAAC,cAAc,CAAC,6BAA6B,CAAC,MAAM,CAAC,CAAC;IACxE,CAAC;IAEO,KAAK,CAAC,uBAAuB,CACpC,QAAgB,EAChB,QAA0B;QAE1B,MAAM,QAAQ,GAAG,IAAI,CAAC,oBAAoB,CAAC,GAAG,CAAC,QAAQ,CAAC,IAAI,OAAO,CAAC,OAAO,EAAE,CAAC;QAC9E,MAAM,IAAI,GAAG,QAAQ,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAC5D,IAAI,CAAC,oBAAoB,CAAC,GAAG,CAAC,QAAQ,EAAE,IAAI,CAAC,CAAC;QAE9C,IAAI,CAAC;YACJ,OAAO,MAAM,IAAI,CAAC;QACnB,CAAC;gBAAS,CAAC;YACV,IAAI,IAAI,CAAC,oBAAoB,CAAC,GAAG,CAAC,QAAQ,CAAC,KAAK,IAAI,EAAE,CAAC;gBACtD,IAAI,CAAC,oBAAoB,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;YAC5C,CAAC;QACF,CAAC;IACF,CAAC;IAEO,cAAc,CAAC,MAAyB;QAC/C,MAAM,OAAO,GAAG,cAAc,CAAC,MAAM,CAAC,CAAC;QACvC,IAAI,CAAC,OAAO,EAAE,CAAC;YACd,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,0CAA0C,EAAE;gBAC5D,SAAS,EAAE,MAAM,CAAC,EAAE;gBACpB,QAAQ,EAAE,MAAM,CAAC,QAAQ;gBACzB,UAAU,EAAE,MAAM,CAAC,UAAU;gBAC7B,IAAI,EAAE,MAAM,CAAC,IAAI;gBACjB,IAAI,EAAE,MAAM,CAAC,IAAI;aACjB,CAAC,CAAC;QACJ,CAAC;QACD,OAAO,OAAO,CAAC;IAChB,CAAC;CACD,CAAA;AA9TY,gDAAkB;6BAAlB,kBAAkB;IAD9B,IAAA,YAAO,GAAE;qCAQqB,0DAA0B;QACzB,4DAA2B;QAC1B,8DAA4B;QAC1C,oEAA+B;QACzB,iFAAqC;QACvC,6EAAmC;QAChD,uBAAM;GAbH,kBAAkB,CA8T9B"}

package/dist/modules/instance-ai/storage/typeorm-observation-log-store.d.ts

@@ -0,0 +1,37 @@
+import { type BuiltObservationLogStore, type BuiltObservationLogTaskLockStore, type NewObservationLogEntry, type ObservationCursor, type ObservationLogEntry, type ObservationLogReadOptions, type ObservationLogReflection, type ObservationLogReflectionResult, type ObservationLogScope, type ObservationLogTaskKind, type ObservationLogTaskLockHandle } from '@n8n/agents';
+import type { AgentDbMessage } from '@n8n/instance-ai';
+import type { InstanceAiMessage } from '../entities/instance-ai-message.entity';
+import type { InstanceAiMessageRepository } from '../repositories/instance-ai-message.repository';
+import type { InstanceAiObservationCursorRepository } from '../repositories/instance-ai-observation-cursor.repository';
+import type { InstanceAiObservationLockRepository } from '../repositories/instance-ai-observation-lock.repository';
+import type { InstanceAiObservationRepository } from '../repositories/instance-ai-observation.repository';
+export declare class TypeORMObservationLogStore implements BuiltObservationLogStore, BuiltObservationLogTaskLockStore {
+    private readonly observationRepo;
+    private readonly observationCursorRepo;
+    private readonly observationLockRepo;
+    private readonly messageRepo;
+    private readonly toAgentMessage;
+    constructor(observationRepo: InstanceAiObservationRepository, observationCursorRepo: InstanceAiObservationCursorRepository, observationLockRepo: InstanceAiObservationLockRepository, messageRepo: InstanceAiMessageRepository, toAgentMessage: (entity: InstanceAiMessage) => AgentDbMessage | undefined);
+    appendObservationLogEntries(rows: NewObservationLogEntry[]): Promise<ObservationLogEntry[]>;
+    getActiveObservationLog(scope: ObservationLogScope & {
+        limit?: number;
+        order?: 'asc' | 'desc';
+    }): Promise<ObservationLogEntry[]>;
+    getObservationLog(opts: ObservationLogReadOptions): Promise<ObservationLogEntry[]>;
+    getMessagesForObservationScope(observationScopeId: string, opts?: {
+        since?: {
+            sinceCreatedAt: Date;
+            sinceMessageId: string;
+        };
+    }): Promise<AgentDbMessage[]>;
+    dropObservationLogEntries(ids: string[]): Promise<void>;
+    supersedeObservationLogEntries(ids: string[], supersededBy: string): Promise<void>;
+    applyObservationLogReflection(scope: ObservationLogScope, reflection: ObservationLogReflection): Promise<ObservationLogReflectionResult>;
+    getCursor(observationScopeId: string): Promise<ObservationCursor | null>;
+    setCursor(cursor: ObservationCursor): Promise<void>;
+    acquireObservationLogTaskLock(observationScopeId: string, taskKind: ObservationLogTaskKind, opts: {
+        ttlMs: number;
+        holderId: string;
+    }): Promise<ObservationLogTaskLockHandle | null>;
+    releaseObservationLogTaskLock(handle: ObservationLogTaskLockHandle): Promise<void>;
+}

package/dist/modules/instance-ai/storage/typeorm-observation-log-store.js

@@ -0,0 +1,212 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.TypeORMObservationLogStore = void 0;
+const agents_1 = require("@n8n/agents");
+const typeorm_1 = require("@n8n/typeorm");
+const instance_ai_observation_lock_entity_1 = require("../entities/instance-ai-observation-lock.entity");
+const instance_ai_observation_entity_1 = require("../entities/instance-ai-observation.entity");
+function toObservationLogEntry(entity) {
+    return {
+        id: entity.id,
+        observationScopeId: entity.observationScopeId,
+        marker: entity.marker,
+        text: entity.text,
+        parentId: entity.parentId,
+        tokenCount: Number(entity.tokenCount),
+        status: entity.status,
+        supersededBy: entity.supersededBy,
+        createdAt: entity.createdAt,
+    };
+}
+class TypeORMObservationLogStore {
+    constructor(observationRepo, observationCursorRepo, observationLockRepo, messageRepo, toAgentMessage) {
+        this.observationRepo = observationRepo;
+        this.observationCursorRepo = observationCursorRepo;
+        this.observationLockRepo = observationLockRepo;
+        this.messageRepo = messageRepo;
+        this.toAgentMessage = toAgentMessage;
+    }
+    async appendObservationLogEntries(rows) {
+        if (rows.length === 0)
+            return [];
+        const entities = rows.map((row) => this.observationRepo.create({
+            observationScopeId: row.observationScopeId,
+            marker: row.marker,
+            text: row.text,
+            parentId: row.parentId ?? null,
+            tokenCount: row.tokenCount ?? (0, agents_1.estimateObservationTokens)(row.text),
+            ...(0, agents_1.activeLifecycleState)(),
+            createdAt: row.createdAt,
+        }));
+        const saved = await this.observationRepo.save(entities);
+        return saved.map((entity) => toObservationLogEntry(entity));
+    }
+    async getActiveObservationLog(scope) {
+        return await this.getObservationLog({ ...scope, status: 'active' });
+    }
+    async getObservationLog(opts) {
+        const baseWhere = {
+            observationScopeId: opts.observationScopeId,
+            ...(opts.status !== undefined && { status: opts.status }),
+            ...(opts.parentId !== undefined && { parentId: opts.parentId ?? (0, typeorm_1.IsNull)() }),
+        };
+        const entities = await this.observationRepo.find({
+            where: [baseWhere],
+            order: {
+                createdAt: opts.order === 'desc' ? 'DESC' : 'ASC',
+                id: opts.order === 'desc' ? 'DESC' : 'ASC',
+            },
+            ...(opts.limit !== undefined && { take: opts.limit }),
+        });
+        return entities.map((entity) => toObservationLogEntry(entity));
+    }
+    async getMessagesForObservationScope(observationScopeId, opts) {
+        const baseWhere = {
+            threadId: observationScopeId,
+        };
+        const where = opts?.since
+            ? [
+                { ...baseWhere, createdAt: (0, typeorm_1.MoreThan)(opts.since.sinceCreatedAt) },
+                {
+                    ...baseWhere,
+                    createdAt: (0, typeorm_1.Equal)(opts.since.sinceCreatedAt),
+                    id: (0, typeorm_1.MoreThan)(opts.since.sinceMessageId),
+                },
+            ]
+            : [baseWhere];
+        const entities = await this.messageRepo.find({
+            where,
+            order: { createdAt: 'ASC', id: 'ASC' },
+        });
+        return entities.flatMap((entity) => {
+            const message = this.toAgentMessage(entity);
+            return message ? [message] : [];
+        });
+    }
+    async dropObservationLogEntries(ids) {
+        if (ids.length === 0)
+            return;
+        await this.observationRepo.update({ id: (0, typeorm_1.In)(ids) }, (0, agents_1.droppedLifecycleState)());
+    }
+    async supersedeObservationLogEntries(ids, supersededBy) {
+        if (ids.length === 0)
+            return;
+        await this.observationRepo.update({ id: (0, typeorm_1.In)(ids) }, (0, agents_1.supersededLifecycleState)(supersededBy));
+    }
+    async applyObservationLogReflection(scope, reflection) {
+        return await this.observationRepo.manager.transaction(async (trx) => {
+            const repo = trx.getRepository(instance_ai_observation_entity_1.InstanceAiObservation);
+            const activeEntries = await repo.find({
+                where: {
+                    observationScopeId: scope.observationScopeId,
+                    status: 'active',
+                },
+                order: { createdAt: 'ASC', id: 'ASC' },
+            });
+            const normalized = (0, agents_1.normalizeObservationLogReflection)(activeEntries.map((entry) => toObservationLogEntry(entry)), reflection);
+            const inserted = normalized.merge.length
+                ? await repo.save(normalized.merge.map((entry) => repo.create({
+                    observationScopeId: scope.observationScopeId,
+                    marker: entry.marker,
+                    text: entry.text,
+                    parentId: entry.parentId ?? null,
+                    tokenCount: entry.tokenCount ?? (0, agents_1.estimateObservationTokens)(entry.text),
+                    ...(0, agents_1.activeLifecycleState)(),
+                    createdAt: entry.createdAt,
+                })))
+                : [];
+            if (normalized.drop.length > 0) {
+                await repo.update({
+                    observationScopeId: scope.observationScopeId,
+                    id: (0, typeorm_1.In)(normalized.drop),
+                }, (0, agents_1.droppedLifecycleState)());
+            }
+            for (const [index, merge] of normalized.merge.entries()) {
+                const replacement = inserted[index];
+                if (replacement && merge.supersedes.length > 0) {
+                    await repo.update({
+                        observationScopeId: scope.observationScopeId,
+                        id: (0, typeorm_1.In)(merge.supersedes),
+                    }, (0, agents_1.supersededLifecycleState)(replacement.id));
+                }
+            }
+            return {
+                droppedIds: [...normalized.drop],
+                supersededIds: normalized.merge.flatMap((entry) => entry.supersedes),
+                inserted: inserted.map((entry) => toObservationLogEntry(entry)),
+            };
+        });
+    }
+    async getCursor(observationScopeId) {
+        const entity = await this.observationCursorRepo.findOneBy({ observationScopeId });
+        if (!entity)
+            return null;
+        return {
+            observationScopeId: entity.observationScopeId,
+            lastObservedMessageId: entity.lastObservedMessageId,
+            lastObservedAt: entity.lastObservedAt,
+            updatedAt: entity.updatedAt,
+        };
+    }
+    async setCursor(cursor) {
+        await this.observationCursorRepo.upsert({
+            observationScopeId: cursor.observationScopeId,
+            lastObservedMessageId: cursor.lastObservedMessageId,
+            lastObservedAt: cursor.lastObservedAt,
+            updatedAt: cursor.updatedAt,
+        }, {
+            conflictPaths: ['observationScopeId'],
+            skipUpdateIfNoValuesChanged: false,
+        });
+    }
+    async acquireObservationLogTaskLock(observationScopeId, taskKind, opts) {
+        const now = new Date();
+        const heldUntil = new Date(now.getTime() + opts.ttlMs);
+        const updateResult = await this.observationLockRepo
+            .createQueryBuilder()
+            .update(instance_ai_observation_lock_entity_1.InstanceAiObservationLock)
+            .set({ taskKind, holderId: opts.holderId, heldUntil })
+            .where('"observationScopeId" = :observationScopeId')
+            .andWhere('"taskKind" = :taskKind')
+            .andWhere('("holderId" = :holderId OR "heldUntil" <= :now)')
+            .setParameters({
+            observationScopeId,
+            taskKind,
+            holderId: opts.holderId,
+            now,
+        })
+            .execute();
+        if ((updateResult.affected ?? 0) > 0) {
+            return { observationScopeId, taskKind, holderId: opts.holderId, heldUntil };
+        }
+        await this.observationLockRepo
+            .createQueryBuilder()
+            .insert()
+            .into(instance_ai_observation_lock_entity_1.InstanceAiObservationLock)
+            .values({
+            observationScopeId,
+            taskKind,
+            holderId: opts.holderId,
+            heldUntil,
+        })
+            .orIgnore()
+            .execute();
+        const claimed = await this.observationLockRepo.findOneBy({
+            observationScopeId,
+            taskKind,
+            holderId: opts.holderId,
+        });
+        if (!claimed)
+            return null;
+        return { observationScopeId, taskKind, holderId: opts.holderId, heldUntil };
+    }
+    async releaseObservationLogTaskLock(handle) {
+        await this.observationLockRepo.delete({
+            observationScopeId: handle.observationScopeId,
+            taskKind: handle.taskKind,
+            holderId: handle.holderId,
+        });
+    }
+}
+exports.TypeORMObservationLogStore = TypeORMObservationLogStore;
+//# sourceMappingURL=typeorm-observation-log-store.js.map

package/dist/modules/instance-ai/storage/typeorm-observation-log-store.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"typeorm-observation-log-store.js","sourceRoot":"","sources":["../../../../src/modules/instance-ai/storage/typeorm-observation-log-store.ts"],"names":[],"mappings":";;;AAAA,wCAiBqB;AAErB,0CAAkF;AAGlF,yGAA4F;AAC5F,+FAAmF;AAMnF,SAAS,qBAAqB,CAAC,MAA6B;IAC3D,OAAO;QACN,EAAE,EAAE,MAAM,CAAC,EAAE;QACb,kBAAkB,EAAE,MAAM,CAAC,kBAAkB;QAC7C,MAAM,EAAE,MAAM,CAAC,MAAM;QACrB,IAAI,EAAE,MAAM,CAAC,IAAI;QACjB,QAAQ,EAAE,MAAM,CAAC,QAAQ;QACzB,UAAU,EAAE,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC;QACrC,MAAM,EAAE,MAAM,CAAC,MAAM;QACrB,YAAY,EAAE,MAAM,CAAC,YAAY;QACjC,SAAS,EAAE,MAAM,CAAC,SAAS;KAC3B,CAAC;AACH,CAAC;AAED,MAAa,0BAA0B;IAGtC,YACkB,eAAgD,EAChD,qBAA4D,EAC5D,mBAAwD,EACxD,WAAwC,EACxC,cAAyE;QAJzE,oBAAe,GAAf,eAAe,CAAiC;QAChD,0BAAqB,GAArB,qBAAqB,CAAuC;QAC5D,wBAAmB,GAAnB,mBAAmB,CAAqC;QACxD,gBAAW,GAAX,WAAW,CAA6B;QACxC,mBAAc,GAAd,cAAc,CAA2D;IACxF,CAAC;IAEJ,KAAK,CAAC,2BAA2B,CAChC,IAA8B;QAE9B,IAAI,IAAI,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO,EAAE,CAAC;QAEjC,MAAM,QAAQ,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CACjC,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC;YAC3B,kBAAkB,EAAE,GAAG,CAAC,kBAAkB;YAC1C,MAAM,EAAE,GAAG,CAAC,MAAM;YAClB,IAAI,EAAE,GAAG,CAAC,IAAI;YACd,QAAQ,EAAE,GAAG,CAAC,QAAQ,IAAI,IAAI;YAC9B,UAAU,EAAE,GAAG,CAAC,UAAU,IAAI,IAAA,kCAAyB,EAAC,GAAG,CAAC,IAAI,CAAC;YACjE,GAAG,IAAA,6BAAoB,GAAE;YACzB,SAAS,EAAE,GAAG,CAAC,SAAS;SACxB,CAAC,CACF,CAAC;QAEF,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAExD,OAAO,KAAK,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,qBAAqB,CAAC,MAAM,CAAC,CAAC,CAAC;IAC7D,CAAC;IAED,KAAK,CAAC,uBAAuB,CAC5B,KAAuE;QAEvE,OAAO,MAAM,IAAI,CAAC,iBAAiB,CAAC,EAAE,GAAG,KAAK,EAAE,MAAM,EAAE,QAAQ,EAAE,CAAC,CAAC;IACrE,CAAC;IAED,KAAK,CAAC,iBAAiB,CAAC,IAA+B;QACtD,MAAM,SAAS,GAA4C;YAC1D,kBAAkB,EAAE,IAAI,CAAC,kBAAkB;YAC3C,GAAG,CAAC,IAAI,CAAC,MAAM,KAAK,SAAS,IAAI,EAAE,MAAM,EAAE,IAAI,CAAC,MAAM,EAAE,CAAC;YACzD,GAAG,CAAC,IAAI,CAAC,QAAQ,KAAK,SAAS,IAAI,EAAE,QAAQ,EAAE,IAAI,CAAC,QAAQ,IAAI,IAAA,gBAAM,GAAE,EAAE,CAAC;SAC3E,CAAC;QAEF,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC;YAChD,KAAK,EAAE,CAAC,SAAS,CAAC;YAClB,KAAK,EAAE;gBACN,SAAS,EAAE,IAAI,CAAC,KAAK,KAAK,MAAM,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,KAAK;gBACjD,EAAE,EAAE,IAAI,CAAC,KAAK,KAAK,MAAM,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,KAAK;aAC1C;YACD,GAAG,CAAC,IAAI,CAAC,KAAK,KAAK,SAAS,IAAI,EAAE,IAAI,EAAE,IAAI,CAAC,KAAK,EAAE,CAAC;SACrD,CAAC,CAAC;QACH,OAAO,QAAQ,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,qBAAqB,CAAC,MAAM,CAAC,CAAC,CAAC;IAChE,CAAC;IAED,KAAK,CAAC,8BAA8B,CACnC,kBAA0B,EAC1B,IAAmE;QAEnE,MAAM,SAAS,GAAwC;YACtD,QAAQ,EAAE,kBAAkB;SAC5B,CAAC;QAEF,MAAM,KAAK,GAA+C,IAAI,EAAE,KAAK;YACpE,CAAC,CAAC;gBACA,EAAE,GAAG,SAAS,EAAE,SAAS,EAAE,IAAA,kBAAQ,EAAC,IAAI,CAAC,KAAK,CAAC,cAAc,CAAC,EAAE;gBAChE;oBACC,GAAG,SAAS;oBACZ,SAAS,EAAE,IAAA,eAAK,EAAC,IAAI,CAAC,KAAK,CAAC,cAAc,CAAC;oBAC3C,EAAE,EAAE,IAAA,kBAAQ,EAAC,IAAI,CAAC,KAAK,CAAC,cAAc,CAAC;iBACvC;aACD;YACF,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC;QAEf,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC;YAC5C,KAAK;YACL,KAAK,EAAE,EAAE,SAAS,EAAE,KAAK,EAAE,EAAE,EAAE,KAAK,EAAE;SACtC,CAAC,CAAC;QAEH,OAAO,QAAQ,CAAC,OAAO,CAAC,CAAC,MAAM,EAAE,EAAE;YAClC,MAAM,OAAO,GAAG,IAAI,CAAC,cAAc,CAAC,MAAM,CAAC,CAAC;YAE5C,OAAO,OAAO,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;QACjC,CAAC,CAAC,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,yBAAyB,CAAC,GAAa;QAC5C,IAAI,GAAG,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO;QAE7B,MAAM,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC,EAAE,EAAE,EAAE,IAAA,YAAE,EAAC,GAAG,CAAC,EAAE,EAAE,IAAA,8BAAqB,GAAE,CAAC,CAAC;IAC7E,CAAC;IAED,KAAK,CAAC,8BAA8B,CAAC,GAAa,EAAE,YAAoB;QACvE,IAAI,GAAG,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO;QAE7B,MAAM,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC,EAAE,EAAE,EAAE,IAAA,YAAE,EAAC,GAAG,CAAC,EAAE,EAAE,IAAA,iCAAwB,EAAC,YAAY,CAAC,CAAC,CAAC;IAC5F,CAAC;IAED,KAAK,CAAC,6BAA6B,CAClC,KAA0B,EAC1B,UAAoC;QAEpC,OAAO,MAAM,IAAI,CAAC,eAAe,CAAC,OAAO,CAAC,WAAW,CAAC,KAAK,EAAE,GAAG,EAAE,EAAE;YACnE,MAAM,IAAI,GAAG,GAAG,CAAC,aAAa,CAAC,sDAAqB,CAAC,CAAC;YAEtD,MAAM,aAAa,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC;gBACrC,KAAK,EAAE;oBACN,kBAAkB,EAAE,KAAK,CAAC,kBAAkB;oBAC5C,MAAM,EAAE,QAAQ;iBAChB;gBACD,KAAK,EAAE,EAAE,SAAS,EAAE,KAAK,EAAE,EAAE,EAAE,KAAK,EAAE;aACtC,CAAC,CAAC;YAEH,MAAM,UAAU,GAAG,IAAA,0CAAiC,EACnD,aAAa,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,qBAAqB,CAAC,KAAK,CAAC,CAAC,EAC1D,UAAU,CACV,CAAC;YAEF,MAAM,QAAQ,GAAG,UAAU,CAAC,KAAK,CAAC,MAAM;gBACvC,CAAC,CAAC,MAAM,IAAI,CAAC,IAAI,CACf,UAAU,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAC9B,IAAI,CAAC,MAAM,CAAC;oBACX,kBAAkB,EAAE,KAAK,CAAC,kBAAkB;oBAC5C,MAAM,EAAE,KAAK,CAAC,MAAM;oBACpB,IAAI,EAAE,KAAK,CAAC,IAAI;oBAChB,QAAQ,EAAE,KAAK,CAAC,QAAQ,IAAI,IAAI;oBAChC,UAAU,EAAE,KAAK,CAAC,UAAU,IAAI,IAAA,kCAAyB,EAAC,KAAK,CAAC,IAAI,CAAC;oBACrE,GAAG,IAAA,6BAAoB,GAAE;oBACzB,SAAS,EAAE,KAAK,CAAC,SAAS;iBAC1B,CAAC,CACF,CACD;gBACF,CAAC,CAAC,EAAE,CAAC;YAEN,IAAI,UAAU,CAAC,IAAI,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBAChC,MAAM,IAAI,CAAC,MAAM,CAChB;oBACC,kBAAkB,EAAE,KAAK,CAAC,kBAAkB;oBAC5C,EAAE,EAAE,IAAA,YAAE,EAAC,UAAU,CAAC,IAAI,CAAC;iBACvB,EACD,IAAA,8BAAqB,GAAE,CACvB,CAAC;YACH,CAAC;YAED,KAAK,MAAM,CAAC,KAAK,EAAE,KAAK,CAAC,IAAI,UAAU,CAAC,KAAK,CAAC,OAAO,EAAE,EAAE,CAAC;gBACzD,MAAM,WAAW,GAAG,QAAQ,CAAC,KAAK,CAAC,CAAC;gBACpC,IAAI,WAAW,IAAI,KAAK,CAAC,UAAU,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;oBAChD,MAAM,IAAI,CAAC,MAAM,CAChB;wBACC,kBAAkB,EAAE,KAAK,CAAC,kBAAkB;wBAC5C,EAAE,EAAE,IAAA,YAAE,EAAC,KAAK,CAAC,UAAU,CAAC;qBACxB,EACD,IAAA,iCAAwB,EAAC,WAAW,CAAC,EAAE,CAAC,CACxC,CAAC;gBACH,CAAC;YACF,CAAC;YAED,OAAO;gBACN,UAAU,EAAE,CAAC,GAAG,UAAU,CAAC,IAAI,CAAC;gBAChC,aAAa,EAAE,UAAU,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,UAAU,CAAC;gBACpE,QAAQ,EAAE,QAAQ,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,qBAAqB,CAAC,KAAK,CAAC,CAAC;aAC/D,CAAC;QACH,CAAC,CAAC,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,SAAS,CAAC,kBAA0B;QACzC,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,qBAAqB,CAAC,SAAS,CAAC,EAAE,kBAAkB,EAAE,CAAC,CAAC;QAElF,IAAI,CAAC,MAAM;YAAE,OAAO,IAAI,CAAC;QAEzB,OAAO;YACN,kBAAkB,EAAE,MAAM,CAAC,kBAAkB;YAC7C,qBAAqB,EAAE,MAAM,CAAC,qBAAqB;YACnD,cAAc,EAAE,MAAM,CAAC,cAAc;YACrC,SAAS,EAAE,MAAM,CAAC,SAAS;SAC3B,CAAC;IACH,CAAC;IAED,KAAK,CAAC,SAAS,CAAC,MAAyB;QACxC,MAAM,IAAI,CAAC,qBAAqB,CAAC,MAAM,CACtC;YACC,kBAAkB,EAAE,MAAM,CAAC,kBAAkB;YAC7C,qBAAqB,EAAE,MAAM,CAAC,qBAAqB;YACnD,cAAc,EAAE,MAAM,CAAC,cAAc;YACrC,SAAS,EAAE,MAAM,CAAC,SAAS;SAC3B,EACD;YACC,aAAa,EAAE,CAAC,oBAAoB,CAAC;YACrC,2BAA2B,EAAE,KAAK;SAClC,CACD,CAAC;IACH,CAAC;IAED,KAAK,CAAC,6BAA6B,CAClC,kBAA0B,EAC1B,QAAgC,EAChC,IAAyC;QAEzC,MAAM,GAAG,GAAG,IAAI,IAAI,EAAE,CAAC;QACvB,MAAM,SAAS,GAAG,IAAI,IAAI,CAAC,GAAG,CAAC,OAAO,EAAE,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC;QAEvD,MAAM,YAAY,GAAG,MAAM,IAAI,CAAC,mBAAmB;aACjD,kBAAkB,EAAE;aACpB,MAAM,CAAC,+DAAyB,CAAC;aACjC,GAAG,CAAC,EAAE,QAAQ,EAAE,QAAQ,EAAE,IAAI,CAAC,QAAQ,EAAE,SAAS,EAAE,CAAC;aACrD,KAAK,CAAC,4CAA4C,CAAC;aACnD,QAAQ,CAAC,wBAAwB,CAAC;aAClC,QAAQ,CAAC,iDAAiD,CAAC;aAC3D,aAAa,CAAC;YACd,kBAAkB;YAClB,QAAQ;YACR,QAAQ,EAAE,IAAI,CAAC,QAAQ;YACvB,GAAG;SACH,CAAC;aACD,OAAO,EAAE,CAAC;QAEZ,IAAI,CAAC,YAAY,CAAC,QAAQ,IAAI,CAAC,CAAC,GAAG,CAAC,EAAE,CAAC;YACtC,OAAO,EAAE,kBAAkB,EAAE,QAAQ,EAAE,QAAQ,EAAE,IAAI,CAAC,QAAQ,EAAE,SAAS,EAAE,CAAC;QAC7E,CAAC;QAED,MAAM,IAAI,CAAC,mBAAmB;aAC5B,kBAAkB,EAAE;aACpB,MAAM,EAAE;aACR,IAAI,CAAC,+DAAyB,CAAC;aAC/B,MAAM,CAAC;YACP,kBAAkB;YAClB,QAAQ;YACR,QAAQ,EAAE,IAAI,CAAC,QAAQ;YACvB,SAAS;SACT,CAAC;aACD,QAAQ,EAAE;aACV,OAAO,EAAE,CAAC;QAEZ,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,mBAAmB,CAAC,SAAS,CAAC;YACxD,kBAAkB;YAClB,QAAQ;YACR,QAAQ,EAAE,IAAI,CAAC,QAAQ;SACvB,CAAC,CAAC;QAEH,IAAI,CAAC,OAAO;YAAE,OAAO,IAAI,CAAC;QAE1B,OAAO,EAAE,kBAAkB,EAAE,QAAQ,EAAE,QAAQ,EAAE,IAAI,CAAC,QAAQ,EAAE,SAAS,EAAE,CAAC;IAC7E,CAAC;IAED,KAAK,CAAC,6BAA6B,CAAC,MAAoC;QACvE,MAAM,IAAI,CAAC,mBAAmB,CAAC,MAAM,CAAC;YACrC,kBAAkB,EAAE,MAAM,CAAC,kBAAkB;YAC7C,QAAQ,EAAE,MAAM,CAAC,QAAQ;YACzB,QAAQ,EAAE,MAAM,CAAC,QAAQ;SACzB,CAAC,CAAC;IACJ,CAAC;CACD;AA7PD,gEA6PC"}