@1mcp/agent 0.10.3 → 0.12.0

package/build/core/server/serverManager.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"serverManager.d.ts","sourceRoot":"","sources":["../../../src/core/server/serverManager.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,SAAS,EAAE,MAAM,+CAA+C,CAAC;AAK1E,OAAO,EAAE,OAAO,EAAE,UAAU,EAAE,eAAe,EAAE,MAAM,mBAAmB,CAAC;AAEzE,qBAAa,aAAa;IACxB,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAgB;IACvC,OAAO,CAAC,OAAO,CAAsC;IACrD,OAAO,CAAC,YAAY,CAAoC;IACxD,OAAO,CAAC,kBAAkB,CAA4C;IAEtE,OAAO,CAAC,OAAO,CAAe;IAC9B,OAAO,CAAC,UAAU,CAAiC;IACnD,OAAO,CAAC,mBAAmB,CAAyC;IACpE,OAAO,CAAC,gBAAgB,CAA0B;IAElD,OAAO;WAYO,mBAAmB,CAC/B,MAAM,EAAE;QAAE,IAAI,EAAE,MAAM,CAAC;QAAC,OAAO,EAAE,MAAM,CAAA;KAAE,EACzC,YAAY,EAAE;QAAE,YAAY,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAA;KAAE,EACvD,OAAO,EAAE,OAAO,EAChB,UAAU,EAAE,MAAM,CAAC,MAAM,EAAE,SAAS,CAAC,GACpC,aAAa;IAOhB,WAAkB,OAAO,IAAI,aAAa,CAEzC;WAGa,aAAa,IAAI,IAAI;IAatB,gBAAgB,CAAC,SAAS,EAAE,SAAS,EAAE,SAAS,EAAE,MAAM,EAAE,IAAI,EAAE,eAAe,GAAG,OAAO,CAAC,IAAI,CAAC;YA2B9F,iBAAiB;YAoBjB,SAAS;IA4BhB,mBAAmB,CAAC,SAAS,EAAE,MAAM,EAAE,UAAU,GAAE,OAAe,GAAG,IAAI;IA8BzE,YAAY,CAAC,SAAS,EAAE,MAAM,GAAG,SAAS,GAAG,SAAS;IAItD,aAAa,IAAI,GAAG,CAAC,MAAM,EAAE,SAAS,CAAC;IAUvC,mBAAmB,IAAI,MAAM,CAAC,MAAM,EAAE,SAAS,CAAC;IAIhD,wBAAwB,IAAI,MAAM;IAIlC,SAAS,CAAC,SAAS,EAAE,MAAM,GAAG,UAAU,GAAG,SAAS;IAIpD,0BAA0B,CAAC,UAAU,EAAE,OAAO,EAAE,aAAa,EAAE,MAAM,CAAC,MAAM,EAAE,SAAS,CAAC,GAAG,IAAI;CAIvG"}

package/build/core/server/serverManager.js

@@ -0,0 +1,159 @@
+import { Server } from '@modelcontextprotocol/sdk/server/index.js';
+import logger from '../../logger/logger.js';
+import configReloadService from '../../services/configReloadService.js';
+import { setupCapabilities } from '../../capabilities/capabilityManager.js';
+import { enhanceServerWithLogging } from '../../transport/http/middleware/loggingMiddleware.js';
+export class ServerManager {
+    static instance;
+    servers = new Map();
+    serverConfig;
+    serverCapabilities;
+    clients = {};
+    transports = {};
+    connectionSemaphore = new Map();
+    disconnectingIds = new Set();
+    constructor(config, capabilities, clients, transports) {
+        this.serverConfig = config;
+        this.serverCapabilities = capabilities;
+        this.clients = clients;
+        this.transports = transports;
+    }
+    static getOrCreateInstance(config, capabilities, clients, transports) {
+        if (!ServerManager.instance) {
+            ServerManager.instance = new ServerManager(config, capabilities, clients, transports);
+        }
+        return ServerManager.instance;
+    }
+    static get current() {
+        return ServerManager.instance;
+    }
+    // Test utility method to reset singleton state
+    static resetInstance() {
+        if (ServerManager.instance) {
+            // Clean up existing connections with forced close
+            for (const [sessionId] of ServerManager.instance.servers) {
+                ServerManager.instance.disconnectTransport(sessionId, true);
+            }
+            ServerManager.instance.servers.clear();
+            ServerManager.instance.connectionSemaphore.clear();
+            ServerManager.instance.disconnectingIds.clear();
+        }
+        ServerManager.instance = undefined;
+    }
+    async connectTransport(transport, sessionId, opts) {
+        // Check if a connection is already in progress for this session
+        const existingConnection = this.connectionSemaphore.get(sessionId);
+        if (existingConnection) {
+            logger.warn(`Connection already in progress for session ${sessionId}, waiting...`);
+            await existingConnection;
+            return;
+        }
+        // Check if transport is already connected
+        if (this.servers.has(sessionId)) {
+            logger.warn(`Transport already connected for session ${sessionId}`);
+            return;
+        }
+        // Create connection promise to prevent race conditions
+        const connectionPromise = this.performConnection(transport, sessionId, opts);
+        this.connectionSemaphore.set(sessionId, connectionPromise);
+        try {
+            await connectionPromise;
+        }
+        finally {
+            // Clean up the semaphore entry
+            this.connectionSemaphore.delete(sessionId);
+        }
+    }
+    async performConnection(transport, sessionId, opts) {
+        // Set connection timeout
+        const connectionTimeoutMs = 30000; // 30 seconds
+        const timeoutPromise = new Promise((_, reject) => {
+            setTimeout(() => reject(new Error(`Connection timeout for session ${sessionId}`)), connectionTimeoutMs);
+        });
+        try {
+            await Promise.race([this.doConnect(transport, sessionId, opts), timeoutPromise]);
+        }
+        catch (error) {
+            // Clean up partial connection on failure
+            if (this.servers.has(sessionId)) {
+                this.servers.delete(sessionId);
+            }
+            logger.error(`Failed to connect transport for session ${sessionId}:`, error);
+            throw error;
+        }
+    }
+    async doConnect(transport, sessionId, opts) {
+        // Create a new server instance for this transport
+        const server = new Server(this.serverConfig, this.serverCapabilities);
+        // Create server info object first
+        const serverInfo = {
+            server,
+            ...opts,
+        };
+        // Enhance server with logging middleware
+        enhanceServerWithLogging(server);
+        // Set up capabilities for this server instance
+        await setupCapabilities(this.clients, serverInfo);
+        // Update the configuration reload service with server info
+        configReloadService.updateServerInfo(sessionId, serverInfo);
+        // Store the server instance
+        this.servers.set(sessionId, serverInfo);
+        // Connect the transport to the new server instance
+        await server.connect(transport);
+        logger.info(`Connected transport for session ${sessionId}`);
+    }
+    disconnectTransport(sessionId, forceClose = false) {
+        // Prevent recursive disconnection calls
+        if (this.disconnectingIds.has(sessionId)) {
+            return;
+        }
+        const server = this.servers.get(sessionId);
+        if (server) {
+            this.disconnectingIds.add(sessionId);
+            try {
+                // Only close the transport if explicitly requested (e.g., during shutdown)
+                // Don't close if this is called from an onclose handler to avoid recursion
+                if (forceClose && server.server.transport) {
+                    try {
+                        server.server.transport.close();
+                    }
+                    catch (error) {
+                        logger.error(`Error closing transport for session ${sessionId}:`, error);
+                    }
+                }
+                this.servers.delete(sessionId);
+                configReloadService.removeServerInfo(sessionId);
+                logger.info(`Disconnected transport for session ${sessionId}`);
+            }
+            finally {
+                this.disconnectingIds.delete(sessionId);
+            }
+        }
+    }
+    getTransport(sessionId) {
+        return this.servers.get(sessionId)?.server.transport;
+    }
+    getTransports() {
+        const transports = new Map();
+        for (const [id, server] of this.servers.entries()) {
+            if (server.server.transport) {
+                transports.set(id, server.server.transport);
+            }
+        }
+        return transports;
+    }
+    getClientTransports() {
+        return this.transports;
+    }
+    getActiveTransportsCount() {
+        return this.servers.size;
+    }
+    getServer(sessionId) {
+        return this.servers.get(sessionId);
+    }
+    updateClientsAndTransports(newClients, newTransports) {
+        this.clients = newClients;
+        this.transports = newTransports;
+    }
+}
+//# sourceMappingURL=serverManager.js.map

package/build/core/server/serverManager.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"serverManager.js","sourceRoot":"","sources":["../../../src/core/server/serverManager.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,EAAE,MAAM,2CAA2C,CAAC;AAEnE,OAAO,MAAM,MAAM,wBAAwB,CAAC;AAC5C,OAAO,mBAAmB,MAAM,uCAAuC,CAAC;AACxE,OAAO,EAAE,iBAAiB,EAAE,MAAM,yCAAyC,CAAC;AAC5E,OAAO,EAAE,wBAAwB,EAAE,MAAM,sDAAsD,CAAC;AAGhG,MAAM,OAAO,aAAa;IAChB,MAAM,CAAC,QAAQ,CAAgB;IAC/B,OAAO,GAA4B,IAAI,GAAG,EAAE,CAAC;IAC7C,YAAY,CAAoC;IAChD,kBAAkB,CAA4C;IAE9D,OAAO,GAAY,EAAE,CAAC;IACtB,UAAU,GAA8B,EAAE,CAAC;IAC3C,mBAAmB,GAA+B,IAAI,GAAG,EAAE,CAAC;IAC5D,gBAAgB,GAAgB,IAAI,GAAG,EAAE,CAAC;IAElD,YACE,MAAyC,EACzC,YAAuD,EACvD,OAAgB,EAChB,UAAqC;QAErC,IAAI,CAAC,YAAY,GAAG,MAAM,CAAC;QAC3B,IAAI,CAAC,kBAAkB,GAAG,YAAY,CAAC;QACvC,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;QACvB,IAAI,CAAC,UAAU,GAAG,UAAU,CAAC;IAC/B,CAAC;IAEM,MAAM,CAAC,mBAAmB,CAC/B,MAAyC,EACzC,YAAuD,EACvD,OAAgB,EAChB,UAAqC;QAErC,IAAI,CAAC,aAAa,CAAC,QAAQ,EAAE,CAAC;YAC5B,aAAa,CAAC,QAAQ,GAAG,IAAI,aAAa,CAAC,MAAM,EAAE,YAAY,EAAE,OAAO,EAAE,UAAU,CAAC,CAAC;QACxF,CAAC;QACD,OAAO,aAAa,CAAC,QAAQ,CAAC;IAChC,CAAC;IAEM,MAAM,KAAK,OAAO;QACvB,OAAO,aAAa,CAAC,QAAQ,CAAC;IAChC,CAAC;IAED,+CAA+C;IACxC,MAAM,CAAC,aAAa;QACzB,IAAI,aAAa,CAAC,QAAQ,EAAE,CAAC;YAC3B,kDAAkD;YAClD,KAAK,MAAM,CAAC,SAAS,CAAC,IAAI,aAAa,CAAC,QAAQ,CAAC,OAAO,EAAE,CAAC;gBACzD,aAAa,CAAC,QAAQ,CAAC,mBAAmB,CAAC,SAAS,EAAE,IAAI,CAAC,CAAC;YAC9D,CAAC;YACD,aAAa,CAAC,QAAQ,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC;YACvC,aAAa,CAAC,QAAQ,CAAC,mBAAmB,CAAC,KAAK,EAAE,CAAC;YACnD,aAAa,CAAC,QAAQ,CAAC,gBAAgB,CAAC,KAAK,EAAE,CAAC;QAClD,CAAC;QACD,aAAa,CAAC,QAAQ,GAAG,SAAgB,CAAC;IAC5C,CAAC;IAEM,KAAK,CAAC,gBAAgB,CAAC,SAAoB,EAAE,SAAiB,EAAE,IAAqB;QAC1F,gEAAgE;QAChE,MAAM,kBAAkB,GAAG,IAAI,CAAC,mBAAmB,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QACnE,IAAI,kBAAkB,EAAE,CAAC;YACvB,MAAM,CAAC,IAAI,CAAC,8CAA8C,SAAS,cAAc,CAAC,CAAC;YACnF,MAAM,kBAAkB,CAAC;YACzB,OAAO;QACT,CAAC;QAED,0CAA0C;QAC1C,IAAI,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,SAAS,CAAC,EAAE,CAAC;YAChC,MAAM,CAAC,IAAI,CAAC,2CAA2C,SAAS,EAAE,CAAC,CAAC;YACpE,OAAO;QACT,CAAC;QAED,uDAAuD;QACvD,MAAM,iBAAiB,GAAG,IAAI,CAAC,iBAAiB,CAAC,SAAS,EAAE,SAAS,EAAE,IAAI,CAAC,CAAC;QAC7E,IAAI,CAAC,mBAAmB,CAAC,GAAG,CAAC,SAAS,EAAE,iBAAiB,CAAC,CAAC;QAE3D,IAAI,CAAC;YACH,MAAM,iBAAiB,CAAC;QAC1B,CAAC;gBAAS,CAAC;YACT,+BAA+B;YAC/B,IAAI,CAAC,mBAAmB,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;QAC7C,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,iBAAiB,CAAC,SAAoB,EAAE,SAAiB,EAAE,IAAqB;QAC5F,yBAAyB;QACzB,MAAM,mBAAmB,GAAG,KAAK,CAAC,CAAC,aAAa;QAEhD,MAAM,cAAc,GAAG,IAAI,OAAO,CAAQ,CAAC,CAAC,EAAE,MAAM,EAAE,EAAE;YACtD,UAAU,CAAC,GAAG,EAAE,CAAC,MAAM,CAAC,IAAI,KAAK,CAAC,kCAAkC,SAAS,EAAE,CAAC,CAAC,EAAE,mBAAmB,CAAC,CAAC;QAC1G,CAAC,CAAC,CAAC;QAEH,IAAI,CAAC;YACH,MAAM,OAAO,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,SAAS,EAAE,SAAS,EAAE,IAAI,CAAC,EAAE,cAAc,CAAC,CAAC,CAAC;QACnF,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,yCAAyC;YACzC,IAAI,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,SAAS,CAAC,EAAE,CAAC;gBAChC,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;YACjC,CAAC;YACD,MAAM,CAAC,KAAK,CAAC,2CAA2C,SAAS,GAAG,EAAE,KAAK,CAAC,CAAC;YAC7E,MAAM,KAAK,CAAC;QACd,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,SAAS,CAAC,SAAoB,EAAE,SAAiB,EAAE,IAAqB;QACpF,kDAAkD;QAClD,MAAM,MAAM,GAAG,IAAI,MAAM,CAAC,IAAI,CAAC,YAAY,EAAE,IAAI,CAAC,kBAAkB,CAAC,CAAC;QAEtE,kCAAkC;QAClC,MAAM,UAAU,GAAe;YAC7B,MAAM;YACN,GAAG,IAAI;SACR,CAAC;QAEF,yCAAyC;QACzC,wBAAwB,CAAC,MAAM,CAAC,CAAC;QAEjC,+CAA+C;QAC/C,MAAM,iBAAiB,CAAC,IAAI,CAAC,OAAO,EAAE,UAAU,CAAC,CAAC;QAElD,2DAA2D;QAC3D,mBAAmB,CAAC,gBAAgB,CAAC,SAAS,EAAE,UAAU,CAAC,CAAC;QAE5D,4BAA4B;QAC5B,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,SAAS,EAAE,UAAU,CAAC,CAAC;QAExC,mDAAmD;QACnD,MAAM,MAAM,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;QAEhC,MAAM,CAAC,IAAI,CAAC,mCAAmC,SAAS,EAAE,CAAC,CAAC;IAC9D,CAAC;IAEM,mBAAmB,CAAC,SAAiB,EAAE,aAAsB,KAAK;QACvE,wCAAwC;QACxC,IAAI,IAAI,CAAC,gBAAgB,CAAC,GAAG,CAAC,SAAS,CAAC,EAAE,CAAC;YACzC,OAAO;QACT,CAAC;QAED,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QAC3C,IAAI,MAAM,EAAE,CAAC;YACX,IAAI,CAAC,gBAAgB,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;YAErC,IAAI,CAAC;gBACH,2EAA2E;gBAC3E,2EAA2E;gBAC3E,IAAI,UAAU,IAAI,MAAM,CAAC,MAAM,CAAC,SAAS,EAAE,CAAC;oBAC1C,IAAI,CAAC;wBACH,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,KAAK,EAAE,CAAC;oBAClC,CAAC;oBAAC,OAAO,KAAK,EAAE,CAAC;wBACf,MAAM,CAAC,KAAK,CAAC,uCAAuC,SAAS,GAAG,EAAE,KAAK,CAAC,CAAC;oBAC3E,CAAC;gBACH,CAAC;gBAED,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;gBAC/B,mBAAmB,CAAC,gBAAgB,CAAC,SAAS,CAAC,CAAC;gBAChD,MAAM,CAAC,IAAI,CAAC,sCAAsC,SAAS,EAAE,CAAC,CAAC;YACjE,CAAC;oBAAS,CAAC;gBACT,IAAI,CAAC,gBAAgB,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;YAC1C,CAAC;QACH,CAAC;IACH,CAAC;IAEM,YAAY,CAAC,SAAiB;QACnC,OAAO,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,SAAS,CAAC,EAAE,MAAM,CAAC,SAAS,CAAC;IACvD,CAAC;IAEM,aAAa;QAClB,MAAM,UAAU,GAAG,IAAI,GAAG,EAAqB,CAAC;QAChD,KAAK,MAAM,CAAC,EAAE,EAAE,MAAM,CAAC,IAAI,IAAI,CAAC,OAAO,CAAC,OAAO,EAAE,EAAE,CAAC;YAClD,IAAI,MAAM,CAAC,MAAM,CAAC,SAAS,EAAE,CAAC;gBAC5B,UAAU,CAAC,GAAG,CAAC,EAAE,EAAE,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;YAC9C,CAAC;QACH,CAAC;QACD,OAAO,UAAU,CAAC;IACpB,CAAC;IAEM,mBAAmB;QACxB,OAAO,IAAI,CAAC,UAAU,CAAC;IACzB,CAAC;IAEM,wBAAwB;QAC7B,OAAO,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC;IAC3B,CAAC;IAEM,SAAS,CAAC,SAAiB;QAChC,OAAO,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;IACrC,CAAC;IAEM,0BAA0B,CAAC,UAAmB,EAAE,aAAwC;QAC7F,IAAI,CAAC,OAAO,GAAG,UAAU,CAAC;QAC1B,IAAI,CAAC,UAAU,GAAG,aAAa,CAAC;IAClC,CAAC;CACF"}

package/build/core/types/client.d.ts

@@ -0,0 +1,38 @@
+import { Client } from '@modelcontextprotocol/sdk/client/index.js';
+import { ServerCapabilities } from '@modelcontextprotocol/sdk/types.js';
+import { EnhancedTransport } from './transport.js';
+/**
+ * Enum representing possible client connection states
+ */
+export declare enum ClientStatus {
+    /** Client is successfully connected */
+    Connected = "connected",
+    /** Client is disconnected */
+    Disconnected = "disconnected",
+    /** Client encountered an error */
+    Error = "error"
+}
+/**
+ * Complete client information including transport, status and history
+ */
+export interface ClientInfo {
+    readonly name: string;
+    readonly transport: EnhancedTransport;
+    readonly client: Client;
+    readonly lastError?: Error;
+    readonly lastConnected?: Date;
+    status: ClientStatus;
+    capabilities?: ServerCapabilities;
+}
+/**
+ * Map of client information indexed by client name
+ */
+export type Clients = Record<string, ClientInfo>;
+/**
+ * Options for client operations
+ */
+export interface OperationOptions {
+    readonly retryCount?: number;
+    readonly retryDelay?: number;
+}
+//# sourceMappingURL=client.d.ts.map

package/build/core/types/client.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"client.d.ts","sourceRoot":"","sources":["../../../src/core/types/client.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,EAAE,MAAM,2CAA2C,CAAC;AACnE,OAAO,EAAE,kBAAkB,EAAE,MAAM,oCAAoC,CAAC;AACxE,OAAO,EAAE,iBAAiB,EAAE,MAAM,gBAAgB,CAAC;AAEnD;;GAEG;AACH,oBAAY,YAAY;IACtB,uCAAuC;IACvC,SAAS,cAAc;IACvB,6BAA6B;IAC7B,YAAY,iBAAiB;IAC7B,kCAAkC;IAClC,KAAK,UAAU;CAChB;AAED;;GAEG;AACH,MAAM,WAAW,UAAU;IACzB,QAAQ,CAAC,IAAI,EAAE,MAAM,CAAC;IACtB,QAAQ,CAAC,SAAS,EAAE,iBAAiB,CAAC;IACtC,QAAQ,CAAC,MAAM,EAAE,MAAM,CAAC;IACxB,QAAQ,CAAC,SAAS,CAAC,EAAE,KAAK,CAAC;IAC3B,QAAQ,CAAC,aAAa,CAAC,EAAE,IAAI,CAAC;IAC9B,MAAM,EAAE,YAAY,CAAC;IACrB,YAAY,CAAC,EAAE,kBAAkB,CAAC;CACnC;AAED;;GAEG;AACH,MAAM,MAAM,OAAO,GAAG,MAAM,CAAC,MAAM,EAAE,UAAU,CAAC,CAAC;AAEjD;;GAEG;AACH,MAAM,WAAW,gBAAgB;IAC/B,QAAQ,CAAC,UAAU,CAAC,EAAE,MAAM,CAAC;IAC7B,QAAQ,CAAC,UAAU,CAAC,EAAE,MAAM,CAAC;CAC9B"}

package/build/core/types/client.js

@@ -0,0 +1,13 @@
+/**
+ * Enum representing possible client connection states
+ */
+export var ClientStatus;
+(function (ClientStatus) {
+    /** Client is successfully connected */
+    ClientStatus["Connected"] = "connected";
+    /** Client is disconnected */
+    ClientStatus["Disconnected"] = "disconnected";
+    /** Client encountered an error */
+    ClientStatus["Error"] = "error";
+})(ClientStatus || (ClientStatus = {}));
+//# sourceMappingURL=client.js.map

package/build/core/types/client.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"client.js","sourceRoot":"","sources":["../../../src/core/types/client.ts"],"names":[],"mappings":"AAIA;;GAEG;AACH,MAAM,CAAN,IAAY,YAOX;AAPD,WAAY,YAAY;IACtB,uCAAuC;IACvC,uCAAuB,CAAA;IACvB,6BAA6B;IAC7B,6CAA6B,CAAA;IAC7B,kCAAkC;IAClC,+BAAe,CAAA;AACjB,CAAC,EAPW,YAAY,KAAZ,YAAY,QAOvB"}

package/build/core/types/index.d.ts

@@ -0,0 +1,4 @@
+export * from './transport.js';
+export * from './server.js';
+export * from './client.js';
+//# sourceMappingURL=index.d.ts.map

package/build/core/types/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/core/types/index.ts"],"names":[],"mappings":"AACA,cAAc,gBAAgB,CAAC;AAC/B,cAAc,aAAa,CAAC;AAC5B,cAAc,aAAa,CAAC"}

package/build/core/types/index.js

@@ -0,0 +1,5 @@
+// Re-export all types from focused modules
+export * from './transport.js';
+export * from './server.js';
+export * from './client.js';
+//# sourceMappingURL=index.js.map

package/build/core/types/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/core/types/index.ts"],"names":[],"mappings":"AAAA,2CAA2C;AAC3C,cAAc,gBAAgB,CAAC;AAC/B,cAAc,aAAa,CAAC;AAC5B,cAAc,aAAa,CAAC"}

package/build/core/types/server.d.ts

@@ -0,0 +1,14 @@
+import { Server } from '@modelcontextprotocol/sdk/server/index.js';
+import { ServerCapabilities } from '@modelcontextprotocol/sdk/types.js';
+export interface ServerInfoExtra {
+    readonly tags?: string[];
+    readonly enablePagination?: boolean;
+}
+/**
+ * Server information including tags
+ */
+export interface ServerInfo extends ServerInfoExtra {
+    readonly server: Server;
+}
+export type ServerCapability = keyof ServerCapabilities;
+//# sourceMappingURL=server.d.ts.map

package/build/core/types/server.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"server.d.ts","sourceRoot":"","sources":["../../../src/core/types/server.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,EAAE,MAAM,2CAA2C,CAAC;AACnE,OAAO,EAAE,kBAAkB,EAAE,MAAM,oCAAoC,CAAC;AAExE,MAAM,WAAW,eAAe;IAC9B,QAAQ,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,CAAC;IACzB,QAAQ,CAAC,gBAAgB,CAAC,EAAE,OAAO,CAAC;CACrC;AAED;;GAEG;AACH,MAAM,WAAW,UAAW,SAAQ,eAAe;IACjD,QAAQ,CAAC,MAAM,EAAE,MAAM,CAAC;CACzB;AAED,MAAM,MAAM,gBAAgB,GAAG,MAAM,kBAAkB,CAAC"}

package/build/core/types/server.js

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=server.js.map

package/build/core/types/server.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"server.js","sourceRoot":"","sources":["../../../src/core/types/server.ts"],"names":[],"mappings":""}

package/build/{types.d.ts → core/types/transport.d.ts}

@@ -1,8 +1,5 @@
 import { z } from 'zod';
-import { Server } from '@modelcontextprotocol/sdk/server/index.js';
-import { Client } from '@modelcontextprotocol/sdk/client/index.js';
 import { Transport } from '@modelcontextprotocol/sdk/shared/transport.js';
-import { ServerCapabilities } from '@modelcontextprotocol/sdk/types.js';
 /**
  * Enhanced transport interface that includes MCP-specific properties
  */
@@ -85,48 +82,4 @@ export type TransportConfig = HTTPBasedTransportConfig | StdioTransportConfig;
  * Type for MCP server parameters derived from transport config schema
  */
 export type MCPServerParams = z.infer<typeof transportConfigSchema>;
-export interface ServerInfoExtra {
-    readonly tags?: string[];
-    readonly enablePagination?: boolean;
-}
-/**
- * Server information including tags
- */
-export interface ServerInfo extends ServerInfoExtra {
-    readonly server: Server;
-}
-/**
- * Enum representing possible client connection states
- */
-export declare enum ClientStatus {
-    /** Client is successfully connected */
-    Connected = "connected",
-    /** Client is disconnected */
-    Disconnected = "disconnected",
-    /** Client encountered an error */
-    Error = "error"
-}
-/**
- * Complete client information including transport, status and history
- */
-export interface ClientInfo {
-    readonly name: string;
-    readonly transport: EnhancedTransport;
-    readonly client: Client;
-    readonly lastError?: Error;
-    readonly lastConnected?: Date;
-    status: ClientStatus;
-    capabilities?: ServerCapabilities;
-}
-/**
- * Map of client information indexed by client name
- */
-export type Clients = Record<string, ClientInfo>;
-/**
- * Options for client operations
- */
-export interface OperationOptions {
-    readonly retryCount?: number;
-    readonly retryDelay?: number;
-}
-//# sourceMappingURL=types.d.ts.map
+//# sourceMappingURL=transport.d.ts.map

package/build/core/types/transport.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"transport.d.ts","sourceRoot":"","sources":["../../../src/core/types/transport.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,EAAE,SAAS,EAAE,MAAM,+CAA+C,CAAC;AAE1E;;GAEG;AACH,MAAM,WAAW,iBAAkB,SAAQ,SAAS;IAClD,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,IAAI,CAAC,EAAE,MAAM,EAAE,CAAC;CACjB;AAED;;GAEG;AACH,MAAM,WAAW,mBAAmB;IAClC,QAAQ,CAAC,OAAO,CAAC,EAAE,MAAM,CAAC;IAC1B,QAAQ,CAAC,QAAQ,CAAC,EAAE,OAAO,CAAC;IAC5B,QAAQ,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,CAAC;CAC1B;AAED;;GAEG;AACH,MAAM,WAAW,wBAAyB,SAAQ,mBAAmB;IACnE,QAAQ,CAAC,IAAI,EAAE,MAAM,GAAG,KAAK,CAAC;IAC9B,QAAQ,CAAC,GAAG,EAAE,MAAM,CAAC;IACrB,QAAQ,CAAC,OAAO,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;CAC3C;AAED;;GAEG;AACH,MAAM,WAAW,oBAAqB,SAAQ,mBAAmB;IAC/D,QAAQ,CAAC,IAAI,EAAE,OAAO,CAAC;IACvB,QAAQ,CAAC,OAAO,EAAE,MAAM,CAAC;IACzB,QAAQ,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,CAAC;IACzB,QAAQ,CAAC,MAAM,CAAC,EAAE,MAAM,GAAG,MAAM,CAAC;IAClC,QAAQ,CAAC,GAAG,CAAC,EAAE,MAAM,CAAC;IACtB,QAAQ,CAAC,GAAG,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;CACvC;AAED;;GAEG;AACH,eAAO,MAAM,qBAAqB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;EAgBhC,CAAC;AAEH;;GAEG;AACH,MAAM,MAAM,eAAe,GAAG,wBAAwB,GAAG,oBAAoB,CAAC;AAE9E;;GAEG;AACH,MAAM,MAAM,eAAe,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,qBAAqB,CAAC,CAAC"}

package/build/{types.js → core/types/transport.js}

@@ -17,16 +17,4 @@ export const transportConfigSchema = z.object({
     cwd: z.string().optional(),
     env: z.record(z.string()).optional(),
 });
-/**
- * Enum representing possible client connection states
- */
-export var ClientStatus;
-(function (ClientStatus) {
-    /** Client is successfully connected */
-    ClientStatus["Connected"] = "connected";
-    /** Client is disconnected */
-    ClientStatus["Disconnected"] = "disconnected";
-    /** Client encountered an error */
-    ClientStatus["Error"] = "error";
-})(ClientStatus || (ClientStatus = {}));
-//# sourceMappingURL=types.js.map
+//# sourceMappingURL=transport.js.map

package/build/core/types/transport.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"transport.js","sourceRoot":"","sources":["../../../src/core/types/transport.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAyCxB;;GAEG;AACH,MAAM,CAAC,MAAM,qBAAqB,GAAG,CAAC,CAAC,MAAM,CAAC;IAC5C,IAAI,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,OAAO,EAAE,KAAK,EAAE,MAAM,EAAE,gBAAgB,CAAC,CAAC,CAAC,QAAQ,EAAE;IACnE,QAAQ,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC,QAAQ,EAAE;IAChC,OAAO,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;IAC9B,IAAI,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC,QAAQ,EAAE;IAEpC,sBAAsB;IACtB,GAAG,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,GAAG,EAAE,CAAC,QAAQ,EAAE;IAChC,OAAO,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC,QAAQ,EAAE;IAExC,+BAA+B;IAC/B,OAAO,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;IAC9B,IAAI,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC,QAAQ,EAAE;IACpC,MAAM,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC,QAAQ,EAAE;IACpD,GAAG,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;IAC1B,GAAG,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC,QAAQ,EAAE;CACrC,CAAC,CAAC"}

package/build/handlers/notificationHandlers.d.ts

@@ -1,4 +1,4 @@
-import { Clients, ServerInfo } from '../types.js';
+import { Clients, ServerInfo } from '../core/types/index.js';
 /**
  * Sets up client-to-server notification handlers
  * @param clients Record of client instances

package/build/handlers/notificationHandlers.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"notificationHandlers.d.ts","sourceRoot":"","sources":["../../src/handlers/notificationHandlers.ts"],"names":[],"mappings":"AAaA,OAAO,EAAE,OAAO,EAAE,UAAU,EAAgB,MAAM,aAAa,CAAC;AAChE;;;;GAIG;AACH,wBAAgB,gCAAgC,CAAC,OAAO,EAAE,OAAO,EAAE,UAAU,EAAE,UAAU,GAAG,IAAI,CA4B/F;AAED;;;;GAIG;AACH,wBAAgB,gCAAgC,CAAC,OAAO,EAAE,OAAO,EAAE,UAAU,EAAE,UAAU,GAAG,IAAI,CA8B/F"}
+{"version":3,"file":"notificationHandlers.d.ts","sourceRoot":"","sources":["../../src/handlers/notificationHandlers.ts"],"names":[],"mappings":"AAaA,OAAO,EAAE,OAAO,EAAE,UAAU,EAAgB,MAAM,wBAAwB,CAAC;AAC3E;;;;GAIG;AACH,wBAAgB,gCAAgC,CAAC,OAAO,EAAE,OAAO,EAAE,UAAU,EAAE,UAAU,GAAG,IAAI,CAwC/F;AAED;;;;GAIG;AACH,wBAAgB,gCAAgC,CAAC,OAAO,EAAE,OAAO,EAAE,UAAU,EAAE,UAAU,GAAG,IAAI,CAyC/F"}

package/build/handlers/notificationHandlers.js

@@ -1,7 +1,7 @@
 import { CancelledNotificationSchema, ProgressNotificationSchema, LoggingMessageNotificationSchema, ResourceUpdatedNotificationSchema, ResourceListChangedNotificationSchema, ToolListChangedNotificationSchema, PromptListChangedNotificationSchema, InitializedNotificationSchema, RootsListChangedNotificationSchema, } from '@modelcontextprotocol/sdk/types.js';
 import logger from '../logger/logger.js';
 import { withErrorHandling } from '../utils/errorHandling.js';
-import { ClientStatus } from '../types.js';
+import { ClientStatus } from '../core/types/index.js';
 /**
  * Sets up client-to-server notification handlers
  * @param clients Record of client instances
@@ -21,13 +21,26 @@ export function setupClientToServerNotifications(clients, serverInfo) {
         clientNotificationSchemas.forEach((schema) => {
             clientInfo.client.setNotificationHandler(schema, withErrorHandling(async (notification) => {
                 logger.info(`Received notification in client: ${name} ${JSON.stringify(notification)}`);
-                serverInfo.server.notification({
-                    ...notification,
-                    params: {
-                        ...notification.params,
-                        server: name,
-                    },
-                });
+                // Try to send notification, catch connection errors gracefully
+                try {
+                    // Preserve original message structure and only modify params
+                    const forwardedNotification = {
+                        method: notification.method,
+                        params: {
+                            ...notification.params,
+                            server: name,
+                        },
+                    };
+                    serverInfo.server.notification(forwardedNotification);
+                }
+                catch (error) {
+                    if (error instanceof Error && error.message.includes('Not connected')) {
+                        logger.warn(`Server transport not connected. Dropping notification from ${name}`);
+                    }
+                    else {
+                        logger.error(`Failed to send notification from ${name}: ${error}`);
+                    }
+                }
             }, `Error handling client notification from ${name}`));
         });
     }
@@ -52,13 +65,26 @@ export function setupServerToClientNotifications(clients, serverInfo) {
                     logger.warn(`Client ${name} is not connected. Notification not sent.`);
                     return;
                 }
-                clientInfo.client.notification({
-                    ...notification,
-                    params: {
-                        ...notification.params,
-                        client: name,
-                    },
-                });
+                // Try to send notification, catch connection errors gracefully
+                try {
+                    // Preserve original message structure and only modify params
+                    const forwardedNotification = {
+                        method: notification.method,
+                        params: {
+                            ...notification.params,
+                            client: name,
+                        },
+                    };
+                    clientInfo.client.notification(forwardedNotification);
+                }
+                catch (error) {
+                    if (error instanceof Error && error.message.includes('Not connected')) {
+                        logger.warn(`Client ${name} transport not connected. Dropping notification.`);
+                    }
+                    else {
+                        logger.error(`Failed to send notification to ${name}: ${error}`);
+                    }
+                }
             }, `Error handling server notification to ${name}`));
         });
     }

package/build/handlers/notificationHandlers.js.map

@@ -1 +1 @@
-{"version":3,"file":"notificationHandlers.js","sourceRoot":"","sources":["../../src/handlers/notificationHandlers.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,2BAA2B,EAC3B,0BAA0B,EAC1B,gCAAgC,EAChC,iCAAiC,EACjC,qCAAqC,EACrC,iCAAiC,EACjC,mCAAmC,EACnC,6BAA6B,EAC7B,kCAAkC,GACnC,MAAM,oCAAoC,CAAC;AAC5C,OAAO,MAAM,MAAM,qBAAqB,CAAC;AACzC,OAAO,EAAE,iBAAiB,EAAE,MAAM,2BAA2B,CAAC;AAC9D,OAAO,EAAuB,YAAY,EAAE,MAAM,aAAa,CAAC;AAChE;;;;GAIG;AACH,MAAM,UAAU,gCAAgC,CAAC,OAAgB,EAAE,UAAsB;IACvF,MAAM,yBAAyB,GAAG;QAChC,2BAA2B;QAC3B,0BAA0B;QAC1B,gCAAgC;QAChC,iCAAiC;QACjC,qCAAqC;QACrC,iCAAiC;QACjC,mCAAmC;KACpC,CAAC;IAEF,KAAK,MAAM,CAAC,IAAI,EAAE,UAAU,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC;QACzD,yBAAyB,CAAC,OAAO,CAAC,CAAC,MAAM,EAAE,EAAE;YAC3C,UAAU,CAAC,MAAM,CAAC,sBAAsB,CACtC,MAAM,EACN,iBAAiB,CAAC,KAAK,EAAE,YAAY,EAAE,EAAE;gBACvC,MAAM,CAAC,IAAI,CAAC,oCAAoC,IAAI,IAAI,IAAI,CAAC,SAAS,CAAC,YAAY,CAAC,EAAE,CAAC,CAAC;gBACxF,UAAU,CAAC,MAAM,CAAC,YAAY,CAAC;oBAC7B,GAAG,YAAY;oBACf,MAAM,EAAE;wBACN,GAAG,YAAY,CAAC,MAAM;wBACtB,MAAM,EAAE,IAAI;qBACb;iBACF,CAAC,CAAC;YACL,CAAC,EAAE,2CAA2C,IAAI,EAAE,CAAC,CACtD,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC;AACH,CAAC;AAED;;;;GAIG;AACH,MAAM,UAAU,gCAAgC,CAAC,OAAgB,EAAE,UAAsB;IACvF,MAAM,yBAAyB,GAAG;QAChC,2BAA2B;QAC3B,0BAA0B;QAC1B,6BAA6B;QAC7B,kCAAkC;KACnC,CAAC;IAEF,KAAK,MAAM,CAAC,IAAI,EAAE,UAAU,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC;QACzD,yBAAyB,CAAC,OAAO,CAAC,CAAC,MAAM,EAAE,EAAE;YAC3C,UAAU,CAAC,MAAM,CAAC,sBAAsB,CACtC,MAAM,EACN,iBAAiB,CAAC,KAAK,EAAE,YAAY,EAAE,EAAE;gBACvC,MAAM,CAAC,IAAI,CAAC,oCAAoC,IAAI,IAAI,IAAI,CAAC,SAAS,CAAC,YAAY,CAAC,EAAE,CAAC,CAAC;gBACxF,IAAI,UAAU,CAAC,MAAM,KAAK,YAAY,CAAC,SAAS,EAAE,CAAC;oBACjD,MAAM,CAAC,IAAI,CAAC,UAAU,IAAI,2CAA2C,CAAC,CAAC;oBACvE,OAAO;gBACT,CAAC;gBAED,UAAU,CAAC,MAAM,CAAC,YAAY,CAAC;oBAC7B,GAAG,YAAY;oBACf,MAAM,EAAE;wBACN,GAAG,YAAY,CAAC,MAAM;wBACtB,MAAM,EAAE,IAAI;qBACb;iBACF,CAAC,CAAC;YACL,CAAC,EAAE,yCAAyC,IAAI,EAAE,CAAC,CACpD,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC;AACH,CAAC"}
+{"version":3,"file":"notificationHandlers.js","sourceRoot":"","sources":["../../src/handlers/notificationHandlers.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,2BAA2B,EAC3B,0BAA0B,EAC1B,gCAAgC,EAChC,iCAAiC,EACjC,qCAAqC,EACrC,iCAAiC,EACjC,mCAAmC,EACnC,6BAA6B,EAC7B,kCAAkC,GACnC,MAAM,oCAAoC,CAAC;AAC5C,OAAO,MAAM,MAAM,qBAAqB,CAAC;AACzC,OAAO,EAAE,iBAAiB,EAAE,MAAM,2BAA2B,CAAC;AAC9D,OAAO,EAAuB,YAAY,EAAE,MAAM,wBAAwB,CAAC;AAC3E;;;;GAIG;AACH,MAAM,UAAU,gCAAgC,CAAC,OAAgB,EAAE,UAAsB;IACvF,MAAM,yBAAyB,GAAG;QAChC,2BAA2B;QAC3B,0BAA0B;QAC1B,gCAAgC;QAChC,iCAAiC;QACjC,qCAAqC;QACrC,iCAAiC;QACjC,mCAAmC;KACpC,CAAC;IAEF,KAAK,MAAM,CAAC,IAAI,EAAE,UAAU,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC;QACzD,yBAAyB,CAAC,OAAO,CAAC,CAAC,MAAM,EAAE,EAAE;YAC3C,UAAU,CAAC,MAAM,CAAC,sBAAsB,CACtC,MAAM,EACN,iBAAiB,CAAC,KAAK,EAAE,YAAY,EAAE,EAAE;gBACvC,MAAM,CAAC,IAAI,CAAC,oCAAoC,IAAI,IAAI,IAAI,CAAC,SAAS,CAAC,YAAY,CAAC,EAAE,CAAC,CAAC;gBAExF,+DAA+D;gBAC/D,IAAI,CAAC;oBACH,6DAA6D;oBAC7D,MAAM,qBAAqB,GAAG;wBAC5B,MAAM,EAAE,YAAY,CAAC,MAAM;wBAC3B,MAAM,EAAE;4BACN,GAAG,YAAY,CAAC,MAAM;4BACtB,MAAM,EAAE,IAAI;yBACb;qBACF,CAAC;oBACF,UAAU,CAAC,MAAM,CAAC,YAAY,CAAC,qBAAqB,CAAC,CAAC;gBACxD,CAAC;gBAAC,OAAO,KAAK,EAAE,CAAC;oBACf,IAAI,KAAK,YAAY,KAAK,IAAI,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,eAAe,CAAC,EAAE,CAAC;wBACtE,MAAM,CAAC,IAAI,CAAC,8DAA8D,IAAI,EAAE,CAAC,CAAC;oBACpF,CAAC;yBAAM,CAAC;wBACN,MAAM,CAAC,KAAK,CAAC,oCAAoC,IAAI,KAAK,KAAK,EAAE,CAAC,CAAC;oBACrE,CAAC;gBACH,CAAC;YACH,CAAC,EAAE,2CAA2C,IAAI,EAAE,CAAC,CACtD,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC;AACH,CAAC;AAED;;;;GAIG;AACH,MAAM,UAAU,gCAAgC,CAAC,OAAgB,EAAE,UAAsB;IACvF,MAAM,yBAAyB,GAAG;QAChC,2BAA2B;QAC3B,0BAA0B;QAC1B,6BAA6B;QAC7B,kCAAkC;KACnC,CAAC;IAEF,KAAK,MAAM,CAAC,IAAI,EAAE,UAAU,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC;QACzD,yBAAyB,CAAC,OAAO,CAAC,CAAC,MAAM,EAAE,EAAE;YAC3C,UAAU,CAAC,MAAM,CAAC,sBAAsB,CACtC,MAAM,EACN,iBAAiB,CAAC,KAAK,EAAE,YAAY,EAAE,EAAE;gBACvC,MAAM,CAAC,IAAI,CAAC,oCAAoC,IAAI,IAAI,IAAI,CAAC,SAAS,CAAC,YAAY,CAAC,EAAE,CAAC,CAAC;gBACxF,IAAI,UAAU,CAAC,MAAM,KAAK,YAAY,CAAC,SAAS,EAAE,CAAC;oBACjD,MAAM,CAAC,IAAI,CAAC,UAAU,IAAI,2CAA2C,CAAC,CAAC;oBACvE,OAAO;gBACT,CAAC;gBAED,+DAA+D;gBAC/D,IAAI,CAAC;oBACH,6DAA6D;oBAC7D,MAAM,qBAAqB,GAAG;wBAC5B,MAAM,EAAE,YAAY,CAAC,MAAM;wBAC3B,MAAM,EAAE;4BACN,GAAG,YAAY,CAAC,MAAM;4BACtB,MAAM,EAAE,IAAI;yBACb;qBACF,CAAC;oBACF,UAAU,CAAC,MAAM,CAAC,YAAY,CAAC,qBAAqB,CAAC,CAAC;gBACxD,CAAC;gBAAC,OAAO,KAAK,EAAE,CAAC;oBACf,IAAI,KAAK,YAAY,KAAK,IAAI,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,eAAe,CAAC,EAAE,CAAC;wBACtE,MAAM,CAAC,IAAI,CAAC,UAAU,IAAI,kDAAkD,CAAC,CAAC;oBAChF,CAAC;yBAAM,CAAC;wBACN,MAAM,CAAC,KAAK,CAAC,kCAAkC,IAAI,KAAK,KAAK,EAAE,CAAC,CAAC;oBACnE,CAAC;gBACH,CAAC;YACH,CAAC,EAAE,yCAAyC,IAAI,EAAE,CAAC,CACpD,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC;AACH,CAAC"}

package/build/handlers/requestHandlers.d.ts

@@ -1,4 +1,4 @@
-import { Clients, ServerInfo } from '../types.js';
+import { Clients, ServerInfo } from '../core/types/index.js';
 /**
  * Registers all request handlers based on available capabilities
  * @param clients Record of client instances

package/build/handlers/requestHandlers.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"requestHandlers.d.ts","sourceRoot":"","sources":["../../src/handlers/requestHandlers.ts"],"names":[],"mappings":"AA0BA,OAAO,EAAE,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AAkClD;;;;;;GAMG;AACH,wBAAgB,uBAAuB,CAAC,OAAO,EAAE,OAAO,EAAE,UAAU,EAAE,UAAU,GAAG,IAAI,CAkBtF"}
+{"version":3,"file":"requestHandlers.d.ts","sourceRoot":"","sources":["../../src/handlers/requestHandlers.ts"],"names":[],"mappings":"AA+BA,OAAO,EAAE,OAAO,EAAE,UAAU,EAAgB,MAAM,wBAAwB,CAAC;AAqD3E;;;;;;GAMG;AACH,wBAAgB,uBAAuB,CAAC,OAAO,EAAE,OAAO,EAAE,UAAU,EAAE,UAAU,GAAG,IAAI,CA6CtF"}

package/build/handlers/requestHandlers.js

@@ -1,10 +1,12 @@
-import { CallToolRequestSchema, ListResourcesRequestSchema, ListToolsRequestSchema, ReadResourceRequestSchema, ListPromptsRequestSchema, GetPromptRequestSchema, ListResourceTemplatesRequestSchema, SubscribeRequestSchema, UnsubscribeRequestSchema, SetLevelRequestSchema, CallToolResultSchema, CreateMessageRequestSchema, ListRootsRequestSchema, } from '@modelcontextprotocol/sdk/types.js';
+import { CallToolRequestSchema, ListResourcesRequestSchema, ListToolsRequestSchema, ReadResourceRequestSchema, ListPromptsRequestSchema, GetPromptRequestSchema, ListResourceTemplatesRequestSchema, SubscribeRequestSchema, UnsubscribeRequestSchema, SetLevelRequestSchema, CallToolResultSchema, CreateMessageRequestSchema, ListRootsRequestSchema, ElicitRequestSchema, PingRequestSchema, CompleteRequestSchema, } from '@modelcontextprotocol/sdk/types.js';
 import { setLogLevel } from '../logger/logger.js';
 import { MCP_URI_SEPARATOR } from '../constants.js';
-import { executeClientOperation, executeServerOperation } from '../clients/clientManager.js';
+import { executeClientOperation, executeServerOperation } from '../core/client/clientManager.js';
 import { parseUri, withErrorHandling } from '../utils/errorHandling.js';
 import { filterClients, byCapabilities, byTags } from '../utils/clientFiltering.js';
+import { ClientStatus } from '../core/types/index.js';
 import { handlePagination } from '../utils/pagination.js';
+import logger from '../logger/logger.js';
 /**
  * Registers server-specific request handlers
  * @param clients Record of client instances
@@ -12,11 +14,19 @@ import { handlePagination } from '../utils/pagination.js';
  */
 function registerServerRequestHandlers(clients, serverInfo) {
     Object.entries(clients).forEach(([_, clientInfo]) => {
+        clientInfo.client.setRequestHandler(PingRequestSchema, withErrorHandling(async () => {
+            return executeServerOperation(serverInfo, (_server) => _server.server.ping());
+        }, 'Error pinging'));
         clientInfo.client.setRequestHandler(CreateMessageRequestSchema, withErrorHandling(async (request) => {
             return executeServerOperation(serverInfo, (_server) => _server.server.createMessage(request.params, {
                 timeout: clientInfo.transport.timeout,
             }));
         }, 'Error creating message'));
+        clientInfo.client.setRequestHandler(ElicitRequestSchema, withErrorHandling(async (request) => {
+            return executeServerOperation(serverInfo, (_server) => _server.server.elicitInput(request.params, {
+                timeout: clientInfo.transport.timeout,
+            }));
+        }, 'Error eliciting input'));
         clientInfo.client.setRequestHandler(ListRootsRequestSchema, withErrorHandling(async (request) => {
             return executeServerOperation(serverInfo, (_server) => _server.server.listRoots(request.params, {
                 timeout: clientInfo.transport.timeout,
@@ -37,12 +47,33 @@ export function registerRequestHandlers(clients, serverInfo) {
         setLogLevel(request.params.level);
         return {};
     });
+    // Register ping handler
+    serverInfo.server.setRequestHandler(PingRequestSchema, withErrorHandling(async () => {
+        // Health check all connected upstream clients
+        const healthCheckPromises = Object.entries(clients).map(async ([clientName, clientInfo]) => {
+            if (clientInfo.status === ClientStatus.Connected) {
+                try {
+                    await clientInfo.client.ping();
+                    logger.info(`Health check successful for client: ${clientName}`);
+                }
+                catch (error) {
+                    logger.warn(`Health check failed for client ${clientName}: ${error}`);
+                }
+            }
+        });
+        // Wait for all health checks to complete (but don't fail if some fail)
+        await Promise.allSettled(healthCheckPromises);
+        // Always return successful pong response
+        return {};
+    }, 'Error handling ping'));
     // Register resource-related handlers
     registerResourceHandlers(clients, serverInfo);
     // Register tool-related handlers
     registerToolHandlers(clients, serverInfo);
     // Register prompt-related handlers
     registerPromptHandlers(clients, serverInfo);
+    // Register completion-related handlers
+    registerCompletionHandlers(clients, serverInfo);
     // Register server-specific request handlers
     registerServerRequestHandlers(clients, serverInfo);
 }
@@ -115,6 +146,8 @@ function registerToolHandlers(clients, serverInfo) {
             name: `${clientInfo.name}${MCP_URI_SEPARATOR}${tool.name}`,
             description: tool.description,
             inputSchema: tool.inputSchema,
+            outputSchema: tool.outputSchema,
+            annotations: tool.annotations,
         })) ?? [], serverInfo.enablePagination ?? false);
         return {
             tools: result.items,
@@ -154,4 +187,34 @@ function registerPromptHandlers(clients, serverInfo) {
         return executeClientOperation(clients, clientName, (clientInfo) => clientInfo.client.getPrompt({ ...request.params, name: promptName }));
     }, 'Error getting prompt'));
 }
+/**
+ * Registers completion-related request handlers
+ * @param clients Record of client instances
+ * @param serverInfo The MCP server instance
+ */
+function registerCompletionHandlers(clients, serverInfo) {
+    serverInfo.server.setRequestHandler(CompleteRequestSchema, withErrorHandling(async (request) => {
+        const { ref } = request.params;
+        let clientName;
+        let updatedRef;
+        if (ref.type === 'ref/prompt') {
+            const { clientName: cn, resourceName } = parseUri(ref.name, MCP_URI_SEPARATOR);
+            clientName = cn;
+            updatedRef = { ...ref, name: resourceName };
+        }
+        else if (ref.type === 'ref/resource') {
+            const { clientName: cn, resourceName } = parseUri(ref.uri, MCP_URI_SEPARATOR);
+            clientName = cn;
+            updatedRef = { ...ref, uri: resourceName };
+        }
+        else {
+            // This should be caught by the schema validation, but as a safeguard:
+            throw new Error(`Unsupported completion reference type: ${ref.type}`);
+        }
+        const params = { ...request.params, ref: updatedRef };
+        return executeClientOperation(clients, clientName, (clientInfo) => clientInfo.client.complete(params, {
+            timeout: clientInfo.transport.timeout,
+        }), {}, 'completions');
+    }, 'Error handling completion'));
+}
 //# sourceMappingURL=requestHandlers.js.map