@tambo-ai/react 0.64.1 → 0.65.2

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

@@ -1,6 +1,7 @@
 import React, { createContext, useContext, useEffect, useMemo, useRef, useState, } from "react";
+import { getMcpServerUniqueKey, } from "../model/mcp-server-info";
 import { useTamboMcpToken } from "../providers/tambo-mcp-token-provider";
-import { useTamboRegistry } from "../providers/tambo-registry-provider";
+import { useTamboMcpServerInfos, useTamboRegistry, } from "../providers/tambo-registry-provider";
 import { isContentPartArray, toText } from "../util/content-parts";
 import { useElicitation } from "./elicitation";
 import { MCPClient, MCPTransport, } from "./mcp-client";
@@ -37,15 +38,19 @@ const TAMBO_INTERNAL_MCP_SERVER_NAME = "__tambo_internal_mcp_server__";
 /**
  * This provider is used to register tools from MCP servers.
  * It automatically includes an internal Tambo MCP server when an MCP access token is available.
+ *
+ * **BREAKING CHANGE**: This provider no longer accepts `mcpServers` as a prop.
+ * Instead, pass `mcpServers` to `TamboProvider` or `TamboRegistryProvider`.
+ * This provider must be wrapped inside `TamboProvider` to access the MCP server registry.
  * @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, handlers, children }) => {
+export const TamboMcpProvider = ({ handlers, children }) => {
     const { registerTool } = useTamboRegistry();
     const { mcpAccessToken, tamboBaseUrl } = useTamboMcpToken();
+    const mcpServers = useTamboMcpServerInfos();
     const providerSamplingHandler = handlers?.sampling;
     // Elicitation state and default handler
     const { elicitation, resolveElicitation, defaultElicitationHandler } = useElicitation();
@@ -71,6 +76,7 @@ export const TamboMcpProvider = ({ mcpServers, handlers, children }) => {
                 name: TAMBO_INTERNAL_MCP_SERVER_NAME,
                 url: tamboMcpUrl,
                 transport: MCPTransport.HTTP,
+                serverKey: "tambo", // Internal server always uses 'tambo' as serverKey
                 customHeaders: {
                     Authorization: `Bearer ${mcpAccessToken}`,
                 },
@@ -207,8 +213,12 @@ export const TamboMcpProvider = ({ mcpServers, handlers, children }) => {
         if (keysToRemove.length > 0) {
             setConnectedMcpServers(Array.from(clientMap.values()));
         }
-        // eslint-disable-next-line react-hooks/exhaustive-deps
-    }, [currentServersMap, registerTool]);
+    }, [
+        currentServersMap,
+        providerElicitationHandler,
+        providerSamplingHandler,
+        registerTool,
+    ]);
     // Update handlers when they change (without recreating clients)
     useEffect(() => {
         const clientMap = clientMapRef.current;
@@ -290,16 +300,16 @@ export const useTamboMcpServers = () => {
     return useContext(McpProviderContext).servers;
 };
 /**
- * Hook to access elicitation context from TamboMcpProvider.
+ * Hook to access MCP elicitation state 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
+ * @returns The elicitation state with current request and response handler
  * @example
  * ```tsx
  * function ElicitationUI() {
- *   const { elicitation, resolveElicitation } = useTamboElicitationContext();
+ *   const { elicitation, resolveElicitation } = useTamboMcpElicitation();
  *
  *   if (!elicitation) return null;
  *
@@ -314,7 +324,7 @@ export const useTamboMcpServers = () => {
  * }
  * ```
  */
-export const useTamboElicitationContext = () => {
+export const useTamboMcpElicitation = () => {
     const context = useContext(McpProviderContext);
     return {
         elicitation: context.elicitation,
@@ -322,86 +332,25 @@ export const useTamboElicitationContext = () => {
     };
 };
 /**
- * Derives a short server key from a URL hostname using heuristics.
- * Attempts to extract the "meaningful" part of the domain name.
- *
- * Examples:
- * - "https://mcp.linear.app/mcp" -> "linear"
- * - "https://api.github.com" -> "github"
- * - "https://google.com" -> "google"
- * - "https://google.co.uk" -> "google"
- * - "https://mcp.company.co.uk" -> "company"
- * @param url - The URL to derive a server key from
- * @returns A short server key derived from the hostname
+ * @deprecated Use `useTamboMcpElicitation` instead.
+ * This hook will be removed in a future version.
  */
-function deriveServerKey(url) {
-    try {
-        const parsed = new URL(url);
-        const hostname = parsed.hostname;
-        // Split hostname into parts
-        const parts = hostname.split(".");
-        // Remove common TLD patterns
-        // Handle cases like: .com, .org, .co.uk, .com.au, etc.
-        let relevantParts = [...parts];
-        // If we have 3+ parts and the last two are short (likely TLD like .co.uk)
-        if (relevantParts.length >= 3 &&
-            relevantParts[relevantParts.length - 1].length <= 3 &&
-            relevantParts[relevantParts.length - 2].length <= 3) {
-            relevantParts = relevantParts.slice(0, -2);
-        }
-        // Otherwise just remove the last part (TLD like .com)
-        else if (relevantParts.length >= 2) {
-            relevantParts = relevantParts.slice(0, -1);
-        }
-        // From what's left, prefer the rightmost part that's not a common prefix
-        // Common prefixes: www, api, mcp, app, etc.
-        const commonPrefixes = new Set([
-            "www",
-            "api",
-            "mcp",
-            "app",
-            "staging",
-            "dev",
-            "prod",
-        ]);
-        // Work backwards through the parts to find a meaningful name
-        for (let i = relevantParts.length - 1; i >= 0; i--) {
-            const part = relevantParts[i];
-            if (part && !commonPrefixes.has(part.toLowerCase())) {
-                return part.toLowerCase();
-            }
-        }
-        // Fallback: use the last relevant part even if it's a common prefix
-        return relevantParts[relevantParts.length - 1]?.toLowerCase() || hostname;
-    }
-    catch {
-        // If URL parsing fails, just return a sanitized version of the input
-        return url.replace(/[^a-zA-Z0-9]/g, "_").toLowerCase();
-    }
-}
-/**
- * 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.HTTP}|${headerStr}`;
-}
+export const useTamboElicitationContext = useTamboMcpElicitation;
 /**
- * Normalizes a server definition (string or object) into a McpServerInfo.
- * @returns The normalized McpServerInfo object
+ * Normalizes registry server metadata into a `McpServerConfig`.
+ *
+ * Accepts a `NormalizedMcpServerInfo`, which already guarantees a concrete
+ * `transport` and a `serverKey` derived by the registry, and narrows the
+ * opaque `handlers` field to `Partial<MCPHandlers>`.
  */
 function normalizeServerInfo(server) {
-    const s = typeof server === "string"
-        ? { url: server, transport: MCPTransport.HTTP }
-        : server;
-    const key = getServerKey(s);
-    const serverKey = s.serverKey ?? deriveServerKey(s.url);
-    return { ...s, key, serverKey };
+    const key = getMcpServerUniqueKey(server);
+    // Cast handlers to proper type if present
+    const handlers = server.handlers;
+    return {
+        ...server,
+        handlers,
+        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,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;AAoGD,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,MAAM,YAAY,GAAG,iBAAiB,CAAC,IAAI,GAAG,CAAC,CAAC;wBAEhD,KAAK,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,EAAE;4BACrB,kEAAkE;4BAClE,MAAM,QAAQ,GAAG,YAAY;gCAC3B,CAAC,CAAC,GAAG,UAAU,CAAC,SAAS,KAAK,IAAI,CAAC,IAAI,EAAE;gCACzC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC;4BAEd,iFAAiF;4BACjF,MAAM,YAAY,GAAG,YAAY,CAAC,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;4BACxD,IAAI,YAAY,IAAI,YAAY,KAAK,GAAG,EAAE,CAAC;gCACzC,OAAO;4BACT,CAAC;4BAED,0DAA0D;4BAC1D,IAAI,CAAC,YAAY,EAAE,CAAC;gCAClB,YAAY,CAAC,OAAO,CAAC,GAAG,CAAC,QAAQ,EAAE,GAAG,CAAC,CAAC;gCACxC,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,QAAQ,CAAC,CAAC;4BAChD,CAAC;4BAED,YAAY,CAAC;gCACX,WAAW,EAAE,IAAI,CAAC,WAAW,IAAI,EAAE;gCACnC,IAAI,EAAE,QAAQ;gCACd,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;;;;;;;;;;;;GAYG;AAEH,SAAS,eAAe,CAAC,GAAW;IAClC,IAAI,CAAC;QACH,MAAM,MAAM,GAAG,IAAI,GAAG,CAAC,GAAG,CAAC,CAAC;QAC5B,MAAM,QAAQ,GAAG,MAAM,CAAC,QAAQ,CAAC;QAEjC,4BAA4B;QAC5B,MAAM,KAAK,GAAG,QAAQ,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;QAElC,6BAA6B;QAC7B,uDAAuD;QACvD,IAAI,aAAa,GAAG,CAAC,GAAG,KAAK,CAAC,CAAC;QAE/B,0EAA0E;QAC1E,IACE,aAAa,CAAC,MAAM,IAAI,CAAC;YACzB,aAAa,CAAC,aAAa,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,MAAM,IAAI,CAAC;YACnD,aAAa,CAAC,aAAa,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,MAAM,IAAI,CAAC,EACnD,CAAC;YACD,aAAa,GAAG,aAAa,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;QAC7C,CAAC;QACD,sDAAsD;aACjD,IAAI,aAAa,CAAC,MAAM,IAAI,CAAC,EAAE,CAAC;YACnC,aAAa,GAAG,aAAa,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;QAC7C,CAAC;QAED,yEAAyE;QACzE,4CAA4C;QAC5C,MAAM,cAAc,GAAG,IAAI,GAAG,CAAC;YAC7B,KAAK;YACL,KAAK;YACL,KAAK;YACL,KAAK;YACL,SAAS;YACT,KAAK;YACL,MAAM;SACP,CAAC,CAAC;QAEH,6DAA6D;QAC7D,KAAK,IAAI,CAAC,GAAG,aAAa,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;YACnD,MAAM,IAAI,GAAG,aAAa,CAAC,CAAC,CAAC,CAAC;YAC9B,IAAI,IAAI,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC,EAAE,CAAC;gBACpD,OAAO,IAAI,CAAC,WAAW,EAAE,CAAC;YAC5B,CAAC;QACH,CAAC;QAED,oEAAoE;QACpE,OAAO,aAAa,CAAC,aAAa,CAAC,MAAM,GAAG,CAAC,CAAC,EAAE,WAAW,EAAE,IAAI,QAAQ,CAAC;IAC5E,CAAC;IAAC,MAAM,CAAC;QACP,qEAAqE;QACrE,OAAO,GAAG,CAAC,OAAO,CAAC,eAAe,EAAE,GAAG,CAAC,CAAC,WAAW,EAAE,CAAC;IACzD,CAAC;AACH,CAAC;AAED;;;;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,IAAI,IAAI,SAAS,EAAE,CAAC;AACvF,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,IAAI,EAAE;QAC/C,CAAC,CAAC,MAAM,CAAC;IACb,MAAM,GAAG,GAAG,YAAY,CAAC,CAAC,CAAC,CAAC;IAC5B,MAAM,SAAS,GAAG,CAAC,CAAC,SAAS,IAAI,eAAe,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC;IACxD,OAAO,EAAE,GAAG,CAAC,EAAE,GAAG,EAAE,SAAS,EAAE,CAAC;AAClC,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 HTTP for string URLs */\n  transport?: MCPTransport;\n  /** Optional custom headers to include in requests */\n  customHeaders?: Record<string, string>;\n  /**\n   * Optional short name for namespacing MCP resources, prompts, and tools.\n   * When multiple MCP servers are configured, this key is used to prefix:\n   * - prompts: `<serverKey>:<promptName>`\n   * - resources: `<serverKey>:<resourceUrl>`\n   * - tools: `<serverKey>__<toolName>`\n   *\n   * If not provided, a key will be derived from the URL hostname.\n   * For example, \"https://mcp.linear.app/mcp\" becomes \"linear\".\n   */\n  serverKey?: 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   * Short name for namespacing, either provided by user or derived from URL.\n   * Used to prefix tools, prompts, and resources when multiple servers are present.\n   */\n  serverKey: 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              const shouldPrefix = currentServersMap.size > 1;\n\n              tools.forEach((tool) => {\n                // Prefix tool name with serverKey if multiple servers are present\n                const toolName = shouldPrefix\n                  ? `${serverInfo.serverKey}__${tool.name}`\n                  : tool.name;\n\n                // Skip if another server already owns this tool (using final name for ownership)\n                const currentOwner = toolOwnerRef.current.get(toolName);\n                if (currentOwner && currentOwner !== key) {\n                  return;\n                }\n\n                // Record ownership for this server key (using final name)\n                if (!currentOwner) {\n                  toolOwnerRef.current.set(toolName, key);\n                  if (!keyToToolsRef.current.has(key)) {\n                    keyToToolsRef.current.set(key, new Set());\n                  }\n                  keyToToolsRef.current.get(key)!.add(toolName);\n                }\n\n                registerTool({\n                  description: tool.description ?? \"\",\n                  name: toolName,\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 * Derives a short server key from a URL hostname using heuristics.\n * Attempts to extract the \"meaningful\" part of the domain name.\n *\n * Examples:\n * - \"https://mcp.linear.app/mcp\" -> \"linear\"\n * - \"https://api.github.com\" -> \"github\"\n * - \"https://google.com\" -> \"google\"\n * - \"https://google.co.uk\" -> \"google\"\n * - \"https://mcp.company.co.uk\" -> \"company\"\n * @param url - The URL to derive a server key from\n * @returns A short server key derived from the hostname\n */\n\nfunction deriveServerKey(url: string): string {\n  try {\n    const parsed = new URL(url);\n    const hostname = parsed.hostname;\n\n    // Split hostname into parts\n    const parts = hostname.split(\".\");\n\n    // Remove common TLD patterns\n    // Handle cases like: .com, .org, .co.uk, .com.au, etc.\n    let relevantParts = [...parts];\n\n    // If we have 3+ parts and the last two are short (likely TLD like .co.uk)\n    if (\n      relevantParts.length >= 3 &&\n      relevantParts[relevantParts.length - 1].length <= 3 &&\n      relevantParts[relevantParts.length - 2].length <= 3\n    ) {\n      relevantParts = relevantParts.slice(0, -2);\n    }\n    // Otherwise just remove the last part (TLD like .com)\n    else if (relevantParts.length >= 2) {\n      relevantParts = relevantParts.slice(0, -1);\n    }\n\n    // From what's left, prefer the rightmost part that's not a common prefix\n    // Common prefixes: www, api, mcp, app, etc.\n    const commonPrefixes = new Set([\n      \"www\",\n      \"api\",\n      \"mcp\",\n      \"app\",\n      \"staging\",\n      \"dev\",\n      \"prod\",\n    ]);\n\n    // Work backwards through the parts to find a meaningful name\n    for (let i = relevantParts.length - 1; i >= 0; i--) {\n      const part = relevantParts[i];\n      if (part && !commonPrefixes.has(part.toLowerCase())) {\n        return part.toLowerCase();\n      }\n    }\n\n    // Fallback: use the last relevant part even if it's a common prefix\n    return relevantParts[relevantParts.length - 1]?.toLowerCase() || hostname;\n  } catch {\n    // If URL parsing fails, just return a sanitized version of the input\n    return url.replace(/[^a-zA-Z0-9]/g, \"_\").toLowerCase();\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.HTTP}|${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.HTTP }\n      : server;\n  const key = getServerKey(s);\n  const serverKey = s.serverKey ?? deriveServerKey(s.url);\n  return { ...s, key, serverKey };\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,EACL,qBAAqB,GAEtB,MAAM,0BAA0B,CAAC;AAClC,OAAO,EAAE,gBAAgB,EAAE,MAAM,uCAAuC,CAAC;AACzE,OAAO,EACL,sBAAsB,EACtB,gBAAgB,GACjB,MAAM,sCAAsC,CAAC;AAC9C,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;AA2ED,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;;;;;;;;;;;GAWG;AACH,MAAM,CAAC,MAAM,gBAAgB,GAGxB,CAAC,EAAE,QAAQ,EAAE,QAAQ,EAAE,EAAE,EAAE;IAC9B,MAAM,EAAE,YAAY,EAAE,GAAG,gBAAgB,EAAE,CAAC;IAC5C,MAAM,EAAE,cAAc,EAAE,YAAY,EAAE,GAAG,gBAAgB,EAAE,CAAC;IAC5D,MAAM,UAAU,GAAG,sBAAsB,EAAE,CAAC;IAC5C,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,SAAS,EAAE,OAAO,EAAE,mDAAmD;gBACvE,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,EACnC,OAA6C,EAC7C,KAA2C,EAC3C,EAAE,CAAC,MAAM,0BAA0B,CAAC,OAAO,EAAE,KAAK,EAAE,UAAU,CAAC,CAAC;oBACpE,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,EAChC,OAA0C,EAC1C,KAAwC,EACxC,EAAE,CAAC,MAAM,uBAAuB,CAAC,OAAO,EAAE,KAAK,EAAE,UAAU,CAAC,CAAC;oBACjE,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,MAAM,YAAY,GAAG,iBAAiB,CAAC,IAAI,GAAG,CAAC,CAAC;wBAEhD,KAAK,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,EAAE;4BACrB,kEAAkE;4BAClE,MAAM,QAAQ,GAAG,YAAY;gCAC3B,CAAC,CAAC,GAAG,UAAU,CAAC,SAAS,KAAK,IAAI,CAAC,IAAI,EAAE;gCACzC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC;4BAEd,iFAAiF;4BACjF,MAAM,YAAY,GAAG,YAAY,CAAC,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;4BACxD,IAAI,YAAY,IAAI,YAAY,KAAK,GAAG,EAAE,CAAC;gCACzC,OAAO;4BACT,CAAC;4BAED,0DAA0D;4BAC1D,IAAI,CAAC,YAAY,EAAE,CAAC;gCAClB,YAAY,CAAC,OAAO,CAAC,GAAG,CAAC,QAAQ,EAAE,GAAG,CAAC,CAAC;gCACxC,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,QAAQ,CAAC,CAAC;4BAChD,CAAC;4BAED,YAAY,CAAC;gCACX,WAAW,EAAE,IAAI,CAAC,WAAW,IAAI,EAAE;gCACnC,IAAI,EAAE,QAAQ;gCACd,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;IACH,CAAC,EAAE;QACD,iBAAiB;QACjB,0BAA0B;QAC1B,uBAAuB;QACvB,YAAY;KACb,CAAC,CAAC;IAEH,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,EACH,OAA6C,EAC7C,KAA2C,EAC3C,EAAE,CAAC,MAAM,0BAA0B,CAAC,OAAO,EAAE,KAAK,EAAE,UAAU,CAAC;oBACnE,CAAC,CAAC,SAAS,CAAC,CAAC;YAEjB,MAAM,wBAAwB,GAC5B,UAAU,CAAC,QAAQ,EAAE,QAAQ;gBAC7B,CAAC,uBAAuB;oBACtB,CAAC,CAAC,KAAK,EACH,OAA0C,EAC1C,KAAwC,EACxC,EAAE,CAAC,MAAM,uBAAuB,CAAC,OAAO,EAAE,KAAK,EAAE,UAAU,CAAC;oBAChE,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,sBAAsB,GAAG,GAA4B,EAAE;IAClE,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;;;GAGG;AACH,MAAM,CAAC,MAAM,0BAA0B,GAAG,sBAAsB,CAAC;AAEjE;;;;;;GAMG;AACH,SAAS,mBAAmB,CAAC,MAA+B;IAC1D,MAAM,GAAG,GAAG,qBAAqB,CAAC,MAAM,CAAC,CAAC;IAC1C,0CAA0C;IAC1C,MAAM,QAAQ,GAAG,MAAM,CAAC,QAA4C,CAAC;IACrE,OAAO;QACL,GAAG,MAAM;QACT,QAAQ;QACR,GAAG;KACJ,CAAC;AACJ,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 {\n  getMcpServerUniqueKey,\n  type NormalizedMcpServerInfo,\n} from \"../model/mcp-server-info\";\nimport { useTamboMcpToken } from \"../providers/tambo-mcp-token-provider\";\nimport {\n  useTamboMcpServerInfos,\n  useTamboRegistry,\n} 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 * Normalized MCP server information as consumed by the provider.\n *\n * Extends `NormalizedMcpServerInfo` from the core model by:\n * - narrowing `handlers` to `Partial<MCPHandlers>`\n * - adding a stable `key` derived from URL/transport/headers\n *\n * The registry is responsible for producing `NormalizedMcpServerInfo`\n * instances; this type adds the MCP-specific wiring needed to connect and\n * track clients.\n */\ninterface McpServerConfig extends NormalizedMcpServerInfo {\n  /**\n   * Optional handlers for elicitation and sampling requests from the server.\n   * Interpreted as a partial set of MCP handlers.\n   */\n  handlers?: Partial<MCPHandlers>;\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 *\n * **BREAKING CHANGE**: This provider no longer accepts `mcpServers` as a prop.\n * Instead, pass `mcpServers` to `TamboProvider` or `TamboRegistryProvider`.\n * This provider must be wrapped inside `TamboProvider` to access the MCP server registry.\n * @param props - The provider props\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  handlers?: ProviderMCPHandlers;\n  children: React.ReactNode;\n}> = ({ handlers, children }) => {\n  const { registerTool } = useTamboRegistry();\n  const { mcpAccessToken, tamboBaseUrl } = useTamboMcpToken();\n  const mcpServers = useTamboMcpServerInfos();\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        serverKey: \"tambo\", // Internal server always uses 'tambo' as serverKey\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 (\n                request: Parameters<MCPElicitationHandler>[0],\n                extra: Parameters<MCPElicitationHandler>[1],\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 (\n                request: Parameters<MCPSamplingHandler>[0],\n                extra: Parameters<MCPSamplingHandler>[1],\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              const shouldPrefix = currentServersMap.size > 1;\n\n              tools.forEach((tool) => {\n                // Prefix tool name with serverKey if multiple servers are present\n                const toolName = shouldPrefix\n                  ? `${serverInfo.serverKey}__${tool.name}`\n                  : tool.name;\n\n                // Skip if another server already owns this tool (using final name for ownership)\n                const currentOwner = toolOwnerRef.current.get(toolName);\n                if (currentOwner && currentOwner !== key) {\n                  return;\n                }\n\n                // Record ownership for this server key (using final name)\n                if (!currentOwner) {\n                  toolOwnerRef.current.set(toolName, key);\n                  if (!keyToToolsRef.current.has(key)) {\n                    keyToToolsRef.current.set(key, new Set());\n                  }\n                  keyToToolsRef.current.get(key)!.add(toolName);\n                }\n\n                registerTool({\n                  description: tool.description ?? \"\",\n                  name: toolName,\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    currentServersMap,\n    providerElicitationHandler,\n    providerSamplingHandler,\n    registerTool,\n  ]);\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 (\n              request: Parameters<MCPElicitationHandler>[0],\n              extra: Parameters<MCPElicitationHandler>[1],\n            ) => await providerElicitationHandler(request, extra, serverInfo)\n          : undefined);\n\n      const effectiveSamplingHandler =\n        serverInfo.handlers?.sampling ??\n        (providerSamplingHandler\n          ? async (\n              request: Parameters<MCPSamplingHandler>[0],\n              extra: Parameters<MCPSamplingHandler>[1],\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 MCP elicitation state 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 state with current request and response handler\n * @example\n * ```tsx\n * function ElicitationUI() {\n *   const { elicitation, resolveElicitation } = useTamboMcpElicitation();\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 useTamboMcpElicitation = (): ElicitationContextState => {\n  const context = useContext(McpProviderContext);\n  return {\n    elicitation: context.elicitation,\n    resolveElicitation: context.resolveElicitation,\n  };\n};\n\n/**\n * @deprecated Use `useTamboMcpElicitation` instead.\n * This hook will be removed in a future version.\n */\nexport const useTamboElicitationContext = useTamboMcpElicitation;\n\n/**\n * Normalizes registry server metadata into a `McpServerConfig`.\n *\n * Accepts a `NormalizedMcpServerInfo`, which already guarantees a concrete\n * `transport` and a `serverKey` derived by the registry, and narrows the\n * opaque `handlers` field to `Partial<MCPHandlers>`.\n */\nfunction normalizeServerInfo(server: NormalizedMcpServerInfo): McpServerConfig {\n  const key = getMcpServerUniqueKey(server);\n  // Cast handlers to proper type if present\n  const handlers = server.handlers as Partial<MCPHandlers> | undefined;\n  return {\n    ...server,\n    handlers,\n    key,\n  };\n}\n"]}

package/esm/model/mcp-server-info.d.ts

@@ -0,0 +1,74 @@
+/**
+ * The transport protocol to use for MCP connections.
+ */
+export declare enum MCPTransport {
+    SSE = "sse",
+    HTTP = "http"
+}
+/**
+ * User-provided configuration for an MCP server.
+ *
+ * This is the type accepted by `TamboProvider` / `TamboRegistryProvider` in
+ * the `mcpServers` prop.
+ *
+ * The `handlers` field is intentionally typed as `unknown` here so the core
+ * SDK does not depend on the MCP subpackage. In the `@tambo-ai/react/mcp`
+ * subpackage this is treated as `Partial<MCPHandlers>` (with
+ * `elicitation` / `sampling` callbacks).
+ */
+export interface McpServerInfo {
+    /** Optional name for the MCP server */
+    name?: string;
+    /** The URL of the MCP server to connect to */
+    url: string;
+    /** Optional description of the MCP server */
+    description?: string;
+    /** The transport type to use (SSE or HTTP). Defaults to HTTP for string URLs */
+    transport?: MCPTransport;
+    /** Optional custom headers to include in requests */
+    customHeaders?: Record<string, string>;
+    /**
+     * Optional short name for namespacing MCP resources, prompts, and tools.
+     * When multiple MCP servers are configured, this key is used to prefix:
+     * - prompts: `<serverKey>:<promptName>`
+     * - resources: `<serverKey>:<resourceUrl>`
+     * - tools: `<serverKey>__<toolName>`
+     *
+     * If not provided, a key will be derived from the URL hostname.
+     * For example, "https://mcp.linear.app/mcp" becomes "linear".
+     */
+    serverKey?: string;
+    /**
+     * Optional handlers for elicitation and sampling requests from the server.
+     *
+     * In the MCP subpackage this is interpreted as `Partial<MCPHandlers>`,
+     * i.e. `{ elicitation?: MCPElicitationHandler; sampling?: MCPSamplingHandler }`.
+     *
+     * Note: These callbacks should be stable (e.g., wrapped in useCallback or
+     * defined outside the component) to avoid constant re-registration of the
+     * MCP server on every render.
+     */
+    handlers?: unknown;
+}
+/**
+ * Normalized MCP server metadata used internally by the registry and MCP
+ * provider.
+ *
+ * This is equivalent to `McpServerInfo` except that:
+ * - `serverKey` is guaranteed to be present
+ * - `transport` is resolved to a concrete value (defaults to HTTP)
+ */
+export interface NormalizedMcpServerInfo extends McpServerInfo {
+    transport: MCPTransport;
+    serverKey: string;
+}
+/**
+ * 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.
+ *
+ * This is used by both the registry and MCP provider to deduplicate servers,
+ * so it lives alongside the shared server metadata type.
+ * @returns A stable string key identifying the server
+ */
+export declare function getMcpServerUniqueKey(serverInfo: Pick<McpServerInfo, "url" | "transport" | "customHeaders">): string;
+//# sourceMappingURL=mcp-server-info.d.ts.map

package/esm/model/mcp-server-info.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"mcp-server-info.d.ts","sourceRoot":"","sources":["../../src/model/mcp-server-info.ts"],"names":[],"mappings":"AAAA;;GAEG;AACH,oBAAY,YAAY;IACtB,GAAG,QAAQ;IACX,IAAI,SAAS;CACd;AAED;;;;;;;;;;GAUG;AACH,MAAM,WAAW,aAAa;IAC5B,uCAAuC;IACvC,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,8CAA8C;IAC9C,GAAG,EAAE,MAAM,CAAC;IACZ,6CAA6C;IAC7C,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,gFAAgF;IAChF,SAAS,CAAC,EAAE,YAAY,CAAC;IACzB,qDAAqD;IACrD,aAAa,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IACvC;;;;;;;;;OASG;IACH,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB;;;;;;;;;OASG;IACH,QAAQ,CAAC,EAAE,OAAO,CAAC;CACpB;AAED;;;;;;;GAOG;AACH,MAAM,WAAW,uBAAwB,SAAQ,aAAa;IAC5D,SAAS,EAAE,YAAY,CAAC;IACxB,SAAS,EAAE,MAAM,CAAC;CACnB;AAED;;;;;;;GAOG;AACH,wBAAgB,qBAAqB,CACnC,UAAU,EAAE,IAAI,CAAC,aAAa,EAAE,KAAK,GAAG,WAAW,GAAG,eAAe,CAAC,GACrE,MAAM,CAUR"}

package/esm/model/mcp-server-info.js

@@ -0,0 +1,25 @@
+/**
+ * The transport protocol to use for MCP connections.
+ */
+export var MCPTransport;
+(function (MCPTransport) {
+    MCPTransport["SSE"] = "sse";
+    MCPTransport["HTTP"] = "http";
+})(MCPTransport || (MCPTransport = {}));
+/**
+ * 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.
+ *
+ * This is used by both the registry and MCP provider to deduplicate servers,
+ * so it lives alongside the shared server metadata type.
+ * @returns A stable string key identifying the server
+ */
+export function getMcpServerUniqueKey(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.HTTP}|${headerStr}`;
+}
+//# sourceMappingURL=mcp-server-info.js.map

package/esm/model/mcp-server-info.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"mcp-server-info.js","sourceRoot":"","sources":["../../src/model/mcp-server-info.ts"],"names":[],"mappings":"AAAA;;GAEG;AACH,MAAM,CAAN,IAAY,YAGX;AAHD,WAAY,YAAY;IACtB,2BAAW,CAAA;IACX,6BAAa,CAAA;AACf,CAAC,EAHW,YAAY,KAAZ,YAAY,QAGvB;AA6DD;;;;;;;GAOG;AACH,MAAM,UAAU,qBAAqB,CACnC,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;IAEP,OAAO,GAAG,UAAU,CAAC,GAAG,IAAI,UAAU,CAAC,SAAS,IAAI,YAAY,CAAC,IAAI,IAAI,SAAS,EAAE,CAAC;AACvF,CAAC","sourcesContent":["/**\n * The transport protocol to use for MCP connections.\n */\nexport enum MCPTransport {\n  SSE = \"sse\",\n  HTTP = \"http\",\n}\n\n/**\n * User-provided configuration for an MCP server.\n *\n * This is the type accepted by `TamboProvider` / `TamboRegistryProvider` in\n * the `mcpServers` prop.\n *\n * The `handlers` field is intentionally typed as `unknown` here so the core\n * SDK does not depend on the MCP subpackage. In the `@tambo-ai/react/mcp`\n * subpackage this is treated as `Partial<MCPHandlers>` (with\n * `elicitation` / `sampling` callbacks).\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 HTTP for string URLs */\n  transport?: MCPTransport;\n  /** Optional custom headers to include in requests */\n  customHeaders?: Record<string, string>;\n  /**\n   * Optional short name for namespacing MCP resources, prompts, and tools.\n   * When multiple MCP servers are configured, this key is used to prefix:\n   * - prompts: `<serverKey>:<promptName>`\n   * - resources: `<serverKey>:<resourceUrl>`\n   * - tools: `<serverKey>__<toolName>`\n   *\n   * If not provided, a key will be derived from the URL hostname.\n   * For example, \"https://mcp.linear.app/mcp\" becomes \"linear\".\n   */\n  serverKey?: string;\n  /**\n   * Optional handlers for elicitation and sampling requests from the server.\n   *\n   * In the MCP subpackage this is interpreted as `Partial<MCPHandlers>`,\n   * i.e. `{ elicitation?: MCPElicitationHandler; sampling?: MCPSamplingHandler }`.\n   *\n   * Note: These callbacks should be stable (e.g., wrapped in useCallback or\n   * defined outside the component) to avoid constant re-registration of the\n   * MCP server on every render.\n   */\n  handlers?: unknown;\n}\n\n/**\n * Normalized MCP server metadata used internally by the registry and MCP\n * provider.\n *\n * This is equivalent to `McpServerInfo` except that:\n * - `serverKey` is guaranteed to be present\n * - `transport` is resolved to a concrete value (defaults to HTTP)\n */\nexport interface NormalizedMcpServerInfo extends McpServerInfo {\n  transport: MCPTransport;\n  serverKey: string;\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 *\n * This is used by both the registry and MCP provider to deduplicate servers,\n * so it lives alongside the shared server metadata type.\n * @returns A stable string key identifying the server\n */\nexport function getMcpServerUniqueKey(\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\n  return `${serverInfo.url}|${serverInfo.transport ?? MCPTransport.HTTP}|${headerStr}`;\n}\n"]}

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

@@ -2,7 +2,7 @@ import { advanceStream } from "@tambo-ai/typescript-sdk";
 import { act, renderHook } from "@testing-library/react";
 import React from "react";
 import { GenerationStage, } from "../../model/generate-component-response";
-import { useTamboClient, useTamboQueryClient } from "../tambo-client-provider";
+import { TamboClientContext, 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";
@@ -14,10 +14,13 @@ Object.defineProperty(global, "crypto", {
     },
 });
 // Mock the required providers
-jest.mock("../tambo-client-provider", () => ({
-    useTamboClient: jest.fn(),
-    useTamboQueryClient: jest.fn(),
-}));
+jest.mock("../tambo-client-provider", () => {
+    return {
+        useTamboClient: jest.fn(),
+        useTamboQueryClient: jest.fn(),
+        TamboClientContext: React.createContext(undefined),
+    };
+});
 jest.mock("@tambo-ai/typescript-sdk", () => ({
     advanceStream: jest.fn(),
 }));
@@ -33,10 +36,21 @@ const createMockMessage = (overrides = {}) => ({
 });
 // Test wrapper
 const createWrapper = (initialMessages = []) => {
-    const TestWrapper = ({ children }) => (React.createElement(TamboRegistryProvider, { components: [], tools: [] },
-        React.createElement(TamboContextHelpersProvider, null,
-            React.createElement(TamboMcpTokenProvider, null,
-                React.createElement(TamboThreadProvider, { initialMessages: initialMessages, autoGenerateThreadName: false }, children)))));
+    const TestWrapper = ({ children }) => {
+        const client = useTamboClient();
+        const queryClient = useTamboQueryClient();
+        return (React.createElement(TamboClientContext.Provider, { value: {
+                client,
+                queryClient,
+                isUpdatingToken: false,
+                mcpAccessToken: null,
+                setMcpAccessToken: () => { },
+            } },
+            React.createElement(TamboRegistryProvider, { components: [], tools: [] },
+                React.createElement(TamboContextHelpersProvider, null,
+                    React.createElement(TamboMcpTokenProvider, null,
+                        React.createElement(TamboThreadProvider, { initialMessages: initialMessages, autoGenerateThreadName: false }, 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,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,IAClB,eAAe,EAAE,eAAe,EAChC,sBAAsB,EAAE,KAAK,IAE5B,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\n            initialMessages={initialMessages}\n            autoGenerateThreadName={false}\n          >\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"]}
+{"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,EACL,kBAAkB,EAClB,cAAc,EACd,mBAAmB,GACpB,MAAM,0BAA0B,CAAC;AAClC,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;IACzC,OAAO;QACL,cAAc,EAAE,IAAI,CAAC,EAAE,EAAE;QACzB,mBAAmB,EAAE,IAAI,CAAC,EAAE,EAAE;QAC9B,kBAAkB,EAAE,KAAK,CAAC,aAAa,CAAC,SAAS,CAAC;KACnD,CAAC;AACJ,CAAC,CAAC,CAAC;AACH,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;QAClE,MAAM,MAAM,GAAG,cAAc,EAAE,CAAC;QAChC,MAAM,WAAW,GAAG,mBAAmB,EAAE,CAAC;QAE1C,OAAO,CACL,oBAAC,kBAAkB,CAAC,QAAQ,IAC1B,KAAK,EAAE;gBACL,MAAM;gBACN,WAAW;gBACX,eAAe,EAAE,KAAK;gBACtB,cAAc,EAAE,IAAI;gBACpB,iBAAiB,EAAE,GAAG,EAAE,GAAE,CAAC;aAC5B;YAED,oBAAC,qBAAqB,IAAC,UAAU,EAAE,EAAE,EAAE,KAAK,EAAE,EAAE;gBAC9C,oBAAC,2BAA2B;oBAC1B,oBAAC,qBAAqB;wBACpB,oBAAC,mBAAmB,IAClB,eAAe,EAAE,eAAe,EAChC,sBAAsB,EAAE,KAAK,IAE5B,QAAQ,CACW,CACA,CACI,CACR,CACI,CAC/B,CAAC;IACJ,CAAC,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 {\n  TamboClientContext,\n  useTamboClient,\n  useTamboQueryClient,\n} 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  return {\n    useTamboClient: jest.fn(),\n    useTamboQueryClient: jest.fn(),\n    TamboClientContext: React.createContext(undefined),\n  };\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    const client = useTamboClient();\n    const queryClient = useTamboQueryClient();\n\n    return (\n      <TamboClientContext.Provider\n        value={{\n          client,\n          queryClient,\n          isUpdatingToken: false,\n          mcpAccessToken: null,\n          setMcpAccessToken: () => {},\n        }}\n      >\n        <TamboRegistryProvider components={[]} tools={[]}>\n          <TamboContextHelpersProvider>\n            <TamboMcpTokenProvider>\n              <TamboThreadProvider\n                initialMessages={initialMessages}\n                autoGenerateThreadName={false}\n              >\n                {children}\n              </TamboThreadProvider>\n            </TamboMcpTokenProvider>\n          </TamboContextHelpersProvider>\n        </TamboRegistryProvider>\n      </TamboClientContext.Provider>\n    );\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"]}