@tambo-ai/react 0.55.0 → 0.57.0

package/dist/mcp/mcp-client.js

@@ -4,6 +4,7 @@ exports.MCPClient = exports.MCPTransport = void 0;
 const index_js_1 = require("@modelcontextprotocol/sdk/client/index.js");
 const sse_js_1 = require("@modelcontextprotocol/sdk/client/sse.js");
 const streamableHttp_js_1 = require("@modelcontextprotocol/sdk/client/streamableHttp.js");
+const types_js_1 = require("@modelcontextprotocol/sdk/types.js");
 var MCPTransport;
 (function (MCPTransport) {
     MCPTransport["SSE"] = "sse";
@@ -27,6 +28,7 @@ class MCPClient {
     endpoint;
     headers;
     authProvider;
+    handlers;
     /**
      * Tracks an in-flight reconnect so concurrent triggers coalesce
      * (single-flight). When set, additional calls to `reconnect()` or
@@ -62,17 +64,19 @@ class MCPClient {
     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
      * @param transportType - The transport to use for the MCP client
      * @param headers - Optional custom headers to include in requests
      */
-    constructor(endpoint, transportType, headers, authProvider, sessionId) {
+    constructor(endpoint, transportType, headers, authProvider, sessionId, handlers = {}) {
         this.endpoint = endpoint;
         this.headers = headers ?? {};
         this.authProvider = authProvider;
         this.transportType = transportType;
+        this.handlers = handlers;
         this.transport = this.initializeTransport(sessionId);
         this.client = this.initializeClient();
     }
@@ -89,8 +93,8 @@ class MCPClient {
      * @returns A connected MCPClient instance ready for use
      * @throws {Error} Will throw an error if connection fails
      */
-    static async create(endpoint, transportType = MCPTransport.HTTP, headers, authProvider, sessionId) {
-        const mcpClient = new MCPClient(endpoint, transportType, headers, authProvider, sessionId);
+    static async create(endpoint, transportType = MCPTransport.HTTP, headers, authProvider, sessionId, handlers = {}) {
+        const mcpClient = new MCPClient(endpoint, transportType, headers, authProvider, sessionId, handlers);
         await mcpClient.client.connect(mcpClient.transport);
         if ("sessionId" in mcpClient.transport) {
             mcpClient.sessionId = mcpClient.transport.sessionId;
@@ -233,12 +237,31 @@ class MCPClient {
             });
         }
     }
+    /**
+     * Initializes the MCP client with the appropriate capabilities and handlers
+     * @returns The initialized MCP client
+     */
     initializeClient() {
+        const elicitationCapability = this.handlers.elicitation
+            ? { elicitation: {} }
+            : {};
+        const samplingCapability = this.handlers.sampling ? { sampling: {} } : {};
         const client = new index_js_1.Client({
             name: "tambo-mcp-client",
             version: "1.0.0",
+        }, {
+            capabilities: {
+                ...elicitationCapability,
+                ...samplingCapability,
+            },
         });
         client.onclose = this.onclose.bind(this);
+        if (this.handlers.elicitation) {
+            client.setRequestHandler(types_js_1.ElicitRequestSchema, this.handlers.elicitation);
+        }
+        if (this.handlers.sampling) {
+            client.setRequestHandler(types_js_1.CreateMessageRequestSchema, this.handlers.sampling);
+        }
         return client;
     }
     /**
@@ -254,7 +277,9 @@ class MCPClient {
         while (hasMore) {
             const response = await this.client.listTools({ cursor }, {});
             allTools.push(...response.tools.map((tool) => {
-                if (tool.inputSchema.type !== "object") {
+                // make sure the right type is used
+                const inputSchemaType = tool.inputSchema.type;
+                if (inputSchemaType !== "object") {
                     throw new Error(`Input schema for tool ${tool.name} is not an object`);
                 }
                 return {
@@ -295,6 +320,40 @@ class MCPClient {
         });
         return result;
     }
+    updateElicitationHandler(handler) {
+        // 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) {
+            throw new Error("Elicitation handler must be set on create");
+        }
+        this.handlers = {
+            ...this.handlers,
+            elicitation: handler,
+        };
+        if (!handler) {
+            const method = types_js_1.ElicitRequestSchema.shape.method.value;
+            this.client.removeRequestHandler(method);
+            return;
+        }
+        this.client.setRequestHandler(types_js_1.ElicitRequestSchema, handler);
+    }
+    updateSamplingHandler(handler) {
+        // 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) {
+            throw new Error("Sampling handler must be set on create");
+        }
+        this.handlers = {
+            ...this.handlers,
+            sampling: handler,
+        };
+        if (!handler) {
+            const method = types_js_1.CreateMessageRequestSchema.shape.method.value;
+            this.client.removeRequestHandler(method);
+            return;
+        }
+        this.client.setRequestHandler(types_js_1.CreateMessageRequestSchema, handler);
+    }
 }
 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;AAGnG,IAAY,YAGX;AAHD,WAAY,YAAY;IACtB,2BAAW,CAAA;IACX,6BAAa,CAAA;AACf,CAAC,EAHW,YAAY,4BAAZ,YAAY,QAGvB;AACD;;;;;;;;;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;IAC3C;;;;;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;IAE3C;;;;;OAKG;IACH,YACE,QAAgB,EAChB,aAA2B,EAC3B,OAAgC,EAChC,YAAkC,EAClC,SAAkB;QAElB,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,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;QAE7B,MAAM,SAAS,GAAG,IAAI,SAAS,CAC7B,QAAQ,EACR,aAAa,EACb,OAAO,EACP,YAAY,EACZ,SAAS,CACV,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;IAEO,gBAAgB;QACtB,MAAM,MAAM,GAAG,IAAI,iBAAM,CAAC;YACxB,IAAI,EAAE,kBAAkB;YACxB,OAAO,EAAE,OAAO;SACjB,CAAC,CAAC;QACH,MAAM,CAAC,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACzC,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,IAAI,IAAI,CAAC,WAAW,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;oBACvC,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;;AAzUH,8BA0UC","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 { JSONSchema7 } from \"json-schema\";\n\nexport enum MCPTransport {\n  SSE = \"sse\",\n  HTTP = \"http\",\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  /**\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  /**\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  ) {\n    this.endpoint = endpoint;\n    this.headers = headers ?? {};\n    this.authProvider = authProvider;\n    this.transportType = transportType;\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  ): Promise<MCPClient> {\n    const mcpClient = new MCPClient(\n      endpoint,\n      transportType,\n      headers,\n      authProvider,\n      sessionId,\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  private initializeClient() {\n    const client = new Client({\n      name: \"tambo-mcp-client\",\n      version: \"1.0.0\",\n    });\n    client.onclose = this.onclose.bind(this);\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          if (tool.inputSchema.type !== \"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\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;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"]}

package/dist/providers/__tests__/tambo-thread-provider-initial-messages.test.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=tambo-thread-provider-initial-messages.test.d.ts.map

package/dist/providers/__tests__/tambo-thread-provider-initial-messages.test.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"tambo-thread-provider-initial-messages.test.d.ts","sourceRoot":"","sources":["../../../src/providers/__tests__/tambo-thread-provider-initial-messages.test.tsx"],"names":[],"mappings":""}

package/dist/providers/__tests__/tambo-thread-provider-initial-messages.test.js

@@ -0,0 +1,179 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+const typescript_sdk_1 = require("@tambo-ai/typescript-sdk");
+const react_1 = require("@testing-library/react");
+const react_2 = __importDefault(require("react"));
+const generate_component_response_1 = require("../../model/generate-component-response");
+const tambo_client_provider_1 = require("../tambo-client-provider");
+const tambo_context_helpers_provider_1 = require("../tambo-context-helpers-provider");
+const tambo_registry_provider_1 = require("../tambo-registry-provider");
+const tambo_thread_provider_1 = require("../tambo-thread-provider");
+// Mock crypto.randomUUID
+Object.defineProperty(global, "crypto", {
+    value: {
+        randomUUID: jest.fn().mockReturnValue("test-uuid"),
+    },
+});
+// Mock the required providers
+jest.mock("../tambo-client-provider", () => ({
+    useTamboClient: jest.fn(),
+    useTamboQueryClient: jest.fn(),
+}));
+jest.mock("@tambo-ai/typescript-sdk", () => ({
+    advanceStream: jest.fn(),
+}));
+// Test utilities
+const createMockMessage = (overrides = {}) => ({
+    id: "test-message-1",
+    content: [{ type: "text", text: "Hello" }],
+    role: "user",
+    threadId: "test-thread-1",
+    createdAt: new Date().toISOString(),
+    componentState: {},
+    ...overrides,
+});
+// Test wrapper
+const createWrapper = (initialMessages = []) => {
+    const TestWrapper = ({ children }) => (react_2.default.createElement(tambo_registry_provider_1.TamboRegistryProvider, { components: [], tools: [] },
+        react_2.default.createElement(tambo_context_helpers_provider_1.TamboContextHelpersProvider, null,
+            react_2.default.createElement(tambo_thread_provider_1.TamboThreadProvider, { initialMessages: initialMessages }, children))));
+    TestWrapper.displayName = "TestWrapper";
+    return TestWrapper;
+};
+describe("TamboThreadProvider with initial messages", () => {
+    const mockClient = {
+        beta: {
+            threads: {
+                advance: jest.fn(),
+                advanceByID: jest.fn(),
+                cancel: jest.fn(),
+                messages: {
+                    create: jest.fn(),
+                },
+            },
+        },
+    };
+    beforeEach(() => {
+        jest.clearAllMocks();
+        tambo_client_provider_1.useTamboClient.mockReturnValue(mockClient);
+        // Provide a minimal mock for the query client used by the provider
+        const mockQueryClient = {
+            setQueryData: jest.fn(),
+            invalidateQueries: jest.fn(),
+        };
+        tambo_client_provider_1.useTamboQueryClient.mockReturnValue(mockQueryClient);
+        typescript_sdk_1.advanceStream.mockImplementation(async function* () {
+            yield {
+                responseMessageDto: {
+                    id: "response-1",
+                    role: "assistant",
+                    content: [{ type: "text", text: "Hello back!" }],
+                    threadId: "new-thread-id",
+                    componentState: {},
+                    createdAt: new Date().toISOString(),
+                },
+                generationStage: generate_component_response_1.GenerationStage.COMPLETE,
+            };
+        });
+    });
+    it("should initialize with empty messages when no initial messages provided", () => {
+        const { result } = (0, react_1.renderHook)(() => (0, tambo_thread_provider_1.useTamboThread)(), {
+            wrapper: createWrapper(),
+        });
+        expect(result.current.thread.messages).toEqual([]);
+    });
+    it("should initialize with provided initial messages", () => {
+        const initialMessages = [
+            createMockMessage({
+                id: "initial-1",
+                role: "system",
+                content: [{ type: "text", text: "You are a helpful assistant." }],
+            }),
+            createMockMessage({
+                id: "initial-2",
+                role: "user",
+                content: [{ type: "text", text: "Hello!" }],
+            }),
+        ];
+        const { result } = (0, react_1.renderHook)(() => (0, tambo_thread_provider_1.useTamboThread)(), {
+            wrapper: createWrapper(initialMessages),
+        });
+        expect(result.current.thread.messages).toHaveLength(2);
+        expect(result.current.thread.messages[0].content[0].text).toBe("You are a helpful assistant.");
+        expect(result.current.thread.messages[1].content[0].text).toBe("Hello!");
+    });
+    it("should include initial messages when sending a message to a new thread", async () => {
+        const initialMessages = [
+            createMockMessage({
+                id: "initial-1",
+                role: "system",
+                content: [{ type: "text", text: "You are a helpful assistant." }],
+            }),
+        ];
+        const { result } = (0, react_1.renderHook)(() => (0, tambo_thread_provider_1.useTamboThread)(), {
+            wrapper: createWrapper(initialMessages),
+        });
+        await (0, react_1.act)(async () => {
+            await result.current.sendThreadMessage("Test message");
+        });
+        // Check that advanceStream was called with initial messages
+        expect(typescript_sdk_1.advanceStream).toHaveBeenCalledWith(mockClient, expect.objectContaining({
+            initialMessages: [
+                {
+                    content: [{ type: "text", text: "You are a helpful assistant." }],
+                    role: "system",
+                    additionalContext: undefined,
+                },
+            ],
+        }), undefined);
+    });
+    it("should not include initial messages when sending to an existing thread", async () => {
+        const initialMessages = [
+            createMockMessage({
+                id: "initial-1",
+                role: "system",
+                content: [{ type: "text", text: "You are a helpful assistant." }],
+            }),
+        ];
+        const { result } = (0, react_1.renderHook)(() => (0, tambo_thread_provider_1.useTamboThread)(), {
+            wrapper: createWrapper(initialMessages),
+        });
+        // Switch to an existing thread first
+        await (0, react_1.act)(async () => {
+            result.current.switchCurrentThread("existing-thread-id", false);
+        });
+        await (0, react_1.act)(async () => {
+            await result.current.sendThreadMessage("Test message");
+        });
+        // Check that advanceStream was called without initial messages
+        expect(typescript_sdk_1.advanceStream).toHaveBeenCalledWith(mockClient, expect.not.objectContaining({
+            initialMessages: expect.anything(),
+        }), "existing-thread-id");
+    });
+    it("should reset to initial messages when starting a new thread", () => {
+        const initialMessages = [
+            createMockMessage({
+                id: "initial-1",
+                role: "system",
+                content: [{ type: "text", text: "You are a helpful assistant." }],
+            }),
+        ];
+        const { result } = (0, react_1.renderHook)(() => (0, tambo_thread_provider_1.useTamboThread)(), {
+            wrapper: createWrapper(initialMessages),
+        });
+        // Switch to an existing thread
+        (0, react_1.act)(() => {
+            result.current.switchCurrentThread("existing-thread-id", false);
+        });
+        // Start a new thread
+        (0, react_1.act)(() => {
+            result.current.startNewThread();
+        });
+        expect(result.current.thread.messages).toHaveLength(1);
+        expect(result.current.thread.messages[0].content[0].text).toBe("You are a helpful assistant.");
+    });
+});
+//# sourceMappingURL=tambo-thread-provider-initial-messages.test.js.map

package/dist/providers/__tests__/tambo-thread-provider-initial-messages.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"tambo-thread-provider-initial-messages.test.js","sourceRoot":"","sources":["../../../src/providers/__tests__/tambo-thread-provider-initial-messages.test.tsx"],"names":[],"mappings":";;;;;AAAA,6DAAkE;AAClE,kDAAyD;AACzD,kDAA0B;AAE1B,yFAGiD;AACjD,oEAA+E;AAC/E,sFAAgF;AAChF,wEAAmE;AACnE,oEAA+E;AAI/E,yBAAyB;AACzB,MAAM,CAAC,cAAc,CAAC,MAAM,EAAE,QAAQ,EAAE;IACtC,KAAK,EAAE;QACL,UAAU,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,WAAW,CAAC;KACnD;CACF,CAAC,CAAC;AAEH,8BAA8B;AAC9B,IAAI,CAAC,IAAI,CAAC,0BAA0B,EAAE,GAAG,EAAE,CAAC,CAAC;IAC3C,cAAc,EAAE,IAAI,CAAC,EAAE,EAAE;IACzB,mBAAmB,EAAE,IAAI,CAAC,EAAE,EAAE;CAC/B,CAAC,CAAC,CAAC;AACJ,IAAI,CAAC,IAAI,CAAC,0BAA0B,EAAE,GAAG,EAAE,CAAC,CAAC;IAC3C,aAAa,EAAE,IAAI,CAAC,EAAE,EAAE;CACzB,CAAC,CAAC,CAAC;AAEJ,iBAAiB;AACjB,MAAM,iBAAiB,GAAG,CACxB,YAAyC,EAAE,EACvB,EAAE,CAAC,CAAC;IACxB,EAAE,EAAE,gBAAgB;IACpB,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC;IAC1C,IAAI,EAAE,MAAM;IACZ,QAAQ,EAAE,eAAe;IACzB,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;IACnC,cAAc,EAAE,EAAE;IAClB,GAAG,SAAS;CACb,CAAC,CAAC;AAEH,eAAe;AACf,MAAM,aAAa,GAAG,CAAC,kBAAwC,EAAE,EAAE,EAAE;IACnE,MAAM,WAAW,GAAG,CAAC,EAAE,QAAQ,EAAiC,EAAE,EAAE,CAAC,CACnE,8BAAC,+CAAqB,IAAC,UAAU,EAAE,EAAE,EAAE,KAAK,EAAE,EAAE;QAC9C,8BAAC,4DAA2B;YAC1B,8BAAC,2CAAmB,IAAC,eAAe,EAAE,eAAe,IAClD,QAAQ,CACW,CACM,CACR,CACzB,CAAC;IACF,WAAW,CAAC,WAAW,GAAG,aAAa,CAAC;IACxC,OAAO,WAAW,CAAC;AACrB,CAAC,CAAC;AAEF,QAAQ,CAAC,2CAA2C,EAAE,GAAG,EAAE;IACzD,MAAM,UAAU,GAAmB;QACjC,IAAI,EAAE;YACJ,OAAO,EAAE;gBACP,OAAO,EAAE,IAAI,CAAC,EAAE,EAAE;gBAClB,WAAW,EAAE,IAAI,CAAC,EAAE,EAAE;gBACtB,MAAM,EAAE,IAAI,CAAC,EAAE,EAAE;gBACjB,QAAQ,EAAE;oBACR,MAAM,EAAE,IAAI,CAAC,EAAE,EAAE;iBAClB;aACF;SACF;KACF,CAAC;IAEF,UAAU,CAAC,GAAG,EAAE;QACd,IAAI,CAAC,aAAa,EAAE,CAAC;QACpB,sCAA4B,CAAC,eAAe,CAAC,UAAU,CAAC,CAAC;QAC1D,mEAAmE;QACnE,MAAM,eAAe,GAAG;YACtB,YAAY,EAAE,IAAI,CAAC,EAAE,EAAE;YACvB,iBAAiB,EAAE,IAAI,CAAC,EAAE,EAAE;SAC7B,CAAC;QACD,2CAAiC,CAAC,eAAe,CAAC,eAAe,CAAC,CAAC;QACnE,8BAA2B,CAAC,kBAAkB,CAAC,KAAK,SAAS,CAAC;YAC7D,MAAM;gBACJ,kBAAkB,EAAE;oBAClB,EAAE,EAAE,YAAY;oBAChB,IAAI,EAAE,WAAW;oBACjB,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,aAAa,EAAE,CAAC;oBAChD,QAAQ,EAAE,eAAe;oBACzB,cAAc,EAAE,EAAE;oBAClB,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;iBACpC;gBACD,eAAe,EAAE,6CAAe,CAAC,QAAQ;aAC1C,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yEAAyE,EAAE,GAAG,EAAE;QACjF,MAAM,EAAE,MAAM,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CAAC,IAAA,sCAAc,GAAE,EAAE;YACpD,OAAO,EAAE,aAAa,EAAE;SACzB,CAAC,CAAC;QAEH,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;IACrD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kDAAkD,EAAE,GAAG,EAAE;QAC1D,MAAM,eAAe,GAAyB;YAC5C,iBAAiB,CAAC;gBAChB,EAAE,EAAE,WAAW;gBACf,IAAI,EAAE,QAAQ;gBACd,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,8BAA8B,EAAE,CAAC;aAClE,CAAC;YACF,iBAAiB,CAAC;gBAChB,EAAE,EAAE,WAAW;gBACf,IAAI,EAAE,MAAM;gBACZ,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC;aAC5C,CAAC;SACH,CAAC;QAEF,MAAM,EAAE,MAAM,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CAAC,IAAA,sCAAc,GAAE,EAAE;YACpD,OAAO,EAAE,aAAa,CAAC,eAAe,CAAC;SACxC,CAAC,CAAC;QAEH,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QACvD,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAC5D,8BAA8B,CAC/B,CAAC;QACF,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;IAC3E,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,wEAAwE,EAAE,KAAK,IAAI,EAAE;QACtF,MAAM,eAAe,GAAyB;YAC5C,iBAAiB,CAAC;gBAChB,EAAE,EAAE,WAAW;gBACf,IAAI,EAAE,QAAQ;gBACd,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,8BAA8B,EAAE,CAAC;aAClE,CAAC;SACH,CAAC;QAEF,MAAM,EAAE,MAAM,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CAAC,IAAA,sCAAc,GAAE,EAAE;YACpD,OAAO,EAAE,aAAa,CAAC,eAAe,CAAC;SACxC,CAAC,CAAC;QAEH,MAAM,IAAA,WAAG,EAAC,KAAK,IAAI,EAAE;YACnB,MAAM,MAAM,CAAC,OAAO,CAAC,iBAAiB,CAAC,cAAc,CAAC,CAAC;QACzD,CAAC,CAAC,CAAC;QAEH,4DAA4D;QAC5D,MAAM,CAAC,8BAAa,CAAC,CAAC,oBAAoB,CACxC,UAAU,EACV,MAAM,CAAC,gBAAgB,CAAC;YACtB,eAAe,EAAE;gBACf;oBACE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,8BAA8B,EAAE,CAAC;oBACjE,IAAI,EAAE,QAAQ;oBACd,iBAAiB,EAAE,SAAS;iBAC7B;aACF;SACF,CAAC,EACF,SAAS,CACV,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,wEAAwE,EAAE,KAAK,IAAI,EAAE;QACtF,MAAM,eAAe,GAAyB;YAC5C,iBAAiB,CAAC;gBAChB,EAAE,EAAE,WAAW;gBACf,IAAI,EAAE,QAAQ;gBACd,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,8BAA8B,EAAE,CAAC;aAClE,CAAC;SACH,CAAC;QAEF,MAAM,EAAE,MAAM,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CAAC,IAAA,sCAAc,GAAE,EAAE;YACpD,OAAO,EAAE,aAAa,CAAC,eAAe,CAAC;SACxC,CAAC,CAAC;QAEH,qCAAqC;QACrC,MAAM,IAAA,WAAG,EAAC,KAAK,IAAI,EAAE;YACnB,MAAM,CAAC,OAAO,CAAC,mBAAmB,CAAC,oBAAoB,EAAE,KAAK,CAAC,CAAC;QAClE,CAAC,CAAC,CAAC;QAEH,MAAM,IAAA,WAAG,EAAC,KAAK,IAAI,EAAE;YACnB,MAAM,MAAM,CAAC,OAAO,CAAC,iBAAiB,CAAC,cAAc,CAAC,CAAC;QACzD,CAAC,CAAC,CAAC;QAEH,+DAA+D;QAC/D,MAAM,CAAC,8BAAa,CAAC,CAAC,oBAAoB,CACxC,UAAU,EACV,MAAM,CAAC,GAAG,CAAC,gBAAgB,CAAC;YAC1B,eAAe,EAAE,MAAM,CAAC,QAAQ,EAAE;SACnC,CAAC,EACF,oBAAoB,CACrB,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6DAA6D,EAAE,GAAG,EAAE;QACrE,MAAM,eAAe,GAAyB;YAC5C,iBAAiB,CAAC;gBAChB,EAAE,EAAE,WAAW;gBACf,IAAI,EAAE,QAAQ;gBACd,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,8BAA8B,EAAE,CAAC;aAClE,CAAC;SACH,CAAC;QAEF,MAAM,EAAE,MAAM,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CAAC,IAAA,sCAAc,GAAE,EAAE;YACpD,OAAO,EAAE,aAAa,CAAC,eAAe,CAAC;SACxC,CAAC,CAAC;QAEH,+BAA+B;QAC/B,IAAA,WAAG,EAAC,GAAG,EAAE;YACP,MAAM,CAAC,OAAO,CAAC,mBAAmB,CAAC,oBAAoB,EAAE,KAAK,CAAC,CAAC;QAClE,CAAC,CAAC,CAAC;QAEH,qBAAqB;QACrB,IAAA,WAAG,EAAC,GAAG,EAAE;YACP,MAAM,CAAC,OAAO,CAAC,cAAc,EAAE,CAAC;QAClC,CAAC,CAAC,CAAC;QAEH,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QACvD,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAC5D,8BAA8B,CAC/B,CAAC;IACJ,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC","sourcesContent":["import TamboAI, { advanceStream } from \"@tambo-ai/typescript-sdk\";\nimport { act, renderHook } from \"@testing-library/react\";\nimport React from \"react\";\nimport { DeepPartial } from \"ts-essentials\";\nimport {\n  GenerationStage,\n  TamboThreadMessage,\n} from \"../../model/generate-component-response\";\nimport { useTamboClient, useTamboQueryClient } from \"../tambo-client-provider\";\nimport { TamboContextHelpersProvider } from \"../tambo-context-helpers-provider\";\nimport { TamboRegistryProvider } from \"../tambo-registry-provider\";\nimport { TamboThreadProvider, useTamboThread } from \"../tambo-thread-provider\";\n\ntype PartialTamboAI = DeepPartial<TamboAI>;\n\n// Mock crypto.randomUUID\nObject.defineProperty(global, \"crypto\", {\n  value: {\n    randomUUID: jest.fn().mockReturnValue(\"test-uuid\"),\n  },\n});\n\n// Mock the required providers\njest.mock(\"../tambo-client-provider\", () => ({\n  useTamboClient: jest.fn(),\n  useTamboQueryClient: jest.fn(),\n}));\njest.mock(\"@tambo-ai/typescript-sdk\", () => ({\n  advanceStream: jest.fn(),\n}));\n\n// Test utilities\nconst createMockMessage = (\n  overrides: Partial<TamboThreadMessage> = {},\n): TamboThreadMessage => ({\n  id: \"test-message-1\",\n  content: [{ type: \"text\", text: \"Hello\" }],\n  role: \"user\",\n  threadId: \"test-thread-1\",\n  createdAt: new Date().toISOString(),\n  componentState: {},\n  ...overrides,\n});\n\n// Test wrapper\nconst createWrapper = (initialMessages: TamboThreadMessage[] = []) => {\n  const TestWrapper = ({ children }: { children: React.ReactNode }) => (\n    <TamboRegistryProvider components={[]} tools={[]}>\n      <TamboContextHelpersProvider>\n        <TamboThreadProvider initialMessages={initialMessages}>\n          {children}\n        </TamboThreadProvider>\n      </TamboContextHelpersProvider>\n    </TamboRegistryProvider>\n  );\n  TestWrapper.displayName = \"TestWrapper\";\n  return TestWrapper;\n};\n\ndescribe(\"TamboThreadProvider with initial messages\", () => {\n  const mockClient: PartialTamboAI = {\n    beta: {\n      threads: {\n        advance: jest.fn(),\n        advanceByID: jest.fn(),\n        cancel: jest.fn(),\n        messages: {\n          create: jest.fn(),\n        },\n      },\n    },\n  };\n\n  beforeEach(() => {\n    jest.clearAllMocks();\n    (useTamboClient as jest.Mock).mockReturnValue(mockClient);\n    // Provide a minimal mock for the query client used by the provider\n    const mockQueryClient = {\n      setQueryData: jest.fn(),\n      invalidateQueries: jest.fn(),\n    };\n    (useTamboQueryClient as jest.Mock).mockReturnValue(mockQueryClient);\n    (advanceStream as jest.Mock).mockImplementation(async function* () {\n      yield {\n        responseMessageDto: {\n          id: \"response-1\",\n          role: \"assistant\",\n          content: [{ type: \"text\", text: \"Hello back!\" }],\n          threadId: \"new-thread-id\",\n          componentState: {},\n          createdAt: new Date().toISOString(),\n        },\n        generationStage: GenerationStage.COMPLETE,\n      };\n    });\n  });\n\n  it(\"should initialize with empty messages when no initial messages provided\", () => {\n    const { result } = renderHook(() => useTamboThread(), {\n      wrapper: createWrapper(),\n    });\n\n    expect(result.current.thread.messages).toEqual([]);\n  });\n\n  it(\"should initialize with provided initial messages\", () => {\n    const initialMessages: TamboThreadMessage[] = [\n      createMockMessage({\n        id: \"initial-1\",\n        role: \"system\",\n        content: [{ type: \"text\", text: \"You are a helpful assistant.\" }],\n      }),\n      createMockMessage({\n        id: \"initial-2\",\n        role: \"user\",\n        content: [{ type: \"text\", text: \"Hello!\" }],\n      }),\n    ];\n\n    const { result } = renderHook(() => useTamboThread(), {\n      wrapper: createWrapper(initialMessages),\n    });\n\n    expect(result.current.thread.messages).toHaveLength(2);\n    expect(result.current.thread.messages[0].content[0].text).toBe(\n      \"You are a helpful assistant.\",\n    );\n    expect(result.current.thread.messages[1].content[0].text).toBe(\"Hello!\");\n  });\n\n  it(\"should include initial messages when sending a message to a new thread\", async () => {\n    const initialMessages: TamboThreadMessage[] = [\n      createMockMessage({\n        id: \"initial-1\",\n        role: \"system\",\n        content: [{ type: \"text\", text: \"You are a helpful assistant.\" }],\n      }),\n    ];\n\n    const { result } = renderHook(() => useTamboThread(), {\n      wrapper: createWrapper(initialMessages),\n    });\n\n    await act(async () => {\n      await result.current.sendThreadMessage(\"Test message\");\n    });\n\n    // Check that advanceStream was called with initial messages\n    expect(advanceStream).toHaveBeenCalledWith(\n      mockClient,\n      expect.objectContaining({\n        initialMessages: [\n          {\n            content: [{ type: \"text\", text: \"You are a helpful assistant.\" }],\n            role: \"system\",\n            additionalContext: undefined,\n          },\n        ],\n      }),\n      undefined,\n    );\n  });\n\n  it(\"should not include initial messages when sending to an existing thread\", async () => {\n    const initialMessages: TamboThreadMessage[] = [\n      createMockMessage({\n        id: \"initial-1\",\n        role: \"system\",\n        content: [{ type: \"text\", text: \"You are a helpful assistant.\" }],\n      }),\n    ];\n\n    const { result } = renderHook(() => useTamboThread(), {\n      wrapper: createWrapper(initialMessages),\n    });\n\n    // Switch to an existing thread first\n    await act(async () => {\n      result.current.switchCurrentThread(\"existing-thread-id\", false);\n    });\n\n    await act(async () => {\n      await result.current.sendThreadMessage(\"Test message\");\n    });\n\n    // Check that advanceStream was called without initial messages\n    expect(advanceStream).toHaveBeenCalledWith(\n      mockClient,\n      expect.not.objectContaining({\n        initialMessages: expect.anything(),\n      }),\n      \"existing-thread-id\",\n    );\n  });\n\n  it(\"should reset to initial messages when starting a new thread\", () => {\n    const initialMessages: TamboThreadMessage[] = [\n      createMockMessage({\n        id: \"initial-1\",\n        role: \"system\",\n        content: [{ type: \"text\", text: \"You are a helpful assistant.\" }],\n      }),\n    ];\n\n    const { result } = renderHook(() => useTamboThread(), {\n      wrapper: createWrapper(initialMessages),\n    });\n\n    // Switch to an existing thread\n    act(() => {\n      result.current.switchCurrentThread(\"existing-thread-id\", false);\n    });\n\n    // Start a new thread\n    act(() => {\n      result.current.startNewThread();\n    });\n\n    expect(result.current.thread.messages).toHaveLength(1);\n    expect(result.current.thread.messages[0].content[0].text).toBe(\n      \"You are a helpful assistant.\",\n    );\n  });\n});\n"]}

package/dist/providers/__tests__/tambo-thread-provider.test.js

@@ -157,9 +157,7 @@ describe("TamboThreadProvider", () => {
         await (0, react_1.act)(async () => {
             await result.current.switchCurrentThread("test-thread-1");
         });
-        expect(mockThreadsApi.retrieve).toHaveBeenCalledWith("test-thread-1", {
-            includeInternal: true,
-        });
+        expect(mockThreadsApi.retrieve).toHaveBeenCalledWith("test-thread-1");
         expect(result.current.thread.id).toBe("test-thread-1");
     });
     it("should start a new thread", async () => {
@@ -843,9 +841,7 @@ describe("TamboThreadProvider", () => {
                 await result.current.switchCurrentThread("existing-thread-123");
             });
             // Verify that the thread retrieval was called
-            expect(mockThreadsApi.retrieve).toHaveBeenCalledWith("existing-thread-123", {
-                includeInternal: true,
-            });
+            expect(mockThreadsApi.retrieve).toHaveBeenCalledWith("existing-thread-123");
             // Verify that neither setQueryData nor refetchQueries were called
             expect(mockQueryClient.setQueryData).not.toHaveBeenCalled();
             expect(mockQueryClient.invalidateQueries).not.toHaveBeenCalled();