@tambo-ai/react 0.70.0 → 0.72.0

package/dist/v1/providers/tambo-v1-provider.js

@@ -0,0 +1,131 @@
+"use strict";
+"use client";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __importStar = (this && this.__importStar) || (function () {
+    var ownKeys = function(o) {
+        ownKeys = Object.getOwnPropertyNames || function (o) {
+            var ar = [];
+            for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
+            return ar;
+        };
+        return ownKeys(o);
+    };
+    return function (mod) {
+        if (mod && mod.__esModule) return mod;
+        var result = {};
+        if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
+        __setModuleDefault(result, mod);
+        return result;
+    };
+})();
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.useContextKey = useContextKey;
+exports.TamboV1Provider = TamboV1Provider;
+/**
+ * TamboV1Provider - Main Provider for v1 API
+ *
+ * Composes the necessary providers for the v1 SDK:
+ * - TamboClientProvider: API client and authentication
+ * - TamboRegistryProvider: Component and tool registration
+ * - TamboContextHelpersProvider: Context helper functions
+ * - TamboV1StreamProvider: Streaming state management
+ *
+ * This provider should wrap your entire application or the portion
+ * that needs access to Tambo v1 functionality.
+ */
+const react_1 = __importStar(require("react"));
+const react_query_1 = require("@tanstack/react-query");
+const tambo_client_provider_1 = require("../../providers/tambo-client-provider");
+const tambo_registry_provider_1 = require("../../providers/tambo-registry-provider");
+const tambo_context_helpers_provider_1 = require("../../providers/tambo-context-helpers-provider");
+const tambo_v1_stream_context_1 = require("./tambo-v1-stream-context");
+/**
+ * Context for providing contextKey to child components.
+ */
+const ContextKeyContext = (0, react_1.createContext)(undefined);
+/**
+ * Hook to access the contextKey from TamboV1Provider.
+ * @returns The contextKey if provided, undefined otherwise
+ */
+function useContextKey() {
+    return (0, react_1.useContext)(ContextKeyContext);
+}
+/**
+ * Main provider for the Tambo v1 SDK.
+ *
+ * Composes TamboClientProvider, TamboRegistryProvider, and TamboV1StreamProvider
+ * to provide a complete context for building AI-powered applications.
+ *
+ * Threads are managed dynamically through useTamboV1() hook functions:
+ * - startNewThread() - Begin a new conversation
+ * - switchThread(threadId) - Switch to an existing thread
+ * - initThread(threadId) - Initialize a thread for receiving events
+ * @param props - Provider configuration
+ * @param props.apiKey - Tambo API key for authentication
+ * @param props.tamboUrl - Optional custom Tambo API URL
+ * @param props.environment - Optional environment configuration
+ * @param props.userToken - Optional OAuth token for user authentication
+ * @param props.components - Components to register with the AI
+ * @param props.tools - Tools to register for client-side execution
+ * @param props.mcpServers - MCP servers to register for additional tools/resources
+ * @param props.onCallUnregisteredTool - Callback for handling unknown tool calls
+ * @param props.resources - Static resources to register with the AI
+ * @param props.listResources - Dynamic resource search function (must be paired with getResource)
+ * @param props.getResource - Dynamic resource fetch function (must be paired with listResources)
+ * @param props.contextHelpers - Configuration for context helper functions
+ * @param props.contextKey - Optional context key for thread scoping/isolation
+ * @param props.queryClient - Optional custom React Query client
+ * @param props.children - Child components
+ * @returns Provider component tree
+ * @example
+ * ```tsx
+ * import { TamboV1Provider } from '@tambo-ai/react/v1';
+ *
+ * function App() {
+ *   return (
+ *     <TamboV1Provider
+ *       apiKey={process.env.NEXT_PUBLIC_TAMBO_API_KEY!}
+ *       components={[WeatherCard, StockChart]}
+ *       tools={[searchTool, calculatorTool]}
+ *     >
+ *       <ChatInterface />
+ *     </TamboV1Provider>
+ *   );
+ * }
+ * ```
+ */
+function TamboV1Provider({ apiKey, tamboUrl, environment, userToken, components, tools, mcpServers, onCallUnregisteredTool, resources, listResources, getResource, contextHelpers, contextKey, queryClient, children, }) {
+    // Create a stable default QueryClient if none provided.
+    // Using useState to avoid SSR issues with module-level singletons.
+    const [defaultQueryClient] = (0, react_1.useState)(() => new react_query_1.QueryClient({
+        defaultOptions: {
+            queries: {
+                staleTime: 1000,
+                retry: 1,
+            },
+        },
+    }));
+    const client = queryClient ?? defaultQueryClient;
+    return (react_1.default.createElement(react_query_1.QueryClientProvider, { client: client },
+        react_1.default.createElement(tambo_client_provider_1.TamboClientProvider, { apiKey: apiKey, tamboUrl: tamboUrl, environment: environment, userToken: userToken },
+            react_1.default.createElement(tambo_registry_provider_1.TamboRegistryProvider, { components: components, tools: tools, mcpServers: mcpServers, onCallUnregisteredTool: onCallUnregisteredTool, resources: resources, listResources: listResources, getResource: getResource },
+                react_1.default.createElement(tambo_context_helpers_provider_1.TamboContextHelpersProvider, { contextHelpers: contextHelpers },
+                    react_1.default.createElement(ContextKeyContext.Provider, { value: contextKey },
+                        react_1.default.createElement(tambo_v1_stream_context_1.TamboV1StreamProvider, null, children)))))));
+}
+//# sourceMappingURL=tambo-v1-provider.js.map

package/dist/v1/providers/tambo-v1-provider.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"tambo-v1-provider.js","sourceRoot":"","sources":["../../../src/v1/providers/tambo-v1-provider.tsx"],"names":[],"mappings":";AAAA,YAAY,CAAC;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAgDb,sCAEC;AA2HD,0CA2DC;AAtOD;;;;;;;;;;;GAWG;AAEH,+CAKe;AACf,uDAAyE;AACzE,iFAG+C;AAC/C,qFAGiD;AACjD,mGAA6F;AAO7F,uEAAkE;AAElE;;GAEG;AACH,MAAM,iBAAiB,GAAG,IAAA,qBAAa,EAAqB,SAAS,CAAC,CAAC;AAEvE;;;GAGG;AACH,SAAgB,aAAa;IAC3B,OAAO,IAAA,kBAAU,EAAC,iBAAiB,CAAC,CAAC;AACvC,CAAC;AA+ED;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GA2CG;AACH,SAAgB,eAAe,CAAC,EAC9B,MAAM,EACN,QAAQ,EACR,WAAW,EACX,SAAS,EACT,UAAU,EACV,KAAK,EACL,UAAU,EACV,sBAAsB,EACtB,SAAS,EACT,aAAa,EACb,WAAW,EACX,cAAc,EACd,UAAU,EACV,WAAW,EACX,QAAQ,GACgC;IACxC,wDAAwD;IACxD,mEAAmE;IACnE,MAAM,CAAC,kBAAkB,CAAC,GAAG,IAAA,gBAAQ,EACnC,GAAG,EAAE,CACH,IAAI,yBAAW,CAAC;QACd,cAAc,EAAE;YACd,OAAO,EAAE;gBACP,SAAS,EAAE,IAAI;gBACf,KAAK,EAAE,CAAC;aACT;SACF;KACF,CAAC,CACL,CAAC;IAEF,MAAM,MAAM,GAAG,WAAW,IAAI,kBAAkB,CAAC;IAEjD,OAAO,CACL,8BAAC,iCAAmB,IAAC,MAAM,EAAE,MAAM;QACjC,8BAAC,2CAAmB,IAClB,MAAM,EAAE,MAAM,EACd,QAAQ,EAAE,QAAQ,EAClB,WAAW,EAAE,WAAW,EACxB,SAAS,EAAE,SAAS;YAEpB,8BAAC,+CAAqB,IACpB,UAAU,EAAE,UAAU,EACtB,KAAK,EAAE,KAAK,EACZ,UAAU,EAAE,UAAU,EACtB,sBAAsB,EAAE,sBAAsB,EAC9C,SAAS,EAAE,SAAS,EACpB,aAAa,EAAE,aAAa,EAC5B,WAAW,EAAE,WAAW;gBAExB,8BAAC,4DAA2B,IAAC,cAAc,EAAE,cAAc;oBACzD,8BAAC,iBAAiB,CAAC,QAAQ,IAAC,KAAK,EAAE,UAAU;wBAC3C,8BAAC,+CAAqB,QAAE,QAAQ,CAAyB,CAC9B,CACD,CACR,CACJ,CACF,CACvB,CAAC;AACJ,CAAC","sourcesContent":["\"use client\";\n\n/**\n * TamboV1Provider - Main Provider for v1 API\n *\n * Composes the necessary providers for the v1 SDK:\n * - TamboClientProvider: API client and authentication\n * - TamboRegistryProvider: Component and tool registration\n * - TamboContextHelpersProvider: Context helper functions\n * - TamboV1StreamProvider: Streaming state management\n *\n * This provider should wrap your entire application or the portion\n * that needs access to Tambo v1 functionality.\n */\n\nimport React, {\n  createContext,\n  useContext,\n  type PropsWithChildren,\n  useState,\n} from \"react\";\nimport { QueryClient, QueryClientProvider } from \"@tanstack/react-query\";\nimport {\n  TamboClientProvider,\n  type TamboClientProviderProps,\n} from \"../../providers/tambo-client-provider\";\nimport {\n  TamboRegistryProvider,\n  type TamboRegistryProviderProps,\n} from \"../../providers/tambo-registry-provider\";\nimport { TamboContextHelpersProvider } from \"../../providers/tambo-context-helpers-provider\";\nimport type { ContextHelpers } from \"../../context-helpers\";\nimport type { McpServerInfo } from \"../../model/mcp-server-info\";\nimport type {\n  ListResourceItem,\n  ResourceSource,\n} from \"../../model/resource-info\";\nimport { TamboV1StreamProvider } from \"./tambo-v1-stream-context\";\n\n/**\n * Context for providing contextKey to child components.\n */\nconst ContextKeyContext = createContext<string | undefined>(undefined);\n\n/**\n * Hook to access the contextKey from TamboV1Provider.\n * @returns The contextKey if provided, undefined otherwise\n */\nexport function useContextKey(): string | undefined {\n  return useContext(ContextKeyContext);\n}\n\n/**\n * Props for TamboV1Provider\n */\nexport interface TamboV1ProviderProps extends Pick<\n  TamboClientProviderProps,\n  \"apiKey\" | \"tamboUrl\" | \"environment\" | \"userToken\"\n> {\n  /**\n   * Components to register with the registry.\n   * These will be available for the AI to use in responses.\n   */\n  components?: TamboRegistryProviderProps[\"components\"];\n\n  /**\n   * Tools to register with the registry.\n   * These will be executed client-side when requested by the AI.\n   */\n  tools?: TamboRegistryProviderProps[\"tools\"];\n\n  /**\n   * MCP servers to register with the registry.\n   * These provide additional tools and resources from MCP-compatible servers.\n   */\n  mcpServers?: (McpServerInfo | string)[];\n\n  /**\n   * Callback function called when an unregistered tool is called.\n   * If not provided, an error will be thrown for unknown tools.\n   */\n  onCallUnregisteredTool?: TamboRegistryProviderProps[\"onCallUnregisteredTool\"];\n\n  /**\n   * Static resources to register with the registry.\n   * These will be available for the AI to access.\n   */\n  resources?: ListResourceItem[];\n\n  /**\n   * Dynamic resource search function.\n   * Must be paired with getResource.\n   * Called when searching for resources dynamically.\n   */\n  listResources?: ResourceSource[\"listResources\"];\n\n  /**\n   * Dynamic resource fetch function.\n   * Must be paired with listResources.\n   * Called when fetching a specific resource by URI.\n   */\n  getResource?: ResourceSource[\"getResource\"];\n\n  /**\n   * Configuration for context helpers.\n   * A dictionary of functions that provide additional context to the AI.\n   * Each key becomes the context name, and the function returns the value.\n   */\n  contextHelpers?: ContextHelpers;\n\n  /**\n   * Optional context key for thread scoping/isolation.\n   * Threads created with the same contextKey are grouped together.\n   * Useful for multi-tenant applications or separating conversation contexts.\n   */\n  contextKey?: string;\n\n  /**\n   * Optional custom QueryClient instance.\n   * If not provided, a default client will be created.\n   */\n  queryClient?: QueryClient;\n\n  /**\n   * Children components\n   */\n  children: React.ReactNode;\n}\n\n/**\n * Main provider for the Tambo v1 SDK.\n *\n * Composes TamboClientProvider, TamboRegistryProvider, and TamboV1StreamProvider\n * to provide a complete context for building AI-powered applications.\n *\n * Threads are managed dynamically through useTamboV1() hook functions:\n * - startNewThread() - Begin a new conversation\n * - switchThread(threadId) - Switch to an existing thread\n * - initThread(threadId) - Initialize a thread for receiving events\n * @param props - Provider configuration\n * @param props.apiKey - Tambo API key for authentication\n * @param props.tamboUrl - Optional custom Tambo API URL\n * @param props.environment - Optional environment configuration\n * @param props.userToken - Optional OAuth token for user authentication\n * @param props.components - Components to register with the AI\n * @param props.tools - Tools to register for client-side execution\n * @param props.mcpServers - MCP servers to register for additional tools/resources\n * @param props.onCallUnregisteredTool - Callback for handling unknown tool calls\n * @param props.resources - Static resources to register with the AI\n * @param props.listResources - Dynamic resource search function (must be paired with getResource)\n * @param props.getResource - Dynamic resource fetch function (must be paired with listResources)\n * @param props.contextHelpers - Configuration for context helper functions\n * @param props.contextKey - Optional context key for thread scoping/isolation\n * @param props.queryClient - Optional custom React Query client\n * @param props.children - Child components\n * @returns Provider component tree\n * @example\n * ```tsx\n * import { TamboV1Provider } from '@tambo-ai/react/v1';\n *\n * function App() {\n *   return (\n *     <TamboV1Provider\n *       apiKey={process.env.NEXT_PUBLIC_TAMBO_API_KEY!}\n *       components={[WeatherCard, StockChart]}\n *       tools={[searchTool, calculatorTool]}\n *     >\n *       <ChatInterface />\n *     </TamboV1Provider>\n *   );\n * }\n * ```\n */\nexport function TamboV1Provider({\n  apiKey,\n  tamboUrl,\n  environment,\n  userToken,\n  components,\n  tools,\n  mcpServers,\n  onCallUnregisteredTool,\n  resources,\n  listResources,\n  getResource,\n  contextHelpers,\n  contextKey,\n  queryClient,\n  children,\n}: PropsWithChildren<TamboV1ProviderProps>) {\n  // Create a stable default QueryClient if none provided.\n  // Using useState to avoid SSR issues with module-level singletons.\n  const [defaultQueryClient] = useState(\n    () =>\n      new QueryClient({\n        defaultOptions: {\n          queries: {\n            staleTime: 1000,\n            retry: 1,\n          },\n        },\n      }),\n  );\n\n  const client = queryClient ?? defaultQueryClient;\n\n  return (\n    <QueryClientProvider client={client}>\n      <TamboClientProvider\n        apiKey={apiKey}\n        tamboUrl={tamboUrl}\n        environment={environment}\n        userToken={userToken}\n      >\n        <TamboRegistryProvider\n          components={components}\n          tools={tools}\n          mcpServers={mcpServers}\n          onCallUnregisteredTool={onCallUnregisteredTool}\n          resources={resources}\n          listResources={listResources}\n          getResource={getResource}\n        >\n          <TamboContextHelpersProvider contextHelpers={contextHelpers}>\n            <ContextKeyContext.Provider value={contextKey}>\n              <TamboV1StreamProvider>{children}</TamboV1StreamProvider>\n            </ContextKeyContext.Provider>\n          </TamboContextHelpersProvider>\n        </TamboRegistryProvider>\n      </TamboClientProvider>\n    </QueryClientProvider>\n  );\n}\n"]}

package/dist/v1/providers/tambo-v1-provider.test.d.ts

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

package/dist/v1/providers/tambo-v1-provider.test.d.ts.map

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

package/dist/v1/providers/tambo-v1-provider.test.js

@@ -0,0 +1,181 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+const react_query_1 = require("@tanstack/react-query");
+const react_1 = require("@testing-library/react");
+const react_2 = __importDefault(require("react"));
+const zod_1 = require("zod");
+const tambo_client_provider_1 = require("../../providers/tambo-client-provider");
+const tambo_registry_provider_1 = require("../../providers/tambo-registry-provider");
+const tambo_context_helpers_provider_1 = require("../../providers/tambo-context-helpers-provider");
+const tambo_v1_stream_context_1 = require("./tambo-v1-stream-context");
+const tambo_v1_provider_1 = require("./tambo-v1-provider");
+// Mock the client provider to capture the apiKey
+jest.mock("../../providers/tambo-client-provider", () => ({
+    useTamboClient: jest.fn(),
+    TamboClientProvider: ({ children }) => children,
+}));
+describe("TamboV1Provider", () => {
+    const mockClient = {
+        apiKey: "test-api-key",
+        threads: {},
+    };
+    beforeEach(() => {
+        jest.mocked(tambo_client_provider_1.useTamboClient).mockReturnValue(mockClient);
+    });
+    it("provides access to registry context", () => {
+        const wrapper = ({ children }) => (react_2.default.createElement(tambo_v1_provider_1.TamboV1Provider, { apiKey: "test-api-key" }, children));
+        const { result } = (0, react_1.renderHook)(() => (0, tambo_registry_provider_1.useTamboRegistry)(), { wrapper });
+        expect(result.current.componentList).toBeDefined();
+        expect(result.current.toolRegistry).toBeDefined();
+        expect(typeof result.current.registerComponent).toBe("function");
+        expect(typeof result.current.registerTool).toBe("function");
+    });
+    it("provides access to stream context", () => {
+        const wrapper = ({ children }) => (react_2.default.createElement(tambo_v1_provider_1.TamboV1Provider, { apiKey: "test-api-key" }, children));
+        const { result } = (0, react_1.renderHook)(() => (0, tambo_v1_stream_context_1.useStreamState)(), { wrapper });
+        expect(result.current.threadMap).toBeDefined();
+        expect(result.current.currentThreadId).toBeNull();
+    });
+    it("manages threads via useThreadManagement", () => {
+        const wrapper = ({ children }) => (react_2.default.createElement(tambo_v1_provider_1.TamboV1Provider, { apiKey: "test-api-key" }, children));
+        const { result } = (0, react_1.renderHook)(() => ({
+            state: (0, tambo_v1_stream_context_1.useStreamState)(),
+            management: (0, tambo_v1_stream_context_1.useThreadManagement)(),
+        }), { wrapper });
+        // Initially no thread
+        expect(result.current.state.currentThreadId).toBeNull();
+        // Initialize and switch to a thread
+        (0, react_1.act)(() => {
+            result.current.management.initThread("thread_123");
+            result.current.management.switchThread("thread_123");
+        });
+        expect(result.current.state.currentThreadId).toBe("thread_123");
+        expect(result.current.state.threadMap.thread_123).toBeDefined();
+    });
+    it("provides access to query client", () => {
+        const wrapper = ({ children }) => (react_2.default.createElement(tambo_v1_provider_1.TamboV1Provider, { apiKey: "test-api-key" }, children));
+        const { result } = (0, react_1.renderHook)(() => (0, react_query_1.useQueryClient)(), { wrapper });
+        expect(result.current).toBeInstanceOf(react_query_1.QueryClient);
+    });
+    it("uses custom query client when provided", () => {
+        const customClient = new react_query_1.QueryClient();
+        const wrapper = ({ children }) => (react_2.default.createElement(tambo_v1_provider_1.TamboV1Provider, { apiKey: "test-api-key", queryClient: customClient }, children));
+        const { result } = (0, react_1.renderHook)(() => (0, react_query_1.useQueryClient)(), { wrapper });
+        expect(result.current).toBe(customClient);
+    });
+    it("registers components when provided", () => {
+        const TestComponent = () => react_2.default.createElement("div", null, "Test");
+        const propsSchema = zod_1.z.object({
+            title: zod_1.z.string().describe("The title"),
+        });
+        const components = [
+            {
+                name: "TestComponent",
+                description: "A test component",
+                component: TestComponent,
+                propsSchema,
+            },
+        ];
+        const wrapper = ({ children }) => (react_2.default.createElement(tambo_v1_provider_1.TamboV1Provider, { apiKey: "test-api-key", components: components }, children));
+        const { result } = (0, react_1.renderHook)(() => (0, tambo_registry_provider_1.useTamboRegistry)(), { wrapper });
+        expect(result.current.componentList.TestComponent).toBeDefined();
+        expect(result.current.componentList.TestComponent.name).toBe("TestComponent");
+    });
+    it("registers tools when provided", () => {
+        const inputSchema = zod_1.z.object({
+            query: zod_1.z.string().describe("Search query"),
+        });
+        const outputSchema = zod_1.z.string().describe("Result string");
+        const tools = [
+            {
+                name: "testTool",
+                description: "A test tool",
+                tool: async () => "result",
+                inputSchema,
+                outputSchema,
+            },
+        ];
+        const wrapper = ({ children }) => (react_2.default.createElement(tambo_v1_provider_1.TamboV1Provider, { apiKey: "test-api-key", tools: tools }, children));
+        const { result } = (0, react_1.renderHook)(() => (0, tambo_registry_provider_1.useTamboRegistry)(), { wrapper });
+        expect(result.current.toolRegistry.testTool).toBeDefined();
+        expect(result.current.toolRegistry.testTool.name).toBe("testTool");
+    });
+    it("registers MCP servers when provided", () => {
+        const mcpServers = [
+            { url: "https://mcp.example.com", name: "Example MCP" },
+        ];
+        const wrapper = ({ children }) => (react_2.default.createElement(tambo_v1_provider_1.TamboV1Provider, { apiKey: "test-api-key", mcpServers: mcpServers }, children));
+        const { result } = (0, react_1.renderHook)(() => (0, tambo_registry_provider_1.useTamboRegistry)(), { wrapper });
+        expect(result.current.mcpServerInfos).toHaveLength(1);
+        expect(result.current.mcpServerInfos[0].url).toBe("https://mcp.example.com");
+    });
+    it("provides onCallUnregisteredTool to registry", () => {
+        const onCallUnregisteredTool = jest
+            .fn()
+            .mockResolvedValue("fallback result");
+        const wrapper = ({ children }) => (react_2.default.createElement(tambo_v1_provider_1.TamboV1Provider, { apiKey: "test-api-key", onCallUnregisteredTool: onCallUnregisteredTool }, children));
+        const { result } = (0, react_1.renderHook)(() => (0, tambo_registry_provider_1.useTamboRegistry)(), { wrapper });
+        expect(result.current.onCallUnregisteredTool).toBe(onCallUnregisteredTool);
+    });
+    it("registers static resources when provided", () => {
+        const resources = [
+            {
+                uri: "resource://test/example",
+                name: "Test Resource",
+                description: "A test resource",
+                mimeType: "text/plain",
+            },
+        ];
+        const wrapper = ({ children }) => (react_2.default.createElement(tambo_v1_provider_1.TamboV1Provider, { apiKey: "test-api-key", resources: resources }, children));
+        const { result } = (0, react_1.renderHook)(() => (0, tambo_registry_provider_1.useTamboRegistry)(), { wrapper });
+        expect(result.current.resources).toHaveLength(1);
+        expect(result.current.resources[0].uri).toBe("resource://test/example");
+        expect(result.current.resources[0].name).toBe("Test Resource");
+    });
+    it("registers resource source when listResources and getResource provided", () => {
+        const listResources = jest.fn().mockResolvedValue({ resources: [] });
+        const getResource = jest.fn().mockResolvedValue({ contents: [] });
+        const wrapper = ({ children }) => (react_2.default.createElement(tambo_v1_provider_1.TamboV1Provider, { apiKey: "test-api-key", listResources: listResources, getResource: getResource }, children));
+        const { result } = (0, react_1.renderHook)(() => (0, tambo_registry_provider_1.useTamboRegistry)(), { wrapper });
+        expect(result.current.resourceSource).toBeDefined();
+        expect(result.current.resourceSource?.listResources).toBe(listResources);
+        expect(result.current.resourceSource?.getResource).toBe(getResource);
+    });
+    it("provides contextKey via useContextKey hook", () => {
+        const wrapper = ({ children }) => (react_2.default.createElement(tambo_v1_provider_1.TamboV1Provider, { apiKey: "test-api-key", contextKey: "my-context-key" }, children));
+        const { result } = (0, react_1.renderHook)(() => (0, tambo_v1_provider_1.useContextKey)(), { wrapper });
+        expect(result.current).toBe("my-context-key");
+    });
+    it("returns undefined from useContextKey when no contextKey provided", () => {
+        const wrapper = ({ children }) => (react_2.default.createElement(tambo_v1_provider_1.TamboV1Provider, { apiKey: "test-api-key" }, children));
+        const { result } = (0, react_1.renderHook)(() => (0, tambo_v1_provider_1.useContextKey)(), { wrapper });
+        expect(result.current).toBeUndefined();
+    });
+    it("provides context helpers via useTamboContextHelpers hook", async () => {
+        const contextHelpers = {
+            getUserName: () => "Test User",
+            getCurrentTime: () => new Date().toISOString(),
+        };
+        const wrapper = ({ children }) => (react_2.default.createElement(tambo_v1_provider_1.TamboV1Provider, { apiKey: "test-api-key", contextHelpers: contextHelpers }, children));
+        const { result, rerender } = (0, react_1.renderHook)(() => (0, tambo_context_helpers_provider_1.useTamboContextHelpers)(), {
+            wrapper,
+        });
+        // Helpers are registered via useEffect, so we need to trigger a rerender
+        await (0, react_1.act)(async () => {
+            rerender();
+        });
+        const helpers = result.current.getContextHelpers();
+        expect(helpers.getUserName).toBe(contextHelpers.getUserName);
+        expect(helpers.getCurrentTime).toBe(contextHelpers.getCurrentTime);
+    });
+    it("returns empty contextHelpers when none provided", async () => {
+        const wrapper = ({ children }) => (react_2.default.createElement(tambo_v1_provider_1.TamboV1Provider, { apiKey: "test-api-key" }, children));
+        const { result } = (0, react_1.renderHook)(() => (0, tambo_context_helpers_provider_1.useTamboContextHelpers)(), { wrapper });
+        const helpers = result.current.getContextHelpers();
+        expect(Object.keys(helpers)).toHaveLength(0);
+    });
+});
+//# sourceMappingURL=tambo-v1-provider.test.js.map

package/dist/v1/providers/tambo-v1-provider.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"tambo-v1-provider.test.js","sourceRoot":"","sources":["../../../src/v1/providers/tambo-v1-provider.test.tsx"],"names":[],"mappings":";;;;;AACA,uDAAoE;AACpE,kDAAyD;AACzD,kDAA0B;AAC1B,6BAAwB;AACxB,iFAAuE;AACvE,qFAA2E;AAC3E,mGAAwF;AACxF,uEAAgF;AAChF,2DAAqE;AAErE,iDAAiD;AACjD,IAAI,CAAC,IAAI,CAAC,uCAAuC,EAAE,GAAG,EAAE,CAAC,CAAC;IACxD,cAAc,EAAE,IAAI,CAAC,EAAE,EAAE;IACzB,mBAAmB,EAAE,CAAC,EAAE,QAAQ,EAAiC,EAAE,EAAE,CACnE,QAAQ;CACX,CAAC,CAAC,CAAC;AAEJ,QAAQ,CAAC,iBAAiB,EAAE,GAAG,EAAE;IAC/B,MAAM,UAAU,GAAG;QACjB,MAAM,EAAE,cAAc;QACtB,OAAO,EAAE,EAAE;KACU,CAAC;IAExB,UAAU,CAAC,GAAG,EAAE;QACd,IAAI,CAAC,MAAM,CAAC,sCAAc,CAAC,CAAC,eAAe,CAAC,UAAU,CAAC,CAAC;IAC1D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qCAAqC,EAAE,GAAG,EAAE;QAC7C,MAAM,OAAO,GAAG,CAAC,EAAE,QAAQ,EAAiC,EAAE,EAAE,CAAC,CAC/D,8BAAC,mCAAe,IAAC,MAAM,EAAC,cAAc,IAAE,QAAQ,CAAmB,CACpE,CAAC;QAEF,MAAM,EAAE,MAAM,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CAAC,IAAA,0CAAgB,GAAE,EAAE,EAAE,OAAO,EAAE,CAAC,CAAC;QAErE,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC,WAAW,EAAE,CAAC;QACnD,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC,WAAW,EAAE,CAAC;QAClD,MAAM,CAAC,OAAO,MAAM,CAAC,OAAO,CAAC,iBAAiB,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QACjE,MAAM,CAAC,OAAO,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;IAC9D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,mCAAmC,EAAE,GAAG,EAAE;QAC3C,MAAM,OAAO,GAAG,CAAC,EAAE,QAAQ,EAAiC,EAAE,EAAE,CAAC,CAC/D,8BAAC,mCAAe,IAAC,MAAM,EAAC,cAAc,IAAE,QAAQ,CAAmB,CACpE,CAAC;QAEF,MAAM,EAAE,MAAM,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CAAC,IAAA,wCAAc,GAAE,EAAE,EAAE,OAAO,EAAE,CAAC,CAAC;QAEnE,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC,WAAW,EAAE,CAAC;QAC/C,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC,QAAQ,EAAE,CAAC;IACpD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yCAAyC,EAAE,GAAG,EAAE;QACjD,MAAM,OAAO,GAAG,CAAC,EAAE,QAAQ,EAAiC,EAAE,EAAE,CAAC,CAC/D,8BAAC,mCAAe,IAAC,MAAM,EAAC,cAAc,IAAE,QAAQ,CAAmB,CACpE,CAAC;QAEF,MAAM,EAAE,MAAM,EAAE,GAAG,IAAA,kBAAU,EAC3B,GAAG,EAAE,CAAC,CAAC;YACL,KAAK,EAAE,IAAA,wCAAc,GAAE;YACvB,UAAU,EAAE,IAAA,6CAAmB,GAAE;SAClC,CAAC,EACF,EAAE,OAAO,EAAE,CACZ,CAAC;QAEF,sBAAsB;QACtB,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,eAAe,CAAC,CAAC,QAAQ,EAAE,CAAC;QAExD,oCAAoC;QACpC,IAAA,WAAG,EAAC,GAAG,EAAE;YACP,MAAM,CAAC,OAAO,CAAC,UAAU,CAAC,UAAU,CAAC,YAAY,CAAC,CAAC;YACnD,MAAM,CAAC,OAAO,CAAC,UAAU,CAAC,YAAY,CAAC,YAAY,CAAC,CAAC;QACvD,CAAC,CAAC,CAAC;QAEH,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,eAAe,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;QAChE,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,SAAS,CAAC,UAAU,CAAC,CAAC,WAAW,EAAE,CAAC;IAClE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,iCAAiC,EAAE,GAAG,EAAE;QACzC,MAAM,OAAO,GAAG,CAAC,EAAE,QAAQ,EAAiC,EAAE,EAAE,CAAC,CAC/D,8BAAC,mCAAe,IAAC,MAAM,EAAC,cAAc,IAAE,QAAQ,CAAmB,CACpE,CAAC;QAEF,MAAM,EAAE,MAAM,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CAAC,IAAA,4BAAc,GAAE,EAAE,EAAE,OAAO,EAAE,CAAC,CAAC;QAEnE,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,cAAc,CAAC,yBAAW,CAAC,CAAC;IACrD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,wCAAwC,EAAE,GAAG,EAAE;QAChD,MAAM,YAAY,GAAG,IAAI,yBAAW,EAAE,CAAC;QACvC,MAAM,OAAO,GAAG,CAAC,EAAE,QAAQ,EAAiC,EAAE,EAAE,CAAC,CAC/D,8BAAC,mCAAe,IAAC,MAAM,EAAC,cAAc,EAAC,WAAW,EAAE,YAAY,IAC7D,QAAQ,CACO,CACnB,CAAC;QAEF,MAAM,EAAE,MAAM,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CAAC,IAAA,4BAAc,GAAE,EAAE,EAAE,OAAO,EAAE,CAAC,CAAC;QAEnE,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;IAC5C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,oCAAoC,EAAE,GAAG,EAAE;QAC5C,MAAM,aAAa,GAAG,GAAG,EAAE,CAAC,kDAAe,CAAC;QAC5C,MAAM,WAAW,GAAG,OAAC,CAAC,MAAM,CAAC;YAC3B,KAAK,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,WAAW,CAAC;SACxC,CAAC,CAAC;QACH,MAAM,UAAU,GAAG;YACjB;gBACE,IAAI,EAAE,eAAe;gBACrB,WAAW,EAAE,kBAAkB;gBAC/B,SAAS,EAAE,aAAa;gBACxB,WAAW;aACZ;SACF,CAAC;QAEF,MAAM,OAAO,GAAG,CAAC,EAAE,QAAQ,EAAiC,EAAE,EAAE,CAAC,CAC/D,8BAAC,mCAAe,IAAC,MAAM,EAAC,cAAc,EAAC,UAAU,EAAE,UAAU,IAC1D,QAAQ,CACO,CACnB,CAAC;QAEF,MAAM,EAAE,MAAM,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CAAC,IAAA,0CAAgB,GAAE,EAAE,EAAE,OAAO,EAAE,CAAC,CAAC;QAErE,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,aAAa,CAAC,aAAa,CAAC,CAAC,WAAW,EAAE,CAAC;QACjE,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,aAAa,CAAC,aAAa,CAAC,IAAI,CAAC,CAAC,IAAI,CAC1D,eAAe,CAChB,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+BAA+B,EAAE,GAAG,EAAE;QACvC,MAAM,WAAW,GAAG,OAAC,CAAC,MAAM,CAAC;YAC3B,KAAK,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,cAAc,CAAC;SAC3C,CAAC,CAAC;QACH,MAAM,YAAY,GAAG,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,eAAe,CAAC,CAAC;QAC1D,MAAM,KAAK,GAAG;YACZ;gBACE,IAAI,EAAE,UAAU;gBAChB,WAAW,EAAE,aAAa;gBAC1B,IAAI,EAAE,KAAK,IAAI,EAAE,CAAC,QAAQ;gBAC1B,WAAW;gBACX,YAAY;aACb;SACF,CAAC;QAEF,MAAM,OAAO,GAAG,CAAC,EAAE,QAAQ,EAAiC,EAAE,EAAE,CAAC,CAC/D,8BAAC,mCAAe,IAAC,MAAM,EAAC,cAAc,EAAC,KAAK,EAAE,KAAK,IAChD,QAAQ,CACO,CACnB,CAAC;QAEF,MAAM,EAAE,MAAM,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CAAC,IAAA,0CAAgB,GAAE,EAAE,EAAE,OAAO,EAAE,CAAC,CAAC;QAErE,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,QAAQ,CAAC,CAAC,WAAW,EAAE,CAAC;QAC3D,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;IACrE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qCAAqC,EAAE,GAAG,EAAE;QAC7C,MAAM,UAAU,GAAG;YACjB,EAAE,GAAG,EAAE,yBAAyB,EAAE,IAAI,EAAE,aAAa,EAAE;SACxD,CAAC;QAEF,MAAM,OAAO,GAAG,CAAC,EAAE,QAAQ,EAAiC,EAAE,EAAE,CAAC,CAC/D,8BAAC,mCAAe,IAAC,MAAM,EAAC,cAAc,EAAC,UAAU,EAAE,UAAU,IAC1D,QAAQ,CACO,CACnB,CAAC;QAEF,MAAM,EAAE,MAAM,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CAAC,IAAA,0CAAgB,GAAE,EAAE,EAAE,OAAO,EAAE,CAAC,CAAC;QAErE,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,cAAc,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QACtD,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,IAAI,CAC/C,yBAAyB,CAC1B,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6CAA6C,EAAE,GAAG,EAAE;QACrD,MAAM,sBAAsB,GAAG,IAAI;aAChC,EAAE,EAAE;aACJ,iBAAiB,CAAC,iBAAiB,CAAC,CAAC;QAExC,MAAM,OAAO,GAAG,CAAC,EAAE,QAAQ,EAAiC,EAAE,EAAE,CAAC,CAC/D,8BAAC,mCAAe,IACd,MAAM,EAAC,cAAc,EACrB,sBAAsB,EAAE,sBAAsB,IAE7C,QAAQ,CACO,CACnB,CAAC;QAEF,MAAM,EAAE,MAAM,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CAAC,IAAA,0CAAgB,GAAE,EAAE,EAAE,OAAO,EAAE,CAAC,CAAC;QAErE,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,sBAAsB,CAAC,CAAC,IAAI,CAAC,sBAAsB,CAAC,CAAC;IAC7E,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,0CAA0C,EAAE,GAAG,EAAE;QAClD,MAAM,SAAS,GAAG;YAChB;gBACE,GAAG,EAAE,yBAAyB;gBAC9B,IAAI,EAAE,eAAe;gBACrB,WAAW,EAAE,iBAAiB;gBAC9B,QAAQ,EAAE,YAAY;aACvB;SACF,CAAC;QAEF,MAAM,OAAO,GAAG,CAAC,EAAE,QAAQ,EAAiC,EAAE,EAAE,CAAC,CAC/D,8BAAC,mCAAe,IAAC,MAAM,EAAC,cAAc,EAAC,SAAS,EAAE,SAAS,IACxD,QAAQ,CACO,CACnB,CAAC;QAEF,MAAM,EAAE,MAAM,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CAAC,IAAA,0CAAgB,GAAE,EAAE,EAAE,OAAO,EAAE,CAAC,CAAC;QAErE,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QACjD,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,yBAAyB,CAAC,CAAC;QACxE,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;IACjE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,uEAAuE,EAAE,GAAG,EAAE;QAC/E,MAAM,aAAa,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,EAAE,SAAS,EAAE,EAAE,EAAE,CAAC,CAAC;QACrE,MAAM,WAAW,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,EAAE,QAAQ,EAAE,EAAE,EAAE,CAAC,CAAC;QAElE,MAAM,OAAO,GAAG,CAAC,EAAE,QAAQ,EAAiC,EAAE,EAAE,CAAC,CAC/D,8BAAC,mCAAe,IACd,MAAM,EAAC,cAAc,EACrB,aAAa,EAAE,aAAa,EAC5B,WAAW,EAAE,WAAW,IAEvB,QAAQ,CACO,CACnB,CAAC;QAEF,MAAM,EAAE,MAAM,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CAAC,IAAA,0CAAgB,GAAE,EAAE,EAAE,OAAO,EAAE,CAAC,CAAC;QAErE,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,cAAc,CAAC,CAAC,WAAW,EAAE,CAAC;QACpD,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,cAAc,EAAE,aAAa,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;QACzE,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,cAAc,EAAE,WAAW,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;IACvE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,4CAA4C,EAAE,GAAG,EAAE;QACpD,MAAM,OAAO,GAAG,CAAC,EAAE,QAAQ,EAAiC,EAAE,EAAE,CAAC,CAC/D,8BAAC,mCAAe,IAAC,MAAM,EAAC,cAAc,EAAC,UAAU,EAAC,gBAAgB,IAC/D,QAAQ,CACO,CACnB,CAAC;QAEF,MAAM,EAAE,MAAM,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CAAC,IAAA,iCAAa,GAAE,EAAE,EAAE,OAAO,EAAE,CAAC,CAAC;QAElE,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;IAChD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kEAAkE,EAAE,GAAG,EAAE;QAC1E,MAAM,OAAO,GAAG,CAAC,EAAE,QAAQ,EAAiC,EAAE,EAAE,CAAC,CAC/D,8BAAC,mCAAe,IAAC,MAAM,EAAC,cAAc,IAAE,QAAQ,CAAmB,CACpE,CAAC;QAEF,MAAM,EAAE,MAAM,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CAAC,IAAA,iCAAa,GAAE,EAAE,EAAE,OAAO,EAAE,CAAC,CAAC;QAElE,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,aAAa,EAAE,CAAC;IACzC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,0DAA0D,EAAE,KAAK,IAAI,EAAE;QACxE,MAAM,cAAc,GAAG;YACrB,WAAW,EAAE,GAAG,EAAE,CAAC,WAAW;YAC9B,cAAc,EAAE,GAAG,EAAE,CAAC,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;SAC/C,CAAC;QAEF,MAAM,OAAO,GAAG,CAAC,EAAE,QAAQ,EAAiC,EAAE,EAAE,CAAC,CAC/D,8BAAC,mCAAe,IAAC,MAAM,EAAC,cAAc,EAAC,cAAc,EAAE,cAAc,IAClE,QAAQ,CACO,CACnB,CAAC;QAEF,MAAM,EAAE,MAAM,EAAE,QAAQ,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CAAC,IAAA,uDAAsB,GAAE,EAAE;YACtE,OAAO;SACR,CAAC,CAAC;QAEH,yEAAyE;QACzE,MAAM,IAAA,WAAG,EAAC,KAAK,IAAI,EAAE;YACnB,QAAQ,EAAE,CAAC;QACb,CAAC,CAAC,CAAC;QAEH,MAAM,OAAO,GAAG,MAAM,CAAC,OAAO,CAAC,iBAAiB,EAAE,CAAC;QACnD,MAAM,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,WAAW,CAAC,CAAC;QAC7D,MAAM,CAAC,OAAO,CAAC,cAAc,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,cAAc,CAAC,CAAC;IACrE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,iDAAiD,EAAE,KAAK,IAAI,EAAE;QAC/D,MAAM,OAAO,GAAG,CAAC,EAAE,QAAQ,EAAiC,EAAE,EAAE,CAAC,CAC/D,8BAAC,mCAAe,IAAC,MAAM,EAAC,cAAc,IAAE,QAAQ,CAAmB,CACpE,CAAC;QAEF,MAAM,EAAE,MAAM,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CAAC,IAAA,uDAAsB,GAAE,EAAE,EAAE,OAAO,EAAE,CAAC,CAAC;QAE3E,MAAM,OAAO,GAAG,MAAM,CAAC,OAAO,CAAC,iBAAiB,EAAE,CAAC;QACnD,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;IAC/C,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC","sourcesContent":["import TamboAI from \"@tambo-ai/typescript-sdk\";\nimport { QueryClient, useQueryClient } from \"@tanstack/react-query\";\nimport { renderHook, act } from \"@testing-library/react\";\nimport React from \"react\";\nimport { z } from \"zod\";\nimport { useTamboClient } from \"../../providers/tambo-client-provider\";\nimport { useTamboRegistry } from \"../../providers/tambo-registry-provider\";\nimport { useTamboContextHelpers } from \"../../providers/tambo-context-helpers-provider\";\nimport { useStreamState, useThreadManagement } from \"./tambo-v1-stream-context\";\nimport { TamboV1Provider, useContextKey } from \"./tambo-v1-provider\";\n\n// Mock the client provider to capture the apiKey\njest.mock(\"../../providers/tambo-client-provider\", () => ({\n  useTamboClient: jest.fn(),\n  TamboClientProvider: ({ children }: { children: React.ReactNode }) =>\n    children,\n}));\n\ndescribe(\"TamboV1Provider\", () => {\n  const mockClient = {\n    apiKey: \"test-api-key\",\n    threads: {},\n  } as unknown as TamboAI;\n\n  beforeEach(() => {\n    jest.mocked(useTamboClient).mockReturnValue(mockClient);\n  });\n\n  it(\"provides access to registry context\", () => {\n    const wrapper = ({ children }: { children: React.ReactNode }) => (\n      <TamboV1Provider apiKey=\"test-api-key\">{children}</TamboV1Provider>\n    );\n\n    const { result } = renderHook(() => useTamboRegistry(), { wrapper });\n\n    expect(result.current.componentList).toBeDefined();\n    expect(result.current.toolRegistry).toBeDefined();\n    expect(typeof result.current.registerComponent).toBe(\"function\");\n    expect(typeof result.current.registerTool).toBe(\"function\");\n  });\n\n  it(\"provides access to stream context\", () => {\n    const wrapper = ({ children }: { children: React.ReactNode }) => (\n      <TamboV1Provider apiKey=\"test-api-key\">{children}</TamboV1Provider>\n    );\n\n    const { result } = renderHook(() => useStreamState(), { wrapper });\n\n    expect(result.current.threadMap).toBeDefined();\n    expect(result.current.currentThreadId).toBeNull();\n  });\n\n  it(\"manages threads via useThreadManagement\", () => {\n    const wrapper = ({ children }: { children: React.ReactNode }) => (\n      <TamboV1Provider apiKey=\"test-api-key\">{children}</TamboV1Provider>\n    );\n\n    const { result } = renderHook(\n      () => ({\n        state: useStreamState(),\n        management: useThreadManagement(),\n      }),\n      { wrapper },\n    );\n\n    // Initially no thread\n    expect(result.current.state.currentThreadId).toBeNull();\n\n    // Initialize and switch to a thread\n    act(() => {\n      result.current.management.initThread(\"thread_123\");\n      result.current.management.switchThread(\"thread_123\");\n    });\n\n    expect(result.current.state.currentThreadId).toBe(\"thread_123\");\n    expect(result.current.state.threadMap.thread_123).toBeDefined();\n  });\n\n  it(\"provides access to query client\", () => {\n    const wrapper = ({ children }: { children: React.ReactNode }) => (\n      <TamboV1Provider apiKey=\"test-api-key\">{children}</TamboV1Provider>\n    );\n\n    const { result } = renderHook(() => useQueryClient(), { wrapper });\n\n    expect(result.current).toBeInstanceOf(QueryClient);\n  });\n\n  it(\"uses custom query client when provided\", () => {\n    const customClient = new QueryClient();\n    const wrapper = ({ children }: { children: React.ReactNode }) => (\n      <TamboV1Provider apiKey=\"test-api-key\" queryClient={customClient}>\n        {children}\n      </TamboV1Provider>\n    );\n\n    const { result } = renderHook(() => useQueryClient(), { wrapper });\n\n    expect(result.current).toBe(customClient);\n  });\n\n  it(\"registers components when provided\", () => {\n    const TestComponent = () => <div>Test</div>;\n    const propsSchema = z.object({\n      title: z.string().describe(\"The title\"),\n    });\n    const components = [\n      {\n        name: \"TestComponent\",\n        description: \"A test component\",\n        component: TestComponent,\n        propsSchema,\n      },\n    ];\n\n    const wrapper = ({ children }: { children: React.ReactNode }) => (\n      <TamboV1Provider apiKey=\"test-api-key\" components={components}>\n        {children}\n      </TamboV1Provider>\n    );\n\n    const { result } = renderHook(() => useTamboRegistry(), { wrapper });\n\n    expect(result.current.componentList.TestComponent).toBeDefined();\n    expect(result.current.componentList.TestComponent.name).toBe(\n      \"TestComponent\",\n    );\n  });\n\n  it(\"registers tools when provided\", () => {\n    const inputSchema = z.object({\n      query: z.string().describe(\"Search query\"),\n    });\n    const outputSchema = z.string().describe(\"Result string\");\n    const tools = [\n      {\n        name: \"testTool\",\n        description: \"A test tool\",\n        tool: async () => \"result\",\n        inputSchema,\n        outputSchema,\n      },\n    ];\n\n    const wrapper = ({ children }: { children: React.ReactNode }) => (\n      <TamboV1Provider apiKey=\"test-api-key\" tools={tools}>\n        {children}\n      </TamboV1Provider>\n    );\n\n    const { result } = renderHook(() => useTamboRegistry(), { wrapper });\n\n    expect(result.current.toolRegistry.testTool).toBeDefined();\n    expect(result.current.toolRegistry.testTool.name).toBe(\"testTool\");\n  });\n\n  it(\"registers MCP servers when provided\", () => {\n    const mcpServers = [\n      { url: \"https://mcp.example.com\", name: \"Example MCP\" },\n    ];\n\n    const wrapper = ({ children }: { children: React.ReactNode }) => (\n      <TamboV1Provider apiKey=\"test-api-key\" mcpServers={mcpServers}>\n        {children}\n      </TamboV1Provider>\n    );\n\n    const { result } = renderHook(() => useTamboRegistry(), { wrapper });\n\n    expect(result.current.mcpServerInfos).toHaveLength(1);\n    expect(result.current.mcpServerInfos[0].url).toBe(\n      \"https://mcp.example.com\",\n    );\n  });\n\n  it(\"provides onCallUnregisteredTool to registry\", () => {\n    const onCallUnregisteredTool = jest\n      .fn()\n      .mockResolvedValue(\"fallback result\");\n\n    const wrapper = ({ children }: { children: React.ReactNode }) => (\n      <TamboV1Provider\n        apiKey=\"test-api-key\"\n        onCallUnregisteredTool={onCallUnregisteredTool}\n      >\n        {children}\n      </TamboV1Provider>\n    );\n\n    const { result } = renderHook(() => useTamboRegistry(), { wrapper });\n\n    expect(result.current.onCallUnregisteredTool).toBe(onCallUnregisteredTool);\n  });\n\n  it(\"registers static resources when provided\", () => {\n    const resources = [\n      {\n        uri: \"resource://test/example\",\n        name: \"Test Resource\",\n        description: \"A test resource\",\n        mimeType: \"text/plain\",\n      },\n    ];\n\n    const wrapper = ({ children }: { children: React.ReactNode }) => (\n      <TamboV1Provider apiKey=\"test-api-key\" resources={resources}>\n        {children}\n      </TamboV1Provider>\n    );\n\n    const { result } = renderHook(() => useTamboRegistry(), { wrapper });\n\n    expect(result.current.resources).toHaveLength(1);\n    expect(result.current.resources[0].uri).toBe(\"resource://test/example\");\n    expect(result.current.resources[0].name).toBe(\"Test Resource\");\n  });\n\n  it(\"registers resource source when listResources and getResource provided\", () => {\n    const listResources = jest.fn().mockResolvedValue({ resources: [] });\n    const getResource = jest.fn().mockResolvedValue({ contents: [] });\n\n    const wrapper = ({ children }: { children: React.ReactNode }) => (\n      <TamboV1Provider\n        apiKey=\"test-api-key\"\n        listResources={listResources}\n        getResource={getResource}\n      >\n        {children}\n      </TamboV1Provider>\n    );\n\n    const { result } = renderHook(() => useTamboRegistry(), { wrapper });\n\n    expect(result.current.resourceSource).toBeDefined();\n    expect(result.current.resourceSource?.listResources).toBe(listResources);\n    expect(result.current.resourceSource?.getResource).toBe(getResource);\n  });\n\n  it(\"provides contextKey via useContextKey hook\", () => {\n    const wrapper = ({ children }: { children: React.ReactNode }) => (\n      <TamboV1Provider apiKey=\"test-api-key\" contextKey=\"my-context-key\">\n        {children}\n      </TamboV1Provider>\n    );\n\n    const { result } = renderHook(() => useContextKey(), { wrapper });\n\n    expect(result.current).toBe(\"my-context-key\");\n  });\n\n  it(\"returns undefined from useContextKey when no contextKey provided\", () => {\n    const wrapper = ({ children }: { children: React.ReactNode }) => (\n      <TamboV1Provider apiKey=\"test-api-key\">{children}</TamboV1Provider>\n    );\n\n    const { result } = renderHook(() => useContextKey(), { wrapper });\n\n    expect(result.current).toBeUndefined();\n  });\n\n  it(\"provides context helpers via useTamboContextHelpers hook\", async () => {\n    const contextHelpers = {\n      getUserName: () => \"Test User\",\n      getCurrentTime: () => new Date().toISOString(),\n    };\n\n    const wrapper = ({ children }: { children: React.ReactNode }) => (\n      <TamboV1Provider apiKey=\"test-api-key\" contextHelpers={contextHelpers}>\n        {children}\n      </TamboV1Provider>\n    );\n\n    const { result, rerender } = renderHook(() => useTamboContextHelpers(), {\n      wrapper,\n    });\n\n    // Helpers are registered via useEffect, so we need to trigger a rerender\n    await act(async () => {\n      rerender();\n    });\n\n    const helpers = result.current.getContextHelpers();\n    expect(helpers.getUserName).toBe(contextHelpers.getUserName);\n    expect(helpers.getCurrentTime).toBe(contextHelpers.getCurrentTime);\n  });\n\n  it(\"returns empty contextHelpers when none provided\", async () => {\n    const wrapper = ({ children }: { children: React.ReactNode }) => (\n      <TamboV1Provider apiKey=\"test-api-key\">{children}</TamboV1Provider>\n    );\n\n    const { result } = renderHook(() => useTamboContextHelpers(), { wrapper });\n\n    const helpers = result.current.getContextHelpers();\n    expect(Object.keys(helpers)).toHaveLength(0);\n  });\n});\n"]}

package/dist/v1/providers/tambo-v1-stream-context.d.ts

@@ -0,0 +1,136 @@
+/**
+ * Stream Context Provider for v1 API
+ *
+ * Manages streaming state using React Context and useReducer.
+ * Provides state and dispatch to child components via separate contexts
+ * following the split-context pattern for optimal re-render performance.
+ */
+import React, { type ReactNode, type Dispatch } from "react";
+import { type StreamState, type StreamAction } from "../utils/event-accumulator";
+import type { TamboV1Thread } from "../types/thread";
+/**
+ * Thread management functions exposed by the stream context.
+ */
+export interface ThreadManagement {
+    /**
+     * Initialize a new thread in the stream context.
+     * Use this before sending messages to a new thread.
+     * @param threadId - The thread ID to initialize
+     * @param initialThread - Optional initial thread data
+     */
+    initThread: (threadId: string, initialThread?: Partial<TamboV1Thread>) => void;
+    /**
+     * Switch the current active thread.
+     * Does not fetch thread data - use useTamboV1Thread for that.
+     * @param threadId - The thread ID to switch to, or null to clear
+     */
+    switchThread: (threadId: string | null) => void;
+    /**
+     * Start a new thread (generates a temporary ID).
+     * The actual thread ID will be assigned when the first message is sent.
+     * @returns The temporary thread ID
+     */
+    startNewThread: () => string;
+}
+/**
+ * Props for TamboV1StreamProvider
+ */
+export interface TamboV1StreamProviderProps {
+    children: ReactNode;
+    /**
+     * Initial thread state (optional).
+     * If not provided, an empty thread will be created.
+     */
+    initialThread?: Partial<TamboV1Thread>;
+    /**
+     * Thread ID for the stream context.
+     * Used to initialize the thread if initialThread is not provided.
+     */
+    threadId?: string;
+}
+/**
+ * Provider component for stream state management.
+ *
+ * Uses useReducer with streamReducer to accumulate AG-UI events into
+ * thread state. Provides state, dispatch, and thread management via separate contexts.
+ * @returns JSX element wrapping children with stream contexts
+ * @example
+ * ```tsx
+ * <TamboV1StreamProvider threadId="thread_123">
+ *   <ChatInterface />
+ * </TamboV1StreamProvider>
+ * ```
+ */
+export declare function TamboV1StreamProvider(props: TamboV1StreamProviderProps): React.JSX.Element;
+/**
+ * Hook to access stream state.
+ *
+ * Must be used within TamboV1StreamProvider.
+ * @returns Current stream state
+ * @throws {Error} if used outside TamboV1StreamProvider
+ * @example
+ * ```tsx
+ * function ChatMessages() {
+ *   const { thread, streaming } = useStreamState();
+ *
+ *   return (
+ *     <div>
+ *       {thread.messages.map(msg => <Message key={msg.id} message={msg} />)}
+ *       {streaming.status === 'streaming' && <LoadingIndicator />}
+ *     </div>
+ *   );
+ * }
+ * ```
+ */
+export declare function useStreamState(): StreamState;
+/**
+ * Hook to access stream dispatch function.
+ *
+ * Must be used within TamboV1StreamProvider.
+ * @returns Dispatch function for sending events to reducer
+ * @throws {Error} if used outside TamboV1StreamProvider
+ * @example
+ * ```tsx
+ * function StreamHandler() {
+ *   const dispatch = useStreamDispatch();
+ *
+ *   useEffect(() => {
+ *     async function handleStream() {
+ *       for await (const event of streamEvents) {
+ *         dispatch({ type: 'EVENT', event });
+ *       }
+ *     }
+ *     handleStream();
+ *   }, [dispatch]);
+ *
+ *   return null;
+ * }
+ * ```
+ */
+export declare function useStreamDispatch(): Dispatch<StreamAction>;
+/**
+ * Hook to access thread management functions.
+ *
+ * Must be used within TamboV1StreamProvider.
+ * @returns Thread management functions
+ * @throws {Error} if used outside TamboV1StreamProvider
+ * @example
+ * ```tsx
+ * function ThreadSwitcher() {
+ *   const { switchThread, startNewThread } = useThreadManagement();
+ *
+ *   return (
+ *     <div>
+ *       <button onClick={() => switchThread('thread_123')}>
+ *         Load Thread
+ *       </button>
+ *       <button onClick={startNewThread}>
+ *         New Chat
+ *       </button>
+ *     </div>
+ *   );
+ * }
+ * ```
+ */
+export declare function useThreadManagement(): ThreadManagement;
+//# sourceMappingURL=tambo-v1-stream-context.d.ts.map

package/dist/v1/providers/tambo-v1-stream-context.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"tambo-v1-stream-context.d.ts","sourceRoot":"","sources":["../../../src/v1/providers/tambo-v1-stream-context.tsx"],"names":[],"mappings":"AAEA;;;;;;GAMG;AAIH,OAAO,KAAK,EAAE,EAMZ,KAAK,SAAS,EACd,KAAK,QAAQ,EACd,MAAM,OAAO,CAAC;AACf,OAAO,EAGL,KAAK,WAAW,EAChB,KAAK,YAAY,EAClB,MAAM,4BAA4B,CAAC;AACpC,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,iBAAiB,CAAC;AAErD;;GAEG;AACH,MAAM,WAAW,gBAAgB;IAC/B;;;;;OAKG;IACH,UAAU,EAAE,CACV,QAAQ,EAAE,MAAM,EAChB,aAAa,CAAC,EAAE,OAAO,CAAC,aAAa,CAAC,KACnC,IAAI,CAAC;IAEV;;;;OAIG;IACH,YAAY,EAAE,CAAC,QAAQ,EAAE,MAAM,GAAG,IAAI,KAAK,IAAI,CAAC;IAEhD;;;;OAIG;IACH,cAAc,EAAE,MAAM,MAAM,CAAC;CAC9B;AAsBD;;GAEG;AACH,MAAM,WAAW,0BAA0B;IACzC,QAAQ,EAAE,SAAS,CAAC;IAEpB;;;OAGG;IACH,aAAa,CAAC,EAAE,OAAO,CAAC,aAAa,CAAC,CAAC;IAEvC;;;OAGG;IACH,QAAQ,CAAC,EAAE,MAAM,CAAC;CACnB;AA0CD;;;;;;;;;;;;GAYG;AACH,wBAAgB,qBAAqB,CAAC,KAAK,EAAE,0BAA0B,qBAkDtE;AAED;;;;;;;;;;;;;;;;;;;GAmBG;AACH,wBAAgB,cAAc,IAAI,WAAW,CAQ5C;AAED;;;;;;;;;;;;;;;;;;;;;;;GAuBG;AACH,wBAAgB,iBAAiB,IAAI,QAAQ,CAAC,YAAY,CAAC,CAU1D;AAED;;;;;;;;;;;;;;;;;;;;;;;GAuBG;AACH,wBAAgB,mBAAmB,IAAI,gBAAgB,CAUtD"}