@tambo-ai/react 0.58.1 → 0.59.0

package/esm/mcp/tambo-mcp-provider.d.ts

@@ -1,5 +1,5 @@
 import React, { FC } from "react";
-import { MCPClient, MCPTransport } from "./mcp-client";
+import { MCPClient, MCPHandlers, MCPTransport } from "./mcp-client";
 /**
  * Extracts error message from MCP tool result content.
  * Handles both array and string content formats.
@@ -7,12 +7,26 @@ import { MCPClient, MCPTransport } from "./mcp-client";
  * @returns The extracted error message as a string
  */
 export declare function extractErrorMessage(content: unknown): string;
+/**
+ * Configuration for connecting to an MCP server.
+ */
 export interface McpServerInfo {
+    /** Optional name for the MCP server */
     name?: string;
+    /** The URL of the MCP server to connect to */
     url: string;
+    /** Optional description of the MCP server */
     description?: string;
+    /** The transport type to use (SSE or HTTP). Defaults to SSE for string URLs */
     transport?: MCPTransport;
+    /** Optional custom headers to include in requests */
     customHeaders?: Record<string, string>;
+    /**
+     * Optional handlers for elicitation and sampling requests from the server.
+     * Note: These callbacks should be stable (e.g., wrapped in useCallback or defined outside the component)
+     * to avoid constant re-registration of the MCP server on every render.
+     */
+    handlers?: Partial<MCPHandlers>;
 }
 export interface ConnectedMcpServer extends McpServerInfo {
     client: MCPClient;
@@ -22,12 +36,26 @@ export interface FailedMcpServer extends McpServerInfo {
     connectionError: Error;
 }
 export type McpServer = ConnectedMcpServer | FailedMcpServer;
+/**
+ * Provider-level MCP handlers that receive the McpServerInfo as context in addition to the request.
+ * These handlers are applied to all MCP servers unless overridden by per-server handlers.
+ */
+export interface ProviderMCPHandlers {
+    elicitation?: (request: Parameters<MCPHandlers["elicitation"]>[0], serverInfo: McpServerInfo) => ReturnType<MCPHandlers["elicitation"]>;
+    sampling?: (request: Parameters<MCPHandlers["sampling"]>[0], serverInfo: McpServerInfo) => ReturnType<MCPHandlers["sampling"]>;
+}
 /**
  * This provider is used to register tools from MCP servers.
- * @returns the wrapped children
+ * It automatically includes an internal Tambo MCP server when an MCP access token is available.
+ * @param props - The provider props
+ * @param props.mcpServers - Array of MCP server configurations
+ * @param props.handlers - Optional handlers applied to all MCP servers unless overridden per-server
+ * @param props.children - The children to wrap
+ * @returns The TamboMcpProvider component
  */
 export declare const TamboMcpProvider: FC<{
     mcpServers: (McpServerInfo | string)[];
+    handlers?: ProviderMCPHandlers;
     children: React.ReactNode;
 }>;
 /**

package/esm/mcp/tambo-mcp-provider.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"tambo-mcp-provider.d.ts","sourceRoot":"","sources":["../../src/mcp/tambo-mcp-provider.tsx"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,EAEZ,EAAE,EAIH,MAAM,OAAO,CAAC;AAIf,OAAO,EAAE,SAAS,EAAE,YAAY,EAAE,MAAM,cAAc,CAAC;AAEvD;;;;;GAKG;AACH,wBAAgB,mBAAmB,CAAC,OAAO,EAAE,OAAO,GAAG,MAAM,CAsB5D;AAED,MAAM,WAAW,aAAa;IAC5B,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,GAAG,EAAE,MAAM,CAAC;IACZ,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,SAAS,CAAC,EAAE,YAAY,CAAC;IACzB,aAAa,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;CACxC;AAED,MAAM,WAAW,kBAAmB,SAAQ,aAAa;IACvD,MAAM,EAAE,SAAS,CAAC;CACnB;AAED,MAAM,WAAW,eAAgB,SAAQ,aAAa;IACpD,MAAM,CAAC,EAAE,KAAK,CAAC;IACf,eAAe,EAAE,KAAK,CAAC;CACxB;AAED,MAAM,MAAM,SAAS,GAAG,kBAAkB,GAAG,eAAe,CAAC;AAG7D;;;GAGG;AACH,eAAO,MAAM,gBAAgB,EAAE,EAAE,CAAC;IAChC,UAAU,EAAE,CAAC,aAAa,GAAG,MAAM,CAAC,EAAE,CAAC;IACvC,QAAQ,EAAE,KAAK,CAAC,SAAS,CAAC;CAC3B,CAgJA,CAAC;AAEF;;;;;;;;;;;;;;;;;;;;;;GAsBG;AACH,eAAO,MAAM,kBAAkB,mBAE9B,CAAC"}
+{"version":3,"file":"tambo-mcp-provider.d.ts","sourceRoot":"","sources":["../../src/mcp/tambo-mcp-provider.tsx"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,EAEZ,EAAE,EAKH,MAAM,OAAO,CAAC;AAKf,OAAO,EAAE,SAAS,EAAE,WAAW,EAAE,YAAY,EAAE,MAAM,cAAc,CAAC;AAEpE;;;;;GAKG;AACH,wBAAgB,mBAAmB,CAAC,OAAO,EAAE,OAAO,GAAG,MAAM,CAsB5D;AAED;;GAEG;AACH,MAAM,WAAW,aAAa;IAC5B,uCAAuC;IACvC,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,8CAA8C;IAC9C,GAAG,EAAE,MAAM,CAAC;IACZ,6CAA6C;IAC7C,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,+EAA+E;IAC/E,SAAS,CAAC,EAAE,YAAY,CAAC;IACzB,qDAAqD;IACrD,aAAa,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IACvC;;;;OAIG;IACH,QAAQ,CAAC,EAAE,OAAO,CAAC,WAAW,CAAC,CAAC;CACjC;AAED,MAAM,WAAW,kBAAmB,SAAQ,aAAa;IACvD,MAAM,EAAE,SAAS,CAAC;CACnB;AAED,MAAM,WAAW,eAAgB,SAAQ,aAAa;IACpD,MAAM,CAAC,EAAE,KAAK,CAAC;IACf,eAAe,EAAE,KAAK,CAAC;CACxB;AAED,MAAM,MAAM,SAAS,GAAG,kBAAkB,GAAG,eAAe,CAAC;AAE7D;;;GAGG;AACH,MAAM,WAAW,mBAAmB;IAClC,WAAW,CAAC,EAAE,CACZ,OAAO,EAAE,UAAU,CAAC,WAAW,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC,CAAC,EAClD,UAAU,EAAE,aAAa,KACtB,UAAU,CAAC,WAAW,CAAC,aAAa,CAAC,CAAC,CAAC;IAC5C,QAAQ,CAAC,EAAE,CACT,OAAO,EAAE,UAAU,CAAC,WAAW,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,CAAC,EAC/C,UAAU,EAAE,aAAa,KACtB,UAAU,CAAC,WAAW,CAAC,UAAU,CAAC,CAAC,CAAC;CAC1C;AAOD;;;;;;;;GAQG;AACH,eAAO,MAAM,gBAAgB,EAAE,EAAE,CAAC;IAChC,UAAU,EAAE,CAAC,aAAa,GAAG,MAAM,CAAC,EAAE,CAAC;IACvC,QAAQ,CAAC,EAAE,mBAAmB,CAAC;IAC/B,QAAQ,EAAE,KAAK,CAAC,SAAS,CAAC;CAC3B,CA8QA,CAAC;AAEF;;;;;;;;;;;;;;;;;;;;;;GAsBG;AACH,eAAO,MAAM,kBAAkB,mBAE9B,CAAC"}

package/esm/mcp/tambo-mcp-provider.js

@@ -1,5 +1,6 @@
 import { deepEqual } from "fast-equals";
-import React, { createContext, useContext, useEffect, useState, } from "react";
+import React, { createContext, useContext, useEffect, useMemo, useState, } from "react";
+import { useTamboMcpToken } from "../providers/tambo-mcp-token-provider";
 import { useTamboRegistry } from "../providers/tambo-registry-provider";
 import { isContentPartArray, toText } from "../util/content-parts";
 import { MCPClient, MCPTransport } from "./mcp-client";
@@ -27,15 +28,45 @@ export function extractErrorMessage(content) {
     return `${content}`;
 }
 const McpProviderContext = createContext([]);
+// Constant for the internal Tambo MCP server name
+const TAMBO_INTERNAL_MCP_SERVER_NAME = "__tambo_internal_mcp_server__";
 /**
  * This provider is used to register tools from MCP servers.
- * @returns the wrapped children
+ * It automatically includes an internal Tambo MCP server when an MCP access token is available.
+ * @param props - The provider props
+ * @param props.mcpServers - Array of MCP server configurations
+ * @param props.handlers - Optional handlers applied to all MCP servers unless overridden per-server
+ * @param props.children - The children to wrap
+ * @returns The TamboMcpProvider component
  */
-export const TamboMcpProvider = ({ mcpServers, children }) => {
+export const TamboMcpProvider = ({ mcpServers, handlers, children }) => {
     const { registerTool } = useTamboRegistry();
+    const { mcpAccessToken, tamboBaseUrl } = useTamboMcpToken();
+    const providerElicitationHandler = handlers?.elicitation;
+    const providerSamplingHandler = handlers?.sampling;
     const [connectedMcpServers, setConnectedMcpServers] = useState([]);
+    // Combine user-provided MCP servers with the internal Tambo MCP server
+    const allMcpServers = useMemo(() => {
+        const servers = [...mcpServers];
+        // Add internal Tambo MCP server if we have an access token and a base URL
+        if (mcpAccessToken && tamboBaseUrl) {
+            // Build the internal MCP URL robustly, preserving any existing base path
+            const base = new URL(tamboBaseUrl);
+            base.pathname = `${base.pathname.replace(/\/+$/, "")}/mcp`;
+            const tamboMcpUrl = base.toString();
+            servers.push({
+                name: TAMBO_INTERNAL_MCP_SERVER_NAME,
+                url: tamboMcpUrl,
+                transport: MCPTransport.HTTP,
+                customHeaders: {
+                    Authorization: `Bearer ${mcpAccessToken}`,
+                },
+            });
+        }
+        return servers;
+    }, [mcpServers, mcpAccessToken, tamboBaseUrl]);
     useEffect(() => {
-        if (!mcpServers) {
+        if (!allMcpServers.length) {
             return;
         }
         async function registerMcpServers(mcpServerInfos) {
@@ -47,19 +78,48 @@ export const TamboMcpProvider = ({ mcpServers, children }) => {
             // initialize the MCP clients, converting McpServerInfo -> McpServer
             const mcpServers = await Promise.allSettled(mcpServerInfos.map(async (mcpServerInfo) => {
                 try {
+                    // Merge provider handlers with per-server handlers (per-server takes precedence)
+                    const effectiveHandlers = {};
+                    // Apply provider elicitation handler if present and not overridden
+                    if (mcpServerInfo.handlers?.elicitation) {
+                        effectiveHandlers.elicitation =
+                            mcpServerInfo.handlers.elicitation;
+                    }
+                    else if (providerElicitationHandler) {
+                        const elicitationHandler = providerElicitationHandler;
+                        effectiveHandlers.elicitation = async (request) => await elicitationHandler(request, mcpServerInfo);
+                    }
+                    // Apply provider sampling handler if present and not overridden
+                    if (mcpServerInfo.handlers?.sampling) {
+                        effectiveHandlers.sampling = mcpServerInfo.handlers.sampling;
+                    }
+                    else if (providerSamplingHandler) {
+                        const samplingHandler = providerSamplingHandler;
+                        effectiveHandlers.sampling = async (request) => await samplingHandler(request, mcpServerInfo);
+                    }
                     const client = await MCPClient.create(mcpServerInfo.url, mcpServerInfo.transport, mcpServerInfo.customHeaders, undefined, // no oauth support yet
-                    undefined);
+                    undefined, // starting with no session id at first.
+                    effectiveHandlers);
                     const connectedMcpServer = {
                         ...mcpServerInfo,
                         client: client,
                     };
                     // note because the promises may resolve in any order, the resulting
                     // array may not be in the same order as the input array
-                    setConnectedMcpServers((prev) => [
-                        // replace the server if it already exists
-                        ...prev.filter((s) => !equalsMcpServer(s, mcpServerInfo)),
-                        connectedMcpServer,
-                    ]);
+                    setConnectedMcpServers((prev) => {
+                        const existing = prev.find((s) => equalsMcpServer(s, mcpServerInfo));
+                        try {
+                            existing?.client?.close();
+                        }
+                        catch {
+                            // best-effort cleanup
+                        }
+                        return [
+                            // replace the server if it already exists
+                            ...prev.filter((s) => !equalsMcpServer(s, mcpServerInfo)),
+                            connectedMcpServer,
+                        ];
+                    });
                     return connectedMcpServer;
                 }
                 catch (error) {
@@ -69,11 +129,20 @@ export const TamboMcpProvider = ({ mcpServers, children }) => {
                     };
                     // note because the promises may resolve in any order, the resulting
                     // array may not be in the same order as the input array
-                    setConnectedMcpServers((prev) => [
-                        // replace the server if it already exists
-                        ...prev.filter((s) => !equalsMcpServer(s, mcpServerInfo)),
-                        failedMcpServer,
-                    ]);
+                    setConnectedMcpServers((prev) => {
+                        const existing = prev.find((s) => equalsMcpServer(s, mcpServerInfo));
+                        try {
+                            existing?.client?.close();
+                        }
+                        catch {
+                            // best-effort cleanup
+                        }
+                        return [
+                            // replace the server if it already exists
+                            ...prev.filter((s) => !equalsMcpServer(s, mcpServerInfo)),
+                            failedMcpServer,
+                        ];
+                    });
                     return failedMcpServer;
                 }
             }));
@@ -134,11 +203,52 @@ export const TamboMcpProvider = ({ mcpServers, children }) => {
             });
         }
         // normalize the server infos
-        const mcpServerInfos = mcpServers.map((mcpServer) => typeof mcpServer === "string"
+        const mcpServerInfos = allMcpServers.map((mcpServer) => typeof mcpServer === "string"
             ? { url: mcpServer, transport: MCPTransport.SSE }
             : mcpServer);
         registerMcpServers(mcpServerInfos);
-    }, [mcpServers, registerTool]);
+        // eslint-disable-next-line react-hooks/exhaustive-deps
+    }, [allMcpServers, registerTool]);
+    // Update handlers when they change
+    // We depend only on the handler refs and server identities. The linter cannot
+    // infer this shape with optional chaining and derived wrapper functions.
+    useEffect(() => {
+        const mcpServerInfos = allMcpServers.map((mcpServer) => typeof mcpServer === "string"
+            ? { url: mcpServer, transport: MCPTransport.SSE }
+            : mcpServer);
+        connectedMcpServers.forEach((connectedServer) => {
+            if (!connectedServer.client) {
+                // Skip failed servers
+                return;
+            }
+            // Find the matching server info to get the current handlers
+            const serverInfo = mcpServerInfos.find((info) => equalsMcpServer(connectedServer, info));
+            if (!serverInfo) {
+                // Server was removed, will be handled by the main effect
+                return;
+            }
+            // Determine effective elicitation handler (per-server overrides provider)
+            const defaultElicitationHandler = providerElicitationHandler;
+            const effectiveElicitationHandler = serverInfo.handlers?.elicitation ??
+                (defaultElicitationHandler
+                    ? async (request) => await defaultElicitationHandler(request, serverInfo)
+                    : undefined);
+            // Determine effective sampling handler (per-server overrides provider)
+            const defaultSamplingHandler = providerSamplingHandler;
+            const effectiveSamplingHandler = serverInfo.handlers?.sampling ??
+                (defaultSamplingHandler
+                    ? async (request) => await defaultSamplingHandler(request, serverInfo)
+                    : undefined);
+            // Apply handler updates unconditionally so removals propagate
+            connectedServer.client.updateElicitationHandler?.(effectiveElicitationHandler);
+            connectedServer.client.updateSamplingHandler?.(effectiveSamplingHandler);
+        });
+    }, [
+        allMcpServers,
+        connectedMcpServers,
+        providerElicitationHandler,
+        providerSamplingHandler,
+    ]);
     return (React.createElement(McpProviderContext.Provider, { value: connectedMcpServers }, children));
 };
 /**

package/esm/mcp/tambo-mcp-provider.js.map

@@ -1 +1 @@
-{"version":3,"file":"tambo-mcp-provider.js","sourceRoot":"","sources":["../../src/mcp/tambo-mcp-provider.tsx"],"names":[],"mappings":"AAAA,OAAO,EAAE,SAAS,EAAE,MAAM,aAAa,CAAC;AACxC,OAAO,KAAK,EAAE,EACZ,aAAa,EAEb,UAAU,EACV,SAAS,EACT,QAAQ,GACT,MAAM,OAAO,CAAC;AAEf,OAAO,EAAE,gBAAgB,EAAE,MAAM,sCAAsC,CAAC;AACxE,OAAO,EAAE,kBAAkB,EAAE,MAAM,EAAE,MAAM,uBAAuB,CAAC;AACnE,OAAO,EAAE,SAAS,EAAE,YAAY,EAAE,MAAM,cAAc,CAAC;AAEvD;;;;;GAKG;AACH,MAAM,UAAU,mBAAmB,CAAC,OAAgB;IAClD,IAAI,OAAO,KAAK,SAAS,IAAI,OAAO,KAAK,IAAI,EAAE,CAAC;QAC9C,OAAO,wBAAwB,CAAC;IAClC,CAAC;IAED,IAAI,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC;QAC3B,MAAM,SAAS,GAAG,OAAO;aACtB,MAAM,CACL,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,IAAI,IAAI,CAAC,IAAI,KAAK,MAAM,IAAI,OAAO,IAAI,CAAC,IAAI,KAAK,QAAQ,CACxE;aACA,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAE5B,OAAO,SAAS,CAAC,MAAM,GAAG,CAAC;YACzB,CAAC,CAAC,SAAS,CAAC,IAAI,CAAC,GAAG,CAAC;YACrB,CAAC,CAAC,wCAAwC,CAAC;IAC/C,CAAC;IAED,IAAI,OAAO,OAAO,KAAK,QAAQ,EAAE,CAAC;QAChC,OAAO,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;IACjC,CAAC;IAED,OAAO,GAAG,OAAO,EAAE,CAAC;AACtB,CAAC;AAqBD,MAAM,kBAAkB,GAAG,aAAa,CAAc,EAAE,CAAC,CAAC;AAC1D;;;GAGG;AACH,MAAM,CAAC,MAAM,gBAAgB,GAGxB,CAAC,EAAE,UAAU,EAAE,QAAQ,EAAE,EAAE,EAAE;IAChC,MAAM,EAAE,YAAY,EAAE,GAAG,gBAAgB,EAAE,CAAC;IAC5C,MAAM,CAAC,mBAAmB,EAAE,sBAAsB,CAAC,GAAG,QAAQ,CAC5D,EAAE,CACH,CAAC;IAEF,SAAS,CAAC,GAAG,EAAE;QACb,IAAI,CAAC,UAAU,EAAE,CAAC;YAChB,OAAO;QACT,CAAC;QACD,KAAK,UAAU,kBAAkB,CAAC,cAA+B;YAC/D,yDAAyD;YACzD,MAAM,YAAY,GAAG,IAAI,GAAG,EAAqB,CAAC;YAClD,sBAAsB,CAAC,CAAC,IAAI,EAAE,EAAE;YAC9B,kDAAkD;YAClD,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAChB,cAAc,CAAC,IAAI,CAAC,CAAC,aAAa,EAAE,EAAE,CACpC,eAAe,CAAC,CAAC,EAAE,aAAa,CAAC,CAClC,CACF,CACF,CAAC;YAEF,oEAAoE;YACpE,MAAM,UAAU,GAAG,MAAM,OAAO,CAAC,UAAU,CACzC,cAAc,CAAC,GAAG,CAAC,KAAK,EAAE,aAAa,EAAsB,EAAE;gBAC7D,IAAI,CAAC;oBACH,MAAM,MAAM,GAAG,MAAM,SAAS,CAAC,MAAM,CACnC,aAAa,CAAC,GAAG,EACjB,aAAa,CAAC,SAAS,EACvB,aAAa,CAAC,aAAa,EAC3B,SAAS,EAAE,uBAAuB;oBAClC,SAAS,CACV,CAAC;oBACF,MAAM,kBAAkB,GAAG;wBACzB,GAAG,aAAa;wBAChB,MAAM,EAAE,MAAM;qBACf,CAAC;oBACF,oEAAoE;oBACpE,wDAAwD;oBACxD,sBAAsB,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC;wBAC/B,0CAA0C;wBAC1C,GAAG,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,eAAe,CAAC,CAAC,EAAE,aAAa,CAAC,CAAC;wBACzD,kBAAkB;qBACnB,CAAC,CAAC;oBACH,OAAO,kBAAkB,CAAC;gBAC5B,CAAC;gBAAC,OAAO,KAAK,EAAE,CAAC;oBACf,MAAM,eAAe,GAAG;wBACtB,GAAG,aAAa;wBAChB,eAAe,EAAE,KAAc;qBAChC,CAAC;oBACF,oEAAoE;oBACpE,wDAAwD;oBACxD,sBAAsB,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC;wBAC/B,0CAA0C;wBAC1C,GAAG,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,eAAe,CAAC,CAAC,EAAE,aAAa,CAAC,CAAC;wBACzD,eAAe;qBAChB,CAAC,CAAC;oBACH,OAAO,eAAe,CAAC;gBACzB,CAAC;YACH,CAAC,CAAC,CACH,CAAC;YAEF,gCAAgC;YAChC,MAAM,mBAAmB,GAAG,UAAU;iBACnC,MAAM,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,MAAM,KAAK,WAAW,CAAC;iBACjD,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;YAEjC,8CAA8C;YAC9C,MAAM,eAAe,GAAG,mBAAmB,CAAC,GAAG,CAAC,KAAK,EAAE,SAAS,EAAE,EAAE;gBAClE,MAAM,KAAK,GAAG,CAAC,MAAM,SAAS,CAAC,MAAM,EAAE,SAAS,EAAE,CAAC,IAAI,EAAE,CAAC;gBAC1D,KAAK,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,EAAE;oBACrB,YAAY,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,EAAE,SAAS,CAAC,CAAC;gBACzC,CAAC,CAAC,CAAC;gBACH,OAAO,KAAK,CAAC;YACf,CAAC,CAAC,CAAC;YACH,MAAM,WAAW,GAAG,MAAM,OAAO,CAAC,UAAU,CAAC,eAAe,CAAC,CAAC;YAE9D,6DAA6D;YAC7D,MAAM,WAAW,GAAG,WAAW,CAAC,MAAM,CACpC,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,MAAM,KAAK,UAAU,CACzC,CAAC;YACF,IAAI,WAAW,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBAC3B,OAAO,CAAC,KAAK,CACX,4CAA4C,EAC5C,WAAW,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,MAAM,CAAC,CAC3C,CAAC;YACJ,CAAC;YAED,gCAAgC;YAChC,MAAM,QAAQ,GAAG,WAAW;iBACzB,MAAM,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,MAAM,KAAK,WAAW,CAAC;iBACjD,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,KAAK,CAAC;iBAC7B,IAAI,EAAE,CAAC;YACV,QAAQ,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,EAAE;gBACxB,YAAY,CAAC;oBACX,WAAW,EAAE,IAAI,CAAC,WAAW,IAAI,EAAE;oBACnC,IAAI,EAAE,IAAI,CAAC,IAAI;oBACf,IAAI,EAAE,KAAK,EAAE,IAA6B,EAAE,EAAE;wBAC5C,MAAM,SAAS,GAAG,YAAY,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;wBAC9C,IAAI,CAAC,SAAS,EAAE,CAAC;4BACf,sBAAsB;4BACtB,MAAM,IAAI,KAAK,CAAC,uBAAuB,IAAI,CAAC,IAAI,YAAY,CAAC,CAAC;wBAChE,CAAC;wBACD,IAAI,CAAC,SAAS,CAAC,MAAM,EAAE,CAAC;4BACtB,iGAAiG;4BACjG,MAAM,IAAI,KAAK,CACb,uBAAuB,IAAI,CAAC,IAAI,mBAAmB,CACpD,CAAC;wBACJ,CAAC;wBACD,MAAM,MAAM,GAAG,MAAM,SAAS,CAAC,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;wBAChE,IAAI,MAAM,CAAC,OAAO,EAAE,CAAC;4BACnB,MAAM,YAAY,GAAG,mBAAmB,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;4BACzD,MAAM,IAAI,KAAK,CAAC,YAAY,CAAC,CAAC;wBAChC,CAAC;wBACD,OAAO,MAAM,CAAC,OAAO,CAAC;oBACxB,CAAC;oBACD,UAAU,EAAE,IAAI,CAAC,WAAsC;oBACvD,kBAAkB,EAAE,CAAC,OAAgB,EAAE,EAAE;wBACvC,wFAAwF;wBACxF,2DAA2D;wBAC3D,IAAI,kBAAkB,CAAC,OAAO,CAAC,EAAE,CAAC;4BAChC,OAAO,OAAO,CAAC;wBACjB,CAAC;wBACD,OAAO,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;oBACnD,CAAC;iBACF,CAAC,CAAC;YACL,CAAC,CAAC,CAAC;QACL,CAAC;QAED,6BAA6B;QAC7B,MAAM,cAAc,GAAG,UAAU,CAAC,GAAG,CAAC,CAAC,SAAS,EAAE,EAAE,CAClD,OAAO,SAAS,KAAK,QAAQ;YAC3B,CAAC,CAAC,EAAE,GAAG,EAAE,SAAS,EAAE,SAAS,EAAE,YAAY,CAAC,GAAG,EAAE;YACjD,CAAC,CAAC,SAAS,CACd,CAAC;QAEF,kBAAkB,CAAC,cAAc,CAAC,CAAC;IACrC,CAAC,EAAE,CAAC,UAAU,EAAE,YAAY,CAAC,CAAC,CAAC;IAE/B,OAAO,CACL,oBAAC,kBAAkB,CAAC,QAAQ,IAAC,KAAK,EAAE,mBAAmB,IACpD,QAAQ,CACmB,CAC/B,CAAC;AACJ,CAAC,CAAC;AAEF;;;;;;;;;;;;;;;;;;;;;;GAsBG;AACH,MAAM,CAAC,MAAM,kBAAkB,GAAG,GAAG,EAAE;IACrC,OAAO,UAAU,CAAC,kBAAkB,CAAC,CAAC;AACxC,CAAC,CAAC;AACF,SAAS,eAAe,CAAC,CAAY,EAAE,aAA4B;IACjE,OAAO,CACL,CAAC,CAAC,GAAG,KAAK,aAAa,CAAC,GAAG;QAC3B,CAAC,CAAC,SAAS,KAAK,aAAa,CAAC,SAAS;QACvC,SAAS,CAAC,CAAC,CAAC,aAAa,EAAE,aAAa,CAAC,aAAa,CAAC,CACxD,CAAC;AACJ,CAAC","sourcesContent":["import { deepEqual } from \"fast-equals\";\nimport React, {\n  createContext,\n  FC,\n  useContext,\n  useEffect,\n  useState,\n} from \"react\";\nimport { TamboTool } from \"../model/component-metadata\";\nimport { useTamboRegistry } from \"../providers/tambo-registry-provider\";\nimport { isContentPartArray, toText } from \"../util/content-parts\";\nimport { MCPClient, MCPTransport } from \"./mcp-client\";\n\n/**\n * Extracts error message from MCP tool result content.\n * Handles both array and string content formats.\n * Always returns a string, even for invalid/null inputs.\n * @returns The extracted error message as a string\n */\nexport function extractErrorMessage(content: unknown): string {\n  if (content === undefined || content === null) {\n    return \"Unknown error occurred\";\n  }\n\n  if (Array.isArray(content)) {\n    const textItems = content\n      .filter(\n        (item) => item && item.type === \"text\" && typeof item.text === \"string\",\n      )\n      .map((item) => item.text);\n\n    return textItems.length > 0\n      ? textItems.join(\" \")\n      : \"Error occurred but no details provided\";\n  }\n\n  if (typeof content === \"object\") {\n    return JSON.stringify(content);\n  }\n\n  return `${content}`;\n}\n\nexport interface McpServerInfo {\n  name?: string;\n  url: string;\n  description?: string;\n  transport?: MCPTransport;\n  customHeaders?: Record<string, string>;\n}\n\nexport interface ConnectedMcpServer extends McpServerInfo {\n  client: MCPClient;\n}\n\nexport interface FailedMcpServer extends McpServerInfo {\n  client?: never;\n  connectionError: Error;\n}\n\nexport type McpServer = ConnectedMcpServer | FailedMcpServer;\n\nconst McpProviderContext = createContext<McpServer[]>([]);\n/**\n * This provider is used to register tools from MCP servers.\n * @returns the wrapped children\n */\nexport const TamboMcpProvider: FC<{\n  mcpServers: (McpServerInfo | string)[];\n  children: React.ReactNode;\n}> = ({ mcpServers, children }) => {\n  const { registerTool } = useTamboRegistry();\n  const [connectedMcpServers, setConnectedMcpServers] = useState<McpServer[]>(\n    [],\n  );\n\n  useEffect(() => {\n    if (!mcpServers) {\n      return;\n    }\n    async function registerMcpServers(mcpServerInfos: McpServerInfo[]) {\n      // Maps tool names to the MCP client that registered them\n      const mcpServerMap = new Map<string, McpServer>();\n      setConnectedMcpServers((prev) =>\n        // remove any servers that are not in the new list\n        prev.filter((s) =>\n          mcpServerInfos.some((mcpServerInfo) =>\n            equalsMcpServer(s, mcpServerInfo),\n          ),\n        ),\n      );\n\n      // initialize the MCP clients, converting McpServerInfo -> McpServer\n      const mcpServers = await Promise.allSettled(\n        mcpServerInfos.map(async (mcpServerInfo): Promise<McpServer> => {\n          try {\n            const client = await MCPClient.create(\n              mcpServerInfo.url,\n              mcpServerInfo.transport,\n              mcpServerInfo.customHeaders,\n              undefined, // no oauth support yet\n              undefined, // starting with no session id at first.\n            );\n            const connectedMcpServer = {\n              ...mcpServerInfo,\n              client: client,\n            };\n            // note because the promises may resolve in any order, the resulting\n            // array may not be in the same order as the input array\n            setConnectedMcpServers((prev) => [\n              // replace the server if it already exists\n              ...prev.filter((s) => !equalsMcpServer(s, mcpServerInfo)),\n              connectedMcpServer,\n            ]);\n            return connectedMcpServer;\n          } catch (error) {\n            const failedMcpServer = {\n              ...mcpServerInfo,\n              connectionError: error as Error,\n            };\n            // note because the promises may resolve in any order, the resulting\n            // array may not be in the same order as the input array\n            setConnectedMcpServers((prev) => [\n              // replace the server if it already exists\n              ...prev.filter((s) => !equalsMcpServer(s, mcpServerInfo)),\n              failedMcpServer,\n            ]);\n            return failedMcpServer;\n          }\n        }),\n      );\n\n      // note do not rely on the state\n      const connectedMcpServers = mcpServers\n        .filter((result) => result.status === \"fulfilled\")\n        .map((result) => result.value);\n\n      // Now create a map of tool name to MCP client\n      const serverToolLists = connectedMcpServers.map(async (mcpServer) => {\n        const tools = (await mcpServer.client?.listTools()) ?? [];\n        tools.forEach((tool) => {\n          mcpServerMap.set(tool.name, mcpServer);\n        });\n        return tools;\n      });\n      const toolResults = await Promise.allSettled(serverToolLists);\n\n      // Just log the failed tools, we can't do anything about them\n      const failedTools = toolResults.filter(\n        (result) => result.status === \"rejected\",\n      );\n      if (failedTools.length > 0) {\n        console.error(\n          \"Failed to register tools from MCP servers:\",\n          failedTools.map((result) => result.reason),\n        );\n      }\n\n      // Register the successful tools\n      const allTools = toolResults\n        .filter((result) => result.status === \"fulfilled\")\n        .map((result) => result.value)\n        .flat();\n      allTools.forEach((tool) => {\n        registerTool({\n          description: tool.description ?? \"\",\n          name: tool.name,\n          tool: async (args: Record<string, unknown>) => {\n            const mcpServer = mcpServerMap.get(tool.name);\n            if (!mcpServer) {\n              // should never happen\n              throw new Error(`MCP server for tool ${tool.name} not found`);\n            }\n            if (!mcpServer.client) {\n              // this can't actually happen because the tool can't be registered if the server is not connected\n              throw new Error(\n                `MCP server for tool ${tool.name} is not connected`,\n              );\n            }\n            const result = await mcpServer.client.callTool(tool.name, args);\n            if (result.isError) {\n              const errorMessage = extractErrorMessage(result.content);\n              throw new Error(errorMessage);\n            }\n            return result.content;\n          },\n          toolSchema: tool.inputSchema as TamboTool[\"toolSchema\"],\n          transformToContent: (content: unknown) => {\n            // MCP tools can return content in various formats; pass through arrays of content parts\n            // unchanged, otherwise stringify into a text content part.\n            if (isContentPartArray(content)) {\n              return content;\n            }\n            return [{ type: \"text\", text: toText(content) }];\n          },\n        });\n      });\n    }\n\n    // normalize the server infos\n    const mcpServerInfos = mcpServers.map((mcpServer) =>\n      typeof mcpServer === \"string\"\n        ? { url: mcpServer, transport: MCPTransport.SSE }\n        : mcpServer,\n    );\n\n    registerMcpServers(mcpServerInfos);\n  }, [mcpServers, registerTool]);\n\n  return (\n    <McpProviderContext.Provider value={connectedMcpServers}>\n      {children}\n    </McpProviderContext.Provider>\n  );\n};\n\n/**\n * Hook to access the actual MCP servers, as they are connected (or fail to\n * connect).\n *\n * You can call methods on the MCP client that is included in the MCP server\n * object.\n *\n * If the server fails to connect, the `client` property will be `undefined` and\n * the `connectionError` property will be set.\n *\n * For example, to forcibly disconnect and reconnect all MCP servers:\n *\n * ```tsx\n * const mcpServers = useTamboMcpServers();\n * mcpServers.forEach((mcpServer) => {\n *   mcpServer.client?.reconnect();\n * });\n * ```\n *\n * Note that the MCP servers are not guaranteed to be in the same order as the\n * input array, because they are added as they are connected.\n * @returns The MCP servers\n */\nexport const useTamboMcpServers = () => {\n  return useContext(McpProviderContext);\n};\nfunction equalsMcpServer(s: McpServer, mcpServerInfo: McpServerInfo): boolean {\n  return (\n    s.url === mcpServerInfo.url &&\n    s.transport === mcpServerInfo.transport &&\n    deepEqual(s.customHeaders, mcpServerInfo.customHeaders)\n  );\n}\n"]}
+{"version":3,"file":"tambo-mcp-provider.js","sourceRoot":"","sources":["../../src/mcp/tambo-mcp-provider.tsx"],"names":[],"mappings":"AAAA,OAAO,EAAE,SAAS,EAAE,MAAM,aAAa,CAAC;AACxC,OAAO,KAAK,EAAE,EACZ,aAAa,EAEb,UAAU,EACV,SAAS,EACT,OAAO,EACP,QAAQ,GACT,MAAM,OAAO,CAAC;AAEf,OAAO,EAAE,gBAAgB,EAAE,MAAM,uCAAuC,CAAC;AACzE,OAAO,EAAE,gBAAgB,EAAE,MAAM,sCAAsC,CAAC;AACxE,OAAO,EAAE,kBAAkB,EAAE,MAAM,EAAE,MAAM,uBAAuB,CAAC;AACnE,OAAO,EAAE,SAAS,EAAe,YAAY,EAAE,MAAM,cAAc,CAAC;AAEpE;;;;;GAKG;AACH,MAAM,UAAU,mBAAmB,CAAC,OAAgB;IAClD,IAAI,OAAO,KAAK,SAAS,IAAI,OAAO,KAAK,IAAI,EAAE,CAAC;QAC9C,OAAO,wBAAwB,CAAC;IAClC,CAAC;IAED,IAAI,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC;QAC3B,MAAM,SAAS,GAAG,OAAO;aACtB,MAAM,CACL,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,IAAI,IAAI,CAAC,IAAI,KAAK,MAAM,IAAI,OAAO,IAAI,CAAC,IAAI,KAAK,QAAQ,CACxE;aACA,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAE5B,OAAO,SAAS,CAAC,MAAM,GAAG,CAAC;YACzB,CAAC,CAAC,SAAS,CAAC,IAAI,CAAC,GAAG,CAAC;YACrB,CAAC,CAAC,wCAAwC,CAAC;IAC/C,CAAC;IAED,IAAI,OAAO,OAAO,KAAK,QAAQ,EAAE,CAAC;QAChC,OAAO,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;IACjC,CAAC;IAED,OAAO,GAAG,OAAO,EAAE,CAAC;AACtB,CAAC;AAkDD,MAAM,kBAAkB,GAAG,aAAa,CAAc,EAAE,CAAC,CAAC;AAE1D,kDAAkD;AAClD,MAAM,8BAA8B,GAAG,+BAA+B,CAAC;AAEvE;;;;;;;;GAQG;AACH,MAAM,CAAC,MAAM,gBAAgB,GAIxB,CAAC,EAAE,UAAU,EAAE,QAAQ,EAAE,QAAQ,EAAE,EAAE,EAAE;IAC1C,MAAM,EAAE,YAAY,EAAE,GAAG,gBAAgB,EAAE,CAAC;IAC5C,MAAM,EAAE,cAAc,EAAE,YAAY,EAAE,GAAG,gBAAgB,EAAE,CAAC;IAC5D,MAAM,0BAA0B,GAAG,QAAQ,EAAE,WAAW,CAAC;IACzD,MAAM,uBAAuB,GAAG,QAAQ,EAAE,QAAQ,CAAC;IACnD,MAAM,CAAC,mBAAmB,EAAE,sBAAsB,CAAC,GAAG,QAAQ,CAC5D,EAAE,CACH,CAAC;IAEF,uEAAuE;IACvE,MAAM,aAAa,GAAG,OAAO,CAAC,GAAG,EAAE;QACjC,MAAM,OAAO,GAAG,CAAC,GAAG,UAAU,CAAC,CAAC;QAEhC,0EAA0E;QAC1E,IAAI,cAAc,IAAI,YAAY,EAAE,CAAC;YACnC,yEAAyE;YACzE,MAAM,IAAI,GAAG,IAAI,GAAG,CAAC,YAAY,CAAC,CAAC;YACnC,IAAI,CAAC,QAAQ,GAAG,GAAG,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC;YAC3D,MAAM,WAAW,GAAG,IAAI,CAAC,QAAQ,EAAE,CAAC;YACpC,OAAO,CAAC,IAAI,CAAC;gBACX,IAAI,EAAE,8BAA8B;gBACpC,GAAG,EAAE,WAAW;gBAChB,SAAS,EAAE,YAAY,CAAC,IAAI;gBAC5B,aAAa,EAAE;oBACb,aAAa,EAAE,UAAU,cAAc,EAAE;iBAC1C;aACF,CAAC,CAAC;QACL,CAAC;QAED,OAAO,OAAO,CAAC;IACjB,CAAC,EAAE,CAAC,UAAU,EAAE,cAAc,EAAE,YAAY,CAAC,CAAC,CAAC;IAE/C,SAAS,CAAC,GAAG,EAAE;QACb,IAAI,CAAC,aAAa,CAAC,MAAM,EAAE,CAAC;YAC1B,OAAO;QACT,CAAC;QACD,KAAK,UAAU,kBAAkB,CAAC,cAA+B;YAC/D,yDAAyD;YACzD,MAAM,YAAY,GAAG,IAAI,GAAG,EAAqB,CAAC;YAClD,sBAAsB,CAAC,CAAC,IAAI,EAAE,EAAE;YAC9B,kDAAkD;YAClD,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAChB,cAAc,CAAC,IAAI,CAAC,CAAC,aAAa,EAAE,EAAE,CACpC,eAAe,CAAC,CAAC,EAAE,aAAa,CAAC,CAClC,CACF,CACF,CAAC;YAEF,oEAAoE;YACpE,MAAM,UAAU,GAAG,MAAM,OAAO,CAAC,UAAU,CACzC,cAAc,CAAC,GAAG,CAAC,KAAK,EAAE,aAAa,EAAsB,EAAE;gBAC7D,IAAI,CAAC;oBACH,iFAAiF;oBACjF,MAAM,iBAAiB,GAAyB,EAAE,CAAC;oBAEnD,mEAAmE;oBACnE,IAAI,aAAa,CAAC,QAAQ,EAAE,WAAW,EAAE,CAAC;wBACxC,iBAAiB,CAAC,WAAW;4BAC3B,aAAa,CAAC,QAAQ,CAAC,WAAW,CAAC;oBACvC,CAAC;yBAAM,IAAI,0BAA0B,EAAE,CAAC;wBACtC,MAAM,kBAAkB,GAAG,0BAA0B,CAAC;wBACtD,iBAAiB,CAAC,WAAW,GAAG,KAAK,EAAE,OAAO,EAAE,EAAE,CAChD,MAAM,kBAAkB,CAAC,OAAO,EAAE,aAAa,CAAC,CAAC;oBACrD,CAAC;oBAED,gEAAgE;oBAChE,IAAI,aAAa,CAAC,QAAQ,EAAE,QAAQ,EAAE,CAAC;wBACrC,iBAAiB,CAAC,QAAQ,GAAG,aAAa,CAAC,QAAQ,CAAC,QAAQ,CAAC;oBAC/D,CAAC;yBAAM,IAAI,uBAAuB,EAAE,CAAC;wBACnC,MAAM,eAAe,GAAG,uBAAuB,CAAC;wBAChD,iBAAiB,CAAC,QAAQ,GAAG,KAAK,EAAE,OAAO,EAAE,EAAE,CAC7C,MAAM,eAAe,CAAC,OAAO,EAAE,aAAa,CAAC,CAAC;oBAClD,CAAC;oBAED,MAAM,MAAM,GAAG,MAAM,SAAS,CAAC,MAAM,CACnC,aAAa,CAAC,GAAG,EACjB,aAAa,CAAC,SAAS,EACvB,aAAa,CAAC,aAAa,EAC3B,SAAS,EAAE,uBAAuB;oBAClC,SAAS,EAAE,wCAAwC;oBACnD,iBAAiB,CAClB,CAAC;oBACF,MAAM,kBAAkB,GAAG;wBACzB,GAAG,aAAa;wBAChB,MAAM,EAAE,MAAM;qBACf,CAAC;oBACF,oEAAoE;oBACpE,wDAAwD;oBACxD,sBAAsB,CAAC,CAAC,IAAI,EAAE,EAAE;wBAC9B,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAC/B,eAAe,CAAC,CAAC,EAAE,aAAa,CAAC,CAClC,CAAC;wBACF,IAAI,CAAC;4BACH,QAAQ,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC;wBAC5B,CAAC;wBAAC,MAAM,CAAC;4BACP,sBAAsB;wBACxB,CAAC;wBACD,OAAO;4BACL,0CAA0C;4BAC1C,GAAG,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,eAAe,CAAC,CAAC,EAAE,aAAa,CAAC,CAAC;4BACzD,kBAAkB;yBACnB,CAAC;oBACJ,CAAC,CAAC,CAAC;oBACH,OAAO,kBAAkB,CAAC;gBAC5B,CAAC;gBAAC,OAAO,KAAK,EAAE,CAAC;oBACf,MAAM,eAAe,GAAG;wBACtB,GAAG,aAAa;wBAChB,eAAe,EAAE,KAAc;qBAChC,CAAC;oBACF,oEAAoE;oBACpE,wDAAwD;oBACxD,sBAAsB,CAAC,CAAC,IAAI,EAAE,EAAE;wBAC9B,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAC/B,eAAe,CAAC,CAAC,EAAE,aAAa,CAAC,CAClC,CAAC;wBACF,IAAI,CAAC;4BACH,QAAQ,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC;wBAC5B,CAAC;wBAAC,MAAM,CAAC;4BACP,sBAAsB;wBACxB,CAAC;wBACD,OAAO;4BACL,0CAA0C;4BAC1C,GAAG,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,eAAe,CAAC,CAAC,EAAE,aAAa,CAAC,CAAC;4BACzD,eAAe;yBAChB,CAAC;oBACJ,CAAC,CAAC,CAAC;oBACH,OAAO,eAAe,CAAC;gBACzB,CAAC;YACH,CAAC,CAAC,CACH,CAAC;YAEF,gCAAgC;YAChC,MAAM,mBAAmB,GAAG,UAAU;iBACnC,MAAM,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,MAAM,KAAK,WAAW,CAAC;iBACjD,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;YAEjC,8CAA8C;YAC9C,MAAM,eAAe,GAAG,mBAAmB,CAAC,GAAG,CAAC,KAAK,EAAE,SAAS,EAAE,EAAE;gBAClE,MAAM,KAAK,GAAG,CAAC,MAAM,SAAS,CAAC,MAAM,EAAE,SAAS,EAAE,CAAC,IAAI,EAAE,CAAC;gBAC1D,KAAK,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,EAAE;oBACrB,YAAY,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,EAAE,SAAS,CAAC,CAAC;gBACzC,CAAC,CAAC,CAAC;gBACH,OAAO,KAAK,CAAC;YACf,CAAC,CAAC,CAAC;YACH,MAAM,WAAW,GAAG,MAAM,OAAO,CAAC,UAAU,CAAC,eAAe,CAAC,CAAC;YAE9D,6DAA6D;YAC7D,MAAM,WAAW,GAAG,WAAW,CAAC,MAAM,CACpC,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,MAAM,KAAK,UAAU,CACzC,CAAC;YACF,IAAI,WAAW,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBAC3B,OAAO,CAAC,KAAK,CACX,4CAA4C,EAC5C,WAAW,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,MAAM,CAAC,CAC3C,CAAC;YACJ,CAAC;YAED,gCAAgC;YAChC,MAAM,QAAQ,GAAG,WAAW;iBACzB,MAAM,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,MAAM,KAAK,WAAW,CAAC;iBACjD,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,KAAK,CAAC;iBAC7B,IAAI,EAAE,CAAC;YACV,QAAQ,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,EAAE;gBACxB,YAAY,CAAC;oBACX,WAAW,EAAE,IAAI,CAAC,WAAW,IAAI,EAAE;oBACnC,IAAI,EAAE,IAAI,CAAC,IAAI;oBACf,IAAI,EAAE,KAAK,EAAE,IAA6B,EAAE,EAAE;wBAC5C,MAAM,SAAS,GAAG,YAAY,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;wBAC9C,IAAI,CAAC,SAAS,EAAE,CAAC;4BACf,sBAAsB;4BACtB,MAAM,IAAI,KAAK,CAAC,uBAAuB,IAAI,CAAC,IAAI,YAAY,CAAC,CAAC;wBAChE,CAAC;wBACD,IAAI,CAAC,SAAS,CAAC,MAAM,EAAE,CAAC;4BACtB,iGAAiG;4BACjG,MAAM,IAAI,KAAK,CACb,uBAAuB,IAAI,CAAC,IAAI,mBAAmB,CACpD,CAAC;wBACJ,CAAC;wBACD,MAAM,MAAM,GAAG,MAAM,SAAS,CAAC,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;wBAChE,IAAI,MAAM,CAAC,OAAO,EAAE,CAAC;4BACnB,MAAM,YAAY,GAAG,mBAAmB,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;4BACzD,MAAM,IAAI,KAAK,CAAC,YAAY,CAAC,CAAC;wBAChC,CAAC;wBACD,OAAO,MAAM,CAAC,OAAO,CAAC;oBACxB,CAAC;oBACD,UAAU,EAAE,IAAI,CAAC,WAAsC;oBACvD,kBAAkB,EAAE,CAAC,OAAgB,EAAE,EAAE;wBACvC,wFAAwF;wBACxF,2DAA2D;wBAC3D,IAAI,kBAAkB,CAAC,OAAO,CAAC,EAAE,CAAC;4BAChC,OAAO,OAAO,CAAC;wBACjB,CAAC;wBACD,OAAO,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;oBACnD,CAAC;iBACF,CAAC,CAAC;YACL,CAAC,CAAC,CAAC;QACL,CAAC;QAED,6BAA6B;QAC7B,MAAM,cAAc,GAAG,aAAa,CAAC,GAAG,CAAC,CAAC,SAAS,EAAE,EAAE,CACrD,OAAO,SAAS,KAAK,QAAQ;YAC3B,CAAC,CAAC,EAAE,GAAG,EAAE,SAAS,EAAE,SAAS,EAAE,YAAY,CAAC,GAAG,EAAE;YACjD,CAAC,CAAC,SAAS,CACd,CAAC;QAEF,kBAAkB,CAAC,cAAc,CAAC,CAAC;QACnC,uDAAuD;IACzD,CAAC,EAAE,CAAC,aAAa,EAAE,YAAY,CAAC,CAAC,CAAC;IAElC,mCAAmC;IACnC,8EAA8E;IAC9E,yEAAyE;IACzE,SAAS,CAAC,GAAG,EAAE;QACb,MAAM,cAAc,GAAG,aAAa,CAAC,GAAG,CAAC,CAAC,SAAS,EAAE,EAAE,CACrD,OAAO,SAAS,KAAK,QAAQ;YAC3B,CAAC,CAAC,EAAE,GAAG,EAAE,SAAS,EAAE,SAAS,EAAE,YAAY,CAAC,GAAG,EAAE;YACjD,CAAC,CAAC,SAAS,CACd,CAAC;QAEF,mBAAmB,CAAC,OAAO,CAAC,CAAC,eAAe,EAAE,EAAE;YAC9C,IAAI,CAAC,eAAe,CAAC,MAAM,EAAE,CAAC;gBAC5B,sBAAsB;gBACtB,OAAO;YACT,CAAC;YAED,4DAA4D;YAC5D,MAAM,UAAU,GAAG,cAAc,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAC9C,eAAe,CAAC,eAAe,EAAE,IAAI,CAAC,CACvC,CAAC;YAEF,IAAI,CAAC,UAAU,EAAE,CAAC;gBAChB,yDAAyD;gBACzD,OAAO;YACT,CAAC;YAED,0EAA0E;YAC1E,MAAM,yBAAyB,GAAG,0BAA0B,CAAC;YAC7D,MAAM,2BAA2B,GAC/B,UAAU,CAAC,QAAQ,EAAE,WAAW;gBAChC,CAAC,yBAAyB;oBACxB,CAAC,CAAC,KAAK,EAAE,OAAO,EAAE,EAAE,CAChB,MAAM,yBAAyB,CAAC,OAAO,EAAE,UAAU,CAAC;oBACxD,CAAC,CAAC,SAAS,CAAC,CAAC;YAEjB,uEAAuE;YACvE,MAAM,sBAAsB,GAAG,uBAAuB,CAAC;YACvD,MAAM,wBAAwB,GAC5B,UAAU,CAAC,QAAQ,EAAE,QAAQ;gBAC7B,CAAC,sBAAsB;oBACrB,CAAC,CAAC,KAAK,EAAE,OAAO,EAAE,EAAE,CAAC,MAAM,sBAAsB,CAAC,OAAO,EAAE,UAAU,CAAC;oBACtE,CAAC,CAAC,SAAS,CAAC,CAAC;YAEjB,8DAA8D;YAC9D,eAAe,CAAC,MAAM,CAAC,wBAAwB,EAAE,CAC/C,2BAA2B,CAC5B,CAAC;YACF,eAAe,CAAC,MAAM,CAAC,qBAAqB,EAAE,CAAC,wBAAwB,CAAC,CAAC;QAC3E,CAAC,CAAC,CAAC;IACL,CAAC,EAAE;QACD,aAAa;QACb,mBAAmB;QACnB,0BAA0B;QAC1B,uBAAuB;KACxB,CAAC,CAAC;IAEH,OAAO,CACL,oBAAC,kBAAkB,CAAC,QAAQ,IAAC,KAAK,EAAE,mBAAmB,IACpD,QAAQ,CACmB,CAC/B,CAAC;AACJ,CAAC,CAAC;AAEF;;;;;;;;;;;;;;;;;;;;;;GAsBG;AACH,MAAM,CAAC,MAAM,kBAAkB,GAAG,GAAG,EAAE;IACrC,OAAO,UAAU,CAAC,kBAAkB,CAAC,CAAC;AACxC,CAAC,CAAC;AACF,SAAS,eAAe,CAAC,CAAY,EAAE,aAA4B;IACjE,OAAO,CACL,CAAC,CAAC,GAAG,KAAK,aAAa,CAAC,GAAG;QAC3B,CAAC,CAAC,SAAS,KAAK,aAAa,CAAC,SAAS;QACvC,SAAS,CAAC,CAAC,CAAC,aAAa,EAAE,aAAa,CAAC,aAAa,CAAC,CACxD,CAAC;AACJ,CAAC","sourcesContent":["import { deepEqual } from \"fast-equals\";\nimport React, {\n  createContext,\n  FC,\n  useContext,\n  useEffect,\n  useMemo,\n  useState,\n} from \"react\";\nimport { TamboTool } from \"../model/component-metadata\";\nimport { useTamboMcpToken } from \"../providers/tambo-mcp-token-provider\";\nimport { useTamboRegistry } from \"../providers/tambo-registry-provider\";\nimport { isContentPartArray, toText } from \"../util/content-parts\";\nimport { MCPClient, MCPHandlers, MCPTransport } from \"./mcp-client\";\n\n/**\n * Extracts error message from MCP tool result content.\n * Handles both array and string content formats.\n * Always returns a string, even for invalid/null inputs.\n * @returns The extracted error message as a string\n */\nexport function extractErrorMessage(content: unknown): string {\n  if (content === undefined || content === null) {\n    return \"Unknown error occurred\";\n  }\n\n  if (Array.isArray(content)) {\n    const textItems = content\n      .filter(\n        (item) => item && item.type === \"text\" && typeof item.text === \"string\",\n      )\n      .map((item) => item.text);\n\n    return textItems.length > 0\n      ? textItems.join(\" \")\n      : \"Error occurred but no details provided\";\n  }\n\n  if (typeof content === \"object\") {\n    return JSON.stringify(content);\n  }\n\n  return `${content}`;\n}\n\n/**\n * Configuration for connecting to an MCP server.\n */\nexport interface McpServerInfo {\n  /** Optional name for the MCP server */\n  name?: string;\n  /** The URL of the MCP server to connect to */\n  url: string;\n  /** Optional description of the MCP server */\n  description?: string;\n  /** The transport type to use (SSE or HTTP). Defaults to SSE for string URLs */\n  transport?: MCPTransport;\n  /** Optional custom headers to include in requests */\n  customHeaders?: Record<string, string>;\n  /**\n   * Optional handlers for elicitation and sampling requests from the server.\n   * Note: These callbacks should be stable (e.g., wrapped in useCallback or defined outside the component)\n   * to avoid constant re-registration of the MCP server on every render.\n   */\n  handlers?: Partial<MCPHandlers>;\n}\n\nexport interface ConnectedMcpServer extends McpServerInfo {\n  client: MCPClient;\n}\n\nexport interface FailedMcpServer extends McpServerInfo {\n  client?: never;\n  connectionError: Error;\n}\n\nexport type McpServer = ConnectedMcpServer | FailedMcpServer;\n\n/**\n * Provider-level MCP handlers that receive the McpServerInfo as context in addition to the request.\n * These handlers are applied to all MCP servers unless overridden by per-server handlers.\n */\nexport interface ProviderMCPHandlers {\n  elicitation?: (\n    request: Parameters<MCPHandlers[\"elicitation\"]>[0],\n    serverInfo: McpServerInfo,\n  ) => ReturnType<MCPHandlers[\"elicitation\"]>;\n  sampling?: (\n    request: Parameters<MCPHandlers[\"sampling\"]>[0],\n    serverInfo: McpServerInfo,\n  ) => ReturnType<MCPHandlers[\"sampling\"]>;\n}\n\nconst McpProviderContext = createContext<McpServer[]>([]);\n\n// Constant for the internal Tambo MCP server name\nconst TAMBO_INTERNAL_MCP_SERVER_NAME = \"__tambo_internal_mcp_server__\";\n\n/**\n * This provider is used to register tools from MCP servers.\n * It automatically includes an internal Tambo MCP server when an MCP access token is available.\n * @param props - The provider props\n * @param props.mcpServers - Array of MCP server configurations\n * @param props.handlers - Optional handlers applied to all MCP servers unless overridden per-server\n * @param props.children - The children to wrap\n * @returns The TamboMcpProvider component\n */\nexport const TamboMcpProvider: FC<{\n  mcpServers: (McpServerInfo | string)[];\n  handlers?: ProviderMCPHandlers;\n  children: React.ReactNode;\n}> = ({ mcpServers, handlers, children }) => {\n  const { registerTool } = useTamboRegistry();\n  const { mcpAccessToken, tamboBaseUrl } = useTamboMcpToken();\n  const providerElicitationHandler = handlers?.elicitation;\n  const providerSamplingHandler = handlers?.sampling;\n  const [connectedMcpServers, setConnectedMcpServers] = useState<McpServer[]>(\n    [],\n  );\n\n  // Combine user-provided MCP servers with the internal Tambo MCP server\n  const allMcpServers = useMemo(() => {\n    const servers = [...mcpServers];\n\n    // Add internal Tambo MCP server if we have an access token and a base URL\n    if (mcpAccessToken && tamboBaseUrl) {\n      // Build the internal MCP URL robustly, preserving any existing base path\n      const base = new URL(tamboBaseUrl);\n      base.pathname = `${base.pathname.replace(/\\/+$/, \"\")}/mcp`;\n      const tamboMcpUrl = base.toString();\n      servers.push({\n        name: TAMBO_INTERNAL_MCP_SERVER_NAME,\n        url: tamboMcpUrl,\n        transport: MCPTransport.HTTP,\n        customHeaders: {\n          Authorization: `Bearer ${mcpAccessToken}`,\n        },\n      });\n    }\n\n    return servers;\n  }, [mcpServers, mcpAccessToken, tamboBaseUrl]);\n\n  useEffect(() => {\n    if (!allMcpServers.length) {\n      return;\n    }\n    async function registerMcpServers(mcpServerInfos: McpServerInfo[]) {\n      // Maps tool names to the MCP client that registered them\n      const mcpServerMap = new Map<string, McpServer>();\n      setConnectedMcpServers((prev) =>\n        // remove any servers that are not in the new list\n        prev.filter((s) =>\n          mcpServerInfos.some((mcpServerInfo) =>\n            equalsMcpServer(s, mcpServerInfo),\n          ),\n        ),\n      );\n\n      // initialize the MCP clients, converting McpServerInfo -> McpServer\n      const mcpServers = await Promise.allSettled(\n        mcpServerInfos.map(async (mcpServerInfo): Promise<McpServer> => {\n          try {\n            // Merge provider handlers with per-server handlers (per-server takes precedence)\n            const effectiveHandlers: Partial<MCPHandlers> = {};\n\n            // Apply provider elicitation handler if present and not overridden\n            if (mcpServerInfo.handlers?.elicitation) {\n              effectiveHandlers.elicitation =\n                mcpServerInfo.handlers.elicitation;\n            } else if (providerElicitationHandler) {\n              const elicitationHandler = providerElicitationHandler;\n              effectiveHandlers.elicitation = async (request) =>\n                await elicitationHandler(request, mcpServerInfo);\n            }\n\n            // Apply provider sampling handler if present and not overridden\n            if (mcpServerInfo.handlers?.sampling) {\n              effectiveHandlers.sampling = mcpServerInfo.handlers.sampling;\n            } else if (providerSamplingHandler) {\n              const samplingHandler = providerSamplingHandler;\n              effectiveHandlers.sampling = async (request) =>\n                await samplingHandler(request, mcpServerInfo);\n            }\n\n            const client = await MCPClient.create(\n              mcpServerInfo.url,\n              mcpServerInfo.transport,\n              mcpServerInfo.customHeaders,\n              undefined, // no oauth support yet\n              undefined, // starting with no session id at first.\n              effectiveHandlers,\n            );\n            const connectedMcpServer = {\n              ...mcpServerInfo,\n              client: client,\n            };\n            // note because the promises may resolve in any order, the resulting\n            // array may not be in the same order as the input array\n            setConnectedMcpServers((prev) => {\n              const existing = prev.find((s) =>\n                equalsMcpServer(s, mcpServerInfo),\n              );\n              try {\n                existing?.client?.close();\n              } catch {\n                // best-effort cleanup\n              }\n              return [\n                // replace the server if it already exists\n                ...prev.filter((s) => !equalsMcpServer(s, mcpServerInfo)),\n                connectedMcpServer,\n              ];\n            });\n            return connectedMcpServer;\n          } catch (error) {\n            const failedMcpServer = {\n              ...mcpServerInfo,\n              connectionError: error as Error,\n            };\n            // note because the promises may resolve in any order, the resulting\n            // array may not be in the same order as the input array\n            setConnectedMcpServers((prev) => {\n              const existing = prev.find((s) =>\n                equalsMcpServer(s, mcpServerInfo),\n              );\n              try {\n                existing?.client?.close();\n              } catch {\n                // best-effort cleanup\n              }\n              return [\n                // replace the server if it already exists\n                ...prev.filter((s) => !equalsMcpServer(s, mcpServerInfo)),\n                failedMcpServer,\n              ];\n            });\n            return failedMcpServer;\n          }\n        }),\n      );\n\n      // note do not rely on the state\n      const connectedMcpServers = mcpServers\n        .filter((result) => result.status === \"fulfilled\")\n        .map((result) => result.value);\n\n      // Now create a map of tool name to MCP client\n      const serverToolLists = connectedMcpServers.map(async (mcpServer) => {\n        const tools = (await mcpServer.client?.listTools()) ?? [];\n        tools.forEach((tool) => {\n          mcpServerMap.set(tool.name, mcpServer);\n        });\n        return tools;\n      });\n      const toolResults = await Promise.allSettled(serverToolLists);\n\n      // Just log the failed tools, we can't do anything about them\n      const failedTools = toolResults.filter(\n        (result) => result.status === \"rejected\",\n      );\n      if (failedTools.length > 0) {\n        console.error(\n          \"Failed to register tools from MCP servers:\",\n          failedTools.map((result) => result.reason),\n        );\n      }\n\n      // Register the successful tools\n      const allTools = toolResults\n        .filter((result) => result.status === \"fulfilled\")\n        .map((result) => result.value)\n        .flat();\n      allTools.forEach((tool) => {\n        registerTool({\n          description: tool.description ?? \"\",\n          name: tool.name,\n          tool: async (args: Record<string, unknown>) => {\n            const mcpServer = mcpServerMap.get(tool.name);\n            if (!mcpServer) {\n              // should never happen\n              throw new Error(`MCP server for tool ${tool.name} not found`);\n            }\n            if (!mcpServer.client) {\n              // this can't actually happen because the tool can't be registered if the server is not connected\n              throw new Error(\n                `MCP server for tool ${tool.name} is not connected`,\n              );\n            }\n            const result = await mcpServer.client.callTool(tool.name, args);\n            if (result.isError) {\n              const errorMessage = extractErrorMessage(result.content);\n              throw new Error(errorMessage);\n            }\n            return result.content;\n          },\n          toolSchema: tool.inputSchema as TamboTool[\"toolSchema\"],\n          transformToContent: (content: unknown) => {\n            // MCP tools can return content in various formats; pass through arrays of content parts\n            // unchanged, otherwise stringify into a text content part.\n            if (isContentPartArray(content)) {\n              return content;\n            }\n            return [{ type: \"text\", text: toText(content) }];\n          },\n        });\n      });\n    }\n\n    // normalize the server infos\n    const mcpServerInfos = allMcpServers.map((mcpServer) =>\n      typeof mcpServer === \"string\"\n        ? { url: mcpServer, transport: MCPTransport.SSE }\n        : mcpServer,\n    );\n\n    registerMcpServers(mcpServerInfos);\n    // eslint-disable-next-line react-hooks/exhaustive-deps\n  }, [allMcpServers, registerTool]);\n\n  // Update handlers when they change\n  // We depend only on the handler refs and server identities. The linter cannot\n  // infer this shape with optional chaining and derived wrapper functions.\n  useEffect(() => {\n    const mcpServerInfos = allMcpServers.map((mcpServer) =>\n      typeof mcpServer === \"string\"\n        ? { url: mcpServer, transport: MCPTransport.SSE }\n        : mcpServer,\n    );\n\n    connectedMcpServers.forEach((connectedServer) => {\n      if (!connectedServer.client) {\n        // Skip failed servers\n        return;\n      }\n\n      // Find the matching server info to get the current handlers\n      const serverInfo = mcpServerInfos.find((info) =>\n        equalsMcpServer(connectedServer, info),\n      );\n\n      if (!serverInfo) {\n        // Server was removed, will be handled by the main effect\n        return;\n      }\n\n      // Determine effective elicitation handler (per-server overrides provider)\n      const defaultElicitationHandler = providerElicitationHandler;\n      const effectiveElicitationHandler =\n        serverInfo.handlers?.elicitation ??\n        (defaultElicitationHandler\n          ? async (request) =>\n              await defaultElicitationHandler(request, serverInfo)\n          : undefined);\n\n      // Determine effective sampling handler (per-server overrides provider)\n      const defaultSamplingHandler = providerSamplingHandler;\n      const effectiveSamplingHandler =\n        serverInfo.handlers?.sampling ??\n        (defaultSamplingHandler\n          ? async (request) => await defaultSamplingHandler(request, serverInfo)\n          : undefined);\n\n      // Apply handler updates unconditionally so removals propagate\n      connectedServer.client.updateElicitationHandler?.(\n        effectiveElicitationHandler,\n      );\n      connectedServer.client.updateSamplingHandler?.(effectiveSamplingHandler);\n    });\n  }, [\n    allMcpServers,\n    connectedMcpServers,\n    providerElicitationHandler,\n    providerSamplingHandler,\n  ]);\n\n  return (\n    <McpProviderContext.Provider value={connectedMcpServers}>\n      {children}\n    </McpProviderContext.Provider>\n  );\n};\n\n/**\n * Hook to access the actual MCP servers, as they are connected (or fail to\n * connect).\n *\n * You can call methods on the MCP client that is included in the MCP server\n * object.\n *\n * If the server fails to connect, the `client` property will be `undefined` and\n * the `connectionError` property will be set.\n *\n * For example, to forcibly disconnect and reconnect all MCP servers:\n *\n * ```tsx\n * const mcpServers = useTamboMcpServers();\n * mcpServers.forEach((mcpServer) => {\n *   mcpServer.client?.reconnect();\n * });\n * ```\n *\n * Note that the MCP servers are not guaranteed to be in the same order as the\n * input array, because they are added as they are connected.\n * @returns The MCP servers\n */\nexport const useTamboMcpServers = () => {\n  return useContext(McpProviderContext);\n};\nfunction equalsMcpServer(s: McpServer, mcpServerInfo: McpServerInfo): boolean {\n  return (\n    s.url === mcpServerInfo.url &&\n    s.transport === mcpServerInfo.transport &&\n    deepEqual(s.customHeaders, mcpServerInfo.customHeaders)\n  );\n}\n"]}

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

@@ -4,6 +4,7 @@ import React from "react";
 import { GenerationStage, } from "../../model/generate-component-response";
 import { useTamboClient, useTamboQueryClient } from "../tambo-client-provider";
 import { TamboContextHelpersProvider } from "../tambo-context-helpers-provider";
+import { TamboMcpTokenProvider } from "../tambo-mcp-token-provider";
 import { TamboRegistryProvider } from "../tambo-registry-provider";
 import { TamboThreadProvider, useTamboThread } from "../tambo-thread-provider";
 // Mock crypto.randomUUID
@@ -34,7 +35,8 @@ const createMockMessage = (overrides = {}) => ({
 const createWrapper = (initialMessages = []) => {
     const TestWrapper = ({ children }) => (React.createElement(TamboRegistryProvider, { components: [], tools: [] },
         React.createElement(TamboContextHelpersProvider, null,
-            React.createElement(TamboThreadProvider, { initialMessages: initialMessages }, children))));
+            React.createElement(TamboMcpTokenProvider, null,
+                React.createElement(TamboThreadProvider, { initialMessages: initialMessages }, children)))));
     TestWrapper.displayName = "TestWrapper";
     return TestWrapper;
 };

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

@@ -1 +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,OAAgB,EAAE,aAAa,EAAE,MAAM,0BAA0B,CAAC;AAClE,OAAO,EAAE,GAAG,EAAE,UAAU,EAAE,MAAM,wBAAwB,CAAC;AACzD,OAAO,KAAK,MAAM,OAAO,CAAC;AAE1B,OAAO,EACL,eAAe,GAEhB,MAAM,yCAAyC,CAAC;AACjD,OAAO,EAAE,cAAc,EAAE,mBAAmB,EAAE,MAAM,0BAA0B,CAAC;AAC/E,OAAO,EAAE,2BAA2B,EAAE,MAAM,mCAAmC,CAAC;AAChF,OAAO,EAAE,qBAAqB,EAAE,MAAM,4BAA4B,CAAC;AACnE,OAAO,EAAE,mBAAmB,EAAE,cAAc,EAAE,MAAM,0BAA0B,CAAC;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,oBAAC,qBAAqB,IAAC,UAAU,EAAE,EAAE,EAAE,KAAK,EAAE,EAAE;QAC9C,oBAAC,2BAA2B;YAC1B,oBAAC,mBAAmB,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,cAA4B,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,mBAAiC,CAAC,eAAe,CAAC,eAAe,CAAC,CAAC;QACnE,aAA2B,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,eAAe,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,UAAU,CAAC,GAAG,EAAE,CAAC,cAAc,EAAE,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,UAAU,CAAC,GAAG,EAAE,CAAC,cAAc,EAAE,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,UAAU,CAAC,GAAG,EAAE,CAAC,cAAc,EAAE,EAAE;YACpD,OAAO,EAAE,aAAa,CAAC,eAAe,CAAC;SACxC,CAAC,CAAC;QAEH,MAAM,GAAG,CAAC,KAAK,IAAI,EAAE;YACnB,MAAM,MAAM,CAAC,OAAO,CAAC,iBAAiB,CAAC,cAAc,CAAC,CAAC;QACzD,CAAC,CAAC,CAAC;QAEH,4DAA4D;QAC5D,MAAM,CAAC,aAAa,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,UAAU,CAAC,GAAG,EAAE,CAAC,cAAc,EAAE,EAAE;YACpD,OAAO,EAAE,aAAa,CAAC,eAAe,CAAC;SACxC,CAAC,CAAC;QAEH,qCAAqC;QACrC,MAAM,GAAG,CAAC,KAAK,IAAI,EAAE;YACnB,MAAM,CAAC,OAAO,CAAC,mBAAmB,CAAC,oBAAoB,EAAE,KAAK,CAAC,CAAC;QAClE,CAAC,CAAC,CAAC;QAEH,MAAM,GAAG,CAAC,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,aAAa,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,UAAU,CAAC,GAAG,EAAE,CAAC,cAAc,EAAE,EAAE;YACpD,OAAO,EAAE,aAAa,CAAC,eAAe,CAAC;SACxC,CAAC,CAAC;QAEH,+BAA+B;QAC/B,GAAG,CAAC,GAAG,EAAE;YACP,MAAM,CAAC,OAAO,CAAC,mBAAmB,CAAC,oBAAoB,EAAE,KAAK,CAAC,CAAC;QAClE,CAAC,CAAC,CAAC;QAEH,qBAAqB;QACrB,GAAG,CAAC,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"]}
+{"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,OAAgB,EAAE,aAAa,EAAE,MAAM,0BAA0B,CAAC;AAClE,OAAO,EAAE,GAAG,EAAE,UAAU,EAAE,MAAM,wBAAwB,CAAC;AACzD,OAAO,KAAK,MAAM,OAAO,CAAC;AAE1B,OAAO,EACL,eAAe,GAEhB,MAAM,yCAAyC,CAAC;AACjD,OAAO,EAAE,cAAc,EAAE,mBAAmB,EAAE,MAAM,0BAA0B,CAAC;AAC/E,OAAO,EAAE,2BAA2B,EAAE,MAAM,mCAAmC,CAAC;AAChF,OAAO,EAAE,qBAAqB,EAAE,MAAM,6BAA6B,CAAC;AACpE,OAAO,EAAE,qBAAqB,EAAE,MAAM,4BAA4B,CAAC;AACnE,OAAO,EAAE,mBAAmB,EAAE,cAAc,EAAE,MAAM,0BAA0B,CAAC;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,oBAAC,qBAAqB,IAAC,UAAU,EAAE,EAAE,EAAE,KAAK,EAAE,EAAE;QAC9C,oBAAC,2BAA2B;YAC1B,oBAAC,qBAAqB;gBACpB,oBAAC,mBAAmB,IAAC,eAAe,EAAE,eAAe,IAClD,QAAQ,CACW,CACA,CACI,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,cAA4B,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,mBAAiC,CAAC,eAAe,CAAC,eAAe,CAAC,CAAC;QACnE,aAA2B,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,eAAe,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,UAAU,CAAC,GAAG,EAAE,CAAC,cAAc,EAAE,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,UAAU,CAAC,GAAG,EAAE,CAAC,cAAc,EAAE,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,UAAU,CAAC,GAAG,EAAE,CAAC,cAAc,EAAE,EAAE;YACpD,OAAO,EAAE,aAAa,CAAC,eAAe,CAAC;SACxC,CAAC,CAAC;QAEH,MAAM,GAAG,CAAC,KAAK,IAAI,EAAE;YACnB,MAAM,MAAM,CAAC,OAAO,CAAC,iBAAiB,CAAC,cAAc,CAAC,CAAC;QACzD,CAAC,CAAC,CAAC;QAEH,4DAA4D;QAC5D,MAAM,CAAC,aAAa,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,UAAU,CAAC,GAAG,EAAE,CAAC,cAAc,EAAE,EAAE;YACpD,OAAO,EAAE,aAAa,CAAC,eAAe,CAAC;SACxC,CAAC,CAAC;QAEH,qCAAqC;QACrC,MAAM,GAAG,CAAC,KAAK,IAAI,EAAE;YACnB,MAAM,CAAC,OAAO,CAAC,mBAAmB,CAAC,oBAAoB,EAAE,KAAK,CAAC,CAAC;QAClE,CAAC,CAAC,CAAC;QAEH,MAAM,GAAG,CAAC,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,aAAa,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,UAAU,CAAC,GAAG,EAAE,CAAC,cAAc,EAAE,EAAE;YACpD,OAAO,EAAE,aAAa,CAAC,eAAe,CAAC;SACxC,CAAC,CAAC;QAEH,+BAA+B;QAC/B,GAAG,CAAC,GAAG,EAAE;YACP,MAAM,CAAC,OAAO,CAAC,mBAAmB,CAAC,oBAAoB,EAAE,KAAK,CAAC,CAAC;QAClE,CAAC,CAAC,CAAC;QAEH,qBAAqB;QACrB,GAAG,CAAC,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 { TamboMcpTokenProvider } from \"../tambo-mcp-token-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        <TamboMcpTokenProvider>\n          <TamboThreadProvider initialMessages={initialMessages}>\n            {children}\n          </TamboThreadProvider>\n        </TamboMcpTokenProvider>\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/esm/providers/__tests__/tambo-thread-provider.test.js

@@ -6,6 +6,7 @@ import { GenerationStage, } from "../../model/generate-component-response";
 import { serializeRegistry } from "../../testing/tools";
 import { useTamboClient, useTamboQueryClient } from "../tambo-client-provider";
 import { TamboContextHelpersProvider } from "../tambo-context-helpers-provider";
+import { TamboMcpTokenProvider } from "../tambo-mcp-token-provider";
 import { TamboRegistryProvider } from "../tambo-registry-provider";
 import { TamboThreadProvider, useTamboThread } from "../tambo-thread-provider";
 // Mock crypto.randomUUID
@@ -111,7 +112,8 @@ describe("TamboThreadProvider", () => {
                 currentTimeContextHelper: () => null,
                 currentPageContextHelper: () => null,
             } },
-            React.createElement(TamboThreadProvider, { streaming: false }, children))));
+            React.createElement(TamboMcpTokenProvider, null,
+                React.createElement(TamboThreadProvider, { streaming: false }, children)))));
     beforeEach(() => {
         jest.clearAllMocks();
         // Setup mock query client
@@ -327,7 +329,8 @@ describe("TamboThreadProvider", () => {
                     currentTimeContextHelper: () => null,
                     currentPageContextHelper: () => null,
                 } },
-                React.createElement(TamboThreadProvider, { streaming: false }, children))));
+                React.createElement(TamboMcpTokenProvider, null,
+                    React.createElement(TamboThreadProvider, { streaming: false }, children)))));
         const mockUnregisteredToolCallResponse = {
             responseMessageDto: {
                 id: "unregistered-tool-call-1",
@@ -425,7 +428,8 @@ describe("TamboThreadProvider", () => {
                         currentTimeContextHelper: () => null,
                         currentPageContextHelper: () => null,
                     } },
-                    React.createElement(TamboThreadProvider, { streaming: true }, children))));
+                    React.createElement(TamboMcpTokenProvider, null,
+                        React.createElement(TamboThreadProvider, { streaming: true }, children)))));
             const mockStreamResponse = {
                 responseMessageDto: {
                     id: "stream-response",
@@ -486,7 +490,8 @@ describe("TamboThreadProvider", () => {
                         currentTimeContextHelper: () => null,
                         currentPageContextHelper: () => null,
                     } },
-                    React.createElement(TamboThreadProvider, { streaming: true }, children))));
+                    React.createElement(TamboMcpTokenProvider, null,
+                        React.createElement(TamboThreadProvider, { streaming: true }, children)))));
             const { result } = renderHook(() => useTamboThread(), {
                 wrapper: wrapperWithStreaming,
             });
@@ -528,7 +533,8 @@ describe("TamboThreadProvider", () => {
                         currentTimeContextHelper: () => null,
                         currentPageContextHelper: () => null,
                     } },
-                    React.createElement(TamboThreadProvider, { streaming: false }, children))));
+                    React.createElement(TamboMcpTokenProvider, null,
+                        React.createElement(TamboThreadProvider, { streaming: false }, children)))));
             const { result } = renderHook(() => useTamboThread(), {
                 wrapper: wrapperWithoutStreaming,
             });
@@ -570,7 +576,8 @@ describe("TamboThreadProvider", () => {
                         currentTimeContextHelper: () => null,
                         currentPageContextHelper: () => null,
                     } },
-                    React.createElement(TamboThreadProvider, null, children))));
+                    React.createElement(TamboMcpTokenProvider, null,
+                        React.createElement(TamboThreadProvider, null, children)))));
             const mockStreamResponse = {
                 responseMessageDto: {
                     id: "stream-response",
@@ -631,7 +638,8 @@ describe("TamboThreadProvider", () => {
                         currentTimeContextHelper: () => null,
                         currentPageContextHelper: () => null,
                     } },
-                    React.createElement(TamboThreadProvider, { streaming: true }, children))));
+                    React.createElement(TamboMcpTokenProvider, null,
+                        React.createElement(TamboThreadProvider, { streaming: true }, children)))));
             const { result } = renderHook(() => useTamboThread(), {
                 wrapper: wrapperWithStreaming,
             });
@@ -675,7 +683,8 @@ describe("TamboThreadProvider", () => {
                         currentTimeContextHelper: () => null,
                         currentPageContextHelper: () => null,
                     } },
-                    React.createElement(TamboThreadProvider, { streaming: false }, children))));
+                    React.createElement(TamboMcpTokenProvider, null,
+                        React.createElement(TamboThreadProvider, { streaming: false }, children)))));
             const mockStreamResponse = {
                 responseMessageDto: {
                     id: "stream-response",
@@ -878,7 +887,8 @@ describe("TamboThreadProvider", () => {
                         currentTimeContextHelper: () => null,
                         currentPageContextHelper: () => null,
                     } },
-                    React.createElement(TamboThreadProvider, { streaming: false }, children))));
+                    React.createElement(TamboMcpTokenProvider, null,
+                        React.createElement(TamboThreadProvider, { streaming: false }, children)))));
             const mockToolCallResponse = {
                 responseMessageDto: {
                     id: "tool-call-1",
@@ -971,7 +981,8 @@ describe("TamboThreadProvider", () => {
                         currentTimeContextHelper: () => null,
                         currentPageContextHelper: () => null,
                     } },
-                    React.createElement(TamboThreadProvider, { streaming: true }, children))));
+                    React.createElement(TamboMcpTokenProvider, null,
+                        React.createElement(TamboThreadProvider, { streaming: true }, children)))));
             const mockToolCallChunk = {
                 responseMessageDto: {
                     id: "tool-call-chunk",
@@ -1072,7 +1083,8 @@ describe("TamboThreadProvider", () => {
                         currentTimeContextHelper: () => null,
                         currentPageContextHelper: () => null,
                     } },
-                    React.createElement(TamboThreadProvider, { streaming: false }, children))));
+                    React.createElement(TamboMcpTokenProvider, null,
+                        React.createElement(TamboThreadProvider, { streaming: false }, children)))));
             const mockToolCallResponse = {
                 responseMessageDto: {
                     id: "tool-call-1",
@@ -1159,7 +1171,8 @@ describe("TamboThreadProvider", () => {
                         currentTimeContextHelper: () => null,
                         currentPageContextHelper: () => null,
                     } },
-                    React.createElement(TamboThreadProvider, { streaming: false }, children))));
+                    React.createElement(TamboMcpTokenProvider, null,
+                        React.createElement(TamboThreadProvider, { streaming: false }, children)))));
             const mockToolCallResponse = {
                 responseMessageDto: {
                     id: "tool-call-1",