@tambo-ai/react 0.41.2 → 0.42.1

package/esm/index.js

@@ -6,9 +6,11 @@ export * from "./hooks/use-suggestions";
 export { useTamboStreamStatus, } from "./hooks/use-tambo-stream-status";
 export { useTamboThreadInput } from "./hooks/use-thread-input";
 // Re-export provider components
-export { TamboClientProvider, TamboComponentProvider, TamboPropStreamProvider, TamboProvider, TamboStubProvider, TamboThreadProvider, useTambo, useTamboClient, useTamboStream, useTamboThread, } from "./providers";
+export { TamboClientProvider, TamboComponentProvider, TamboContextHelpersProvider, TamboPropStreamProvider, TamboProvider, TamboStubProvider, TamboThreadProvider, useTambo, useTamboClient, useTamboContextHelpers, useTamboStream, useTamboThread, } from "./providers";
 export { useTamboThreadList } from "./hooks/use-tambo-threads";
 export { GenerationStage, } from "./model/generate-component-response";
 export { withTamboInteractable as withInteractable, } from "./providers/hoc/with-tambo-interactable";
 export { useTamboInteractable } from "./providers/tambo-interactable-provider";
+// Context helpers exports
+export { DEFAULT_CONTEXT_HELPERS, getUserPageContext, getUserTimeContext, } from "./context-helpers";
 //# sourceMappingURL=index.js.map

package/esm/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,wKAAwK;AAExK,OAAO,EAAE,sBAAsB,EAAE,MAAM,6BAA6B,CAAC;AACrE,OAAO,EACL,oBAAoB,EACpB,sBAAsB,EACtB,sBAAsB,GACvB,MAAM,6BAA6B,CAAC;AACrC,OAAO,EAAE,sBAAsB,EAAE,MAAM,6BAA6B,CAAC;AACrE,cAAc,yBAAyB,CAAC;AACxC,OAAO,EACL,oBAAoB,GAGrB,MAAM,iCAAiC,CAAC;AACzC,OAAO,EAAE,mBAAmB,EAAE,MAAM,0BAA0B,CAAC;AAE/D,gCAAgC;AAChC,OAAO,EACL,mBAAmB,EACnB,sBAAsB,EACtB,uBAAuB,EACvB,aAAa,EACb,iBAAiB,EACjB,mBAAmB,EACnB,QAAQ,EACR,cAAc,EACd,cAAc,EACd,cAAc,GAIf,MAAM,aAAa,CAAC;AAcrB,OAAO,EAAE,kBAAkB,EAAE,MAAM,2BAA2B,CAAC;AAQ/D,OAAO,EACL,eAAe,GAEhB,MAAM,qCAAqC,CAAC;AAO7C,OAAO,EACL,qBAAqB,IAAI,gBAAgB,GAG1C,MAAM,yCAAyC,CAAC;AACjD,OAAO,EAAE,oBAAoB,EAAE,MAAM,yCAAyC,CAAC","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  TamboPropStreamProvider,\n  TamboProvider,\n  TamboStubProvider,\n  TamboThreadProvider,\n  useTambo,\n  useTamboClient,\n  useTamboStream,\n  useTamboThread,\n  type TamboComponent,\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"]}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,wKAAwK;AAExK,OAAO,EAAE,sBAAsB,EAAE,MAAM,6BAA6B,CAAC;AACrE,OAAO,EACL,oBAAoB,EACpB,sBAAsB,EACtB,sBAAsB,GACvB,MAAM,6BAA6B,CAAC;AACrC,OAAO,EAAE,sBAAsB,EAAE,MAAM,6BAA6B,CAAC;AACrE,cAAc,yBAAyB,CAAC;AACxC,OAAO,EACL,oBAAoB,GAGrB,MAAM,iCAAiC,CAAC;AACzC,OAAO,EAAE,mBAAmB,EAAE,MAAM,0BAA0B,CAAC;AAE/D,gCAAgC;AAChC,OAAO,EACL,mBAAmB,EACnB,sBAAsB,EACtB,2BAA2B,EAC3B,uBAAuB,EACvB,aAAa,EACb,iBAAiB,EACjB,mBAAmB,EACnB,QAAQ,EACR,cAAc,EACd,sBAAsB,EACtB,cAAc,EACd,cAAc,GAMf,MAAM,aAAa,CAAC;AAcrB,OAAO,EAAE,kBAAkB,EAAE,MAAM,2BAA2B,CAAC;AAQ/D,OAAO,EACL,eAAe,GAEhB,MAAM,qCAAqC,CAAC;AAO7C,OAAO,EACL,qBAAqB,IAAI,gBAAgB,GAG1C,MAAM,yCAAyC,CAAC;AACjD,OAAO,EAAE,oBAAoB,EAAE,MAAM,yCAAyC,CAAC;AAE/E,0BAA0B;AAC1B,OAAO,EACL,uBAAuB,EACvB,kBAAkB,EAClB,kBAAkB,GACnB,MAAM,mBAAmB,CAAC","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"]}

package/esm/providers/__tests__/tambo-context-helpers-provider.test.d.ts

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

package/esm/providers/__tests__/tambo-context-helpers-provider.test.d.ts.map

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

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

@@ -0,0 +1,156 @@
+import { act, renderHook } from "@testing-library/react";
+import React from "react";
+import * as contextHelpers from "../../context-helpers";
+import { DEFAULT_CONTEXT_HELPERS } from "../../context-helpers";
+import { TamboContextHelpersProvider, useTamboContextHelpers, } from "../tambo-context-helpers-provider";
+describe("TamboContextHelpersProvider", () => {
+    // Create a wrapper component that provides the context for testing hooks
+    const wrapper = ({ children }) => (React.createElement(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
+         */
+        it("should throw error when used outside provider", () => {
+            // Mock console.error to prevent error output in test logs
+            const consoleSpy = jest
+                .spyOn(console, "error")
+                .mockImplementation(() => { });
+            // Verify that using the hook without a provider throws the expected error
+            expect(() => {
+                renderHook(() => useTamboContextHelpers());
+            }).toThrow("useTamboContextHelpers must be used within a TamboContextHelpersProvider");
+            // Restore console.error to its original implementation
+            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
+         */
+        it("should provide context helpers functions", () => {
+            const { result } = renderHook(() => 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 } = renderHook(() => useTamboContextHelpers(), {
+                wrapper,
+            });
+            const helpers = result.current.getContextHelpers();
+            // Verify we have the expected number of default helpers
+            expect(helpers).toHaveLength(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 } = renderHook(() => useTamboContextHelpers(), {
+                wrapper,
+            });
+            // Enable the userPage helper using the provided function
+            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);
+        });
+        /**
+         * Test: Only enabled helpers contribute to additional context
+         * Verifies that getAdditionalContext only runs and returns data from
+         * helpers that are currently enabled
+         */
+        it("should get additional context from enabled helpers", async () => {
+            const { result } = renderHook(() => 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
+         */
+        it("should respect contextHelpers configuration prop", () => {
+            // Create a custom wrapper with specific configuration
+            const customWrapper = ({ children }) => (React.createElement(TamboContextHelpersProvider, { contextHelpers: {
+                    userTime: false, // Disable userTime
+                    userPage: true, // Enable userPage
+                } }, children));
+            const { result } = renderHook(() => useTamboContextHelpers(), {
+                wrapper: customWrapper,
+            });
+            // 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);
+        });
+        /**
+         * 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
+         */
+        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");
+                },
+            };
+            // Temporarily add the error helper to the default helpers
+            // Store original helpers to restore later
+            const originalHelpers = DEFAULT_CONTEXT_HELPERS;
+            Object.defineProperty(contextHelpers, "DEFAULT_CONTEXT_HELPERS", {
+                value: [...originalHelpers, errorHelper],
+                writable: true,
+            });
+            const { result } = renderHook(() => useTamboContextHelpers(), {
+                wrapper,
+            });
+            // 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,
+            });
+            consoleErrorSpy.mockRestore();
+        });
+    });
+});
+//# sourceMappingURL=tambo-context-helpers-provider.test.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"tambo-context-helpers-provider.test.js","sourceRoot":"","sources":["../../../src/providers/__tests__/tambo-context-helpers-provider.test.tsx"],"names":[],"mappings":"AAAA,OAAO,EAAE,GAAG,EAAE,UAAU,EAAE,MAAM,wBAAwB,CAAC;AACzD,OAAO,KAAK,MAAM,OAAO,CAAC;AAC1B,OAAO,KAAK,cAAc,MAAM,uBAAuB,CAAC;AACxD,OAAO,EAAE,uBAAuB,EAAE,MAAM,uBAAuB,CAAC;AAChE,OAAO,EACL,2BAA2B,EAC3B,sBAAsB,GACvB,MAAM,mCAAmC,CAAC;AAE3C,QAAQ,CAAC,6BAA6B,EAAE,GAAG,EAAE;IAC3C,yEAAyE;IACzE,MAAM,OAAO,GAAG,CAAC,EAAE,QAAQ,EAAiC,EAAE,EAAE,CAAC,CAC/D,oBAAC,2BAA2B,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,UAAU,CAAC,GAAG,EAAE,CAAC,sBAAsB,EAAE,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,UAAU,CAAC,GAAG,EAAE,CAAC,sBAAsB,EAAE,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,UAAU,CAAC,GAAG,EAAE,CAAC,sBAAsB,EAAE,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,uBAAuB,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,UAAU,CAAC,GAAG,EAAE,CAAC,sBAAsB,EAAE,EAAE;gBAC5D,OAAO;aACR,CAAC,CAAC;YAEH,yDAAyD;YACzD,GAAG,CAAC,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,UAAU,CAAC,GAAG,EAAE,CAAC,sBAAsB,EAAE,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,oBAAC,2BAA2B,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,UAAU,CAAC,GAAG,EAAE,CAAC,sBAAsB,EAAE,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,uBAAuB,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,UAAU,CAAC,GAAG,EAAE,CAAC,sBAAsB,EAAE,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"]}

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

@@ -5,6 +5,7 @@ import { z } from "zod";
 import { GenerationStage, } from "../../model/generate-component-response";
 import { serializeRegistry } from "../../testing/tools";
 import { useTamboClient } from "../tambo-client-provider";
+import { TamboContextHelpersProvider } from "../tambo-context-helpers-provider";
 import { TamboRegistryProvider } from "../tambo-registry-provider";
 import { TamboThreadProvider, useTamboThread } from "../tambo-thread-provider";
 // Mock crypto.randomUUID
@@ -20,29 +21,13 @@ jest.mock("../tambo-client-provider", () => ({
 jest.mock("@tambo-ai/typescript-sdk", () => ({
     advanceStream: jest.fn(),
 }));
-// Mock the system context and getCustomContext
+// Mock the getCustomContext
 jest.mock("../../util/registry", () => ({
     ...jest.requireActual("../../util/registry"),
-    getSystemContext: () => ({
-        localTime: "7/24/2025, 6:26:21 PM",
-        timezone: "America/New_York",
-    }),
     getCustomContext: () => ({
         message: "additional instructions",
     }),
 }));
-// Mock the combined context which includes the system context and the custom context
-const mockCombinedContext = {
-    system: {
-        localTime: "7/24/2025, 6:26:21 PM",
-        timezone: "America/New_York",
-    },
-    ...{
-        custom: {
-            message: "additional instructions",
-        },
-    },
-};
 // Test utilities
 const createMockMessage = (overrides = {}) => ({
     id: "test-message-1",
@@ -115,7 +100,8 @@ describe("TamboThreadProvider", () => {
         },
     ];
     const wrapper = ({ children }) => (React.createElement(TamboRegistryProvider, { components: mockRegistry },
-        React.createElement(TamboThreadProvider, { streaming: false }, children)));
+        React.createElement(TamboContextHelpersProvider, { contextHelpers: { userTime: false, userPage: false } },
+            React.createElement(TamboThreadProvider, { streaming: false }, children))));
     beforeEach(() => {
         jest.clearAllMocks();
         jest.mocked(mockThreadsApi.retrieve).mockResolvedValue(mockThread);
@@ -223,7 +209,11 @@ describe("TamboThreadProvider", () => {
             messageToAppend: {
                 content: [{ type: "text", text: "Hello" }],
                 role: "user",
-                additionalContext: mockCombinedContext,
+                additionalContext: {
+                    custom: {
+                        message: "additional instructions",
+                    },
+                },
             },
             availableComponents: serializeRegistry(mockRegistry),
             contextKey: undefined,
@@ -316,7 +306,8 @@ describe("TamboThreadProvider", () => {
         it("should call advanceStream when streamResponse=true", async () => {
             // Use wrapper with streaming=true to show that explicit streamResponse=true works
             const wrapperWithStreaming = ({ children, }) => (React.createElement(TamboRegistryProvider, { components: mockRegistry },
-                React.createElement(TamboThreadProvider, { streaming: true }, children)));
+                React.createElement(TamboContextHelpersProvider, { contextHelpers: { userTime: false, userPage: false } },
+                    React.createElement(TamboThreadProvider, { streaming: true }, children))));
             const mockStreamResponse = {
                 responseMessageDto: {
                     id: "stream-response",
@@ -354,7 +345,11 @@ describe("TamboThreadProvider", () => {
                 messageToAppend: {
                     content: [{ type: "text", text: "Hello streaming" }],
                     role: "user",
-                    additionalContext: mockCombinedContext,
+                    additionalContext: {
+                        custom: {
+                            message: "additional instructions",
+                        },
+                    },
                 },
                 availableComponents: serializeRegistry(mockRegistry),
                 contextKey: undefined,
@@ -369,7 +364,8 @@ describe("TamboThreadProvider", () => {
         it("should call advanceById when streamResponse=false for existing thread", async () => {
             // Use wrapper with streaming=true to show that explicit streamResponse=false overrides provider setting
             const wrapperWithStreaming = ({ children, }) => (React.createElement(TamboRegistryProvider, { components: mockRegistry },
-                React.createElement(TamboThreadProvider, { streaming: true }, children)));
+                React.createElement(TamboContextHelpersProvider, { contextHelpers: { userTime: false, userPage: false } },
+                    React.createElement(TamboThreadProvider, { streaming: true }, children))));
             const { result } = renderHook(() => useTamboThread(), {
                 wrapper: wrapperWithStreaming,
             });
@@ -388,7 +384,11 @@ describe("TamboThreadProvider", () => {
                 messageToAppend: {
                     content: [{ type: "text", text: "Hello non-streaming" }],
                     role: "user",
-                    additionalContext: mockCombinedContext,
+                    additionalContext: {
+                        custom: {
+                            message: "additional instructions",
+                        },
+                    },
                 },
                 availableComponents: serializeRegistry(mockRegistry),
                 contextKey: undefined,
@@ -403,7 +403,8 @@ describe("TamboThreadProvider", () => {
         it("should call advanceById when streamResponse is undefined and provider streaming=false", async () => {
             // Use wrapper with streaming=false to test that undefined streamResponse respects provider setting
             const wrapperWithoutStreaming = ({ children, }) => (React.createElement(TamboRegistryProvider, { components: mockRegistry },
-                React.createElement(TamboThreadProvider, { streaming: false }, children)));
+                React.createElement(TamboContextHelpersProvider, { contextHelpers: { userTime: false, userPage: false } },
+                    React.createElement(TamboThreadProvider, { streaming: false }, children))));
             const { result } = renderHook(() => useTamboThread(), {
                 wrapper: wrapperWithoutStreaming,
             });
@@ -422,7 +423,11 @@ describe("TamboThreadProvider", () => {
                 messageToAppend: {
                     content: [{ type: "text", text: "Hello default" }],
                     role: "user",
-                    additionalContext: mockCombinedContext,
+                    additionalContext: {
+                        custom: {
+                            message: "additional instructions",
+                        },
+                    },
                 },
                 availableComponents: serializeRegistry(mockRegistry),
                 contextKey: undefined,
@@ -437,7 +442,8 @@ describe("TamboThreadProvider", () => {
         it("should call advanceStream when streamResponse is undefined and provider streaming=true (default)", async () => {
             // Use wrapper with streaming=true (default) to test that undefined streamResponse respects provider setting
             const wrapperWithDefaultStreaming = ({ children, }) => (React.createElement(TamboRegistryProvider, { components: mockRegistry },
-                React.createElement(TamboThreadProvider, null, children)));
+                React.createElement(TamboContextHelpersProvider, { contextHelpers: { userTime: false, userPage: false } },
+                    React.createElement(TamboThreadProvider, null, children))));
             const mockStreamResponse = {
                 responseMessageDto: {
                     id: "stream-response",
@@ -475,7 +481,11 @@ describe("TamboThreadProvider", () => {
                 messageToAppend: {
                     content: [{ type: "text", text: "Hello default streaming" }],
                     role: "user",
-                    additionalContext: mockCombinedContext,
+                    additionalContext: {
+                        custom: {
+                            message: "additional instructions",
+                        },
+                    },
                 },
                 availableComponents: serializeRegistry(mockRegistry),
                 contextKey: undefined,
@@ -490,7 +500,8 @@ describe("TamboThreadProvider", () => {
         it("should call advance when streamResponse=false for placeholder thread", async () => {
             // Use wrapper with streaming=true to show that explicit streamResponse=false overrides provider setting
             const wrapperWithStreaming = ({ children, }) => (React.createElement(TamboRegistryProvider, { components: mockRegistry },
-                React.createElement(TamboThreadProvider, { streaming: true }, children)));
+                React.createElement(TamboContextHelpersProvider, { contextHelpers: { userTime: false, userPage: false } },
+                    React.createElement(TamboThreadProvider, { streaming: true }, children))));
             const { result } = renderHook(() => useTamboThread(), {
                 wrapper: wrapperWithStreaming,
             });
@@ -511,7 +522,11 @@ describe("TamboThreadProvider", () => {
                 messageToAppend: {
                     content: [{ type: "text", text: "Hello new thread" }],
                     role: "user",
-                    additionalContext: mockCombinedContext,
+                    additionalContext: {
+                        custom: {
+                            message: "additional instructions",
+                        },
+                    },
                 },
                 availableComponents: serializeRegistry(mockRegistry),
                 contextKey: undefined,
@@ -526,7 +541,8 @@ describe("TamboThreadProvider", () => {
         it("should call advanceStream when streamResponse=true for placeholder thread", async () => {
             // Use wrapper with streaming=false to show that explicit streamResponse=true overrides provider setting
             const wrapperWithoutStreaming = ({ children, }) => (React.createElement(TamboRegistryProvider, { components: mockRegistry },
-                React.createElement(TamboThreadProvider, { streaming: false }, children)));
+                React.createElement(TamboContextHelpersProvider, { contextHelpers: { userTime: false, userPage: false } },
+                    React.createElement(TamboThreadProvider, { streaming: false }, children))));
             const mockStreamResponse = {
                 responseMessageDto: {
                     id: "stream-response",
@@ -566,7 +582,11 @@ describe("TamboThreadProvider", () => {
                 messageToAppend: {
                     content: [{ type: "text", text: "Hello streaming new thread" }],
                     role: "user",
-                    additionalContext: mockCombinedContext,
+                    additionalContext: {
+                        custom: {
+                            message: "additional instructions",
+                        },
+                    },
                 },
                 availableComponents: serializeRegistry(mockRegistry),
                 contextKey: undefined,