@tambo-ai/react 0.58.1 → 0.60.0

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;AAEnG,iEAS4C;AAG5C,IAAY,YAGX;AAHD,WAAY,YAAY;IACtB,2BAAW,CAAA;IACX,6BAAa,CAAA;AACf,CAAC,EAHW,YAAY,4BAAZ,YAAY,QAGvB;AAgED;;;;;;;;;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,CAAC,OAA0C;QACjE,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,CAAC,OAAuC;QAC3D,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;AAzPD,8BAyPC","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 type { RequestHandlerExtra } from \"@modelcontextprotocol/sdk/shared/protocol.js\";\nimport {\n  ClientNotification,\n  ClientRequest,\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 * Handler for MCP elicitation requests.\n * Receives the elicit request and a RequestHandlerExtra containing an AbortSignal that fires when the request is cancelled.\n * @param request - The elicitation request from the server\n * @param extra - Additional context including AbortSignal for cancellation\n * @returns Promise resolving to the elicitation result\n * @example\n * ```typescript\n * const handler: MCPElicitationHandler = async (request, extra) => {\n *   // Listen for cancellation\n *   extra.signal.addEventListener('abort', () => {\n *     console.log('Request cancelled');\n *   });\n *\n *   // Return user's response\n *   return {\n *     action: 'accept',\n *     content: { name: 'John' }\n *   };\n * };\n * ```\n */\nexport type MCPElicitationHandler = (\n  e: ElicitRequest,\n  extra: RequestHandlerExtra<ClientRequest, ClientNotification>,\n) => Promise<ElicitResult>;\n\n/**\n * Handler for MCP sampling requests (create_message).\n * Receives the sampling request and a RequestHandlerExtra containing an AbortSignal that fires when the request is cancelled.\n * @param request - The sampling/create_message request from the server\n * @param extra - Additional context including AbortSignal for cancellation\n * @returns Promise resolving to the sampling result\n */\nexport type MCPSamplingHandler = (\n  e: CreateMessageRequest,\n  extra: RequestHandlerExtra<ClientRequest, ClientNotification>,\n) => Promise<CreateMessageResult>;\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 (receives request and RequestHandlerExtra with AbortSignal)\n * @param sampling - Handler for sampling requests (receives request and RequestHandlerExtra with AbortSignal)\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, extra) => Promise.resolve({...}),\n *     },\n * );\n * ```\n */\nexport interface MCPHandlers {\n  elicitation: MCPElicitationHandler;\n  sampling: MCPSamplingHandler;\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(handler: MCPElicitationHandler | undefined) {\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(handler: MCPSamplingHandler | undefined) {\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,107 @@
+import { type ListPromptsResult } from "@modelcontextprotocol/sdk/types.js";
+import { UseQueryResult } from "@tanstack/react-query";
+import { type ConnectedMcpServer } from "./tambo-mcp-provider";
+export type ListPromptItem = ListPromptsResult["prompts"][number];
+export interface ListPromptEntry {
+    server: ConnectedMcpServer;
+    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(): {
+    data: ListPromptEntry[];
+    error: Error | null;
+    errors: Error[];
+    isPending: boolean;
+    isSuccess: boolean;
+    isError: boolean;
+    isPaused: boolean;
+    isRefetching: boolean;
+    isFetching: boolean;
+    isLoading: boolean;
+    refetch: () => Promise<void>;
+};
+/**
+ * Hook to get the prompt for the specified name.
+ * @param promptName - The name of the prompt to get. If the prompt won't return anything
+ * @param args - The arguments to pass to the prompt.
+ * @returns The prompt for the specified name.
+ */
+export declare function useTamboMcpPrompt(promptName: string | undefined, args?: Record<string, string>): 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;
+} | null, 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;AAC5C,OAAO,EAAE,cAAc,EAAE,MAAM,uBAAuB,CAAC;AAEvD,OAAO,EACL,KAAK,kBAAkB,EAGxB,MAAM,sBAAsB,CAAC;AAE9B,MAAM,MAAM,cAAc,GAAG,iBAAiB,CAAC,SAAS,CAAC,CAAC,MAAM,CAAC,CAAC;AAClE,MAAM,WAAW,eAAe;IAE9B,MAAM,EAAE,kBAAkB,CAAC;IAC3B,MAAM,EAAE,cAAc,CAAC;CACxB;AAED;;;GAGG;AACH,wBAAgB,qBAAqB;;WA8B5B,KAAK,GAAG,IAAI;YACX,KAAK,EAAE;eACJ,OAAO;eACP,OAAO;aACT,OAAO;cACN,OAAO;kBACH,OAAO;gBACT,OAAO;eACR,OAAO;aACT,MAAM,OAAO,CAAC,IAAI,CAAC;EAb7B;AAwDD;;;;;GAKG;AACH,wBAAgB,iBAAiB,CAC/B,UAAU,EAAE,MAAM,GAAG,SAAS,EAC9B,IAAI,GAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAM;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;iBAiClC"}

package/dist/mcp/mcp-hooks.js

@@ -0,0 +1,103 @@
+"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)();
+    const queries = (0, hooks_1.useTamboQueries)({
+        queries: mcpServers.map((mcpServer) => ({
+            queryKey: ["mcp-prompts", mcpServer.key],
+            // Only run for connected servers that have a client
+            enabled: isConnectedMcpServer(mcpServer),
+            queryFn: async () => {
+                // Fast path: if this server doesn't have a client, skip work
+                if (!isConnectedMcpServer(mcpServer))
+                    return [];
+                const result = await mcpServer.client.client.listPrompts();
+                const prompts = result?.prompts ?? [];
+                const promptsEntries = prompts.map((prompt) => ({
+                    server: mcpServer,
+                    prompt,
+                }));
+                return promptsEntries;
+            },
+        })),
+        combine: (results) => {
+            return combineArrayResults(results);
+        },
+    });
+    return queries;
+}
+// TODO: find a more general place for this
+function combineArrayResults(results) {
+    const errors = results
+        .filter((result) => result.isError)
+        .map((result) => result.error);
+    // Treat queries that are idle (disabled) as non-blocking for aggregate status
+    const enabledish = results.filter((r) => r.fetchStatus !== "idle" || r.isSuccess || r.isError);
+    return {
+        // Prefer flatMap to avoid extra intermediate arrays
+        data: results.flatMap((result) => result.isSuccess && Array.isArray(result.data) ? result.data : []),
+        // Preserve a single error for compatibility and expose the full list for diagnostics
+        error: errors[0] ?? null,
+        errors,
+        isPending: enabledish.some((result) => result.isPending),
+        isSuccess: enabledish.length > 0 && enabledish.every((result) => result.isSuccess),
+        isError: errors.length > 0,
+        isPaused: enabledish.some((result) => result.isPaused),
+        isRefetching: enabledish.some((result) => result.isRefetching),
+        isFetching: enabledish.some((result) => result.isFetching),
+        isLoading: enabledish.some((result) => result.isLoading),
+        // Aggregate refetch to trigger all underlying queries
+        refetch: async () => {
+            await Promise.all(results.map(async (r) => {
+                await r.refetch();
+            }));
+        },
+    };
+}
+// Type guard for narrowing to connected servers
+function isConnectedMcpServer(server) {
+    return "client" in server && server.client != null;
+}
+/**
+ * Hook to get the prompt for the specified name.
+ * @param promptName - The name of the prompt to get. If the prompt won't return anything
+ * @param args - The arguments to pass to the prompt.
+ * @returns The prompt for the specified name.
+ */
+function useTamboMcpPrompt(promptName, args = {}) {
+    // figure out which server has the prompt
+    const { data: promptEntries } = useTamboMcpPromptList();
+    const promptEntry = promptEntries?.find((prompt) => prompt.prompt.name === promptName);
+    // Use the stable server key (and the server instance itself) instead of brittle
+    // name/url/transport matching.
+    const mcpServer = promptEntry?.server;
+    // Canonicalize args to avoid unstable cache keys from object identity/order
+    const sortedArgsEntries = Object.keys(args)
+        .sort()
+        .map((k) => [k, args[k]]);
+    return (0, hooks_1.useTamboQuery)({
+        // Include server identity and sorted args to prevent stale cache hits
+        queryKey: ["mcp-prompt", promptName, mcpServer?.key, sortedArgsEntries],
+        // Only run when we have a prompt name and a connected server
+        enabled: Boolean(promptName && mcpServer && isConnectedMcpServer(mcpServer)),
+        queryFn: async () => {
+            if (!promptName || !mcpServer || !isConnectedMcpServer(mcpServer)) {
+                return null;
+            }
+            const result = await mcpServer.client.client.getPrompt({
+                name: promptName,
+                arguments: args,
+            });
+            return result ?? null; // return null because react-query doesn't like undefined results
+        },
+    });
+}
+//# 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":";;AAuBA,sDA0BC;AA8DD,8CAmCC;AA7ID,oCAA0D;AAC1D,6DAI8B;AAS9B;;;GAGG;AACH,SAAgB,qBAAqB;IACnC,MAAM,UAAU,GAAG,IAAA,uCAAkB,GAAE,CAAC;IACxC,MAAM,OAAO,GAAG,IAAA,uBAAe,EAAC;QAC9B,OAAO,EAAE,UAAU,CAAC,GAAG,CAAC,CAAC,SAAS,EAAE,EAAE,CAAC,CAAC;YACtC,QAAQ,EAAE,CAAC,aAAa,EAAE,SAAS,CAAC,GAAG,CAAC;YACxC,oDAAoD;YACpD,OAAO,EAAE,oBAAoB,CAAC,SAAS,CAAC;YACxC,OAAO,EAAE,KAAK,IAAgC,EAAE;gBAC9C,6DAA6D;gBAC7D,IAAI,CAAC,oBAAoB,CAAC,SAAS,CAAC;oBAAE,OAAO,EAAE,CAAC;gBAEhD,MAAM,MAAM,GAAG,MAAM,SAAS,CAAC,MAAM,CAAC,MAAM,CAAC,WAAW,EAAE,CAAC;gBAC3D,MAAM,OAAO,GAAqB,MAAM,EAAE,OAAO,IAAI,EAAE,CAAC;gBACxD,MAAM,cAAc,GAAG,OAAO,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC;oBAC9C,MAAM,EAAE,SAAS;oBACjB,MAAM;iBACP,CAAC,CAAC,CAAC;gBACJ,OAAO,cAAc,CAAC;YACxB,CAAC;SACF,CAAC,CAAC;QACH,OAAO,EAAE,CAAC,OAAO,EAAE,EAAE;YACnB,OAAO,mBAAmB,CAAC,OAAO,CAAC,CAAC;QACtC,CAAC;KACF,CAAC,CAAC;IAEH,OAAO,OAAO,CAAC;AACjB,CAAC;AACD,2CAA2C;AAC3C,SAAS,mBAAmB,CAAI,OAAqC;IAanE,MAAM,MAAM,GAAG,OAAO;SACnB,MAAM,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,OAAO,CAAC;SAClC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,KAAc,CAAC,CAAC;IAE1C,8EAA8E;IAC9E,MAAM,UAAU,GAAG,OAAO,CAAC,MAAM,CAC/B,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,WAAW,KAAK,MAAM,IAAI,CAAC,CAAC,SAAS,IAAI,CAAC,CAAC,OAAO,CAC5D,CAAC;IAEF,OAAO;QACL,oDAAoD;QACpD,IAAI,EAAE,OAAO,CAAC,OAAO,CAAC,CAAC,MAAM,EAAE,EAAE,CAC/B,MAAM,CAAC,SAAS,IAAI,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAClE;QACD,qFAAqF;QACrF,KAAK,EAAE,MAAM,CAAC,CAAC,CAAC,IAAI,IAAI;QACxB,MAAM;QACN,SAAS,EAAE,UAAU,CAAC,IAAI,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC;QACxD,SAAS,EACP,UAAU,CAAC,MAAM,GAAG,CAAC,IAAI,UAAU,CAAC,KAAK,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC;QACzE,OAAO,EAAE,MAAM,CAAC,MAAM,GAAG,CAAC;QAC1B,QAAQ,EAAE,UAAU,CAAC,IAAI,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,QAAQ,CAAC;QACtD,YAAY,EAAE,UAAU,CAAC,IAAI,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,YAAY,CAAC;QAC9D,UAAU,EAAE,UAAU,CAAC,IAAI,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,UAAU,CAAC;QAC1D,SAAS,EAAE,UAAU,CAAC,IAAI,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC;QACxD,sDAAsD;QACtD,OAAO,EAAE,KAAK,IAAI,EAAE;YAClB,MAAM,OAAO,CAAC,GAAG,CACf,OAAO,CAAC,GAAG,CAAC,KAAK,EAAE,CAAC,EAAE,EAAE;gBACtB,MAAM,CAAC,CAAC,OAAO,EAAE,CAAC;YACpB,CAAC,CAAC,CACH,CAAC;QACJ,CAAC;KACF,CAAC;AACJ,CAAC;AAED,gDAAgD;AAChD,SAAS,oBAAoB,CAAC,MAAiB;IAC7C,OAAO,QAAQ,IAAI,MAAM,IAAI,MAAM,CAAC,MAAM,IAAI,IAAI,CAAC;AACrD,CAAC;AAED;;;;;GAKG;AACH,SAAgB,iBAAiB,CAC/B,UAA8B,EAC9B,OAA+B,EAAE;IAEjC,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,gFAAgF;IAChF,+BAA+B;IAC/B,MAAM,SAAS,GAAG,WAAW,EAAE,MAAM,CAAC;IAEtC,4EAA4E;IAC5E,MAAM,iBAAiB,GAAG,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC;SACxC,IAAI,EAAE;SACN,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC,CAAC,CAAU,CAAC,CAAC;IACrC,OAAO,IAAA,qBAAa,EAAC;QACnB,sEAAsE;QACtE,QAAQ,EAAE,CAAC,YAAY,EAAE,UAAU,EAAE,SAAS,EAAE,GAAG,EAAE,iBAAiB,CAAC;QACvE,6DAA6D;QAC7D,OAAO,EAAE,OAAO,CACd,UAAU,IAAI,SAAS,IAAI,oBAAoB,CAAC,SAAS,CAAC,CAC3D;QACD,OAAO,EAAE,KAAK,IAAqC,EAAE;YACnD,IAAI,CAAC,UAAU,IAAI,CAAC,SAAS,IAAI,CAAC,oBAAoB,CAAC,SAAS,CAAC,EAAE,CAAC;gBAClE,OAAO,IAAI,CAAC;YACd,CAAC;YACD,MAAM,MAAM,GAAG,MAAM,SAAS,CAAC,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC;gBACrD,IAAI,EAAE,UAAU;gBAChB,SAAS,EAAE,IAAI;aAChB,CAAC,CAAC;YACH,OAAO,MAAM,IAAI,IAAI,CAAC,CAAC,iEAAiE;QAC1F,CAAC;KACF,CAAC,CAAC;AACL,CAAC","sourcesContent":["import {\n  GetPromptResult,\n  type ListPromptsResult,\n} from \"@modelcontextprotocol/sdk/types.js\";\nimport { UseQueryResult } from \"@tanstack/react-query\";\nimport { useTamboQueries, useTamboQuery } from \"../hooks\";\nimport {\n  type ConnectedMcpServer,\n  type McpServer,\n  useTamboMcpServers,\n} from \"./tambo-mcp-provider\";\n\nexport type ListPromptItem = ListPromptsResult[\"prompts\"][number];\nexport interface ListPromptEntry {\n  // Only connected servers produce prompt entries, so expose the connected type\n  server: ConnectedMcpServer;\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  const queries = useTamboQueries({\n    queries: mcpServers.map((mcpServer) => ({\n      queryKey: [\"mcp-prompts\", mcpServer.key],\n      // Only run for connected servers that have a client\n      enabled: isConnectedMcpServer(mcpServer),\n      queryFn: async (): Promise<ListPromptEntry[]> => {\n        // Fast path: if this server doesn't have a client, skip work\n        if (!isConnectedMcpServer(mcpServer)) return [];\n\n        const result = await mcpServer.client.client.listPrompts();\n        const prompts: ListPromptItem[] = result?.prompts ?? [];\n        const promptsEntries = prompts.map((prompt) => ({\n          server: mcpServer,\n          prompt,\n        }));\n        return promptsEntries;\n      },\n    })),\n    combine: (results) => {\n      return combineArrayResults(results);\n    },\n  });\n\n  return queries;\n}\n// TODO: find a more general place for this\nfunction combineArrayResults<T>(results: UseQueryResult<T[], Error>[]): {\n  data: T[];\n  error: Error | null;\n  errors: Error[];\n  isPending: boolean;\n  isSuccess: boolean;\n  isError: boolean;\n  isPaused: boolean;\n  isRefetching: boolean;\n  isFetching: boolean;\n  isLoading: boolean;\n  refetch: () => Promise<void>;\n} {\n  const errors = results\n    .filter((result) => result.isError)\n    .map((result) => result.error as Error);\n\n  // Treat queries that are idle (disabled) as non-blocking for aggregate status\n  const enabledish = results.filter(\n    (r) => r.fetchStatus !== \"idle\" || r.isSuccess || r.isError,\n  );\n\n  return {\n    // Prefer flatMap to avoid extra intermediate arrays\n    data: results.flatMap((result) =>\n      result.isSuccess && Array.isArray(result.data) ? result.data : [],\n    ),\n    // Preserve a single error for compatibility and expose the full list for diagnostics\n    error: errors[0] ?? null,\n    errors,\n    isPending: enabledish.some((result) => result.isPending),\n    isSuccess:\n      enabledish.length > 0 && enabledish.every((result) => result.isSuccess),\n    isError: errors.length > 0,\n    isPaused: enabledish.some((result) => result.isPaused),\n    isRefetching: enabledish.some((result) => result.isRefetching),\n    isFetching: enabledish.some((result) => result.isFetching),\n    isLoading: enabledish.some((result) => result.isLoading),\n    // Aggregate refetch to trigger all underlying queries\n    refetch: async () => {\n      await Promise.all(\n        results.map(async (r) => {\n          await r.refetch();\n        }),\n      );\n    },\n  };\n}\n\n// Type guard for narrowing to connected servers\nfunction isConnectedMcpServer(server: McpServer): server is ConnectedMcpServer {\n  return \"client\" in server && server.client != null;\n}\n\n/**\n * Hook to get the prompt for the specified name.\n * @param promptName - The name of the prompt to get. If the prompt won't return anything\n * @param args - The arguments to pass to the prompt.\n * @returns The prompt for the specified name.\n */\nexport function useTamboMcpPrompt(\n  promptName: string | undefined,\n  args: Record<string, string> = {},\n) {\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  // Use the stable server key (and the server instance itself) instead of brittle\n  // name/url/transport matching.\n  const mcpServer = promptEntry?.server;\n\n  // Canonicalize args to avoid unstable cache keys from object identity/order\n  const sortedArgsEntries = Object.keys(args)\n    .sort()\n    .map((k) => [k, args[k]] as const);\n  return useTamboQuery({\n    // Include server identity and sorted args to prevent stale cache hits\n    queryKey: [\"mcp-prompt\", promptName, mcpServer?.key, sortedArgsEntries],\n    // Only run when we have a prompt name and a connected server\n    enabled: Boolean(\n      promptName && mcpServer && isConnectedMcpServer(mcpServer),\n    ),\n    queryFn: async (): Promise<GetPromptResult | null> => {\n      if (!promptName || !mcpServer || !isConnectedMcpServer(mcpServer)) {\n        return null;\n      }\n      const result = await mcpServer.client.client.getPrompt({\n        name: promptName,\n        arguments: args,\n      });\n      return result ?? null; // return null because react-query doesn't like undefined results\n    },\n  });\n}\n"]}