@tambo-ai/react 0.58.0 → 0.59.0

package/dist/mcp/mcp-client.d.ts

@@ -24,9 +24,9 @@ export declare enum MCPTransport {
  * });
  * ```
  */
-interface MCPHandlers {
-    elicitation?: (e: ElicitRequest) => Promise<ElicitResult>;
-    sampling?: (e: CreateMessageRequest) => Promise<CreateMessageResult>;
+export interface MCPHandlers {
+    elicitation: (e: ElicitRequest) => Promise<ElicitResult>;
+    sampling: (e: CreateMessageRequest) => Promise<CreateMessageResult>;
 }
 /**
  * A client for interacting with MCP (Model Context Protocol) servers.
@@ -39,7 +39,12 @@ interface MCPHandlers {
  * ```
  */
 export declare class MCPClient {
-    private client;
+    /**
+     * The underlying MCP client
+     *
+     * Be careful not to mutate the client directly, use the methods provided instead.
+     */
+    client: Client;
     private transport;
     private transportType;
     sessionId?: string;
@@ -47,42 +52,6 @@ export declare class MCPClient {
     private headers;
     private authProvider?;
     private handlers;
-    /**
-     * Tracks an in-flight reconnect so concurrent triggers coalesce
-     * (single-flight). When set, additional calls to `reconnect()` or
-     * the automatic `onclose` handler will await the same Promise instead of
-     * starting another reconnect sequence.
-     */
-    private reconnecting?;
-    /**
-     * Timer id for a scheduled automatic reconnect (used by `onclose`).
-     * Present only while waiting for the backoff delay to elapse.
-     */
-    private reconnectTimer?;
-    /**
-     * Count of consecutive automatic reconnect failures used to compute
-     * exponential backoff. Reset to 0 after a successful connection.
-     */
-    private backoffAttempts;
-    /**
-     * Backoff policy (discoverable constants)
-     * - BACKOFF_INITIAL_MS: initial delay for the first automatic retry
-     * - BACKOFF_MULTIPLIER: exponential growth factor for each failed attempt
-     * - BACKOFF_MAX_MS: upper bound for the delay
-     * - BACKOFF_JITTER_RATIO: jitter range as a fraction of the base delay
-     *
-     * Jitter is applied symmetrically in [-ratio, +ratio]. For example, with a
-     * 500ms base delay and 0.2 ratio, the actual delay is in [400ms, 600ms].
-     *
-     * The backoff applies only to automatic reconnects started from the
-     * `onclose` handler. Explicit/manual calls to `reconnect()` run immediately
-     * (no backoff), and will preempt any scheduled automatic attempt.
-     */
-    static readonly BACKOFF_INITIAL_MS = 500;
-    static readonly BACKOFF_MULTIPLIER = 2;
-    static readonly BACKOFF_MAX_MS = 30000;
-    static readonly BACKOFF_JITTER_RATIO = 0.2;
-    elicitation: EventTarget;
     /**
      * Private constructor to enforce using the static create method.
      * @param endpoint - The URL of the MCP server to connect to
@@ -103,43 +72,7 @@ export declare class MCPClient {
      * @returns A connected MCPClient instance ready for use
      * @throws {Error} Will throw an error if connection fails
      */
-    static create(endpoint: string, transportType: MCPTransport | undefined, headers: Record<string, string> | undefined, authProvider: OAuthClientProvider | undefined, sessionId: string | undefined, handlers?: MCPHandlers): Promise<MCPClient>;
-    /**
-     * Reconnects to the MCP server, optionally retaining the same session ID.
-     *
-     * Single‑flight semantics:
-     * - If a reconnect is already in progress (triggered either manually or by
-     * the automatic `onclose` handler), additional calls will await the
-     * in-flight reconnect rather than start another one.
-     * - If an automatic reconnect has been scheduled but not yet started (i.e.,
-     * we are waiting in a backoff delay), calling `reconnect()` manually will
-     * cancel the scheduled attempt and perform an immediate reconnect.
-     *
-     * Backoff policy:
-     * - Backoff delays with jitter are applied only for automatic reconnects
-     * (via `onclose`). Manual calls to `reconnect()` do not use backoff.
-     * @param newSession - Whether to create a new session (true) or reuse existing session ID (false)
-     * @param reportErrorOnClose - Whether to report errors when closing the client
-     * Note that only StreamableHTTPClientTransport supports session IDs.
-     * @returns A promise that resolves when the reconnect is complete
-     */
-    reconnect(newSession?: boolean, reportErrorOnClose?: boolean): Promise<void>;
-    /**
-     * Called by the underlying MCP SDK when the connection closes.
-     * Schedules an automatic reconnect with bounded exponential backoff and
-     * jitter. If a reconnect is already scheduled or running, this is a no-op.
-     */
-    private onclose;
-    /**
-     * Compute the next backoff delay with symmetric jitter.
-     * @returns The next backoff delay in milliseconds
-     */
-    private computeBackoffDelayMs;
-    /**
-     * Schedule an automatic reconnect attempt if one is not already scheduled
-     * or running. Uses the backoff policy and self-reschedules on failure.
-     */
-    private scheduleAutoReconnect;
+    static create(endpoint: string, transportType: MCPTransport | undefined, headers: Record<string, string> | undefined, authProvider: OAuthClientProvider | undefined, sessionId: string | undefined, handlers?: Partial<MCPHandlers>): Promise<MCPClient>;
     private initializeTransport;
     /**
      * Initializes the MCP client with the appropriate capabilities and handlers
@@ -199,9 +132,10 @@ export declare class MCPClient {
      * @returns The result from the tool execution
      * @throws {Error} Will throw an error if the tool call fails or if arguments are invalid
      */
-    callTool(name: string, args: Record<string, unknown>): Promise<MCPToolCallResult>;
+    callTool(name: string, args: Record<string, unknown>, _meta?: Record<string, unknown>): Promise<MCPToolCallResult>;
     updateElicitationHandler(handler: ((e: ElicitRequest) => Promise<ElicitResult>) | undefined): void;
     updateSamplingHandler(handler: ((e: CreateMessageRequest) => Promise<CreateMessageResult>) | undefined): void;
+    close(): Promise<void>;
 }
 /**
  * The result of a tool call.
@@ -213,5 +147,4 @@ export interface MCPToolSpec {
     description?: string;
     inputSchema?: JSONSchema7;
 }
-export {};
 //# sourceMappingURL=mcp-client.d.ts.map

package/dist/mcp/mcp-client.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"mcp-client.d.ts","sourceRoot":"","sources":["../../src/mcp/mcp-client.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,KAAK,mBAAmB,EAAE,MAAM,0CAA0C,CAAC;AACpF,OAAO,EAAE,MAAM,EAAE,MAAM,2CAA2C,CAAC;AAGnE,OAAO,EACL,oBAAoB,EAEpB,mBAAmB,EACnB,aAAa,EAEb,YAAY,EACb,MAAM,oCAAoC,CAAC;AAC5C,OAAO,EAAE,WAAW,EAAE,MAAM,aAAa,CAAC;AAE1C,oBAAY,YAAY;IACtB,GAAG,QAAQ;IACX,IAAI,SAAS;CACd;AAED;;;;;;;;;;;;;;;;;GAiBG;AACH,UAAU,WAAW;IACnB,WAAW,CAAC,EAAE,CAAC,CAAC,EAAE,aAAa,KAAK,OAAO,CAAC,YAAY,CAAC,CAAC;IAC1D,QAAQ,CAAC,EAAE,CAAC,CAAC,EAAE,oBAAoB,KAAK,OAAO,CAAC,mBAAmB,CAAC,CAAC;CACtE;AAED;;;;;;;;;GASG;AACH,qBAAa,SAAS;IACpB,OAAO,CAAC,MAAM,CAAS;IACvB,OAAO,CAAC,SAAS,CAAqD;IACtE,OAAO,CAAC,aAAa,CAAe;IAC7B,SAAS,CAAC,EAAE,MAAM,CAAC;IAC1B,OAAO,CAAC,QAAQ,CAAS;IACzB,OAAO,CAAC,OAAO,CAAyB;IACxC,OAAO,CAAC,YAAY,CAAC,CAAsB;IAC3C,OAAO,CAAC,QAAQ,CAAc;IAC9B;;;;;OAKG;IACH,OAAO,CAAC,YAAY,CAAC,CAAgB;IACrC;;;OAGG;IACH,OAAO,CAAC,cAAc,CAAC,CAAgC;IACvD;;;OAGG;IACH,OAAO,CAAC,eAAe,CAAK;IAE5B;;;;;;;;;;;;;OAaG;IACH,MAAM,CAAC,QAAQ,CAAC,kBAAkB,OAAO;IACzC,MAAM,CAAC,QAAQ,CAAC,kBAAkB,KAAK;IACvC,MAAM,CAAC,QAAQ,CAAC,cAAc,SAAU;IACxC,MAAM,CAAC,QAAQ,CAAC,oBAAoB,OAAO;IAEpC,WAAW,EAAE,WAAW,CAAqB;IACpD;;;;;OAKG;IACH,OAAO;IAiBP;;;;;;;;;;;;OAYG;WACU,MAAM,CACjB,QAAQ,EAAE,MAAM,EAChB,aAAa,EAAE,YAAY,YAAoB,EAC/C,OAAO,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,GAAG,SAAS,EAC3C,YAAY,EAAE,mBAAmB,GAAG,SAAS,EAC7C,SAAS,EAAE,MAAM,GAAG,SAAS,EAC7B,QAAQ,GAAE,WAAgB,GACzB,OAAO,CAAC,SAAS,CAAC;IAerB;;;;;;;;;;;;;;;;;;OAkBG;IACG,SAAS,CAAC,UAAU,UAAQ,EAAE,kBAAkB,UAAO;IAuD7D;;;;OAIG;IACH,OAAO,CAAC,OAAO;IAIf;;;OAGG;IACH,OAAO,CAAC,qBAAqB;IAY7B;;;OAGG;IACH,OAAO,CAAC,qBAAqB;IAoC7B,OAAO,CAAC,mBAAmB;IAe3B;;;OAGG;IACH,OAAO,CAAC,gBAAgB;IA8BxB;;;;;OAKG;IACG,SAAS,IAAI,OAAO,CAAC,WAAW,EAAE,CAAC;IAmCzC,qBAAqB;;;;;;;;;;;;;;;;;;;;;;;;;IAIrB,gBAAgB;;;;;;;;;;;;;IAIhB,eAAe;IAIf;;;;;;OAMG;IACG,QAAQ,CACZ,IAAI,EAAE,MAAM,EACZ,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAC5B,OAAO,CAAC,iBAAiB,CAAC;IAQ7B,wBAAwB,CACtB,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,aAAa,KAAK,OAAO,CAAC,YAAY,CAAC,CAAC,GAAG,SAAS;IAmBpE,qBAAqB,CACnB,OAAO,EACH,CAAC,CAAC,CAAC,EAAE,oBAAoB,KAAK,OAAO,CAAC,mBAAmB,CAAC,CAAC,GAC3D,SAAS;CAkBhB;AAED;;;GAGG;AACH,MAAM,MAAM,iBAAiB,GAAG,OAAO,CACrC,UAAU,CAAC,OAAO,MAAM,CAAC,SAAS,CAAC,QAAQ,CAAC,CAC7C,CAAC;AASF,MAAM,WAAW,WAAW;IAC1B,IAAI,EAAE,MAAM,CAAC;IACb,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,WAAW,CAAC,EAAE,WAAW,CAAC;CAC3B"}
+{"version":3,"file":"mcp-client.d.ts","sourceRoot":"","sources":["../../src/mcp/mcp-client.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,KAAK,mBAAmB,EAAE,MAAM,0CAA0C,CAAC;AACpF,OAAO,EAAE,MAAM,EAAE,MAAM,2CAA2C,CAAC;AAGnE,OAAO,EACL,oBAAoB,EAEpB,mBAAmB,EACnB,aAAa,EAEb,YAAY,EACb,MAAM,oCAAoC,CAAC;AAC5C,OAAO,EAAE,WAAW,EAAE,MAAM,aAAa,CAAC;AAE1C,oBAAY,YAAY;IACtB,GAAG,QAAQ;IACX,IAAI,SAAS;CACd;AAED;;;;;;;;;;;;;;;;;GAiBG;AACH,MAAM,WAAW,WAAW;IAC1B,WAAW,EAAE,CAAC,CAAC,EAAE,aAAa,KAAK,OAAO,CAAC,YAAY,CAAC,CAAC;IACzD,QAAQ,EAAE,CAAC,CAAC,EAAE,oBAAoB,KAAK,OAAO,CAAC,mBAAmB,CAAC,CAAC;CACrE;AAED;;;;;;;;;GASG;AACH,qBAAa,SAAS;IACpB;;;;OAIG;IACH,MAAM,EAAE,MAAM,CAAC;IACf,OAAO,CAAC,SAAS,CAAqD;IACtE,OAAO,CAAC,aAAa,CAAe;IAC7B,SAAS,CAAC,EAAE,MAAM,CAAC;IAC1B,OAAO,CAAC,QAAQ,CAAS;IACzB,OAAO,CAAC,OAAO,CAAyB;IACxC,OAAO,CAAC,YAAY,CAAC,CAAsB;IAC3C,OAAO,CAAC,QAAQ,CAAuB;IAEvC;;;;;OAKG;IACH,OAAO;IAiBP;;;;;;;;;;;;OAYG;WACU,MAAM,CACjB,QAAQ,EAAE,MAAM,EAChB,aAAa,EAAE,YAAY,YAAoB,EAC/C,OAAO,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,GAAG,SAAS,EAC3C,YAAY,EAAE,mBAAmB,GAAG,SAAS,EAC7C,SAAS,EAAE,MAAM,GAAG,SAAS,EAC7B,QAAQ,GAAE,OAAO,CAAC,WAAW,CAAM,GAClC,OAAO,CAAC,SAAS,CAAC;IAgBrB,OAAO,CAAC,mBAAmB;IAe3B;;;OAGG;IACH,OAAO,CAAC,gBAAgB;IA8BxB;;;;;OAKG;IACG,SAAS,IAAI,OAAO,CAAC,WAAW,EAAE,CAAC;IAmCzC,qBAAqB;;;;;;;;;;;;;;;;;;;;;;;;;IAIrB,gBAAgB;;;;;;;;;;;;;IAIhB,eAAe;IAIf;;;;;;OAMG;IACG,QAAQ,CACZ,IAAI,EAAE,MAAM,EACZ,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EAC7B,KAAK,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAC9B,OAAO,CAAC,iBAAiB,CAAC;IAS7B,wBAAwB,CACtB,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,aAAa,KAAK,OAAO,CAAC,YAAY,CAAC,CAAC,GAAG,SAAS;IAwBpE,qBAAqB,CACnB,OAAO,EACH,CAAC,CAAC,CAAC,EAAE,oBAAoB,KAAK,OAAO,CAAC,mBAAmB,CAAC,CAAC,GAC3D,SAAS;IAwBT,KAAK;CAOZ;AAED;;;GAGG;AACH,MAAM,MAAM,iBAAiB,GAAG,OAAO,CACrC,UAAU,CAAC,OAAO,MAAM,CAAC,SAAS,CAAC,QAAQ,CAAC,CAC7C,CAAC;AASF,MAAM,WAAW,WAAW;IAC1B,IAAI,EAAE,MAAM,CAAC;IACb,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,WAAW,CAAC,EAAE,WAAW,CAAC;CAC3B"}

package/dist/mcp/mcp-client.js

@@ -21,6 +21,11 @@ var MCPTransport;
  * ```
  */
 class MCPClient {
+    /**
+     * The underlying MCP client
+     *
+     * Be careful not to mutate the client directly, use the methods provided instead.
+     */
     client;
     transport;
     transportType;
@@ -29,42 +34,6 @@ class MCPClient {
     headers;
     authProvider;
     handlers;
-    /**
-     * Tracks an in-flight reconnect so concurrent triggers coalesce
-     * (single-flight). When set, additional calls to `reconnect()` or
-     * the automatic `onclose` handler will await the same Promise instead of
-     * starting another reconnect sequence.
-     */
-    reconnecting;
-    /**
-     * Timer id for a scheduled automatic reconnect (used by `onclose`).
-     * Present only while waiting for the backoff delay to elapse.
-     */
-    reconnectTimer;
-    /**
-     * Count of consecutive automatic reconnect failures used to compute
-     * exponential backoff. Reset to 0 after a successful connection.
-     */
-    backoffAttempts = 0;
-    /**
-     * Backoff policy (discoverable constants)
-     * - BACKOFF_INITIAL_MS: initial delay for the first automatic retry
-     * - BACKOFF_MULTIPLIER: exponential growth factor for each failed attempt
-     * - BACKOFF_MAX_MS: upper bound for the delay
-     * - BACKOFF_JITTER_RATIO: jitter range as a fraction of the base delay
-     *
-     * Jitter is applied symmetrically in [-ratio, +ratio]. For example, with a
-     * 500ms base delay and 0.2 ratio, the actual delay is in [400ms, 600ms].
-     *
-     * The backoff applies only to automatic reconnects started from the
-     * `onclose` handler. Explicit/manual calls to `reconnect()` run immediately
-     * (no backoff), and will preempt any scheduled automatic attempt.
-     */
-    static BACKOFF_INITIAL_MS = 500;
-    static BACKOFF_MULTIPLIER = 2;
-    static BACKOFF_MAX_MS = 30_000;
-    static BACKOFF_JITTER_RATIO = 0.2;
-    elicitation = new EventTarget();
     /**
      * Private constructor to enforce using the static create method.
      * @param endpoint - The URL of the MCP server to connect to
@@ -101,127 +70,6 @@ class MCPClient {
         }
         return mcpClient;
     }
-    /**
-     * Reconnects to the MCP server, optionally retaining the same session ID.
-     *
-     * Single‑flight semantics:
-     * - If a reconnect is already in progress (triggered either manually or by
-     * the automatic `onclose` handler), additional calls will await the
-     * in-flight reconnect rather than start another one.
-     * - If an automatic reconnect has been scheduled but not yet started (i.e.,
-     * we are waiting in a backoff delay), calling `reconnect()` manually will
-     * cancel the scheduled attempt and perform an immediate reconnect.
-     *
-     * Backoff policy:
-     * - Backoff delays with jitter are applied only for automatic reconnects
-     * (via `onclose`). Manual calls to `reconnect()` do not use backoff.
-     * @param newSession - Whether to create a new session (true) or reuse existing session ID (false)
-     * @param reportErrorOnClose - Whether to report errors when closing the client
-     * Note that only StreamableHTTPClientTransport supports session IDs.
-     * @returns A promise that resolves when the reconnect is complete
-     */
-    async reconnect(newSession = false, reportErrorOnClose = true) {
-        // If a reconnect is already running, coalesce into it.
-        if (this.reconnecting) {
-            return await this.reconnecting;
-        }
-        // Manual reconnect preempts any scheduled automatic attempt.
-        if (this.reconnectTimer) {
-            clearTimeout(this.reconnectTimer);
-            this.reconnectTimer = undefined;
-        }
-        const doReconnect = async () => {
-            const sessionId = newSession ? undefined : this.sessionId;
-            // Prevent re-entrant onclose during deliberate close by detaching
-            // the handler from the previous client instance.
-            const prevClient = this.client;
-            // Prevent re-entrant onclose callbacks from the previous client
-            prevClient.onclose = undefined;
-            try {
-                await prevClient.close();
-            }
-            catch (error) {
-                if (reportErrorOnClose) {
-                    console.error("Error closing Tambo MCP Client:", error);
-                }
-            }
-            this.transport = this.initializeTransport(sessionId);
-            this.client = this.initializeClient();
-            await this.client.connect(this.transport);
-            // We may have gotten a session id from the server, so we need to set it
-            if ("sessionId" in this.transport) {
-                this.sessionId = this.transport.sessionId;
-                if (sessionId !== this.sessionId) {
-                    // This is a pretty unusual thing to happen, but it might be possible?
-                    console.warn("Session id mismatch", sessionId, this.sessionId);
-                }
-            }
-        };
-        this.reconnecting = (async () => {
-            try {
-                await doReconnect();
-                // Successful manual reconnect: reset backoff.
-                this.backoffAttempts = 0;
-            }
-            finally {
-                this.reconnecting = undefined;
-            }
-        })();
-        return await this.reconnecting;
-    }
-    /**
-     * Called by the underlying MCP SDK when the connection closes.
-     * Schedules an automatic reconnect with bounded exponential backoff and
-     * jitter. If a reconnect is already scheduled or running, this is a no-op.
-     */
-    onclose() {
-        this.scheduleAutoReconnect();
-    }
-    /**
-     * Compute the next backoff delay with symmetric jitter.
-     * @returns The next backoff delay in milliseconds
-     */
-    computeBackoffDelayMs() {
-        const base = Math.min(MCPClient.BACKOFF_MAX_MS, MCPClient.BACKOFF_INITIAL_MS *
-            Math.pow(MCPClient.BACKOFF_MULTIPLIER, this.backoffAttempts));
-        const jitterRange = MCPClient.BACKOFF_JITTER_RATIO * base;
-        const jitter = (Math.random() * 2 - 1) * jitterRange; // [-range, +range]
-        const ms = Math.max(0, Math.round(base + jitter));
-        return ms;
-    }
-    /**
-     * Schedule an automatic reconnect attempt if one is not already scheduled
-     * or running. Uses the backoff policy and self-reschedules on failure.
-     */
-    scheduleAutoReconnect() {
-        if (this.reconnecting || this.reconnectTimer) {
-            return;
-        }
-        const delayMs = this.computeBackoffDelayMs();
-        console.warn("Tambo MCP Client closed; attempting automatic reconnect in", `${delayMs}ms`);
-        this.reconnectTimer = setTimeout(async () => {
-            this.reconnectTimer = undefined;
-            // Start the actual reconnect (single-flight)
-            const inFlight = (this.reconnecting = this.reconnect(false, false));
-            try {
-                await inFlight;
-                // Success: reset attempts
-                this.backoffAttempts = 0;
-            }
-            catch (err) {
-                // Failure: increase attempts; scheduling occurs in finally below so the
-                // new timer isn't blocked by `this.reconnecting` being truthy.
-                this.backoffAttempts += 1;
-                console.warn("Automatic reconnect failed; will retry with backoff.", err);
-            }
-            finally {
-                this.reconnecting = undefined;
-                if (this.backoffAttempts > 0) {
-                    this.scheduleAutoReconnect();
-                }
-            }
-        }, delayMs);
-    }
     initializeTransport(sessionId) {
         if (this.transportType === MCPTransport.SSE) {
             return new sse_js_1.SSEClientTransport(new URL(this.endpoint), {
@@ -255,7 +103,6 @@ class MCPClient {
                 ...samplingCapability,
             },
         });
-        client.onclose = this.onclose.bind(this);
         if (this.handlers.elicitation) {
             client.setRequestHandler(types_js_1.ElicitRequestSchema, this.handlers.elicitation);
         }
@@ -313,14 +160,19 @@ class MCPClient {
      * @returns The result from the tool execution
      * @throws {Error} Will throw an error if the tool call fails or if arguments are invalid
      */
-    async callTool(name, args) {
+    async callTool(name, args, _meta) {
         const result = await this.client.callTool({
             name,
             arguments: args,
+            _meta,
         });
         return result;
     }
     updateElicitationHandler(handler) {
+        // Skip if handler hasn't changed
+        if (handler === this.handlers.elicitation) {
+            return;
+        }
         // Because we advertise the elicitation capability on initial connection, we can only update
         // an existing handler, not add it if we haven't set it yet.
         if (handler && !this.handlers.elicitation) {
@@ -338,6 +190,10 @@ class MCPClient {
         this.client.setRequestHandler(types_js_1.ElicitRequestSchema, handler);
     }
     updateSamplingHandler(handler) {
+        // Skip if handler hasn't changed
+        if (handler === this.handlers.sampling) {
+            return;
+        }
         // Because we advertise the sampling capability on initial connection, we can only update
         // an existing handler, not add it if we haven't set it yet.
         if (handler && !this.handlers.sampling) {
@@ -354,6 +210,13 @@ class MCPClient {
         }
         this.client.setRequestHandler(types_js_1.CreateMessageRequestSchema, handler);
     }
+    async close() {
+        // Not really sure which one of these to close first, but we'll close the
+        // transport first so that no requests can come in and hit closing/closed
+        // clients
+        await this.transport.close();
+        await this.client.close();
+    }
 }
 exports.MCPClient = MCPClient;
 //# sourceMappingURL=mcp-client.js.map

package/dist/mcp/mcp-client.js.map

@@ -1 +1 @@
-{"version":3,"file":"mcp-client.js","sourceRoot":"","sources":["../../src/mcp/mcp-client.ts"],"names":[],"mappings":";;;AACA,wEAAmE;AACnE,oEAA6E;AAC7E,0FAAmG;AACnG,iEAO4C;AAG5C,IAAY,YAGX;AAHD,WAAY,YAAY;IACtB,2BAAW,CAAA;IACX,6BAAa,CAAA;AACf,CAAC,EAHW,YAAY,4BAAZ,YAAY,QAGvB;AAyBD;;;;;;;;;GASG;AACH,MAAa,SAAS;IACZ,MAAM,CAAS;IACf,SAAS,CAAqD;IAC9D,aAAa,CAAe;IAC7B,SAAS,CAAU;IAClB,QAAQ,CAAS;IACjB,OAAO,CAAyB;IAChC,YAAY,CAAuB;IACnC,QAAQ,CAAc;IAC9B;;;;;OAKG;IACK,YAAY,CAAiB;IACrC;;;OAGG;IACK,cAAc,CAAiC;IACvD;;;OAGG;IACK,eAAe,GAAG,CAAC,CAAC;IAE5B;;;;;;;;;;;;;OAaG;IACH,MAAM,CAAU,kBAAkB,GAAG,GAAG,CAAC;IACzC,MAAM,CAAU,kBAAkB,GAAG,CAAC,CAAC;IACvC,MAAM,CAAU,cAAc,GAAG,MAAM,CAAC;IACxC,MAAM,CAAU,oBAAoB,GAAG,GAAG,CAAC;IAEpC,WAAW,GAAgB,IAAI,WAAW,EAAE,CAAC;IACpD;;;;;OAKG;IACH,YACE,QAAgB,EAChB,aAA2B,EAC3B,OAAgC,EAChC,YAAkC,EAClC,SAAkB,EAClB,WAAwB,EAAE;QAE1B,IAAI,CAAC,QAAQ,GAAG,QAAQ,CAAC;QACzB,IAAI,CAAC,OAAO,GAAG,OAAO,IAAI,EAAE,CAAC;QAC7B,IAAI,CAAC,YAAY,GAAG,YAAY,CAAC;QACjC,IAAI,CAAC,aAAa,GAAG,aAAa,CAAC;QACnC,IAAI,CAAC,QAAQ,GAAG,QAAQ,CAAC;QACzB,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC,mBAAmB,CAAC,SAAS,CAAC,CAAC;QACrD,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC,gBAAgB,EAAE,CAAC;IACxC,CAAC;IAED;;;;;;;;;;;;OAYG;IACH,MAAM,CAAC,KAAK,CAAC,MAAM,CACjB,QAAgB,EAChB,gBAA8B,YAAY,CAAC,IAAI,EAC/C,OAA2C,EAC3C,YAA6C,EAC7C,SAA6B,EAC7B,WAAwB,EAAE;QAE1B,MAAM,SAAS,GAAG,IAAI,SAAS,CAC7B,QAAQ,EACR,aAAa,EACb,OAAO,EACP,YAAY,EACZ,SAAS,EACT,QAAQ,CACT,CAAC;QACF,MAAM,SAAS,CAAC,MAAM,CAAC,OAAO,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC;QACpD,IAAI,WAAW,IAAI,SAAS,CAAC,SAAS,EAAE,CAAC;YACvC,SAAS,CAAC,SAAS,GAAG,SAAS,CAAC,SAAS,CAAC,SAAS,CAAC;QACtD,CAAC;QACD,OAAO,SAAS,CAAC;IACnB,CAAC;IACD;;;;;;;;;;;;;;;;;;OAkBG;IACH,KAAK,CAAC,SAAS,CAAC,UAAU,GAAG,KAAK,EAAE,kBAAkB,GAAG,IAAI;QAC3D,uDAAuD;QACvD,IAAI,IAAI,CAAC,YAAY,EAAE,CAAC;YACtB,OAAO,MAAM,IAAI,CAAC,YAAY,CAAC;QACjC,CAAC;QAED,6DAA6D;QAC7D,IAAI,IAAI,CAAC,cAAc,EAAE,CAAC;YACxB,YAAY,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;YAClC,IAAI,CAAC,cAAc,GAAG,SAAS,CAAC;QAClC,CAAC;QAED,MAAM,WAAW,GAAG,KAAK,IAAI,EAAE;YAC7B,MAAM,SAAS,GAAG,UAAU,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC;YAE1D,kEAAkE;YAClE,iDAAiD;YACjD,MAAM,UAAU,GAAG,IAAI,CAAC,MAAM,CAAC;YAC/B,gEAAgE;YAChE,UAAU,CAAC,OAAO,GAAG,SAAS,CAAC;YAE/B,IAAI,CAAC;gBACH,MAAM,UAAU,CAAC,KAAK,EAAE,CAAC;YAC3B,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,IAAI,kBAAkB,EAAE,CAAC;oBACvB,OAAO,CAAC,KAAK,CAAC,iCAAiC,EAAE,KAAK,CAAC,CAAC;gBAC1D,CAAC;YACH,CAAC;YAED,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC,mBAAmB,CAAC,SAAS,CAAC,CAAC;YACrD,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC,gBAAgB,EAAE,CAAC;YACtC,MAAM,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;YAC1C,wEAAwE;YACxE,IAAI,WAAW,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC;gBAClC,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC,SAAS,CAAC,SAAS,CAAC;gBAC1C,IAAI,SAAS,KAAK,IAAI,CAAC,SAAS,EAAE,CAAC;oBACjC,sEAAsE;oBACtE,OAAO,CAAC,IAAI,CAAC,qBAAqB,EAAE,SAAS,EAAE,IAAI,CAAC,SAAS,CAAC,CAAC;gBACjE,CAAC;YACH,CAAC;QACH,CAAC,CAAC;QAEF,IAAI,CAAC,YAAY,GAAG,CAAC,KAAK,IAAI,EAAE;YAC9B,IAAI,CAAC;gBACH,MAAM,WAAW,EAAE,CAAC;gBACpB,8CAA8C;gBAC9C,IAAI,CAAC,eAAe,GAAG,CAAC,CAAC;YAC3B,CAAC;oBAAS,CAAC;gBACT,IAAI,CAAC,YAAY,GAAG,SAAS,CAAC;YAChC,CAAC;QACH,CAAC,CAAC,EAAE,CAAC;QAEL,OAAO,MAAM,IAAI,CAAC,YAAY,CAAC;IACjC,CAAC;IAED;;;;OAIG;IACK,OAAO;QACb,IAAI,CAAC,qBAAqB,EAAE,CAAC;IAC/B,CAAC;IAED;;;OAGG;IACK,qBAAqB;QAC3B,MAAM,IAAI,GAAG,IAAI,CAAC,GAAG,CACnB,SAAS,CAAC,cAAc,EACxB,SAAS,CAAC,kBAAkB;YAC1B,IAAI,CAAC,GAAG,CAAC,SAAS,CAAC,kBAAkB,EAAE,IAAI,CAAC,eAAe,CAAC,CAC/D,CAAC;QACF,MAAM,WAAW,GAAG,SAAS,CAAC,oBAAoB,GAAG,IAAI,CAAC;QAC1D,MAAM,MAAM,GAAG,CAAC,IAAI,CAAC,MAAM,EAAE,GAAG,CAAC,GAAG,CAAC,CAAC,GAAG,WAAW,CAAC,CAAC,mBAAmB;QACzE,MAAM,EAAE,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,KAAK,CAAC,IAAI,GAAG,MAAM,CAAC,CAAC,CAAC;QAClD,OAAO,EAAE,CAAC;IACZ,CAAC;IAED;;;OAGG;IACK,qBAAqB;QAC3B,IAAI,IAAI,CAAC,YAAY,IAAI,IAAI,CAAC,cAAc,EAAE,CAAC;YAC7C,OAAO;QACT,CAAC;QAED,MAAM,OAAO,GAAG,IAAI,CAAC,qBAAqB,EAAE,CAAC;QAC7C,OAAO,CAAC,IAAI,CACV,4DAA4D,EAC5D,GAAG,OAAO,IAAI,CACf,CAAC;QAEF,IAAI,CAAC,cAAc,GAAG,UAAU,CAAC,KAAK,IAAI,EAAE;YAC1C,IAAI,CAAC,cAAc,GAAG,SAAS,CAAC;YAChC,6CAA6C;YAC7C,MAAM,QAAQ,GAAG,CAAC,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC,SAAS,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC,CAAC;YACpE,IAAI,CAAC;gBACH,MAAM,QAAQ,CAAC;gBACf,0BAA0B;gBAC1B,IAAI,CAAC,eAAe,GAAG,CAAC,CAAC;YAC3B,CAAC;YAAC,OAAO,GAAG,EAAE,CAAC;gBACb,wEAAwE;gBACxE,+DAA+D;gBAC/D,IAAI,CAAC,eAAe,IAAI,CAAC,CAAC;gBAC1B,OAAO,CAAC,IAAI,CACV,sDAAsD,EACtD,GAAG,CACJ,CAAC;YACJ,CAAC;oBAAS,CAAC;gBACT,IAAI,CAAC,YAAY,GAAG,SAAS,CAAC;gBAC9B,IAAI,IAAI,CAAC,eAAe,GAAG,CAAC,EAAE,CAAC;oBAC7B,IAAI,CAAC,qBAAqB,EAAE,CAAC;gBAC/B,CAAC;YACH,CAAC;QACH,CAAC,EAAE,OAAO,CAAC,CAAC;IACd,CAAC;IAEO,mBAAmB,CAAC,SAA6B;QACvD,IAAI,IAAI,CAAC,aAAa,KAAK,YAAY,CAAC,GAAG,EAAE,CAAC;YAC5C,OAAO,IAAI,2BAAkB,CAAC,IAAI,GAAG,CAAC,IAAI,CAAC,QAAQ,CAAC,EAAE;gBACpD,YAAY,EAAE,IAAI,CAAC,YAAY;gBAC/B,WAAW,EAAE,EAAE,OAAO,EAAE,IAAI,CAAC,OAAO,EAAE;aACvC,CAAC,CAAC;QACL,CAAC;aAAM,CAAC;YACN,OAAO,IAAI,iDAA6B,CAAC,IAAI,GAAG,CAAC,IAAI,CAAC,QAAQ,CAAC,EAAE;gBAC/D,YAAY,EAAE,IAAI,CAAC,YAAY;gBAC/B,WAAW,EAAE,EAAE,OAAO,EAAE,IAAI,CAAC,OAAO,EAAE;gBACtC,SAAS;aACV,CAAC,CAAC;QACL,CAAC;IACH,CAAC;IAED;;;OAGG;IACK,gBAAgB;QACtB,MAAM,qBAAqB,GAAG,IAAI,CAAC,QAAQ,CAAC,WAAW;YACrD,CAAC,CAAC,EAAE,WAAW,EAAE,EAAE,EAAE;YACrB,CAAC,CAAC,EAAE,CAAC;QACP,MAAM,kBAAkB,GAAG,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC,CAAC,EAAE,QAAQ,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;QAC1E,MAAM,MAAM,GAAG,IAAI,iBAAM,CACvB;YACE,IAAI,EAAE,kBAAkB;YACxB,OAAO,EAAE,OAAO;SACjB,EACD;YACE,YAAY,EAAE;gBACZ,GAAG,qBAAqB;gBACxB,GAAG,kBAAkB;aACtB;SACF,CACF,CAAC;QACF,MAAM,CAAC,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACzC,IAAI,IAAI,CAAC,QAAQ,CAAC,WAAW,EAAE,CAAC;YAC9B,MAAM,CAAC,iBAAiB,CAAC,8BAAmB,EAAE,IAAI,CAAC,QAAQ,CAAC,WAAW,CAAC,CAAC;QAC3E,CAAC;QACD,IAAI,IAAI,CAAC,QAAQ,CAAC,QAAQ,EAAE,CAAC;YAC3B,MAAM,CAAC,iBAAiB,CACtB,qCAA0B,EAC1B,IAAI,CAAC,QAAQ,CAAC,QAAQ,CACvB,CAAC;QACJ,CAAC;QACD,OAAO,MAAM,CAAC;IAChB,CAAC;IAED;;;;;OAKG;IACH,KAAK,CAAC,SAAS;QACb,MAAM,QAAQ,GAAkB,EAAE,CAAC;QACnC,IAAI,OAAO,GAAG,IAAI,CAAC;QACnB,IAAI,MAAM,GAAuB,SAAS,CAAC;QAE3C,OAAO,OAAO,EAAE,CAAC;YACf,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,EAAE,MAAM,EAAE,EAAE,EAAE,CAAC,CAAC;YAC7D,QAAQ,CAAC,IAAI,CACX,GAAG,QAAQ,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAe,EAAE;gBAC1C,mCAAmC;gBACnC,MAAM,eAAe,GAAW,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC;gBACtD,IAAI,eAAe,KAAK,QAAQ,EAAE,CAAC;oBACjC,MAAM,IAAI,KAAK,CACb,yBAAyB,IAAI,CAAC,IAAI,mBAAmB,CACtD,CAAC;gBACJ,CAAC;gBAED,OAAO;oBACL,IAAI,EAAE,IAAI,CAAC,IAAI;oBACf,WAAW,EAAE,IAAI,CAAC,WAAW;oBAC7B,WAAW,EAAE,IAAI,CAAC,WAA0B;iBAC7C,CAAC;YACJ,CAAC,CAAC,CACH,CAAC;YAEF,IAAI,QAAQ,CAAC,UAAU,EAAE,CAAC;gBACxB,MAAM,GAAG,QAAQ,CAAC,UAAU,CAAC;YAC/B,CAAC;iBAAM,CAAC;gBACN,OAAO,GAAG,KAAK,CAAC;YAClB,CAAC;QACH,CAAC;QAED,OAAO,QAAQ,CAAC;IAClB,CAAC;IAED,qBAAqB;QACnB,OAAO,IAAI,CAAC,MAAM,CAAC,qBAAqB,EAAE,CAAC;IAC7C,CAAC;IAED,gBAAgB;QACd,OAAO,IAAI,CAAC,MAAM,CAAC,gBAAgB,EAAE,CAAC;IACxC,CAAC;IAED,eAAe;QACb,OAAO,IAAI,CAAC,MAAM,CAAC,eAAe,EAAE,CAAC;IACvC,CAAC;IAED;;;;;;OAMG;IACH,KAAK,CAAC,QAAQ,CACZ,IAAY,EACZ,IAA6B;QAE7B,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC;YACxC,IAAI;YACJ,SAAS,EAAE,IAAI;SAChB,CAAC,CAAC;QACH,OAAO,MAAM,CAAC;IAChB,CAAC;IAED,wBAAwB,CACtB,OAAkE;QAElE,4FAA4F;QAC5F,4DAA4D;QAC5D,IAAI,OAAO,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,WAAW,EAAE,CAAC;YAC1C,MAAM,IAAI,KAAK,CAAC,2CAA2C,CAAC,CAAC;QAC/D,CAAC;QACD,IAAI,CAAC,QAAQ,GAAG;YACd,GAAG,IAAI,CAAC,QAAQ;YAChB,WAAW,EAAE,OAAO;SACrB,CAAC;QACF,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,MAAM,MAAM,GAAG,8BAAmB,CAAC,KAAK,CAAC,MAAM,CAAC,KAAK,CAAC;YACtD,IAAI,CAAC,MAAM,CAAC,oBAAoB,CAAC,MAAM,CAAC,CAAC;YACzC,OAAO;QACT,CAAC;QACD,IAAI,CAAC,MAAM,CAAC,iBAAiB,CAAC,8BAAmB,EAAE,OAAO,CAAC,CAAC;IAC9D,CAAC;IAED,qBAAqB,CACnB,OAEa;QAEb,yFAAyF;QACzF,4DAA4D;QAC5D,IAAI,OAAO,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,QAAQ,EAAE,CAAC;YACvC,MAAM,IAAI,KAAK,CAAC,wCAAwC,CAAC,CAAC;QAC5D,CAAC;QACD,IAAI,CAAC,QAAQ,GAAG;YACd,GAAG,IAAI,CAAC,QAAQ;YAChB,QAAQ,EAAE,OAAO;SAClB,CAAC;QACF,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,MAAM,MAAM,GAAG,qCAA0B,CAAC,KAAK,CAAC,MAAM,CAAC,KAAK,CAAC;YAC7D,IAAI,CAAC,MAAM,CAAC,oBAAoB,CAAC,MAAM,CAAC,CAAC;YACzC,OAAO;QACT,CAAC;QACD,IAAI,CAAC,MAAM,CAAC,iBAAiB,CAAC,qCAA0B,EAAE,OAAO,CAAC,CAAC;IACrE,CAAC;;AApZH,8BAqZC","sourcesContent":["import { type OAuthClientProvider } from \"@modelcontextprotocol/sdk/client/auth.js\";\nimport { Client } from \"@modelcontextprotocol/sdk/client/index.js\";\nimport { SSEClientTransport } from \"@modelcontextprotocol/sdk/client/sse.js\";\nimport { StreamableHTTPClientTransport } from \"@modelcontextprotocol/sdk/client/streamableHttp.js\";\nimport {\n  CreateMessageRequest,\n  CreateMessageRequestSchema,\n  CreateMessageResult,\n  ElicitRequest,\n  ElicitRequestSchema,\n  ElicitResult,\n} from \"@modelcontextprotocol/sdk/types.js\";\nimport { JSONSchema7 } from \"json-schema\";\n\nexport enum MCPTransport {\n  SSE = \"sse\",\n  HTTP = \"http\",\n}\n\n/**\n * Handlers for MCP requests - these are only used if the server supports the corresponding capabilities\n * @param elicitation - Handler for elicitation requests\n * @param sampling - Handler for sampling requests\n * @example\n * ```typescript\n * const mcp = await MCPClient.create(\n *     'https://api.example.com/mcp',\n *     MCPTransport.HTTP,\n *     {},\n *     undefined,\n *     undefined,\n *     {\n *       elicitation: (e: ElicitRequest) => Promise.resolve({...}),\n *     },\n * });\n * ```\n */\ninterface MCPHandlers {\n  elicitation?: (e: ElicitRequest) => Promise<ElicitResult>;\n  sampling?: (e: CreateMessageRequest) => Promise<CreateMessageResult>;\n}\n\n/**\n * A client for interacting with MCP (Model Context Protocol) servers.\n * Provides a simple interface for listing and calling tools exposed by the server.\n * @example\n * ```typescript\n * const mcp = await MCPClient.create('https://api.example.com/mcp');\n * const tools = await mcp.listTools();\n * const result = await mcp.callTool('toolName', { arg1: 'value1' });\n * ```\n */\nexport class MCPClient {\n  private client: Client;\n  private transport: SSEClientTransport | StreamableHTTPClientTransport;\n  private transportType: MCPTransport;\n  public sessionId?: string;\n  private endpoint: string;\n  private headers: Record<string, string>;\n  private authProvider?: OAuthClientProvider;\n  private handlers: MCPHandlers;\n  /**\n   * Tracks an in-flight reconnect so concurrent triggers coalesce\n   * (single-flight). When set, additional calls to `reconnect()` or\n   * the automatic `onclose` handler will await the same Promise instead of\n   * starting another reconnect sequence.\n   */\n  private reconnecting?: Promise<void>;\n  /**\n   * Timer id for a scheduled automatic reconnect (used by `onclose`).\n   * Present only while waiting for the backoff delay to elapse.\n   */\n  private reconnectTimer?: ReturnType<typeof setTimeout>;\n  /**\n   * Count of consecutive automatic reconnect failures used to compute\n   * exponential backoff. Reset to 0 after a successful connection.\n   */\n  private backoffAttempts = 0;\n\n  /**\n   * Backoff policy (discoverable constants)\n   * - BACKOFF_INITIAL_MS: initial delay for the first automatic retry\n   * - BACKOFF_MULTIPLIER: exponential growth factor for each failed attempt\n   * - BACKOFF_MAX_MS: upper bound for the delay\n   * - BACKOFF_JITTER_RATIO: jitter range as a fraction of the base delay\n   *\n   * Jitter is applied symmetrically in [-ratio, +ratio]. For example, with a\n   * 500ms base delay and 0.2 ratio, the actual delay is in [400ms, 600ms].\n   *\n   * The backoff applies only to automatic reconnects started from the\n   * `onclose` handler. Explicit/manual calls to `reconnect()` run immediately\n   * (no backoff), and will preempt any scheduled automatic attempt.\n   */\n  static readonly BACKOFF_INITIAL_MS = 500;\n  static readonly BACKOFF_MULTIPLIER = 2;\n  static readonly BACKOFF_MAX_MS = 30_000;\n  static readonly BACKOFF_JITTER_RATIO = 0.2;\n\n  public elicitation: EventTarget = new EventTarget();\n  /**\n   * Private constructor to enforce using the static create method.\n   * @param endpoint - The URL of the MCP server to connect to\n   * @param transportType - The transport to use for the MCP client\n   * @param headers - Optional custom headers to include in requests\n   */\n  private constructor(\n    endpoint: string,\n    transportType: MCPTransport,\n    headers?: Record<string, string>,\n    authProvider?: OAuthClientProvider,\n    sessionId?: string,\n    handlers: MCPHandlers = {},\n  ) {\n    this.endpoint = endpoint;\n    this.headers = headers ?? {};\n    this.authProvider = authProvider;\n    this.transportType = transportType;\n    this.handlers = handlers;\n    this.transport = this.initializeTransport(sessionId);\n    this.client = this.initializeClient();\n  }\n\n  /**\n   * Creates and initializes a new MCPClient instance. This is the recommended\n   * way to create an MCPClient as it handles both instantiation and connection\n   * setup.\n   * @param endpoint - The URL of the MCP server to connect to\n   * @param transportType - The transport type to use for the MCP client. Defaults to HTTP.\n   * @param headers - Optional custom headers to include in requests\n   * @param authProvider - Optional auth provider to use for authentication\n   * @param sessionId - Optional session id to use for the MCP client - if not\n   *   provided, a new session will be created\n   * @returns A connected MCPClient instance ready for use\n   * @throws {Error} Will throw an error if connection fails\n   */\n  static async create(\n    endpoint: string,\n    transportType: MCPTransport = MCPTransport.HTTP,\n    headers: Record<string, string> | undefined,\n    authProvider: OAuthClientProvider | undefined,\n    sessionId: string | undefined,\n    handlers: MCPHandlers = {},\n  ): Promise<MCPClient> {\n    const mcpClient = new MCPClient(\n      endpoint,\n      transportType,\n      headers,\n      authProvider,\n      sessionId,\n      handlers,\n    );\n    await mcpClient.client.connect(mcpClient.transport);\n    if (\"sessionId\" in mcpClient.transport) {\n      mcpClient.sessionId = mcpClient.transport.sessionId;\n    }\n    return mcpClient;\n  }\n  /**\n   * Reconnects to the MCP server, optionally retaining the same session ID.\n   *\n   * Single‑flight semantics:\n   * - If a reconnect is already in progress (triggered either manually or by\n   * the automatic `onclose` handler), additional calls will await the\n   * in-flight reconnect rather than start another one.\n   * - If an automatic reconnect has been scheduled but not yet started (i.e.,\n   * we are waiting in a backoff delay), calling `reconnect()` manually will\n   * cancel the scheduled attempt and perform an immediate reconnect.\n   *\n   * Backoff policy:\n   * - Backoff delays with jitter are applied only for automatic reconnects\n   * (via `onclose`). Manual calls to `reconnect()` do not use backoff.\n   * @param newSession - Whether to create a new session (true) or reuse existing session ID (false)\n   * @param reportErrorOnClose - Whether to report errors when closing the client\n   * Note that only StreamableHTTPClientTransport supports session IDs.\n   * @returns A promise that resolves when the reconnect is complete\n   */\n  async reconnect(newSession = false, reportErrorOnClose = true) {\n    // If a reconnect is already running, coalesce into it.\n    if (this.reconnecting) {\n      return await this.reconnecting;\n    }\n\n    // Manual reconnect preempts any scheduled automatic attempt.\n    if (this.reconnectTimer) {\n      clearTimeout(this.reconnectTimer);\n      this.reconnectTimer = undefined;\n    }\n\n    const doReconnect = async () => {\n      const sessionId = newSession ? undefined : this.sessionId;\n\n      // Prevent re-entrant onclose during deliberate close by detaching\n      // the handler from the previous client instance.\n      const prevClient = this.client;\n      // Prevent re-entrant onclose callbacks from the previous client\n      prevClient.onclose = undefined;\n\n      try {\n        await prevClient.close();\n      } catch (error) {\n        if (reportErrorOnClose) {\n          console.error(\"Error closing Tambo MCP Client:\", error);\n        }\n      }\n\n      this.transport = this.initializeTransport(sessionId);\n      this.client = this.initializeClient();\n      await this.client.connect(this.transport);\n      // We may have gotten a session id from the server, so we need to set it\n      if (\"sessionId\" in this.transport) {\n        this.sessionId = this.transport.sessionId;\n        if (sessionId !== this.sessionId) {\n          // This is a pretty unusual thing to happen, but it might be possible?\n          console.warn(\"Session id mismatch\", sessionId, this.sessionId);\n        }\n      }\n    };\n\n    this.reconnecting = (async () => {\n      try {\n        await doReconnect();\n        // Successful manual reconnect: reset backoff.\n        this.backoffAttempts = 0;\n      } finally {\n        this.reconnecting = undefined;\n      }\n    })();\n\n    return await this.reconnecting;\n  }\n\n  /**\n   * Called by the underlying MCP SDK when the connection closes.\n   * Schedules an automatic reconnect with bounded exponential backoff and\n   * jitter. If a reconnect is already scheduled or running, this is a no-op.\n   */\n  private onclose() {\n    this.scheduleAutoReconnect();\n  }\n\n  /**\n   * Compute the next backoff delay with symmetric jitter.\n   * @returns The next backoff delay in milliseconds\n   */\n  private computeBackoffDelayMs(): number {\n    const base = Math.min(\n      MCPClient.BACKOFF_MAX_MS,\n      MCPClient.BACKOFF_INITIAL_MS *\n        Math.pow(MCPClient.BACKOFF_MULTIPLIER, this.backoffAttempts),\n    );\n    const jitterRange = MCPClient.BACKOFF_JITTER_RATIO * base;\n    const jitter = (Math.random() * 2 - 1) * jitterRange; // [-range, +range]\n    const ms = Math.max(0, Math.round(base + jitter));\n    return ms;\n  }\n\n  /**\n   * Schedule an automatic reconnect attempt if one is not already scheduled\n   * or running. Uses the backoff policy and self-reschedules on failure.\n   */\n  private scheduleAutoReconnect() {\n    if (this.reconnecting || this.reconnectTimer) {\n      return;\n    }\n\n    const delayMs = this.computeBackoffDelayMs();\n    console.warn(\n      \"Tambo MCP Client closed; attempting automatic reconnect in\",\n      `${delayMs}ms`,\n    );\n\n    this.reconnectTimer = setTimeout(async () => {\n      this.reconnectTimer = undefined;\n      // Start the actual reconnect (single-flight)\n      const inFlight = (this.reconnecting = this.reconnect(false, false));\n      try {\n        await inFlight;\n        // Success: reset attempts\n        this.backoffAttempts = 0;\n      } catch (err) {\n        // Failure: increase attempts; scheduling occurs in finally below so the\n        // new timer isn't blocked by `this.reconnecting` being truthy.\n        this.backoffAttempts += 1;\n        console.warn(\n          \"Automatic reconnect failed; will retry with backoff.\",\n          err,\n        );\n      } finally {\n        this.reconnecting = undefined;\n        if (this.backoffAttempts > 0) {\n          this.scheduleAutoReconnect();\n        }\n      }\n    }, delayMs);\n  }\n\n  private initializeTransport(sessionId: string | undefined) {\n    if (this.transportType === MCPTransport.SSE) {\n      return new SSEClientTransport(new URL(this.endpoint), {\n        authProvider: this.authProvider,\n        requestInit: { headers: this.headers },\n      });\n    } else {\n      return new StreamableHTTPClientTransport(new URL(this.endpoint), {\n        authProvider: this.authProvider,\n        requestInit: { headers: this.headers },\n        sessionId,\n      });\n    }\n  }\n\n  /**\n   * Initializes the MCP client with the appropriate capabilities and handlers\n   * @returns The initialized MCP client\n   */\n  private initializeClient() {\n    const elicitationCapability = this.handlers.elicitation\n      ? { elicitation: {} }\n      : {};\n    const samplingCapability = this.handlers.sampling ? { sampling: {} } : {};\n    const client = new Client(\n      {\n        name: \"tambo-mcp-client\",\n        version: \"1.0.0\",\n      },\n      {\n        capabilities: {\n          ...elicitationCapability,\n          ...samplingCapability,\n        },\n      },\n    );\n    client.onclose = this.onclose.bind(this);\n    if (this.handlers.elicitation) {\n      client.setRequestHandler(ElicitRequestSchema, this.handlers.elicitation);\n    }\n    if (this.handlers.sampling) {\n      client.setRequestHandler(\n        CreateMessageRequestSchema,\n        this.handlers.sampling,\n      );\n    }\n    return client;\n  }\n\n  /**\n   * Retrieves a complete list of all available tools from the MCP server.\n   * Handles pagination automatically by following cursors until all tools are fetched.\n   * @returns A complete list of all available tools and their descriptions\n   * @throws {Error} Will throw an error if any server request fails during pagination\n   */\n  async listTools(): Promise<MCPToolSpec[]> {\n    const allTools: MCPToolSpec[] = [];\n    let hasMore = true;\n    let cursor: string | undefined = undefined;\n\n    while (hasMore) {\n      const response = await this.client.listTools({ cursor }, {});\n      allTools.push(\n        ...response.tools.map((tool): MCPToolSpec => {\n          // make sure the right type is used\n          const inputSchemaType: string = tool.inputSchema.type;\n          if (inputSchemaType !== \"object\") {\n            throw new Error(\n              `Input schema for tool ${tool.name} is not an object`,\n            );\n          }\n\n          return {\n            name: tool.name,\n            description: tool.description,\n            inputSchema: tool.inputSchema as JSONSchema7,\n          };\n        }),\n      );\n\n      if (response.nextCursor) {\n        cursor = response.nextCursor;\n      } else {\n        hasMore = false;\n      }\n    }\n\n    return allTools;\n  }\n\n  getServerCapabilities() {\n    return this.client.getServerCapabilities();\n  }\n\n  getServerVersion() {\n    return this.client.getServerVersion();\n  }\n\n  getInstructions() {\n    return this.client.getInstructions();\n  }\n\n  /**\n   * Calls a specific tool on the MCP server with the provided arguments.\n   * @param name - The name of the tool to call\n   * @param args - Arguments to pass to the tool, must match the tool's expected schema\n   * @returns The result from the tool execution\n   * @throws {Error} Will throw an error if the tool call fails or if arguments are invalid\n   */\n  async callTool(\n    name: string,\n    args: Record<string, unknown>,\n  ): Promise<MCPToolCallResult> {\n    const result = await this.client.callTool({\n      name,\n      arguments: args,\n    });\n    return result;\n  }\n\n  updateElicitationHandler(\n    handler: ((e: ElicitRequest) => Promise<ElicitResult>) | undefined,\n  ) {\n    // Because we advertise the elicitation capability on initial connection, we can only update\n    // an existing handler, not add it if we haven't set it yet.\n    if (handler && !this.handlers.elicitation) {\n      throw new Error(\"Elicitation handler must be set on create\");\n    }\n    this.handlers = {\n      ...this.handlers,\n      elicitation: handler,\n    };\n    if (!handler) {\n      const method = ElicitRequestSchema.shape.method.value;\n      this.client.removeRequestHandler(method);\n      return;\n    }\n    this.client.setRequestHandler(ElicitRequestSchema, handler);\n  }\n\n  updateSamplingHandler(\n    handler:\n      | ((e: CreateMessageRequest) => Promise<CreateMessageResult>)\n      | undefined,\n  ) {\n    // Because we advertise the sampling capability on initial connection, we can only update\n    // an existing handler, not add it if we haven't set it yet.\n    if (handler && !this.handlers.sampling) {\n      throw new Error(\"Sampling handler must be set on create\");\n    }\n    this.handlers = {\n      ...this.handlers,\n      sampling: handler,\n    };\n    if (!handler) {\n      const method = CreateMessageRequestSchema.shape.method.value;\n      this.client.removeRequestHandler(method);\n      return;\n    }\n    this.client.setRequestHandler(CreateMessageRequestSchema, handler);\n  }\n}\n\n/**\n * The result of a tool call.\n * This is the same as the result of a tool call in the OpenAI SDK, but is reified here\n */\nexport type MCPToolCallResult = Awaited<\n  ReturnType<typeof Client.prototype.callTool>\n>;\n\n// Example usage:\n/*\nconst mcp = await MCPClient.create('https://api.example.com/mcp', MCPTransport.HTTP);\nconst tools = await mcp.listTools();\nconst result = await mcp.callTool('toolName', { arg1: 'value1' });\n*/\n\nexport interface MCPToolSpec {\n  name: string;\n  description?: string;\n  inputSchema?: JSONSchema7;\n}\n"]}
+{"version":3,"file":"mcp-client.js","sourceRoot":"","sources":["../../src/mcp/mcp-client.ts"],"names":[],"mappings":";;;AACA,wEAAmE;AACnE,oEAA6E;AAC7E,0FAAmG;AACnG,iEAO4C;AAG5C,IAAY,YAGX;AAHD,WAAY,YAAY;IACtB,2BAAW,CAAA;IACX,6BAAa,CAAA;AACf,CAAC,EAHW,YAAY,4BAAZ,YAAY,QAGvB;AAyBD;;;;;;;;;GASG;AACH,MAAa,SAAS;IACpB;;;;OAIG;IACH,MAAM,CAAS;IACP,SAAS,CAAqD;IAC9D,aAAa,CAAe;IAC7B,SAAS,CAAU;IAClB,QAAQ,CAAS;IACjB,OAAO,CAAyB;IAChC,YAAY,CAAuB;IACnC,QAAQ,CAAuB;IAEvC;;;;;OAKG;IACH,YACE,QAAgB,EAChB,aAA2B,EAC3B,OAAgC,EAChC,YAAkC,EAClC,SAAkB,EAClB,WAAiC,EAAE;QAEnC,IAAI,CAAC,QAAQ,GAAG,QAAQ,CAAC;QACzB,IAAI,CAAC,OAAO,GAAG,OAAO,IAAI,EAAE,CAAC;QAC7B,IAAI,CAAC,YAAY,GAAG,YAAY,CAAC;QACjC,IAAI,CAAC,aAAa,GAAG,aAAa,CAAC;QACnC,IAAI,CAAC,QAAQ,GAAG,QAAQ,CAAC;QACzB,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC,mBAAmB,CAAC,SAAS,CAAC,CAAC;QACrD,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC,gBAAgB,EAAE,CAAC;IACxC,CAAC;IAED;;;;;;;;;;;;OAYG;IACH,MAAM,CAAC,KAAK,CAAC,MAAM,CACjB,QAAgB,EAChB,gBAA8B,YAAY,CAAC,IAAI,EAC/C,OAA2C,EAC3C,YAA6C,EAC7C,SAA6B,EAC7B,WAAiC,EAAE;QAEnC,MAAM,SAAS,GAAG,IAAI,SAAS,CAC7B,QAAQ,EACR,aAAa,EACb,OAAO,EACP,YAAY,EACZ,SAAS,EACT,QAAQ,CACT,CAAC;QACF,MAAM,SAAS,CAAC,MAAM,CAAC,OAAO,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC;QACpD,IAAI,WAAW,IAAI,SAAS,CAAC,SAAS,EAAE,CAAC;YACvC,SAAS,CAAC,SAAS,GAAG,SAAS,CAAC,SAAS,CAAC,SAAS,CAAC;QACtD,CAAC;QACD,OAAO,SAAS,CAAC;IACnB,CAAC;IAEO,mBAAmB,CAAC,SAA6B;QACvD,IAAI,IAAI,CAAC,aAAa,KAAK,YAAY,CAAC,GAAG,EAAE,CAAC;YAC5C,OAAO,IAAI,2BAAkB,CAAC,IAAI,GAAG,CAAC,IAAI,CAAC,QAAQ,CAAC,EAAE;gBACpD,YAAY,EAAE,IAAI,CAAC,YAAY;gBAC/B,WAAW,EAAE,EAAE,OAAO,EAAE,IAAI,CAAC,OAAO,EAAE;aACvC,CAAC,CAAC;QACL,CAAC;aAAM,CAAC;YACN,OAAO,IAAI,iDAA6B,CAAC,IAAI,GAAG,CAAC,IAAI,CAAC,QAAQ,CAAC,EAAE;gBAC/D,YAAY,EAAE,IAAI,CAAC,YAAY;gBAC/B,WAAW,EAAE,EAAE,OAAO,EAAE,IAAI,CAAC,OAAO,EAAE;gBACtC,SAAS;aACV,CAAC,CAAC;QACL,CAAC;IACH,CAAC;IAED;;;OAGG;IACK,gBAAgB;QACtB,MAAM,qBAAqB,GAAG,IAAI,CAAC,QAAQ,CAAC,WAAW;YACrD,CAAC,CAAC,EAAE,WAAW,EAAE,EAAE,EAAE;YACrB,CAAC,CAAC,EAAE,CAAC;QACP,MAAM,kBAAkB,GAAG,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC,CAAC,EAAE,QAAQ,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;QAC1E,MAAM,MAAM,GAAG,IAAI,iBAAM,CACvB;YACE,IAAI,EAAE,kBAAkB;YACxB,OAAO,EAAE,OAAO;SACjB,EACD;YACE,YAAY,EAAE;gBACZ,GAAG,qBAAqB;gBACxB,GAAG,kBAAkB;aACtB;SACF,CACF,CAAC;QAEF,IAAI,IAAI,CAAC,QAAQ,CAAC,WAAW,EAAE,CAAC;YAC9B,MAAM,CAAC,iBAAiB,CAAC,8BAAmB,EAAE,IAAI,CAAC,QAAQ,CAAC,WAAW,CAAC,CAAC;QAC3E,CAAC;QACD,IAAI,IAAI,CAAC,QAAQ,CAAC,QAAQ,EAAE,CAAC;YAC3B,MAAM,CAAC,iBAAiB,CACtB,qCAA0B,EAC1B,IAAI,CAAC,QAAQ,CAAC,QAAQ,CACvB,CAAC;QACJ,CAAC;QACD,OAAO,MAAM,CAAC;IAChB,CAAC;IAED;;;;;OAKG;IACH,KAAK,CAAC,SAAS;QACb,MAAM,QAAQ,GAAkB,EAAE,CAAC;QACnC,IAAI,OAAO,GAAG,IAAI,CAAC;QACnB,IAAI,MAAM,GAAuB,SAAS,CAAC;QAE3C,OAAO,OAAO,EAAE,CAAC;YACf,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,EAAE,MAAM,EAAE,EAAE,EAAE,CAAC,CAAC;YAC7D,QAAQ,CAAC,IAAI,CACX,GAAG,QAAQ,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAe,EAAE;gBAC1C,mCAAmC;gBACnC,MAAM,eAAe,GAAW,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC;gBACtD,IAAI,eAAe,KAAK,QAAQ,EAAE,CAAC;oBACjC,MAAM,IAAI,KAAK,CACb,yBAAyB,IAAI,CAAC,IAAI,mBAAmB,CACtD,CAAC;gBACJ,CAAC;gBAED,OAAO;oBACL,IAAI,EAAE,IAAI,CAAC,IAAI;oBACf,WAAW,EAAE,IAAI,CAAC,WAAW;oBAC7B,WAAW,EAAE,IAAI,CAAC,WAA0B;iBAC7C,CAAC;YACJ,CAAC,CAAC,CACH,CAAC;YAEF,IAAI,QAAQ,CAAC,UAAU,EAAE,CAAC;gBACxB,MAAM,GAAG,QAAQ,CAAC,UAAU,CAAC;YAC/B,CAAC;iBAAM,CAAC;gBACN,OAAO,GAAG,KAAK,CAAC;YAClB,CAAC;QACH,CAAC;QAED,OAAO,QAAQ,CAAC;IAClB,CAAC;IAED,qBAAqB;QACnB,OAAO,IAAI,CAAC,MAAM,CAAC,qBAAqB,EAAE,CAAC;IAC7C,CAAC;IAED,gBAAgB;QACd,OAAO,IAAI,CAAC,MAAM,CAAC,gBAAgB,EAAE,CAAC;IACxC,CAAC;IAED,eAAe;QACb,OAAO,IAAI,CAAC,MAAM,CAAC,eAAe,EAAE,CAAC;IACvC,CAAC;IAED;;;;;;OAMG;IACH,KAAK,CAAC,QAAQ,CACZ,IAAY,EACZ,IAA6B,EAC7B,KAA+B;QAE/B,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC;YACxC,IAAI;YACJ,SAAS,EAAE,IAAI;YACf,KAAK;SACN,CAAC,CAAC;QACH,OAAO,MAAM,CAAC;IAChB,CAAC;IAED,wBAAwB,CACtB,OAAkE;QAElE,iCAAiC;QACjC,IAAI,OAAO,KAAK,IAAI,CAAC,QAAQ,CAAC,WAAW,EAAE,CAAC;YAC1C,OAAO;QACT,CAAC;QAED,4FAA4F;QAC5F,4DAA4D;QAC5D,IAAI,OAAO,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,WAAW,EAAE,CAAC;YAC1C,MAAM,IAAI,KAAK,CAAC,2CAA2C,CAAC,CAAC;QAC/D,CAAC;QACD,IAAI,CAAC,QAAQ,GAAG;YACd,GAAG,IAAI,CAAC,QAAQ;YAChB,WAAW,EAAE,OAAO;SACrB,CAAC;QACF,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,MAAM,MAAM,GAAG,8BAAmB,CAAC,KAAK,CAAC,MAAM,CAAC,KAAK,CAAC;YACtD,IAAI,CAAC,MAAM,CAAC,oBAAoB,CAAC,MAAM,CAAC,CAAC;YACzC,OAAO;QACT,CAAC;QACD,IAAI,CAAC,MAAM,CAAC,iBAAiB,CAAC,8BAAmB,EAAE,OAAO,CAAC,CAAC;IAC9D,CAAC;IAED,qBAAqB,CACnB,OAEa;QAEb,iCAAiC;QACjC,IAAI,OAAO,KAAK,IAAI,CAAC,QAAQ,CAAC,QAAQ,EAAE,CAAC;YACvC,OAAO;QACT,CAAC;QAED,yFAAyF;QACzF,4DAA4D;QAC5D,IAAI,OAAO,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,QAAQ,EAAE,CAAC;YACvC,MAAM,IAAI,KAAK,CAAC,wCAAwC,CAAC,CAAC;QAC5D,CAAC;QACD,IAAI,CAAC,QAAQ,GAAG;YACd,GAAG,IAAI,CAAC,QAAQ;YAChB,QAAQ,EAAE,OAAO;SAClB,CAAC;QACF,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,MAAM,MAAM,GAAG,qCAA0B,CAAC,KAAK,CAAC,MAAM,CAAC,KAAK,CAAC;YAC7D,IAAI,CAAC,MAAM,CAAC,oBAAoB,CAAC,MAAM,CAAC,CAAC;YACzC,OAAO;QACT,CAAC;QACD,IAAI,CAAC,MAAM,CAAC,iBAAiB,CAAC,qCAA0B,EAAE,OAAO,CAAC,CAAC;IACrE,CAAC;IAED,KAAK,CAAC,KAAK;QACT,yEAAyE;QACzE,yEAAyE;QACzE,UAAU;QACV,MAAM,IAAI,CAAC,SAAS,CAAC,KAAK,EAAE,CAAC;QAC7B,MAAM,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC;IAC5B,CAAC;CACF;AA/PD,8BA+PC","sourcesContent":["import { type OAuthClientProvider } from \"@modelcontextprotocol/sdk/client/auth.js\";\nimport { Client } from \"@modelcontextprotocol/sdk/client/index.js\";\nimport { SSEClientTransport } from \"@modelcontextprotocol/sdk/client/sse.js\";\nimport { StreamableHTTPClientTransport } from \"@modelcontextprotocol/sdk/client/streamableHttp.js\";\nimport {\n  CreateMessageRequest,\n  CreateMessageRequestSchema,\n  CreateMessageResult,\n  ElicitRequest,\n  ElicitRequestSchema,\n  ElicitResult,\n} from \"@modelcontextprotocol/sdk/types.js\";\nimport { JSONSchema7 } from \"json-schema\";\n\nexport enum MCPTransport {\n  SSE = \"sse\",\n  HTTP = \"http\",\n}\n\n/**\n * Handlers for MCP requests - these are only used if the server supports the corresponding capabilities\n * @param elicitation - Handler for elicitation requests\n * @param sampling - Handler for sampling requests\n * @example\n * ```typescript\n * const mcp = await MCPClient.create(\n *     'https://api.example.com/mcp',\n *     MCPTransport.HTTP,\n *     {},\n *     undefined,\n *     undefined,\n *     {\n *       elicitation: (e: ElicitRequest) => Promise.resolve({...}),\n *     },\n * });\n * ```\n */\nexport interface MCPHandlers {\n  elicitation: (e: ElicitRequest) => Promise<ElicitResult>;\n  sampling: (e: CreateMessageRequest) => Promise<CreateMessageResult>;\n}\n\n/**\n * A client for interacting with MCP (Model Context Protocol) servers.\n * Provides a simple interface for listing and calling tools exposed by the server.\n * @example\n * ```typescript\n * const mcp = await MCPClient.create('https://api.example.com/mcp');\n * const tools = await mcp.listTools();\n * const result = await mcp.callTool('toolName', { arg1: 'value1' });\n * ```\n */\nexport class MCPClient {\n  /**\n   * The underlying MCP client\n   *\n   * Be careful not to mutate the client directly, use the methods provided instead.\n   */\n  client: Client;\n  private transport: SSEClientTransport | StreamableHTTPClientTransport;\n  private transportType: MCPTransport;\n  public sessionId?: string;\n  private endpoint: string;\n  private headers: Record<string, string>;\n  private authProvider?: OAuthClientProvider;\n  private handlers: Partial<MCPHandlers>;\n\n  /**\n   * Private constructor to enforce using the static create method.\n   * @param endpoint - The URL of the MCP server to connect to\n   * @param transportType - The transport to use for the MCP client\n   * @param headers - Optional custom headers to include in requests\n   */\n  private constructor(\n    endpoint: string,\n    transportType: MCPTransport,\n    headers?: Record<string, string>,\n    authProvider?: OAuthClientProvider,\n    sessionId?: string,\n    handlers: Partial<MCPHandlers> = {},\n  ) {\n    this.endpoint = endpoint;\n    this.headers = headers ?? {};\n    this.authProvider = authProvider;\n    this.transportType = transportType;\n    this.handlers = handlers;\n    this.transport = this.initializeTransport(sessionId);\n    this.client = this.initializeClient();\n  }\n\n  /**\n   * Creates and initializes a new MCPClient instance. This is the recommended\n   * way to create an MCPClient as it handles both instantiation and connection\n   * setup.\n   * @param endpoint - The URL of the MCP server to connect to\n   * @param transportType - The transport type to use for the MCP client. Defaults to HTTP.\n   * @param headers - Optional custom headers to include in requests\n   * @param authProvider - Optional auth provider to use for authentication\n   * @param sessionId - Optional session id to use for the MCP client - if not\n   *   provided, a new session will be created\n   * @returns A connected MCPClient instance ready for use\n   * @throws {Error} Will throw an error if connection fails\n   */\n  static async create(\n    endpoint: string,\n    transportType: MCPTransport = MCPTransport.HTTP,\n    headers: Record<string, string> | undefined,\n    authProvider: OAuthClientProvider | undefined,\n    sessionId: string | undefined,\n    handlers: Partial<MCPHandlers> = {},\n  ): Promise<MCPClient> {\n    const mcpClient = new MCPClient(\n      endpoint,\n      transportType,\n      headers,\n      authProvider,\n      sessionId,\n      handlers,\n    );\n    await mcpClient.client.connect(mcpClient.transport);\n    if (\"sessionId\" in mcpClient.transport) {\n      mcpClient.sessionId = mcpClient.transport.sessionId;\n    }\n    return mcpClient;\n  }\n\n  private initializeTransport(sessionId: string | undefined) {\n    if (this.transportType === MCPTransport.SSE) {\n      return new SSEClientTransport(new URL(this.endpoint), {\n        authProvider: this.authProvider,\n        requestInit: { headers: this.headers },\n      });\n    } else {\n      return new StreamableHTTPClientTransport(new URL(this.endpoint), {\n        authProvider: this.authProvider,\n        requestInit: { headers: this.headers },\n        sessionId,\n      });\n    }\n  }\n\n  /**\n   * Initializes the MCP client with the appropriate capabilities and handlers\n   * @returns The initialized MCP client\n   */\n  private initializeClient() {\n    const elicitationCapability = this.handlers.elicitation\n      ? { elicitation: {} }\n      : {};\n    const samplingCapability = this.handlers.sampling ? { sampling: {} } : {};\n    const client = new Client(\n      {\n        name: \"tambo-mcp-client\",\n        version: \"1.0.0\",\n      },\n      {\n        capabilities: {\n          ...elicitationCapability,\n          ...samplingCapability,\n        },\n      },\n    );\n\n    if (this.handlers.elicitation) {\n      client.setRequestHandler(ElicitRequestSchema, this.handlers.elicitation);\n    }\n    if (this.handlers.sampling) {\n      client.setRequestHandler(\n        CreateMessageRequestSchema,\n        this.handlers.sampling,\n      );\n    }\n    return client;\n  }\n\n  /**\n   * Retrieves a complete list of all available tools from the MCP server.\n   * Handles pagination automatically by following cursors until all tools are fetched.\n   * @returns A complete list of all available tools and their descriptions\n   * @throws {Error} Will throw an error if any server request fails during pagination\n   */\n  async listTools(): Promise<MCPToolSpec[]> {\n    const allTools: MCPToolSpec[] = [];\n    let hasMore = true;\n    let cursor: string | undefined = undefined;\n\n    while (hasMore) {\n      const response = await this.client.listTools({ cursor }, {});\n      allTools.push(\n        ...response.tools.map((tool): MCPToolSpec => {\n          // make sure the right type is used\n          const inputSchemaType: string = tool.inputSchema.type;\n          if (inputSchemaType !== \"object\") {\n            throw new Error(\n              `Input schema for tool ${tool.name} is not an object`,\n            );\n          }\n\n          return {\n            name: tool.name,\n            description: tool.description,\n            inputSchema: tool.inputSchema as JSONSchema7,\n          };\n        }),\n      );\n\n      if (response.nextCursor) {\n        cursor = response.nextCursor;\n      } else {\n        hasMore = false;\n      }\n    }\n\n    return allTools;\n  }\n\n  getServerCapabilities() {\n    return this.client.getServerCapabilities();\n  }\n\n  getServerVersion() {\n    return this.client.getServerVersion();\n  }\n\n  getInstructions() {\n    return this.client.getInstructions();\n  }\n\n  /**\n   * Calls a specific tool on the MCP server with the provided arguments.\n   * @param name - The name of the tool to call\n   * @param args - Arguments to pass to the tool, must match the tool's expected schema\n   * @returns The result from the tool execution\n   * @throws {Error} Will throw an error if the tool call fails or if arguments are invalid\n   */\n  async callTool(\n    name: string,\n    args: Record<string, unknown>,\n    _meta?: Record<string, unknown>,\n  ): Promise<MCPToolCallResult> {\n    const result = await this.client.callTool({\n      name,\n      arguments: args,\n      _meta,\n    });\n    return result;\n  }\n\n  updateElicitationHandler(\n    handler: ((e: ElicitRequest) => Promise<ElicitResult>) | undefined,\n  ) {\n    // Skip if handler hasn't changed\n    if (handler === this.handlers.elicitation) {\n      return;\n    }\n\n    // Because we advertise the elicitation capability on initial connection, we can only update\n    // an existing handler, not add it if we haven't set it yet.\n    if (handler && !this.handlers.elicitation) {\n      throw new Error(\"Elicitation handler must be set on create\");\n    }\n    this.handlers = {\n      ...this.handlers,\n      elicitation: handler,\n    };\n    if (!handler) {\n      const method = ElicitRequestSchema.shape.method.value;\n      this.client.removeRequestHandler(method);\n      return;\n    }\n    this.client.setRequestHandler(ElicitRequestSchema, handler);\n  }\n\n  updateSamplingHandler(\n    handler:\n      | ((e: CreateMessageRequest) => Promise<CreateMessageResult>)\n      | undefined,\n  ) {\n    // Skip if handler hasn't changed\n    if (handler === this.handlers.sampling) {\n      return;\n    }\n\n    // Because we advertise the sampling capability on initial connection, we can only update\n    // an existing handler, not add it if we haven't set it yet.\n    if (handler && !this.handlers.sampling) {\n      throw new Error(\"Sampling handler must be set on create\");\n    }\n    this.handlers = {\n      ...this.handlers,\n      sampling: handler,\n    };\n    if (!handler) {\n      const method = CreateMessageRequestSchema.shape.method.value;\n      this.client.removeRequestHandler(method);\n      return;\n    }\n    this.client.setRequestHandler(CreateMessageRequestSchema, handler);\n  }\n\n  async close() {\n    // Not really sure which one of these to close first, but we'll close the\n    // transport first so that no requests can come in and hit closing/closed\n    // clients\n    await this.transport.close();\n    await this.client.close();\n  }\n}\n\n/**\n * The result of a tool call.\n * This is the same as the result of a tool call in the OpenAI SDK, but is reified here\n */\nexport type MCPToolCallResult = Awaited<\n  ReturnType<typeof Client.prototype.callTool>\n>;\n\n// Example usage:\n/*\nconst mcp = await MCPClient.create('https://api.example.com/mcp', MCPTransport.HTTP);\nconst tools = await mcp.listTools();\nconst result = await mcp.callTool('toolName', { arg1: 'value1' });\n*/\n\nexport interface MCPToolSpec {\n  name: string;\n  description?: string;\n  inputSchema?: JSONSchema7;\n}\n"]}

package/dist/mcp/mcp-hooks.d.ts

@@ -0,0 +1,93 @@
+import { type ListPromptsResult } from "@modelcontextprotocol/sdk/types.js";
+import { McpServerInfo } from "./tambo-mcp-provider";
+export type ListPromptItem = ListPromptsResult["prompts"][number];
+export interface ListPromptEntry {
+    server: McpServerInfo;
+    prompt: ListPromptItem;
+}
+/**
+ * Hook to get the prompts for all the registered MCP servers.
+ * @returns The prompts for the MCP servers, including the server that the prompt was found on.
+ */
+export declare function useTamboMcpPromptList(): import("@tanstack/react-query").UseQueryResult<ListPromptEntry[], Error>;
+/**
+ * Hook to get the prompt for the specified name.
+ * @param promptName - The name of the prompt to get.
+ * @returns The prompt for the specified name.
+ */
+export declare function useTamboMcpPrompt(promptName: string): import("@tanstack/react-query").UseQueryResult<{
+    [x: string]: unknown;
+    messages: {
+        [x: string]: unknown;
+        content: {
+            [x: string]: unknown;
+            text: string;
+            type: "text";
+            _meta?: {
+                [x: string]: unknown;
+            } | undefined;
+        } | {
+            [x: string]: unknown;
+            data: string;
+            type: "image";
+            mimeType: string;
+            _meta?: {
+                [x: string]: unknown;
+            } | undefined;
+        } | {
+            [x: string]: unknown;
+            data: string;
+            type: "audio";
+            mimeType: string;
+            _meta?: {
+                [x: string]: unknown;
+            } | undefined;
+        } | {
+            [x: string]: unknown;
+            name: string;
+            type: "resource_link";
+            uri: string;
+            title?: string | undefined;
+            description?: string | undefined;
+            _meta?: {
+                [x: string]: unknown;
+            } | undefined;
+            mimeType?: string | undefined;
+            icons?: {
+                [x: string]: unknown;
+                src: string;
+                mimeType?: string | undefined;
+                sizes?: string[] | undefined;
+            }[] | undefined;
+        } | {
+            [x: string]: unknown;
+            type: "resource";
+            resource: {
+                [x: string]: unknown;
+                text: string;
+                uri: string;
+                _meta?: {
+                    [x: string]: unknown;
+                } | undefined;
+                mimeType?: string | undefined;
+            } | {
+                [x: string]: unknown;
+                uri: string;
+                blob: string;
+                _meta?: {
+                    [x: string]: unknown;
+                } | undefined;
+                mimeType?: string | undefined;
+            };
+            _meta?: {
+                [x: string]: unknown;
+            } | undefined;
+        };
+        role: "user" | "assistant";
+    }[];
+    description?: string | undefined;
+    _meta?: {
+        [x: string]: unknown;
+    } | undefined;
+} | undefined, Error>;
+//# sourceMappingURL=mcp-hooks.d.ts.map

package/dist/mcp/mcp-hooks.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"mcp-hooks.d.ts","sourceRoot":"","sources":["../../src/mcp/mcp-hooks.ts"],"names":[],"mappings":"AAAA,OAAO,EAEL,KAAK,iBAAiB,EACvB,MAAM,oCAAoC,CAAC;AAE5C,OAAO,EAAE,aAAa,EAAsB,MAAM,sBAAsB,CAAC;AAEzE,MAAM,MAAM,cAAc,GAAG,iBAAiB,CAAC,SAAS,CAAC,CAAC,MAAM,CAAC,CAAC;AAClE,MAAM,WAAW,eAAe;IAC9B,MAAM,EAAE,aAAa,CAAC;IACtB,MAAM,EAAE,cAAc,CAAC;CACxB;AAED;;;GAGG;AACH,wBAAgB,qBAAqB,6EAyBpC;AAED;;;;GAIG;AACH,wBAAgB,iBAAiB,CAAC,UAAU,EAAE,MAAM;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;sBAsBnD"}

package/dist/mcp/mcp-hooks.js

@@ -0,0 +1,56 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.useTamboMcpPromptList = useTamboMcpPromptList;
+exports.useTamboMcpPrompt = useTamboMcpPrompt;
+const hooks_1 = require("../hooks");
+const tambo_mcp_provider_1 = require("./tambo-mcp-provider");
+/**
+ * Hook to get the prompts for all the registered MCP servers.
+ * @returns The prompts for the MCP servers, including the server that the prompt was found on.
+ */
+function useTamboMcpPromptList() {
+    const mcpServers = (0, tambo_mcp_provider_1.useTamboMcpServers)();
+    return (0, hooks_1.useTamboQuery)({
+        queryKey: ["mcp-prompts"],
+        queryFn: async () => {
+            const promptResults = await Promise.allSettled(mcpServers.map(async (mcpServer) => ({
+                server: mcpServer,
+                response: await mcpServer.client?.client.listPrompts(),
+            })));
+            const prompts = promptResults
+                .filter((result) => result.status === "fulfilled")
+                .flatMap((result) => {
+                const { server, response } = result.value;
+                return response?.prompts?.map((prompt) => ({
+                    server,
+                    prompt,
+                }));
+            })
+                .filter((prompt) => prompt !== undefined);
+            return prompts;
+        },
+    });
+}
+/**
+ * Hook to get the prompt for the specified name.
+ * @param promptName - The name of the prompt to get.
+ * @returns The prompt for the specified name.
+ */
+function useTamboMcpPrompt(promptName) {
+    // figure out which server has the prompt
+    const { data: promptEntries } = useTamboMcpPromptList();
+    const promptEntry = promptEntries?.find((prompt) => prompt.prompt.name === promptName);
+    const mcpServers = (0, tambo_mcp_provider_1.useTamboMcpServers)();
+    const mcpServer = mcpServers.find((mcpServer) => mcpServer.name === promptEntry?.server.name &&
+        mcpServer.url === promptEntry?.server.url &&
+        mcpServer.transport === promptEntry?.server.transport);
+    return (0, hooks_1.useTamboQuery)({
+        queryKey: ["mcp-prompt", promptName],
+        queryFn: async () => {
+            return await mcpServer?.client?.client.getPrompt({
+                name: promptName,
+            });
+        },
+    });
+}
+//# sourceMappingURL=mcp-hooks.js.map

package/dist/mcp/mcp-hooks.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"mcp-hooks.js","sourceRoot":"","sources":["../../src/mcp/mcp-hooks.ts"],"names":[],"mappings":";;AAiBA,sDAyBC;AAOD,8CAsBC;AAnED,oCAAyC;AACzC,6DAAyE;AAQzE;;;GAGG;AACH,SAAgB,qBAAqB;IACnC,MAAM,UAAU,GAAG,IAAA,uCAAkB,GAAE,CAAC;IACxC,OAAO,IAAA,qBAAa,EAAC;QACnB,QAAQ,EAAE,CAAC,aAAa,CAAC;QACzB,OAAO,EAAE,KAAK,IAAgC,EAAE;YAC9C,MAAM,aAAa,GAAG,MAAM,OAAO,CAAC,UAAU,CAC5C,UAAU,CAAC,GAAG,CAAC,KAAK,EAAE,SAAS,EAAE,EAAE,CAAC,CAAC;gBACnC,MAAM,EAAE,SAAS;gBACjB,QAAQ,EAAE,MAAM,SAAS,CAAC,MAAM,EAAE,MAAM,CAAC,WAAW,EAAE;aACvD,CAAC,CAAC,CACJ,CAAC;YAEF,MAAM,OAAO,GAAsB,aAAa;iBAC7C,MAAM,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,MAAM,KAAK,WAAW,CAAC;iBACjD,OAAO,CAAC,CAAC,MAAM,EAAE,EAAE;gBAClB,MAAM,EAAE,MAAM,EAAE,QAAQ,EAAE,GAAG,MAAM,CAAC,KAAK,CAAC;gBAC1C,OAAO,QAAQ,EAAE,OAAO,EAAE,GAAG,CAAC,CAAC,MAAsB,EAAE,EAAE,CAAC,CAAC;oBACzD,MAAM;oBACN,MAAM;iBACP,CAAC,CAAC,CAAC;YACN,CAAC,CAAC;iBACD,MAAM,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,KAAK,SAAS,CAAC,CAAC;YAC5C,OAAO,OAAO,CAAC;QACjB,CAAC;KACF,CAAC,CAAC;AACL,CAAC;AAED;;;;GAIG;AACH,SAAgB,iBAAiB,CAAC,UAAkB;IAClD,yCAAyC;IACzC,MAAM,EAAE,IAAI,EAAE,aAAa,EAAE,GAAG,qBAAqB,EAAE,CAAC;IACxD,MAAM,WAAW,GAAG,aAAa,EAAE,IAAI,CACrC,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,MAAM,CAAC,IAAI,KAAK,UAAU,CAC9C,CAAC;IACF,MAAM,UAAU,GAAG,IAAA,uCAAkB,GAAE,CAAC;IACxC,MAAM,SAAS,GAAG,UAAU,CAAC,IAAI,CAC/B,CAAC,SAAS,EAAE,EAAE,CACZ,SAAS,CAAC,IAAI,KAAK,WAAW,EAAE,MAAM,CAAC,IAAI;QAC3C,SAAS,CAAC,GAAG,KAAK,WAAW,EAAE,MAAM,CAAC,GAAG;QACzC,SAAS,CAAC,SAAS,KAAK,WAAW,EAAE,MAAM,CAAC,SAAS,CACxD,CAAC;IAEF,OAAO,IAAA,qBAAa,EAAC;QACnB,QAAQ,EAAE,CAAC,YAAY,EAAE,UAAU,CAAC;QACpC,OAAO,EAAE,KAAK,IAA0C,EAAE;YACxD,OAAO,MAAM,SAAS,EAAE,MAAM,EAAE,MAAM,CAAC,SAAS,CAAC;gBAC/C,IAAI,EAAE,UAAU;aACjB,CAAC,CAAC;QACL,CAAC;KACF,CAAC,CAAC;AACL,CAAC","sourcesContent":["import {\n  GetPromptResult,\n  type ListPromptsResult,\n} from \"@modelcontextprotocol/sdk/types.js\";\nimport { useTamboQuery } from \"../hooks\";\nimport { McpServerInfo, useTamboMcpServers } from \"./tambo-mcp-provider\";\n\nexport type ListPromptItem = ListPromptsResult[\"prompts\"][number];\nexport interface ListPromptEntry {\n  server: McpServerInfo;\n  prompt: ListPromptItem;\n}\n\n/**\n * Hook to get the prompts for all the registered MCP servers.\n * @returns The prompts for the MCP servers, including the server that the prompt was found on.\n */\nexport function useTamboMcpPromptList() {\n  const mcpServers = useTamboMcpServers();\n  return useTamboQuery({\n    queryKey: [\"mcp-prompts\"],\n    queryFn: async (): Promise<ListPromptEntry[]> => {\n      const promptResults = await Promise.allSettled(\n        mcpServers.map(async (mcpServer) => ({\n          server: mcpServer,\n          response: await mcpServer.client?.client.listPrompts(),\n        })),\n      );\n\n      const prompts: ListPromptEntry[] = promptResults\n        .filter((result) => result.status === \"fulfilled\")\n        .flatMap((result) => {\n          const { server, response } = result.value;\n          return response?.prompts?.map((prompt: ListPromptItem) => ({\n            server,\n            prompt,\n          }));\n        })\n        .filter((prompt) => prompt !== undefined);\n      return prompts;\n    },\n  });\n}\n\n/**\n * Hook to get the prompt for the specified name.\n * @param promptName - The name of the prompt to get.\n * @returns The prompt for the specified name.\n */\nexport function useTamboMcpPrompt(promptName: string) {\n  // figure out which server has the prompt\n  const { data: promptEntries } = useTamboMcpPromptList();\n  const promptEntry = promptEntries?.find(\n    (prompt) => prompt.prompt.name === promptName,\n  );\n  const mcpServers = useTamboMcpServers();\n  const mcpServer = mcpServers.find(\n    (mcpServer) =>\n      mcpServer.name === promptEntry?.server.name &&\n      mcpServer.url === promptEntry?.server.url &&\n      mcpServer.transport === promptEntry?.server.transport,\n  );\n\n  return useTamboQuery({\n    queryKey: [\"mcp-prompt\", promptName],\n    queryFn: async (): Promise<GetPromptResult | undefined> => {\n      return await mcpServer?.client?.client.getPrompt({\n        name: promptName,\n      });\n    },\n  });\n}\n"]}