@agiflowai/one-mcp 0.3.16 → 0.3.17

package/dist/cli.cjs

@@ -1,5 +1,5 @@
 #!/usr/bin/env node
-const require_src = require('./src-BRqEdbha.cjs');
+const require_src = require('./src-B8Rb84GQ.cjs');
 let node_fs_promises = require("node:fs/promises");
 let node_fs = require("node:fs");
 let node_crypto = require("node:crypto");

package/dist/cli.mjs

@@ -1,5 +1,5 @@
 #!/usr/bin/env node
-import { _ as DefinitionsCacheService, a as SseTransportHandler, b as ConfigFetcherService, c as createServer, d as version, g as McpClientManagerService, h as SkillService, i as StdioTransportHandler, l as createSessionServer, n as RuntimeStateService, o as HttpTransportHandler, p as SearchListToolsTool, r as StdioHttpTransportHandler, s as TRANSPORT_MODE, t as StopServerService, u as initializeSharedServices, v as generateServerId, y as findConfigFile } from "./src-Dn6vMZIk.mjs";
+import { _ as DefinitionsCacheService, a as SseTransportHandler, b as ConfigFetcherService, c as createServer, d as version, g as McpClientManagerService, h as SkillService, i as StdioTransportHandler, l as createSessionServer, n as RuntimeStateService, o as HttpTransportHandler, p as SearchListToolsTool, r as StdioHttpTransportHandler, s as TRANSPORT_MODE, t as StopServerService, u as initializeSharedServices, v as generateServerId, y as findConfigFile } from "./src-D5YMjGgs.mjs";
 import { access, writeFile } from "node:fs/promises";
 import { constants } from "node:fs";
 import { randomUUID } from "node:crypto";

package/dist/index.cjs

@@ -1,4 +1,4 @@
-const require_src = require('./src-BRqEdbha.cjs');
+const require_src = require('./src-B8Rb84GQ.cjs');
 
 exports.ConfigFetcherService = require_src.ConfigFetcherService;
 exports.DefinitionsCacheService = require_src.DefinitionsCacheService;

package/dist/index.mjs

@@ -1,3 +1,3 @@
-import { _ as DefinitionsCacheService, a as SseTransportHandler, b as ConfigFetcherService, c as createServer, f as UseToolTool, g as McpClientManagerService, h as SkillService, i as StdioTransportHandler, l as createSessionServer, m as DescribeToolsTool, n as RuntimeStateService, o as HttpTransportHandler, p as SearchListToolsTool, r as StdioHttpTransportHandler, s as TRANSPORT_MODE, t as StopServerService, u as initializeSharedServices, v as generateServerId, y as findConfigFile } from "./src-Dn6vMZIk.mjs";
+import { _ as DefinitionsCacheService, a as SseTransportHandler, b as ConfigFetcherService, c as createServer, f as UseToolTool, g as McpClientManagerService, h as SkillService, i as StdioTransportHandler, l as createSessionServer, m as DescribeToolsTool, n as RuntimeStateService, o as HttpTransportHandler, p as SearchListToolsTool, r as StdioHttpTransportHandler, s as TRANSPORT_MODE, t as StopServerService, u as initializeSharedServices, v as generateServerId, y as findConfigFile } from "./src-D5YMjGgs.mjs";
 
 export { ConfigFetcherService, DefinitionsCacheService, DescribeToolsTool, HttpTransportHandler, McpClientManagerService, RuntimeStateService, SearchListToolsTool, SkillService, SseTransportHandler, StdioHttpTransportHandler, StdioTransportHandler, StopServerService, TRANSPORT_MODE, UseToolTool, createServer, createSessionServer, findConfigFile, generateServerId, initializeSharedServices };

package/dist/{src-BRqEdbha.cjs → src-B8Rb84GQ.cjs}

@@ -1453,6 +1453,14 @@ var DefinitionsCacheService = class {
 /** Default connection timeout in milliseconds (30 seconds) */
 const DEFAULT_CONNECTION_TIMEOUT_MS = 3e4;
 /**
+* Checks if an error is a session-related error from an HTTP backend
+* (e.g., downstream server restarted and no longer recognizes the session ID).
+*/
+function isSessionError(error) {
+	const message = error instanceof Error ? error.message : String(error);
+	return message.includes("unknown session") || message.includes("Session not found");
+}
+/**
 * MCP Client wrapper for managing individual server connections
 * This is an internal class used by McpClientManagerService
 */
@@ -1466,6 +1474,7 @@ var McpClient = class {
 	client;
 	childProcess;
 	connected = false;
+	reconnectFn;
 	constructor(serverName, transport, client, config) {
 		this.serverName = serverName;
 		this.serverInstruction = config.instruction;
@@ -1481,34 +1490,76 @@ var McpClient = class {
 	setConnected(connected) {
 		this.connected = connected;
 	}
+	/**
+	* Sets a reconnection function that creates a fresh Client and transport.
+	* Called automatically by withSessionRetry when a session error is detected
+	* (e.g., downstream HTTP server restarted and the old session ID is invalid).
+	*/
+	setReconnectFn(fn) {
+		this.reconnectFn = fn;
+	}
+	/**
+	* Wraps an operation with automatic retry on session errors.
+	* If the operation fails with a session error (e.g., downstream server restarted),
+	* reconnects and retries once.
+	*/
+	async withSessionRetry(operation) {
+		try {
+			return await operation();
+		} catch (error) {
+			if (!this.reconnectFn || !isSessionError(error)) throw error;
+			console.error(`Session error for ${this.serverName}, reconnecting: ${error instanceof Error ? error.message : String(error)}`);
+			try {
+				await this.client.close();
+			} catch (closeError) {
+				console.error(`Failed to close stale client for ${this.serverName}:`, closeError);
+			}
+			const result = await this.reconnectFn();
+			this.client = result.client;
+			if (result.childProcess) this.childProcess = result.childProcess;
+			return await operation();
+		}
+	}
 	async listTools() {
 		if (!this.connected) throw new Error(`Client for ${this.serverName} is not connected`);
-		return (await this.client.listTools()).tools;
+		return this.withSessionRetry(async () => {
+			return (await this.client.listTools()).tools;
+		});
 	}
 	async listResources() {
 		if (!this.connected) throw new Error(`Client for ${this.serverName} is not connected`);
-		return (await this.client.listResources()).resources;
+		return this.withSessionRetry(async () => {
+			return (await this.client.listResources()).resources;
+		});
 	}
 	async listPrompts() {
 		if (!this.connected) throw new Error(`Client for ${this.serverName} is not connected`);
-		return (await this.client.listPrompts()).prompts;
+		return this.withSessionRetry(async () => {
+			return (await this.client.listPrompts()).prompts;
+		});
 	}
 	async callTool(name, args) {
 		if (!this.connected) throw new Error(`Client for ${this.serverName} is not connected`);
-		return await this.client.callTool({
-			name,
-			arguments: args
+		return this.withSessionRetry(async () => {
+			return await this.client.callTool({
+				name,
+				arguments: args
+			});
 		});
 	}
 	async readResource(uri) {
 		if (!this.connected) throw new Error(`Client for ${this.serverName} is not connected`);
-		return await this.client.readResource({ uri });
+		return this.withSessionRetry(async () => {
+			return await this.client.readResource({ uri });
+		});
 	}
 	async getPrompt(name, args) {
 		if (!this.connected) throw new Error(`Client for ${this.serverName} is not connected`);
-		return await this.client.getPrompt({
-			name,
-			arguments: args
+		return this.withSessionRetry(async () => {
+			return await this.client.getPrompt({
+				name,
+				arguments: args
+			});
 		});
 	}
 	async close() {
@@ -1589,6 +1640,20 @@ var McpClientManagerService = class {
 		try {
 			await Promise.race([this.performConnection(mcpClient, config), new Promise((_, reject) => setTimeout(() => reject(/* @__PURE__ */ new Error(`Connection timeout after ${timeoutMs}ms`)), timeoutMs))]);
 			mcpClient.setConnected(true);
+			if (config.transport === "http" || config.transport === "sse") mcpClient.setReconnectFn(async () => {
+				try {
+					const newClient = new __modelcontextprotocol_sdk_client_index_js.Client({
+						name: "@agiflowai/one-mcp-client",
+						version: "0.1.0"
+					}, { capabilities: {} });
+					const newMcpClient = new McpClient(serverName, config.transport, newClient, {});
+					await this.performConnection(newMcpClient, config);
+					return { client: newClient };
+				} catch (error) {
+					console.error(`Failed to reconnect to ${serverName}:`, error);
+					throw error;
+				}
+			});
 			if (!mcpClient.serverInstruction) try {
 				const serverInstruction = mcpClient["client"].getInstructions();
 				if (serverInstruction) mcpClient.serverInstruction = serverInstruction;
@@ -2796,7 +2861,7 @@ IMPORTANT: Only use tools discovered from describe_tools with id="${this.serverI
 
 //#endregion
 //#region package.json
-var version = "0.3.15";
+var version = "0.3.16";
 
 //#endregion
 //#region src/server/index.ts

package/dist/{src-Dn6vMZIk.mjs → src-D5YMjGgs.mjs}

@@ -1424,6 +1424,14 @@ var DefinitionsCacheService = class {
 /** Default connection timeout in milliseconds (30 seconds) */
 const DEFAULT_CONNECTION_TIMEOUT_MS = 3e4;
 /**
+* Checks if an error is a session-related error from an HTTP backend
+* (e.g., downstream server restarted and no longer recognizes the session ID).
+*/
+function isSessionError(error) {
+	const message = error instanceof Error ? error.message : String(error);
+	return message.includes("unknown session") || message.includes("Session not found");
+}
+/**
 * MCP Client wrapper for managing individual server connections
 * This is an internal class used by McpClientManagerService
 */
@@ -1437,6 +1445,7 @@ var McpClient = class {
 	client;
 	childProcess;
 	connected = false;
+	reconnectFn;
 	constructor(serverName, transport, client, config) {
 		this.serverName = serverName;
 		this.serverInstruction = config.instruction;
@@ -1452,34 +1461,76 @@ var McpClient = class {
 	setConnected(connected) {
 		this.connected = connected;
 	}
+	/**
+	* Sets a reconnection function that creates a fresh Client and transport.
+	* Called automatically by withSessionRetry when a session error is detected
+	* (e.g., downstream HTTP server restarted and the old session ID is invalid).
+	*/
+	setReconnectFn(fn) {
+		this.reconnectFn = fn;
+	}
+	/**
+	* Wraps an operation with automatic retry on session errors.
+	* If the operation fails with a session error (e.g., downstream server restarted),
+	* reconnects and retries once.
+	*/
+	async withSessionRetry(operation) {
+		try {
+			return await operation();
+		} catch (error) {
+			if (!this.reconnectFn || !isSessionError(error)) throw error;
+			console.error(`Session error for ${this.serverName}, reconnecting: ${error instanceof Error ? error.message : String(error)}`);
+			try {
+				await this.client.close();
+			} catch (closeError) {
+				console.error(`Failed to close stale client for ${this.serverName}:`, closeError);
+			}
+			const result = await this.reconnectFn();
+			this.client = result.client;
+			if (result.childProcess) this.childProcess = result.childProcess;
+			return await operation();
+		}
+	}
 	async listTools() {
 		if (!this.connected) throw new Error(`Client for ${this.serverName} is not connected`);
-		return (await this.client.listTools()).tools;
+		return this.withSessionRetry(async () => {
+			return (await this.client.listTools()).tools;
+		});
 	}
 	async listResources() {
 		if (!this.connected) throw new Error(`Client for ${this.serverName} is not connected`);
-		return (await this.client.listResources()).resources;
+		return this.withSessionRetry(async () => {
+			return (await this.client.listResources()).resources;
+		});
 	}
 	async listPrompts() {
 		if (!this.connected) throw new Error(`Client for ${this.serverName} is not connected`);
-		return (await this.client.listPrompts()).prompts;
+		return this.withSessionRetry(async () => {
+			return (await this.client.listPrompts()).prompts;
+		});
 	}
 	async callTool(name, args) {
 		if (!this.connected) throw new Error(`Client for ${this.serverName} is not connected`);
-		return await this.client.callTool({
-			name,
-			arguments: args
+		return this.withSessionRetry(async () => {
+			return await this.client.callTool({
+				name,
+				arguments: args
+			});
 		});
 	}
 	async readResource(uri) {
 		if (!this.connected) throw new Error(`Client for ${this.serverName} is not connected`);
-		return await this.client.readResource({ uri });
+		return this.withSessionRetry(async () => {
+			return await this.client.readResource({ uri });
+		});
 	}
 	async getPrompt(name, args) {
 		if (!this.connected) throw new Error(`Client for ${this.serverName} is not connected`);
-		return await this.client.getPrompt({
-			name,
-			arguments: args
+		return this.withSessionRetry(async () => {
+			return await this.client.getPrompt({
+				name,
+				arguments: args
+			});
 		});
 	}
 	async close() {
@@ -1560,6 +1611,20 @@ var McpClientManagerService = class {
 		try {
 			await Promise.race([this.performConnection(mcpClient, config), new Promise((_, reject) => setTimeout(() => reject(/* @__PURE__ */ new Error(`Connection timeout after ${timeoutMs}ms`)), timeoutMs))]);
 			mcpClient.setConnected(true);
+			if (config.transport === "http" || config.transport === "sse") mcpClient.setReconnectFn(async () => {
+				try {
+					const newClient = new Client({
+						name: "@agiflowai/one-mcp-client",
+						version: "0.1.0"
+					}, { capabilities: {} });
+					const newMcpClient = new McpClient(serverName, config.transport, newClient, {});
+					await this.performConnection(newMcpClient, config);
+					return { client: newClient };
+				} catch (error) {
+					console.error(`Failed to reconnect to ${serverName}:`, error);
+					throw error;
+				}
+			});
 			if (!mcpClient.serverInstruction) try {
 				const serverInstruction = mcpClient["client"].getInstructions();
 				if (serverInstruction) mcpClient.serverInstruction = serverInstruction;
@@ -2767,7 +2832,7 @@ IMPORTANT: Only use tools discovered from describe_tools with id="${this.serverI
 
 //#endregion
 //#region package.json
-var version = "0.3.15";
+var version = "0.3.16";
 
 //#endregion
 //#region src/server/index.ts

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "@agiflowai/one-mcp",
   "description": "One MCP server package",
-  "version": "0.3.16",
+  "version": "0.3.17",
   "license": "AGPL-3.0",
   "keywords": [
     "mcp",