@tambo-ai/react 0.42.0 → 0.43.0

package/dist/index.js

@@ -15,7 +15,7 @@ var __exportStar = (this && this.__exportStar) || function(m, exports) {
     for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
 };
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.getUserTimeContext = exports.getUserPageContext = exports.DEFAULT_CONTEXT_HELPERS = exports.useTamboInteractable = exports.withInteractable = exports.GenerationStage = exports.useTamboThreadList = exports.useTamboThread = exports.useTamboStream = exports.useTamboContextHelpers = exports.useTamboClient = exports.useTambo = exports.TamboThreadProvider = exports.TamboStubProvider = exports.TamboProvider = exports.TamboPropStreamProvider = exports.TamboContextHelpersProvider = exports.TamboComponentProvider = exports.TamboClientProvider = exports.useTamboThreadInput = exports.useTamboStreamStatus = exports.useTamboStreamingProps = exports.useTamboMessageContext = exports.useTamboCurrentMessage = exports.TamboMessageProvider = exports.useTamboComponentState = void 0;
+exports.currentTimeContextHelper = exports.currentPageContextHelper = exports.useTamboInteractable = exports.withInteractable = exports.GenerationStage = exports.useTamboThreadList = exports.useTamboThread = exports.useTamboStream = exports.useTamboContextHelpers = exports.useTamboClient = exports.useTambo = exports.TamboThreadProvider = exports.TamboStubProvider = exports.TamboProvider = exports.TamboPropStreamProvider = exports.TamboContextHelpersProvider = exports.TamboComponentProvider = exports.TamboClientProvider = exports.useTamboThreadInput = exports.useTamboStreamStatus = exports.useTamboStreamingProps = exports.useTamboMessageContext = exports.useTamboCurrentMessage = exports.TamboMessageProvider = exports.useTamboComponentState = void 0;
 var use_component_state_1 = require("./hooks/use-component-state");
 Object.defineProperty(exports, "useTamboComponentState", { enumerable: true, get: function () { return use_component_state_1.useTamboComponentState; } });
 var use_current_message_1 = require("./hooks/use-current-message");
@@ -53,7 +53,6 @@ var tambo_interactable_provider_1 = require("./providers/tambo-interactable-prov
 Object.defineProperty(exports, "useTamboInteractable", { enumerable: true, get: function () { return tambo_interactable_provider_1.useTamboInteractable; } });
 // Context helpers exports
 var context_helpers_1 = require("./context-helpers");
-Object.defineProperty(exports, "DEFAULT_CONTEXT_HELPERS", { enumerable: true, get: function () { return context_helpers_1.DEFAULT_CONTEXT_HELPERS; } });
-Object.defineProperty(exports, "getUserPageContext", { enumerable: true, get: function () { return context_helpers_1.getUserPageContext; } });
-Object.defineProperty(exports, "getUserTimeContext", { enumerable: true, get: function () { return context_helpers_1.getUserTimeContext; } });
+Object.defineProperty(exports, "currentPageContextHelper", { enumerable: true, get: function () { return context_helpers_1.currentPageContextHelper; } });
+Object.defineProperty(exports, "currentTimeContextHelper", { enumerable: true, get: function () { return context_helpers_1.currentTimeContextHelper; } });
 //# sourceMappingURL=index.js.map

package/dist/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":";AAAA,wKAAwK;;;;;;;;;;;;;;;;;AAExK,mEAAqE;AAA5D,6HAAA,sBAAsB,OAAA;AAC/B,mEAIqC;AAHnC,2HAAA,oBAAoB,OAAA;AACpB,6HAAA,sBAAsB,OAAA;AACtB,6HAAA,sBAAsB,OAAA;AAExB,mEAAqE;AAA5D,6HAAA,sBAAsB,OAAA;AAC/B,0DAAwC;AACxC,2EAIyC;AAHvC,+HAAA,oBAAoB,OAAA;AAItB,6DAA+D;AAAtD,uHAAA,mBAAmB,OAAA;AAE5B,gCAAgC;AAChC,yCAkBqB;AAjBnB,gHAAA,mBAAmB,OAAA;AACnB,mHAAA,sBAAsB,OAAA;AACtB,wHAAA,2BAA2B,OAAA;AAC3B,oHAAA,uBAAuB,OAAA;AACvB,0GAAA,aAAa,OAAA;AACb,8GAAA,iBAAiB,OAAA;AACjB,gHAAA,mBAAmB,OAAA;AACnB,qGAAA,QAAQ,OAAA;AACR,2GAAA,cAAc,OAAA;AACd,mHAAA,sBAAsB,OAAA;AACtB,2GAAA,cAAc,OAAA;AACd,2GAAA,cAAc,OAAA;AAoBhB,+DAA+D;AAAtD,uHAAA,kBAAkB,OAAA;AAQ3B,mFAG6C;AAF3C,8HAAA,eAAe,OAAA;AASjB,mFAIiD;AAH/C,2HAAA,qBAAqB,OAAoB;AAI3C,uFAA+E;AAAtE,mIAAA,oBAAoB,OAAA;AAE7B,0BAA0B;AAC1B,qDAI2B;AAHzB,0HAAA,uBAAuB,OAAA;AACvB,qHAAA,kBAAkB,OAAA;AAClB,qHAAA,kBAAkB,OAAA","sourcesContent":["/** Exports for the library. Only publically available exports are re-exported here. Anything not exported here is not supported and may change or break at any time. */\n\nexport { useTamboComponentState } from \"./hooks/use-component-state\";\nexport {\n  TamboMessageProvider,\n  useTamboCurrentMessage,\n  useTamboMessageContext,\n} from \"./hooks/use-current-message\";\nexport { useTamboStreamingProps } from \"./hooks/use-streaming-props\";\nexport * from \"./hooks/use-suggestions\";\nexport {\n  useTamboStreamStatus,\n  type PropStatus,\n  type StreamStatus,\n} from \"./hooks/use-tambo-stream-status\";\nexport { useTamboThreadInput } from \"./hooks/use-thread-input\";\n\n// Re-export provider components\nexport {\n  TamboClientProvider,\n  TamboComponentProvider,\n  TamboContextHelpersProvider,\n  TamboPropStreamProvider,\n  TamboProvider,\n  TamboStubProvider,\n  TamboThreadProvider,\n  useTambo,\n  useTamboClient,\n  useTamboContextHelpers,\n  useTamboStream,\n  useTamboThread,\n  type TamboComponent,\n  type TamboContextHelpersContextProps,\n  type TamboContextHelpersProviderProps,\n  type TamboRegistryContext,\n  type TamboStubProviderProps,\n} from \"./providers\";\n\n// Re-export types from Tambo Node SDK\nexport type {\n  APIError,\n  RateLimitError,\n  TamboAIError,\n} from \"@tambo-ai/typescript-sdk\";\nexport type {\n  Suggestion,\n  SuggestionGenerateParams,\n  SuggestionGenerateResponse,\n  SuggestionListResponse,\n} from \"@tambo-ai/typescript-sdk/resources/beta/threads/suggestions\";\nexport { useTamboThreadList } from \"./hooks/use-tambo-threads\";\nexport {\n  type ComponentContextToolMetadata,\n  type ComponentRegistry,\n  type ParameterSpec,\n  type RegisteredComponent,\n  type TamboTool,\n} from \"./model/component-metadata\";\nexport {\n  GenerationStage,\n  type TamboThreadMessage,\n} from \"./model/generate-component-response\";\nexport { type TamboThread } from \"./model/tambo-thread\";\n\nexport type {\n  TamboInteractableComponent as InteractableComponent,\n  TamboInteractableContext,\n} from \"./model/tambo-interactable\";\nexport {\n  withTamboInteractable as withInteractable,\n  type InteractableConfig,\n  type WithTamboInteractableProps,\n} from \"./providers/hoc/with-tambo-interactable\";\nexport { useTamboInteractable } from \"./providers/tambo-interactable-provider\";\n\n// Context helpers exports\nexport {\n  DEFAULT_CONTEXT_HELPERS,\n  getUserPageContext,\n  getUserTimeContext,\n} from \"./context-helpers\";\nexport type {\n  AdditionalContext,\n  AdditionalContextHelper,\n  ContextHelpersConfig,\n} from \"./context-helpers\";\n"]}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":";AAAA,wKAAwK;;;;;;;;;;;;;;;;;AAExK,mEAAqE;AAA5D,6HAAA,sBAAsB,OAAA;AAC/B,mEAIqC;AAHnC,2HAAA,oBAAoB,OAAA;AACpB,6HAAA,sBAAsB,OAAA;AACtB,6HAAA,sBAAsB,OAAA;AAExB,mEAAqE;AAA5D,6HAAA,sBAAsB,OAAA;AAC/B,0DAAwC;AACxC,2EAIyC;AAHvC,+HAAA,oBAAoB,OAAA;AAItB,6DAA+D;AAAtD,uHAAA,mBAAmB,OAAA;AAE5B,gCAAgC;AAChC,yCAkBqB;AAjBnB,gHAAA,mBAAmB,OAAA;AACnB,mHAAA,sBAAsB,OAAA;AACtB,wHAAA,2BAA2B,OAAA;AAC3B,oHAAA,uBAAuB,OAAA;AACvB,0GAAA,aAAa,OAAA;AACb,8GAAA,iBAAiB,OAAA;AACjB,gHAAA,mBAAmB,OAAA;AACnB,qGAAA,QAAQ,OAAA;AACR,2GAAA,cAAc,OAAA;AACd,mHAAA,sBAAsB,OAAA;AACtB,2GAAA,cAAc,OAAA;AACd,2GAAA,cAAc,OAAA;AAoBhB,+DAA+D;AAAtD,uHAAA,kBAAkB,OAAA;AAQ3B,mFAG6C;AAF3C,8HAAA,eAAe,OAAA;AASjB,mFAIiD;AAH/C,2HAAA,qBAAqB,OAAoB;AAI3C,uFAA+E;AAAtE,mIAAA,oBAAoB,OAAA;AAE7B,0BAA0B;AAC1B,qDAG2B;AAFzB,2HAAA,wBAAwB,OAAA;AACxB,2HAAA,wBAAwB,OAAA","sourcesContent":["/** Exports for the library. Only publically available exports are re-exported here. Anything not exported here is not supported and may change or break at any time. */\n\nexport { useTamboComponentState } from \"./hooks/use-component-state\";\nexport {\n  TamboMessageProvider,\n  useTamboCurrentMessage,\n  useTamboMessageContext,\n} from \"./hooks/use-current-message\";\nexport { useTamboStreamingProps } from \"./hooks/use-streaming-props\";\nexport * from \"./hooks/use-suggestions\";\nexport {\n  useTamboStreamStatus,\n  type PropStatus,\n  type StreamStatus,\n} from \"./hooks/use-tambo-stream-status\";\nexport { useTamboThreadInput } from \"./hooks/use-thread-input\";\n\n// Re-export provider components\nexport {\n  TamboClientProvider,\n  TamboComponentProvider,\n  TamboContextHelpersProvider,\n  TamboPropStreamProvider,\n  TamboProvider,\n  TamboStubProvider,\n  TamboThreadProvider,\n  useTambo,\n  useTamboClient,\n  useTamboContextHelpers,\n  useTamboStream,\n  useTamboThread,\n  type TamboComponent,\n  type TamboContextHelpersContextProps,\n  type TamboContextHelpersProviderProps,\n  type TamboRegistryContext,\n  type TamboStubProviderProps,\n} from \"./providers\";\n\n// Re-export types from Tambo Node SDK\nexport type {\n  APIError,\n  RateLimitError,\n  TamboAIError,\n} from \"@tambo-ai/typescript-sdk\";\nexport type {\n  Suggestion,\n  SuggestionGenerateParams,\n  SuggestionGenerateResponse,\n  SuggestionListResponse,\n} from \"@tambo-ai/typescript-sdk/resources/beta/threads/suggestions\";\nexport { useTamboThreadList } from \"./hooks/use-tambo-threads\";\nexport {\n  type ComponentContextToolMetadata,\n  type ComponentRegistry,\n  type ParameterSpec,\n  type RegisteredComponent,\n  type TamboTool,\n} from \"./model/component-metadata\";\nexport {\n  GenerationStage,\n  type TamboThreadMessage,\n} from \"./model/generate-component-response\";\nexport { type TamboThread } from \"./model/tambo-thread\";\n\nexport type {\n  TamboInteractableComponent as InteractableComponent,\n  TamboInteractableContext,\n} from \"./model/tambo-interactable\";\nexport {\n  withTamboInteractable as withInteractable,\n  type InteractableConfig,\n  type WithTamboInteractableProps,\n} from \"./providers/hoc/with-tambo-interactable\";\nexport { useTamboInteractable } from \"./providers/tambo-interactable-provider\";\n\n// Context helpers exports\nexport {\n  currentPageContextHelper,\n  currentTimeContextHelper,\n} from \"./context-helpers\";\nexport type {\n  AdditionalContext,\n  ContextHelperFn,\n  ContextHelpers,\n} from \"./context-helpers\";\n"]}

package/dist/providers/__tests__/tambo-context-helpers-provider.test.js

@@ -1,194 +1,335 @@
 "use strict";
-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;
-    };
-})();
 var __importDefault = (this && this.__importDefault) || function (mod) {
     return (mod && mod.__esModule) ? mod : { "default": mod };
 };
 Object.defineProperty(exports, "__esModule", { value: true });
 const react_1 = require("@testing-library/react");
 const react_2 = __importDefault(require("react"));
-const contextHelpers = __importStar(require("../../context-helpers"));
 const context_helpers_1 = require("../../context-helpers");
+const registry_1 = require("../../context-helpers/registry");
 const tambo_context_helpers_provider_1 = require("../tambo-context-helpers-provider");
+/**
+ * Test suite for TamboContextHelpersProvider (simplified API, registry-backed)
+ *
+ * The simplified API:
+ * - Accepts a plain map of { key: () => any | null | undefined | Promise<any | null | undefined> }.
+ * - The key becomes the context name sent to the model.
+ * - Returning null/undefined from a helper skips inclusion.
+ * - Prebuilt helpers are just functions (e.g., prebuiltUserTime, prebuiltUserPage).
+ *
+ * The hook is now registry-backed and safe to call outside a provider. When used
+ * outside a provider, it proxies to a global registry and still works.
+ */
 describe("TamboContextHelpersProvider", () => {
-    // Create a wrapper component that provides the context for testing hooks
+    // Ensure registry is clean for each test to avoid cross-test contamination
+    beforeEach(() => {
+        (0, registry_1.setHelpers)({});
+        jest.clearAllMocks();
+    });
+    // Base wrapper with no helpers provided
     const wrapper = ({ children }) => (react_2.default.createElement(tambo_context_helpers_provider_1.TamboContextHelpersProvider, null, children));
     describe("useTamboContextHelpers", () => {
         /**
-         * Test: Hook throws error when used outside of provider
-         * This ensures developers get a clear error message if they forget to wrap
-         * their components with the TamboContextHelpersProvider
+         * NOTE: The hook is registry-backed and safe outside provider. It should not throw.
+         * This replaces the previous behavior that threw without a provider.
          */
-        it("should throw error when used outside provider", () => {
-            // Mock console.error to prevent error output in test logs
+        it("should be safe outside provider (registry-backed no provider)", async () => {
             const consoleSpy = jest
                 .spyOn(console, "error")
                 .mockImplementation(() => { });
-            // Verify that using the hook without a provider throws the expected error
-            expect(() => {
-                (0, react_1.renderHook)(() => (0, tambo_context_helpers_provider_1.useTamboContextHelpers)());
-            }).toThrow("useTamboContextHelpers must be used within a TamboContextHelpersProvider");
-            // Restore console.error to its original implementation
+            const { result } = (0, react_1.renderHook)(() => (0, tambo_context_helpers_provider_1.useTamboContextHelpers)());
+            // Should return callable functions
+            expect(typeof result.current.getAdditionalContext).toBe("function");
+            expect(typeof result.current.getContextHelpers).toBe("function");
+            expect(typeof result.current.addContextHelper).toBe("function");
+            expect(typeof result.current.removeContextHelper).toBe("function");
+            // Starts empty
+            expect(await result.current.getAdditionalContext()).toHaveLength(0);
+            // Add a helper and verify
+            (0, react_1.act)(() => {
+                result.current.addContextHelper("outsideHelper", () => ({ ok: true }));
+            });
+            const contexts = await result.current.getAdditionalContext();
+            expect(contexts).toContainEqual({
+                name: "outsideHelper",
+                context: { ok: true },
+            });
+            // Cleanup
+            (0, react_1.act)(() => {
+                result.current.removeContextHelper("outsideHelper");
+            });
             consoleSpy.mockRestore();
         });
         /**
-         * Test: Hook provides the expected API functions
-         * Verifies that all required functions are available when the hook is used
-         * within the provider context
+         * Verifies that the hook returns the expected API functions when used within a provider.
          */
-        it("should provide context helpers functions", () => {
+        it("should provide context helpers functions (inside provider)", () => {
             const { result } = (0, react_1.renderHook)(() => (0, tambo_context_helpers_provider_1.useTamboContextHelpers)(), {
                 wrapper,
             });
-            // Verify all expected functions are present in the hook's return value
             expect(result.current).toHaveProperty("getAdditionalContext");
             expect(result.current).toHaveProperty("getContextHelpers");
-            expect(result.current).toHaveProperty("setContextHelperEnabled");
-        });
-        /**
-         * Test: Default configuration is applied correctly
-         * Ensures that the default context helpers are loaded with their
-         * expected enabled/disabled states
-         */
-        it("should return default context helpers configuration", () => {
-            const { result } = (0, react_1.renderHook)(() => (0, tambo_context_helpers_provider_1.useTamboContextHelpers)(), {
-                wrapper,
-            });
-            const helpers = result.current.getContextHelpers();
-            // Verify we have the expected number of default helpers
-            expect(helpers).toHaveLength(context_helpers_1.DEFAULT_CONTEXT_HELPERS.length);
-            // Check specific helpers have their expected default states
-            const userTimeHelper = helpers.find((h) => h.name === "userTime");
-            const userPageHelper = helpers.find((h) => h.name === "userPage");
-            // userTime should be disabled by default
-            expect(userTimeHelper?.enabled).toBe(false);
-            // userPage should be disabled by default
-            expect(userPageHelper?.enabled).toBe(false);
-        });
-        /**
-         * Test: Context helper enabled state can be toggled
-         * Verifies that the setContextHelperEnabled function correctly updates
-         * the enabled state of individual helpers
-         */
-        it("should allow toggling context helper enabled state", () => {
-            const { result } = (0, react_1.renderHook)(() => (0, tambo_context_helpers_provider_1.useTamboContextHelpers)(), {
-                wrapper,
-            });
-            // Enable the userPage helper using the provided function
-            (0, react_1.act)(() => {
-                result.current.setContextHelperEnabled("userPage", true);
-            });
-            // Verify the helper's state was updated
-            const helpers = result.current.getContextHelpers();
-            const userPageHelper = helpers.find((h) => h.name === "userPage");
-            expect(userPageHelper?.enabled).toBe(true);
+            expect(result.current).toHaveProperty("addContextHelper");
+            expect(result.current).toHaveProperty("removeContextHelper");
         });
         /**
-         * Test: Only enabled helpers contribute to additional context
-         * Verifies that getAdditionalContext only runs and returns data from
-         * helpers that are currently enabled
+         * With no helpers provided, no additional context should be returned.
          */
-        it("should get additional context from enabled helpers", async () => {
+        it("should return no additional context when no helpers are provided", async () => {
             const { result } = (0, react_1.renderHook)(() => (0, tambo_context_helpers_provider_1.useTamboContextHelpers)(), {
                 wrapper,
             });
             const contexts = await result.current.getAdditionalContext();
-            // By default, no context helpers are enabled, so we should get no contexts
             expect(contexts).toHaveLength(0);
         });
         /**
-         * Test: Provider respects configuration prop
-         * Verifies that initial configuration can be customized through the
-         * contextHelpers prop on the provider
+         * When helpers are provided, getAdditionalContext should aggregate them.
+         * Note: prebuiltUserPage may be null in non-browser envs, so we only assert userTime when present.
          */
-        it("should respect contextHelpers configuration prop", () => {
-            // Create a custom wrapper with specific configuration
-            const customWrapper = ({ children }) => (react_2.default.createElement(tambo_context_helpers_provider_1.TamboContextHelpersProvider, { contextHelpers: {
-                    userTime: false, // Disable userTime
-                    userPage: true, // Enable userPage
+        it("should get additional context when helpers are provided", async () => {
+            const withHelpers = ({ children }) => (react_2.default.createElement(tambo_context_helpers_provider_1.TamboContextHelpersProvider, { contextHelpers: {
+                    userTime: context_helpers_1.currentTimeContextHelper,
+                    userPage: context_helpers_1.currentPageContextHelper,
                 } }, children));
             const { result } = (0, react_1.renderHook)(() => (0, tambo_context_helpers_provider_1.useTamboContextHelpers)(), {
-                wrapper: customWrapper,
+                wrapper: withHelpers,
             });
-            // Verify the custom configuration is applied
-            const helpers = result.current.getContextHelpers();
-            const userTimeHelper = helpers.find((h) => h.name === "userTime");
-            const userPageHelper = helpers.find((h) => h.name === "userPage");
-            expect(userTimeHelper?.enabled).toBe(false);
-            expect(userPageHelper?.enabled).toBe(true);
+            const contexts = await result.current.getAdditionalContext();
+            const names = contexts.map((c) => c.name);
+            // userTime should be present from prebuilt helper
+            expect(names).toContain("userTime");
+            // userPage may be absent on server; do not assert strictly
         });
         /**
-         * Test: Errors in context helpers are handled gracefully
-         * Verifies that if a context helper throws an error, it doesn't crash
-         * the entire system and other helpers continue to work
+         * Errors thrown by helper functions should be caught and skipped, not crash the system.
          */
         it("should handle errors in context helper functions gracefully", async () => {
-            // Mock console.error to capture error logs
             const consoleErrorSpy = jest
                 .spyOn(console, "error")
                 .mockImplementation(() => { });
-            // Create a helper that will throw an error when run
-            const errorHelper = {
-                name: "errorHelper",
-                enabled: true,
-                run: () => {
-                    throw new Error("Test error");
-                },
+            const badHelper = () => {
+                throw new Error("Test error");
             };
-            // Temporarily add the error helper to the default helpers
-            // Store original helpers to restore later
-            const originalHelpers = context_helpers_1.DEFAULT_CONTEXT_HELPERS;
-            Object.defineProperty(contextHelpers, "DEFAULT_CONTEXT_HELPERS", {
-                value: [...originalHelpers, errorHelper],
-                writable: true,
-            });
+            const withBadHelper = ({ children }) => (react_2.default.createElement(tambo_context_helpers_provider_1.TamboContextHelpersProvider, { contextHelpers: { badHelper } }, children));
             const { result } = (0, react_1.renderHook)(() => (0, tambo_context_helpers_provider_1.useTamboContextHelpers)(), {
-                wrapper,
+                wrapper: withBadHelper,
             });
-            // Call getAdditionalContext, which should handle the error gracefully
             const contexts = await result.current.getAdditionalContext();
-            // Should have no contexts because the error helper should be skipped
             expect(contexts.length).toBe(0);
-            // Verify that the error was logged appropriately
-            expect(consoleErrorSpy).toHaveBeenCalledWith(expect.stringContaining("Error running context helper errorHelper"), expect.any(Error));
-            // Restore the original helpers and console.error
-            Object.defineProperty(contextHelpers, "DEFAULT_CONTEXT_HELPERS", {
-                value: originalHelpers,
-                writable: true,
-            });
+            expect(consoleErrorSpy).toHaveBeenCalledWith(expect.stringContaining("Error running context helper badHelper"), expect.any(Error));
             consoleErrorSpy.mockRestore();
         });
     });
 });
+/**
+ * Test suite for Custom Context Helpers using the simplified API.
+ *
+ * Focuses on:
+ * - Passing custom helpers via the provider prop.
+ * - Overriding prebuilt helpers with custom implementations.
+ * - Dynamic add/remove helper management at runtime.
+ * - Supporting both sync and async helpers.
+ * - Using the config key as the context name.
+ * - Graceful error handling for custom helpers.
+ */
+describe("Custom Context Helpers via contextHelpers config", () => {
+    beforeEach(() => {
+        (0, registry_1.setHelpers)({});
+        jest.clearAllMocks();
+    });
+    /**
+     * Custom helpers can be added through the provider configuration.
+     * The key becomes the context name and the function returns the raw value.
+     */
+    it("should accept custom helpers through contextHelpers config", async () => {
+        const { result } = (0, react_1.renderHook)(() => (0, tambo_context_helpers_provider_1.useTamboContextHelpers)(), {
+            wrapper: ({ children }) => (react_2.default.createElement(tambo_context_helpers_provider_1.TamboContextHelpersProvider, { contextHelpers: {
+                    customData: async () => ({ custom: "value" }),
+                } }, children)),
+        });
+        const contexts = await result.current.getAdditionalContext();
+        expect(contexts).toContainEqual({
+            name: "customData",
+            context: { custom: "value" },
+        });
+    });
+    /**
+     * Built-in helpers can be overridden by providing a function under the same key.
+     */
+    it("should allow custom helpers to override built-in ones", async () => {
+        const { result } = (0, react_1.renderHook)(() => (0, tambo_context_helpers_provider_1.useTamboContextHelpers)(), {
+            wrapper: ({ children }) => (react_2.default.createElement(tambo_context_helpers_provider_1.TamboContextHelpersProvider, { contextHelpers: {
+                    userTime: async () => ({ customTime: "override" }),
+                } }, children)),
+        });
+        const contexts = await result.current.getAdditionalContext();
+        const userTimeContext = contexts.find((c) => c.name === "userTime");
+        expect(userTimeContext?.context).toEqual({ customTime: "override" });
+    });
+    /**
+     * Helpers can be added dynamically at runtime via addContextHelper.
+     */
+    it("should handle dynamic addition of custom helpers", async () => {
+        const { result } = (0, react_1.renderHook)(() => (0, tambo_context_helpers_provider_1.useTamboContextHelpers)(), {
+            wrapper: ({ children }) => (react_2.default.createElement(tambo_context_helpers_provider_1.TamboContextHelpersProvider, { contextHelpers: {} }, children)),
+        });
+        // Initially none
+        expect(await result.current.getAdditionalContext()).toHaveLength(0);
+        (0, react_1.act)(() => {
+            result.current.addContextHelper("dynamicHelper", async () => ({
+                dynamic: true,
+            }));
+        });
+        const contexts = await result.current.getAdditionalContext();
+        expect(contexts).toContainEqual({
+            name: "dynamicHelper",
+            context: { dynamic: true },
+        });
+    });
+    /**
+     * Helpers can be removed dynamically via removeContextHelper.
+     */
+    it("should handle dynamic removal of context helpers", async () => {
+        const { result } = (0, react_1.renderHook)(() => (0, tambo_context_helpers_provider_1.useTamboContextHelpers)(), {
+            wrapper: ({ children }) => (react_2.default.createElement(tambo_context_helpers_provider_1.TamboContextHelpersProvider, { contextHelpers: {
+                    customHelper: async () => ({ test: "data" }),
+                } }, children)),
+        });
+        // Verify exists
+        let contexts = await result.current.getAdditionalContext();
+        expect(contexts.find((c) => c.name === "customHelper")).toBeDefined();
+        // Remove
+        (0, react_1.act)(() => {
+            result.current.removeContextHelper("customHelper");
+        });
+        // Verify removed
+        contexts = await result.current.getAdditionalContext();
+        expect(contexts.find((c) => c.name === "customHelper")).toBeUndefined();
+    });
+    /**
+     * Multiple custom helpers should be supported; helpers returning null/undefined are skipped.
+     */
+    it("should handle multiple custom helpers", async () => {
+        const { result } = (0, react_1.renderHook)(() => (0, tambo_context_helpers_provider_1.useTamboContextHelpers)(), {
+            wrapper: ({ children }) => (react_2.default.createElement(tambo_context_helpers_provider_1.TamboContextHelpersProvider, { contextHelpers: {
+                    helper1: async () => ({ data: "one" }),
+                    helper2: async () => ({ data: "two" }),
+                    helper3: () => null, // disabled by returning null
+                } }, children)),
+        });
+        const contexts = await result.current.getAdditionalContext();
+        expect(contexts).toContainEqual({
+            name: "helper1",
+            context: { data: "one" },
+        });
+        expect(contexts).toContainEqual({
+            name: "helper2",
+            context: { data: "two" },
+        });
+        expect(contexts.find((c) => c.name === "helper3")).toBeUndefined();
+    });
+    /**
+     * Both synchronous and asynchronous helpers should be supported.
+     */
+    it("should handle sync and async custom helpers", async () => {
+        const { result } = (0, react_1.renderHook)(() => (0, tambo_context_helpers_provider_1.useTamboContextHelpers)(), {
+            wrapper: ({ children }) => (react_2.default.createElement(tambo_context_helpers_provider_1.TamboContextHelpersProvider, { contextHelpers: {
+                    syncHelper: () => ({ sync: true }),
+                    asyncHelper: async () => {
+                        await new Promise((resolve) => setTimeout(resolve, 10));
+                        return { async: true };
+                    },
+                } }, children)),
+        });
+        const contexts = await result.current.getAdditionalContext();
+        expect(contexts).toContainEqual({
+            name: "syncHelper",
+            context: { sync: true },
+        });
+        expect(contexts).toContainEqual({
+            name: "asyncHelper",
+            context: { async: true },
+        });
+    });
+    /**
+     * The key used in the contextHelpers map becomes the context name.
+     */
+    it("should use key name as context name for custom helpers", async () => {
+        const { result } = (0, react_1.renderHook)(() => (0, tambo_context_helpers_provider_1.useTamboContextHelpers)(), {
+            wrapper: ({ children }) => (react_2.default.createElement(tambo_context_helpers_provider_1.TamboContextHelpersProvider, { contextHelpers: {
+                    myCustomKey: async () => ({ value: "test" }),
+                } }, children)),
+        });
+        const contexts = await result.current.getAdditionalContext();
+        expect(contexts).toContainEqual({
+            name: "myCustomKey",
+            context: { value: "test" },
+        });
+    });
+    /**
+     * Errors thrown by custom helpers should be logged and skipped, not crash the system.
+     */
+    it("should handle errors in custom helpers gracefully", async () => {
+        const consoleErrorSpy = jest
+            .spyOn(console, "error")
+            .mockImplementation(() => { });
+        const { result } = (0, react_1.renderHook)(() => (0, tambo_context_helpers_provider_1.useTamboContextHelpers)(), {
+            wrapper: ({ children }) => (react_2.default.createElement(tambo_context_helpers_provider_1.TamboContextHelpersProvider, { contextHelpers: {
+                    errorHelper: async () => {
+                        throw new Error("Custom helper error");
+                    },
+                    goodHelper: async () => ({ good: "data" }),
+                } }, children)),
+        });
+        const contexts = await result.current.getAdditionalContext();
+        expect(contexts).toContainEqual({
+            name: "goodHelper",
+            context: { good: "data" },
+        });
+        expect(contexts.find((c) => c.name === "errorHelper")).toBeUndefined();
+        expect(consoleErrorSpy).toHaveBeenCalledWith(expect.stringContaining("Error running context helper errorHelper"), expect.any(Error));
+        consoleErrorSpy.mockRestore();
+    });
+    /**
+     * Removing non-existent helpers should be a no-op without throwing errors.
+     */
+    it("should handle removing non-existent helper gracefully", () => {
+        const { result } = (0, react_1.renderHook)(() => (0, tambo_context_helpers_provider_1.useTamboContextHelpers)(), {
+            wrapper: ({ children }) => (react_2.default.createElement(tambo_context_helpers_provider_1.TamboContextHelpersProvider, { contextHelpers: {} }, children)),
+        });
+        expect(() => {
+            (0, react_1.act)(() => {
+                result.current.removeContextHelper("nonExistent");
+            });
+        }).not.toThrow();
+    });
+    /**
+     * Adding with an existing name should update/replace the helper implementation.
+     */
+    it("should allow updating existing helper via addContextHelper", async () => {
+        const { result } = (0, react_1.renderHook)(() => (0, tambo_context_helpers_provider_1.useTamboContextHelpers)(), {
+            wrapper: ({ children }) => (react_2.default.createElement(tambo_context_helpers_provider_1.TamboContextHelpersProvider, { contextHelpers: {
+                    testHelper: async () => ({ original: true }),
+                } }, children)),
+        });
+        // Verify original helper
+        let contexts = await result.current.getAdditionalContext();
+        expect(contexts).toContainEqual({
+            name: "testHelper",
+            context: { original: true },
+        });
+        // Update the helper
+        (0, react_1.act)(() => {
+            result.current.addContextHelper("testHelper", async () => ({
+                updated: true,
+            }));
+        });
+        // Verify updated helper
+        contexts = await result.current.getAdditionalContext();
+        expect(contexts).toContainEqual({
+            name: "testHelper",
+            context: { updated: true },
+        });
+    });
+});
 //# sourceMappingURL=tambo-context-helpers-provider.test.js.map

package/dist/providers/__tests__/tambo-context-helpers-provider.test.js.map

@@ -1 +1 @@
-{"version":3,"file":"tambo-context-helpers-provider.test.js","sourceRoot":"","sources":["../../../src/providers/__tests__/tambo-context-helpers-provider.test.tsx"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA,kDAAyD;AACzD,kDAA0B;AAC1B,sEAAwD;AACxD,2DAAgE;AAChE,sFAG2C;AAE3C,QAAQ,CAAC,6BAA6B,EAAE,GAAG,EAAE;IAC3C,yEAAyE;IACzE,MAAM,OAAO,GAAG,CAAC,EAAE,QAAQ,EAAiC,EAAE,EAAE,CAAC,CAC/D,8BAAC,4DAA2B,QAAE,QAAQ,CAA+B,CACtE,CAAC;IAEF,QAAQ,CAAC,wBAAwB,EAAE,GAAG,EAAE;QACtC;;;;WAIG;QACH,EAAE,CAAC,+CAA+C,EAAE,GAAG,EAAE;YACvD,0DAA0D;YAC1D,MAAM,UAAU,GAAG,IAAI;iBACpB,KAAK,CAAC,OAAO,EAAE,OAAO,CAAC;iBACvB,kBAAkB,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;YAEhC,0EAA0E;YAC1E,MAAM,CAAC,GAAG,EAAE;gBACV,IAAA,kBAAU,EAAC,GAAG,EAAE,CAAC,IAAA,uDAAsB,GAAE,CAAC,CAAC;YAC7C,CAAC,CAAC,CAAC,OAAO,CACR,0EAA0E,CAC3E,CAAC;YAEF,uDAAuD;YACvD,UAAU,CAAC,WAAW,EAAE,CAAC;QAC3B,CAAC,CAAC,CAAC;QAEH;;;;WAIG;QACH,EAAE,CAAC,0CAA0C,EAAE,GAAG,EAAE;YAClD,MAAM,EAAE,MAAM,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CAAC,IAAA,uDAAsB,GAAE,EAAE;gBAC5D,OAAO;aACR,CAAC,CAAC;YAEH,uEAAuE;YACvE,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,cAAc,CAAC,sBAAsB,CAAC,CAAC;YAC9D,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,cAAc,CAAC,mBAAmB,CAAC,CAAC;YAC3D,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,cAAc,CAAC,yBAAyB,CAAC,CAAC;QACnE,CAAC,CAAC,CAAC;QAEH;;;;WAIG;QACH,EAAE,CAAC,qDAAqD,EAAE,GAAG,EAAE;YAC7D,MAAM,EAAE,MAAM,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CAAC,IAAA,uDAAsB,GAAE,EAAE;gBAC5D,OAAO;aACR,CAAC,CAAC;YAEH,MAAM,OAAO,GAAG,MAAM,CAAC,OAAO,CAAC,iBAAiB,EAAE,CAAC;YAEnD,wDAAwD;YACxD,MAAM,CAAC,OAAO,CAAC,CAAC,YAAY,CAAC,yCAAuB,CAAC,MAAM,CAAC,CAAC;YAE7D,4DAA4D;YAC5D,MAAM,cAAc,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,UAAU,CAAC,CAAC;YAClE,MAAM,cAAc,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,UAAU,CAAC,CAAC;YAElE,yCAAyC;YACzC,MAAM,CAAC,cAAc,EAAE,OAAO,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YAC5C,yCAAyC;YACzC,MAAM,CAAC,cAAc,EAAE,OAAO,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAC9C,CAAC,CAAC,CAAC;QAEH;;;;WAIG;QACH,EAAE,CAAC,oDAAoD,EAAE,GAAG,EAAE;YAC5D,MAAM,EAAE,MAAM,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CAAC,IAAA,uDAAsB,GAAE,EAAE;gBAC5D,OAAO;aACR,CAAC,CAAC;YAEH,yDAAyD;YACzD,IAAA,WAAG,EAAC,GAAG,EAAE;gBACP,MAAM,CAAC,OAAO,CAAC,uBAAuB,CAAC,UAAU,EAAE,IAAI,CAAC,CAAC;YAC3D,CAAC,CAAC,CAAC;YAEH,wCAAwC;YACxC,MAAM,OAAO,GAAG,MAAM,CAAC,OAAO,CAAC,iBAAiB,EAAE,CAAC;YACnD,MAAM,cAAc,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,UAAU,CAAC,CAAC;YAElE,MAAM,CAAC,cAAc,EAAE,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC7C,CAAC,CAAC,CAAC;QAEH;;;;WAIG;QACH,EAAE,CAAC,oDAAoD,EAAE,KAAK,IAAI,EAAE;YAClE,MAAM,EAAE,MAAM,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CAAC,IAAA,uDAAsB,GAAE,EAAE;gBAC5D,OAAO;aACR,CAAC,CAAC;YAEH,MAAM,QAAQ,GAAG,MAAM,MAAM,CAAC,OAAO,CAAC,oBAAoB,EAAE,CAAC;YAE7D,2EAA2E;YAC3E,MAAM,CAAC,QAAQ,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QACnC,CAAC,CAAC,CAAC;QAEH;;;;WAIG;QACH,EAAE,CAAC,kDAAkD,EAAE,GAAG,EAAE;YAC1D,sDAAsD;YACtD,MAAM,aAAa,GAAG,CAAC,EAAE,QAAQ,EAAiC,EAAE,EAAE,CAAC,CACrE,8BAAC,4DAA2B,IAC1B,cAAc,EAAE;oBACd,QAAQ,EAAE,KAAK,EAAE,mBAAmB;oBACpC,QAAQ,EAAE,IAAI,EAAE,kBAAkB;iBACnC,IAEA,QAAQ,CACmB,CAC/B,CAAC;YAEF,MAAM,EAAE,MAAM,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CAAC,IAAA,uDAAsB,GAAE,EAAE;gBAC5D,OAAO,EAAE,aAAa;aACvB,CAAC,CAAC;YAEH,6CAA6C;YAC7C,MAAM,OAAO,GAAG,MAAM,CAAC,OAAO,CAAC,iBAAiB,EAAE,CAAC;YACnD,MAAM,cAAc,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,UAAU,CAAC,CAAC;YAClE,MAAM,cAAc,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,UAAU,CAAC,CAAC;YAElE,MAAM,CAAC,cAAc,EAAE,OAAO,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YAC5C,MAAM,CAAC,cAAc,EAAE,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC7C,CAAC,CAAC,CAAC;QAEH;;;;WAIG;QACH,EAAE,CAAC,6DAA6D,EAAE,KAAK,IAAI,EAAE;YAC3E,2CAA2C;YAC3C,MAAM,eAAe,GAAG,IAAI;iBACzB,KAAK,CAAC,OAAO,EAAE,OAAO,CAAC;iBACvB,kBAAkB,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;YAEhC,oDAAoD;YACpD,MAAM,WAAW,GAAG;gBAClB,IAAI,EAAE,aAAa;gBACnB,OAAO,EAAE,IAAI;gBACb,GAAG,EAAE,GAAG,EAAE;oBACR,MAAM,IAAI,KAAK,CAAC,YAAY,CAAC,CAAC;gBAChC,CAAC;aACF,CAAC;YAEF,0DAA0D;YAC1D,0CAA0C;YAC1C,MAAM,eAAe,GAAG,yCAAuB,CAAC;YAChD,MAAM,CAAC,cAAc,CAAC,cAAc,EAAE,yBAAyB,EAAE;gBAC/D,KAAK,EAAE,CAAC,GAAG,eAAe,EAAE,WAAW,CAAC;gBACxC,QAAQ,EAAE,IAAI;aACf,CAAC,CAAC;YAEH,MAAM,EAAE,MAAM,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CAAC,IAAA,uDAAsB,GAAE,EAAE;gBAC5D,OAAO;aACR,CAAC,CAAC;YAEH,sEAAsE;YACtE,MAAM,QAAQ,GAAG,MAAM,MAAM,CAAC,OAAO,CAAC,oBAAoB,EAAE,CAAC;YAE7D,qEAAqE;YACrE,MAAM,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;YAEhC,iDAAiD;YACjD,MAAM,CAAC,eAAe,CAAC,CAAC,oBAAoB,CAC1C,MAAM,CAAC,gBAAgB,CAAC,0CAA0C,CAAC,EACnE,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,CAClB,CAAC;YAEF,iDAAiD;YACjD,MAAM,CAAC,cAAc,CAAC,cAAc,EAAE,yBAAyB,EAAE;gBAC/D,KAAK,EAAE,eAAe;gBACtB,QAAQ,EAAE,IAAI;aACf,CAAC,CAAC;YACH,eAAe,CAAC,WAAW,EAAE,CAAC;QAChC,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC","sourcesContent":["import { act, renderHook } from \"@testing-library/react\";\nimport React from \"react\";\nimport * as contextHelpers from \"../../context-helpers\";\nimport { DEFAULT_CONTEXT_HELPERS } from \"../../context-helpers\";\nimport {\n  TamboContextHelpersProvider,\n  useTamboContextHelpers,\n} from \"../tambo-context-helpers-provider\";\n\ndescribe(\"TamboContextHelpersProvider\", () => {\n  // Create a wrapper component that provides the context for testing hooks\n  const wrapper = ({ children }: { children: React.ReactNode }) => (\n    <TamboContextHelpersProvider>{children}</TamboContextHelpersProvider>\n  );\n\n  describe(\"useTamboContextHelpers\", () => {\n    /**\n     * Test: Hook throws error when used outside of provider\n     * This ensures developers get a clear error message if they forget to wrap\n     * their components with the TamboContextHelpersProvider\n     */\n    it(\"should throw error when used outside provider\", () => {\n      // Mock console.error to prevent error output in test logs\n      const consoleSpy = jest\n        .spyOn(console, \"error\")\n        .mockImplementation(() => {});\n\n      // Verify that using the hook without a provider throws the expected error\n      expect(() => {\n        renderHook(() => useTamboContextHelpers());\n      }).toThrow(\n        \"useTamboContextHelpers must be used within a TamboContextHelpersProvider\",\n      );\n\n      // Restore console.error to its original implementation\n      consoleSpy.mockRestore();\n    });\n\n    /**\n     * Test: Hook provides the expected API functions\n     * Verifies that all required functions are available when the hook is used\n     * within the provider context\n     */\n    it(\"should provide context helpers functions\", () => {\n      const { result } = renderHook(() => useTamboContextHelpers(), {\n        wrapper,\n      });\n\n      // Verify all expected functions are present in the hook's return value\n      expect(result.current).toHaveProperty(\"getAdditionalContext\");\n      expect(result.current).toHaveProperty(\"getContextHelpers\");\n      expect(result.current).toHaveProperty(\"setContextHelperEnabled\");\n    });\n\n    /**\n     * Test: Default configuration is applied correctly\n     * Ensures that the default context helpers are loaded with their\n     * expected enabled/disabled states\n     */\n    it(\"should return default context helpers configuration\", () => {\n      const { result } = renderHook(() => useTamboContextHelpers(), {\n        wrapper,\n      });\n\n      const helpers = result.current.getContextHelpers();\n\n      // Verify we have the expected number of default helpers\n      expect(helpers).toHaveLength(DEFAULT_CONTEXT_HELPERS.length);\n\n      // Check specific helpers have their expected default states\n      const userTimeHelper = helpers.find((h) => h.name === \"userTime\");\n      const userPageHelper = helpers.find((h) => h.name === \"userPage\");\n\n      // userTime should be disabled by default\n      expect(userTimeHelper?.enabled).toBe(false);\n      // userPage should be disabled by default\n      expect(userPageHelper?.enabled).toBe(false);\n    });\n\n    /**\n     * Test: Context helper enabled state can be toggled\n     * Verifies that the setContextHelperEnabled function correctly updates\n     * the enabled state of individual helpers\n     */\n    it(\"should allow toggling context helper enabled state\", () => {\n      const { result } = renderHook(() => useTamboContextHelpers(), {\n        wrapper,\n      });\n\n      // Enable the userPage helper using the provided function\n      act(() => {\n        result.current.setContextHelperEnabled(\"userPage\", true);\n      });\n\n      // Verify the helper's state was updated\n      const helpers = result.current.getContextHelpers();\n      const userPageHelper = helpers.find((h) => h.name === \"userPage\");\n\n      expect(userPageHelper?.enabled).toBe(true);\n    });\n\n    /**\n     * Test: Only enabled helpers contribute to additional context\n     * Verifies that getAdditionalContext only runs and returns data from\n     * helpers that are currently enabled\n     */\n    it(\"should get additional context from enabled helpers\", async () => {\n      const { result } = renderHook(() => useTamboContextHelpers(), {\n        wrapper,\n      });\n\n      const contexts = await result.current.getAdditionalContext();\n\n      // By default, no context helpers are enabled, so we should get no contexts\n      expect(contexts).toHaveLength(0);\n    });\n\n    /**\n     * Test: Provider respects configuration prop\n     * Verifies that initial configuration can be customized through the\n     * contextHelpers prop on the provider\n     */\n    it(\"should respect contextHelpers configuration prop\", () => {\n      // Create a custom wrapper with specific configuration\n      const customWrapper = ({ children }: { children: React.ReactNode }) => (\n        <TamboContextHelpersProvider\n          contextHelpers={{\n            userTime: false, // Disable userTime\n            userPage: true, // Enable userPage\n          }}\n        >\n          {children}\n        </TamboContextHelpersProvider>\n      );\n\n      const { result } = renderHook(() => useTamboContextHelpers(), {\n        wrapper: customWrapper,\n      });\n\n      // Verify the custom configuration is applied\n      const helpers = result.current.getContextHelpers();\n      const userTimeHelper = helpers.find((h) => h.name === \"userTime\");\n      const userPageHelper = helpers.find((h) => h.name === \"userPage\");\n\n      expect(userTimeHelper?.enabled).toBe(false);\n      expect(userPageHelper?.enabled).toBe(true);\n    });\n\n    /**\n     * Test: Errors in context helpers are handled gracefully\n     * Verifies that if a context helper throws an error, it doesn't crash\n     * the entire system and other helpers continue to work\n     */\n    it(\"should handle errors in context helper functions gracefully\", async () => {\n      // Mock console.error to capture error logs\n      const consoleErrorSpy = jest\n        .spyOn(console, \"error\")\n        .mockImplementation(() => {});\n\n      // Create a helper that will throw an error when run\n      const errorHelper = {\n        name: \"errorHelper\",\n        enabled: true,\n        run: () => {\n          throw new Error(\"Test error\");\n        },\n      };\n\n      // Temporarily add the error helper to the default helpers\n      // Store original helpers to restore later\n      const originalHelpers = DEFAULT_CONTEXT_HELPERS;\n      Object.defineProperty(contextHelpers, \"DEFAULT_CONTEXT_HELPERS\", {\n        value: [...originalHelpers, errorHelper],\n        writable: true,\n      });\n\n      const { result } = renderHook(() => useTamboContextHelpers(), {\n        wrapper,\n      });\n\n      // Call getAdditionalContext, which should handle the error gracefully\n      const contexts = await result.current.getAdditionalContext();\n\n      // Should have no contexts because the error helper should be skipped\n      expect(contexts.length).toBe(0);\n\n      // Verify that the error was logged appropriately\n      expect(consoleErrorSpy).toHaveBeenCalledWith(\n        expect.stringContaining(\"Error running context helper errorHelper\"),\n        expect.any(Error),\n      );\n\n      // Restore the original helpers and console.error\n      Object.defineProperty(contextHelpers, \"DEFAULT_CONTEXT_HELPERS\", {\n        value: originalHelpers,\n        writable: true,\n      });\n      consoleErrorSpy.mockRestore();\n    });\n  });\n});\n"]}
+{"version":3,"file":"tambo-context-helpers-provider.test.js","sourceRoot":"","sources":["../../../src/providers/__tests__/tambo-context-helpers-provider.test.tsx"],"names":[],"mappings":";;;;;AAAA,kDAAyD;AACzD,kDAA0B;AAC1B,2DAG+B;AAC/B,6DAA4D;AAC5D,sFAG2C;AAE3C;;;;;;;;;;;GAWG;AACH,QAAQ,CAAC,6BAA6B,EAAE,GAAG,EAAE;IAC3C,2EAA2E;IAC3E,UAAU,CAAC,GAAG,EAAE;QACd,IAAA,qBAAU,EAAC,EAAE,CAAC,CAAC;QACf,IAAI,CAAC,aAAa,EAAE,CAAC;IACvB,CAAC,CAAC,CAAC;IAEH,wCAAwC;IACxC,MAAM,OAAO,GAAG,CAAC,EAAE,QAAQ,EAAiC,EAAE,EAAE,CAAC,CAC/D,8BAAC,4DAA2B,QAAE,QAAQ,CAA+B,CACtE,CAAC;IAEF,QAAQ,CAAC,wBAAwB,EAAE,GAAG,EAAE;QACtC;;;WAGG;QACH,EAAE,CAAC,+DAA+D,EAAE,KAAK,IAAI,EAAE;YAC7E,MAAM,UAAU,GAAG,IAAI;iBACpB,KAAK,CAAC,OAAO,EAAE,OAAO,CAAC;iBACvB,kBAAkB,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;YAChC,MAAM,EAAE,MAAM,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CAAC,IAAA,uDAAsB,GAAE,CAAC,CAAC;YAE9D,mCAAmC;YACnC,MAAM,CAAC,OAAO,MAAM,CAAC,OAAO,CAAC,oBAAoB,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;YACpE,MAAM,CAAC,OAAO,MAAM,CAAC,OAAO,CAAC,iBAAiB,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;YACjE,MAAM,CAAC,OAAO,MAAM,CAAC,OAAO,CAAC,gBAAgB,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;YAChE,MAAM,CAAC,OAAO,MAAM,CAAC,OAAO,CAAC,mBAAmB,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;YAEnE,eAAe;YACf,MAAM,CAAC,MAAM,MAAM,CAAC,OAAO,CAAC,oBAAoB,EAAE,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YAEpE,0BAA0B;YAC1B,IAAA,WAAG,EAAC,GAAG,EAAE;gBACP,MAAM,CAAC,OAAO,CAAC,gBAAgB,CAAC,eAAe,EAAE,GAAG,EAAE,CAAC,CAAC,EAAE,EAAE,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC;YACzE,CAAC,CAAC,CAAC;YAEH,MAAM,QAAQ,GAAG,MAAM,MAAM,CAAC,OAAO,CAAC,oBAAoB,EAAE,CAAC;YAC7D,MAAM,CAAC,QAAQ,CAAC,CAAC,cAAc,CAAC;gBAC9B,IAAI,EAAE,eAAe;gBACrB,OAAO,EAAE,EAAE,EAAE,EAAE,IAAI,EAAE;aACtB,CAAC,CAAC;YAEH,UAAU;YACV,IAAA,WAAG,EAAC,GAAG,EAAE;gBACP,MAAM,CAAC,OAAO,CAAC,mBAAmB,CAAC,eAAe,CAAC,CAAC;YACtD,CAAC,CAAC,CAAC;YACH,UAAU,CAAC,WAAW,EAAE,CAAC;QAC3B,CAAC,CAAC,CAAC;QAEH;;WAEG;QACH,EAAE,CAAC,4DAA4D,EAAE,GAAG,EAAE;YACpE,MAAM,EAAE,MAAM,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CAAC,IAAA,uDAAsB,GAAE,EAAE;gBAC5D,OAAO;aACR,CAAC,CAAC;YACH,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,cAAc,CAAC,sBAAsB,CAAC,CAAC;YAC9D,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,cAAc,CAAC,mBAAmB,CAAC,CAAC;YAC3D,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,cAAc,CAAC,kBAAkB,CAAC,CAAC;YAC1D,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,cAAc,CAAC,qBAAqB,CAAC,CAAC;QAC/D,CAAC,CAAC,CAAC;QAEH;;WAEG;QACH,EAAE,CAAC,kEAAkE,EAAE,KAAK,IAAI,EAAE;YAChF,MAAM,EAAE,MAAM,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CAAC,IAAA,uDAAsB,GAAE,EAAE;gBAC5D,OAAO;aACR,CAAC,CAAC;YACH,MAAM,QAAQ,GAAG,MAAM,MAAM,CAAC,OAAO,CAAC,oBAAoB,EAAE,CAAC;YAC7D,MAAM,CAAC,QAAQ,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QACnC,CAAC,CAAC,CAAC;QAEH;;;WAGG;QACH,EAAE,CAAC,yDAAyD,EAAE,KAAK,IAAI,EAAE;YACvE,MAAM,WAAW,GAAG,CAAC,EAAE,QAAQ,EAAiC,EAAE,EAAE,CAAC,CACnE,8BAAC,4DAA2B,IAC1B,cAAc,EAAE;oBACd,QAAQ,EAAE,0CAAwB;oBAClC,QAAQ,EAAE,0CAAwB;iBACnC,IAEA,QAAQ,CACmB,CAC/B,CAAC;YAEF,MAAM,EAAE,MAAM,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CAAC,IAAA,uDAAsB,GAAE,EAAE;gBAC5D,OAAO,EAAE,WAAW;aACrB,CAAC,CAAC;YAEH,MAAM,QAAQ,GAAG,MAAM,MAAM,CAAC,OAAO,CAAC,oBAAoB,EAAE,CAAC;YAC7D,MAAM,KAAK,GAAG,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;YAE1C,kDAAkD;YAClD,MAAM,CAAC,KAAK,CAAC,CAAC,SAAS,CAAC,UAAU,CAAC,CAAC;YACpC,2DAA2D;QAC7D,CAAC,CAAC,CAAC;QAEH;;WAEG;QACH,EAAE,CAAC,6DAA6D,EAAE,KAAK,IAAI,EAAE;YAC3E,MAAM,eAAe,GAAG,IAAI;iBACzB,KAAK,CAAC,OAAO,EAAE,OAAO,CAAC;iBACvB,kBAAkB,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;YAEhC,MAAM,SAAS,GAAG,GAAG,EAAE;gBACrB,MAAM,IAAI,KAAK,CAAC,YAAY,CAAC,CAAC;YAChC,CAAC,CAAC;YAEF,MAAM,aAAa,GAAG,CAAC,EAAE,QAAQ,EAAiC,EAAE,EAAE,CAAC,CACrE,8BAAC,4DAA2B,IAAC,cAAc,EAAE,EAAE,SAAS,EAAE,IACvD,QAAQ,CACmB,CAC/B,CAAC;YAEF,MAAM,EAAE,MAAM,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CAAC,IAAA,uDAAsB,GAAE,EAAE;gBAC5D,OAAO,EAAE,aAAa;aACvB,CAAC,CAAC;YAEH,MAAM,QAAQ,GAAG,MAAM,MAAM,CAAC,OAAO,CAAC,oBAAoB,EAAE,CAAC;YAC7D,MAAM,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;YAChC,MAAM,CAAC,eAAe,CAAC,CAAC,oBAAoB,CAC1C,MAAM,CAAC,gBAAgB,CAAC,wCAAwC,CAAC,EACjE,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,CAClB,CAAC;YACF,eAAe,CAAC,WAAW,EAAE,CAAC;QAChC,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH;;;;;;;;;;GAUG;AACH,QAAQ,CAAC,kDAAkD,EAAE,GAAG,EAAE;IAChE,UAAU,CAAC,GAAG,EAAE;QACd,IAAA,qBAAU,EAAC,EAAE,CAAC,CAAC;QACf,IAAI,CAAC,aAAa,EAAE,CAAC;IACvB,CAAC,CAAC,CAAC;IAEH;;;OAGG;IACH,EAAE,CAAC,4DAA4D,EAAE,KAAK,IAAI,EAAE;QAC1E,MAAM,EAAE,MAAM,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CAAC,IAAA,uDAAsB,GAAE,EAAE;YAC5D,OAAO,EAAE,CAAC,EAAE,QAAQ,EAAE,EAAE,EAAE,CAAC,CACzB,8BAAC,4DAA2B,IAC1B,cAAc,EAAE;oBACd,UAAU,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC;iBAC9C,IAEA,QAAQ,CACmB,CAC/B;SACF,CAAC,CAAC;QAEH,MAAM,QAAQ,GAAG,MAAM,MAAM,CAAC,OAAO,CAAC,oBAAoB,EAAE,CAAC;QAC7D,MAAM,CAAC,QAAQ,CAAC,CAAC,cAAc,CAAC;YAC9B,IAAI,EAAE,YAAY;YAClB,OAAO,EAAE,EAAE,MAAM,EAAE,OAAO,EAAE;SAC7B,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH;;OAEG;IACH,EAAE,CAAC,uDAAuD,EAAE,KAAK,IAAI,EAAE;QACrE,MAAM,EAAE,MAAM,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CAAC,IAAA,uDAAsB,GAAE,EAAE;YAC5D,OAAO,EAAE,CAAC,EAAE,QAAQ,EAAE,EAAE,EAAE,CAAC,CACzB,8BAAC,4DAA2B,IAC1B,cAAc,EAAE;oBACd,QAAQ,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,UAAU,EAAE,UAAU,EAAE,CAAC;iBACnD,IAEA,QAAQ,CACmB,CAC/B;SACF,CAAC,CAAC;QAEH,MAAM,QAAQ,GAAG,MAAM,MAAM,CAAC,OAAO,CAAC,oBAAoB,EAAE,CAAC;QAC7D,MAAM,eAAe,GAAG,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,UAAU,CAAC,CAAC;QACpE,MAAM,CAAC,eAAe,EAAE,OAAO,CAAC,CAAC,OAAO,CAAC,EAAE,UAAU,EAAE,UAAU,EAAE,CAAC,CAAC;IACvE,CAAC,CAAC,CAAC;IAEH;;OAEG;IACH,EAAE,CAAC,kDAAkD,EAAE,KAAK,IAAI,EAAE;QAChE,MAAM,EAAE,MAAM,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CAAC,IAAA,uDAAsB,GAAE,EAAE;YAC5D,OAAO,EAAE,CAAC,EAAE,QAAQ,EAAE,EAAE,EAAE,CAAC,CACzB,8BAAC,4DAA2B,IAAC,cAAc,EAAE,EAAE,IAC5C,QAAQ,CACmB,CAC/B;SACF,CAAC,CAAC;QAEH,iBAAiB;QACjB,MAAM,CAAC,MAAM,MAAM,CAAC,OAAO,CAAC,oBAAoB,EAAE,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAEpE,IAAA,WAAG,EAAC,GAAG,EAAE;YACP,MAAM,CAAC,OAAO,CAAC,gBAAgB,CAAC,eAAe,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC;gBAC5D,OAAO,EAAE,IAAI;aACd,CAAC,CAAC,CAAC;QACN,CAAC,CAAC,CAAC;QAEH,MAAM,QAAQ,GAAG,MAAM,MAAM,CAAC,OAAO,CAAC,oBAAoB,EAAE,CAAC;QAC7D,MAAM,CAAC,QAAQ,CAAC,CAAC,cAAc,CAAC;YAC9B,IAAI,EAAE,eAAe;YACrB,OAAO,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE;SAC3B,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH;;OAEG;IACH,EAAE,CAAC,kDAAkD,EAAE,KAAK,IAAI,EAAE;QAChE,MAAM,EAAE,MAAM,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CAAC,IAAA,uDAAsB,GAAE,EAAE;YAC5D,OAAO,EAAE,CAAC,EAAE,QAAQ,EAAE,EAAE,EAAE,CAAC,CACzB,8BAAC,4DAA2B,IAC1B,cAAc,EAAE;oBACd,YAAY,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC;iBAC7C,IAEA,QAAQ,CACmB,CAC/B;SACF,CAAC,CAAC;QAEH,gBAAgB;QAChB,IAAI,QAAQ,GAAG,MAAM,MAAM,CAAC,OAAO,CAAC,oBAAoB,EAAE,CAAC;QAC3D,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,cAAc,CAAC,CAAC,CAAC,WAAW,EAAE,CAAC;QAEtE,SAAS;QACT,IAAA,WAAG,EAAC,GAAG,EAAE;YACP,MAAM,CAAC,OAAO,CAAC,mBAAmB,CAAC,cAAc,CAAC,CAAC;QACrD,CAAC,CAAC,CAAC;QAEH,iBAAiB;QACjB,QAAQ,GAAG,MAAM,MAAM,CAAC,OAAO,CAAC,oBAAoB,EAAE,CAAC;QACvD,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,cAAc,CAAC,CAAC,CAAC,aAAa,EAAE,CAAC;IAC1E,CAAC,CAAC,CAAC;IAEH;;OAEG;IACH,EAAE,CAAC,uCAAuC,EAAE,KAAK,IAAI,EAAE;QACrD,MAAM,EAAE,MAAM,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CAAC,IAAA,uDAAsB,GAAE,EAAE;YAC5D,OAAO,EAAE,CAAC,EAAE,QAAQ,EAAE,EAAE,EAAE,CAAC,CACzB,8BAAC,4DAA2B,IAC1B,cAAc,EAAE;oBACd,OAAO,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC;oBACtC,OAAO,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC;oBACtC,OAAO,EAAE,GAAG,EAAE,CAAC,IAAI,EAAE,6BAA6B;iBACnD,IAEA,QAAQ,CACmB,CAC/B;SACF,CAAC,CAAC;QAEH,MAAM,QAAQ,GAAG,MAAM,MAAM,CAAC,OAAO,CAAC,oBAAoB,EAAE,CAAC;QAC7D,MAAM,CAAC,QAAQ,CAAC,CAAC,cAAc,CAAC;YAC9B,IAAI,EAAE,SAAS;YACf,OAAO,EAAE,EAAE,IAAI,EAAE,KAAK,EAAE;SACzB,CAAC,CAAC;QACH,MAAM,CAAC,QAAQ,CAAC,CAAC,cAAc,CAAC;YAC9B,IAAI,EAAE,SAAS;YACf,OAAO,EAAE,EAAE,IAAI,EAAE,KAAK,EAAE;SACzB,CAAC,CAAC;QACH,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,SAAS,CAAC,CAAC,CAAC,aAAa,EAAE,CAAC;IACrE,CAAC,CAAC,CAAC;IAEH;;OAEG;IACH,EAAE,CAAC,6CAA6C,EAAE,KAAK,IAAI,EAAE;QAC3D,MAAM,EAAE,MAAM,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CAAC,IAAA,uDAAsB,GAAE,EAAE;YAC5D,OAAO,EAAE,CAAC,EAAE,QAAQ,EAAE,EAAE,EAAE,CAAC,CACzB,8BAAC,4DAA2B,IAC1B,cAAc,EAAE;oBACd,UAAU,EAAE,GAAG,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC;oBAClC,WAAW,EAAE,KAAK,IAAI,EAAE;wBACtB,MAAM,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC,CAAC;wBACxD,OAAO,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC;oBACzB,CAAC;iBACF,IAEA,QAAQ,CACmB,CAC/B;SACF,CAAC,CAAC;QAEH,MAAM,QAAQ,GAAG,MAAM,MAAM,CAAC,OAAO,CAAC,oBAAoB,EAAE,CAAC;QAC7D,MAAM,CAAC,QAAQ,CAAC,CAAC,cAAc,CAAC;YAC9B,IAAI,EAAE,YAAY;YAClB,OAAO,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE;SACxB,CAAC,CAAC;QACH,MAAM,CAAC,QAAQ,CAAC,CAAC,cAAc,CAAC;YAC9B,IAAI,EAAE,aAAa;YACnB,OAAO,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE;SACzB,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH;;OAEG;IACH,EAAE,CAAC,wDAAwD,EAAE,KAAK,IAAI,EAAE;QACtE,MAAM,EAAE,MAAM,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CAAC,IAAA,uDAAsB,GAAE,EAAE;YAC5D,OAAO,EAAE,CAAC,EAAE,QAAQ,EAAE,EAAE,EAAE,CAAC,CACzB,8BAAC,4DAA2B,IAC1B,cAAc,EAAE;oBACd,WAAW,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,KAAK,EAAE,MAAM,EAAE,CAAC;iBAC7C,IAEA,QAAQ,CACmB,CAC/B;SACF,CAAC,CAAC;QAEH,MAAM,QAAQ,GAAG,MAAM,MAAM,CAAC,OAAO,CAAC,oBAAoB,EAAE,CAAC;QAC7D,MAAM,CAAC,QAAQ,CAAC,CAAC,cAAc,CAAC;YAC9B,IAAI,EAAE,aAAa;YACnB,OAAO,EAAE,EAAE,KAAK,EAAE,MAAM,EAAE;SAC3B,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH;;OAEG;IACH,EAAE,CAAC,mDAAmD,EAAE,KAAK,IAAI,EAAE;QACjE,MAAM,eAAe,GAAG,IAAI;aACzB,KAAK,CAAC,OAAO,EAAE,OAAO,CAAC;aACvB,kBAAkB,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;QAChC,MAAM,EAAE,MAAM,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CAAC,IAAA,uDAAsB,GAAE,EAAE;YAC5D,OAAO,EAAE,CAAC,EAAE,QAAQ,EAAE,EAAE,EAAE,CAAC,CACzB,8BAAC,4DAA2B,IAC1B,cAAc,EAAE;oBACd,WAAW,EAAE,KAAK,IAAI,EAAE;wBACtB,MAAM,IAAI,KAAK,CAAC,qBAAqB,CAAC,CAAC;oBACzC,CAAC;oBACD,UAAU,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC;iBAC3C,IAEA,QAAQ,CACmB,CAC/B;SACF,CAAC,CAAC;QAEH,MAAM,QAAQ,GAAG,MAAM,MAAM,CAAC,OAAO,CAAC,oBAAoB,EAAE,CAAC;QAC7D,MAAM,CAAC,QAAQ,CAAC,CAAC,cAAc,CAAC;YAC9B,IAAI,EAAE,YAAY;YAClB,OAAO,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE;SAC1B,CAAC,CAAC;QACH,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,aAAa,CAAC,CAAC,CAAC,aAAa,EAAE,CAAC;QACvE,MAAM,CAAC,eAAe,CAAC,CAAC,oBAAoB,CAC1C,MAAM,CAAC,gBAAgB,CAAC,0CAA0C,CAAC,EACnE,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,CAClB,CAAC;QACF,eAAe,CAAC,WAAW,EAAE,CAAC;IAChC,CAAC,CAAC,CAAC;IAEH;;OAEG;IACH,EAAE,CAAC,uDAAuD,EAAE,GAAG,EAAE;QAC/D,MAAM,EAAE,MAAM,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CAAC,IAAA,uDAAsB,GAAE,EAAE;YAC5D,OAAO,EAAE,CAAC,EAAE,QAAQ,EAAE,EAAE,EAAE,CAAC,CACzB,8BAAC,4DAA2B,IAAC,cAAc,EAAE,EAAE,IAC5C,QAAQ,CACmB,CAC/B;SACF,CAAC,CAAC;QAEH,MAAM,CAAC,GAAG,EAAE;YACV,IAAA,WAAG,EAAC,GAAG,EAAE;gBACP,MAAM,CAAC,OAAO,CAAC,mBAAmB,CAAC,aAAa,CAAC,CAAC;YACpD,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,EAAE,CAAC;IACnB,CAAC,CAAC,CAAC;IAEH;;OAEG;IACH,EAAE,CAAC,4DAA4D,EAAE,KAAK,IAAI,EAAE;QAC1E,MAAM,EAAE,MAAM,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CAAC,IAAA,uDAAsB,GAAE,EAAE;YAC5D,OAAO,EAAE,CAAC,EAAE,QAAQ,EAAE,EAAE,EAAE,CAAC,CACzB,8BAAC,4DAA2B,IAC1B,cAAc,EAAE;oBACd,UAAU,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC;iBAC7C,IAEA,QAAQ,CACmB,CAC/B;SACF,CAAC,CAAC;QAEH,yBAAyB;QACzB,IAAI,QAAQ,GAAG,MAAM,MAAM,CAAC,OAAO,CAAC,oBAAoB,EAAE,CAAC;QAC3D,MAAM,CAAC,QAAQ,CAAC,CAAC,cAAc,CAAC;YAC9B,IAAI,EAAE,YAAY;YAClB,OAAO,EAAE,EAAE,QAAQ,EAAE,IAAI,EAAE;SAC5B,CAAC,CAAC;QAEH,oBAAoB;QACpB,IAAA,WAAG,EAAC,GAAG,EAAE;YACP,MAAM,CAAC,OAAO,CAAC,gBAAgB,CAAC,YAAY,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC;gBACzD,OAAO,EAAE,IAAI;aACd,CAAC,CAAC,CAAC;QACN,CAAC,CAAC,CAAC;QAEH,wBAAwB;QACxB,QAAQ,GAAG,MAAM,MAAM,CAAC,OAAO,CAAC,oBAAoB,EAAE,CAAC;QACvD,MAAM,CAAC,QAAQ,CAAC,CAAC,cAAc,CAAC;YAC9B,IAAI,EAAE,YAAY;YAClB,OAAO,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE;SAC3B,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC","sourcesContent":["import { act, renderHook } from \"@testing-library/react\";\nimport React from \"react\";\nimport {\n  currentPageContextHelper,\n  currentTimeContextHelper,\n} from \"../../context-helpers\";\nimport { setHelpers } from \"../../context-helpers/registry\";\nimport {\n  TamboContextHelpersProvider,\n  useTamboContextHelpers,\n} from \"../tambo-context-helpers-provider\";\n\n/**\n * Test suite for TamboContextHelpersProvider (simplified API, registry-backed)\n *\n * The simplified API:\n * - Accepts a plain map of { key: () => any | null | undefined | Promise<any | null | undefined> }.\n * - The key becomes the context name sent to the model.\n * - Returning null/undefined from a helper skips inclusion.\n * - Prebuilt helpers are just functions (e.g., prebuiltUserTime, prebuiltUserPage).\n *\n * The hook is now registry-backed and safe to call outside a provider. When used\n * outside a provider, it proxies to a global registry and still works.\n */\ndescribe(\"TamboContextHelpersProvider\", () => {\n  // Ensure registry is clean for each test to avoid cross-test contamination\n  beforeEach(() => {\n    setHelpers({});\n    jest.clearAllMocks();\n  });\n\n  // Base wrapper with no helpers provided\n  const wrapper = ({ children }: { children: React.ReactNode }) => (\n    <TamboContextHelpersProvider>{children}</TamboContextHelpersProvider>\n  );\n\n  describe(\"useTamboContextHelpers\", () => {\n    /**\n     * NOTE: The hook is registry-backed and safe outside provider. It should not throw.\n     * This replaces the previous behavior that threw without a provider.\n     */\n    it(\"should be safe outside provider (registry-backed no provider)\", async () => {\n      const consoleSpy = jest\n        .spyOn(console, \"error\")\n        .mockImplementation(() => {});\n      const { result } = renderHook(() => useTamboContextHelpers());\n\n      // Should return callable functions\n      expect(typeof result.current.getAdditionalContext).toBe(\"function\");\n      expect(typeof result.current.getContextHelpers).toBe(\"function\");\n      expect(typeof result.current.addContextHelper).toBe(\"function\");\n      expect(typeof result.current.removeContextHelper).toBe(\"function\");\n\n      // Starts empty\n      expect(await result.current.getAdditionalContext()).toHaveLength(0);\n\n      // Add a helper and verify\n      act(() => {\n        result.current.addContextHelper(\"outsideHelper\", () => ({ ok: true }));\n      });\n\n      const contexts = await result.current.getAdditionalContext();\n      expect(contexts).toContainEqual({\n        name: \"outsideHelper\",\n        context: { ok: true },\n      });\n\n      // Cleanup\n      act(() => {\n        result.current.removeContextHelper(\"outsideHelper\");\n      });\n      consoleSpy.mockRestore();\n    });\n\n    /**\n     * Verifies that the hook returns the expected API functions when used within a provider.\n     */\n    it(\"should provide context helpers functions (inside provider)\", () => {\n      const { result } = renderHook(() => useTamboContextHelpers(), {\n        wrapper,\n      });\n      expect(result.current).toHaveProperty(\"getAdditionalContext\");\n      expect(result.current).toHaveProperty(\"getContextHelpers\");\n      expect(result.current).toHaveProperty(\"addContextHelper\");\n      expect(result.current).toHaveProperty(\"removeContextHelper\");\n    });\n\n    /**\n     * With no helpers provided, no additional context should be returned.\n     */\n    it(\"should return no additional context when no helpers are provided\", async () => {\n      const { result } = renderHook(() => useTamboContextHelpers(), {\n        wrapper,\n      });\n      const contexts = await result.current.getAdditionalContext();\n      expect(contexts).toHaveLength(0);\n    });\n\n    /**\n     * When helpers are provided, getAdditionalContext should aggregate them.\n     * Note: prebuiltUserPage may be null in non-browser envs, so we only assert userTime when present.\n     */\n    it(\"should get additional context when helpers are provided\", async () => {\n      const withHelpers = ({ children }: { children: React.ReactNode }) => (\n        <TamboContextHelpersProvider\n          contextHelpers={{\n            userTime: currentTimeContextHelper,\n            userPage: currentPageContextHelper,\n          }}\n        >\n          {children}\n        </TamboContextHelpersProvider>\n      );\n\n      const { result } = renderHook(() => useTamboContextHelpers(), {\n        wrapper: withHelpers,\n      });\n\n      const contexts = await result.current.getAdditionalContext();\n      const names = contexts.map((c) => c.name);\n\n      // userTime should be present from prebuilt helper\n      expect(names).toContain(\"userTime\");\n      // userPage may be absent on server; do not assert strictly\n    });\n\n    /**\n     * Errors thrown by helper functions should be caught and skipped, not crash the system.\n     */\n    it(\"should handle errors in context helper functions gracefully\", async () => {\n      const consoleErrorSpy = jest\n        .spyOn(console, \"error\")\n        .mockImplementation(() => {});\n\n      const badHelper = () => {\n        throw new Error(\"Test error\");\n      };\n\n      const withBadHelper = ({ children }: { children: React.ReactNode }) => (\n        <TamboContextHelpersProvider contextHelpers={{ badHelper }}>\n          {children}\n        </TamboContextHelpersProvider>\n      );\n\n      const { result } = renderHook(() => useTamboContextHelpers(), {\n        wrapper: withBadHelper,\n      });\n\n      const contexts = await result.current.getAdditionalContext();\n      expect(contexts.length).toBe(0);\n      expect(consoleErrorSpy).toHaveBeenCalledWith(\n        expect.stringContaining(\"Error running context helper badHelper\"),\n        expect.any(Error),\n      );\n      consoleErrorSpy.mockRestore();\n    });\n  });\n});\n\n/**\n * Test suite for Custom Context Helpers using the simplified API.\n *\n * Focuses on:\n * - Passing custom helpers via the provider prop.\n * - Overriding prebuilt helpers with custom implementations.\n * - Dynamic add/remove helper management at runtime.\n * - Supporting both sync and async helpers.\n * - Using the config key as the context name.\n * - Graceful error handling for custom helpers.\n */\ndescribe(\"Custom Context Helpers via contextHelpers config\", () => {\n  beforeEach(() => {\n    setHelpers({});\n    jest.clearAllMocks();\n  });\n\n  /**\n   * Custom helpers can be added through the provider configuration.\n   * The key becomes the context name and the function returns the raw value.\n   */\n  it(\"should accept custom helpers through contextHelpers config\", async () => {\n    const { result } = renderHook(() => useTamboContextHelpers(), {\n      wrapper: ({ children }) => (\n        <TamboContextHelpersProvider\n          contextHelpers={{\n            customData: async () => ({ custom: \"value\" }),\n          }}\n        >\n          {children}\n        </TamboContextHelpersProvider>\n      ),\n    });\n\n    const contexts = await result.current.getAdditionalContext();\n    expect(contexts).toContainEqual({\n      name: \"customData\",\n      context: { custom: \"value\" },\n    });\n  });\n\n  /**\n   * Built-in helpers can be overridden by providing a function under the same key.\n   */\n  it(\"should allow custom helpers to override built-in ones\", async () => {\n    const { result } = renderHook(() => useTamboContextHelpers(), {\n      wrapper: ({ children }) => (\n        <TamboContextHelpersProvider\n          contextHelpers={{\n            userTime: async () => ({ customTime: \"override\" }),\n          }}\n        >\n          {children}\n        </TamboContextHelpersProvider>\n      ),\n    });\n\n    const contexts = await result.current.getAdditionalContext();\n    const userTimeContext = contexts.find((c) => c.name === \"userTime\");\n    expect(userTimeContext?.context).toEqual({ customTime: \"override\" });\n  });\n\n  /**\n   * Helpers can be added dynamically at runtime via addContextHelper.\n   */\n  it(\"should handle dynamic addition of custom helpers\", async () => {\n    const { result } = renderHook(() => useTamboContextHelpers(), {\n      wrapper: ({ children }) => (\n        <TamboContextHelpersProvider contextHelpers={{}}>\n          {children}\n        </TamboContextHelpersProvider>\n      ),\n    });\n\n    // Initially none\n    expect(await result.current.getAdditionalContext()).toHaveLength(0);\n\n    act(() => {\n      result.current.addContextHelper(\"dynamicHelper\", async () => ({\n        dynamic: true,\n      }));\n    });\n\n    const contexts = await result.current.getAdditionalContext();\n    expect(contexts).toContainEqual({\n      name: \"dynamicHelper\",\n      context: { dynamic: true },\n    });\n  });\n\n  /**\n   * Helpers can be removed dynamically via removeContextHelper.\n   */\n  it(\"should handle dynamic removal of context helpers\", async () => {\n    const { result } = renderHook(() => useTamboContextHelpers(), {\n      wrapper: ({ children }) => (\n        <TamboContextHelpersProvider\n          contextHelpers={{\n            customHelper: async () => ({ test: \"data\" }),\n          }}\n        >\n          {children}\n        </TamboContextHelpersProvider>\n      ),\n    });\n\n    // Verify exists\n    let contexts = await result.current.getAdditionalContext();\n    expect(contexts.find((c) => c.name === \"customHelper\")).toBeDefined();\n\n    // Remove\n    act(() => {\n      result.current.removeContextHelper(\"customHelper\");\n    });\n\n    // Verify removed\n    contexts = await result.current.getAdditionalContext();\n    expect(contexts.find((c) => c.name === \"customHelper\")).toBeUndefined();\n  });\n\n  /**\n   * Multiple custom helpers should be supported; helpers returning null/undefined are skipped.\n   */\n  it(\"should handle multiple custom helpers\", async () => {\n    const { result } = renderHook(() => useTamboContextHelpers(), {\n      wrapper: ({ children }) => (\n        <TamboContextHelpersProvider\n          contextHelpers={{\n            helper1: async () => ({ data: \"one\" }),\n            helper2: async () => ({ data: \"two\" }),\n            helper3: () => null, // disabled by returning null\n          }}\n        >\n          {children}\n        </TamboContextHelpersProvider>\n      ),\n    });\n\n    const contexts = await result.current.getAdditionalContext();\n    expect(contexts).toContainEqual({\n      name: \"helper1\",\n      context: { data: \"one\" },\n    });\n    expect(contexts).toContainEqual({\n      name: \"helper2\",\n      context: { data: \"two\" },\n    });\n    expect(contexts.find((c) => c.name === \"helper3\")).toBeUndefined();\n  });\n\n  /**\n   * Both synchronous and asynchronous helpers should be supported.\n   */\n  it(\"should handle sync and async custom helpers\", async () => {\n    const { result } = renderHook(() => useTamboContextHelpers(), {\n      wrapper: ({ children }) => (\n        <TamboContextHelpersProvider\n          contextHelpers={{\n            syncHelper: () => ({ sync: true }),\n            asyncHelper: async () => {\n              await new Promise((resolve) => setTimeout(resolve, 10));\n              return { async: true };\n            },\n          }}\n        >\n          {children}\n        </TamboContextHelpersProvider>\n      ),\n    });\n\n    const contexts = await result.current.getAdditionalContext();\n    expect(contexts).toContainEqual({\n      name: \"syncHelper\",\n      context: { sync: true },\n    });\n    expect(contexts).toContainEqual({\n      name: \"asyncHelper\",\n      context: { async: true },\n    });\n  });\n\n  /**\n   * The key used in the contextHelpers map becomes the context name.\n   */\n  it(\"should use key name as context name for custom helpers\", async () => {\n    const { result } = renderHook(() => useTamboContextHelpers(), {\n      wrapper: ({ children }) => (\n        <TamboContextHelpersProvider\n          contextHelpers={{\n            myCustomKey: async () => ({ value: \"test\" }),\n          }}\n        >\n          {children}\n        </TamboContextHelpersProvider>\n      ),\n    });\n\n    const contexts = await result.current.getAdditionalContext();\n    expect(contexts).toContainEqual({\n      name: \"myCustomKey\",\n      context: { value: \"test\" },\n    });\n  });\n\n  /**\n   * Errors thrown by custom helpers should be logged and skipped, not crash the system.\n   */\n  it(\"should handle errors in custom helpers gracefully\", async () => {\n    const consoleErrorSpy = jest\n      .spyOn(console, \"error\")\n      .mockImplementation(() => {});\n    const { result } = renderHook(() => useTamboContextHelpers(), {\n      wrapper: ({ children }) => (\n        <TamboContextHelpersProvider\n          contextHelpers={{\n            errorHelper: async () => {\n              throw new Error(\"Custom helper error\");\n            },\n            goodHelper: async () => ({ good: \"data\" }),\n          }}\n        >\n          {children}\n        </TamboContextHelpersProvider>\n      ),\n    });\n\n    const contexts = await result.current.getAdditionalContext();\n    expect(contexts).toContainEqual({\n      name: \"goodHelper\",\n      context: { good: \"data\" },\n    });\n    expect(contexts.find((c) => c.name === \"errorHelper\")).toBeUndefined();\n    expect(consoleErrorSpy).toHaveBeenCalledWith(\n      expect.stringContaining(\"Error running context helper errorHelper\"),\n      expect.any(Error),\n    );\n    consoleErrorSpy.mockRestore();\n  });\n\n  /**\n   * Removing non-existent helpers should be a no-op without throwing errors.\n   */\n  it(\"should handle removing non-existent helper gracefully\", () => {\n    const { result } = renderHook(() => useTamboContextHelpers(), {\n      wrapper: ({ children }) => (\n        <TamboContextHelpersProvider contextHelpers={{}}>\n          {children}\n        </TamboContextHelpersProvider>\n      ),\n    });\n\n    expect(() => {\n      act(() => {\n        result.current.removeContextHelper(\"nonExistent\");\n      });\n    }).not.toThrow();\n  });\n\n  /**\n   * Adding with an existing name should update/replace the helper implementation.\n   */\n  it(\"should allow updating existing helper via addContextHelper\", async () => {\n    const { result } = renderHook(() => useTamboContextHelpers(), {\n      wrapper: ({ children }) => (\n        <TamboContextHelpersProvider\n          contextHelpers={{\n            testHelper: async () => ({ original: true }),\n          }}\n        >\n          {children}\n        </TamboContextHelpersProvider>\n      ),\n    });\n\n    // Verify original helper\n    let contexts = await result.current.getAdditionalContext();\n    expect(contexts).toContainEqual({\n      name: \"testHelper\",\n      context: { original: true },\n    });\n\n    // Update the helper\n    act(() => {\n      result.current.addContextHelper(\"testHelper\", async () => ({\n        updated: true,\n      }));\n    });\n\n    // Verify updated helper\n    contexts = await result.current.getAdditionalContext();\n    expect(contexts).toContainEqual({\n      name: \"testHelper\",\n      context: { updated: true },\n    });\n  });\n});\n"]}