@tambo-ai/react 0.58.1 → 0.60.0

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

@@ -1,8 +1,9 @@
-import { deepEqual } from "fast-equals";
-import React, { createContext, useContext, useEffect, useState, } from "react";
+import React, { createContext, useContext, useEffect, useMemo, useRef, 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";
+import { useElicitation } from "./elicitation";
+import { MCPClient, MCPTransport, } from "./mcp-client";
 /**
  * Extracts error message from MCP tool result content.
  * Handles both array and string content formats.
@@ -15,7 +16,7 @@ export function extractErrorMessage(content) {
     }
     if (Array.isArray(content)) {
         const textItems = content
-            .filter((item) => item && item.type === "text" && typeof item.text === "string")
+            .filter((item) => item?.type === "text" && typeof item.text === "string")
             .map((item) => item.text);
         return textItems.length > 0
             ? textItems.join(" ")
@@ -26,120 +27,236 @@ export function extractErrorMessage(content) {
     }
     return `${content}`;
 }
-const McpProviderContext = createContext([]);
+const McpProviderContext = createContext({
+    servers: [],
+    elicitation: null,
+    resolveElicitation: null,
+});
+// 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 providerSamplingHandler = handlers?.sampling;
+    // Elicitation state and default handler
+    const { elicitation, resolveElicitation, defaultElicitationHandler } = useElicitation();
+    // Use provided handler or fall back to default
+    const providerElicitationHandler = handlers?.elicitation ?? defaultElicitationHandler;
+    // Stable reference to track active clients by server key
+    const clientMapRef = useRef(new Map());
+    // Track tool ownership to prevent duplicate registrations across servers
+    // toolOwnerRef: tool name -> server key; keyToToolsRef: server key -> set of tool names
+    const toolOwnerRef = useRef(new Map());
+    const keyToToolsRef = useRef(new Map());
+    // State for exposing connected servers to consumers
     const [connectedMcpServers, setConnectedMcpServers] = useState([]);
-    useEffect(() => {
-        if (!mcpServers) {
-            return;
+    // Stable map of current server configurations keyed by server key
+    const currentServersMap = useMemo(() => {
+        const servers = [...mcpServers];
+        // Add internal Tambo MCP server if we have an access token and a base URL
+        if (mcpAccessToken && tamboBaseUrl) {
+            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}`,
+                },
+            });
         }
-        async function registerMcpServers(mcpServerInfos) {
-            // Maps tool names to the MCP client that registered them
-            const mcpServerMap = new Map();
-            setConnectedMcpServers((prev) => 
-            // remove any servers that are not in the new list
-            prev.filter((s) => mcpServerInfos.some((mcpServerInfo) => equalsMcpServer(s, mcpServerInfo))));
-            // initialize the MCP clients, converting McpServerInfo -> McpServer
-            const mcpServers = await Promise.allSettled(mcpServerInfos.map(async (mcpServerInfo) => {
+        // Create a map of server key -> server info for efficient lookups
+        const serverMap = new Map();
+        servers.forEach((server) => {
+            const serverInfo = normalizeServerInfo(server);
+            // Store without cloning to avoid unnecessary allocation
+            serverMap.set(serverInfo.key, serverInfo);
+        });
+        return serverMap;
+    }, [mcpServers, mcpAccessToken, tamboBaseUrl]);
+    // Main effect: manage client lifecycle (create/remove)
+    useEffect(() => {
+        const clientMap = clientMapRef.current;
+        const currentKeys = new Set(currentServersMap.keys());
+        const existingKeys = new Set(clientMap.keys());
+        // 1. Remove clients that are no longer in the current server list
+        const keysToRemove = Array.from(existingKeys).filter((key) => !currentKeys.has(key));
+        keysToRemove.forEach((key) => {
+            const server = clientMap.get(key);
+            if (server?.client) {
                 try {
-                    const client = await MCPClient.create(mcpServerInfo.url, mcpServerInfo.transport, mcpServerInfo.customHeaders, undefined, // no oauth support yet
-                    undefined);
-                    const connectedMcpServer = {
-                        ...mcpServerInfo,
-                        client: client,
+                    server.client.close();
+                }
+                catch (error) {
+                    // Avoid logging sensitive data embedded in the key (headers)
+                    const url = server.url ?? "(unknown url)";
+                    console.error(`Error closing MCP client for ${url}:`, error);
+                }
+            }
+            // Release tool ownership for this server
+            const owned = keyToToolsRef.current.get(key);
+            if (owned) {
+                for (const name of owned)
+                    toolOwnerRef.current.delete(name);
+                keyToToolsRef.current.delete(key);
+            }
+            clientMap.delete(key);
+        });
+        // 2. Add new clients for servers that don't exist yet
+        const keysToAdd = Array.from(currentKeys).filter((key) => !existingKeys.has(key));
+        if (keysToAdd.length > 0) {
+            // Async initialization of new clients
+            Promise.allSettled(keysToAdd.map(async (key) => {
+                const serverInfo = currentServersMap.get(key);
+                try {
+                    // Build effective handlers (per-server overrides provider)
+                    const effectiveHandlers = {};
+                    if (serverInfo.handlers?.elicitation) {
+                        effectiveHandlers.elicitation = serverInfo.handlers.elicitation;
+                    }
+                    else if (providerElicitationHandler) {
+                        effectiveHandlers.elicitation = async (request, extra) => await providerElicitationHandler(request, extra, serverInfo);
+                    }
+                    if (serverInfo.handlers?.sampling) {
+                        effectiveHandlers.sampling = serverInfo.handlers.sampling;
+                    }
+                    else if (providerSamplingHandler) {
+                        effectiveHandlers.sampling = async (request, extra) => await providerSamplingHandler(request, extra, serverInfo);
+                    }
+                    const client = await MCPClient.create(serverInfo.url, serverInfo.transport, serverInfo.customHeaders, undefined, undefined, effectiveHandlers);
+                    const connectedServer = {
+                        ...serverInfo,
+                        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,
-                    ]);
-                    return connectedMcpServer;
+                    clientMap.set(key, connectedServer);
+                    setConnectedMcpServers(Array.from(clientMap.values()));
+                    // Register tools from this server (deduplicated by ownership)
+                    try {
+                        const tools = await client.listTools();
+                        tools.forEach((tool) => {
+                            // Skip if another server already owns this tool
+                            const currentOwner = toolOwnerRef.current.get(tool.name);
+                            if (currentOwner && currentOwner !== key) {
+                                return;
+                            }
+                            // Record ownership for this server key
+                            if (!currentOwner) {
+                                toolOwnerRef.current.set(tool.name, key);
+                                if (!keyToToolsRef.current.has(key)) {
+                                    keyToToolsRef.current.set(key, new Set());
+                                }
+                                keyToToolsRef.current.get(key).add(tool.name);
+                            }
+                            registerTool({
+                                description: tool.description ?? "",
+                                name: tool.name,
+                                tool: async (args = {}) => {
+                                    const server = clientMap.get(key);
+                                    if (!server?.client) {
+                                        throw new Error(`MCP server for tool ${tool.name} is not connected`);
+                                    }
+                                    const result = await server.client.callTool(tool.name, args);
+                                    if (result.isError) {
+                                        const errorMessage = extractErrorMessage(result.content);
+                                        throw new Error(errorMessage);
+                                    }
+                                    return result.content;
+                                },
+                                toolSchema: tool.inputSchema,
+                                transformToContent: (content) => {
+                                    if (isContentPartArray(content)) {
+                                        return content;
+                                    }
+                                    return [{ type: "text", text: toText(content) }];
+                                },
+                            });
+                        });
+                    }
+                    catch (error) {
+                        console.error(`Failed to register tools from MCP server ${serverInfo.url}:`, error);
+                    }
                 }
                 catch (error) {
-                    const failedMcpServer = {
-                        ...mcpServerInfo,
+                    const failedServer = {
+                        ...serverInfo,
                         connectionError: error,
                     };
-                    // 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,
-                    ]);
-                    return failedMcpServer;
+                    clientMap.set(key, failedServer);
+                    setConnectedMcpServers(Array.from(clientMap.values()));
+                    console.error(`Failed to connect to MCP server ${serverInfo.url}:`, error);
                 }
             }));
-            // note do not rely on the state
-            const connectedMcpServers = mcpServers
-                .filter((result) => result.status === "fulfilled")
-                .map((result) => result.value);
-            // Now create a map of tool name to MCP client
-            const serverToolLists = connectedMcpServers.map(async (mcpServer) => {
-                const tools = (await mcpServer.client?.listTools()) ?? [];
-                tools.forEach((tool) => {
-                    mcpServerMap.set(tool.name, mcpServer);
-                });
-                return tools;
-            });
-            const toolResults = await Promise.allSettled(serverToolLists);
-            // Just log the failed tools, we can't do anything about them
-            const failedTools = toolResults.filter((result) => result.status === "rejected");
-            if (failedTools.length > 0) {
-                console.error("Failed to register tools from MCP servers:", failedTools.map((result) => result.reason));
+        }
+        // Update state after removals (additions update state asynchronously above)
+        if (keysToRemove.length > 0) {
+            setConnectedMcpServers(Array.from(clientMap.values()));
+        }
+        // eslint-disable-next-line react-hooks/exhaustive-deps
+    }, [currentServersMap, registerTool]);
+    // Update handlers when they change (without recreating clients)
+    useEffect(() => {
+        const clientMap = clientMapRef.current;
+        clientMap.forEach((server, key) => {
+            if (!server.client) {
+                return; // Skip failed servers
+            }
+            const serverInfo = currentServersMap.get(key);
+            if (!serverInfo) {
+                return; // Server was removed, handled by main effect
             }
-            // Register the successful tools
-            const allTools = toolResults
-                .filter((result) => result.status === "fulfilled")
-                .map((result) => result.value)
-                .flat();
-            allTools.forEach((tool) => {
-                registerTool({
-                    description: tool.description ?? "",
-                    name: tool.name,
-                    tool: async (args) => {
-                        const mcpServer = mcpServerMap.get(tool.name);
-                        if (!mcpServer) {
-                            // should never happen
-                            throw new Error(`MCP server for tool ${tool.name} not found`);
-                        }
-                        if (!mcpServer.client) {
-                            // this can't actually happen because the tool can't be registered if the server is not connected
-                            throw new Error(`MCP server for tool ${tool.name} is not connected`);
-                        }
-                        const result = await mcpServer.client.callTool(tool.name, args);
-                        if (result.isError) {
-                            const errorMessage = extractErrorMessage(result.content);
-                            throw new Error(errorMessage);
-                        }
-                        return result.content;
-                    },
-                    toolSchema: tool.inputSchema,
-                    transformToContent: (content) => {
-                        // MCP tools can return content in various formats; pass through arrays of content parts
-                        // unchanged, otherwise stringify into a text content part.
-                        if (isContentPartArray(content)) {
-                            return content;
-                        }
-                        return [{ type: "text", text: toText(content) }];
-                    },
-                });
+            // Determine effective handlers
+            const effectiveElicitationHandler = serverInfo.handlers?.elicitation ??
+                (providerElicitationHandler
+                    ? async (request, extra) => await providerElicitationHandler(request, extra, serverInfo)
+                    : undefined);
+            const effectiveSamplingHandler = serverInfo.handlers?.sampling ??
+                (providerSamplingHandler
+                    ? async (request, extra) => await providerSamplingHandler(request, extra, serverInfo)
+                    : undefined);
+            // Update handlers unconditionally (allows removal by passing undefined)
+            server.client.updateElicitationHandler?.(effectiveElicitationHandler);
+            server.client.updateSamplingHandler?.(effectiveSamplingHandler);
+        });
+    }, [currentServersMap, providerElicitationHandler, providerSamplingHandler]);
+    // Cleanup on unmount: close all clients
+    useEffect(() => {
+        const clientMap = clientMapRef.current;
+        const ownerMapAtMount = toolOwnerRef.current;
+        const keyToToolsAtMount = keyToToolsRef.current;
+        return () => {
+            clientMap.forEach((server, _key) => {
+                if (server.client) {
+                    try {
+                        server.client.close();
+                    }
+                    catch (error) {
+                        const url = server.url ?? "(unknown url)";
+                        console.error(`Error closing MCP client on unmount for ${url}:`, error);
+                    }
+                }
             });
-        }
-        // normalize the server infos
-        const mcpServerInfos = mcpServers.map((mcpServer) => typeof mcpServer === "string"
-            ? { url: mcpServer, transport: MCPTransport.SSE }
-            : mcpServer);
-        registerMcpServers(mcpServerInfos);
-    }, [mcpServers, registerTool]);
-    return (React.createElement(McpProviderContext.Provider, { value: connectedMcpServers }, children));
+            clientMap.clear();
+            ownerMapAtMount.clear();
+            keyToToolsAtMount.clear();
+        };
+    }, []);
+    const contextValue = useMemo(() => ({
+        servers: connectedMcpServers,
+        elicitation,
+        resolveElicitation,
+    }), [connectedMcpServers, elicitation, resolveElicitation]);
+    return (React.createElement(McpProviderContext.Provider, { value: contextValue }, children));
 };
 /**
  * Hook to access the actual MCP servers, as they are connected (or fail to
@@ -165,11 +282,62 @@ export const TamboMcpProvider = ({ mcpServers, children }) => {
  * @returns The MCP servers
  */
 export const useTamboMcpServers = () => {
-    return useContext(McpProviderContext);
+    return useContext(McpProviderContext).servers;
 };
-function equalsMcpServer(s, mcpServerInfo) {
-    return (s.url === mcpServerInfo.url &&
-        s.transport === mcpServerInfo.transport &&
-        deepEqual(s.customHeaders, mcpServerInfo.customHeaders));
+/**
+ * Hook to access elicitation context from TamboMcpProvider.
+ * This provides access to the current elicitation request and methods to respond to it.
+ *
+ * The elicitation state is automatically managed by TamboMcpProvider when MCP servers
+ * request user input through the elicitation protocol.
+ * @returns The elicitation context with current request and response handlers
+ * @example
+ * ```tsx
+ * function ElicitationUI() {
+ *   const { elicitation, resolveElicitation } = useTamboElicitationContext();
+ *
+ *   if (!elicitation) return null;
+ *
+ *   return (
+ *     <div>
+ *       <p>{elicitation.message}</p>
+ *       <button onClick={() => resolveElicitation?.({ action: "accept", content: {} })}>
+ *         Accept
+ *       </button>
+ *     </div>
+ *   );
+ * }
+ * ```
+ */
+export const useTamboElicitationContext = () => {
+    const context = useContext(McpProviderContext);
+    return {
+        elicitation: context.elicitation,
+        resolveElicitation: context.resolveElicitation,
+    };
+};
+/**
+ * Creates a stable identifier for an MCP server based on its connection properties.
+ * Two servers with the same URL, transport, and headers will have the same key.
+ * @returns A stable string key identifying the server
+ */
+function getServerKey(serverInfo) {
+    const headerStr = serverInfo.customHeaders
+        ? JSON.stringify(Object.entries(serverInfo.customHeaders)
+            .map(([k, v]) => [k.toLowerCase(), v])
+            .sort(([a], [b]) => a.localeCompare(b)))
+        : "";
+    return `${serverInfo.url}|${serverInfo.transport ?? MCPTransport.SSE}|${headerStr}`;
+}
+/**
+ * Normalizes a server definition (string or object) into a McpServerInfo.
+ * @returns The normalized McpServerInfo object
+ */
+function normalizeServerInfo(server) {
+    const s = typeof server === "string"
+        ? { url: server, transport: MCPTransport.SSE }
+        : server;
+    const key = getServerKey(s);
+    return { ...s, key };
 }
 //# sourceMappingURL=tambo-mcp-provider.js.map

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,KAAK,EAAE,EACZ,aAAa,EAEb,UAAU,EACV,SAAS,EACT,OAAO,EACP,MAAM,EACN,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,EAAgC,cAAc,EAAE,MAAM,eAAe,CAAC;AAC7E,OAAO,EACL,SAAS,EAIT,YAAY,GACb,MAAM,cAAc,CAAC;AAEtB;;;;;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,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,EAAE,IAAI,KAAK,MAAM,IAAI,OAAO,IAAI,CAAC,IAAI,KAAK,QAAQ,CAAC;aACxE,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;AAoFD,MAAM,kBAAkB,GAAG,aAAa,CAA0B;IAChE,OAAO,EAAE,EAAE;IACX,WAAW,EAAE,IAAI;IACjB,kBAAkB,EAAE,IAAI;CACzB,CAAC,CAAC;AAEH,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,uBAAuB,GAAG,QAAQ,EAAE,QAAQ,CAAC;IAEnD,wCAAwC;IACxC,MAAM,EAAE,WAAW,EAAE,kBAAkB,EAAE,yBAAyB,EAAE,GAClE,cAAc,EAAE,CAAC;IAEnB,+CAA+C;IAC/C,MAAM,0BAA0B,GAC9B,QAAQ,EAAE,WAAW,IAAI,yBAAyB,CAAC;IAErD,yDAAyD;IACzD,MAAM,YAAY,GAAG,MAAM,CAAyB,IAAI,GAAG,EAAE,CAAC,CAAC;IAC/D,yEAAyE;IACzE,wFAAwF;IACxF,MAAM,YAAY,GAAG,MAAM,CAAsB,IAAI,GAAG,EAAE,CAAC,CAAC;IAC5D,MAAM,aAAa,GAAG,MAAM,CAA2B,IAAI,GAAG,EAAE,CAAC,CAAC;IAElE,oDAAoD;IACpD,MAAM,CAAC,mBAAmB,EAAE,sBAAsB,CAAC,GAAG,QAAQ,CAC5D,EAAE,CACH,CAAC;IAEF,kEAAkE;IAClE,MAAM,iBAAiB,GAAG,OAAO,CAAC,GAAG,EAAE;QACrC,MAAM,OAAO,GAAG,CAAC,GAAG,UAAU,CAAC,CAAC;QAEhC,0EAA0E;QAC1E,IAAI,cAAc,IAAI,YAAY,EAAE,CAAC;YACnC,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,kEAAkE;QAClE,MAAM,SAAS,GAAG,IAAI,GAAG,EAA2B,CAAC;QACrD,OAAO,CAAC,OAAO,CAAC,CAAC,MAAM,EAAE,EAAE;YACzB,MAAM,UAAU,GAAG,mBAAmB,CAAC,MAAM,CAAC,CAAC;YAC/C,wDAAwD;YACxD,SAAS,CAAC,GAAG,CAAC,UAAU,CAAC,GAAG,EAAE,UAAU,CAAC,CAAC;QAC5C,CAAC,CAAC,CAAC;QAEH,OAAO,SAAS,CAAC;IACnB,CAAC,EAAE,CAAC,UAAU,EAAE,cAAc,EAAE,YAAY,CAAC,CAAC,CAAC;IAE/C,uDAAuD;IACvD,SAAS,CAAC,GAAG,EAAE;QACb,MAAM,SAAS,GAAG,YAAY,CAAC,OAAO,CAAC;QACvC,MAAM,WAAW,GAAG,IAAI,GAAG,CAAC,iBAAiB,CAAC,IAAI,EAAE,CAAC,CAAC;QACtD,MAAM,YAAY,GAAG,IAAI,GAAG,CAAC,SAAS,CAAC,IAAI,EAAE,CAAC,CAAC;QAE/C,kEAAkE;QAClE,MAAM,YAAY,GAAG,KAAK,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,MAAM,CAClD,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC,WAAW,CAAC,GAAG,CAAC,GAAG,CAAC,CAC/B,CAAC;QACF,YAAY,CAAC,OAAO,CAAC,CAAC,GAAG,EAAE,EAAE;YAC3B,MAAM,MAAM,GAAG,SAAS,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;YAClC,IAAI,MAAM,EAAE,MAAM,EAAE,CAAC;gBACnB,IAAI,CAAC;oBACH,MAAM,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC;gBACxB,CAAC;gBAAC,OAAO,KAAK,EAAE,CAAC;oBACf,6DAA6D;oBAC7D,MAAM,GAAG,GAAI,MAAoB,CAAC,GAAG,IAAI,eAAe,CAAC;oBACzD,OAAO,CAAC,KAAK,CAAC,gCAAgC,GAAG,GAAG,EAAE,KAAK,CAAC,CAAC;gBAC/D,CAAC;YACH,CAAC;YACD,yCAAyC;YACzC,MAAM,KAAK,GAAG,aAAa,CAAC,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;YAC7C,IAAI,KAAK,EAAE,CAAC;gBACV,KAAK,MAAM,IAAI,IAAI,KAAK;oBAAE,YAAY,CAAC,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;gBAC5D,aAAa,CAAC,OAAO,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;YACpC,CAAC;YACD,SAAS,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;QACxB,CAAC,CAAC,CAAC;QAEH,sDAAsD;QACtD,MAAM,SAAS,GAAG,KAAK,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,MAAM,CAC9C,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC,YAAY,CAAC,GAAG,CAAC,GAAG,CAAC,CAChC,CAAC;QAEF,IAAI,SAAS,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACzB,sCAAsC;YACtC,OAAO,CAAC,UAAU,CAChB,SAAS,CAAC,GAAG,CAAC,KAAK,EAAE,GAAG,EAAE,EAAE;gBAC1B,MAAM,UAAU,GAAG,iBAAiB,CAAC,GAAG,CAAC,GAAG,CAAE,CAAC;gBAE/C,IAAI,CAAC;oBACH,2DAA2D;oBAC3D,MAAM,iBAAiB,GAAyB,EAAE,CAAC;oBAEnD,IAAI,UAAU,CAAC,QAAQ,EAAE,WAAW,EAAE,CAAC;wBACrC,iBAAiB,CAAC,WAAW,GAAG,UAAU,CAAC,QAAQ,CAAC,WAAW,CAAC;oBAClE,CAAC;yBAAM,IAAI,0BAA0B,EAAE,CAAC;wBACtC,iBAAiB,CAAC,WAAW,GAAG,KAAK,EAAE,OAAO,EAAE,KAAK,EAAE,EAAE,CACvD,MAAM,0BAA0B,CAAC,OAAO,EAAE,KAAK,EAAE,UAAU,CAAC,CAAC;oBACjE,CAAC;oBAED,IAAI,UAAU,CAAC,QAAQ,EAAE,QAAQ,EAAE,CAAC;wBAClC,iBAAiB,CAAC,QAAQ,GAAG,UAAU,CAAC,QAAQ,CAAC,QAAQ,CAAC;oBAC5D,CAAC;yBAAM,IAAI,uBAAuB,EAAE,CAAC;wBACnC,iBAAiB,CAAC,QAAQ,GAAG,KAAK,EAAE,OAAO,EAAE,KAAK,EAAE,EAAE,CACpD,MAAM,uBAAuB,CAAC,OAAO,EAAE,KAAK,EAAE,UAAU,CAAC,CAAC;oBAC9D,CAAC;oBAED,MAAM,MAAM,GAAG,MAAM,SAAS,CAAC,MAAM,CACnC,UAAU,CAAC,GAAG,EACd,UAAU,CAAC,SAAS,EACpB,UAAU,CAAC,aAAa,EACxB,SAAS,EACT,SAAS,EACT,iBAAiB,CAClB,CAAC;oBAEF,MAAM,eAAe,GAAuB;wBAC1C,GAAG,UAAU;wBACb,MAAM;qBACP,CAAC;oBAEF,SAAS,CAAC,GAAG,CAAC,GAAG,EAAE,eAAe,CAAC,CAAC;oBACpC,sBAAsB,CAAC,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;oBAEvD,8DAA8D;oBAC9D,IAAI,CAAC;wBACH,MAAM,KAAK,GAAG,MAAM,MAAM,CAAC,SAAS,EAAE,CAAC;wBACvC,KAAK,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,EAAE;4BACrB,gDAAgD;4BAChD,MAAM,YAAY,GAAG,YAAY,CAAC,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;4BACzD,IAAI,YAAY,IAAI,YAAY,KAAK,GAAG,EAAE,CAAC;gCACzC,OAAO;4BACT,CAAC;4BAED,uCAAuC;4BACvC,IAAI,CAAC,YAAY,EAAE,CAAC;gCAClB,YAAY,CAAC,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,EAAE,GAAG,CAAC,CAAC;gCACzC,IAAI,CAAC,aAAa,CAAC,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC;oCACpC,aAAa,CAAC,OAAO,CAAC,GAAG,CAAC,GAAG,EAAE,IAAI,GAAG,EAAE,CAAC,CAAC;gCAC5C,CAAC;gCACD,aAAa,CAAC,OAAO,CAAC,GAAG,CAAC,GAAG,CAAE,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;4BACjD,CAAC;4BAED,YAAY,CAAC;gCACX,WAAW,EAAE,IAAI,CAAC,WAAW,IAAI,EAAE;gCACnC,IAAI,EAAE,IAAI,CAAC,IAAI;gCACf,IAAI,EAAE,KAAK,EAAE,OAAgC,EAAE,EAAE,EAAE;oCACjD,MAAM,MAAM,GAAG,SAAS,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;oCAClC,IAAI,CAAC,MAAM,EAAE,MAAM,EAAE,CAAC;wCACpB,MAAM,IAAI,KAAK,CACb,uBAAuB,IAAI,CAAC,IAAI,mBAAmB,CACpD,CAAC;oCACJ,CAAC;oCACD,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,MAAM,CAAC,QAAQ,CACzC,IAAI,CAAC,IAAI,EACT,IAAI,CACL,CAAC;oCACF,IAAI,MAAM,CAAC,OAAO,EAAE,CAAC;wCACnB,MAAM,YAAY,GAAG,mBAAmB,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;wCACzD,MAAM,IAAI,KAAK,CAAC,YAAY,CAAC,CAAC;oCAChC,CAAC;oCACD,OAAO,MAAM,CAAC,OAAO,CAAC;gCACxB,CAAC;gCACD,UAAU,EAAE,IAAI,CAAC,WAAsC;gCACvD,kBAAkB,EAAE,CAAC,OAAgB,EAAE,EAAE;oCACvC,IAAI,kBAAkB,CAAC,OAAO,CAAC,EAAE,CAAC;wCAChC,OAAO,OAAO,CAAC;oCACjB,CAAC;oCACD,OAAO,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;gCACnD,CAAC;6BACF,CAAC,CAAC;wBACL,CAAC,CAAC,CAAC;oBACL,CAAC;oBAAC,OAAO,KAAK,EAAE,CAAC;wBACf,OAAO,CAAC,KAAK,CACX,4CAA4C,UAAU,CAAC,GAAG,GAAG,EAC7D,KAAK,CACN,CAAC;oBACJ,CAAC;gBACH,CAAC;gBAAC,OAAO,KAAK,EAAE,CAAC;oBACf,MAAM,YAAY,GAAoB;wBACpC,GAAG,UAAU;wBACb,eAAe,EAAE,KAAc;qBAChC,CAAC;oBACF,SAAS,CAAC,GAAG,CAAC,GAAG,EAAE,YAAY,CAAC,CAAC;oBACjC,sBAAsB,CAAC,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;oBACvD,OAAO,CAAC,KAAK,CACX,mCAAmC,UAAU,CAAC,GAAG,GAAG,EACpD,KAAK,CACN,CAAC;gBACJ,CAAC;YACH,CAAC,CAAC,CACH,CAAC;QACJ,CAAC;QAED,4EAA4E;QAC5E,IAAI,YAAY,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC5B,sBAAsB,CAAC,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;QACzD,CAAC;QAED,uDAAuD;IACzD,CAAC,EAAE,CAAC,iBAAiB,EAAE,YAAY,CAAC,CAAC,CAAC;IAEtC,gEAAgE;IAChE,SAAS,CAAC,GAAG,EAAE;QACb,MAAM,SAAS,GAAG,YAAY,CAAC,OAAO,CAAC;QAEvC,SAAS,CAAC,OAAO,CAAC,CAAC,MAAM,EAAE,GAAG,EAAE,EAAE;YAChC,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC;gBACnB,OAAO,CAAC,sBAAsB;YAChC,CAAC;YAED,MAAM,UAAU,GAAG,iBAAiB,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;YAC9C,IAAI,CAAC,UAAU,EAAE,CAAC;gBAChB,OAAO,CAAC,6CAA6C;YACvD,CAAC;YAED,+BAA+B;YAC/B,MAAM,2BAA2B,GAC/B,UAAU,CAAC,QAAQ,EAAE,WAAW;gBAChC,CAAC,0BAA0B;oBACzB,CAAC,CAAC,KAAK,EAAE,OAAO,EAAE,KAAK,EAAE,EAAE,CACvB,MAAM,0BAA0B,CAAC,OAAO,EAAE,KAAK,EAAE,UAAU,CAAC;oBAChE,CAAC,CAAC,SAAS,CAAC,CAAC;YAEjB,MAAM,wBAAwB,GAC5B,UAAU,CAAC,QAAQ,EAAE,QAAQ;gBAC7B,CAAC,uBAAuB;oBACtB,CAAC,CAAC,KAAK,EAAE,OAAO,EAAE,KAAK,EAAE,EAAE,CACvB,MAAM,uBAAuB,CAAC,OAAO,EAAE,KAAK,EAAE,UAAU,CAAC;oBAC7D,CAAC,CAAC,SAAS,CAAC,CAAC;YAEjB,wEAAwE;YACxE,MAAM,CAAC,MAAM,CAAC,wBAAwB,EAAE,CAAC,2BAA2B,CAAC,CAAC;YACtE,MAAM,CAAC,MAAM,CAAC,qBAAqB,EAAE,CAAC,wBAAwB,CAAC,CAAC;QAClE,CAAC,CAAC,CAAC;IACL,CAAC,EAAE,CAAC,iBAAiB,EAAE,0BAA0B,EAAE,uBAAuB,CAAC,CAAC,CAAC;IAE7E,wCAAwC;IACxC,SAAS,CAAC,GAAG,EAAE;QACb,MAAM,SAAS,GAAG,YAAY,CAAC,OAAO,CAAC;QACvC,MAAM,eAAe,GAAG,YAAY,CAAC,OAAO,CAAC;QAC7C,MAAM,iBAAiB,GAAG,aAAa,CAAC,OAAO,CAAC;QAChD,OAAO,GAAG,EAAE;YACV,SAAS,CAAC,OAAO,CAAC,CAAC,MAAM,EAAE,IAAI,EAAE,EAAE;gBACjC,IAAI,MAAM,CAAC,MAAM,EAAE,CAAC;oBAClB,IAAI,CAAC;wBACH,MAAM,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC;oBACxB,CAAC;oBAAC,OAAO,KAAK,EAAE,CAAC;wBACf,MAAM,GAAG,GAAI,MAAoB,CAAC,GAAG,IAAI,eAAe,CAAC;wBACzD,OAAO,CAAC,KAAK,CACX,2CAA2C,GAAG,GAAG,EACjD,KAAK,CACN,CAAC;oBACJ,CAAC;gBACH,CAAC;YACH,CAAC,CAAC,CAAC;YACH,SAAS,CAAC,KAAK,EAAE,CAAC;YAClB,eAAe,CAAC,KAAK,EAAE,CAAC;YACxB,iBAAiB,CAAC,KAAK,EAAE,CAAC;QAC5B,CAAC,CAAC;IACJ,CAAC,EAAE,EAAE,CAAC,CAAC;IAEP,MAAM,YAAY,GAAG,OAAO,CAC1B,GAAG,EAAE,CAAC,CAAC;QACL,OAAO,EAAE,mBAAmB;QAC5B,WAAW;QACX,kBAAkB;KACnB,CAAC,EACF,CAAC,mBAAmB,EAAE,WAAW,EAAE,kBAAkB,CAAC,CACvD,CAAC;IAEF,OAAO,CACL,oBAAC,kBAAkB,CAAC,QAAQ,IAAC,KAAK,EAAE,YAAY,IAC7C,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,OAAO,CAAC;AAChD,CAAC,CAAC;AAEF;;;;;;;;;;;;;;;;;;;;;;;;GAwBG;AACH,MAAM,CAAC,MAAM,0BAA0B,GAAG,GAAG,EAAE;IAC7C,MAAM,OAAO,GAAG,UAAU,CAAC,kBAAkB,CAAC,CAAC;IAC/C,OAAO;QACL,WAAW,EAAE,OAAO,CAAC,WAAW;QAChC,kBAAkB,EAAE,OAAO,CAAC,kBAAkB;KAC/C,CAAC;AACJ,CAAC,CAAC;AAEF;;;;GAIG;AACH,SAAS,YAAY,CACnB,UAAsE;IAEtE,MAAM,SAAS,GAAG,UAAU,CAAC,aAAa;QACxC,CAAC,CAAC,IAAI,CAAC,SAAS,CACZ,MAAM,CAAC,OAAO,CAAC,UAAU,CAAC,aAAa,CAAC;aACrC,GAAG,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,WAAW,EAAE,EAAE,CAAC,CAAU,CAAC;aAC9C,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC,CAC1C;QACH,CAAC,CAAC,EAAE,CAAC;IACP,OAAO,GAAG,UAAU,CAAC,GAAG,IAAI,UAAU,CAAC,SAAS,IAAI,YAAY,CAAC,GAAG,IAAI,SAAS,EAAE,CAAC;AACtF,CAAC;AAED;;;GAGG;AACH,SAAS,mBAAmB,CAAC,MAA8B;IACzD,MAAM,CAAC,GACL,OAAO,MAAM,KAAK,QAAQ;QACxB,CAAC,CAAC,EAAE,GAAG,EAAE,MAAM,EAAE,SAAS,EAAE,YAAY,CAAC,GAAG,EAAE;QAC9C,CAAC,CAAC,MAAM,CAAC;IACb,MAAM,GAAG,GAAG,YAAY,CAAC,CAAC,CAAC,CAAC;IAC5B,OAAO,EAAE,GAAG,CAAC,EAAE,GAAG,EAAE,CAAC;AACvB,CAAC","sourcesContent":["import React, {\n  createContext,\n  FC,\n  useContext,\n  useEffect,\n  useMemo,\n  useRef,\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 { type ElicitationContextState, useElicitation } from \"./elicitation\";\nimport {\n  MCPClient,\n  MCPElicitationHandler,\n  MCPHandlers,\n  MCPSamplingHandler,\n  MCPTransport,\n} 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((item) => item?.type === \"text\" && typeof item.text === \"string\")\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 * User-provided configuration for 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\n/**\n * Normalized server information with a stable derived key.\n */\ninterface McpServerConfig extends McpServerInfo {\n  /**\n   * Stable identity for this server derived from its URL/transport/headers.\n   * Present for all server states (connected or failed).\n   */\n  key: string;\n}\n\n/**\n * Connected MCP server with an active client.\n */\nexport interface ConnectedMcpServer extends McpServerConfig {\n  client: MCPClient;\n}\n\n/**\n * Failed MCP server with a connection error.\n */\nexport interface FailedMcpServer extends McpServerConfig {\n  client?: never;\n  connectionError: Error;\n}\n\n/**\n * An active or failed MCP server, with access to the MCP client.\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 *\n * Handlers receive three parameters:\n * 1. request - The MCP request\n * 2. extra - RequestHandlerExtra containing AbortSignal and other metadata\n * 3. serverInfo - Configuration of the MCP server that triggered this request\n */\nexport interface ProviderMCPHandlers {\n  elicitation?: (\n    request: Parameters<MCPElicitationHandler>[0],\n    extra: Parameters<MCPElicitationHandler>[1],\n    serverInfo: McpServerConfig,\n  ) => ReturnType<MCPElicitationHandler>;\n  sampling?: (\n    request: Parameters<MCPSamplingHandler>[0],\n    extra: Parameters<MCPSamplingHandler>[1],\n    serverInfo: McpServerConfig,\n  ) => ReturnType<MCPSamplingHandler>;\n}\n\n/**\n * Context value for MCP provider including server list and elicitation state\n */\ninterface McpProviderContextValue extends ElicitationContextState {\n  servers: McpServer[];\n}\n\nconst McpProviderContext = createContext<McpProviderContextValue>({\n  servers: [],\n  elicitation: null,\n  resolveElicitation: null,\n});\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 providerSamplingHandler = handlers?.sampling;\n\n  // Elicitation state and default handler\n  const { elicitation, resolveElicitation, defaultElicitationHandler } =\n    useElicitation();\n\n  // Use provided handler or fall back to default\n  const providerElicitationHandler =\n    handlers?.elicitation ?? defaultElicitationHandler;\n\n  // Stable reference to track active clients by server key\n  const clientMapRef = useRef<Map<string, McpServer>>(new Map());\n  // Track tool ownership to prevent duplicate registrations across servers\n  // toolOwnerRef: tool name -> server key; keyToToolsRef: server key -> set of tool names\n  const toolOwnerRef = useRef<Map<string, string>>(new Map());\n  const keyToToolsRef = useRef<Map<string, Set<string>>>(new Map());\n\n  // State for exposing connected servers to consumers\n  const [connectedMcpServers, setConnectedMcpServers] = useState<McpServer[]>(\n    [],\n  );\n\n  // Stable map of current server configurations keyed by server key\n  const currentServersMap = 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      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    // Create a map of server key -> server info for efficient lookups\n    const serverMap = new Map<string, McpServerConfig>();\n    servers.forEach((server) => {\n      const serverInfo = normalizeServerInfo(server);\n      // Store without cloning to avoid unnecessary allocation\n      serverMap.set(serverInfo.key, serverInfo);\n    });\n\n    return serverMap;\n  }, [mcpServers, mcpAccessToken, tamboBaseUrl]);\n\n  // Main effect: manage client lifecycle (create/remove)\n  useEffect(() => {\n    const clientMap = clientMapRef.current;\n    const currentKeys = new Set(currentServersMap.keys());\n    const existingKeys = new Set(clientMap.keys());\n\n    // 1. Remove clients that are no longer in the current server list\n    const keysToRemove = Array.from(existingKeys).filter(\n      (key) => !currentKeys.has(key),\n    );\n    keysToRemove.forEach((key) => {\n      const server = clientMap.get(key);\n      if (server?.client) {\n        try {\n          server.client.close();\n        } catch (error) {\n          // Avoid logging sensitive data embedded in the key (headers)\n          const url = (server as McpServer).url ?? \"(unknown url)\";\n          console.error(`Error closing MCP client for ${url}:`, error);\n        }\n      }\n      // Release tool ownership for this server\n      const owned = keyToToolsRef.current.get(key);\n      if (owned) {\n        for (const name of owned) toolOwnerRef.current.delete(name);\n        keyToToolsRef.current.delete(key);\n      }\n      clientMap.delete(key);\n    });\n\n    // 2. Add new clients for servers that don't exist yet\n    const keysToAdd = Array.from(currentKeys).filter(\n      (key) => !existingKeys.has(key),\n    );\n\n    if (keysToAdd.length > 0) {\n      // Async initialization of new clients\n      Promise.allSettled(\n        keysToAdd.map(async (key) => {\n          const serverInfo = currentServersMap.get(key)!;\n\n          try {\n            // Build effective handlers (per-server overrides provider)\n            const effectiveHandlers: Partial<MCPHandlers> = {};\n\n            if (serverInfo.handlers?.elicitation) {\n              effectiveHandlers.elicitation = serverInfo.handlers.elicitation;\n            } else if (providerElicitationHandler) {\n              effectiveHandlers.elicitation = async (request, extra) =>\n                await providerElicitationHandler(request, extra, serverInfo);\n            }\n\n            if (serverInfo.handlers?.sampling) {\n              effectiveHandlers.sampling = serverInfo.handlers.sampling;\n            } else if (providerSamplingHandler) {\n              effectiveHandlers.sampling = async (request, extra) =>\n                await providerSamplingHandler(request, extra, serverInfo);\n            }\n\n            const client = await MCPClient.create(\n              serverInfo.url,\n              serverInfo.transport,\n              serverInfo.customHeaders,\n              undefined,\n              undefined,\n              effectiveHandlers,\n            );\n\n            const connectedServer: ConnectedMcpServer = {\n              ...serverInfo,\n              client,\n            };\n\n            clientMap.set(key, connectedServer);\n            setConnectedMcpServers(Array.from(clientMap.values()));\n\n            // Register tools from this server (deduplicated by ownership)\n            try {\n              const tools = await client.listTools();\n              tools.forEach((tool) => {\n                // Skip if another server already owns this tool\n                const currentOwner = toolOwnerRef.current.get(tool.name);\n                if (currentOwner && currentOwner !== key) {\n                  return;\n                }\n\n                // Record ownership for this server key\n                if (!currentOwner) {\n                  toolOwnerRef.current.set(tool.name, key);\n                  if (!keyToToolsRef.current.has(key)) {\n                    keyToToolsRef.current.set(key, new Set());\n                  }\n                  keyToToolsRef.current.get(key)!.add(tool.name);\n                }\n\n                registerTool({\n                  description: tool.description ?? \"\",\n                  name: tool.name,\n                  tool: async (args: Record<string, unknown> = {}) => {\n                    const server = clientMap.get(key);\n                    if (!server?.client) {\n                      throw new Error(\n                        `MCP server for tool ${tool.name} is not connected`,\n                      );\n                    }\n                    const result = await server.client.callTool(\n                      tool.name,\n                      args,\n                    );\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                    if (isContentPartArray(content)) {\n                      return content;\n                    }\n                    return [{ type: \"text\", text: toText(content) }];\n                  },\n                });\n              });\n            } catch (error) {\n              console.error(\n                `Failed to register tools from MCP server ${serverInfo.url}:`,\n                error,\n              );\n            }\n          } catch (error) {\n            const failedServer: FailedMcpServer = {\n              ...serverInfo,\n              connectionError: error as Error,\n            };\n            clientMap.set(key, failedServer);\n            setConnectedMcpServers(Array.from(clientMap.values()));\n            console.error(\n              `Failed to connect to MCP server ${serverInfo.url}:`,\n              error,\n            );\n          }\n        }),\n      );\n    }\n\n    // Update state after removals (additions update state asynchronously above)\n    if (keysToRemove.length > 0) {\n      setConnectedMcpServers(Array.from(clientMap.values()));\n    }\n\n    // eslint-disable-next-line react-hooks/exhaustive-deps\n  }, [currentServersMap, registerTool]);\n\n  // Update handlers when they change (without recreating clients)\n  useEffect(() => {\n    const clientMap = clientMapRef.current;\n\n    clientMap.forEach((server, key) => {\n      if (!server.client) {\n        return; // Skip failed servers\n      }\n\n      const serverInfo = currentServersMap.get(key);\n      if (!serverInfo) {\n        return; // Server was removed, handled by main effect\n      }\n\n      // Determine effective handlers\n      const effectiveElicitationHandler =\n        serverInfo.handlers?.elicitation ??\n        (providerElicitationHandler\n          ? async (request, extra) =>\n              await providerElicitationHandler(request, extra, serverInfo)\n          : undefined);\n\n      const effectiveSamplingHandler =\n        serverInfo.handlers?.sampling ??\n        (providerSamplingHandler\n          ? async (request, extra) =>\n              await providerSamplingHandler(request, extra, serverInfo)\n          : undefined);\n\n      // Update handlers unconditionally (allows removal by passing undefined)\n      server.client.updateElicitationHandler?.(effectiveElicitationHandler);\n      server.client.updateSamplingHandler?.(effectiveSamplingHandler);\n    });\n  }, [currentServersMap, providerElicitationHandler, providerSamplingHandler]);\n\n  // Cleanup on unmount: close all clients\n  useEffect(() => {\n    const clientMap = clientMapRef.current;\n    const ownerMapAtMount = toolOwnerRef.current;\n    const keyToToolsAtMount = keyToToolsRef.current;\n    return () => {\n      clientMap.forEach((server, _key) => {\n        if (server.client) {\n          try {\n            server.client.close();\n          } catch (error) {\n            const url = (server as McpServer).url ?? \"(unknown url)\";\n            console.error(\n              `Error closing MCP client on unmount for ${url}:`,\n              error,\n            );\n          }\n        }\n      });\n      clientMap.clear();\n      ownerMapAtMount.clear();\n      keyToToolsAtMount.clear();\n    };\n  }, []);\n\n  const contextValue = useMemo(\n    () => ({\n      servers: connectedMcpServers,\n      elicitation,\n      resolveElicitation,\n    }),\n    [connectedMcpServers, elicitation, resolveElicitation],\n  );\n\n  return (\n    <McpProviderContext.Provider value={contextValue}>\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).servers;\n};\n\n/**\n * Hook to access elicitation context from TamboMcpProvider.\n * This provides access to the current elicitation request and methods to respond to it.\n *\n * The elicitation state is automatically managed by TamboMcpProvider when MCP servers\n * request user input through the elicitation protocol.\n * @returns The elicitation context with current request and response handlers\n * @example\n * ```tsx\n * function ElicitationUI() {\n *   const { elicitation, resolveElicitation } = useTamboElicitationContext();\n *\n *   if (!elicitation) return null;\n *\n *   return (\n *     <div>\n *       <p>{elicitation.message}</p>\n *       <button onClick={() => resolveElicitation?.({ action: \"accept\", content: {} })}>\n *         Accept\n *       </button>\n *     </div>\n *   );\n * }\n * ```\n */\nexport const useTamboElicitationContext = () => {\n  const context = useContext(McpProviderContext);\n  return {\n    elicitation: context.elicitation,\n    resolveElicitation: context.resolveElicitation,\n  };\n};\n\n/**\n * Creates a stable identifier for an MCP server based on its connection properties.\n * Two servers with the same URL, transport, and headers will have the same key.\n * @returns A stable string key identifying the server\n */\nfunction getServerKey(\n  serverInfo: Pick<McpServerInfo, \"url\" | \"transport\" | \"customHeaders\">,\n): string {\n  const headerStr = serverInfo.customHeaders\n    ? JSON.stringify(\n        Object.entries(serverInfo.customHeaders)\n          .map(([k, v]) => [k.toLowerCase(), v] as const)\n          .sort(([a], [b]) => a.localeCompare(b)),\n      )\n    : \"\";\n  return `${serverInfo.url}|${serverInfo.transport ?? MCPTransport.SSE}|${headerStr}`;\n}\n\n/**\n * Normalizes a server definition (string or object) into a McpServerInfo.\n * @returns The normalized McpServerInfo object\n */\nfunction normalizeServerInfo(server: McpServerInfo | string): McpServerConfig {\n  const s =\n    typeof server === \"string\"\n      ? { url: server, transport: MCPTransport.SSE }\n      : server;\n  const key = getServerKey(s);\n  return { ...s, key };\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"]}