@hexabot-ai/api 3.0.0-alpha.3 → 3.1.1-alpha.0

package/dist/workflow/services/mcp-client-pool.service.js

@@ -0,0 +1,344 @@
+"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);
+};
+var McpClientPoolService_1;
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.McpClientPoolService = void 0;
+const mcp_1 = require("@ai-sdk/mcp");
+const common_1 = require("@nestjs/common");
+const logger_service_1 = require("../../logger/logger.service");
+const user_1 = require("../../user");
+const mcp_server_repository_1 = require("../repositories/mcp-server.repository");
+const types_1 = require("../types");
+const stdio_stderr_capture_transport_1 = require("./stdio-stderr-capture.transport");
+let McpClientPoolService = McpClientPoolService_1 = class McpClientPoolService {
+    constructor(mcpServerRepository, credentialService, logger) {
+        this.mcpServerRepository = mcpServerRepository;
+        this.credentialService = credentialService;
+        this.logger = logger;
+        this.sampledToolNamesCount = 10;
+        this.clientPool = new Map();
+        this.initFlights = new Map();
+        this.stdioCapturedStderr = new Map();
+    }
+    async onModuleDestroy() {
+        const entries = Array.from(this.clientPool.entries());
+        this.clientPool.clear();
+        this.initFlights.clear();
+        this.stdioCapturedStderr.clear();
+        await Promise.allSettled(entries.map(async ([serverId, entry]) => {
+            this.clearIdleTimeout(entry);
+            try {
+                await entry.client.close();
+            }
+            catch (error) {
+                this.logger.warn(`Unable to close MCP client during shutdown for server "${serverId}"`, error);
+            }
+        }));
+    }
+    async getOrCreateClient(serverId) {
+        return await this.getOrCreateClientInternal(serverId, {
+            allowDisabled: false,
+        });
+    }
+    async listTools(serverId) {
+        const client = await this.getOrCreateClient(serverId);
+        return await client.listTools();
+    }
+    async listToolsForDiagnostics(serverId) {
+        const server = await this.findServerOrFail(serverId);
+        const client = await this.getOrCreateClientInternal(serverId, {
+            allowDisabled: true,
+        });
+        const definitions = await client.listTools();
+        const tools = this.normalizeTools(serverId, definitions);
+        return {
+            server: this.toServerConnectionInfo(server),
+            toolCount: tools.length,
+            tools,
+            ...(this.toObject(definitions._meta)
+                ? { meta: this.toObject(definitions._meta) }
+                : {}),
+        };
+    }
+    async testServer(serverId) {
+        const server = await this.findServerOrFail(serverId);
+        const startedAt = Date.now();
+        this.stdioCapturedStderr.delete(serverId);
+        try {
+            const discovery = await this.listToolsForDiagnostics(serverId);
+            const latencyMs = Date.now() - startedAt;
+            return {
+                ok: true,
+                checkedAt: new Date().toISOString(),
+                latencyMs,
+                server: discovery.server,
+                toolCount: discovery.toolCount,
+                sampledToolNames: discovery.tools
+                    .slice(0, this.sampledToolNamesCount)
+                    .map((tool) => tool.name),
+                ...(discovery.meta ? { meta: discovery.meta } : {}),
+            };
+        }
+        catch (error) {
+            const latencyMs = Date.now() - startedAt;
+            const message = error instanceof Error ? error.message : 'Unknown connection error';
+            const stderrMessage = this.stdioCapturedStderr.get(serverId);
+            const effectiveError = stderrMessage || message;
+            return {
+                ok: false,
+                checkedAt: new Date().toISOString(),
+                latencyMs,
+                server: this.toServerConnectionInfo(server),
+                toolCount: 0,
+                sampledToolNames: [],
+                error: effectiveError,
+            };
+        }
+    }
+    async buildToolSet(bindingDefs) {
+        const toolSet = {};
+        for (const [bindingName, definition] of Object.entries(bindingDefs ?? {})) {
+            const normalizedBindingName = bindingName.trim();
+            if (!normalizedBindingName) {
+                throw new common_1.BadRequestException('Tool binding name cannot be empty');
+            }
+            const serverId = definition?.settings?.server_id?.trim();
+            if (!serverId) {
+                throw new common_1.BadRequestException(`bindings.mcp.${normalizedBindingName}.settings.server_id is required`);
+            }
+            const selectedToolNames = new Set((definition?.settings?.tool_names ?? [])
+                .filter((name) => typeof name === 'string')
+                .map((name) => name.trim())
+                .filter(Boolean));
+            const client = await this.getOrCreateClient(serverId);
+            const definitions = await client.listTools();
+            const filteredTools = selectedToolNames.size > 0
+                ? definitions.tools.filter((tool) => selectedToolNames.has(tool.name))
+                : definitions.tools;
+            if (!filteredTools.length) {
+                continue;
+            }
+            const scopedDefinitions = {
+                ...definitions,
+                tools: filteredTools,
+            };
+            const bindingTools = client.toolsFromDefinitions(scopedDefinitions);
+            for (const [toolName, tool] of Object.entries(bindingTools)) {
+                const prefixedToolName = `${normalizedBindingName}__${toolName}`;
+                if (prefixedToolName in toolSet) {
+                    throw new common_1.BadRequestException(`Duplicate MCP tool name "${prefixedToolName}"`);
+                }
+                toolSet[prefixedToolName] = tool;
+            }
+        }
+        return toolSet;
+    }
+    async getOrCreateClientInternal(serverId, options) {
+        const server = await this.findServerOrFail(serverId);
+        if (!options.allowDisabled && !server.enabled) {
+            throw new common_1.BadRequestException(`MCP server "${server.name}" (${serverId}) is disabled`);
+        }
+        const signature = this.computeServerSignature(server);
+        const existingEntry = this.clientPool.get(serverId);
+        if (existingEntry) {
+            if (existingEntry.signature !== signature) {
+                await this.evictClient(serverId);
+            }
+            else {
+                this.touchClient(serverId, existingEntry);
+                return existingEntry.client;
+            }
+        }
+        const existingInit = this.initFlights.get(serverId);
+        if (existingInit) {
+            return await existingInit;
+        }
+        const initPromise = this.initializeClient(server, signature).finally(() => {
+            this.initFlights.delete(serverId);
+        });
+        this.initFlights.set(serverId, initPromise);
+        return await initPromise;
+    }
+    async initializeClient(server, signature) {
+        const client = await this.createClient(server);
+        const entry = {
+            client,
+            signature,
+            idleTimeout: null,
+        };
+        this.clientPool.set(server.id, entry);
+        this.touchClient(server.id, entry);
+        return client;
+    }
+    async createClient(server) {
+        this.stdioCapturedStderr.delete(server.id);
+        const onUncaughtError = (error) => {
+            this.logger.error(`Unhandled error in MCP client for server "${server.id}"`, error);
+        };
+        if (server.transport === types_1.McpServerTransport.http) {
+            const url = this.trimmedOrNull(server.url);
+            if (!url) {
+                throw new common_1.BadRequestException(`Missing URL for HTTP MCP server "${server.name}"`);
+            }
+            const headers = await this.buildHeaders(server);
+            return await (0, mcp_1.createMCPClient)({
+                transport: {
+                    type: 'http',
+                    url,
+                    ...(headers ? { headers } : {}),
+                },
+                onUncaughtError,
+            });
+        }
+        if (server.transport === types_1.McpServerTransport.stdio) {
+            if (server.credential) {
+                throw new common_1.BadRequestException(`Credential is not supported for stdio MCP server "${server.name}"`);
+            }
+            const command = this.trimmedOrNull(server.command);
+            if (!command) {
+                throw new common_1.BadRequestException(`Missing command for stdio MCP server "${server.name}"`);
+            }
+            const cwd = this.trimmedOrNull(server.cwd);
+            const env = this.buildStdioEnv();
+            return await (0, mcp_1.createMCPClient)({
+                transport: new stdio_stderr_capture_transport_1.StdioStderrCaptureTransport({
+                    command,
+                    ...(server.args ? { args: server.args } : {}),
+                    ...(cwd ? { cwd } : {}),
+                    env,
+                    stderr: 'pipe',
+                }, (message) => {
+                    this.stdioCapturedStderr.set(server.id, message);
+                }),
+                onUncaughtError,
+            });
+        }
+        throw new common_1.BadRequestException(`Unsupported MCP transport "${server.transport}" for server "${server.name}"`);
+    }
+    async findServerOrFail(serverId) {
+        const server = await this.mcpServerRepository.findOne(serverId);
+        if (!server) {
+            throw new common_1.NotFoundException(`MCP server with ID ${serverId} not found`);
+        }
+        return server;
+    }
+    async buildHeaders(server) {
+        const credentialId = server.credential ?? undefined;
+        if (!credentialId) {
+            return undefined;
+        }
+        const value = await this.credentialService.findOneValue(credentialId);
+        if (!value) {
+            throw new common_1.BadRequestException(`Credential "${credentialId}" configured for MCP server "${server.id}" has no value`);
+        }
+        return {
+            Authorization: `Bearer ${value}`,
+        };
+    }
+    computeServerSignature(server) {
+        return JSON.stringify({
+            transport: server.transport,
+            url: server.url,
+            credential: server.credential ?? null,
+            command: server.command,
+            args: server.args ?? null,
+            cwd: server.cwd,
+        });
+    }
+    toServerConnectionInfo(server) {
+        return {
+            id: server.id,
+            name: server.name,
+            enabled: server.enabled,
+            transport: server.transport,
+            url: server.url,
+            ...(server.command ? { command: server.command } : {}),
+            ...(server.args ? { args: server.args } : {}),
+            ...(server.cwd ? { cwd: server.cwd } : {}),
+        };
+    }
+    trimmedOrNull(value) {
+        if (typeof value !== 'string') {
+            return null;
+        }
+        const normalized = value.trim();
+        return normalized ? normalized : null;
+    }
+    buildStdioEnv() {
+        return Object.entries(process.env).reduce((acc, [key, value]) => {
+            if (typeof value === 'string') {
+                acc[key] = value;
+            }
+            return acc;
+        }, {});
+    }
+    touchClient(serverId, entry) {
+        this.clearIdleTimeout(entry);
+        entry.idleTimeout = setTimeout(() => {
+            void this.evictClient(serverId);
+        }, McpClientPoolService_1.DEFAULT_IDLE_TTL_MS);
+        entry.idleTimeout.unref?.();
+    }
+    clearIdleTimeout(entry) {
+        if (entry.idleTimeout) {
+            clearTimeout(entry.idleTimeout);
+            entry.idleTimeout = null;
+        }
+    }
+    async evictClient(serverId) {
+        const entry = this.clientPool.get(serverId);
+        if (!entry) {
+            return;
+        }
+        this.clientPool.delete(serverId);
+        this.stdioCapturedStderr.delete(serverId);
+        this.clearIdleTimeout(entry);
+        try {
+            await entry.client.close();
+        }
+        catch (error) {
+            this.logger.warn(`Unable to close MCP client for server "${serverId}"`, error);
+        }
+    }
+    normalizeTools(serverId, definitions) {
+        return definitions.tools.map((tool) => {
+            const meta = this.toObject(tool._meta);
+            const annotations = this.toObject(tool.annotations);
+            const outputSchema = this.toObject(tool.outputSchema);
+            return {
+                id: `${serverId}:${tool.name}`,
+                serverId,
+                name: tool.name,
+                ...(tool.title ? { title: tool.title } : {}),
+                ...(tool.description ? { description: tool.description } : {}),
+                inputSchema: tool.inputSchema,
+                ...(outputSchema ? { outputSchema } : {}),
+                ...(annotations ? { annotations } : {}),
+                ...(meta ? { meta } : {}),
+            };
+        });
+    }
+    toObject(value) {
+        if (!value || typeof value !== 'object' || Array.isArray(value)) {
+            return undefined;
+        }
+        return value;
+    }
+};
+exports.McpClientPoolService = McpClientPoolService;
+McpClientPoolService.DEFAULT_IDLE_TTL_MS = 5 * 60 * 1000;
+exports.McpClientPoolService = McpClientPoolService = McpClientPoolService_1 = __decorate([
+    (0, common_1.Injectable)(),
+    __metadata("design:paramtypes", [mcp_server_repository_1.McpServerRepository,
+        user_1.CredentialService,
+        logger_service_1.LoggerService])
+], McpClientPoolService);
+//# sourceMappingURL=mcp-client-pool.service.js.map

package/dist/workflow/services/mcp-client-pool.service.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"mcp-client-pool.service.js","sourceRoot":"","sources":["../../../src/workflow/services/mcp-client-pool.service.ts"],"names":[],"mappings":";;;;;;;;;;;;;AAMA,qCAIqB;AACrB,2CAKwB;AAGxB,gEAAwD;AACxD,qCAA2C;AAG3C,iFAA4E;AAC5E,oCAAyE;AAEzE,qFAA+E;AAsDxE,IAAM,oBAAoB,4BAA1B,MAAM,oBAAoB;IAmB/B,YACmB,mBAAwC,EACxC,iBAAoC,EACpC,MAAqB;QAFrB,wBAAmB,GAAnB,mBAAmB,CAAqB;QACxC,sBAAiB,GAAjB,iBAAiB,CAAmB;QACpC,WAAM,GAAN,MAAM,CAAe;QAnBvB,0BAAqB,GAAG,EAAE,CAAC;QAE3B,eAAU,GAAG,IAAI,GAAG,EAA6B,CAAC;QAElD,gBAAW,GAAG,IAAI,GAAG,EAA8B,CAAC;QAEpD,wBAAmB,GAAG,IAAI,GAAG,EAAkB,CAAC;IAc9D,CAAC;IAOJ,KAAK,CAAC,eAAe;QACnB,MAAM,OAAO,GAAG,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,OAAO,EAAE,CAAC,CAAC;QACtD,IAAI,CAAC,UAAU,CAAC,KAAK,EAAE,CAAC;QACxB,IAAI,CAAC,WAAW,CAAC,KAAK,EAAE,CAAC;QACzB,IAAI,CAAC,mBAAmB,CAAC,KAAK,EAAE,CAAC;QAEjC,MAAM,OAAO,CAAC,UAAU,CACtB,OAAO,CAAC,GAAG,CAAC,KAAK,EAAE,CAAC,QAAQ,EAAE,KAAK,CAAC,EAAE,EAAE;YACtC,IAAI,CAAC,gBAAgB,CAAC,KAAK,CAAC,CAAC;YAC7B,IAAI,CAAC;gBACH,MAAM,KAAK,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC;YAC7B,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,IAAI,CAAC,MAAM,CAAC,IAAI,CACd,0DAA0D,QAAQ,GAAG,EACrE,KAAK,CACN,CAAC;YACJ,CAAC;QACH,CAAC,CAAC,CACH,CAAC;IACJ,CAAC;IAQD,KAAK,CAAC,iBAAiB,CAAC,QAAgB;QACtC,OAAO,MAAM,IAAI,CAAC,yBAAyB,CAAC,QAAQ,EAAE;YACpD,aAAa,EAAE,KAAK;SACrB,CAAC,CAAC;IACL,CAAC;IAQD,KAAK,CAAC,SAAS,CAAC,QAAgB;QAC9B,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,iBAAiB,CAAC,QAAQ,CAAC,CAAC;QAEtD,OAAO,MAAM,MAAM,CAAC,SAAS,EAAE,CAAC;IAClC,CAAC;IAQD,KAAK,CAAC,uBAAuB,CAAC,QAAgB;QAC5C,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,gBAAgB,CAAC,QAAQ,CAAC,CAAC;QACrD,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,yBAAyB,CAAC,QAAQ,EAAE;YAC5D,aAAa,EAAE,IAAI;SACpB,CAAC,CAAC;QACH,MAAM,WAAW,GAAG,MAAM,MAAM,CAAC,SAAS,EAAE,CAAC;QAC7C,MAAM,KAAK,GAAG,IAAI,CAAC,cAAc,CAAC,QAAQ,EAAE,WAAW,CAAC,CAAC;QAEzD,OAAO;YACL,MAAM,EAAE,IAAI,CAAC,sBAAsB,CAAC,MAAM,CAAC;YAC3C,SAAS,EAAE,KAAK,CAAC,MAAM;YACvB,KAAK;YACL,GAAG,CAAC,IAAI,CAAC,QAAQ,CAAC,WAAW,CAAC,KAAK,CAAC;gBAClC,CAAC,CAAC,EAAE,IAAI,EAAE,IAAI,CAAC,QAAQ,CAAC,WAAW,CAAC,KAAK,CAAC,EAAE;gBAC5C,CAAC,CAAC,EAAE,CAAC;SACR,CAAC;IACJ,CAAC;IAQD,KAAK,CAAC,UAAU,CAAC,QAAgB;QAC/B,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,gBAAgB,CAAC,QAAQ,CAAC,CAAC;QACrD,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAC7B,IAAI,CAAC,mBAAmB,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;QAE1C,IAAI,CAAC;YACH,MAAM,SAAS,GAAG,MAAM,IAAI,CAAC,uBAAuB,CAAC,QAAQ,CAAC,CAAC;YAC/D,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC;YAEzC,OAAO;gBACL,EAAE,EAAE,IAAI;gBACR,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;gBACnC,SAAS;gBACT,MAAM,EAAE,SAAS,CAAC,MAAM;gBACxB,SAAS,EAAE,SAAS,CAAC,SAAS;gBAC9B,gBAAgB,EAAE,SAAS,CAAC,KAAK;qBAC9B,KAAK,CAAC,CAAC,EAAE,IAAI,CAAC,qBAAqB,CAAC;qBACpC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC;gBAC3B,GAAG,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,SAAS,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;aACpD,CAAC;QACJ,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC;YACzC,MAAM,OAAO,GACX,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,0BAA0B,CAAC;YACtE,MAAM,aAAa,GAAG,IAAI,CAAC,mBAAmB,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;YAC7D,MAAM,cAAc,GAAG,aAAa,IAAI,OAAO,CAAC;YAEhD,OAAO;gBACL,EAAE,EAAE,KAAK;gBACT,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;gBACnC,SAAS;gBACT,MAAM,EAAE,IAAI,CAAC,sBAAsB,CAAC,MAAM,CAAC;gBAC3C,SAAS,EAAE,CAAC;gBACZ,gBAAgB,EAAE,EAAE;gBACpB,KAAK,EAAE,cAAc;aACtB,CAAC;QACJ,CAAC;IACH,CAAC;IAQD,KAAK,CAAC,YAAY,CAAC,WAAsC;QACvD,MAAM,OAAO,GAAY,EAAE,CAAC;QAE5B,KAAK,MAAM,CAAC,WAAW,EAAE,UAAU,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,WAAW,IAAI,EAAE,CAAC,EAAE,CAAC;YAC1E,MAAM,qBAAqB,GAAG,WAAW,CAAC,IAAI,EAAE,CAAC;YACjD,IAAI,CAAC,qBAAqB,EAAE,CAAC;gBAC3B,MAAM,IAAI,4BAAmB,CAAC,mCAAmC,CAAC,CAAC;YACrE,CAAC;YAED,MAAM,QAAQ,GAAG,UAAU,EAAE,QAAQ,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC;YACzD,IAAI,CAAC,QAAQ,EAAE,CAAC;gBACd,MAAM,IAAI,4BAAmB,CAC3B,gBAAgB,qBAAqB,iCAAiC,CACvE,CAAC;YACJ,CAAC;YAED,MAAM,iBAAiB,GAAG,IAAI,GAAG,CAC/B,CAAC,UAAU,EAAE,QAAQ,EAAE,UAAU,IAAI,EAAE,CAAC;iBACrC,MAAM,CAAC,CAAC,IAAI,EAAkB,EAAE,CAAC,OAAO,IAAI,KAAK,QAAQ,CAAC;iBAC1D,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC;iBAC1B,MAAM,CAAC,OAAO,CAAC,CACnB,CAAC;YACF,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,iBAAiB,CAAC,QAAQ,CAAC,CAAC;YACtD,MAAM,WAAW,GAAG,MAAM,MAAM,CAAC,SAAS,EAAE,CAAC;YAC7C,MAAM,aAAa,GACjB,iBAAiB,CAAC,IAAI,GAAG,CAAC;gBACxB,CAAC,CAAC,WAAW,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,iBAAiB,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;gBACtE,CAAC,CAAC,WAAW,CAAC,KAAK,CAAC;YAExB,IAAI,CAAC,aAAa,CAAC,MAAM,EAAE,CAAC;gBAC1B,SAAS;YACX,CAAC;YAED,MAAM,iBAAiB,GAAoB;gBACzC,GAAG,WAAW;gBACd,KAAK,EAAE,aAAa;aACrB,CAAC;YACF,MAAM,YAAY,GAAG,MAAM,CAAC,oBAAoB,CAC9C,iBAAiB,CACP,CAAC;YAEb,KAAK,MAAM,CAAC,QAAQ,EAAE,IAAI,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,EAAE,CAAC;gBAC5D,MAAM,gBAAgB,GAAG,GAAG,qBAAqB,KAAK,QAAQ,EAAE,CAAC;gBACjE,IAAI,gBAAgB,IAAI,OAAO,EAAE,CAAC;oBAChC,MAAM,IAAI,4BAAmB,CAC3B,4BAA4B,gBAAgB,GAAG,CAChD,CAAC;gBACJ,CAAC;gBAED,OAAO,CAAC,gBAAgB,CAAC,GAAG,IAAuB,CAAC;YACtD,CAAC;QACH,CAAC;QAED,OAAO,OAAO,CAAC;IACjB,CAAC;IASO,KAAK,CAAC,yBAAyB,CACrC,QAAgB,EAChB,OAAyB;QAEzB,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,gBAAgB,CAAC,QAAQ,CAAC,CAAC;QACrD,IAAI,CAAC,OAAO,CAAC,aAAa,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;YAC9C,MAAM,IAAI,4BAAmB,CAC3B,eAAe,MAAM,CAAC,IAAI,MAAM,QAAQ,eAAe,CACxD,CAAC;QACJ,CAAC;QAED,MAAM,SAAS,GAAG,IAAI,CAAC,sBAAsB,CAAC,MAAM,CAAC,CAAC;QACtD,MAAM,aAAa,GAAG,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QACpD,IAAI,aAAa,EAAE,CAAC;YAClB,IAAI,aAAa,CAAC,SAAS,KAAK,SAAS,EAAE,CAAC;gBAC1C,MAAM,IAAI,CAAC,WAAW,CAAC,QAAQ,CAAC,CAAC;YACnC,CAAC;iBAAM,CAAC;gBACN,IAAI,CAAC,WAAW,CAAC,QAAQ,EAAE,aAAa,CAAC,CAAC;gBAE1C,OAAO,aAAa,CAAC,MAAM,CAAC;YAC9B,CAAC;QACH,CAAC;QAED,MAAM,YAAY,GAAG,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QACpD,IAAI,YAAY,EAAE,CAAC;YACjB,OAAO,MAAM,YAAY,CAAC;QAC5B,CAAC;QAED,MAAM,WAAW,GAAG,IAAI,CAAC,gBAAgB,CAAC,MAAM,EAAE,SAAS,CAAC,CAAC,OAAO,CAAC,GAAG,EAAE;YACxE,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;QACpC,CAAC,CAAC,CAAC;QACH,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,QAAQ,EAAE,WAAW,CAAC,CAAC;QAE5C,OAAO,MAAM,WAAW,CAAC;IAC3B,CAAC;IASO,KAAK,CAAC,gBAAgB,CAC5B,MAAiB,EACjB,SAAiB;QAEjB,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC;QAC/C,MAAM,KAAK,GAAsB;YAC/B,MAAM;YACN,SAAS;YACT,WAAW,EAAE,IAAI;SAClB,CAAC;QACF,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,EAAE,KAAK,CAAC,CAAC;QACtC,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,EAAE,EAAE,KAAK,CAAC,CAAC;QAEnC,OAAO,MAAM,CAAC;IAChB,CAAC;IAQO,KAAK,CAAC,YAAY,CAAC,MAAiB;QAC1C,IAAI,CAAC,mBAAmB,CAAC,MAAM,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;QAC3C,MAAM,eAAe,GAAG,CAAC,KAAc,EAAE,EAAE;YACzC,IAAI,CAAC,MAAM,CAAC,KAAK,CACf,6CAA6C,MAAM,CAAC,EAAE,GAAG,EACzD,KAAK,CACN,CAAC;QACJ,CAAC,CAAC;QAEF,IAAI,MAAM,CAAC,SAAS,KAAK,0BAAkB,CAAC,IAAI,EAAE,CAAC;YACjD,MAAM,GAAG,GAAG,IAAI,CAAC,aAAa,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;YAC3C,IAAI,CAAC,GAAG,EAAE,CAAC;gBACT,MAAM,IAAI,4BAAmB,CAC3B,oCAAoC,MAAM,CAAC,IAAI,GAAG,CACnD,CAAC;YACJ,CAAC;YAED,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC;YAEhD,OAAO,MAAM,IAAA,qBAAe,EAAC;gBAC3B,SAAS,EAAE;oBACT,IAAI,EAAE,MAAM;oBACZ,GAAG;oBACH,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;iBAChC;gBACD,eAAe;aAChB,CAAC,CAAC;QACL,CAAC;QAED,IAAI,MAAM,CAAC,SAAS,KAAK,0BAAkB,CAAC,KAAK,EAAE,CAAC;YAClD,IAAI,MAAM,CAAC,UAAU,EAAE,CAAC;gBACtB,MAAM,IAAI,4BAAmB,CAC3B,qDAAqD,MAAM,CAAC,IAAI,GAAG,CACpE,CAAC;YACJ,CAAC;YAED,MAAM,OAAO,GAAG,IAAI,CAAC,aAAa,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;YACnD,IAAI,CAAC,OAAO,EAAE,CAAC;gBACb,MAAM,IAAI,4BAAmB,CAC3B,yCAAyC,MAAM,CAAC,IAAI,GAAG,CACxD,CAAC;YACJ,CAAC;YAED,MAAM,GAAG,GAAG,IAAI,CAAC,aAAa,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;YAC3C,MAAM,GAAG,GAAG,IAAI,CAAC,aAAa,EAAE,CAAC;YAEjC,OAAO,MAAM,IAAA,qBAAe,EAAC;gBAC3B,SAAS,EAAE,IAAI,4DAA2B,CACxC;oBACE,OAAO;oBACP,GAAG,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,MAAM,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;oBAC7C,GAAG,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,GAAG,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;oBACvB,GAAG;oBACH,MAAM,EAAE,MAAM;iBACf,EACD,CAAC,OAAO,EAAE,EAAE;oBACV,IAAI,CAAC,mBAAmB,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,EAAE,OAAO,CAAC,CAAC;gBACnD,CAAC,CACF;gBACD,eAAe;aAChB,CAAC,CAAC;QACL,CAAC;QAED,MAAM,IAAI,4BAAmB,CAC3B,8BAA8B,MAAM,CAAC,SAAS,iBAAiB,MAAM,CAAC,IAAI,GAAG,CAC9E,CAAC;IACJ,CAAC;IAQO,KAAK,CAAC,gBAAgB,CAAC,QAAgB;QAC7C,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,mBAAmB,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;QAChE,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,MAAM,IAAI,0BAAiB,CAAC,sBAAsB,QAAQ,YAAY,CAAC,CAAC;QAC1E,CAAC;QAED,OAAO,MAAM,CAAC;IAChB,CAAC;IAQO,KAAK,CAAC,YAAY,CACxB,MAAiB;QAEjB,MAAM,YAAY,GAAG,MAAM,CAAC,UAAU,IAAI,SAAS,CAAC;QACpD,IAAI,CAAC,YAAY,EAAE,CAAC;YAClB,OAAO,SAAS,CAAC;QACnB,CAAC;QAED,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,iBAAiB,CAAC,YAAY,CAAC,YAAY,CAAC,CAAC;QACtE,IAAI,CAAC,KAAK,EAAE,CAAC;YACX,MAAM,IAAI,4BAAmB,CAC3B,eAAe,YAAY,gCAAgC,MAAM,CAAC,EAAE,gBAAgB,CACrF,CAAC;QACJ,CAAC;QAED,OAAO;YACL,aAAa,EAAE,UAAU,KAAK,EAAE;SACjC,CAAC;IACJ,CAAC;IAQO,sBAAsB,CAAC,MAAiB;QAC9C,OAAO,IAAI,CAAC,SAAS,CAAC;YACpB,SAAS,EAAE,MAAM,CAAC,SAAS;YAC3B,GAAG,EAAE,MAAM,CAAC,GAAG;YACf,UAAU,EAAE,MAAM,CAAC,UAAU,IAAI,IAAI;YACrC,OAAO,EAAE,MAAM,CAAC,OAAO;YACvB,IAAI,EAAE,MAAM,CAAC,IAAI,IAAI,IAAI;YACzB,GAAG,EAAE,MAAM,CAAC,GAAG;SAChB,CAAC,CAAC;IACL,CAAC;IAQO,sBAAsB,CAAC,MAAiB;QAC9C,OAAO;YACL,EAAE,EAAE,MAAM,CAAC,EAAE;YACb,IAAI,EAAE,MAAM,CAAC,IAAI;YACjB,OAAO,EAAE,MAAM,CAAC,OAAO;YACvB,SAAS,EAAE,MAAM,CAAC,SAAS;YAC3B,GAAG,EAAE,MAAM,CAAC,GAAG;YACf,GAAG,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,MAAM,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YACtD,GAAG,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,MAAM,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YAC7C,GAAG,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,GAAG,EAAE,MAAM,CAAC,GAAG,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;SAC3C,CAAC;IACJ,CAAC;IAQO,aAAa,CAAC,KAAgC;QACpD,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;YAC9B,OAAO,IAAI,CAAC;QACd,CAAC;QAED,MAAM,UAAU,GAAG,KAAK,CAAC,IAAI,EAAE,CAAC;QAEhC,OAAO,UAAU,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,IAAI,CAAC;IACxC,CAAC;IAOO,aAAa;QACnB,OAAO,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,MAAM,CACvC,CAAC,GAAG,EAAE,CAAC,GAAG,EAAE,KAAK,CAAC,EAAE,EAAE;YACpB,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;gBAC9B,GAAG,CAAC,GAAG,CAAC,GAAG,KAAK,CAAC;YACnB,CAAC;YAED,OAAO,GAAG,CAAC;QACb,CAAC,EACD,EAAE,CACH,CAAC;IACJ,CAAC;IASO,WAAW,CAAC,QAAgB,EAAE,KAAwB;QAC5D,IAAI,CAAC,gBAAgB,CAAC,KAAK,CAAC,CAAC;QAC7B,KAAK,CAAC,WAAW,GAAG,UAAU,CAAC,GAAG,EAAE;YAClC,KAAK,IAAI,CAAC,WAAW,CAAC,QAAQ,CAAC,CAAC;QAClC,CAAC,EAAE,sBAAoB,CAAC,mBAAmB,CAAC,CAAC;QAC7C,KAAK,CAAC,WAAW,CAAC,KAAK,EAAE,EAAE,CAAC;IAC9B,CAAC;IAQO,gBAAgB,CAAC,KAAwB;QAC/C,IAAI,KAAK,CAAC,WAAW,EAAE,CAAC;YACtB,YAAY,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC;YAChC,KAAK,CAAC,WAAW,GAAG,IAAI,CAAC;QAC3B,CAAC;IACH,CAAC;IAQO,KAAK,CAAC,WAAW,CAAC,QAAgB;QACxC,MAAM,KAAK,GAAG,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QAC5C,IAAI,CAAC,KAAK,EAAE,CAAC;YACX,OAAO;QACT,CAAC;QAED,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;QACjC,IAAI,CAAC,mBAAmB,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;QAC1C,IAAI,CAAC,gBAAgB,CAAC,KAAK,CAAC,CAAC;QAC7B,IAAI,CAAC;YACH,MAAM,KAAK,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC;QAC7B,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,CAAC,MAAM,CAAC,IAAI,CACd,0CAA0C,QAAQ,GAAG,EACrD,KAAK,CACN,CAAC;QACJ,CAAC;IACH,CAAC;IAQO,cAAc,CACpB,QAAgB,EAChB,WAA4B;QAE5B,OAAO,WAAW,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE;YACpC,MAAM,IAAI,GAAG,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YACvC,MAAM,WAAW,GAAG,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;YACpD,MAAM,YAAY,GAAG,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;YAEtD,OAAO;gBACL,EAAE,EAAE,GAAG,QAAQ,IAAI,IAAI,CAAC,IAAI,EAAE;gBAC9B,QAAQ;gBACR,IAAI,EAAE,IAAI,CAAC,IAAI;gBACf,GAAG,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,IAAI,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;gBAC5C,GAAG,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,CAAC,EAAE,WAAW,EAAE,IAAI,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;gBAC9D,WAAW,EAAE,IAAI,CAAC,WAAsC;gBACxD,GAAG,CAAC,YAAY,CAAC,CAAC,CAAC,EAAE,YAAY,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;gBACzC,GAAG,CAAC,WAAW,CAAC,CAAC,CAAC,EAAE,WAAW,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;gBACvC,GAAG,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;aAC1B,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC;IAQO,QAAQ,CAAC,KAAc;QAC7B,IAAI,CAAC,KAAK,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;YAChE,OAAO,SAAS,CAAC;QACnB,CAAC;QAED,OAAO,KAAgC,CAAC;IAC1C,CAAC;;AAviBU,oDAAoB;AACf,wCAAmB,GAAG,CAAC,GAAG,EAAE,GAAG,IAAI,AAAhB,CAAiB;+BADzC,oBAAoB;IADhC,IAAA,mBAAU,GAAE;qCAqB6B,2CAAmB;QACrB,wBAAiB;QAC5B,8BAAa;GAtB7B,oBAAoB,CAwiBhC"}

package/dist/workflow/services/mcp-server.service.d.ts

@@ -0,0 +1,11 @@
+import { BaseOrmService } from '@/utils/generics/base-orm.service';
+import { McpServerOrmEntity } from '../entities/mcp-server.entity';
+import { McpServerRepository } from '../repositories/mcp-server.repository';
+import { McpClientPoolService, McpServerDiagnostics, McpToolSummary } from './mcp-client-pool.service';
+export declare class McpServerService extends BaseOrmService<McpServerOrmEntity> {
+    readonly repository: McpServerRepository;
+    private readonly mcpClientPoolService;
+    constructor(repository: McpServerRepository, mcpClientPoolService: McpClientPoolService);
+    testConnection(id: string): Promise<McpServerDiagnostics>;
+    discoverTools(id: string): Promise<McpToolSummary[]>;
+}

package/dist/workflow/services/mcp-server.service.js

@@ -0,0 +1,37 @@
+"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.McpServerService = void 0;
+const common_1 = require("@nestjs/common");
+const base_orm_service_1 = require("../../utils/generics/base-orm.service");
+const mcp_server_repository_1 = require("../repositories/mcp-server.repository");
+const mcp_client_pool_service_1 = require("./mcp-client-pool.service");
+let McpServerService = class McpServerService extends base_orm_service_1.BaseOrmService {
+    constructor(repository, mcpClientPoolService) {
+        super(repository);
+        this.repository = repository;
+        this.mcpClientPoolService = mcpClientPoolService;
+    }
+    async testConnection(id) {
+        return await this.mcpClientPoolService.testServer(id);
+    }
+    async discoverTools(id) {
+        const discovery = await this.mcpClientPoolService.listToolsForDiagnostics(id);
+        return discovery.tools;
+    }
+};
+exports.McpServerService = McpServerService;
+exports.McpServerService = McpServerService = __decorate([
+    (0, common_1.Injectable)(),
+    __metadata("design:paramtypes", [mcp_server_repository_1.McpServerRepository,
+        mcp_client_pool_service_1.McpClientPoolService])
+], McpServerService);
+//# sourceMappingURL=mcp-server.service.js.map

package/dist/workflow/services/mcp-server.service.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"mcp-server.service.js","sourceRoot":"","sources":["../../../src/workflow/services/mcp-server.service.ts"],"names":[],"mappings":";;;;;;;;;;;;AAMA,2CAA4C;AAE5C,4EAAmE;AAGnE,iFAA4E;AAE5E,uEAImC;AAG5B,IAAM,gBAAgB,GAAtB,MAAM,gBAAiB,SAAQ,iCAAkC;IAQtE,YACW,UAA+B,EACvB,oBAA0C;QAE3D,KAAK,CAAC,UAAU,CAAC,CAAC;QAHT,eAAU,GAAV,UAAU,CAAqB;QACvB,yBAAoB,GAApB,oBAAoB,CAAsB;IAG7D,CAAC;IAQD,KAAK,CAAC,cAAc,CAAC,EAAU;QAC7B,OAAO,MAAM,IAAI,CAAC,oBAAoB,CAAC,UAAU,CAAC,EAAE,CAAC,CAAC;IACxD,CAAC;IAQD,KAAK,CAAC,aAAa,CAAC,EAAU;QAC5B,MAAM,SAAS,GACb,MAAM,IAAI,CAAC,oBAAoB,CAAC,uBAAuB,CAAC,EAAE,CAAC,CAAC;QAE9D,OAAO,SAAS,CAAC,KAAK,CAAC;IACzB,CAAC;CACF,CAAA;AArCY,4CAAgB;2BAAhB,gBAAgB;IAD5B,IAAA,mBAAU,GAAE;qCAUY,2CAAmB;QACD,8CAAoB;GAVlD,gBAAgB,CAqC5B"}

package/dist/workflow/services/memory-definition.service.d.ts

@@ -0,0 +1,10 @@
+import { BaseOrmService } from '@/utils/generics/base-orm.service';
+import { MemoryDefinition } from '../dto/memory-definition.dto';
+import { MemoryDefinitionOrmEntity } from '../entities/memory-definition.entity';
+import { MemoryDefinitionRepository } from '../repositories/memory-definition.repository';
+export declare class MemoryDefinitionService extends BaseOrmService<MemoryDefinitionOrmEntity> {
+    readonly repository: MemoryDefinitionRepository;
+    constructor(repository: MemoryDefinitionRepository);
+    findBySlug(slug: string): Promise<MemoryDefinition | null>;
+    buildDefinitionCache(memoryDefinitionIds?: string[] | null): Promise<Map<string, MemoryDefinition>>;
+}

package/dist/workflow/services/memory-definition.service.js

@@ -0,0 +1,56 @@
+"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.MemoryDefinitionService = void 0;
+const common_1 = require("@nestjs/common");
+const typeorm_1 = require("typeorm");
+const base_orm_service_1 = require("../../utils/generics/base-orm.service");
+const memory_definition_repository_1 = require("../repositories/memory-definition.repository");
+const types_1 = require("../types");
+let MemoryDefinitionService = class MemoryDefinitionService extends base_orm_service_1.BaseOrmService {
+    constructor(repository) {
+        super(repository);
+        this.repository = repository;
+    }
+    async findBySlug(slug) {
+        return await this.findOne({ where: { slug } });
+    }
+    async buildDefinitionCache(memoryDefinitionIds) {
+        const where = [
+            { scope: types_1.MemoryScope.global },
+        ];
+        const requestedIds = Array.from(new Set(memoryDefinitionIds ?? [])).filter(Boolean);
+        if (requestedIds.length > 0) {
+            where.push({ id: (0, typeorm_1.In)(requestedIds) });
+        }
+        const definitions = await this.find({ where });
+        if (requestedIds.length > 0) {
+            const resolvedIds = new Set(definitions
+                .map((definition) => definition.id)
+                .filter((id) => Boolean(id)));
+            const missingIds = requestedIds.filter((id) => !resolvedIds.has(id));
+            if (missingIds.length > 0) {
+                throw new Error(`Unable to find memory definition(s): ${missingIds.join(', ')}`);
+            }
+        }
+        const definitionCache = new Map();
+        for (const definition of definitions) {
+            definitionCache.set(definition.slug, definition);
+        }
+        return definitionCache;
+    }
+};
+exports.MemoryDefinitionService = MemoryDefinitionService;
+exports.MemoryDefinitionService = MemoryDefinitionService = __decorate([
+    (0, common_1.Injectable)(),
+    __metadata("design:paramtypes", [memory_definition_repository_1.MemoryDefinitionRepository])
+], MemoryDefinitionService);
+//# sourceMappingURL=memory-definition.service.js.map

package/dist/workflow/services/memory-definition.service.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"memory-definition.service.js","sourceRoot":"","sources":["../../../src/workflow/services/memory-definition.service.ts"],"names":[],"mappings":";;;;;;;;;;;;AAMA,2CAA4C;AAC5C,qCAA+C;AAE/C,4EAAmE;AAInE,+FAA0F;AAC1F,oCAAuC;AAGhC,IAAM,uBAAuB,GAA7B,MAAM,uBAAwB,SAAQ,iCAAyC;IACpF,YAAqB,UAAsC;QACzD,KAAK,CAAC,UAAU,CAAC,CAAC;QADC,eAAU,GAAV,UAAU,CAA4B;IAE3D,CAAC;IAQD,KAAK,CAAC,UAAU,CAAC,IAAY;QAC3B,OAAO,MAAM,IAAI,CAAC,OAAO,CAAC,EAAE,KAAK,EAAE,EAAE,IAAI,EAAE,EAAE,CAAC,CAAC;IACjD,CAAC;IAWD,KAAK,CAAC,oBAAoB,CACxB,mBAAqC;QAErC,MAAM,KAAK,GAAkD;YAC3D,EAAE,KAAK,EAAE,mBAAW,CAAC,MAAM,EAAE;SAC9B,CAAC;QACF,MAAM,YAAY,GAAG,KAAK,CAAC,IAAI,CAAC,IAAI,GAAG,CAAC,mBAAmB,IAAI,EAAE,CAAC,CAAC,CAAC,MAAM,CACxE,OAAO,CACR,CAAC;QAEF,IAAI,YAAY,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC5B,KAAK,CAAC,IAAI,CAAC,EAAE,EAAE,EAAE,IAAA,YAAE,EAAC,YAAY,CAAC,EAAE,CAAC,CAAC;QACvC,CAAC;QAED,MAAM,WAAW,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,CAAC,CAAC;QAC/C,IAAI,YAAY,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC5B,MAAM,WAAW,GAAG,IAAI,GAAG,CACzB,WAAW;iBACR,GAAG,CAAC,CAAC,UAAU,EAAE,EAAE,CAAC,UAAU,CAAC,EAAE,CAAC;iBAClC,MAAM,CAAC,CAAC,EAAE,EAAgB,EAAE,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC,CAC7C,CAAC;YACF,MAAM,UAAU,GAAG,YAAY,CAAC,MAAM,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC,WAAW,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,CAAC;YACrE,IAAI,UAAU,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBAC1B,MAAM,IAAI,KAAK,CACb,wCAAwC,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAChE,CAAC;YACJ,CAAC;QACH,CAAC;QAED,MAAM,eAAe,GAAG,IAAI,GAAG,EAA4B,CAAC;QAC5D,KAAK,MAAM,UAAU,IAAI,WAAW,EAAE,CAAC;YACrC,eAAe,CAAC,GAAG,CAAC,UAAU,CAAC,IAAI,EAAE,UAAU,CAAC,CAAC;QACnD,CAAC;QAED,OAAO,eAAe,CAAC;IACzB,CAAC;CACF,CAAA;AA5DY,0DAAuB;kCAAvB,uBAAuB;IADnC,IAAA,mBAAU,GAAE;qCAEsB,yDAA0B;GADhD,uBAAuB,CA4DnC"}

package/dist/workflow/services/memory-record.service.d.ts

@@ -0,0 +1,19 @@
+import { BaseOrmService } from '@/utils/generics/base-orm.service';
+import type { MemoryDefinition } from '../dto/memory-definition.dto';
+import { MemoryRecordFull } from '../dto/memory-record.dto';
+import { MemoryRecordOrmEntity } from '../entities/memory-record.entity';
+import { MemoryRecordRepository } from '../repositories/memory-record.repository';
+import { MemoryStoreIdentifier, MemoryValue } from '../types';
+export declare class MemoryRecordService extends BaseOrmService<MemoryRecordOrmEntity> {
+    readonly repository: MemoryRecordRepository;
+    constructor(repository: MemoryRecordRepository);
+    findActiveByScope({ ownerId, workflowId, threadId, runId, }: MemoryStoreIdentifier): Promise<MemoryRecordFull[]>;
+    upsertScopedRecord({ definition, ownerId, workflowId, threadId, runId, value, }: {
+        definition: Pick<MemoryDefinition, 'id' | 'scope' | 'ttlSeconds'>;
+        ownerId: string;
+        workflowId?: string | null;
+        threadId?: string | null;
+        runId?: string | null;
+        value: MemoryValue;
+    }): Promise<void>;
+}

package/dist/workflow/services/memory-record.service.js

@@ -0,0 +1,126 @@
+"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.MemoryRecordService = void 0;
+const common_1 = require("@nestjs/common");
+const typeorm_1 = require("typeorm");
+const base_orm_service_1 = require("../../utils/generics/base-orm.service");
+const memory_record_repository_1 = require("../repositories/memory-record.repository");
+const types_1 = require("../types");
+let MemoryRecordService = class MemoryRecordService extends base_orm_service_1.BaseOrmService {
+    constructor(repository) {
+        super(repository);
+        this.repository = repository;
+    }
+    async findActiveByScope({ ownerId, workflowId, threadId, runId, }) {
+        if (!ownerId) {
+            throw new Error('An owner id is required to build memory store.');
+        }
+        const where = [
+            {
+                owner: { id: ownerId },
+                definition: { scope: types_1.MemoryScope.global },
+            },
+        ];
+        if (workflowId) {
+            where.push({
+                owner: { id: ownerId },
+                workflow: { id: workflowId },
+                definition: { scope: types_1.MemoryScope.workflow },
+            });
+        }
+        if (threadId) {
+            where.push({
+                owner: { id: ownerId },
+                thread: { id: threadId },
+                definition: { scope: types_1.MemoryScope.thread },
+            });
+        }
+        if (runId) {
+            where.push({
+                owner: { id: ownerId },
+                run: { id: runId },
+                definition: { scope: types_1.MemoryScope.run },
+            });
+        }
+        const now = new Date();
+        const scopedWhere = where.flatMap((scope) => [
+            { ...scope, expiresAt: (0, typeorm_1.IsNull)() },
+            { ...scope, expiresAt: (0, typeorm_1.MoreThan)(now) },
+        ]);
+        return this.findAndPopulate({
+            where: scopedWhere,
+            order: {
+                updatedAt: 'DESC',
+                createdAt: 'DESC',
+            },
+        });
+    }
+    async upsertScopedRecord({ definition, ownerId, workflowId, threadId, runId, value, }) {
+        const ttlSeconds = definition.ttlSeconds ?? null;
+        const expiresAt = ttlSeconds && ttlSeconds > 0
+            ? new Date(Date.now() + ttlSeconds * 1000)
+            : null;
+        const where = {
+            definition: { id: definition.id },
+            owner: { id: ownerId },
+        };
+        if (definition.scope === types_1.MemoryScope.workflow) {
+            where.workflow = { id: workflowId };
+        }
+        if (definition.scope === types_1.MemoryScope.thread) {
+            where.thread = { id: threadId };
+        }
+        if (definition.scope === types_1.MemoryScope.run) {
+            where.run = { id: runId };
+        }
+        const existing = await this.findOne({
+            where,
+            order: {
+                updatedAt: 'DESC',
+                createdAt: 'DESC',
+            },
+        });
+        if (existing) {
+            await this.updateOne(existing.id, {
+                value,
+                ttlSeconds,
+                expiresAt,
+            });
+            return;
+        }
+        await this.create({
+            definition: definition.id,
+            owner: ownerId,
+            workflow: definition.scope === types_1.MemoryScope.global
+                ? null
+                : definition.scope === types_1.MemoryScope.workflow ||
+                    definition.scope === types_1.MemoryScope.run
+                    ? (workflowId ?? null)
+                    : null,
+            thread: definition.scope === types_1.MemoryScope.thread
+                ? (threadId ?? null)
+                : definition.scope === types_1.MemoryScope.run
+                    ? (threadId ?? null)
+                    : null,
+            run: definition.scope === types_1.MemoryScope.run ? (runId ?? null) : null,
+            value,
+            ttlSeconds,
+            expiresAt,
+        });
+    }
+};
+exports.MemoryRecordService = MemoryRecordService;
+exports.MemoryRecordService = MemoryRecordService = __decorate([
+    (0, common_1.Injectable)(),
+    __metadata("design:paramtypes", [memory_record_repository_1.MemoryRecordRepository])
+], MemoryRecordService);
+//# sourceMappingURL=memory-record.service.js.map

package/dist/workflow/services/memory-record.service.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"memory-record.service.js","sourceRoot":"","sources":["../../../src/workflow/services/memory-record.service.ts"],"names":[],"mappings":";;;;;;;;;;;;AAMA,2CAA4C;AAC5C,qCAA6D;AAE7D,4EAAmE;AAKnE,uFAAkF;AAClF,oCAA2E;AAGpE,IAAM,mBAAmB,GAAzB,MAAM,mBAAoB,SAAQ,iCAAqC;IAC5E,YAAqB,UAAkC;QACrD,KAAK,CAAC,UAAU,CAAC,CAAC;QADC,eAAU,GAAV,UAAU,CAAwB;IAEvD,CAAC;IAYD,KAAK,CAAC,iBAAiB,CAAC,EACtB,OAAO,EACP,UAAU,EACV,QAAQ,EACR,KAAK,GACiB;QACtB,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,MAAM,IAAI,KAAK,CAAC,gDAAgD,CAAC,CAAC;QACpE,CAAC;QAED,MAAM,KAAK,GAA8C;YACvD;gBACE,KAAK,EAAE,EAAE,EAAE,EAAE,OAAO,EAAE;gBACtB,UAAU,EAAE,EAAE,KAAK,EAAE,mBAAW,CAAC,MAAM,EAAE;aAC1C;SACF,CAAC;QAEF,IAAI,UAAU,EAAE,CAAC;YACf,KAAK,CAAC,IAAI,CAAC;gBACT,KAAK,EAAE,EAAE,EAAE,EAAE,OAAO,EAAE;gBACtB,QAAQ,EAAE,EAAE,EAAE,EAAE,UAAU,EAAE;gBAC5B,UAAU,EAAE,EAAE,KAAK,EAAE,mBAAW,CAAC,QAAQ,EAAE;aAC5C,CAAC,CAAC;QACL,CAAC;QAED,IAAI,QAAQ,EAAE,CAAC;YACb,KAAK,CAAC,IAAI,CAAC;gBACT,KAAK,EAAE,EAAE,EAAE,EAAE,OAAO,EAAE;gBACtB,MAAM,EAAE,EAAE,EAAE,EAAE,QAAQ,EAAE;gBACxB,UAAU,EAAE,EAAE,KAAK,EAAE,mBAAW,CAAC,MAAM,EAAE;aAC1C,CAAC,CAAC;QACL,CAAC;QAED,IAAI,KAAK,EAAE,CAAC;YACV,KAAK,CAAC,IAAI,CAAC;gBACT,KAAK,EAAE,EAAE,EAAE,EAAE,OAAO,EAAE;gBACtB,GAAG,EAAE,EAAE,EAAE,EAAE,KAAK,EAAE;gBAClB,UAAU,EAAE,EAAE,KAAK,EAAE,mBAAW,CAAC,GAAG,EAAE;aACvC,CAAC,CAAC;QACL,CAAC;QAED,MAAM,GAAG,GAAG,IAAI,IAAI,EAAE,CAAC;QACvB,MAAM,WAAW,GAAG,KAAK,CAAC,OAAO,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC;YAC3C,EAAE,GAAG,KAAK,EAAE,SAAS,EAAE,IAAA,gBAAM,GAAE,EAAE;YACjC,EAAE,GAAG,KAAK,EAAE,SAAS,EAAE,IAAA,kBAAQ,EAAC,GAAG,CAAC,EAAE;SACvC,CAAC,CAAC;QAEH,OAAO,IAAI,CAAC,eAAe,CAAC;YAC1B,KAAK,EAAE,WAAW;YAClB,KAAK,EAAE;gBACL,SAAS,EAAE,MAAM;gBACjB,SAAS,EAAE,MAAM;aAClB;SACF,CAAC,CAAC;IACL,CAAC;IAWD,KAAK,CAAC,kBAAkB,CAAC,EACvB,UAAU,EACV,OAAO,EACP,UAAU,EACV,QAAQ,EACR,KAAK,EACL,KAAK,GAQN;QACC,MAAM,UAAU,GAAG,UAAU,CAAC,UAAU,IAAI,IAAI,CAAC;QACjD,MAAM,SAAS,GACb,UAAU,IAAI,UAAU,GAAG,CAAC;YAC1B,CAAC,CAAC,IAAI,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,UAAU,GAAG,IAAI,CAAC;YAC1C,CAAC,CAAC,IAAI,CAAC;QACX,MAAM,KAAK,GAA4C;YACrD,UAAU,EAAE,EAAE,EAAE,EAAE,UAAU,CAAC,EAAE,EAAE;YACjC,KAAK,EAAE,EAAE,EAAE,EAAE,OAAO,EAAE;SACvB,CAAC;QAEF,IAAI,UAAU,CAAC,KAAK,KAAK,mBAAW,CAAC,QAAQ,EAAE,CAAC;YAC9C,KAAK,CAAC,QAAQ,GAAG,EAAE,EAAE,EAAE,UAAW,EAAE,CAAC;QACvC,CAAC;QAED,IAAI,UAAU,CAAC,KAAK,KAAK,mBAAW,CAAC,MAAM,EAAE,CAAC;YAC5C,KAAK,CAAC,MAAM,GAAG,EAAE,EAAE,EAAE,QAAS,EAAE,CAAC;QACnC,CAAC;QAED,IAAI,UAAU,CAAC,KAAK,KAAK,mBAAW,CAAC,GAAG,EAAE,CAAC;YACzC,KAAK,CAAC,GAAG,GAAG,EAAE,EAAE,EAAE,KAAM,EAAE,CAAC;QAC7B,CAAC;QAED,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC;YAClC,KAAK;YACL,KAAK,EAAE;gBACL,SAAS,EAAE,MAAM;gBACjB,SAAS,EAAE,MAAM;aAClB;SACF,CAAC,CAAC;QAEH,IAAI,QAAQ,EAAE,CAAC;YACb,MAAM,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,EAAE,EAAE;gBAChC,KAAK;gBACL,UAAU;gBACV,SAAS;aACV,CAAC,CAAC;YAEH,OAAO;QACT,CAAC;QAED,MAAM,IAAI,CAAC,MAAM,CAAC;YAChB,UAAU,EAAE,UAAU,CAAC,EAAE;YACzB,KAAK,EAAE,OAAO;YACd,QAAQ,EACN,UAAU,CAAC,KAAK,KAAK,mBAAW,CAAC,MAAM;gBACrC,CAAC,CAAC,IAAI;gBACN,CAAC,CAAC,UAAU,CAAC,KAAK,KAAK,mBAAW,CAAC,QAAQ;oBACvC,UAAU,CAAC,KAAK,KAAK,mBAAW,CAAC,GAAG;oBACtC,CAAC,CAAC,CAAC,UAAU,IAAI,IAAI,CAAC;oBACtB,CAAC,CAAC,IAAI;YACZ,MAAM,EACJ,UAAU,CAAC,KAAK,KAAK,mBAAW,CAAC,MAAM;gBACrC,CAAC,CAAC,CAAC,QAAQ,IAAI,IAAI,CAAC;gBACpB,CAAC,CAAC,UAAU,CAAC,KAAK,KAAK,mBAAW,CAAC,GAAG;oBACpC,CAAC,CAAC,CAAC,QAAQ,IAAI,IAAI,CAAC;oBACpB,CAAC,CAAC,IAAI;YACZ,GAAG,EAAE,UAAU,CAAC,KAAK,KAAK,mBAAW,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,KAAK,IAAI,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI;YAClE,KAAK;YACL,UAAU;YACV,SAAS;SACV,CAAC,CAAC;IACL,CAAC;CACF,CAAA;AA7JY,kDAAmB;8BAAnB,mBAAmB;IAD/B,IAAA,mBAAU,GAAE;qCAEsB,iDAAsB;GAD5C,mBAAmB,CA6J/B"}