@context-engine-bridge/context-engine-mcp-bridge 0.0.3 → 0.0.4

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@context-engine-bridge/context-engine-mcp-bridge",
-  "version": "0.0.3",
+  "version": "0.0.4",
   "description": "Context Engine MCP bridge (http/stdio proxy combining indexer + memory servers)",
   "bin": {
     "ctxce": "bin/ctxce.js",

package/src/mcpServer.js

@@ -94,6 +94,22 @@ function withTimeout(promise, ms, label) {
   });
 }
 
+function getBridgeToolTimeoutMs() {
+  try {
+    const raw = process.env.CTXCE_TOOL_TIMEOUT_MSEC;
+    if (!raw) {
+      return 300000;
+    }
+    const parsed = Number.parseInt(String(raw), 10);
+    if (!Number.isFinite(parsed) || parsed <= 0) {
+      return 300000;
+    }
+    return parsed;
+  } catch {
+    return 300000;
+  }
+}
+
 function selectClientForTool(name, indexerClient, memoryClient) {
   if (!name) {
     return indexerClient;
@@ -300,10 +316,15 @@ async function createBridgeServer(options) {
       throw new Error(`Tool ${name} not available on any configured MCP server`);
     }
 
-    const result = await targetClient.callTool({
-      name,
-      arguments: args,
-    });
+    const timeoutMs = getBridgeToolTimeoutMs();
+    const result = await targetClient.callTool(
+      {
+        name,
+        arguments: args,
+      },
+      undefined,
+      { timeout: timeoutMs },
+    );
     return result;
   });
 
@@ -314,6 +335,27 @@ export async function runMcpServer(options) {
   const server = await createBridgeServer(options);
   const transport = new StdioServerTransport();
   await server.connect(transport);
+
+  const exitOnStdinClose = process.env.CTXCE_EXIT_ON_STDIN_CLOSE !== "0";
+  if (exitOnStdinClose) {
+    const handleStdioClosed = () => {
+      try {
+        debugLog("[ctxce] Stdio transport closed; exiting MCP bridge process.");
+      } catch {
+        // ignore
+      }
+      // Allow any in-flight logs to flush, then exit.
+      setTimeout(() => {
+        process.exit(0);
+      }, 10).unref();
+    };
+
+    if (process.stdin && typeof process.stdin.on === "function") {
+      process.stdin.on("end", handleStdioClosed);
+      process.stdin.on("close", handleStdioClosed);
+      process.stdin.on("error", handleStdioClosed);
+    }
+  }
 }
 
 export async function runHttpMcpServer(options) {