@tambo-ai/react 0.65.2 → 0.65.3

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

@@ -1,14 +1,21 @@
 "use client";
-import React, { createContext, useContext, useMemo, } from "react";
+import React, { createContext, useContext, useEffect, useMemo, useRef, useState, } from "react";
 import { TamboClientContext } from "./tambo-client-provider";
+import { PLACEHOLDER_THREAD, TamboThreadContext, } from "./tambo-thread-provider";
 const TamboMcpTokenContext = createContext(undefined);
 /**
  * Provider for managing the MCP access token that is returned by the Tambo API.
  * This token is used to authenticate with the internal Tambo MCP server.
- * The base URL is derived from the TamboClient's baseURL property.
  *
- * **NOTE**: This provider now delegates token state management to TamboClientProvider.
- * It exists primarily for backward compatibility and to add the tamboBaseUrl.
+ * Token selection logic:
+ * - Returns the current thread's mcpAccessToken if available
+ * - Returns null if no thread or thread has no token
+ *
+ * Token fetching:
+ * - Thread-specific token: always fetched when switching to thread without token
+ * - Tokens are fetched regardless of client-side MCP server configuration
+ *   to support server-side MCP servers that are not visible to the provider
+ * - Threadless tokens must be fetched by individual hooks/callbacks using contextKey
  * @internal
  * @param props - The provider props
  * @param props.children - The children to wrap
@@ -19,21 +26,126 @@ export const TamboMcpTokenProvider = ({ children, }) => {
     if (!clientContext) {
         throw new Error("TamboMcpTokenProvider must be used within a TamboClientProvider");
     }
-    const { client, mcpAccessToken, setMcpAccessToken } = clientContext;
+    const { client } = clientContext;
     const tamboBaseUrl = client.baseURL;
-    const value = useMemo(() => ({ mcpAccessToken, tamboBaseUrl, setMcpAccessToken }), [mcpAccessToken, tamboBaseUrl, setMcpAccessToken]);
+    // Optional thread context - may not be available in all contexts
+    const threadContext = useContext(TamboThreadContext);
+    const currentThread = threadContext?.currentThread ?? null;
+    const currentThreadId = threadContext?.currentThreadId ?? null;
+    const setThreadMap = useMemo(() => threadContext?.setThreadMap ?? (() => { }), [threadContext?.setThreadMap]);
+    // Track previous thread ID to detect thread switches
+    const previousThreadId = useRef(null);
+    // Return the current thread's token or null
+    const selectedToken = useMemo(() => {
+        return currentThread?.mcpAccessToken ?? null;
+    }, [currentThread]);
+    // Fetch thread-specific token when switching to a thread without one
+    // Always fetch to support server-side MCP servers (not just client-side)
+    // Skip PLACEHOLDER_THREAD - never send it to the server
+    useEffect(() => {
+        const hasThreadChanged = previousThreadId.current !== currentThreadId;
+        previousThreadId.current = currentThreadId;
+        const isPlaceholderThread = currentThreadId === PLACEHOLDER_THREAD.id;
+        const shouldFetchThreadToken = hasThreadChanged &&
+            currentThreadId &&
+            !isPlaceholderThread &&
+            currentThread &&
+            !currentThread.mcpAccessToken;
+        if (shouldFetchThreadToken) {
+            const fetchThreadToken = async () => {
+                try {
+                    const response = await client.beta.auth.getMcpToken({
+                        threadId: currentThreadId,
+                    });
+                    if (response.mcpAccessToken) {
+                        // Update thread in threadMap with new token
+                        setThreadMap((prev) => {
+                            const thread = prev[currentThreadId];
+                            if (thread) {
+                                return {
+                                    ...prev,
+                                    [currentThreadId]: {
+                                        ...thread,
+                                        mcpAccessToken: response.mcpAccessToken,
+                                    },
+                                };
+                            }
+                            return prev;
+                        });
+                    }
+                }
+                catch (error) {
+                    console.error(`Failed to fetch MCP token for thread ${currentThreadId}:`, error);
+                }
+            };
+            void fetchThreadToken();
+        }
+    }, [currentThreadId, currentThread, client, setThreadMap]);
+    const value = useMemo(() => ({ mcpAccessToken: selectedToken, tamboBaseUrl }), [selectedToken, tamboBaseUrl]);
     return (React.createElement(TamboMcpTokenContext.Provider, { value: value }, children));
 };
 /**
- * Hook to access the current MCP access token.
- * @internal
- * @returns The current MCP access token and a setter function
+ * Hook to access the current MCP access token with optional threadless token fetching.
+ *
+ * Token selection logic:
+ * 1. If current thread has mcpAccessToken → use thread-specific token
+ * 2. Else if contextKey provided and no thread (or placeholder thread) → fetch and use threadless token
+ * 3. Else → null
+ * @param contextKey - Optional context key for fetching threadless tokens when not in a thread
+ * @returns The current MCP access token and base URL
  */
-export const useTamboMcpToken = () => {
+export const useTamboMcpToken = (contextKey) => {
     const context = useContext(TamboMcpTokenContext);
     if (context === undefined) {
         throw new Error("useTamboMcpToken must be used within a TamboMcpTokenProvider");
     }
-    return context;
+    const clientContext = useContext(TamboClientContext);
+    if (!clientContext) {
+        throw new Error("useTamboMcpToken must be used within a TamboClientProvider");
+    }
+    const { client } = clientContext;
+    const threadContext = useContext(TamboThreadContext);
+    const currentThreadId = threadContext?.currentThreadId ?? null;
+    // State for threadless token
+    const [threadlessToken, setThreadlessToken] = useState(null);
+    const hasAttemptedFetch = useRef(false);
+    // Determine if we should fetch a threadless token
+    const isPlaceholderThread = currentThreadId === PLACEHOLDER_THREAD.id;
+    const shouldFetchThreadless = contextKey &&
+        (!currentThreadId || isPlaceholderThread) &&
+        !context.mcpAccessToken &&
+        !threadlessToken &&
+        !hasAttemptedFetch.current;
+    // Fetch threadless token when needed
+    useEffect(() => {
+        if (shouldFetchThreadless) {
+            hasAttemptedFetch.current = true;
+            const fetchThreadlessToken = async () => {
+                try {
+                    const response = await client.beta.auth.getMcpToken({ contextKey });
+                    if (response.mcpAccessToken) {
+                        setThreadlessToken(response.mcpAccessToken);
+                    }
+                }
+                catch (error) {
+                    console.error("Failed to fetch threadless MCP token:", error);
+                }
+            };
+            void fetchThreadlessToken();
+        }
+    }, [shouldFetchThreadless, client, contextKey]);
+    // Reset threadless token when switching to a real thread
+    useEffect(() => {
+        if (currentThreadId && currentThreadId !== PLACEHOLDER_THREAD.id) {
+            setThreadlessToken(null);
+            hasAttemptedFetch.current = false;
+        }
+    }, [currentThreadId]);
+    // Return thread token if available, otherwise threadless token
+    const selectedToken = context.mcpAccessToken ?? threadlessToken;
+    return {
+        mcpAccessToken: selectedToken,
+        tamboBaseUrl: context.tamboBaseUrl,
+    };
 };
 //# sourceMappingURL=tambo-mcp-token-provider.js.map

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

@@ -1 +1 @@
-{"version":3,"file":"tambo-mcp-token-provider.js","sourceRoot":"","sources":["../../src/providers/tambo-mcp-token-provider.tsx"],"names":[],"mappings":"AAAA,YAAY,CAAC;AACb,OAAO,KAAK,EAAE,EACZ,aAAa,EAEb,UAAU,EACV,OAAO,GACR,MAAM,OAAO,CAAC;AACf,OAAO,EAAE,kBAAkB,EAAE,MAAM,yBAAyB,CAAC;AAmB7D,MAAM,oBAAoB,GAAG,aAAa,CAExC,SAAS,CAAC,CAAC;AAEb;;;;;;;;;;;GAWG;AACH,MAAM,CAAC,MAAM,qBAAqB,GAAgC,CAAC,EACjE,QAAQ,GACT,EAAE,EAAE;IACH,MAAM,aAAa,GAAG,UAAU,CAAC,kBAAkB,CAAC,CAAC;IACrD,IAAI,CAAC,aAAa,EAAE,CAAC;QACnB,MAAM,IAAI,KAAK,CACb,iEAAiE,CAClE,CAAC;IACJ,CAAC;IACD,MAAM,EAAE,MAAM,EAAE,cAAc,EAAE,iBAAiB,EAAE,GAAG,aAAa,CAAC;IACpE,MAAM,YAAY,GAAG,MAAM,CAAC,OAAO,CAAC;IAEpC,MAAM,KAAK,GAAG,OAAO,CACnB,GAAG,EAAE,CAAC,CAAC,EAAE,cAAc,EAAE,YAAY,EAAE,iBAAiB,EAAE,CAAC,EAC3D,CAAC,cAAc,EAAE,YAAY,EAAE,iBAAiB,CAAC,CAClD,CAAC;IAEF,OAAO,CACL,oBAAC,oBAAoB,CAAC,QAAQ,IAAC,KAAK,EAAE,KAAK,IACxC,QAAQ,CACqB,CACjC,CAAC;AACJ,CAAC,CAAC;AAEF;;;;GAIG;AACH,MAAM,CAAC,MAAM,gBAAgB,GAAG,GAA8B,EAAE;IAC9D,MAAM,OAAO,GAAG,UAAU,CAAC,oBAAoB,CAAC,CAAC;IACjD,IAAI,OAAO,KAAK,SAAS,EAAE,CAAC;QAC1B,MAAM,IAAI,KAAK,CACb,8DAA8D,CAC/D,CAAC;IACJ,CAAC;IACD,OAAO,OAAO,CAAC;AACjB,CAAC,CAAC","sourcesContent":["\"use client\";\nimport React, {\n  createContext,\n  PropsWithChildren,\n  useContext,\n  useMemo,\n} from \"react\";\nimport { TamboClientContext } from \"./tambo-client-provider\";\n\nexport interface TamboMcpTokenContextProps {\n  /**\n   * The current MCP access token for the internal Tambo MCP server\n   */\n  mcpAccessToken: string | null;\n  /**\n   * The base URL for the Tambo API (used to construct the MCP server URL)\n   * Returns undefined if the client is not yet initialized\n   */\n  tamboBaseUrl: string | undefined;\n  /**\n   * Update the MCP access token (for internal use by TamboThreadProvider)\n   * @internal\n   */\n  setMcpAccessToken: (token: string | null) => void;\n}\n\nconst TamboMcpTokenContext = createContext<\n  TamboMcpTokenContextProps | undefined\n>(undefined);\n\n/**\n * Provider for managing the MCP access token that is returned by the Tambo API.\n * This token is used to authenticate with the internal Tambo MCP server.\n * The base URL is derived from the TamboClient's baseURL property.\n *\n * **NOTE**: This provider now delegates token state management to TamboClientProvider.\n * It exists primarily for backward compatibility and to add the tamboBaseUrl.\n * @internal\n * @param props - The provider props\n * @param props.children - The children to wrap\n * @returns The TamboMcpTokenProvider component\n */\nexport const TamboMcpTokenProvider: React.FC<PropsWithChildren> = ({\n  children,\n}) => {\n  const clientContext = useContext(TamboClientContext);\n  if (!clientContext) {\n    throw new Error(\n      \"TamboMcpTokenProvider must be used within a TamboClientProvider\",\n    );\n  }\n  const { client, mcpAccessToken, setMcpAccessToken } = clientContext;\n  const tamboBaseUrl = client.baseURL;\n\n  const value = useMemo(\n    () => ({ mcpAccessToken, tamboBaseUrl, setMcpAccessToken }),\n    [mcpAccessToken, tamboBaseUrl, setMcpAccessToken],\n  );\n\n  return (\n    <TamboMcpTokenContext.Provider value={value}>\n      {children}\n    </TamboMcpTokenContext.Provider>\n  );\n};\n\n/**\n * Hook to access the current MCP access token.\n * @internal\n * @returns The current MCP access token and a setter function\n */\nexport const useTamboMcpToken = (): TamboMcpTokenContextProps => {\n  const context = useContext(TamboMcpTokenContext);\n  if (context === undefined) {\n    throw new Error(\n      \"useTamboMcpToken must be used within a TamboMcpTokenProvider\",\n    );\n  }\n  return context;\n};\n"]}
+{"version":3,"file":"tambo-mcp-token-provider.js","sourceRoot":"","sources":["../../src/providers/tambo-mcp-token-provider.tsx"],"names":[],"mappings":"AAAA,YAAY,CAAC;AACb,OAAO,KAAK,EAAE,EACZ,aAAa,EAEb,UAAU,EACV,SAAS,EACT,OAAO,EACP,MAAM,EACN,QAAQ,GACT,MAAM,OAAO,CAAC;AAEf,OAAO,EAAE,kBAAkB,EAAE,MAAM,yBAAyB,CAAC;AAC7D,OAAO,EACL,kBAAkB,EAClB,kBAAkB,GACnB,MAAM,yBAAyB,CAAC;AAejC,MAAM,oBAAoB,GAAG,aAAa,CAExC,SAAS,CAAC,CAAC;AAEb;;;;;;;;;;;;;;;;;GAiBG;AACH,MAAM,CAAC,MAAM,qBAAqB,GAAgC,CAAC,EACjE,QAAQ,GACT,EAAE,EAAE;IACH,MAAM,aAAa,GAAG,UAAU,CAAC,kBAAkB,CAAC,CAAC;IACrD,IAAI,CAAC,aAAa,EAAE,CAAC;QACnB,MAAM,IAAI,KAAK,CACb,iEAAiE,CAClE,CAAC;IACJ,CAAC;IACD,MAAM,EAAE,MAAM,EAAE,GAAG,aAAa,CAAC;IACjC,MAAM,YAAY,GAAG,MAAM,CAAC,OAAO,CAAC;IAEpC,iEAAiE;IACjE,MAAM,aAAa,GAAG,UAAU,CAAC,kBAAkB,CAAC,CAAC;IACrD,MAAM,aAAa,GAAG,aAAa,EAAE,aAAa,IAAI,IAAI,CAAC;IAC3D,MAAM,eAAe,GAAG,aAAa,EAAE,eAAe,IAAI,IAAI,CAAC;IAC/D,MAAM,YAAY,GAAG,OAAO,CAC1B,GAAG,EAAE,CAAC,aAAa,EAAE,YAAY,IAAI,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,EAC/C,CAAC,aAAa,EAAE,YAAY,CAAC,CAC9B,CAAC;IAEF,qDAAqD;IACrD,MAAM,gBAAgB,GAAG,MAAM,CAAgB,IAAI,CAAC,CAAC;IAErD,4CAA4C;IAC5C,MAAM,aAAa,GAAG,OAAO,CAAC,GAAG,EAAE;QACjC,OAAO,aAAa,EAAE,cAAc,IAAI,IAAI,CAAC;IAC/C,CAAC,EAAE,CAAC,aAAa,CAAC,CAAC,CAAC;IAEpB,qEAAqE;IACrE,yEAAyE;IACzE,wDAAwD;IACxD,SAAS,CAAC,GAAG,EAAE;QACb,MAAM,gBAAgB,GAAG,gBAAgB,CAAC,OAAO,KAAK,eAAe,CAAC;QACtE,gBAAgB,CAAC,OAAO,GAAG,eAAe,CAAC;QAE3C,MAAM,mBAAmB,GAAG,eAAe,KAAK,kBAAkB,CAAC,EAAE,CAAC;QACtE,MAAM,sBAAsB,GAC1B,gBAAgB;YAChB,eAAe;YACf,CAAC,mBAAmB;YACpB,aAAa;YACb,CAAC,aAAa,CAAC,cAAc,CAAC;QAEhC,IAAI,sBAAsB,EAAE,CAAC;YAC3B,MAAM,gBAAgB,GAAG,KAAK,IAAI,EAAE;gBAClC,IAAI,CAAC;oBACH,MAAM,QAAQ,GAAG,MAAM,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,WAAW,CAAC;wBAClD,QAAQ,EAAE,eAAe;qBAC1B,CAAC,CAAC;oBACH,IAAI,QAAQ,CAAC,cAAc,EAAE,CAAC;wBAC5B,4CAA4C;wBAC5C,YAAY,CAAC,CAAC,IAAiC,EAAE,EAAE;4BACjD,MAAM,MAAM,GAAG,IAAI,CAAC,eAAe,CAAC,CAAC;4BACrC,IAAI,MAAM,EAAE,CAAC;gCACX,OAAO;oCACL,GAAG,IAAI;oCACP,CAAC,eAAe,CAAC,EAAE;wCACjB,GAAG,MAAM;wCACT,cAAc,EAAE,QAAQ,CAAC,cAAc;qCACxC;iCACF,CAAC;4BACJ,CAAC;4BACD,OAAO,IAAI,CAAC;wBACd,CAAC,CAAC,CAAC;oBACL,CAAC;gBACH,CAAC;gBAAC,OAAO,KAAK,EAAE,CAAC;oBACf,OAAO,CAAC,KAAK,CACX,wCAAwC,eAAe,GAAG,EAC1D,KAAK,CACN,CAAC;gBACJ,CAAC;YACH,CAAC,CAAC;YACF,KAAK,gBAAgB,EAAE,CAAC;QAC1B,CAAC;IACH,CAAC,EAAE,CAAC,eAAe,EAAE,aAAa,EAAE,MAAM,EAAE,YAAY,CAAC,CAAC,CAAC;IAE3D,MAAM,KAAK,GAAG,OAAO,CACnB,GAAG,EAAE,CAAC,CAAC,EAAE,cAAc,EAAE,aAAa,EAAE,YAAY,EAAE,CAAC,EACvD,CAAC,aAAa,EAAE,YAAY,CAAC,CAC9B,CAAC;IAEF,OAAO,CACL,oBAAC,oBAAoB,CAAC,QAAQ,IAAC,KAAK,EAAE,KAAK,IACxC,QAAQ,CACqB,CACjC,CAAC;AACJ,CAAC,CAAC;AAEF;;;;;;;;;GASG;AACH,MAAM,CAAC,MAAM,gBAAgB,GAAG,CAC9B,UAAmB,EACQ,EAAE;IAC7B,MAAM,OAAO,GAAG,UAAU,CAAC,oBAAoB,CAAC,CAAC;IACjD,IAAI,OAAO,KAAK,SAAS,EAAE,CAAC;QAC1B,MAAM,IAAI,KAAK,CACb,8DAA8D,CAC/D,CAAC;IACJ,CAAC;IAED,MAAM,aAAa,GAAG,UAAU,CAAC,kBAAkB,CAAC,CAAC;IACrD,IAAI,CAAC,aAAa,EAAE,CAAC;QACnB,MAAM,IAAI,KAAK,CACb,4DAA4D,CAC7D,CAAC;IACJ,CAAC;IACD,MAAM,EAAE,MAAM,EAAE,GAAG,aAAa,CAAC;IAEjC,MAAM,aAAa,GAAG,UAAU,CAAC,kBAAkB,CAAC,CAAC;IACrD,MAAM,eAAe,GAAG,aAAa,EAAE,eAAe,IAAI,IAAI,CAAC;IAE/D,6BAA6B;IAC7B,MAAM,CAAC,eAAe,EAAE,kBAAkB,CAAC,GAAG,QAAQ,CAAgB,IAAI,CAAC,CAAC;IAC5E,MAAM,iBAAiB,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC;IAExC,kDAAkD;IAClD,MAAM,mBAAmB,GAAG,eAAe,KAAK,kBAAkB,CAAC,EAAE,CAAC;IACtE,MAAM,qBAAqB,GACzB,UAAU;QACV,CAAC,CAAC,eAAe,IAAI,mBAAmB,CAAC;QACzC,CAAC,OAAO,CAAC,cAAc;QACvB,CAAC,eAAe;QAChB,CAAC,iBAAiB,CAAC,OAAO,CAAC;IAE7B,qCAAqC;IACrC,SAAS,CAAC,GAAG,EAAE;QACb,IAAI,qBAAqB,EAAE,CAAC;YAC1B,iBAAiB,CAAC,OAAO,GAAG,IAAI,CAAC;YACjC,MAAM,oBAAoB,GAAG,KAAK,IAAI,EAAE;gBACtC,IAAI,CAAC;oBACH,MAAM,QAAQ,GAAG,MAAM,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,WAAW,CAAC,EAAE,UAAU,EAAE,CAAC,CAAC;oBACpE,IAAI,QAAQ,CAAC,cAAc,EAAE,CAAC;wBAC5B,kBAAkB,CAAC,QAAQ,CAAC,cAAc,CAAC,CAAC;oBAC9C,CAAC;gBACH,CAAC;gBAAC,OAAO,KAAK,EAAE,CAAC;oBACf,OAAO,CAAC,KAAK,CAAC,uCAAuC,EAAE,KAAK,CAAC,CAAC;gBAChE,CAAC;YACH,CAAC,CAAC;YACF,KAAK,oBAAoB,EAAE,CAAC;QAC9B,CAAC;IACH,CAAC,EAAE,CAAC,qBAAqB,EAAE,MAAM,EAAE,UAAU,CAAC,CAAC,CAAC;IAEhD,yDAAyD;IACzD,SAAS,CAAC,GAAG,EAAE;QACb,IAAI,eAAe,IAAI,eAAe,KAAK,kBAAkB,CAAC,EAAE,EAAE,CAAC;YACjE,kBAAkB,CAAC,IAAI,CAAC,CAAC;YACzB,iBAAiB,CAAC,OAAO,GAAG,KAAK,CAAC;QACpC,CAAC;IACH,CAAC,EAAE,CAAC,eAAe,CAAC,CAAC,CAAC;IAEtB,+DAA+D;IAC/D,MAAM,aAAa,GAAG,OAAO,CAAC,cAAc,IAAI,eAAe,CAAC;IAEhE,OAAO;QACL,cAAc,EAAE,aAAa;QAC7B,YAAY,EAAE,OAAO,CAAC,YAAY;KACnC,CAAC;AACJ,CAAC,CAAC","sourcesContent":["\"use client\";\nimport React, {\n  createContext,\n  PropsWithChildren,\n  useContext,\n  useEffect,\n  useMemo,\n  useRef,\n  useState,\n} from \"react\";\nimport { TamboThread } from \"../model/tambo-thread\";\nimport { TamboClientContext } from \"./tambo-client-provider\";\nimport {\n  PLACEHOLDER_THREAD,\n  TamboThreadContext,\n} from \"./tambo-thread-provider\";\n\nexport interface TamboMcpTokenContextProps {\n  /**\n   * The current MCP access token for the internal Tambo MCP server.\n   * Prefers thread-specific token over threadless token.\n   */\n  mcpAccessToken: string | null;\n  /**\n   * The base URL for the Tambo API (used to construct the MCP server URL)\n   * Returns undefined if the client is not yet initialized\n   */\n  tamboBaseUrl: string | undefined;\n}\n\nconst TamboMcpTokenContext = createContext<\n  TamboMcpTokenContextProps | undefined\n>(undefined);\n\n/**\n * Provider for managing the MCP access token that is returned by the Tambo API.\n * This token is used to authenticate with the internal Tambo MCP server.\n *\n * Token selection logic:\n * - Returns the current thread's mcpAccessToken if available\n * - Returns null if no thread or thread has no token\n *\n * Token fetching:\n * - Thread-specific token: always fetched when switching to thread without token\n * - Tokens are fetched regardless of client-side MCP server configuration\n *   to support server-side MCP servers that are not visible to the provider\n * - Threadless tokens must be fetched by individual hooks/callbacks using contextKey\n * @internal\n * @param props - The provider props\n * @param props.children - The children to wrap\n * @returns The TamboMcpTokenProvider component\n */\nexport const TamboMcpTokenProvider: React.FC<PropsWithChildren> = ({\n  children,\n}) => {\n  const clientContext = useContext(TamboClientContext);\n  if (!clientContext) {\n    throw new Error(\n      \"TamboMcpTokenProvider must be used within a TamboClientProvider\",\n    );\n  }\n  const { client } = clientContext;\n  const tamboBaseUrl = client.baseURL;\n\n  // Optional thread context - may not be available in all contexts\n  const threadContext = useContext(TamboThreadContext);\n  const currentThread = threadContext?.currentThread ?? null;\n  const currentThreadId = threadContext?.currentThreadId ?? null;\n  const setThreadMap = useMemo(\n    () => threadContext?.setThreadMap ?? (() => {}),\n    [threadContext?.setThreadMap],\n  );\n\n  // Track previous thread ID to detect thread switches\n  const previousThreadId = useRef<string | null>(null);\n\n  // Return the current thread's token or null\n  const selectedToken = useMemo(() => {\n    return currentThread?.mcpAccessToken ?? null;\n  }, [currentThread]);\n\n  // Fetch thread-specific token when switching to a thread without one\n  // Always fetch to support server-side MCP servers (not just client-side)\n  // Skip PLACEHOLDER_THREAD - never send it to the server\n  useEffect(() => {\n    const hasThreadChanged = previousThreadId.current !== currentThreadId;\n    previousThreadId.current = currentThreadId;\n\n    const isPlaceholderThread = currentThreadId === PLACEHOLDER_THREAD.id;\n    const shouldFetchThreadToken =\n      hasThreadChanged &&\n      currentThreadId &&\n      !isPlaceholderThread &&\n      currentThread &&\n      !currentThread.mcpAccessToken;\n\n    if (shouldFetchThreadToken) {\n      const fetchThreadToken = async () => {\n        try {\n          const response = await client.beta.auth.getMcpToken({\n            threadId: currentThreadId,\n          });\n          if (response.mcpAccessToken) {\n            // Update thread in threadMap with new token\n            setThreadMap((prev: Record<string, TamboThread>) => {\n              const thread = prev[currentThreadId];\n              if (thread) {\n                return {\n                  ...prev,\n                  [currentThreadId]: {\n                    ...thread,\n                    mcpAccessToken: response.mcpAccessToken,\n                  },\n                };\n              }\n              return prev;\n            });\n          }\n        } catch (error) {\n          console.error(\n            `Failed to fetch MCP token for thread ${currentThreadId}:`,\n            error,\n          );\n        }\n      };\n      void fetchThreadToken();\n    }\n  }, [currentThreadId, currentThread, client, setThreadMap]);\n\n  const value = useMemo(\n    () => ({ mcpAccessToken: selectedToken, tamboBaseUrl }),\n    [selectedToken, tamboBaseUrl],\n  );\n\n  return (\n    <TamboMcpTokenContext.Provider value={value}>\n      {children}\n    </TamboMcpTokenContext.Provider>\n  );\n};\n\n/**\n * Hook to access the current MCP access token with optional threadless token fetching.\n *\n * Token selection logic:\n * 1. If current thread has mcpAccessToken → use thread-specific token\n * 2. Else if contextKey provided and no thread (or placeholder thread) → fetch and use threadless token\n * 3. Else → null\n * @param contextKey - Optional context key for fetching threadless tokens when not in a thread\n * @returns The current MCP access token and base URL\n */\nexport const useTamboMcpToken = (\n  contextKey?: string,\n): TamboMcpTokenContextProps => {\n  const context = useContext(TamboMcpTokenContext);\n  if (context === undefined) {\n    throw new Error(\n      \"useTamboMcpToken must be used within a TamboMcpTokenProvider\",\n    );\n  }\n\n  const clientContext = useContext(TamboClientContext);\n  if (!clientContext) {\n    throw new Error(\n      \"useTamboMcpToken must be used within a TamboClientProvider\",\n    );\n  }\n  const { client } = clientContext;\n\n  const threadContext = useContext(TamboThreadContext);\n  const currentThreadId = threadContext?.currentThreadId ?? null;\n\n  // State for threadless token\n  const [threadlessToken, setThreadlessToken] = useState<string | null>(null);\n  const hasAttemptedFetch = useRef(false);\n\n  // Determine if we should fetch a threadless token\n  const isPlaceholderThread = currentThreadId === PLACEHOLDER_THREAD.id;\n  const shouldFetchThreadless =\n    contextKey &&\n    (!currentThreadId || isPlaceholderThread) &&\n    !context.mcpAccessToken &&\n    !threadlessToken &&\n    !hasAttemptedFetch.current;\n\n  // Fetch threadless token when needed\n  useEffect(() => {\n    if (shouldFetchThreadless) {\n      hasAttemptedFetch.current = true;\n      const fetchThreadlessToken = async () => {\n        try {\n          const response = await client.beta.auth.getMcpToken({ contextKey });\n          if (response.mcpAccessToken) {\n            setThreadlessToken(response.mcpAccessToken);\n          }\n        } catch (error) {\n          console.error(\"Failed to fetch threadless MCP token:\", error);\n        }\n      };\n      void fetchThreadlessToken();\n    }\n  }, [shouldFetchThreadless, client, contextKey]);\n\n  // Reset threadless token when switching to a real thread\n  useEffect(() => {\n    if (currentThreadId && currentThreadId !== PLACEHOLDER_THREAD.id) {\n      setThreadlessToken(null);\n      hasAttemptedFetch.current = false;\n    }\n  }, [currentThreadId]);\n\n  // Return thread token if available, otherwise threadless token\n  const selectedToken = context.mcpAccessToken ?? threadlessToken;\n\n  return {\n    mcpAccessToken: selectedToken,\n    tamboBaseUrl: context.tamboBaseUrl,\n  };\n};\n"]}

package/esm/providers/tambo-prop-stream-provider.test.d.ts.map

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

package/esm/providers/{__tests__/tambo-prop-stream-provider.test.js → tambo-prop-stream-provider.test.js}

@@ -1,15 +1,15 @@
 import { render, screen } from "@testing-library/react";
 import React from "react";
-import { useTamboCurrentMessage } from "../../hooks/use-current-message";
-import { GenerationStage, } from "../../model/generate-component-response";
-import { useTamboGenerationStage, useTamboThread, } from "../../providers/tambo-thread-provider";
-import { TamboPropStreamProvider, useTamboStream, } from "../tambo-prop-stream-provider";
+import { useTamboCurrentMessage } from "../hooks/use-current-message";
+import { GenerationStage, } from "../model/generate-component-response";
+import { useTamboGenerationStage, useTamboThread, } from "../providers/tambo-thread-provider";
+import { TamboPropStreamProvider, useTamboStream, } from "./tambo-prop-stream-provider";
 // Mock the required providers
-jest.mock("../../providers/tambo-thread-provider", () => ({
+jest.mock("../providers/tambo-thread-provider", () => ({
     useTamboThread: jest.fn(),
     useTamboGenerationStage: jest.fn(),
 }));
-jest.mock("../../hooks/use-current-message", () => ({
+jest.mock("../hooks/use-current-message", () => ({
     useTamboCurrentMessage: jest.fn(),
 }));
 // Mock window for SSR tests
@@ -33,31 +33,38 @@ const createMockMessage = (overrides = {}) => ({
     ...overrides,
 });
 // Helper function to create mock CombinedTamboThreadContextProps
-const createMockThreadContext = (overrides = {}) => ({
-    // TamboThreadContextProps properties
-    thread: {
+const createMockThreadContext = (overrides = {}) => {
+    const mockThread = {
         id: "test-thread",
         projectId: "test-project",
         messages: [],
         name: "Test Thread",
         createdAt: new Date().toISOString(),
         updatedAt: new Date().toISOString(),
-    },
-    switchCurrentThread: jest.fn(),
-    startNewThread: jest.fn(),
-    updateThreadName: jest.fn(),
-    generateThreadName: jest.fn(),
-    addThreadMessage: jest.fn(),
-    updateThreadMessage: jest.fn(),
-    cancel: jest.fn(),
-    streaming: false,
-    sendThreadMessage: jest.fn(),
-    // GenerationStageContextProps properties
-    generationStage: GenerationStage.IDLE,
-    generationStatusMessage: "",
-    isIdle: true,
-    ...overrides,
-});
+    };
+    return {
+        // TamboThreadContextProps properties
+        thread: mockThread,
+        currentThreadId: mockThread.id,
+        currentThread: mockThread,
+        threadMap: { [mockThread.id]: mockThread },
+        setThreadMap: jest.fn(),
+        switchCurrentThread: jest.fn(),
+        startNewThread: jest.fn(),
+        updateThreadName: jest.fn(),
+        generateThreadName: jest.fn(),
+        addThreadMessage: jest.fn(),
+        updateThreadMessage: jest.fn(),
+        cancel: jest.fn(),
+        streaming: false,
+        sendThreadMessage: jest.fn(),
+        // GenerationStageContextProps properties
+        generationStage: GenerationStage.IDLE,
+        generationStatusMessage: "",
+        isIdle: true,
+        ...overrides,
+    };
+};
 // Helper component to test hook usage
 const TestHookComponent = ({ testKey = "default", }) => {
     const { streamStatus, getStatusForKey } = useTamboStream();

package/esm/providers/tambo-prop-stream-provider.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"tambo-prop-stream-provider.test.js","sourceRoot":"","sources":["../../src/providers/tambo-prop-stream-provider.test.tsx"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,wBAAwB,CAAC;AACxD,OAAO,KAAK,MAAM,OAAO,CAAC;AAC1B,OAAO,EAAE,sBAAsB,EAAE,MAAM,8BAA8B,CAAC;AACtE,OAAO,EACL,eAAe,GAEhB,MAAM,sCAAsC,CAAC;AAC9C,OAAO,EAEL,uBAAuB,EACvB,cAAc,GACf,MAAM,oCAAoC,CAAC;AAC5C,OAAO,EACL,uBAAuB,EACvB,cAAc,GACf,MAAM,8BAA8B,CAAC;AAEtC,8BAA8B;AAC9B,IAAI,CAAC,IAAI,CAAC,oCAAoC,EAAE,GAAG,EAAE,CAAC,CAAC;IACrD,cAAc,EAAE,IAAI,CAAC,EAAE,EAAE;IACzB,uBAAuB,EAAE,IAAI,CAAC,EAAE,EAAE;CACnC,CAAC,CAAC,CAAC;AAEJ,IAAI,CAAC,IAAI,CAAC,8BAA8B,EAAE,GAAG,EAAE,CAAC,CAAC;IAC/C,sBAAsB,EAAE,IAAI,CAAC,EAAE,EAAE;CAClC,CAAC,CAAC,CAAC;AAEJ,4BAA4B;AAC5B,MAAM,cAAc,GAAG,MAAM,CAAC,MAAM,CAAC;AAErC,qDAAqD;AACrD,MAAM,mBAAmB,GAAG,CAAC,QAAiC,EAAE,EAAO,EAAE,CAAC,CAAC;IACzE,aAAa,EAAE,eAAe;IAC9B,cAAc,EAAE,EAAE;IAClB,OAAO,EAAE,qBAAqB;IAC9B,KAAK;IACL,SAAS,EAAE,gBAAgB;CAC5B,CAAC,CAAC;AAEH,oDAAoD;AACpD,MAAM,iBAAiB,GAAG,CACxB,YAAyC,EAAE,EACvB,EAAE,CAAC,CAAC;IACxB,EAAE,EAAE,cAAc;IAClB,cAAc,EAAE,EAAE;IAClB,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,cAAc,EAAE,CAAC;IACjD,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;IACnC,IAAI,EAAE,WAAW;IACjB,QAAQ,EAAE,aAAa;IACvB,GAAG,SAAS;CACb,CAAC,CAAC;AAEH,iEAAiE;AACjE,MAAM,uBAAuB,GAAG,CAC9B,YAAsD,EAAE,EACvB,EAAE;IACnC,MAAM,UAAU,GAAG;QACjB,EAAE,EAAE,aAAa;QACjB,SAAS,EAAE,cAAc;QACzB,QAAQ,EAAE,EAAE;QACZ,IAAI,EAAE,aAAa;QACnB,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;QACnC,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;KACpC,CAAC;IACF,OAAO;QACL,qCAAqC;QACrC,MAAM,EAAE,UAAU;QAClB,eAAe,EAAE,UAAU,CAAC,EAAE;QAC9B,aAAa,EAAE,UAAU;QACzB,SAAS,EAAE,EAAE,CAAC,UAAU,CAAC,EAAE,CAAC,EAAE,UAAU,EAAE;QAC1C,YAAY,EAAE,IAAI,CAAC,EAAE,EAAE;QACvB,mBAAmB,EAAE,IAAI,CAAC,EAAE,EAAE;QAC9B,cAAc,EAAE,IAAI,CAAC,EAAE,EAAE;QACzB,gBAAgB,EAAE,IAAI,CAAC,EAAE,EAAE;QAC3B,kBAAkB,EAAE,IAAI,CAAC,EAAE,EAAE;QAC7B,gBAAgB,EAAE,IAAI,CAAC,EAAE,EAAE;QAC3B,mBAAmB,EAAE,IAAI,CAAC,EAAE,EAAE;QAC9B,MAAM,EAAE,IAAI,CAAC,EAAE,EAAE;QACjB,SAAS,EAAE,KAAK;QAChB,iBAAiB,EAAE,IAAI,CAAC,EAAE,EAAE;QAC5B,yCAAyC;QACzC,eAAe,EAAE,eAAe,CAAC,IAAI;QACrC,uBAAuB,EAAE,EAAE;QAC3B,MAAM,EAAE,IAAI;QACZ,GAAG,SAAS;KACb,CAAC;AACJ,CAAC,CAAC;AAEF,sCAAsC;AACtC,MAAM,iBAAiB,GAAmC,CAAC,EACzD,OAAO,GAAG,SAAS,GACpB,EAAE,EAAE;IACH,MAAM,EAAE,YAAY,EAAE,eAAe,EAAE,GAAG,cAAc,EAAE,CAAC;IAC3D,MAAM,MAAM,GAAG,eAAe,CAAC,OAAO,CAAC,CAAC;IAExC,OAAO,CACL;QACE,4CAAiB,eAAe,IAAE,IAAI,CAAC,SAAS,CAAC,YAAY,CAAC,CAAO;QACrE,4CAAiB,YAAY,IAAE,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,CAAO,CACxD,CACP,CAAC;AACJ,CAAC,CAAC;AAEF,QAAQ,CAAC,yBAAyB,EAAE,GAAG,EAAE;IACvC,UAAU,CAAC,GAAG,EAAE;QACd,uCAAuC;QACvC,MAAM,CAAC,MAAM,GAAG,cAAc,CAAC;QAE/B,+BAA+B;QAC/B,IAAI,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,eAAe,CAAC,uBAAuB,EAAE,CAAC,CAAC;QAEvE,IAAI,CAAC,MAAM,CAAC,uBAAuB,CAAC,CAAC,eAAe,CAAC;YACnD,eAAe,EAAE,eAAe,CAAC,IAAI;YACrC,uBAAuB,EAAE,EAAE;YAC3B,MAAM,EAAE,IAAI;SACb,CAAC,CAAC;QAEH,IAAI,CAAC,MAAM,CAAC,sBAAsB,CAAC,CAAC,eAAe,CAAC;YAClD,EAAE,EAAE,cAAc;YAClB,SAAS,EAAE;gBACT,KAAK,EAAE,EAAE;aACV;SACoB,CAAC,CAAC;IAC3B,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,IAAI,CAAC,aAAa,EAAE,CAAC;IACvB,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,qBAAqB,EAAE,GAAG,EAAE;QACnC,EAAE,CAAC,iEAAiE,EAAE,GAAG,EAAE;YACzE,uCAAuC;YACvC,MAAM,aAAa,GAAG,OAAO,CAAC,KAAK,CAAC;YACpC,OAAO,CAAC,KAAK,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC;YAE1B,MAAM,CAAC,GAAG,EAAE;gBACV,MAAM,CAAC,oBAAC,iBAAiB,OAAG,CAAC,CAAC;YAChC,CAAC,CAAC,CAAC,OAAO,CACR,8DAA8D,CAC/D,CAAC;YAEF,OAAO,CAAC,KAAK,GAAG,aAAa,CAAC;QAChC,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,qBAAqB,EAAE,GAAG,EAAE;QACnC,QAAQ,CAAC,qBAAqB,EAAE,GAAG,EAAE;YACnC,EAAE,CAAC,gDAAgD,EAAE,GAAG,EAAE;gBACxD,IAAI,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,eAAe,CACzC,uBAAuB,CAAC;oBACtB,eAAe,EAAE,eAAe,CAAC,IAAI;iBACtC,CAAC,CACH,CAAC;gBAEF,IAAI,CAAC,MAAM,CAAC,uBAAuB,CAAC,CAAC,eAAe,CAAC;oBACnD,eAAe,EAAE,eAAe,CAAC,IAAI;oBACrC,uBAAuB,EAAE,EAAE;oBAC3B,MAAM,EAAE,IAAI;iBACb,CAAC,CAAC;gBAEH,IAAI,CAAC,MAAM,CAAC,sBAAsB,CAAC,CAAC,eAAe,CACjD,iBAAiB,CAAC;oBAChB,SAAS,EAAE,mBAAmB,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC;iBAC9C,CAAC,CACH,CAAC;gBAEF,MAAM,CACJ,oBAAC,uBAAuB;oBACtB,oBAAC,uBAAuB,CAAC,SAAS;wBAChC,4CAAiB,SAAS,iBAAiB,CACT,CACZ,CAC3B,CAAC;gBAEF,MAAM,CAAC,MAAM,CAAC,WAAW,CAAC,SAAS,CAAC,CAAC,CAAC,iBAAiB,EAAE,CAAC;YAC5D,CAAC,CAAC,CAAC;YAEH,EAAE,CAAC,kDAAkD,EAAE,GAAG,EAAE;gBAC1D,IAAI,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,eAAe,CACzC,uBAAuB,CAAC;oBACtB,eAAe,EAAE,eAAe,CAAC,kBAAkB;iBACpD,CAAC,CACH,CAAC;gBAEF,IAAI,CAAC,MAAM,CAAC,uBAAuB,CAAC,CAAC,eAAe,CAAC;oBACnD,eAAe,EAAE,eAAe,CAAC,kBAAkB;oBACnD,uBAAuB,EAAE,EAAE;oBAC3B,MAAM,EAAE,KAAK;iBACd,CAAC,CAAC;gBAEH,IAAI,CAAC,MAAM,CAAC,sBAAsB,CAAC,CAAC,eAAe,CACjD,iBAAiB,CAAC;oBAChB,SAAS,EAAE,mBAAmB,CAAC,EAAE,KAAK,EAAE,SAAS,EAAE,CAAC;iBACrD,CAAC,CACH,CAAC;gBAEF,MAAM,CACJ,oBAAC,uBAAuB;oBACtB,oBAAC,uBAAuB,CAAC,SAAS;wBAChC,4CAAiB,SAAS,iBAAiB,CACT,CACZ,CAC3B,CAAC;gBAEF,MAAM,CAAC,MAAM,CAAC,WAAW,CAAC,SAAS,CAAC,CAAC,CAAC,iBAAiB,EAAE,CAAC;YAC5D,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,QAAQ,CAAC,mBAAmB,EAAE,GAAG,EAAE;YACjC,EAAE,CAAC,mDAAmD,EAAE,GAAG,EAAE;gBAC3D,IAAI,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,eAAe,CACzC,uBAAuB,CAAC;oBACtB,eAAe,EAAE,eAAe,CAAC,IAAI;iBACtC,CAAC,CACH,CAAC;gBAEF,IAAI,CAAC,MAAM,CAAC,uBAAuB,CAAC,CAAC,eAAe,CAAC;oBACnD,eAAe,EAAE,eAAe,CAAC,IAAI;oBACrC,uBAAuB,EAAE,EAAE;oBAC3B,MAAM,EAAE,IAAI;iBACb,CAAC,CAAC;gBAEH,IAAI,CAAC,MAAM,CAAC,sBAAsB,CAAC,CAAC,eAAe,CACjD,iBAAiB,CAAC;oBAChB,SAAS,EAAE,mBAAmB,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC;iBAC9C,CAAC,CACH,CAAC;gBAEF,MAAM,CACJ,oBAAC,uBAAuB;oBACtB,oBAAC,uBAAuB,CAAC,OAAO;wBAC9B,4CAAiB,UAAU,gBAAgB,CACX,CACV,CAC3B,CAAC;gBAEF,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,UAAU,CAAC,CAAC,CAAC,GAAG,CAAC,iBAAiB,EAAE,CAAC;YACnE,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,QAAQ,CAAC,mBAAmB,EAAE,GAAG,EAAE;YACjC,EAAE,CAAC,6CAA6C,EAAE,GAAG,EAAE;gBACrD,IAAI,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,eAAe,CACzC,uBAAuB,CAAC;oBACtB,eAAe,EAAE,eAAe,CAAC,IAAI;iBACtC,CAAC,CACH,CAAC;gBAEF,IAAI,CAAC,MAAM,CAAC,uBAAuB,CAAC,CAAC,eAAe,CAAC;oBACnD,eAAe,EAAE,eAAe,CAAC,IAAI;oBACrC,uBAAuB,EAAE,EAAE;oBAC3B,MAAM,EAAE,IAAI;iBACb,CAAC,CAAC;gBAEH,IAAI,CAAC,MAAM,CAAC,sBAAsB,CAAC,CAAC,eAAe,CACjD,iBAAiB,CAAC;oBAChB,SAAS,EAAE,mBAAmB,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC;iBAC9C,CAAC,CACH,CAAC;gBAEF,MAAM,CACJ,oBAAC,uBAAuB;oBACtB,oBAAC,uBAAuB,CAAC,OAAO;wBAC9B,4CAAiB,OAAO,aAAa,CACL,CACV,CAC3B,CAAC;gBAEF,MAAM,CAAC,MAAM,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC,CAAC,iBAAiB,EAAE,CAAC;YAC1D,CAAC,CAAC,CAAC;YAEH,EAAE,CAAC,kDAAkD,EAAE,GAAG,EAAE;gBAC1D,IAAI,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,eAAe,CACzC,uBAAuB,CAAC;oBACtB,eAAe,EAAE,eAAe,CAAC,kBAAkB;iBACpD,CAAC,CACH,CAAC;gBAEF,IAAI,CAAC,MAAM,CAAC,uBAAuB,CAAC,CAAC,eAAe,CAAC;oBACnD,eAAe,EAAE,eAAe,CAAC,kBAAkB;oBACnD,uBAAuB,EAAE,EAAE;oBAC3B,MAAM,EAAE,KAAK;iBACd,CAAC,CAAC;gBAEH,IAAI,CAAC,MAAM,CAAC,sBAAsB,CAAC,CAAC,eAAe,CACjD,iBAAiB,CAAC;oBAChB,SAAS,EAAE,mBAAmB,CAAC,EAAE,KAAK,EAAE,SAAS,EAAE,CAAC;iBACrD,CAAC,CACH,CAAC;gBAEF,MAAM,CACJ,oBAAC,uBAAuB;oBACtB,oBAAC,uBAAuB,CAAC,OAAO;wBAC9B,4CAAiB,OAAO,aAAa,CACL,CACV,CAC3B,CAAC;gBAEF,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,OAAO,CAAC,CAAC,CAAC,GAAG,CAAC,iBAAiB,EAAE,CAAC;YAChE,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,kBAAkB,EAAE,GAAG,EAAE;QAChC,EAAE,CAAC,kDAAkD,EAAE,GAAG,EAAE;YAC1D,IAAI,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,eAAe,CACzC,uBAAuB,CAAC;gBACtB,eAAe,EAAE,eAAe,CAAC,QAAQ;aAC1C,CAAC,CACH,CAAC;YAEF,IAAI,CAAC,MAAM,CAAC,uBAAuB,CAAC,CAAC,eAAe,CAAC;gBACnD,eAAe,EAAE,eAAe,CAAC,QAAQ;gBACzC,uBAAuB,EAAE,EAAE;gBAC3B,MAAM,EAAE,KAAK;aACd,CAAC,CAAC;YAEH,IAAI,CAAC,MAAM,CAAC,sBAAsB,CAAC,CAAC,eAAe,CACjD,iBAAiB,CAAC;gBAChB,SAAS,EAAE,mBAAmB,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC;aACjD,CAAC,CACH,CAAC;YAEF,MAAM,CACJ,oBAAC,uBAAuB;gBACtB,oBAAC,iBAAiB,IAAC,OAAO,EAAC,aAAa,GAAG,CACnB,CAC3B,CAAC;YAEF,MAAM,SAAS,GAAG,IAAI,CAAC,KAAK,CAC1B,MAAM,CAAC,WAAW,CAAC,YAAY,CAAC,CAAC,WAAW,CAC7C,CAAC;YACF,MAAM,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACzC,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,+BAA+B,EAAE,GAAG,EAAE;QAC7C,EAAE,CAAC,qDAAqD,EAAE,GAAG,EAAE;YAC7D,IAAI,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,eAAe,CACzC,uBAAuB,CAAC;gBACtB,eAAe,EAAE,eAAe,CAAC,kBAAkB;aACpD,CAAC,CACH,CAAC;YAEF,IAAI,CAAC,MAAM,CAAC,uBAAuB,CAAC,CAAC,eAAe,CAAC;gBACnD,eAAe,EAAE,eAAe,CAAC,kBAAkB;gBACnD,uBAAuB,EAAE,EAAE;gBAC3B,MAAM,EAAE,KAAK;aACd,CAAC,CAAC;YAEH,IAAI,CAAC,MAAM,CAAC,sBAAsB,CAAC,CAAC,eAAe,CACjD,iBAAiB,CAAC;gBAChB,SAAS,EAAE,mBAAmB,CAAC;oBAC7B,IAAI,EAAE,SAAS;oBACf,GAAG,EAAE,EAAE;iBACR,CAAC;aACH,CAAC,CACH,CAAC;YAEF,MAAM,CACJ,oBAAC,uBAAuB;gBACtB,oBAAC,uBAAuB,CAAC,SAAS,IAAC,SAAS,EAAC,MAAM;oBACjD,4CAAiB,cAAc,sBAAsB,CACnB;gBACpC,oBAAC,uBAAuB,CAAC,SAAS,IAAC,SAAS,EAAC,KAAK;oBAChD,4CAAiB,aAAa,qBAAqB,CACjB;gBACpC,oBAAC,uBAAuB,CAAC,OAAO,IAAC,SAAS,EAAC,MAAM;oBAC/C,4CAAiB,eAAe,qBAAqB,CACrB,CACV,CAC3B,CAAC;YAEF,gEAAgE;YAChE,MAAM,CAAC,MAAM,CAAC,WAAW,CAAC,cAAc,CAAC,CAAC,CAAC,iBAAiB,EAAE,CAAC;YAC/D,MAAM,CAAC,MAAM,CAAC,WAAW,CAAC,aAAa,CAAC,CAAC,CAAC,iBAAiB,EAAE,CAAC;YAC9D,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,eAAe,CAAC,CAAC,CAAC,GAAG,CAAC,iBAAiB,EAAE,CAAC;QACxE,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,SAAS,EAAE,GAAG,EAAE;QACvB,EAAE,CAAC,6CAA6C,EAAE,GAAG,EAAE;YACrD,IAAI,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,eAAe,CACzC,uBAAuB,CAAC;gBACtB,eAAe,EAAE,eAAe,CAAC,kBAAkB;aACpD,CAAC,CACH,CAAC;YAEF,IAAI,CAAC,MAAM,CAAC,uBAAuB,CAAC,CAAC,eAAe,CAAC;gBACnD,eAAe,EAAE,eAAe,CAAC,kBAAkB;gBACnD,uBAAuB,EAAE,EAAE;gBAC3B,MAAM,EAAE,KAAK;aACd,CAAC,CAAC;YAEH,IAAI,CAAC,MAAM,CAAC,sBAAsB,CAAC,CAAC,eAAe,CACjD,iBAAiB,CAAC;gBAChB,SAAS,EAAE,mBAAmB,CAAC,EAAE,KAAK,EAAE,SAAS,EAAE,CAAC;aACrD,CAAC,CACH,CAAC;YAEF,MAAM,CACJ,oBAAC,uBAAuB;gBACtB,oBAAC,uBAAuB,CAAC,SAAS,IAAC,SAAS,EAAC,eAAe;oBAC1D,4CAAiB,SAAS,iBAAiB,CACT,CACZ,CAC3B,CAAC;YAEF,MAAM,cAAc,GAAG,MAAM,CAAC,WAAW,CAAC,SAAS,CAAC,CAAC,aAAa,CAAC;YACnE,MAAM,CAAC,cAAc,CAAC,CAAC,WAAW,CAAC,eAAe,CAAC,CAAC;QACtD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC","sourcesContent":["import { render, screen } from \"@testing-library/react\";\nimport React from \"react\";\nimport { useTamboCurrentMessage } from \"../hooks/use-current-message\";\nimport {\n  GenerationStage,\n  TamboThreadMessage,\n} from \"../model/generate-component-response\";\nimport {\n  CombinedTamboThreadContextProps,\n  useTamboGenerationStage,\n  useTamboThread,\n} from \"../providers/tambo-thread-provider\";\nimport {\n  TamboPropStreamProvider,\n  useTamboStream,\n} from \"./tambo-prop-stream-provider\";\n\n// Mock the required providers\njest.mock(\"../providers/tambo-thread-provider\", () => ({\n  useTamboThread: jest.fn(),\n  useTamboGenerationStage: jest.fn(),\n}));\n\njest.mock(\"../hooks/use-current-message\", () => ({\n  useTamboCurrentMessage: jest.fn(),\n}));\n\n// Mock window for SSR tests\nconst originalWindow = global.window;\n\n// Helper function to create mock ComponentDecisionV2\nconst createMockComponent = (props: Record<string, unknown> = {}): any => ({\n  componentName: \"TestComponent\",\n  componentState: {},\n  message: \"Component generated\",\n  props,\n  reasoning: \"Test reasoning\",\n});\n\n// Helper function to create mock TamboThreadMessage\nconst createMockMessage = (\n  overrides: Partial<TamboThreadMessage> = {},\n): TamboThreadMessage => ({\n  id: \"test-message\",\n  componentState: {},\n  content: [{ type: \"text\", text: \"test content\" }],\n  createdAt: new Date().toISOString(),\n  role: \"assistant\",\n  threadId: \"test-thread\",\n  ...overrides,\n});\n\n// Helper function to create mock CombinedTamboThreadContextProps\nconst createMockThreadContext = (\n  overrides: Partial<CombinedTamboThreadContextProps> = {},\n): CombinedTamboThreadContextProps => {\n  const mockThread = {\n    id: \"test-thread\",\n    projectId: \"test-project\",\n    messages: [],\n    name: \"Test Thread\",\n    createdAt: new Date().toISOString(),\n    updatedAt: new Date().toISOString(),\n  };\n  return {\n    // TamboThreadContextProps properties\n    thread: mockThread,\n    currentThreadId: mockThread.id,\n    currentThread: mockThread,\n    threadMap: { [mockThread.id]: mockThread },\n    setThreadMap: jest.fn(),\n    switchCurrentThread: jest.fn(),\n    startNewThread: jest.fn(),\n    updateThreadName: jest.fn(),\n    generateThreadName: jest.fn(),\n    addThreadMessage: jest.fn(),\n    updateThreadMessage: jest.fn(),\n    cancel: jest.fn(),\n    streaming: false,\n    sendThreadMessage: jest.fn(),\n    // GenerationStageContextProps properties\n    generationStage: GenerationStage.IDLE,\n    generationStatusMessage: \"\",\n    isIdle: true,\n    ...overrides,\n  };\n};\n\n// Helper component to test hook usage\nconst TestHookComponent: React.FC<{ testKey?: string }> = ({\n  testKey = \"default\",\n}) => {\n  const { streamStatus, getStatusForKey } = useTamboStream();\n  const status = getStatusForKey(testKey);\n\n  return (\n    <div>\n      <div data-testid=\"stream-status\">{JSON.stringify(streamStatus)}</div>\n      <div data-testid=\"key-status\">{JSON.stringify(status)}</div>\n    </div>\n  );\n};\n\ndescribe(\"TamboPropStreamProvider\", () => {\n  beforeEach(() => {\n    // Restore window for client-side tests\n    global.window = originalWindow;\n\n    // Default mock implementations\n    jest.mocked(useTamboThread).mockReturnValue(createMockThreadContext());\n\n    jest.mocked(useTamboGenerationStage).mockReturnValue({\n      generationStage: GenerationStage.IDLE,\n      generationStatusMessage: \"\",\n      isIdle: true,\n    });\n\n    jest.mocked(useTamboCurrentMessage).mockReturnValue({\n      id: \"test-message\",\n      component: {\n        props: {},\n      },\n    } as TamboThreadMessage);\n  });\n\n  afterEach(() => {\n    jest.clearAllMocks();\n  });\n\n  describe(\"Hook Error Handling\", () => {\n    it(\"should throw error when useTamboStream is used outside provider\", () => {\n      // Suppress console.error for this test\n      const originalError = console.error;\n      console.error = jest.fn();\n\n      expect(() => {\n        render(<TestHookComponent />);\n      }).toThrow(\n        \"useTamboStream must be used within a TamboPropStreamProvider\",\n      );\n\n      console.error = originalError;\n    });\n  });\n\n  describe(\"Compound Components\", () => {\n    describe(\"Streaming Component\", () => {\n      it(\"should render streaming when isPending is true\", () => {\n        jest.mocked(useTamboThread).mockReturnValue(\n          createMockThreadContext({\n            generationStage: GenerationStage.IDLE,\n          }),\n        );\n\n        jest.mocked(useTamboGenerationStage).mockReturnValue({\n          generationStage: GenerationStage.IDLE,\n          generationStatusMessage: \"\",\n          isIdle: true,\n        });\n\n        jest.mocked(useTamboCurrentMessage).mockReturnValue(\n          createMockMessage({\n            component: createMockComponent({ title: \"\" }),\n          }),\n        );\n\n        render(\n          <TamboPropStreamProvider>\n            <TamboPropStreamProvider.Streaming>\n              <div data-testid=\"loading\">Loading...</div>\n            </TamboPropStreamProvider.Streaming>\n          </TamboPropStreamProvider>,\n        );\n\n        expect(screen.getByTestId(\"loading\")).toBeInTheDocument();\n      });\n\n      it(\"should render streaming when isStreaming is true\", () => {\n        jest.mocked(useTamboThread).mockReturnValue(\n          createMockThreadContext({\n            generationStage: GenerationStage.STREAMING_RESPONSE,\n          }),\n        );\n\n        jest.mocked(useTamboGenerationStage).mockReturnValue({\n          generationStage: GenerationStage.STREAMING_RESPONSE,\n          generationStatusMessage: \"\",\n          isIdle: false,\n        });\n\n        jest.mocked(useTamboCurrentMessage).mockReturnValue(\n          createMockMessage({\n            component: createMockComponent({ title: \"Partial\" }),\n          }),\n        );\n\n        render(\n          <TamboPropStreamProvider>\n            <TamboPropStreamProvider.Streaming>\n              <div data-testid=\"loading\">Loading...</div>\n            </TamboPropStreamProvider.Streaming>\n          </TamboPropStreamProvider>,\n        );\n\n        expect(screen.getByTestId(\"loading\")).toBeInTheDocument();\n      });\n    });\n\n    describe(\"Success Component\", () => {\n      it(\"should not render success when isSuccess is false\", () => {\n        jest.mocked(useTamboThread).mockReturnValue(\n          createMockThreadContext({\n            generationStage: GenerationStage.IDLE,\n          }),\n        );\n\n        jest.mocked(useTamboGenerationStage).mockReturnValue({\n          generationStage: GenerationStage.IDLE,\n          generationStatusMessage: \"\",\n          isIdle: true,\n        });\n\n        jest.mocked(useTamboCurrentMessage).mockReturnValue(\n          createMockMessage({\n            component: createMockComponent({ title: \"\" }),\n          }),\n        );\n\n        render(\n          <TamboPropStreamProvider>\n            <TamboPropStreamProvider.Success>\n              <div data-testid=\"complete\">Complete!</div>\n            </TamboPropStreamProvider.Success>\n          </TamboPropStreamProvider>,\n        );\n\n        expect(screen.queryByTestId(\"complete\")).not.toBeInTheDocument();\n      });\n    });\n\n    describe(\"Pending Component\", () => {\n      it(\"should render pending when no active status\", () => {\n        jest.mocked(useTamboThread).mockReturnValue(\n          createMockThreadContext({\n            generationStage: GenerationStage.IDLE,\n          }),\n        );\n\n        jest.mocked(useTamboGenerationStage).mockReturnValue({\n          generationStage: GenerationStage.IDLE,\n          generationStatusMessage: \"\",\n          isIdle: true,\n        });\n\n        jest.mocked(useTamboCurrentMessage).mockReturnValue(\n          createMockMessage({\n            component: createMockComponent({ title: \"\" }),\n          }),\n        );\n\n        render(\n          <TamboPropStreamProvider>\n            <TamboPropStreamProvider.Pending>\n              <div data-testid=\"empty\">Empty!</div>\n            </TamboPropStreamProvider.Pending>\n          </TamboPropStreamProvider>,\n        );\n\n        expect(screen.getByTestId(\"empty\")).toBeInTheDocument();\n      });\n\n      it(\"should not render pending when isPending is true\", () => {\n        jest.mocked(useTamboThread).mockReturnValue(\n          createMockThreadContext({\n            generationStage: GenerationStage.STREAMING_RESPONSE,\n          }),\n        );\n\n        jest.mocked(useTamboGenerationStage).mockReturnValue({\n          generationStage: GenerationStage.STREAMING_RESPONSE,\n          generationStatusMessage: \"\",\n          isIdle: false,\n        });\n\n        jest.mocked(useTamboCurrentMessage).mockReturnValue(\n          createMockMessage({\n            component: createMockComponent({ title: \"Partial\" }),\n          }),\n        );\n\n        render(\n          <TamboPropStreamProvider>\n            <TamboPropStreamProvider.Pending>\n              <div data-testid=\"empty\">Empty!</div>\n            </TamboPropStreamProvider.Pending>\n          </TamboPropStreamProvider>,\n        );\n\n        expect(screen.queryByTestId(\"empty\")).not.toBeInTheDocument();\n      });\n    });\n  });\n\n  describe(\"Key-based Status\", () => {\n    it(\"should provide status for keys not in propStatus\", () => {\n      jest.mocked(useTamboThread).mockReturnValue(\n        createMockThreadContext({\n          generationStage: GenerationStage.COMPLETE,\n        }),\n      );\n\n      jest.mocked(useTamboGenerationStage).mockReturnValue({\n        generationStage: GenerationStage.COMPLETE,\n        generationStatusMessage: \"\",\n        isIdle: false,\n      });\n\n      jest.mocked(useTamboCurrentMessage).mockReturnValue(\n        createMockMessage({\n          component: createMockComponent({ name: \"John\" }),\n        }),\n      );\n\n      render(\n        <TamboPropStreamProvider>\n          <TestHookComponent testKey=\"nonexistent\" />\n        </TamboPropStreamProvider>,\n      );\n\n      const keyStatus = JSON.parse(\n        screen.getByTestId(\"key-status\").textContent,\n      );\n      expect(keyStatus.isPending).toBe(true);\n    });\n  });\n\n  describe(\"Compound Components with Keys\", () => {\n    it(\"should render loading for specific key when pending\", () => {\n      jest.mocked(useTamboThread).mockReturnValue(\n        createMockThreadContext({\n          generationStage: GenerationStage.STREAMING_RESPONSE,\n        }),\n      );\n\n      jest.mocked(useTamboGenerationStage).mockReturnValue({\n        generationStage: GenerationStage.STREAMING_RESPONSE,\n        generationStatusMessage: \"\",\n        isIdle: false,\n      });\n\n      jest.mocked(useTamboCurrentMessage).mockReturnValue(\n        createMockMessage({\n          component: createMockComponent({\n            name: \"Partial\",\n            age: 25,\n          }),\n        }),\n      );\n\n      render(\n        <TamboPropStreamProvider>\n          <TamboPropStreamProvider.Streaming streamKey=\"name\">\n            <div data-testid=\"name-loading\">Name Loading...</div>\n          </TamboPropStreamProvider.Streaming>\n          <TamboPropStreamProvider.Streaming streamKey=\"age\">\n            <div data-testid=\"age-loading\">Age Loading...</div>\n          </TamboPropStreamProvider.Streaming>\n          <TamboPropStreamProvider.Success streamKey=\"name\">\n            <div data-testid=\"name-complete\">Name Complete!</div>\n          </TamboPropStreamProvider.Success>\n        </TamboPropStreamProvider>,\n      );\n\n      // Both props should be loading since they're in streaming stage\n      expect(screen.getByTestId(\"name-loading\")).toBeInTheDocument();\n      expect(screen.getByTestId(\"age-loading\")).toBeInTheDocument();\n      expect(screen.queryByTestId(\"name-complete\")).not.toBeInTheDocument();\n    });\n  });\n\n  describe(\"Styling\", () => {\n    it(\"should apply className to loading component\", () => {\n      jest.mocked(useTamboThread).mockReturnValue(\n        createMockThreadContext({\n          generationStage: GenerationStage.STREAMING_RESPONSE,\n        }),\n      );\n\n      jest.mocked(useTamboGenerationStage).mockReturnValue({\n        generationStage: GenerationStage.STREAMING_RESPONSE,\n        generationStatusMessage: \"\",\n        isIdle: false,\n      });\n\n      jest.mocked(useTamboCurrentMessage).mockReturnValue(\n        createMockMessage({\n          component: createMockComponent({ title: \"Partial\" }),\n        }),\n      );\n\n      render(\n        <TamboPropStreamProvider>\n          <TamboPropStreamProvider.Streaming className=\"loading-class\">\n            <div data-testid=\"loading\">Loading...</div>\n          </TamboPropStreamProvider.Streaming>\n        </TamboPropStreamProvider>,\n      );\n\n      const loadingElement = screen.getByTestId(\"loading\").parentElement;\n      expect(loadingElement).toHaveClass(\"loading-class\");\n    });\n  });\n});\n"]}

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

@@ -22,8 +22,8 @@ import { TamboGenerationStageContextProps, TamboThreadContextProps, TamboThreadP
  * @param props.autoGenerateThreadName - Whether to automatically generate thread names. Defaults to true.
  * @param props.autoGenerateNameThreshold - The message count threshold at which the thread name will be auto-generated. Defaults to 3.
  * @param props.contextHelpers - Configuration for which context helpers are enabled/disabled
- * @param props.userToken - The user's OAuth token (access or ID) used to identify the user and exchange for a Tambo session token (preferred over contextKey)
- * @param props.contextKey - Optional context key to be used in the thread input provider
+ * @param props.userToken - The user's OAuth token (access or ID) used to identify the user and exchange for a Tambo session token
+ * @param props.contextKey - Optional context key passed to thread input provider for scoping threads
  * @param props.onCallUnregisteredTool - Callback function called when an unregistered tool is called
  * @param props.initialMessages - Initial messages to be included in new threads
  * @param props.getContextHelperData - Optional function to customize the data sent to the AI for each context attachment

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

@@ -1 +1 @@
-{"version":3,"file":"tambo-provider.d.ts","sourceRoot":"","sources":["../../src/providers/tambo-provider.tsx"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,EAAE,iBAAiB,EAA6B,MAAM,OAAO,CAAC;AAC5E,OAAO,EAAE,wBAAwB,EAAE,MAAM,6BAA6B,CAAC;AACvE,OAAO,EAEL,uBAAuB,EAEvB,wBAAwB,EACzB,MAAM,yBAAyB,CAAC;AACjC,OAAO,EACL,0BAA0B,EAG3B,MAAM,4BAA4B,CAAC;AACpC,OAAO,EACL,sBAAsB,EAEtB,mCAAmC,EAEpC,MAAM,qCAAqC,CAAC;AAC7C,OAAO,EACL,+BAA+B,EAE/B,gCAAgC,EAEjC,MAAM,kCAAkC,CAAC;AAM1C,OAAO,EAEL,0BAA0B,EAC3B,MAAM,2BAA2B,CAAC;AACnC,OAAO,EAEL,6BAA6B,EAC9B,MAAM,+BAA+B,CAAC;AACvC,OAAO,EACL,gCAAgC,EAChC,uBAAuB,EAEvB,wBAAwB,EAEzB,MAAM,yBAAyB,CAAC;AAEjC;;;;;;;;;;;;;;;;;;;;;GAqBG;AACH,eAAO,MAAM,aAAa,EAAE,KAAK,CAAC,EAAE,CAClC,iBAAiB,CACf,wBAAwB,GACtB,0BAA0B,GAC1B,wBAAwB,GACxB,gCAAgC,GAChC,6BAA6B,GAC7B,OAAO,CAAC,IAAI,CAAC,mCAAmC,EAAE,sBAAsB,CAAC,CAAC,CAC7E,CA2DF,CAAC;AAEF,MAAM,MAAM,iBAAiB,GAAG,uBAAuB,GACrD,uBAAuB,GACvB,gCAAgC,GAChC,0BAA0B,GAC1B,wBAAwB,GACxB,+BAA+B,GAC/B,sBAAsB,CAAC;AAEzB,eAAO,MAAM,YAAY,kCAExB,CAAC;AAEF;;;;;;GAMG;AACH,eAAO,MAAM,sBAAsB,EAAE,KAAK,CAAC,EAAE,CAAC,iBAAiB,CAwC9D,CAAC;AAEF;;;;;;;GAOG;AACH,eAAO,MAAM,QAAQ,yBAEpB,CAAC"}
+{"version":3,"file":"tambo-provider.d.ts","sourceRoot":"","sources":["../../src/providers/tambo-provider.tsx"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,EAAE,iBAAiB,EAA6B,MAAM,OAAO,CAAC;AAC5E,OAAO,EAAE,wBAAwB,EAAE,MAAM,6BAA6B,CAAC;AACvE,OAAO,EAEL,uBAAuB,EAEvB,wBAAwB,EACzB,MAAM,yBAAyB,CAAC;AACjC,OAAO,EACL,0BAA0B,EAG3B,MAAM,4BAA4B,CAAC;AACpC,OAAO,EACL,sBAAsB,EAEtB,mCAAmC,EAEpC,MAAM,qCAAqC,CAAC;AAC7C,OAAO,EACL,+BAA+B,EAE/B,gCAAgC,EAEjC,MAAM,kCAAkC,CAAC;AAM1C,OAAO,EAEL,0BAA0B,EAC3B,MAAM,2BAA2B,CAAC;AACnC,OAAO,EAEL,6BAA6B,EAC9B,MAAM,+BAA+B,CAAC;AACvC,OAAO,EACL,gCAAgC,EAChC,uBAAuB,EAEvB,wBAAwB,EAEzB,MAAM,yBAAyB,CAAC;AAEjC;;;;;;;;;;;;;;;;;;;;;GAqBG;AACH,eAAO,MAAM,aAAa,EAAE,KAAK,CAAC,EAAE,CAClC,iBAAiB,CACf,wBAAwB,GACtB,0BAA0B,GAC1B,wBAAwB,GACxB,gCAAgC,GAChC,6BAA6B,GAC7B,OAAO,CAAC,IAAI,CAAC,mCAAmC,EAAE,sBAAsB,CAAC,CAAC,CAC7E,CA2DF,CAAC;AAEF,MAAM,MAAM,iBAAiB,GAAG,uBAAuB,GACrD,uBAAuB,GACvB,gCAAgC,GAChC,0BAA0B,GAC1B,wBAAwB,GACxB,+BAA+B,GAC/B,sBAAsB,CAAC;AAEzB,eAAO,MAAM,YAAY,kCAExB,CAAC;AAEF;;;;;;GAMG;AACH,eAAO,MAAM,sBAAsB,EAAE,KAAK,CAAC,EAAE,CAAC,iBAAiB,CAgC9D,CAAC;AAEF;;;;;;;GAOG;AACH,eAAO,MAAM,QAAQ,yBAEpB,CAAC"}

package/esm/providers/tambo-provider.js

@@ -24,8 +24,8 @@ import { TamboThreadProvider, useTamboThread, } from "./tambo-thread-provider";
  * @param props.autoGenerateThreadName - Whether to automatically generate thread names. Defaults to true.
  * @param props.autoGenerateNameThreshold - The message count threshold at which the thread name will be auto-generated. Defaults to 3.
  * @param props.contextHelpers - Configuration for which context helpers are enabled/disabled
- * @param props.userToken - The user's OAuth token (access or ID) used to identify the user and exchange for a Tambo session token (preferred over contextKey)
- * @param props.contextKey - Optional context key to be used in the thread input provider
+ * @param props.userToken - The user's OAuth token (access or ID) used to identify the user and exchange for a Tambo session token
+ * @param props.contextKey - Optional context key passed to thread input provider for scoping threads
  * @param props.onCallUnregisteredTool - Callback function called when an unregistered tool is called
  * @param props.initialMessages - Initial messages to be included in new threads
  * @param props.getContextHelperData - Optional function to customize the data sent to the AI for each context attachment
@@ -35,8 +35,8 @@ export const TamboProvider = ({ children, tamboUrl, apiKey, userToken, component
     return (React.createElement(TamboClientProvider, { tamboUrl: tamboUrl, apiKey: apiKey, environment: environment, userToken: userToken },
         React.createElement(TamboRegistryProvider, { components: components, tools: tools, mcpServers: mcpServers, onCallUnregisteredTool: onCallUnregisteredTool },
             React.createElement(TamboContextHelpersProvider, { contextHelpers: contextHelpers },
-                React.createElement(TamboMcpTokenProvider, null,
-                    React.createElement(TamboThreadProvider, { streaming: streaming, autoGenerateThreadName: autoGenerateThreadName, autoGenerateNameThreshold: autoGenerateNameThreshold, initialMessages: initialMessages },
+                React.createElement(TamboThreadProvider, { streaming: streaming, autoGenerateThreadName: autoGenerateThreadName, autoGenerateNameThreshold: autoGenerateNameThreshold, initialMessages: initialMessages },
+                    React.createElement(TamboMcpTokenProvider, null,
                         React.createElement(TamboThreadInputProvider, { contextKey: contextKey },
                             React.createElement(TamboContextAttachmentProvider, { getContextHelperData: getContextHelperData },
                                 React.createElement(TamboComponentProvider, null,
@@ -57,7 +57,7 @@ export const TamboCompositeProvider = ({ children, }) => {
     if (!clientContext) {
         throw new Error("TamboCompositeProvider must be used within a TamboClientProvider");
     }
-    const { client, queryClient, isUpdatingToken, mcpAccessToken, setMcpAccessToken, } = clientContext;
+    const { client, queryClient, isUpdatingToken } = clientContext;
     const componentRegistry = useTamboComponent();
     const interactableComponents = useTamboInteractable();
     const contextHelpers = useTamboContextHelpers();
@@ -66,8 +66,6 @@ export const TamboCompositeProvider = ({ children, }) => {
             client,
             queryClient,
             isUpdatingToken,
-            mcpAccessToken,
-            setMcpAccessToken,
             ...componentRegistry,
             ...threads,
             ...interactableComponents,

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

@@ -1 +1 @@
-{"version":3,"file":"tambo-provider.js","sourceRoot":"","sources":["../../src/providers/tambo-provider.tsx"],"names":[],"mappings":"AAAA,YAAY,CAAC;AACb,OAAO,KAAK,EAAE,EAAqB,aAAa,EAAE,UAAU,EAAE,MAAM,OAAO,CAAC;AAE5E,OAAO,EACL,kBAAkB,EAElB,mBAAmB,GAEpB,MAAM,yBAAyB,CAAC;AACjC,OAAO,EAEL,sBAAsB,EACtB,iBAAiB,GAClB,MAAM,4BAA4B,CAAC;AACpC,OAAO,EAEL,8BAA8B,EAE9B,yBAAyB,GAC1B,MAAM,qCAAqC,CAAC;AAC7C,OAAO,EAEL,2BAA2B,EAE3B,sBAAsB,GACvB,MAAM,kCAAkC,CAAC;AAC1C,OAAO,EACL,yBAAyB,EACzB,oBAAoB,GACrB,MAAM,+BAA+B,CAAC;AACvC,OAAO,EAAE,qBAAqB,EAAE,MAAM,4BAA4B,CAAC;AACnE,OAAO,EACL,qBAAqB,GAEtB,MAAM,2BAA2B,CAAC;AACnC,OAAO,EACL,wBAAwB,GAEzB,MAAM,+BAA+B,CAAC;AACvC,OAAO,EAGL,mBAAmB,EAEnB,cAAc,GACf,MAAM,yBAAyB,CAAC;AAEjC;;;;;;;;;;;;;;;;;;;;;GAqBG;AACH,MAAM,CAAC,MAAM,aAAa,GAStB,CAAC,EACH,QAAQ,EACR,QAAQ,EACR,MAAM,EACN,SAAS,EACT,UAAU,EACV,WAAW,EACX,KAAK,EACL,UAAU,EACV,SAAS,EACT,sBAAsB,EACtB,yBAAyB,EACzB,cAAc,EACd,UAAU,EACV,eAAe,EACf,sBAAsB,EACtB,oBAAoB,GACrB,EAAE,EAAE;IACH,OAAO,CACL,oBAAC,mBAAmB,IAClB,QAAQ,EAAE,QAAQ,EAClB,MAAM,EAAE,MAAM,EACd,WAAW,EAAE,WAAW,EACxB,SAAS,EAAE,SAAS;QAEpB,oBAAC,qBAAqB,IACpB,UAAU,EAAE,UAAU,EACtB,KAAK,EAAE,KAAK,EACZ,UAAU,EAAE,UAAU,EACtB,sBAAsB,EAAE,sBAAsB;YAE9C,oBAAC,2BAA2B,IAAC,cAAc,EAAE,cAAc;gBACzD,oBAAC,qBAAqB;oBACpB,oBAAC,mBAAmB,IAClB,SAAS,EAAE,SAAS,EACpB,sBAAsB,EAAE,sBAAsB,EAC9C,yBAAyB,EAAE,yBAAyB,EACpD,eAAe,EAAE,eAAe;wBAEhC,oBAAC,wBAAwB,IAAC,UAAU,EAAE,UAAU;4BAC9C,oBAAC,8BAA8B,IAC7B,oBAAoB,EAAE,oBAAoB;gCAE1C,oBAAC,sBAAsB;oCACrB,oBAAC,yBAAyB;wCACxB,oBAAC,sBAAsB,QACpB,QAAQ,CACc,CACC,CACL,CACM,CACR,CACP,CACA,CACI,CACR,CACJ,CACvB,CAAC;AACJ,CAAC,CAAC;AAUF,MAAM,CAAC,MAAM,YAAY,GAAG,aAAa,CACvC,EAAuB,CACxB,CAAC;AAEF;;;;;;GAMG;AACH,MAAM,CAAC,MAAM,sBAAsB,GAAgC,CAAC,EAClE,QAAQ,GACT,EAAE,EAAE;IACH,MAAM,OAAO,GAAG,cAAc,EAAE,CAAC;IACjC,MAAM,aAAa,GAAG,UAAU,CAAC,kBAAkB,CAAC,CAAC;IACrD,IAAI,CAAC,aAAa,EAAE,CAAC;QACnB,MAAM,IAAI,KAAK,CACb,kEAAkE,CACnE,CAAC;IACJ,CAAC;IACD,MAAM,EACJ,MAAM,EACN,WAAW,EACX,eAAe,EACf,cAAc,EACd,iBAAiB,GAClB,GAAG,aAAa,CAAC;IAClB,MAAM,iBAAiB,GAAG,iBAAiB,EAAE,CAAC;IAC9C,MAAM,sBAAsB,GAAG,oBAAoB,EAAE,CAAC;IACtD,MAAM,cAAc,GAAG,sBAAsB,EAAE,CAAC;IAChD,MAAM,iBAAiB,GAAG,yBAAyB,EAAE,CAAC;IAEtD,OAAO,CACL,oBAAC,YAAY,CAAC,QAAQ,IACpB,KAAK,EAAE;YACL,MAAM;YACN,WAAW;YACX,eAAe;YACf,cAAc;YACd,iBAAiB;YACjB,GAAG,iBAAiB;YACpB,GAAG,OAAO;YACV,GAAG,sBAAsB;YACzB,GAAG,cAAc;YACjB,GAAG,iBAAiB;SACrB,IAEA,QAAQ,CACa,CACzB,CAAC;AACJ,CAAC,CAAC;AAEF;;;;;;;GAOG;AACH,MAAM,CAAC,MAAM,QAAQ,GAAG,GAAG,EAAE;IAC3B,OAAO,UAAU,CAAC,YAAY,CAAC,CAAC;AAClC,CAAC,CAAC","sourcesContent":["\"use client\";\nimport React, { PropsWithChildren, createContext, useContext } from \"react\";\nimport { TamboInteractableContext } from \"../model/tambo-interactable\";\nimport {\n  TamboClientContext,\n  TamboClientContextProps,\n  TamboClientProvider,\n  TamboClientProviderProps,\n} from \"./tambo-client-provider\";\nimport {\n  TamboComponentContextProps,\n  TamboComponentProvider,\n  useTamboComponent,\n} from \"./tambo-component-provider\";\nimport {\n  ContextAttachmentState,\n  TamboContextAttachmentProvider,\n  TamboContextAttachmentProviderProps,\n  useTamboContextAttachment,\n} from \"./tambo-context-attachment-provider\";\nimport {\n  TamboContextHelpersContextProps,\n  TamboContextHelpersProvider,\n  TamboContextHelpersProviderProps,\n  useTamboContextHelpers,\n} from \"./tambo-context-helpers-provider\";\nimport {\n  TamboInteractableProvider,\n  useTamboInteractable,\n} from \"./tambo-interactable-provider\";\nimport { TamboMcpTokenProvider } from \"./tambo-mcp-token-provider\";\nimport {\n  TamboRegistryProvider,\n  TamboRegistryProviderProps,\n} from \"./tambo-registry-provider\";\nimport {\n  TamboThreadInputProvider,\n  TamboThreadInputProviderProps,\n} from \"./tambo-thread-input-provider\";\nimport {\n  TamboGenerationStageContextProps,\n  TamboThreadContextProps,\n  TamboThreadProvider,\n  TamboThreadProviderProps,\n  useTamboThread,\n} from \"./tambo-thread-provider\";\n\n/**\n * The TamboProvider gives full access to the whole Tambo API. This includes the\n * TamboAI client, the component registry, the current thread context, and interactable components.\n * @param props - The props for the TamboProvider\n * @param props.children - The children to wrap\n * @param props.tamboUrl - The URL of the Tambo API\n * @param props.apiKey - The API key for the Tambo API\n * @param props.components - The components to register\n * @param props.environment - The environment to use for the Tambo API\n * @param props.tools - The tools to register\n * @param props.mcpServers - The MCP servers to register (metadata only - use TamboMcpProvider for connections)\n * @param props.streaming - Whether to stream the response by default. Defaults to true.\n * @param props.autoGenerateThreadName - Whether to automatically generate thread names. Defaults to true.\n * @param props.autoGenerateNameThreshold - The message count threshold at which the thread name will be auto-generated. Defaults to 3.\n * @param props.contextHelpers - Configuration for which context helpers are enabled/disabled\n * @param props.userToken - The user's OAuth token (access or ID) used to identify the user and exchange for a Tambo session token (preferred over contextKey)\n * @param props.contextKey - Optional context key to be used in the thread input provider\n * @param props.onCallUnregisteredTool - Callback function called when an unregistered tool is called\n * @param props.initialMessages - Initial messages to be included in new threads\n * @param props.getContextHelperData - Optional function to customize the data sent to the AI for each context attachment\n * @returns The TamboProvider component\n */\nexport const TamboProvider: React.FC<\n  PropsWithChildren<\n    TamboClientProviderProps &\n      TamboRegistryProviderProps &\n      TamboThreadProviderProps &\n      TamboContextHelpersProviderProps &\n      TamboThreadInputProviderProps &\n      Partial<Pick<TamboContextAttachmentProviderProps, \"getContextHelperData\">>\n  >\n> = ({\n  children,\n  tamboUrl,\n  apiKey,\n  userToken,\n  components,\n  environment,\n  tools,\n  mcpServers,\n  streaming,\n  autoGenerateThreadName,\n  autoGenerateNameThreshold,\n  contextHelpers,\n  contextKey,\n  initialMessages,\n  onCallUnregisteredTool,\n  getContextHelperData,\n}) => {\n  return (\n    <TamboClientProvider\n      tamboUrl={tamboUrl}\n      apiKey={apiKey}\n      environment={environment}\n      userToken={userToken}\n    >\n      <TamboRegistryProvider\n        components={components}\n        tools={tools}\n        mcpServers={mcpServers}\n        onCallUnregisteredTool={onCallUnregisteredTool}\n      >\n        <TamboContextHelpersProvider contextHelpers={contextHelpers}>\n          <TamboMcpTokenProvider>\n            <TamboThreadProvider\n              streaming={streaming}\n              autoGenerateThreadName={autoGenerateThreadName}\n              autoGenerateNameThreshold={autoGenerateNameThreshold}\n              initialMessages={initialMessages}\n            >\n              <TamboThreadInputProvider contextKey={contextKey}>\n                <TamboContextAttachmentProvider\n                  getContextHelperData={getContextHelperData}\n                >\n                  <TamboComponentProvider>\n                    <TamboInteractableProvider>\n                      <TamboCompositeProvider>\n                        {children}\n                      </TamboCompositeProvider>\n                    </TamboInteractableProvider>\n                  </TamboComponentProvider>\n                </TamboContextAttachmentProvider>\n              </TamboThreadInputProvider>\n            </TamboThreadProvider>\n          </TamboMcpTokenProvider>\n        </TamboContextHelpersProvider>\n      </TamboRegistryProvider>\n    </TamboClientProvider>\n  );\n};\n\nexport type TamboContextProps = TamboClientContextProps &\n  TamboThreadContextProps &\n  TamboGenerationStageContextProps &\n  TamboComponentContextProps &\n  TamboInteractableContext &\n  TamboContextHelpersContextProps &\n  ContextAttachmentState;\n\nexport const TamboContext = createContext<TamboContextProps>(\n  {} as TamboContextProps,\n);\n\n/**\n * TamboCompositeProvider is a provider that combines the TamboClient,\n * TamboThread, TamboComponent, and TamboInteractable providers\n * @param props - The props for the TamboCompositeProvider\n * @param props.children - The children to wrap\n * @returns The wrapped component\n */\nexport const TamboCompositeProvider: React.FC<PropsWithChildren> = ({\n  children,\n}) => {\n  const threads = useTamboThread();\n  const clientContext = useContext(TamboClientContext);\n  if (!clientContext) {\n    throw new Error(\n      \"TamboCompositeProvider must be used within a TamboClientProvider\",\n    );\n  }\n  const {\n    client,\n    queryClient,\n    isUpdatingToken,\n    mcpAccessToken,\n    setMcpAccessToken,\n  } = clientContext;\n  const componentRegistry = useTamboComponent();\n  const interactableComponents = useTamboInteractable();\n  const contextHelpers = useTamboContextHelpers();\n  const contextAttachment = useTamboContextAttachment();\n\n  return (\n    <TamboContext.Provider\n      value={{\n        client,\n        queryClient,\n        isUpdatingToken,\n        mcpAccessToken,\n        setMcpAccessToken,\n        ...componentRegistry,\n        ...threads,\n        ...interactableComponents,\n        ...contextHelpers,\n        ...contextAttachment,\n      }}\n    >\n      {children}\n    </TamboContext.Provider>\n  );\n};\n\n/**\n * The useTambo hook provides access to the Tambo API. This is the primary entrypoint\n * for the Tambo React SDK.\n *\n * This includes the TamboAI client, the component registry, the current thread context,\n * and interactable component management.\n * @returns The Tambo API\n */\nexport const useTambo = () => {\n  return useContext(TamboContext);\n};\n"]}
+{"version":3,"file":"tambo-provider.js","sourceRoot":"","sources":["../../src/providers/tambo-provider.tsx"],"names":[],"mappings":"AAAA,YAAY,CAAC;AACb,OAAO,KAAK,EAAE,EAAqB,aAAa,EAAE,UAAU,EAAE,MAAM,OAAO,CAAC;AAE5E,OAAO,EACL,kBAAkB,EAElB,mBAAmB,GAEpB,MAAM,yBAAyB,CAAC;AACjC,OAAO,EAEL,sBAAsB,EACtB,iBAAiB,GAClB,MAAM,4BAA4B,CAAC;AACpC,OAAO,EAEL,8BAA8B,EAE9B,yBAAyB,GAC1B,MAAM,qCAAqC,CAAC;AAC7C,OAAO,EAEL,2BAA2B,EAE3B,sBAAsB,GACvB,MAAM,kCAAkC,CAAC;AAC1C,OAAO,EACL,yBAAyB,EACzB,oBAAoB,GACrB,MAAM,+BAA+B,CAAC;AACvC,OAAO,EAAE,qBAAqB,EAAE,MAAM,4BAA4B,CAAC;AACnE,OAAO,EACL,qBAAqB,GAEtB,MAAM,2BAA2B,CAAC;AACnC,OAAO,EACL,wBAAwB,GAEzB,MAAM,+BAA+B,CAAC;AACvC,OAAO,EAGL,mBAAmB,EAEnB,cAAc,GACf,MAAM,yBAAyB,CAAC;AAEjC;;;;;;;;;;;;;;;;;;;;;GAqBG;AACH,MAAM,CAAC,MAAM,aAAa,GAStB,CAAC,EACH,QAAQ,EACR,QAAQ,EACR,MAAM,EACN,SAAS,EACT,UAAU,EACV,WAAW,EACX,KAAK,EACL,UAAU,EACV,SAAS,EACT,sBAAsB,EACtB,yBAAyB,EACzB,cAAc,EACd,UAAU,EACV,eAAe,EACf,sBAAsB,EACtB,oBAAoB,GACrB,EAAE,EAAE;IACH,OAAO,CACL,oBAAC,mBAAmB,IAClB,QAAQ,EAAE,QAAQ,EAClB,MAAM,EAAE,MAAM,EACd,WAAW,EAAE,WAAW,EACxB,SAAS,EAAE,SAAS;QAEpB,oBAAC,qBAAqB,IACpB,UAAU,EAAE,UAAU,EACtB,KAAK,EAAE,KAAK,EACZ,UAAU,EAAE,UAAU,EACtB,sBAAsB,EAAE,sBAAsB;YAE9C,oBAAC,2BAA2B,IAAC,cAAc,EAAE,cAAc;gBACzD,oBAAC,mBAAmB,IAClB,SAAS,EAAE,SAAS,EACpB,sBAAsB,EAAE,sBAAsB,EAC9C,yBAAyB,EAAE,yBAAyB,EACpD,eAAe,EAAE,eAAe;oBAEhC,oBAAC,qBAAqB;wBACpB,oBAAC,wBAAwB,IAAC,UAAU,EAAE,UAAU;4BAC9C,oBAAC,8BAA8B,IAC7B,oBAAoB,EAAE,oBAAoB;gCAE1C,oBAAC,sBAAsB;oCACrB,oBAAC,yBAAyB;wCACxB,oBAAC,sBAAsB,QACpB,QAAQ,CACc,CACC,CACL,CACM,CACR,CACL,CACJ,CACM,CACR,CACJ,CACvB,CAAC;AACJ,CAAC,CAAC;AAUF,MAAM,CAAC,MAAM,YAAY,GAAG,aAAa,CACvC,EAAuB,CACxB,CAAC;AAEF;;;;;;GAMG;AACH,MAAM,CAAC,MAAM,sBAAsB,GAAgC,CAAC,EAClE,QAAQ,GACT,EAAE,EAAE;IACH,MAAM,OAAO,GAAG,cAAc,EAAE,CAAC;IACjC,MAAM,aAAa,GAAG,UAAU,CAAC,kBAAkB,CAAC,CAAC;IACrD,IAAI,CAAC,aAAa,EAAE,CAAC;QACnB,MAAM,IAAI,KAAK,CACb,kEAAkE,CACnE,CAAC;IACJ,CAAC;IACD,MAAM,EAAE,MAAM,EAAE,WAAW,EAAE,eAAe,EAAE,GAAG,aAAa,CAAC;IAC/D,MAAM,iBAAiB,GAAG,iBAAiB,EAAE,CAAC;IAC9C,MAAM,sBAAsB,GAAG,oBAAoB,EAAE,CAAC;IACtD,MAAM,cAAc,GAAG,sBAAsB,EAAE,CAAC;IAChD,MAAM,iBAAiB,GAAG,yBAAyB,EAAE,CAAC;IAEtD,OAAO,CACL,oBAAC,YAAY,CAAC,QAAQ,IACpB,KAAK,EAAE;YACL,MAAM;YACN,WAAW;YACX,eAAe;YACf,GAAG,iBAAiB;YACpB,GAAG,OAAO;YACV,GAAG,sBAAsB;YACzB,GAAG,cAAc;YACjB,GAAG,iBAAiB;SACrB,IAEA,QAAQ,CACa,CACzB,CAAC;AACJ,CAAC,CAAC;AAEF;;;;;;;GAOG;AACH,MAAM,CAAC,MAAM,QAAQ,GAAG,GAAG,EAAE;IAC3B,OAAO,UAAU,CAAC,YAAY,CAAC,CAAC;AAClC,CAAC,CAAC","sourcesContent":["\"use client\";\nimport React, { PropsWithChildren, createContext, useContext } from \"react\";\nimport { TamboInteractableContext } from \"../model/tambo-interactable\";\nimport {\n  TamboClientContext,\n  TamboClientContextProps,\n  TamboClientProvider,\n  TamboClientProviderProps,\n} from \"./tambo-client-provider\";\nimport {\n  TamboComponentContextProps,\n  TamboComponentProvider,\n  useTamboComponent,\n} from \"./tambo-component-provider\";\nimport {\n  ContextAttachmentState,\n  TamboContextAttachmentProvider,\n  TamboContextAttachmentProviderProps,\n  useTamboContextAttachment,\n} from \"./tambo-context-attachment-provider\";\nimport {\n  TamboContextHelpersContextProps,\n  TamboContextHelpersProvider,\n  TamboContextHelpersProviderProps,\n  useTamboContextHelpers,\n} from \"./tambo-context-helpers-provider\";\nimport {\n  TamboInteractableProvider,\n  useTamboInteractable,\n} from \"./tambo-interactable-provider\";\nimport { TamboMcpTokenProvider } from \"./tambo-mcp-token-provider\";\nimport {\n  TamboRegistryProvider,\n  TamboRegistryProviderProps,\n} from \"./tambo-registry-provider\";\nimport {\n  TamboThreadInputProvider,\n  TamboThreadInputProviderProps,\n} from \"./tambo-thread-input-provider\";\nimport {\n  TamboGenerationStageContextProps,\n  TamboThreadContextProps,\n  TamboThreadProvider,\n  TamboThreadProviderProps,\n  useTamboThread,\n} from \"./tambo-thread-provider\";\n\n/**\n * The TamboProvider gives full access to the whole Tambo API. This includes the\n * TamboAI client, the component registry, the current thread context, and interactable components.\n * @param props - The props for the TamboProvider\n * @param props.children - The children to wrap\n * @param props.tamboUrl - The URL of the Tambo API\n * @param props.apiKey - The API key for the Tambo API\n * @param props.components - The components to register\n * @param props.environment - The environment to use for the Tambo API\n * @param props.tools - The tools to register\n * @param props.mcpServers - The MCP servers to register (metadata only - use TamboMcpProvider for connections)\n * @param props.streaming - Whether to stream the response by default. Defaults to true.\n * @param props.autoGenerateThreadName - Whether to automatically generate thread names. Defaults to true.\n * @param props.autoGenerateNameThreshold - The message count threshold at which the thread name will be auto-generated. Defaults to 3.\n * @param props.contextHelpers - Configuration for which context helpers are enabled/disabled\n * @param props.userToken - The user's OAuth token (access or ID) used to identify the user and exchange for a Tambo session token\n * @param props.contextKey - Optional context key passed to thread input provider for scoping threads\n * @param props.onCallUnregisteredTool - Callback function called when an unregistered tool is called\n * @param props.initialMessages - Initial messages to be included in new threads\n * @param props.getContextHelperData - Optional function to customize the data sent to the AI for each context attachment\n * @returns The TamboProvider component\n */\nexport const TamboProvider: React.FC<\n  PropsWithChildren<\n    TamboClientProviderProps &\n      TamboRegistryProviderProps &\n      TamboThreadProviderProps &\n      TamboContextHelpersProviderProps &\n      TamboThreadInputProviderProps &\n      Partial<Pick<TamboContextAttachmentProviderProps, \"getContextHelperData\">>\n  >\n> = ({\n  children,\n  tamboUrl,\n  apiKey,\n  userToken,\n  components,\n  environment,\n  tools,\n  mcpServers,\n  streaming,\n  autoGenerateThreadName,\n  autoGenerateNameThreshold,\n  contextHelpers,\n  contextKey,\n  initialMessages,\n  onCallUnregisteredTool,\n  getContextHelperData,\n}) => {\n  return (\n    <TamboClientProvider\n      tamboUrl={tamboUrl}\n      apiKey={apiKey}\n      environment={environment}\n      userToken={userToken}\n    >\n      <TamboRegistryProvider\n        components={components}\n        tools={tools}\n        mcpServers={mcpServers}\n        onCallUnregisteredTool={onCallUnregisteredTool}\n      >\n        <TamboContextHelpersProvider contextHelpers={contextHelpers}>\n          <TamboThreadProvider\n            streaming={streaming}\n            autoGenerateThreadName={autoGenerateThreadName}\n            autoGenerateNameThreshold={autoGenerateNameThreshold}\n            initialMessages={initialMessages}\n          >\n            <TamboMcpTokenProvider>\n              <TamboThreadInputProvider contextKey={contextKey}>\n                <TamboContextAttachmentProvider\n                  getContextHelperData={getContextHelperData}\n                >\n                  <TamboComponentProvider>\n                    <TamboInteractableProvider>\n                      <TamboCompositeProvider>\n                        {children}\n                      </TamboCompositeProvider>\n                    </TamboInteractableProvider>\n                  </TamboComponentProvider>\n                </TamboContextAttachmentProvider>\n              </TamboThreadInputProvider>\n            </TamboMcpTokenProvider>\n          </TamboThreadProvider>\n        </TamboContextHelpersProvider>\n      </TamboRegistryProvider>\n    </TamboClientProvider>\n  );\n};\n\nexport type TamboContextProps = TamboClientContextProps &\n  TamboThreadContextProps &\n  TamboGenerationStageContextProps &\n  TamboComponentContextProps &\n  TamboInteractableContext &\n  TamboContextHelpersContextProps &\n  ContextAttachmentState;\n\nexport const TamboContext = createContext<TamboContextProps>(\n  {} as TamboContextProps,\n);\n\n/**\n * TamboCompositeProvider is a provider that combines the TamboClient,\n * TamboThread, TamboComponent, and TamboInteractable providers\n * @param props - The props for the TamboCompositeProvider\n * @param props.children - The children to wrap\n * @returns The wrapped component\n */\nexport const TamboCompositeProvider: React.FC<PropsWithChildren> = ({\n  children,\n}) => {\n  const threads = useTamboThread();\n  const clientContext = useContext(TamboClientContext);\n  if (!clientContext) {\n    throw new Error(\n      \"TamboCompositeProvider must be used within a TamboClientProvider\",\n    );\n  }\n  const { client, queryClient, isUpdatingToken } = clientContext;\n  const componentRegistry = useTamboComponent();\n  const interactableComponents = useTamboInteractable();\n  const contextHelpers = useTamboContextHelpers();\n  const contextAttachment = useTamboContextAttachment();\n\n  return (\n    <TamboContext.Provider\n      value={{\n        client,\n        queryClient,\n        isUpdatingToken,\n        ...componentRegistry,\n        ...threads,\n        ...interactableComponents,\n        ...contextHelpers,\n        ...contextAttachment,\n      }}\n    >\n      {children}\n    </TamboContext.Provider>\n  );\n};\n\n/**\n * The useTambo hook provides access to the Tambo API. This is the primary entrypoint\n * for the Tambo React SDK.\n *\n * This includes the TamboAI client, the component registry, the current thread context,\n * and interactable component management.\n * @returns The Tambo API\n */\nexport const useTambo = () => {\n  return useContext(TamboContext);\n};\n"]}

package/esm/providers/tambo-registry-provider.js

@@ -1,6 +1,6 @@
 "use client";
 import React, { createContext, useCallback, useContext, useEffect, useMemo, useState, } from "react";
-import { ZodSchema } from "zod";
+import { ZodSchema } from "zod/v3";
 import zodToJsonSchema from "zod-to-json-schema";
 import { getMcpServerUniqueKey, MCPTransport, } from "../model/mcp-server-info";
 import { assertValidName } from "../util/validate-component-name";

package/esm/providers/tambo-registry-provider.js.map

@@ -1 +1 @@
-{"version":3,"file":"tambo-registry-provider.js","sourceRoot":"","sources":["../../src/providers/tambo-registry-provider.tsx"],"names":[],"mappings":"AAAA,YAAY,CAAC;AAEb,OAAO,KAAK,EAAE,EACZ,aAAa,EAEb,WAAW,EACX,UAAU,EACV,SAAS,EACT,OAAO,EACP,QAAQ,GACT,MAAM,OAAO,CAAC;AACf,OAAO,EAAE,SAAS,EAAE,MAAM,KAAK,CAAC;AAChC,OAAO,eAAe,MAAM,oBAAoB,CAAC;AAMjD,OAAO,EACL,qBAAqB,EAErB,YAAY,GAEb,MAAM,0BAA0B,CAAC;AAClC,OAAO,EAAE,eAAe,EAAE,MAAM,iCAAiC,CAAC;AAClE,OAAO,EAAE,iBAAiB,EAAE,MAAM,6BAA6B,CAAC;AAEhE;;;;;GAKG;AACH,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,mBAAmB,CAC1B,MAA8B;IAE9B,MAAM,IAAI,GACR,OAAO,MAAM,KAAK,QAAQ;QACxB,CAAC,CAAC;YACE,GAAG,EAAE,MAAM;YACX,SAAS,EAAE,YAAY,CAAC,IAAI;SAC7B;QACH,CAAC,CAAC,MAAM,CAAC;IAEb,MAAM,SAAS,GAAG,IAAI,CAAC,SAAS,IAAI,eAAe,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;IAC9D,MAAM,SAAS,GAAG,IAAI,CAAC,SAAS,IAAI,YAAY,CAAC,IAAI,CAAC;IAEtD,OAAO,EAAE,GAAG,IAAI,EAAE,SAAS,EAAE,SAAS,EAAE,CAAC;AAC3C,CAAC;AAmBD,MAAM,CAAC,MAAM,oBAAoB,GAAG,aAAa,CAAuB;IACtE,aAAa,EAAE,EAAE;IACjB,YAAY,EAAE,EAAE;IAChB,yBAAyB,EAAE,EAAE;IAC7B,cAAc,EAAE,EAAE;IAClB;;OAEG;IACH,iBAAiB,EAAE,GAAG,EAAE,GAAE,CAAC;IAC3B;;OAEG;IACH,YAAY,EAAE,GAAG,EAAE,GAAE,CAAC;IACtB;;OAEG;IACH,aAAa,EAAE,GAAG,EAAE,GAAE,CAAC;IACvB;;OAEG;IACH,kBAAkB,EAAE,GAAG,EAAE,GAAE,CAAC;IAC5B;;OAEG;IACH,iBAAiB,EAAE,GAAG,EAAE,GAAE,CAAC;IAC3B;;OAEG;IACH,kBAAkB,EAAE,GAAG,EAAE,GAAE,CAAC;CAC7B,CAAC,CAAC;AAwBH;;;;;;;;;;GAUG;AACH,MAAM,CAAC,MAAM,qBAAqB,GAE9B,CAAC,EACH,QAAQ,EACR,UAAU,EAAE,cAAc,EAC1B,KAAK,EAAE,SAAS,EAChB,UAAU,EAAE,cAAc,EAC1B,sBAAsB,GACvB,EAAE,EAAE;IACH,MAAM,CAAC,aAAa,EAAE,gBAAgB,CAAC,GAAG,QAAQ,CAAoB,EAAE,CAAC,CAAC;IAC1E,MAAM,CAAC,YAAY,EAAE,eAAe,CAAC,GAAG,QAAQ,CAC9C,EAAE,CACH,CAAC;IACF,MAAM,CAAC,yBAAyB,EAAE,4BAA4B,CAAC,GAAG,QAAQ,CAExE,EAAE,CAAC,CAAC;IACN,MAAM,CAAC,oBAAoB,EAAE,uBAAuB,CAAC,GAAG,QAAQ,CAE9D,EAAE,CAAC,CAAC;IACN,MAAM,CAAC,qBAAqB,EAAE,wBAAwB,CAAC,GAAG,QAAQ,CAEhE,EAAE,CAAC,CAAC;IAEN,MAAM,YAAY,GAAG,WAAW,CAC9B,CAAC,IAAe,EAAE,eAAe,GAAG,IAAI,EAAE,EAAE;QAC1C,qBAAqB;QACrB,eAAe,CAAC,IAAI,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;QAEnC,wBAAwB;QACxB,IAAI,IAAI,CAAC,UAAU,IAAI,WAAW,CAAC,IAAI,CAAC,UAAU,CAAC,EAAE,CAAC;YACpD,iBAAiB,CAAC,IAAI,CAAC,UAAU,EAAE,uBAAuB,IAAI,CAAC,IAAI,GAAG,CAAC,CAAC;QAC1E,CAAC;QACD,eAAe,CAAC,CAAC,IAAI,EAAE,EAAE;YACvB,IAAI,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,eAAe,EAAE,CAAC;gBACvC,OAAO,CAAC,IAAI,CAAC,oBAAoB,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC;YAChD,CAAC;YACD,OAAO;gBACL,GAAG,IAAI;gBACP,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,IAAI;aAClB,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,EACD,EAAE,CACH,CAAC;IAEF,MAAM,aAAa,GAAG,WAAW,CAC/B,CAAC,KAAkB,EAAE,eAAe,GAAG,IAAI,EAAE,EAAE;QAC7C,KAAK,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,YAAY,CAAC,IAAI,EAAE,eAAe,CAAC,CAAC,CAAC;IAC/D,CAAC,EACD,CAAC,YAAY,CAAC,CACf,CAAC;IAEF,MAAM,iBAAiB,GAAG,WAAW,CAAC,CAAC,IAA4B,EAAE,EAAE;QACrE,MAAM,UAAU,GAAG,mBAAmB,CAAC,IAAI,CAAC,CAAC;QAC7C,wBAAwB,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,GAAG,IAAI,EAAE,UAAU,CAAC,CAAC,CAAC;IAC5D,CAAC,EAAE,EAAE,CAAC,CAAC;IAEP,MAAM,kBAAkB,GAAG,WAAW,CACpC,CAAC,KAAiC,EAAE,EAAE;QACpC,MAAM,UAAU,GAAG,KAAK,CAAC,GAAG,CAAC,mBAAmB,CAAC,CAAC;QAClD,wBAAwB,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,GAAG,IAAI,EAAE,GAAG,UAAU,CAAC,CAAC,CAAC;IAC/D,CAAC,EACD,EAAE,CACH,CAAC;IAEF,MAAM,kBAAkB,GAAG,WAAW,CACpC,CAAC,aAAqB,EAAE,IAAe,EAAE,EAAE;QACzC,oCAAoC;QACpC,eAAe,CAAC,aAAa,EAAE,WAAW,CAAC,CAAC;QAC5C,eAAe,CAAC,IAAI,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;QAEnC,IAAI,CAAC,aAAa,CAAC,aAAa,CAAC,EAAE,CAAC;YAClC,MAAM,IAAI,KAAK,CAAC,aAAa,aAAa,wBAAwB,CAAC,CAAC;QACtE,CAAC;QACD,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC;YAC7B,MAAM,IAAI,KAAK,CAAC,QAAQ,IAAI,CAAC,IAAI,wBAAwB,CAAC,CAAC;QAC7D,CAAC;QAED,4BAA4B,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;YACtC,GAAG,IAAI;YACP,CAAC,aAAa,CAAC,EAAE,CAAC,GAAG,CAAC,IAAI,CAAC,aAAa,CAAC,IAAI,EAAE,CAAC,EAAE,IAAI,CAAC,IAAI,CAAC;SAC7D,CAAC,CAAC,CAAC;IACN,CAAC,EACD,CAAC,aAAa,EAAE,YAAY,CAAC,CAC9B,CAAC;IAEF,MAAM,iBAAiB,GAAG,WAAW,CACnC,CAAC,OAAuB,EAAE,eAAe,GAAG,IAAI,EAAE,EAAE;QAClD,MAAM,EACJ,IAAI,EACJ,WAAW,EACX,SAAS,EACT,WAAW,EACX,eAAe,EACf,gBAAgB,EAChB,eAAe,GAChB,GAAG,OAAO,CAAC;QAEZ,0BAA0B;QAC1B,eAAe,CAAC,IAAI,EAAE,WAAW,CAAC,CAAC;QAEnC,0DAA0D;QAC1D,IAAI,CAAC,WAAW,IAAI,CAAC,eAAe,EAAE,CAAC;YACrC,MAAM,IAAI,KAAK,CACb,aAAa,IAAI,wEAAwE,CAC1F,CAAC;QACJ,CAAC;QAED,sDAAsD;QACtD,IAAI,WAAW,IAAI,eAAe,EAAE,CAAC;YACnC,MAAM,IAAI,KAAK,CACb,aAAa,IAAI,0GAA0G,CAC5H,CAAC;QACJ,CAAC;QAED,4DAA4D;QAC5D,IAAI,WAAW,IAAI,WAAW,CAAC,WAAW,CAAC,EAAE,CAAC;YAC5C,iBAAiB,CAAC,WAAW,EAAE,6BAA6B,IAAI,GAAG,CAAC,CAAC;QACvE,CAAC;QAED,kDAAkD;QAClD,MAAM,KAAK,GAAG,kBAAkB,CAAC,eAAe,EAAE,WAAW,EAAE,IAAI,CAAC,CAAC;QAErE,gBAAgB,CAAC,CAAC,IAAI,EAAE,EAAE;YACxB,IAAI,IAAI,CAAC,IAAI,CAAC,IAAI,eAAe,EAAE,CAAC;gBAClC,OAAO,CAAC,IAAI,CAAC,yBAAyB,IAAI,EAAE,CAAC,CAAC;YAChD,CAAC;YACD,OAAO;gBACL,GAAG,IAAI;gBACP,CAAC,IAAI,CAAC,EAAE;oBACN,SAAS;oBACT,gBAAgB;oBAChB,IAAI;oBACJ,WAAW;oBACX,KAAK;oBACL,YAAY,EAAE,EAAE;iBACjB;aACF,CAAC;QACJ,CAAC,CAAC,CAAC;QACH,IAAI,eAAe,EAAE,CAAC;YACpB,aAAa,CAAC,eAAe,CAAC,CAAC;YAC/B,4BAA4B,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;gBACtC,GAAG,IAAI;gBACP,CAAC,IAAI,CAAC,EAAE,eAAe,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC;aACjD,CAAC,CAAC,CAAC;QACN,CAAC;IACH,CAAC,EACD,CAAC,aAAa,CAAC,CAChB,CAAC;IACF,SAAS,CAAC,GAAG,EAAE;QACb,IAAI,cAAc,EAAE,CAAC;YACnB,cAAc,CAAC,OAAO,CAAC,CAAC,SAAS,EAAE,EAAE;gBACnC,iBAAiB,CAAC,SAAS,EAAE,KAAK,CAAC,CAAC;YACtC,CAAC,CAAC,CAAC;QACL,CAAC;IACH,CAAC,EAAE,CAAC,iBAAiB,EAAE,cAAc,CAAC,CAAC,CAAC;IAExC,SAAS,CAAC,GAAG,EAAE;QACb,IAAI,SAAS,EAAE,CAAC;YACd,aAAa,CAAC,SAAS,EAAE,KAAK,CAAC,CAAC;QAClC,CAAC;IACH,CAAC,EAAE,CAAC,aAAa,EAAE,SAAS,CAAC,CAAC,CAAC;IAE/B,SAAS,CAAC,GAAG,EAAE;QACb,IAAI,CAAC,cAAc,IAAI,cAAc,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACnD,uBAAuB,CAAC,EAAE,CAAC,CAAC;YAC5B,OAAO;QACT,CAAC;QAED,2EAA2E;QAC3E,MAAM,UAAU,GAAG,cAAc,CAAC,GAAG,CAAC,mBAAmB,CAAC,CAAC;QAC3D,uBAAuB,CAAC,UAAU,CAAC,CAAC;IACtC,CAAC,EAAE,CAAC,cAAc,CAAC,CAAC,CAAC;IAErB,MAAM,cAAc,GAA8B,OAAO,CAAC,GAAG,EAAE;QAC7D,MAAM,UAAU,GAAG,CAAC,GAAG,oBAAoB,EAAE,GAAG,qBAAqB,CAAC,CAAC;QACvE,IAAI,UAAU,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC5B,OAAO,UAAU,CAAC;QACpB,CAAC;QAED,2DAA2D;QAC3D,MAAM,KAAK,GAAG,IAAI,GAAG,EAAmC,CAAC;QACzD,KAAK,MAAM,MAAM,IAAI,UAAU,EAAE,CAAC;YAChC,MAAM,GAAG,GAAG,qBAAqB,CAAC,MAAM,CAAC,CAAC;YAC1C,KAAK,CAAC,GAAG,CAAC,GAAG,EAAE,MAAM,CAAC,CAAC;QACzB,CAAC;QAED,MAAM,OAAO,GAAG,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,EAAE,CAAC,CAAC;QAE3C,oEAAoE;QACpE,MAAM,IAAI,GAAG,IAAI,GAAG,EAAkB,CAAC;QACvC,OAAO,OAAO,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE;YAC5B,MAAM,OAAO,GAAG,MAAM,CAAC,SAAS,CAAC;YACjC,MAAM,KAAK,GAAG,CAAC,IAAI,CAAC,GAAG,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC;YAC3C,IAAI,CAAC,GAAG,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC;YAEzB,IAAI,KAAK,KAAK,CAAC,EAAE,CAAC;gBAChB,OAAO,MAAM,CAAC;YAChB,CAAC;YAED,OAAO;gBACL,GAAG,MAAM;gBACT,SAAS,EAAE,GAAG,OAAO,IAAI,KAAK,EAAE;aACjC,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,EAAE,CAAC,oBAAoB,EAAE,qBAAqB,CAAC,CAAC,CAAC;IAElD,MAAM,KAAK,GAAG;QACZ,aAAa;QACb,YAAY;QACZ,yBAAyB;QACzB,cAAc;QACd,iBAAiB;QACjB,YAAY;QACZ,aAAa;QACb,kBAAkB;QAClB,iBAAiB;QACjB,kBAAkB;QAClB,sBAAsB;KACvB,CAAC;IAEF,OAAO,CACL,oBAAC,oBAAoB,CAAC,QAAQ,IAAC,KAAK,EAAE,KAAK,IACxC,QAAQ,CACqB,CACjC,CAAC;AACJ,CAAC,CAAC;AAEF;;;;GAIG;AACH,MAAM,CAAC,MAAM,gBAAgB,GAAG,GAAG,EAAE;IACnC,OAAO,UAAU,CAAC,oBAAoB,CAAC,CAAC;AAC1C,CAAC,CAAC;AAEF;;;;;;;;;;;;;;;;;;;;;;;;;;;;GA4BG;AACH,MAAM,CAAC,MAAM,sBAAsB,GAAG,GAA8B,EAAE;IACpE,OAAO,UAAU,CAAC,oBAAoB,CAAC,CAAC,cAAc,CAAC;AACzD,CAAC,CAAC;AACF,SAAS,kBAAkB,CACzB,eAAoB,EACpB,WAAgB,EAChB,IAAY;IAEZ,IAAI,eAAe,EAAE,CAAC;QACpB,OAAO,CAAC,IAAI,CAAC,yDAAyD,CAAC,CAAC;QACxE,OAAO,eAAe,CAAC;IACzB,CAAC;IAED,IAAI,WAAW,CAAC,WAAW,CAAC,EAAE,CAAC;QAC7B,IAAI,CAAC;YACH,OAAO,eAAe,CAAC,WAAW,CAAC,CAAC;QACtC,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CACX,oBAAoB,IAAI,+BAA+B,EACvD,KAAK,CACN,CAAC;QACJ,CAAC;IACH,CAAC;IACD,qFAAqF;IACrF,IAAI,YAAY,CAAC,WAAW,CAAC,EAAE,CAAC;QAC9B,OAAO,WAAW,CAAC;IACrB,CAAC;IAED,MAAM,IAAI,KAAK,CAAC,4BAA4B,IAAI,EAAE,CAAC,CAAC;AACtD,CAAC;AAED;;;;;GAKG;AACH,SAAS,YAAY,CAAC,WAAgB;IACpC,OAAO,CACL,WAAW;QACX,OAAO,WAAW,KAAK,QAAQ;QAC/B,WAAW,CAAC,IAAI,KAAK,QAAQ;QAC7B,WAAW,CAAC,UAAU,CACvB,CAAC;AACJ,CAAC;AAED;;;;GAIG;AACH,SAAS,WAAW,CAAC,GAAY;IAC/B,IAAI,GAAG,YAAY,SAAS,EAAE,CAAC;QAC7B,OAAO,IAAI,CAAC;IACd,CAAC;IACD,6CAA6C;IAC7C,OAAO,CACL,OAAO,GAAG,KAAK,QAAQ;QACvB,GAAG,KAAK,IAAI;QACZ,OAAQ,GAAW,CAAC,SAAS,KAAK,UAAU;QAC5C,OAAQ,GAAW,CAAC,IAAI,KAAK,QAAQ,CACtC,CAAC;AACJ,CAAC","sourcesContent":["\"use client\";\nimport type TamboAI from \"@tambo-ai/typescript-sdk\";\nimport React, {\n  createContext,\n  PropsWithChildren,\n  useCallback,\n  useContext,\n  useEffect,\n  useMemo,\n  useState,\n} from \"react\";\nimport { ZodSchema } from \"zod\";\nimport zodToJsonSchema from \"zod-to-json-schema\";\nimport {\n  ComponentRegistry,\n  TamboComponent,\n  TamboTool,\n} from \"../model/component-metadata\";\nimport {\n  getMcpServerUniqueKey,\n  McpServerInfo,\n  MCPTransport,\n  type NormalizedMcpServerInfo,\n} from \"../model/mcp-server-info\";\nimport { assertValidName } from \"../util/validate-component-name\";\nimport { assertNoZodRecord } from \"../util/validate-zod-schema\";\n\n/**\n * Derives a short, meaningful key from a server URL.\n * Strips TLDs and common prefixes to get a human-readable identifier.\n * For example, \"https://mcp.linear.app/mcp\" becomes \"linear\".\n * @returns A lowercased, human-readable key derived from the URL\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 * Normalizes an MCP server info object, ensuring it has a serverKey.\n * If serverKey is not provided, derives it from the URL.\n * @returns The normalized MCP server info object\n */\nfunction normalizeServerInfo(\n  server: McpServerInfo | string,\n): NormalizedMcpServerInfo {\n  const base: McpServerInfo =\n    typeof server === \"string\"\n      ? {\n          url: server,\n          transport: MCPTransport.HTTP,\n        }\n      : server;\n\n  const serverKey = base.serverKey ?? deriveServerKey(base.url);\n  const transport = base.transport ?? MCPTransport.HTTP;\n\n  return { ...base, transport, serverKey };\n}\n\nexport interface TamboRegistryContext {\n  componentList: ComponentRegistry;\n  toolRegistry: Record<string, TamboTool>;\n  componentToolAssociations: Record<string, string[]>;\n  mcpServerInfos: NormalizedMcpServerInfo[];\n  registerComponent: (options: TamboComponent) => void;\n  registerTool: (tool: TamboTool) => void;\n  registerTools: (tools: TamboTool[]) => void;\n  addToolAssociation: (componentName: string, tool: TamboTool) => void;\n  registerMcpServer: (info: McpServerInfo) => void;\n  registerMcpServers: (infos: McpServerInfo[]) => void;\n  onCallUnregisteredTool?: (\n    toolName: string,\n    args: TamboAI.ToolCallParameter[],\n  ) => Promise<string>;\n}\n\nexport const TamboRegistryContext = createContext<TamboRegistryContext>({\n  componentList: {},\n  toolRegistry: {},\n  componentToolAssociations: {},\n  mcpServerInfos: [],\n  /**\n   *\n   */\n  registerComponent: () => {},\n  /**\n   *\n   */\n  registerTool: () => {},\n  /**\n   *\n   */\n  registerTools: () => {},\n  /**\n   *\n   */\n  addToolAssociation: () => {},\n  /**\n   *\n   */\n  registerMcpServer: () => {},\n  /**\n   *\n   */\n  registerMcpServers: () => {},\n});\n\nexport interface TamboRegistryProviderProps {\n  /** The components to register */\n  components?: TamboComponent[];\n  /** The tools to register */\n  tools?: TamboTool[];\n  /** The MCP servers to register */\n  mcpServers?: (McpServerInfo | string)[];\n\n  /**\n   * A function to call when an unknown tool is called. If this function is not\n   * provided, an error will be thrown when a tool call is requested by the\n   * server.\n   *\n   * Note that this is generally only for agents, where the agent code may\n   * request tool calls that are not registered in the tool registry.\n   */\n  onCallUnregisteredTool?: (\n    toolName: string,\n    args: TamboAI.ToolCallParameter[],\n  ) => Promise<string>;\n}\n\n/**\n * The TamboRegistryProvider is a React provider that provides a component\n * registry to the descendants of the provider.\n * @param props - The props for the TamboRegistryProvider\n * @param props.children - The children to wrap\n * @param props.components - The components to register\n * @param props.tools - The tools to register\n * @param props.mcpServers - The MCP servers to register\n * @param props.onCallUnregisteredTool - The function to call when an unknown tool is called (optional)\n * @returns The TamboRegistryProvider component\n */\nexport const TamboRegistryProvider: React.FC<\n  PropsWithChildren<TamboRegistryProviderProps>\n> = ({\n  children,\n  components: userComponents,\n  tools: userTools,\n  mcpServers: userMcpServers,\n  onCallUnregisteredTool,\n}) => {\n  const [componentList, setComponentList] = useState<ComponentRegistry>({});\n  const [toolRegistry, setToolRegistry] = useState<Record<string, TamboTool>>(\n    {},\n  );\n  const [componentToolAssociations, setComponentToolAssociations] = useState<\n    Record<string, string[]>\n  >({});\n  const [staticMcpServerInfos, setStaticMcpServerInfos] = useState<\n    NormalizedMcpServerInfo[]\n  >([]);\n  const [dynamicMcpServerInfos, setDynamicMcpServerInfos] = useState<\n    NormalizedMcpServerInfo[]\n  >([]);\n\n  const registerTool = useCallback(\n    (tool: TamboTool, warnOnOverwrite = true) => {\n      // Validate tool name\n      assertValidName(tool.name, \"tool\");\n\n      // Validate tool schemas\n      if (tool.toolSchema && isZodSchema(tool.toolSchema)) {\n        assertNoZodRecord(tool.toolSchema, `toolSchema of tool \"${tool.name}\"`);\n      }\n      setToolRegistry((prev) => {\n        if (prev[tool.name] && warnOnOverwrite) {\n          console.warn(`Overwriting tool ${tool.name}`);\n        }\n        return {\n          ...prev,\n          [tool.name]: tool,\n        };\n      });\n    },\n    [],\n  );\n\n  const registerTools = useCallback(\n    (tools: TamboTool[], warnOnOverwrite = true) => {\n      tools.forEach((tool) => registerTool(tool, warnOnOverwrite));\n    },\n    [registerTool],\n  );\n\n  const registerMcpServer = useCallback((info: McpServerInfo | string) => {\n    const normalized = normalizeServerInfo(info);\n    setDynamicMcpServerInfos((prev) => [...prev, normalized]);\n  }, []);\n\n  const registerMcpServers = useCallback(\n    (infos: (McpServerInfo | string)[]) => {\n      const normalized = infos.map(normalizeServerInfo);\n      setDynamicMcpServerInfos((prev) => [...prev, ...normalized]);\n    },\n    [],\n  );\n\n  const addToolAssociation = useCallback(\n    (componentName: string, tool: TamboTool) => {\n      // Validate component and tool names\n      assertValidName(componentName, \"component\");\n      assertValidName(tool.name, \"tool\");\n\n      if (!componentList[componentName]) {\n        throw new Error(`Component ${componentName} not found in registry`);\n      }\n      if (!toolRegistry[tool.name]) {\n        throw new Error(`Tool ${tool.name} not found in registry`);\n      }\n\n      setComponentToolAssociations((prev) => ({\n        ...prev,\n        [componentName]: [...(prev[componentName] || []), tool.name],\n      }));\n    },\n    [componentList, toolRegistry],\n  );\n\n  const registerComponent = useCallback(\n    (options: TamboComponent, warnOnOverwrite = true) => {\n      const {\n        name,\n        description,\n        component,\n        propsSchema,\n        propsDefinition,\n        loadingComponent,\n        associatedTools,\n      } = options;\n\n      // Validate component name\n      assertValidName(name, \"component\");\n\n      // Validate that at least one props definition is provided\n      if (!propsSchema && !propsDefinition) {\n        throw new Error(\n          `Component ${name} must have either propsSchema (recommended) or propsDefinition defined`,\n        );\n      }\n\n      // Validate that only one props definition is provided\n      if (propsSchema && propsDefinition) {\n        throw new Error(\n          `Component ${name} cannot have both propsSchema and propsDefinition defined. Use only one. We recommend using propsSchema.`,\n        );\n      }\n\n      // Validate that the propsSchema does not include z.record()\n      if (propsSchema && isZodSchema(propsSchema)) {\n        assertNoZodRecord(propsSchema, `propsSchema of component \"${name}\"`);\n      }\n\n      // Convert propsSchema to JSON Schema if it exists\n      const props = getSerializedProps(propsDefinition, propsSchema, name);\n\n      setComponentList((prev) => {\n        if (prev[name] && warnOnOverwrite) {\n          console.warn(`overwriting component ${name}`);\n        }\n        return {\n          ...prev,\n          [name]: {\n            component,\n            loadingComponent,\n            name,\n            description,\n            props,\n            contextTools: [],\n          },\n        };\n      });\n      if (associatedTools) {\n        registerTools(associatedTools);\n        setComponentToolAssociations((prev) => ({\n          ...prev,\n          [name]: associatedTools.map((tool) => tool.name),\n        }));\n      }\n    },\n    [registerTools],\n  );\n  useEffect(() => {\n    if (userComponents) {\n      userComponents.forEach((component) => {\n        registerComponent(component, false);\n      });\n    }\n  }, [registerComponent, userComponents]);\n\n  useEffect(() => {\n    if (userTools) {\n      registerTools(userTools, false);\n    }\n  }, [registerTools, userTools]);\n\n  useEffect(() => {\n    if (!userMcpServers || userMcpServers.length === 0) {\n      setStaticMcpServerInfos([]);\n      return;\n    }\n\n    // Normalize servers from props and ensure all have serverKey and transport\n    const normalized = userMcpServers.map(normalizeServerInfo);\n    setStaticMcpServerInfos(normalized);\n  }, [userMcpServers]);\n\n  const mcpServerInfos: NormalizedMcpServerInfo[] = useMemo(() => {\n    const allServers = [...staticMcpServerInfos, ...dynamicMcpServerInfos];\n    if (allServers.length === 0) {\n      return allServers;\n    }\n\n    // 1. Deduplicate by connection identity using a stable key\n    const byKey = new Map<string, NormalizedMcpServerInfo>();\n    for (const server of allServers) {\n      const key = getMcpServerUniqueKey(server);\n      byKey.set(key, server);\n    }\n\n    const deduped = Array.from(byKey.values());\n\n    // 2. Ensure serverKey uniqueness for readable, unambiguous prefixes\n    const seen = new Map<string, number>();\n    return deduped.map((server) => {\n      const baseKey = server.serverKey;\n      const count = (seen.get(baseKey) ?? 0) + 1;\n      seen.set(baseKey, count);\n\n      if (count === 1) {\n        return server;\n      }\n\n      return {\n        ...server,\n        serverKey: `${baseKey}-${count}`,\n      };\n    });\n  }, [staticMcpServerInfos, dynamicMcpServerInfos]);\n\n  const value = {\n    componentList,\n    toolRegistry,\n    componentToolAssociations,\n    mcpServerInfos,\n    registerComponent,\n    registerTool,\n    registerTools,\n    addToolAssociation,\n    registerMcpServer,\n    registerMcpServers,\n    onCallUnregisteredTool,\n  };\n\n  return (\n    <TamboRegistryContext.Provider value={value}>\n      {children}\n    </TamboRegistryContext.Provider>\n  );\n};\n\n/**\n * The useTamboRegistry hook provides access to the component registry\n * to the descendants of the TamboRegistryProvider.\n * @returns The component registry\n */\nexport const useTamboRegistry = () => {\n  return useContext(TamboRegistryContext);\n};\n\n/**\n * Hook to access the MCP server metadata from TamboRegistryProvider.\n * This provides access to the registered MCP server configurations (metadata only, not connections).\n *\n * This hook can be used anywhere within the TamboProvider hierarchy to access\n * the list of configured MCP servers without needing to be inside TamboMcpProvider.\n * @returns Array of MCP server metadata\n * @example\n * ```tsx\n * function MyComponent() {\n *   const mcpServers = useTamboMcpServerInfos();\n *\n *   return (\n *     <div>\n *       <h3>Configured MCP Servers:</h3>\n *       {mcpServers.map((server) => (\n *         <div key={server.url}>\n *           {server.name || server.url}\n *         </div>\n *       ))}\n *     </div>\n *   );\n * }\n * ```\n *\n * The returned objects are `NormalizedMcpServerInfo` instances, meaning both\n * `serverKey` and `transport` are always populated (with `transport`\n * defaulting to HTTP when not explicitly specified).\n */\nexport const useTamboMcpServerInfos = (): NormalizedMcpServerInfo[] => {\n  return useContext(TamboRegistryContext).mcpServerInfos;\n};\nfunction getSerializedProps(\n  propsDefinition: any,\n  propsSchema: any,\n  name: string,\n) {\n  if (propsDefinition) {\n    console.warn(`propsDefinition is deprecated. Use propsSchema instead.`);\n    return propsDefinition;\n  }\n\n  if (isZodSchema(propsSchema)) {\n    try {\n      return zodToJsonSchema(propsSchema);\n    } catch (error) {\n      console.error(\n        `Error converting ${name} props schema to JSON Schema:`,\n        error,\n      );\n    }\n  }\n  // try to roughly detect JSONSchema, should always be an object with a properties key\n  if (isJSONSchema(propsSchema)) {\n    return propsSchema;\n  }\n\n  throw new Error(`Invalid props schema for ${name}`);\n}\n\n/**\n * Checks if the propsSchema is a JSON Schema. This is a rough check, and the\n * server will provide the definitive check.\n * @param propsSchema - The props schema to check\n * @returns True if the props schema is a JSON Schema, false otherwise\n */\nfunction isJSONSchema(propsSchema: any) {\n  return (\n    propsSchema &&\n    typeof propsSchema === \"object\" &&\n    propsSchema.type === \"object\" &&\n    propsSchema.properties\n  );\n}\n\n/**\n * Since we require a certain zod version, we need to check if the object is a ZodSchema\n * @param obj - The object to check\n * @returns True if the object is a ZodSchema, false otherwise\n */\nfunction isZodSchema(obj: unknown): obj is ZodSchema {\n  if (obj instanceof ZodSchema) {\n    return true;\n  }\n  // try to detect if the object is a ZodSchema\n  return (\n    typeof obj === \"object\" &&\n    obj !== null &&\n    typeof (obj as any).safeParse === \"function\" &&\n    typeof (obj as any)._def === \"object\"\n  );\n}\n"]}
+{"version":3,"file":"tambo-registry-provider.js","sourceRoot":"","sources":["../../src/providers/tambo-registry-provider.tsx"],"names":[],"mappings":"AAAA,YAAY,CAAC;AAEb,OAAO,KAAK,EAAE,EACZ,aAAa,EAEb,WAAW,EACX,UAAU,EACV,SAAS,EACT,OAAO,EACP,QAAQ,GACT,MAAM,OAAO,CAAC;AACf,OAAO,EAAE,SAAS,EAAE,MAAM,QAAQ,CAAC;AACnC,OAAO,eAAe,MAAM,oBAAoB,CAAC;AAMjD,OAAO,EACL,qBAAqB,EAErB,YAAY,GAEb,MAAM,0BAA0B,CAAC;AAClC,OAAO,EAAE,eAAe,EAAE,MAAM,iCAAiC,CAAC;AAClE,OAAO,EAAE,iBAAiB,EAAE,MAAM,6BAA6B,CAAC;AAEhE;;;;;GAKG;AACH,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,mBAAmB,CAC1B,MAA8B;IAE9B,MAAM,IAAI,GACR,OAAO,MAAM,KAAK,QAAQ;QACxB,CAAC,CAAC;YACE,GAAG,EAAE,MAAM;YACX,SAAS,EAAE,YAAY,CAAC,IAAI;SAC7B;QACH,CAAC,CAAC,MAAM,CAAC;IAEb,MAAM,SAAS,GAAG,IAAI,CAAC,SAAS,IAAI,eAAe,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;IAC9D,MAAM,SAAS,GAAG,IAAI,CAAC,SAAS,IAAI,YAAY,CAAC,IAAI,CAAC;IAEtD,OAAO,EAAE,GAAG,IAAI,EAAE,SAAS,EAAE,SAAS,EAAE,CAAC;AAC3C,CAAC;AAmBD,MAAM,CAAC,MAAM,oBAAoB,GAAG,aAAa,CAAuB;IACtE,aAAa,EAAE,EAAE;IACjB,YAAY,EAAE,EAAE;IAChB,yBAAyB,EAAE,EAAE;IAC7B,cAAc,EAAE,EAAE;IAClB;;OAEG;IACH,iBAAiB,EAAE,GAAG,EAAE,GAAE,CAAC;IAC3B;;OAEG;IACH,YAAY,EAAE,GAAG,EAAE,GAAE,CAAC;IACtB;;OAEG;IACH,aAAa,EAAE,GAAG,EAAE,GAAE,CAAC;IACvB;;OAEG;IACH,kBAAkB,EAAE,GAAG,EAAE,GAAE,CAAC;IAC5B;;OAEG;IACH,iBAAiB,EAAE,GAAG,EAAE,GAAE,CAAC;IAC3B;;OAEG;IACH,kBAAkB,EAAE,GAAG,EAAE,GAAE,CAAC;CAC7B,CAAC,CAAC;AAwBH;;;;;;;;;;GAUG;AACH,MAAM,CAAC,MAAM,qBAAqB,GAE9B,CAAC,EACH,QAAQ,EACR,UAAU,EAAE,cAAc,EAC1B,KAAK,EAAE,SAAS,EAChB,UAAU,EAAE,cAAc,EAC1B,sBAAsB,GACvB,EAAE,EAAE;IACH,MAAM,CAAC,aAAa,EAAE,gBAAgB,CAAC,GAAG,QAAQ,CAAoB,EAAE,CAAC,CAAC;IAC1E,MAAM,CAAC,YAAY,EAAE,eAAe,CAAC,GAAG,QAAQ,CAC9C,EAAE,CACH,CAAC;IACF,MAAM,CAAC,yBAAyB,EAAE,4BAA4B,CAAC,GAAG,QAAQ,CAExE,EAAE,CAAC,CAAC;IACN,MAAM,CAAC,oBAAoB,EAAE,uBAAuB,CAAC,GAAG,QAAQ,CAE9D,EAAE,CAAC,CAAC;IACN,MAAM,CAAC,qBAAqB,EAAE,wBAAwB,CAAC,GAAG,QAAQ,CAEhE,EAAE,CAAC,CAAC;IAEN,MAAM,YAAY,GAAG,WAAW,CAC9B,CAAC,IAAe,EAAE,eAAe,GAAG,IAAI,EAAE,EAAE;QAC1C,qBAAqB;QACrB,eAAe,CAAC,IAAI,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;QAEnC,wBAAwB;QACxB,IAAI,IAAI,CAAC,UAAU,IAAI,WAAW,CAAC,IAAI,CAAC,UAAU,CAAC,EAAE,CAAC;YACpD,iBAAiB,CAAC,IAAI,CAAC,UAAU,EAAE,uBAAuB,IAAI,CAAC,IAAI,GAAG,CAAC,CAAC;QAC1E,CAAC;QACD,eAAe,CAAC,CAAC,IAAI,EAAE,EAAE;YACvB,IAAI,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,eAAe,EAAE,CAAC;gBACvC,OAAO,CAAC,IAAI,CAAC,oBAAoB,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC;YAChD,CAAC;YACD,OAAO;gBACL,GAAG,IAAI;gBACP,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,IAAI;aAClB,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,EACD,EAAE,CACH,CAAC;IAEF,MAAM,aAAa,GAAG,WAAW,CAC/B,CAAC,KAAkB,EAAE,eAAe,GAAG,IAAI,EAAE,EAAE;QAC7C,KAAK,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,YAAY,CAAC,IAAI,EAAE,eAAe,CAAC,CAAC,CAAC;IAC/D,CAAC,EACD,CAAC,YAAY,CAAC,CACf,CAAC;IAEF,MAAM,iBAAiB,GAAG,WAAW,CAAC,CAAC,IAA4B,EAAE,EAAE;QACrE,MAAM,UAAU,GAAG,mBAAmB,CAAC,IAAI,CAAC,CAAC;QAC7C,wBAAwB,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,GAAG,IAAI,EAAE,UAAU,CAAC,CAAC,CAAC;IAC5D,CAAC,EAAE,EAAE,CAAC,CAAC;IAEP,MAAM,kBAAkB,GAAG,WAAW,CACpC,CAAC,KAAiC,EAAE,EAAE;QACpC,MAAM,UAAU,GAAG,KAAK,CAAC,GAAG,CAAC,mBAAmB,CAAC,CAAC;QAClD,wBAAwB,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,GAAG,IAAI,EAAE,GAAG,UAAU,CAAC,CAAC,CAAC;IAC/D,CAAC,EACD,EAAE,CACH,CAAC;IAEF,MAAM,kBAAkB,GAAG,WAAW,CACpC,CAAC,aAAqB,EAAE,IAAe,EAAE,EAAE;QACzC,oCAAoC;QACpC,eAAe,CAAC,aAAa,EAAE,WAAW,CAAC,CAAC;QAC5C,eAAe,CAAC,IAAI,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;QAEnC,IAAI,CAAC,aAAa,CAAC,aAAa,CAAC,EAAE,CAAC;YAClC,MAAM,IAAI,KAAK,CAAC,aAAa,aAAa,wBAAwB,CAAC,CAAC;QACtE,CAAC;QACD,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC;YAC7B,MAAM,IAAI,KAAK,CAAC,QAAQ,IAAI,CAAC,IAAI,wBAAwB,CAAC,CAAC;QAC7D,CAAC;QAED,4BAA4B,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;YACtC,GAAG,IAAI;YACP,CAAC,aAAa,CAAC,EAAE,CAAC,GAAG,CAAC,IAAI,CAAC,aAAa,CAAC,IAAI,EAAE,CAAC,EAAE,IAAI,CAAC,IAAI,CAAC;SAC7D,CAAC,CAAC,CAAC;IACN,CAAC,EACD,CAAC,aAAa,EAAE,YAAY,CAAC,CAC9B,CAAC;IAEF,MAAM,iBAAiB,GAAG,WAAW,CACnC,CAAC,OAAuB,EAAE,eAAe,GAAG,IAAI,EAAE,EAAE;QAClD,MAAM,EACJ,IAAI,EACJ,WAAW,EACX,SAAS,EACT,WAAW,EACX,eAAe,EACf,gBAAgB,EAChB,eAAe,GAChB,GAAG,OAAO,CAAC;QAEZ,0BAA0B;QAC1B,eAAe,CAAC,IAAI,EAAE,WAAW,CAAC,CAAC;QAEnC,0DAA0D;QAC1D,IAAI,CAAC,WAAW,IAAI,CAAC,eAAe,EAAE,CAAC;YACrC,MAAM,IAAI,KAAK,CACb,aAAa,IAAI,wEAAwE,CAC1F,CAAC;QACJ,CAAC;QAED,sDAAsD;QACtD,IAAI,WAAW,IAAI,eAAe,EAAE,CAAC;YACnC,MAAM,IAAI,KAAK,CACb,aAAa,IAAI,0GAA0G,CAC5H,CAAC;QACJ,CAAC;QAED,4DAA4D;QAC5D,IAAI,WAAW,IAAI,WAAW,CAAC,WAAW,CAAC,EAAE,CAAC;YAC5C,iBAAiB,CAAC,WAAW,EAAE,6BAA6B,IAAI,GAAG,CAAC,CAAC;QACvE,CAAC;QAED,kDAAkD;QAClD,MAAM,KAAK,GAAG,kBAAkB,CAAC,eAAe,EAAE,WAAW,EAAE,IAAI,CAAC,CAAC;QAErE,gBAAgB,CAAC,CAAC,IAAI,EAAE,EAAE;YACxB,IAAI,IAAI,CAAC,IAAI,CAAC,IAAI,eAAe,EAAE,CAAC;gBAClC,OAAO,CAAC,IAAI,CAAC,yBAAyB,IAAI,EAAE,CAAC,CAAC;YAChD,CAAC;YACD,OAAO;gBACL,GAAG,IAAI;gBACP,CAAC,IAAI,CAAC,EAAE;oBACN,SAAS;oBACT,gBAAgB;oBAChB,IAAI;oBACJ,WAAW;oBACX,KAAK;oBACL,YAAY,EAAE,EAAE;iBACjB;aACF,CAAC;QACJ,CAAC,CAAC,CAAC;QACH,IAAI,eAAe,EAAE,CAAC;YACpB,aAAa,CAAC,eAAe,CAAC,CAAC;YAC/B,4BAA4B,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;gBACtC,GAAG,IAAI;gBACP,CAAC,IAAI,CAAC,EAAE,eAAe,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC;aACjD,CAAC,CAAC,CAAC;QACN,CAAC;IACH,CAAC,EACD,CAAC,aAAa,CAAC,CAChB,CAAC;IACF,SAAS,CAAC,GAAG,EAAE;QACb,IAAI,cAAc,EAAE,CAAC;YACnB,cAAc,CAAC,OAAO,CAAC,CAAC,SAAS,EAAE,EAAE;gBACnC,iBAAiB,CAAC,SAAS,EAAE,KAAK,CAAC,CAAC;YACtC,CAAC,CAAC,CAAC;QACL,CAAC;IACH,CAAC,EAAE,CAAC,iBAAiB,EAAE,cAAc,CAAC,CAAC,CAAC;IAExC,SAAS,CAAC,GAAG,EAAE;QACb,IAAI,SAAS,EAAE,CAAC;YACd,aAAa,CAAC,SAAS,EAAE,KAAK,CAAC,CAAC;QAClC,CAAC;IACH,CAAC,EAAE,CAAC,aAAa,EAAE,SAAS,CAAC,CAAC,CAAC;IAE/B,SAAS,CAAC,GAAG,EAAE;QACb,IAAI,CAAC,cAAc,IAAI,cAAc,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACnD,uBAAuB,CAAC,EAAE,CAAC,CAAC;YAC5B,OAAO;QACT,CAAC;QAED,2EAA2E;QAC3E,MAAM,UAAU,GAAG,cAAc,CAAC,GAAG,CAAC,mBAAmB,CAAC,CAAC;QAC3D,uBAAuB,CAAC,UAAU,CAAC,CAAC;IACtC,CAAC,EAAE,CAAC,cAAc,CAAC,CAAC,CAAC;IAErB,MAAM,cAAc,GAA8B,OAAO,CAAC,GAAG,EAAE;QAC7D,MAAM,UAAU,GAAG,CAAC,GAAG,oBAAoB,EAAE,GAAG,qBAAqB,CAAC,CAAC;QACvE,IAAI,UAAU,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC5B,OAAO,UAAU,CAAC;QACpB,CAAC;QAED,2DAA2D;QAC3D,MAAM,KAAK,GAAG,IAAI,GAAG,EAAmC,CAAC;QACzD,KAAK,MAAM,MAAM,IAAI,UAAU,EAAE,CAAC;YAChC,MAAM,GAAG,GAAG,qBAAqB,CAAC,MAAM,CAAC,CAAC;YAC1C,KAAK,CAAC,GAAG,CAAC,GAAG,EAAE,MAAM,CAAC,CAAC;QACzB,CAAC;QAED,MAAM,OAAO,GAAG,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,EAAE,CAAC,CAAC;QAE3C,oEAAoE;QACpE,MAAM,IAAI,GAAG,IAAI,GAAG,EAAkB,CAAC;QACvC,OAAO,OAAO,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE;YAC5B,MAAM,OAAO,GAAG,MAAM,CAAC,SAAS,CAAC;YACjC,MAAM,KAAK,GAAG,CAAC,IAAI,CAAC,GAAG,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC;YAC3C,IAAI,CAAC,GAAG,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC;YAEzB,IAAI,KAAK,KAAK,CAAC,EAAE,CAAC;gBAChB,OAAO,MAAM,CAAC;YAChB,CAAC;YAED,OAAO;gBACL,GAAG,MAAM;gBACT,SAAS,EAAE,GAAG,OAAO,IAAI,KAAK,EAAE;aACjC,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,EAAE,CAAC,oBAAoB,EAAE,qBAAqB,CAAC,CAAC,CAAC;IAElD,MAAM,KAAK,GAAG;QACZ,aAAa;QACb,YAAY;QACZ,yBAAyB;QACzB,cAAc;QACd,iBAAiB;QACjB,YAAY;QACZ,aAAa;QACb,kBAAkB;QAClB,iBAAiB;QACjB,kBAAkB;QAClB,sBAAsB;KACvB,CAAC;IAEF,OAAO,CACL,oBAAC,oBAAoB,CAAC,QAAQ,IAAC,KAAK,EAAE,KAAK,IACxC,QAAQ,CACqB,CACjC,CAAC;AACJ,CAAC,CAAC;AAEF;;;;GAIG;AACH,MAAM,CAAC,MAAM,gBAAgB,GAAG,GAAG,EAAE;IACnC,OAAO,UAAU,CAAC,oBAAoB,CAAC,CAAC;AAC1C,CAAC,CAAC;AAEF;;;;;;;;;;;;;;;;;;;;;;;;;;;;GA4BG;AACH,MAAM,CAAC,MAAM,sBAAsB,GAAG,GAA8B,EAAE;IACpE,OAAO,UAAU,CAAC,oBAAoB,CAAC,CAAC,cAAc,CAAC;AACzD,CAAC,CAAC;AACF,SAAS,kBAAkB,CACzB,eAAoB,EACpB,WAAgB,EAChB,IAAY;IAEZ,IAAI,eAAe,EAAE,CAAC;QACpB,OAAO,CAAC,IAAI,CAAC,yDAAyD,CAAC,CAAC;QACxE,OAAO,eAAe,CAAC;IACzB,CAAC;IAED,IAAI,WAAW,CAAC,WAAW,CAAC,EAAE,CAAC;QAC7B,IAAI,CAAC;YACH,OAAO,eAAe,CAAC,WAAW,CAAC,CAAC;QACtC,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CACX,oBAAoB,IAAI,+BAA+B,EACvD,KAAK,CACN,CAAC;QACJ,CAAC;IACH,CAAC;IACD,qFAAqF;IACrF,IAAI,YAAY,CAAC,WAAW,CAAC,EAAE,CAAC;QAC9B,OAAO,WAAW,CAAC;IACrB,CAAC;IAED,MAAM,IAAI,KAAK,CAAC,4BAA4B,IAAI,EAAE,CAAC,CAAC;AACtD,CAAC;AAED;;;;;GAKG;AACH,SAAS,YAAY,CAAC,WAAgB;IACpC,OAAO,CACL,WAAW;QACX,OAAO,WAAW,KAAK,QAAQ;QAC/B,WAAW,CAAC,IAAI,KAAK,QAAQ;QAC7B,WAAW,CAAC,UAAU,CACvB,CAAC;AACJ,CAAC;AAED;;;;GAIG;AACH,SAAS,WAAW,CAAC,GAAY;IAC/B,IAAI,GAAG,YAAY,SAAS,EAAE,CAAC;QAC7B,OAAO,IAAI,CAAC;IACd,CAAC;IACD,6CAA6C;IAC7C,OAAO,CACL,OAAO,GAAG,KAAK,QAAQ;QACvB,GAAG,KAAK,IAAI;QACZ,OAAQ,GAAW,CAAC,SAAS,KAAK,UAAU;QAC5C,OAAQ,GAAW,CAAC,IAAI,KAAK,QAAQ,CACtC,CAAC;AACJ,CAAC","sourcesContent":["\"use client\";\nimport type TamboAI from \"@tambo-ai/typescript-sdk\";\nimport React, {\n  createContext,\n  PropsWithChildren,\n  useCallback,\n  useContext,\n  useEffect,\n  useMemo,\n  useState,\n} from \"react\";\nimport { ZodSchema } from \"zod/v3\";\nimport zodToJsonSchema from \"zod-to-json-schema\";\nimport {\n  ComponentRegistry,\n  TamboComponent,\n  TamboTool,\n} from \"../model/component-metadata\";\nimport {\n  getMcpServerUniqueKey,\n  McpServerInfo,\n  MCPTransport,\n  type NormalizedMcpServerInfo,\n} from \"../model/mcp-server-info\";\nimport { assertValidName } from \"../util/validate-component-name\";\nimport { assertNoZodRecord } from \"../util/validate-zod-schema\";\n\n/**\n * Derives a short, meaningful key from a server URL.\n * Strips TLDs and common prefixes to get a human-readable identifier.\n * For example, \"https://mcp.linear.app/mcp\" becomes \"linear\".\n * @returns A lowercased, human-readable key derived from the URL\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 * Normalizes an MCP server info object, ensuring it has a serverKey.\n * If serverKey is not provided, derives it from the URL.\n * @returns The normalized MCP server info object\n */\nfunction normalizeServerInfo(\n  server: McpServerInfo | string,\n): NormalizedMcpServerInfo {\n  const base: McpServerInfo =\n    typeof server === \"string\"\n      ? {\n          url: server,\n          transport: MCPTransport.HTTP,\n        }\n      : server;\n\n  const serverKey = base.serverKey ?? deriveServerKey(base.url);\n  const transport = base.transport ?? MCPTransport.HTTP;\n\n  return { ...base, transport, serverKey };\n}\n\nexport interface TamboRegistryContext {\n  componentList: ComponentRegistry;\n  toolRegistry: Record<string, TamboTool>;\n  componentToolAssociations: Record<string, string[]>;\n  mcpServerInfos: NormalizedMcpServerInfo[];\n  registerComponent: (options: TamboComponent) => void;\n  registerTool: (tool: TamboTool) => void;\n  registerTools: (tools: TamboTool[]) => void;\n  addToolAssociation: (componentName: string, tool: TamboTool) => void;\n  registerMcpServer: (info: McpServerInfo) => void;\n  registerMcpServers: (infos: McpServerInfo[]) => void;\n  onCallUnregisteredTool?: (\n    toolName: string,\n    args: TamboAI.ToolCallParameter[],\n  ) => Promise<string>;\n}\n\nexport const TamboRegistryContext = createContext<TamboRegistryContext>({\n  componentList: {},\n  toolRegistry: {},\n  componentToolAssociations: {},\n  mcpServerInfos: [],\n  /**\n   *\n   */\n  registerComponent: () => {},\n  /**\n   *\n   */\n  registerTool: () => {},\n  /**\n   *\n   */\n  registerTools: () => {},\n  /**\n   *\n   */\n  addToolAssociation: () => {},\n  /**\n   *\n   */\n  registerMcpServer: () => {},\n  /**\n   *\n   */\n  registerMcpServers: () => {},\n});\n\nexport interface TamboRegistryProviderProps {\n  /** The components to register */\n  components?: TamboComponent[];\n  /** The tools to register */\n  tools?: TamboTool[];\n  /** The MCP servers to register */\n  mcpServers?: (McpServerInfo | string)[];\n\n  /**\n   * A function to call when an unknown tool is called. If this function is not\n   * provided, an error will be thrown when a tool call is requested by the\n   * server.\n   *\n   * Note that this is generally only for agents, where the agent code may\n   * request tool calls that are not registered in the tool registry.\n   */\n  onCallUnregisteredTool?: (\n    toolName: string,\n    args: TamboAI.ToolCallParameter[],\n  ) => Promise<string>;\n}\n\n/**\n * The TamboRegistryProvider is a React provider that provides a component\n * registry to the descendants of the provider.\n * @param props - The props for the TamboRegistryProvider\n * @param props.children - The children to wrap\n * @param props.components - The components to register\n * @param props.tools - The tools to register\n * @param props.mcpServers - The MCP servers to register\n * @param props.onCallUnregisteredTool - The function to call when an unknown tool is called (optional)\n * @returns The TamboRegistryProvider component\n */\nexport const TamboRegistryProvider: React.FC<\n  PropsWithChildren<TamboRegistryProviderProps>\n> = ({\n  children,\n  components: userComponents,\n  tools: userTools,\n  mcpServers: userMcpServers,\n  onCallUnregisteredTool,\n}) => {\n  const [componentList, setComponentList] = useState<ComponentRegistry>({});\n  const [toolRegistry, setToolRegistry] = useState<Record<string, TamboTool>>(\n    {},\n  );\n  const [componentToolAssociations, setComponentToolAssociations] = useState<\n    Record<string, string[]>\n  >({});\n  const [staticMcpServerInfos, setStaticMcpServerInfos] = useState<\n    NormalizedMcpServerInfo[]\n  >([]);\n  const [dynamicMcpServerInfos, setDynamicMcpServerInfos] = useState<\n    NormalizedMcpServerInfo[]\n  >([]);\n\n  const registerTool = useCallback(\n    (tool: TamboTool, warnOnOverwrite = true) => {\n      // Validate tool name\n      assertValidName(tool.name, \"tool\");\n\n      // Validate tool schemas\n      if (tool.toolSchema && isZodSchema(tool.toolSchema)) {\n        assertNoZodRecord(tool.toolSchema, `toolSchema of tool \"${tool.name}\"`);\n      }\n      setToolRegistry((prev) => {\n        if (prev[tool.name] && warnOnOverwrite) {\n          console.warn(`Overwriting tool ${tool.name}`);\n        }\n        return {\n          ...prev,\n          [tool.name]: tool,\n        };\n      });\n    },\n    [],\n  );\n\n  const registerTools = useCallback(\n    (tools: TamboTool[], warnOnOverwrite = true) => {\n      tools.forEach((tool) => registerTool(tool, warnOnOverwrite));\n    },\n    [registerTool],\n  );\n\n  const registerMcpServer = useCallback((info: McpServerInfo | string) => {\n    const normalized = normalizeServerInfo(info);\n    setDynamicMcpServerInfos((prev) => [...prev, normalized]);\n  }, []);\n\n  const registerMcpServers = useCallback(\n    (infos: (McpServerInfo | string)[]) => {\n      const normalized = infos.map(normalizeServerInfo);\n      setDynamicMcpServerInfos((prev) => [...prev, ...normalized]);\n    },\n    [],\n  );\n\n  const addToolAssociation = useCallback(\n    (componentName: string, tool: TamboTool) => {\n      // Validate component and tool names\n      assertValidName(componentName, \"component\");\n      assertValidName(tool.name, \"tool\");\n\n      if (!componentList[componentName]) {\n        throw new Error(`Component ${componentName} not found in registry`);\n      }\n      if (!toolRegistry[tool.name]) {\n        throw new Error(`Tool ${tool.name} not found in registry`);\n      }\n\n      setComponentToolAssociations((prev) => ({\n        ...prev,\n        [componentName]: [...(prev[componentName] || []), tool.name],\n      }));\n    },\n    [componentList, toolRegistry],\n  );\n\n  const registerComponent = useCallback(\n    (options: TamboComponent, warnOnOverwrite = true) => {\n      const {\n        name,\n        description,\n        component,\n        propsSchema,\n        propsDefinition,\n        loadingComponent,\n        associatedTools,\n      } = options;\n\n      // Validate component name\n      assertValidName(name, \"component\");\n\n      // Validate that at least one props definition is provided\n      if (!propsSchema && !propsDefinition) {\n        throw new Error(\n          `Component ${name} must have either propsSchema (recommended) or propsDefinition defined`,\n        );\n      }\n\n      // Validate that only one props definition is provided\n      if (propsSchema && propsDefinition) {\n        throw new Error(\n          `Component ${name} cannot have both propsSchema and propsDefinition defined. Use only one. We recommend using propsSchema.`,\n        );\n      }\n\n      // Validate that the propsSchema does not include z.record()\n      if (propsSchema && isZodSchema(propsSchema)) {\n        assertNoZodRecord(propsSchema, `propsSchema of component \"${name}\"`);\n      }\n\n      // Convert propsSchema to JSON Schema if it exists\n      const props = getSerializedProps(propsDefinition, propsSchema, name);\n\n      setComponentList((prev) => {\n        if (prev[name] && warnOnOverwrite) {\n          console.warn(`overwriting component ${name}`);\n        }\n        return {\n          ...prev,\n          [name]: {\n            component,\n            loadingComponent,\n            name,\n            description,\n            props,\n            contextTools: [],\n          },\n        };\n      });\n      if (associatedTools) {\n        registerTools(associatedTools);\n        setComponentToolAssociations((prev) => ({\n          ...prev,\n          [name]: associatedTools.map((tool) => tool.name),\n        }));\n      }\n    },\n    [registerTools],\n  );\n  useEffect(() => {\n    if (userComponents) {\n      userComponents.forEach((component) => {\n        registerComponent(component, false);\n      });\n    }\n  }, [registerComponent, userComponents]);\n\n  useEffect(() => {\n    if (userTools) {\n      registerTools(userTools, false);\n    }\n  }, [registerTools, userTools]);\n\n  useEffect(() => {\n    if (!userMcpServers || userMcpServers.length === 0) {\n      setStaticMcpServerInfos([]);\n      return;\n    }\n\n    // Normalize servers from props and ensure all have serverKey and transport\n    const normalized = userMcpServers.map(normalizeServerInfo);\n    setStaticMcpServerInfos(normalized);\n  }, [userMcpServers]);\n\n  const mcpServerInfos: NormalizedMcpServerInfo[] = useMemo(() => {\n    const allServers = [...staticMcpServerInfos, ...dynamicMcpServerInfos];\n    if (allServers.length === 0) {\n      return allServers;\n    }\n\n    // 1. Deduplicate by connection identity using a stable key\n    const byKey = new Map<string, NormalizedMcpServerInfo>();\n    for (const server of allServers) {\n      const key = getMcpServerUniqueKey(server);\n      byKey.set(key, server);\n    }\n\n    const deduped = Array.from(byKey.values());\n\n    // 2. Ensure serverKey uniqueness for readable, unambiguous prefixes\n    const seen = new Map<string, number>();\n    return deduped.map((server) => {\n      const baseKey = server.serverKey;\n      const count = (seen.get(baseKey) ?? 0) + 1;\n      seen.set(baseKey, count);\n\n      if (count === 1) {\n        return server;\n      }\n\n      return {\n        ...server,\n        serverKey: `${baseKey}-${count}`,\n      };\n    });\n  }, [staticMcpServerInfos, dynamicMcpServerInfos]);\n\n  const value = {\n    componentList,\n    toolRegistry,\n    componentToolAssociations,\n    mcpServerInfos,\n    registerComponent,\n    registerTool,\n    registerTools,\n    addToolAssociation,\n    registerMcpServer,\n    registerMcpServers,\n    onCallUnregisteredTool,\n  };\n\n  return (\n    <TamboRegistryContext.Provider value={value}>\n      {children}\n    </TamboRegistryContext.Provider>\n  );\n};\n\n/**\n * The useTamboRegistry hook provides access to the component registry\n * to the descendants of the TamboRegistryProvider.\n * @returns The component registry\n */\nexport const useTamboRegistry = () => {\n  return useContext(TamboRegistryContext);\n};\n\n/**\n * Hook to access the MCP server metadata from TamboRegistryProvider.\n * This provides access to the registered MCP server configurations (metadata only, not connections).\n *\n * This hook can be used anywhere within the TamboProvider hierarchy to access\n * the list of configured MCP servers without needing to be inside TamboMcpProvider.\n * @returns Array of MCP server metadata\n * @example\n * ```tsx\n * function MyComponent() {\n *   const mcpServers = useTamboMcpServerInfos();\n *\n *   return (\n *     <div>\n *       <h3>Configured MCP Servers:</h3>\n *       {mcpServers.map((server) => (\n *         <div key={server.url}>\n *           {server.name || server.url}\n *         </div>\n *       ))}\n *     </div>\n *   );\n * }\n * ```\n *\n * The returned objects are `NormalizedMcpServerInfo` instances, meaning both\n * `serverKey` and `transport` are always populated (with `transport`\n * defaulting to HTTP when not explicitly specified).\n */\nexport const useTamboMcpServerInfos = (): NormalizedMcpServerInfo[] => {\n  return useContext(TamboRegistryContext).mcpServerInfos;\n};\nfunction getSerializedProps(\n  propsDefinition: any,\n  propsSchema: any,\n  name: string,\n) {\n  if (propsDefinition) {\n    console.warn(`propsDefinition is deprecated. Use propsSchema instead.`);\n    return propsDefinition;\n  }\n\n  if (isZodSchema(propsSchema)) {\n    try {\n      return zodToJsonSchema(propsSchema);\n    } catch (error) {\n      console.error(\n        `Error converting ${name} props schema to JSON Schema:`,\n        error,\n      );\n    }\n  }\n  // try to roughly detect JSONSchema, should always be an object with a properties key\n  if (isJSONSchema(propsSchema)) {\n    return propsSchema;\n  }\n\n  throw new Error(`Invalid props schema for ${name}`);\n}\n\n/**\n * Checks if the propsSchema is a JSON Schema. This is a rough check, and the\n * server will provide the definitive check.\n * @param propsSchema - The props schema to check\n * @returns True if the props schema is a JSON Schema, false otherwise\n */\nfunction isJSONSchema(propsSchema: any) {\n  return (\n    propsSchema &&\n    typeof propsSchema === \"object\" &&\n    propsSchema.type === \"object\" &&\n    propsSchema.properties\n  );\n}\n\n/**\n * Since we require a certain zod version, we need to check if the object is a ZodSchema\n * @param obj - The object to check\n * @returns True if the object is a ZodSchema, false otherwise\n */\nfunction isZodSchema(obj: unknown): obj is ZodSchema {\n  if (obj instanceof ZodSchema) {\n    return true;\n  }\n  // try to detect if the object is a ZodSchema\n  return (\n    typeof obj === \"object\" &&\n    obj !== null &&\n    typeof (obj as any).safeParse === \"function\" &&\n    typeof (obj as any)._def === \"object\"\n  );\n}\n"]}

package/esm/providers/tambo-registry-provider.test.d.ts.map

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

package/esm/providers/{__tests__/tambo-registry-provider.test.js → tambo-registry-provider.test.js}

@@ -1,7 +1,7 @@
 import { act, renderHook } from "@testing-library/react";
 import React from "react";
-import { z } from "zod";
-import { TamboRegistryProvider, useTamboRegistry, } from "../tambo-registry-provider";
+import { z } from "zod/v3";
+import { TamboRegistryProvider, useTamboRegistry, } from "./tambo-registry-provider";
 // Shared tool registry for all tests
 const createMockTools = () => [
     {