From ec093e9b916b7b5123759b656ec639c4d10310d8 Mon Sep 17 00:00:00 2001
From: Chulki Lee <chulki.lee@gmail.com>
Date: Wed, 29 Oct 2025 12:42:53 -0700
Subject: [PATCH] add sessionIdGenerator config

---
 src/handler/mcp-api-handler.ts | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/src/handler/mcp-api-handler.ts b/src/handler/mcp-api-handler.ts
index 8859984..8ef1f24 100644
--- a/src/handler/mcp-api-handler.ts
+++ b/src/handler/mcp-api-handler.ts
@@ -125,6 +125,11 @@ export type Config = {
    * @default false
    */
   disableSse?: boolean;
+
+  /**
+   * sessionIdGenerator for the streamable HTTP transport
+   */
+  sessionIdGenerator?: undefined;
 };
 
 /**
@@ -247,6 +252,7 @@ export function initializeMcpApiHandler(
     maxDuration,
     verboseLogs,
     disableSse,
+    sessionIdGenerator,
   } = config;
 
   const {
@@ -272,7 +278,7 @@ export function initializeMcpApiHandler(
 
   let statelessServer: McpServer;
   const statelessTransport = new StreamableHTTPServerTransport({
-    sessionIdGenerator: undefined,
+    sessionIdGenerator: sessionIdGenerator,
   });
   
   // Start periodic cleanup if not already running