@tambo-ai/react 0.64.1 → 0.65.1

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

@@ -21,10 +21,13 @@ Object.defineProperty(global, "crypto", {
     },
 });
 // Mock the required providers
-jest.mock("../tambo-client-provider", () => ({
-    useTamboClient: jest.fn(),
-    useTamboQueryClient: jest.fn(),
-}));
+jest.mock("../tambo-client-provider", () => {
+    return {
+        useTamboClient: jest.fn(),
+        useTamboQueryClient: jest.fn(),
+        TamboClientContext: react_2.default.createContext(undefined),
+    };
+});
 jest.mock("@tambo-ai/typescript-sdk", () => ({
     advanceStream: jest.fn(),
 }));
@@ -113,13 +116,24 @@ describe("TamboThreadProvider", () => {
         },
     ];
     // Use helpers that explicitly return null so they don't appear in additionalContext
-    const wrapper = ({ children }) => (react_2.default.createElement(tambo_registry_provider_1.TamboRegistryProvider, { components: mockRegistry },
-        react_2.default.createElement(tambo_context_helpers_provider_1.TamboContextHelpersProvider, { contextHelpers: {
-                currentTimeContextHelper: () => null,
-                currentPageContextHelper: () => null,
+    const Wrapper = ({ children }) => {
+        const client = (0, tambo_client_provider_1.useTamboClient)();
+        const queryClient = (0, tambo_client_provider_1.useTamboQueryClient)();
+        return (react_2.default.createElement(tambo_client_provider_1.TamboClientContext.Provider, { value: {
+                client,
+                queryClient,
+                isUpdatingToken: false,
+                mcpAccessToken: null,
+                setMcpAccessToken: () => { },
             } },
-            react_2.default.createElement(tambo_mcp_token_provider_1.TamboMcpTokenProvider, null,
-                react_2.default.createElement(tambo_thread_provider_1.TamboThreadProvider, { streaming: false }, children)))));
+            react_2.default.createElement(tambo_registry_provider_1.TamboRegistryProvider, { components: mockRegistry },
+                react_2.default.createElement(tambo_context_helpers_provider_1.TamboContextHelpersProvider, { contextHelpers: {
+                        currentTimeContextHelper: () => null,
+                        currentPageContextHelper: () => null,
+                    } },
+                    react_2.default.createElement(tambo_mcp_token_provider_1.TamboMcpTokenProvider, null,
+                        react_2.default.createElement(tambo_thread_provider_1.TamboThreadProvider, { streaming: false }, children))))));
+    };
     beforeEach(() => {
         jest.clearAllMocks();
         // Setup mock query client
@@ -154,13 +168,13 @@ describe("TamboThreadProvider", () => {
         jest.mocked(tambo_client_provider_1.useTamboClient).mockReturnValue(mockTamboAI);
     });
     it("should initialize with placeholder thread", () => {
-        const { result } = (0, react_1.renderHook)(() => (0, tambo_thread_provider_1.useTamboThread)(), { wrapper });
+        const { result } = (0, react_1.renderHook)(() => (0, tambo_thread_provider_1.useTamboThread)(), { wrapper: Wrapper });
         expect(result.current.thread.id).toBe("placeholder");
         expect(result.current.isIdle).toBe(true);
         expect(result.current.generationStage).toBe(generate_component_response_1.GenerationStage.IDLE);
     });
     it("should switch to a new thread", async () => {
-        const { result } = (0, react_1.renderHook)(() => (0, tambo_thread_provider_1.useTamboThread)(), { wrapper });
+        const { result } = (0, react_1.renderHook)(() => (0, tambo_thread_provider_1.useTamboThread)(), { wrapper: Wrapper });
         await (0, react_1.act)(async () => {
             await result.current.switchCurrentThread("test-thread-1");
         });
@@ -168,7 +182,7 @@ describe("TamboThreadProvider", () => {
         expect(result.current.thread.id).toBe("test-thread-1");
     });
     it("should start a new thread", async () => {
-        const { result } = (0, react_1.renderHook)(() => (0, tambo_thread_provider_1.useTamboThread)(), { wrapper });
+        const { result } = (0, react_1.renderHook)(() => (0, tambo_thread_provider_1.useTamboThread)(), { wrapper: Wrapper });
         await (0, react_1.act)(async () => {
             result.current.startNewThread();
         });
@@ -176,7 +190,7 @@ describe("TamboThreadProvider", () => {
         expect(result.current.isIdle).toBe(true);
     });
     it("should add a message to the thread", async () => {
-        const { result } = (0, react_1.renderHook)(() => (0, tambo_thread_provider_1.useTamboThread)(), { wrapper });
+        const { result } = (0, react_1.renderHook)(() => (0, tambo_thread_provider_1.useTamboThread)(), { wrapper: Wrapper });
         const testMessage = {
             id: "test-message-1",
             content: [{ type: "text", text: "Hello" }],
@@ -194,7 +208,7 @@ describe("TamboThreadProvider", () => {
         });
     });
     it("should update a message in the thread", async () => {
-        const { result } = (0, react_1.renderHook)(() => (0, tambo_thread_provider_1.useTamboThread)(), { wrapper });
+        const { result } = (0, react_1.renderHook)(() => (0, tambo_thread_provider_1.useTamboThread)(), { wrapper: Wrapper });
         const testMessage = {
             id: "test-message-1",
             content: [{ type: "text", text: "Updated message" }],
@@ -228,7 +242,7 @@ describe("TamboThreadProvider", () => {
         jest
             .mocked(mockThreadsApi.advanceByID)
             .mockResolvedValue(mockAdvanceResponse);
-        const { result } = (0, react_1.renderHook)(() => (0, tambo_thread_provider_1.useTamboThread)(), { wrapper });
+        const { result } = (0, react_1.renderHook)(() => (0, tambo_thread_provider_1.useTamboThread)(), { wrapper: Wrapper });
         await (0, react_1.act)(async () => {
             await result.current.sendThreadMessage("Hello", {
                 threadId: "test-thread-1",
@@ -279,7 +293,7 @@ describe("TamboThreadProvider", () => {
         };
         // Mock advanceStream to return our async iterator
         jest.mocked(typescript_sdk_1.advanceStream).mockResolvedValue(mockAsyncIterator);
-        const { result } = (0, react_1.renderHook)(() => (0, tambo_thread_provider_1.useTamboThread)(), { wrapper });
+        const { result } = (0, react_1.renderHook)(() => (0, tambo_thread_provider_1.useTamboThread)(), { wrapper: Wrapper });
         await (0, react_1.act)(async () => {
             await result.current.sendThreadMessage("Hello", {
                 threadId: "test-thread-1",
@@ -320,7 +334,7 @@ describe("TamboThreadProvider", () => {
             generationStage: generate_component_response_1.GenerationStage.COMPLETE,
             mcpAccessToken: "test-mcp-access-token",
         });
-        const { result } = (0, react_1.renderHook)(() => (0, tambo_thread_provider_1.useTamboThread)(), { wrapper });
+        const { result } = (0, react_1.renderHook)(() => (0, tambo_thread_provider_1.useTamboThread)(), { wrapper: Wrapper });
         await (0, react_1.act)(async () => {
             await result.current.sendThreadMessage("Use tool", {
                 threadId: "test-thread-1",
@@ -334,13 +348,24 @@ describe("TamboThreadProvider", () => {
         const mockOnCallUnregisteredTool = jest
             .fn()
             .mockResolvedValue("unregistered-tool-result");
-        const wrapperWithUnregisteredTool = ({ children, }) => (react_2.default.createElement(tambo_registry_provider_1.TamboRegistryProvider, { components: mockRegistry, onCallUnregisteredTool: mockOnCallUnregisteredTool },
-            react_2.default.createElement(tambo_context_helpers_provider_1.TamboContextHelpersProvider, { contextHelpers: {
-                    currentTimeContextHelper: () => null,
-                    currentPageContextHelper: () => null,
+        const WrapperWithUnregisteredTool = ({ children, }) => {
+            const client = (0, tambo_client_provider_1.useTamboClient)();
+            const queryClient = (0, tambo_client_provider_1.useTamboQueryClient)();
+            return (react_2.default.createElement(tambo_client_provider_1.TamboClientContext.Provider, { value: {
+                    client,
+                    queryClient,
+                    isUpdatingToken: false,
+                    mcpAccessToken: null,
+                    setMcpAccessToken: () => { },
                 } },
-                react_2.default.createElement(tambo_mcp_token_provider_1.TamboMcpTokenProvider, null,
-                    react_2.default.createElement(tambo_thread_provider_1.TamboThreadProvider, { streaming: false }, children)))));
+                react_2.default.createElement(tambo_registry_provider_1.TamboRegistryProvider, { components: mockRegistry, onCallUnregisteredTool: mockOnCallUnregisteredTool },
+                    react_2.default.createElement(tambo_context_helpers_provider_1.TamboContextHelpersProvider, { contextHelpers: {
+                            currentTimeContextHelper: () => null,
+                            currentPageContextHelper: () => null,
+                        } },
+                        react_2.default.createElement(tambo_mcp_token_provider_1.TamboMcpTokenProvider, null,
+                            react_2.default.createElement(tambo_thread_provider_1.TamboThreadProvider, { streaming: false }, children))))));
+        };
         const mockUnregisteredToolCallResponse = {
             responseMessageDto: {
                 id: "unregistered-tool-call-1",
@@ -375,7 +400,7 @@ describe("TamboThreadProvider", () => {
             mcpAccessToken: "test-mcp-access-token",
         });
         const { result } = (0, react_1.renderHook)(() => (0, tambo_thread_provider_1.useTamboThread)(), {
-            wrapper: wrapperWithUnregisteredTool,
+            wrapper: WrapperWithUnregisteredTool,
         });
         await (0, react_1.act)(async () => {
             await result.current.sendThreadMessage("Use unregistered tool", {
@@ -420,7 +445,7 @@ describe("TamboThreadProvider", () => {
             generationStage: generate_component_response_1.GenerationStage.COMPLETE,
             mcpAccessToken: "test-mcp-access-token",
         });
-        const { result } = (0, react_1.renderHook)(() => (0, tambo_thread_provider_1.useTamboThread)(), { wrapper });
+        const { result } = (0, react_1.renderHook)(() => (0, tambo_thread_provider_1.useTamboThread)(), { wrapper: Wrapper });
         await (0, react_1.act)(async () => {
             await result.current.sendThreadMessage("Use unregistered tool", {
                 threadId: "test-thread-1",
@@ -433,13 +458,24 @@ describe("TamboThreadProvider", () => {
     describe("streaming behavior", () => {
         it("should call advanceStream when streamResponse=true", async () => {
             // Use wrapper with streaming=true to show that explicit streamResponse=true works
-            const wrapperWithStreaming = ({ children, }) => (react_2.default.createElement(tambo_registry_provider_1.TamboRegistryProvider, { components: mockRegistry },
-                react_2.default.createElement(tambo_context_helpers_provider_1.TamboContextHelpersProvider, { contextHelpers: {
-                        currentTimeContextHelper: () => null,
-                        currentPageContextHelper: () => null,
+            const WrapperWithStreaming = ({ children, }) => {
+                const client = (0, tambo_client_provider_1.useTamboClient)();
+                const queryClient = (0, tambo_client_provider_1.useTamboQueryClient)();
+                return (react_2.default.createElement(tambo_client_provider_1.TamboClientContext.Provider, { value: {
+                        client,
+                        queryClient,
+                        isUpdatingToken: false,
+                        mcpAccessToken: null,
+                        setMcpAccessToken: () => { },
                     } },
-                    react_2.default.createElement(tambo_mcp_token_provider_1.TamboMcpTokenProvider, null,
-                        react_2.default.createElement(tambo_thread_provider_1.TamboThreadProvider, { streaming: true }, children)))));
+                    react_2.default.createElement(tambo_registry_provider_1.TamboRegistryProvider, { components: mockRegistry },
+                        react_2.default.createElement(tambo_context_helpers_provider_1.TamboContextHelpersProvider, { contextHelpers: {
+                                currentTimeContextHelper: () => null,
+                                currentPageContextHelper: () => null,
+                            } },
+                            react_2.default.createElement(tambo_mcp_token_provider_1.TamboMcpTokenProvider, null,
+                                react_2.default.createElement(tambo_thread_provider_1.TamboThreadProvider, { streaming: true }, children))))));
+            };
             const mockStreamResponse = {
                 responseMessageDto: {
                     id: "stream-response",
@@ -460,7 +496,7 @@ describe("TamboThreadProvider", () => {
             };
             jest.mocked(typescript_sdk_1.advanceStream).mockResolvedValue(mockAsyncIterator);
             const { result } = (0, react_1.renderHook)(() => (0, tambo_thread_provider_1.useTamboThread)(), {
-                wrapper: wrapperWithStreaming,
+                wrapper: WrapperWithStreaming,
             });
             await (0, react_1.act)(async () => {
                 await result.current.sendThreadMessage("Hello streaming", {
@@ -495,15 +531,26 @@ 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_2.default.createElement(tambo_registry_provider_1.TamboRegistryProvider, { components: mockRegistry },
-                react_2.default.createElement(tambo_context_helpers_provider_1.TamboContextHelpersProvider, { contextHelpers: {
-                        currentTimeContextHelper: () => null,
-                        currentPageContextHelper: () => null,
+            const WrapperWithStreaming = ({ children, }) => {
+                const client = (0, tambo_client_provider_1.useTamboClient)();
+                const queryClient = (0, tambo_client_provider_1.useTamboQueryClient)();
+                return (react_2.default.createElement(tambo_client_provider_1.TamboClientContext.Provider, { value: {
+                        client,
+                        queryClient,
+                        isUpdatingToken: false,
+                        mcpAccessToken: null,
+                        setMcpAccessToken: () => { },
                     } },
-                    react_2.default.createElement(tambo_mcp_token_provider_1.TamboMcpTokenProvider, null,
-                        react_2.default.createElement(tambo_thread_provider_1.TamboThreadProvider, { streaming: true }, children)))));
+                    react_2.default.createElement(tambo_registry_provider_1.TamboRegistryProvider, { components: mockRegistry },
+                        react_2.default.createElement(tambo_context_helpers_provider_1.TamboContextHelpersProvider, { contextHelpers: {
+                                currentTimeContextHelper: () => null,
+                                currentPageContextHelper: () => null,
+                            } },
+                            react_2.default.createElement(tambo_mcp_token_provider_1.TamboMcpTokenProvider, null,
+                                react_2.default.createElement(tambo_thread_provider_1.TamboThreadProvider, { streaming: true }, children))))));
+            };
             const { result } = (0, react_1.renderHook)(() => (0, tambo_thread_provider_1.useTamboThread)(), {
-                wrapper: wrapperWithStreaming,
+                wrapper: WrapperWithStreaming,
             });
             await (0, react_1.act)(async () => {
                 await result.current.sendThreadMessage("Hello non-streaming", {
@@ -538,15 +585,26 @@ 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_2.default.createElement(tambo_registry_provider_1.TamboRegistryProvider, { components: mockRegistry },
-                react_2.default.createElement(tambo_context_helpers_provider_1.TamboContextHelpersProvider, { contextHelpers: {
-                        currentTimeContextHelper: () => null,
-                        currentPageContextHelper: () => null,
+            const WrapperWithoutStreaming = ({ children, }) => {
+                const client = (0, tambo_client_provider_1.useTamboClient)();
+                const queryClient = (0, tambo_client_provider_1.useTamboQueryClient)();
+                return (react_2.default.createElement(tambo_client_provider_1.TamboClientContext.Provider, { value: {
+                        client,
+                        queryClient,
+                        isUpdatingToken: false,
+                        mcpAccessToken: null,
+                        setMcpAccessToken: () => { },
                     } },
-                    react_2.default.createElement(tambo_mcp_token_provider_1.TamboMcpTokenProvider, null,
-                        react_2.default.createElement(tambo_thread_provider_1.TamboThreadProvider, { streaming: false }, children)))));
+                    react_2.default.createElement(tambo_registry_provider_1.TamboRegistryProvider, { components: mockRegistry },
+                        react_2.default.createElement(tambo_context_helpers_provider_1.TamboContextHelpersProvider, { contextHelpers: {
+                                currentTimeContextHelper: () => null,
+                                currentPageContextHelper: () => null,
+                            } },
+                            react_2.default.createElement(tambo_mcp_token_provider_1.TamboMcpTokenProvider, null,
+                                react_2.default.createElement(tambo_thread_provider_1.TamboThreadProvider, { streaming: false }, children))))));
+            };
             const { result } = (0, react_1.renderHook)(() => (0, tambo_thread_provider_1.useTamboThread)(), {
-                wrapper: wrapperWithoutStreaming,
+                wrapper: WrapperWithoutStreaming,
             });
             await (0, react_1.act)(async () => {
                 await result.current.sendThreadMessage("Hello default", {
@@ -581,13 +639,24 @@ 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_2.default.createElement(tambo_registry_provider_1.TamboRegistryProvider, { components: mockRegistry },
-                react_2.default.createElement(tambo_context_helpers_provider_1.TamboContextHelpersProvider, { contextHelpers: {
-                        currentTimeContextHelper: () => null,
-                        currentPageContextHelper: () => null,
+            const WrapperWithDefaultStreaming = ({ children, }) => {
+                const client = (0, tambo_client_provider_1.useTamboClient)();
+                const queryClient = (0, tambo_client_provider_1.useTamboQueryClient)();
+                return (react_2.default.createElement(tambo_client_provider_1.TamboClientContext.Provider, { value: {
+                        client,
+                        queryClient,
+                        isUpdatingToken: false,
+                        mcpAccessToken: null,
+                        setMcpAccessToken: () => { },
                     } },
-                    react_2.default.createElement(tambo_mcp_token_provider_1.TamboMcpTokenProvider, null,
-                        react_2.default.createElement(tambo_thread_provider_1.TamboThreadProvider, null, children)))));
+                    react_2.default.createElement(tambo_registry_provider_1.TamboRegistryProvider, { components: mockRegistry },
+                        react_2.default.createElement(tambo_context_helpers_provider_1.TamboContextHelpersProvider, { contextHelpers: {
+                                currentTimeContextHelper: () => null,
+                                currentPageContextHelper: () => null,
+                            } },
+                            react_2.default.createElement(tambo_mcp_token_provider_1.TamboMcpTokenProvider, null,
+                                react_2.default.createElement(tambo_thread_provider_1.TamboThreadProvider, null, children))))));
+            };
             const mockStreamResponse = {
                 responseMessageDto: {
                     id: "stream-response",
@@ -608,7 +677,7 @@ describe("TamboThreadProvider", () => {
             };
             jest.mocked(typescript_sdk_1.advanceStream).mockResolvedValue(mockAsyncIterator);
             const { result } = (0, react_1.renderHook)(() => (0, tambo_thread_provider_1.useTamboThread)(), {
-                wrapper: wrapperWithDefaultStreaming,
+                wrapper: WrapperWithDefaultStreaming,
             });
             await (0, react_1.act)(async () => {
                 await result.current.sendThreadMessage("Hello default streaming", {
@@ -643,15 +712,26 @@ 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_2.default.createElement(tambo_registry_provider_1.TamboRegistryProvider, { components: mockRegistry },
-                react_2.default.createElement(tambo_context_helpers_provider_1.TamboContextHelpersProvider, { contextHelpers: {
-                        currentTimeContextHelper: () => null,
-                        currentPageContextHelper: () => null,
+            const WrapperWithStreaming = ({ children, }) => {
+                const client = (0, tambo_client_provider_1.useTamboClient)();
+                const queryClient = (0, tambo_client_provider_1.useTamboQueryClient)();
+                return (react_2.default.createElement(tambo_client_provider_1.TamboClientContext.Provider, { value: {
+                        client,
+                        queryClient,
+                        isUpdatingToken: false,
+                        mcpAccessToken: null,
+                        setMcpAccessToken: () => { },
                     } },
-                    react_2.default.createElement(tambo_mcp_token_provider_1.TamboMcpTokenProvider, null,
-                        react_2.default.createElement(tambo_thread_provider_1.TamboThreadProvider, { streaming: true }, children)))));
+                    react_2.default.createElement(tambo_registry_provider_1.TamboRegistryProvider, { components: mockRegistry },
+                        react_2.default.createElement(tambo_context_helpers_provider_1.TamboContextHelpersProvider, { contextHelpers: {
+                                currentTimeContextHelper: () => null,
+                                currentPageContextHelper: () => null,
+                            } },
+                            react_2.default.createElement(tambo_mcp_token_provider_1.TamboMcpTokenProvider, null,
+                                react_2.default.createElement(tambo_thread_provider_1.TamboThreadProvider, { streaming: true }, children))))));
+            };
             const { result } = (0, react_1.renderHook)(() => (0, tambo_thread_provider_1.useTamboThread)(), {
-                wrapper: wrapperWithStreaming,
+                wrapper: WrapperWithStreaming,
             });
             // Start with placeholder thread (which is the default state)
             expect(result.current.thread.id).toBe("placeholder");
@@ -688,13 +768,24 @@ 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_2.default.createElement(tambo_registry_provider_1.TamboRegistryProvider, { components: mockRegistry },
-                react_2.default.createElement(tambo_context_helpers_provider_1.TamboContextHelpersProvider, { contextHelpers: {
-                        currentTimeContextHelper: () => null,
-                        currentPageContextHelper: () => null,
+            const WrapperWithoutStreaming = ({ children, }) => {
+                const client = (0, tambo_client_provider_1.useTamboClient)();
+                const queryClient = (0, tambo_client_provider_1.useTamboQueryClient)();
+                return (react_2.default.createElement(tambo_client_provider_1.TamboClientContext.Provider, { value: {
+                        client,
+                        queryClient,
+                        isUpdatingToken: false,
+                        mcpAccessToken: null,
+                        setMcpAccessToken: () => { },
                     } },
-                    react_2.default.createElement(tambo_mcp_token_provider_1.TamboMcpTokenProvider, null,
-                        react_2.default.createElement(tambo_thread_provider_1.TamboThreadProvider, { streaming: false }, children)))));
+                    react_2.default.createElement(tambo_registry_provider_1.TamboRegistryProvider, { components: mockRegistry },
+                        react_2.default.createElement(tambo_context_helpers_provider_1.TamboContextHelpersProvider, { contextHelpers: {
+                                currentTimeContextHelper: () => null,
+                                currentPageContextHelper: () => null,
+                            } },
+                            react_2.default.createElement(tambo_mcp_token_provider_1.TamboMcpTokenProvider, null,
+                                react_2.default.createElement(tambo_thread_provider_1.TamboThreadProvider, { streaming: false }, children))))));
+            };
             const mockStreamResponse = {
                 responseMessageDto: {
                     id: "stream-response",
@@ -715,7 +806,7 @@ describe("TamboThreadProvider", () => {
             };
             jest.mocked(typescript_sdk_1.advanceStream).mockResolvedValue(mockAsyncIterator);
             const { result } = (0, react_1.renderHook)(() => (0, tambo_thread_provider_1.useTamboThread)(), {
-                wrapper: wrapperWithoutStreaming,
+                wrapper: WrapperWithoutStreaming,
             });
             // Start with placeholder thread (which is the default state)
             expect(result.current.thread.id).toBe("placeholder");
@@ -756,7 +847,9 @@ describe("TamboThreadProvider", () => {
             const testError = new Error("API call failed");
             // Mock advanceById to throw an error
             jest.mocked(mockThreadsApi.advanceByID).mockRejectedValue(testError);
-            const { result } = (0, react_1.renderHook)(() => (0, tambo_thread_provider_1.useTamboThread)(), { wrapper });
+            const { result } = (0, react_1.renderHook)(() => (0, tambo_thread_provider_1.useTamboThread)(), {
+                wrapper: Wrapper,
+            });
             // Expect the error to be thrown
             await (0, react_1.act)(async () => {
                 await result.current.switchCurrentThread("test-thread-1");
@@ -772,7 +865,9 @@ describe("TamboThreadProvider", () => {
             const testError = new Error("Streaming API call failed");
             // Mock advanceStream to throw an error
             jest.mocked(typescript_sdk_1.advanceStream).mockRejectedValue(testError);
-            const { result } = (0, react_1.renderHook)(() => (0, tambo_thread_provider_1.useTamboThread)(), { wrapper });
+            const { result } = (0, react_1.renderHook)(() => (0, tambo_thread_provider_1.useTamboThread)(), {
+                wrapper: Wrapper,
+            });
             // Expect the error to be thrown
             await (0, react_1.act)(async () => {
                 await result.current.switchCurrentThread("test-thread-1");
@@ -788,7 +883,9 @@ describe("TamboThreadProvider", () => {
             const testError = new Error("Advance API call failed");
             // Mock advance to throw an error
             jest.mocked(mockThreadsApi.advance).mockRejectedValue(testError);
-            const { result } = (0, react_1.renderHook)(() => (0, tambo_thread_provider_1.useTamboThread)(), { wrapper });
+            const { result } = (0, react_1.renderHook)(() => (0, tambo_thread_provider_1.useTamboThread)(), {
+                wrapper: Wrapper,
+            });
             // Start with placeholder thread (which is the default state)
             expect(result.current.thread.id).toBe("placeholder");
             // Expect the error to be thrown
@@ -804,7 +901,9 @@ describe("TamboThreadProvider", () => {
     });
     describe("refetch threads list behavior", () => {
         it("should refetch threads list when creating a new thread via sendThreadMessage", async () => {
-            const { result } = (0, react_1.renderHook)(() => (0, tambo_thread_provider_1.useTamboThread)(), { wrapper });
+            const { result } = (0, react_1.renderHook)(() => (0, tambo_thread_provider_1.useTamboThread)(), {
+                wrapper: Wrapper,
+            });
             // Mock the advance response to return a new thread ID
             const mockAdvanceResponse = {
                 responseMessageDto: {
@@ -840,7 +939,9 @@ describe("TamboThreadProvider", () => {
             });
         });
         it("should not refetch threads list when switching between existing threads", async () => {
-            const { result } = (0, react_1.renderHook)(() => (0, tambo_thread_provider_1.useTamboThread)(), { wrapper });
+            const { result } = (0, react_1.renderHook)(() => (0, tambo_thread_provider_1.useTamboThread)(), {
+                wrapper: Wrapper,
+            });
             // Start with placeholder thread
             expect(result.current.thread.id).toBe("placeholder");
             // Clear any previous mock calls
@@ -892,13 +993,24 @@ describe("TamboThreadProvider", () => {
                     ],
                 },
             ];
-            const wrapperWithCustomTool = ({ children, }) => (react_2.default.createElement(tambo_registry_provider_1.TamboRegistryProvider, { components: customToolRegistry },
-                react_2.default.createElement(tambo_context_helpers_provider_1.TamboContextHelpersProvider, { contextHelpers: {
-                        currentTimeContextHelper: () => null,
-                        currentPageContextHelper: () => null,
+            const WrapperWithCustomTool = ({ children, }) => {
+                const client = (0, tambo_client_provider_1.useTamboClient)();
+                const queryClient = (0, tambo_client_provider_1.useTamboQueryClient)();
+                return (react_2.default.createElement(tambo_client_provider_1.TamboClientContext.Provider, { value: {
+                        client,
+                        queryClient,
+                        isUpdatingToken: false,
+                        mcpAccessToken: null,
+                        setMcpAccessToken: () => { },
                     } },
-                    react_2.default.createElement(tambo_mcp_token_provider_1.TamboMcpTokenProvider, null,
-                        react_2.default.createElement(tambo_thread_provider_1.TamboThreadProvider, { streaming: false }, children)))));
+                    react_2.default.createElement(tambo_registry_provider_1.TamboRegistryProvider, { components: customToolRegistry },
+                        react_2.default.createElement(tambo_context_helpers_provider_1.TamboContextHelpersProvider, { contextHelpers: {
+                                currentTimeContextHelper: () => null,
+                                currentPageContextHelper: () => null,
+                            } },
+                            react_2.default.createElement(tambo_mcp_token_provider_1.TamboMcpTokenProvider, null,
+                                react_2.default.createElement(tambo_thread_provider_1.TamboThreadProvider, { streaming: false }, children))))));
+            };
             const mockToolCallResponse = {
                 responseMessageDto: {
                     id: "tool-call-1",
@@ -931,7 +1043,7 @@ describe("TamboThreadProvider", () => {
                 mcpAccessToken: "test-mcp-access-token",
             });
             const { result } = (0, react_1.renderHook)(() => (0, tambo_thread_provider_1.useTamboThread)(), {
-                wrapper: wrapperWithCustomTool,
+                wrapper: WrapperWithCustomTool,
             });
             await (0, react_1.act)(async () => {
                 await result.current.sendThreadMessage("Use custom tool", {
@@ -986,13 +1098,24 @@ describe("TamboThreadProvider", () => {
                     ],
                 },
             ];
-            const wrapperWithAsyncTool = ({ children, }) => (react_2.default.createElement(tambo_registry_provider_1.TamboRegistryProvider, { components: customToolRegistry },
-                react_2.default.createElement(tambo_context_helpers_provider_1.TamboContextHelpersProvider, { contextHelpers: {
-                        currentTimeContextHelper: () => null,
-                        currentPageContextHelper: () => null,
+            const WrapperWithAsyncTool = ({ children, }) => {
+                const client = (0, tambo_client_provider_1.useTamboClient)();
+                const queryClient = (0, tambo_client_provider_1.useTamboQueryClient)();
+                return (react_2.default.createElement(tambo_client_provider_1.TamboClientContext.Provider, { value: {
+                        client,
+                        queryClient,
+                        isUpdatingToken: false,
+                        mcpAccessToken: null,
+                        setMcpAccessToken: () => { },
                     } },
-                    react_2.default.createElement(tambo_mcp_token_provider_1.TamboMcpTokenProvider, null,
-                        react_2.default.createElement(tambo_thread_provider_1.TamboThreadProvider, { streaming: true }, children)))));
+                    react_2.default.createElement(tambo_registry_provider_1.TamboRegistryProvider, { components: customToolRegistry },
+                        react_2.default.createElement(tambo_context_helpers_provider_1.TamboContextHelpersProvider, { contextHelpers: {
+                                currentTimeContextHelper: () => null,
+                                currentPageContextHelper: () => null,
+                            } },
+                            react_2.default.createElement(tambo_mcp_token_provider_1.TamboMcpTokenProvider, null,
+                                react_2.default.createElement(tambo_thread_provider_1.TamboThreadProvider, { streaming: true }, children))))));
+            };
             const mockToolCallChunk = {
                 responseMessageDto: {
                     id: "tool-call-chunk",
@@ -1038,7 +1161,7 @@ describe("TamboThreadProvider", () => {
                 },
             });
             const { result } = (0, react_1.renderHook)(() => (0, tambo_thread_provider_1.useTamboThread)(), {
-                wrapper: wrapperWithAsyncTool,
+                wrapper: WrapperWithAsyncTool,
             });
             await (0, react_1.act)(async () => {
                 await result.current.sendThreadMessage("Use async tool", {
@@ -1088,13 +1211,24 @@ describe("TamboThreadProvider", () => {
                     ],
                 },
             ];
-            const wrapperWithoutTransform = ({ children, }) => (react_2.default.createElement(tambo_registry_provider_1.TamboRegistryProvider, { components: toolWithoutTransform },
-                react_2.default.createElement(tambo_context_helpers_provider_1.TamboContextHelpersProvider, { contextHelpers: {
-                        currentTimeContextHelper: () => null,
-                        currentPageContextHelper: () => null,
+            const WrapperWithoutTransform = ({ children, }) => {
+                const client = (0, tambo_client_provider_1.useTamboClient)();
+                const queryClient = (0, tambo_client_provider_1.useTamboQueryClient)();
+                return (react_2.default.createElement(tambo_client_provider_1.TamboClientContext.Provider, { value: {
+                        client,
+                        queryClient,
+                        isUpdatingToken: false,
+                        mcpAccessToken: null,
+                        setMcpAccessToken: () => { },
                     } },
-                    react_2.default.createElement(tambo_mcp_token_provider_1.TamboMcpTokenProvider, null,
-                        react_2.default.createElement(tambo_thread_provider_1.TamboThreadProvider, { streaming: false }, children)))));
+                    react_2.default.createElement(tambo_registry_provider_1.TamboRegistryProvider, { components: toolWithoutTransform },
+                        react_2.default.createElement(tambo_context_helpers_provider_1.TamboContextHelpersProvider, { contextHelpers: {
+                                currentTimeContextHelper: () => null,
+                                currentPageContextHelper: () => null,
+                            } },
+                            react_2.default.createElement(tambo_mcp_token_provider_1.TamboMcpTokenProvider, null,
+                                react_2.default.createElement(tambo_thread_provider_1.TamboThreadProvider, { streaming: false }, children))))));
+            };
             const mockToolCallResponse = {
                 responseMessageDto: {
                     id: "tool-call-1",
@@ -1127,7 +1261,7 @@ describe("TamboThreadProvider", () => {
                 mcpAccessToken: "test-mcp-access-token",
             });
             const { result } = (0, react_1.renderHook)(() => (0, tambo_thread_provider_1.useTamboThread)(), {
-                wrapper: wrapperWithoutTransform,
+                wrapper: WrapperWithoutTransform,
             });
             await (0, react_1.act)(async () => {
                 await result.current.sendThreadMessage("Use tool without transform", {
@@ -1176,13 +1310,24 @@ describe("TamboThreadProvider", () => {
                     ],
                 },
             ];
-            const wrapperWithErrorTool = ({ children, }) => (react_2.default.createElement(tambo_registry_provider_1.TamboRegistryProvider, { components: toolWithTransform },
-                react_2.default.createElement(tambo_context_helpers_provider_1.TamboContextHelpersProvider, { contextHelpers: {
-                        currentTimeContextHelper: () => null,
-                        currentPageContextHelper: () => null,
+            const WrapperWithErrorTool = ({ children, }) => {
+                const client = (0, tambo_client_provider_1.useTamboClient)();
+                const queryClient = (0, tambo_client_provider_1.useTamboQueryClient)();
+                return (react_2.default.createElement(tambo_client_provider_1.TamboClientContext.Provider, { value: {
+                        client,
+                        queryClient,
+                        isUpdatingToken: false,
+                        mcpAccessToken: null,
+                        setMcpAccessToken: () => { },
                     } },
-                    react_2.default.createElement(tambo_mcp_token_provider_1.TamboMcpTokenProvider, null,
-                        react_2.default.createElement(tambo_thread_provider_1.TamboThreadProvider, { streaming: false }, children)))));
+                    react_2.default.createElement(tambo_registry_provider_1.TamboRegistryProvider, { components: toolWithTransform },
+                        react_2.default.createElement(tambo_context_helpers_provider_1.TamboContextHelpersProvider, { contextHelpers: {
+                                currentTimeContextHelper: () => null,
+                                currentPageContextHelper: () => null,
+                            } },
+                            react_2.default.createElement(tambo_mcp_token_provider_1.TamboMcpTokenProvider, null,
+                                react_2.default.createElement(tambo_thread_provider_1.TamboThreadProvider, { streaming: false }, children))))));
+            };
             const mockToolCallResponse = {
                 responseMessageDto: {
                     id: "tool-call-1",
@@ -1215,7 +1360,7 @@ describe("TamboThreadProvider", () => {
                 mcpAccessToken: "test-mcp-access-token",
             });
             const { result } = (0, react_1.renderHook)(() => (0, tambo_thread_provider_1.useTamboThread)(), {
-                wrapper: wrapperWithErrorTool,
+                wrapper: WrapperWithErrorTool,
             });
             await (0, react_1.act)(async () => {
                 await result.current.sendThreadMessage("Use error tool", {
@@ -1243,15 +1388,26 @@ describe("TamboThreadProvider", () => {
     });
     describe("auto-generate thread name", () => {
         it("should auto-generate thread name after reaching threshold", async () => {
-            const wrapperWithAutoGenerate = ({ children, }) => (react_2.default.createElement(tambo_registry_provider_1.TamboRegistryProvider, { components: mockRegistry },
-                react_2.default.createElement(tambo_context_helpers_provider_1.TamboContextHelpersProvider, { contextHelpers: {
-                        currentTimeContextHelper: () => null,
-                        currentPageContextHelper: () => null,
+            const WrapperWithAutoGenerate = ({ children, }) => {
+                const client = (0, tambo_client_provider_1.useTamboClient)();
+                const queryClient = (0, tambo_client_provider_1.useTamboQueryClient)();
+                return (react_2.default.createElement(tambo_client_provider_1.TamboClientContext.Provider, { value: {
+                        client,
+                        queryClient,
+                        isUpdatingToken: false,
+                        mcpAccessToken: null,
+                        setMcpAccessToken: () => { },
                     } },
-                    react_2.default.createElement(tambo_mcp_token_provider_1.TamboMcpTokenProvider, null,
-                        react_2.default.createElement(tambo_thread_provider_1.TamboThreadProvider, { streaming: false, autoGenerateNameThreshold: 2 }, children)))));
+                    react_2.default.createElement(tambo_registry_provider_1.TamboRegistryProvider, { components: mockRegistry },
+                        react_2.default.createElement(tambo_context_helpers_provider_1.TamboContextHelpersProvider, { contextHelpers: {
+                                currentTimeContextHelper: () => null,
+                                currentPageContextHelper: () => null,
+                            } },
+                            react_2.default.createElement(tambo_mcp_token_provider_1.TamboMcpTokenProvider, null,
+                                react_2.default.createElement(tambo_thread_provider_1.TamboThreadProvider, { streaming: false, autoGenerateNameThreshold: 2 }, children))))));
+            };
             const { result } = (0, react_1.renderHook)(() => (0, tambo_thread_provider_1.useTamboThread)(), {
-                wrapper: wrapperWithAutoGenerate,
+                wrapper: WrapperWithAutoGenerate,
             });
             const existingThread = createMockThread({
                 id: "test-thread-1",
@@ -1288,15 +1444,26 @@ describe("TamboThreadProvider", () => {
             expect(mockQueryClient.setQueryData).toHaveBeenCalledWith(["threads", "test-project-id", undefined], expect.any(Function));
         });
         it("should NOT auto-generate when autoGenerateThreadName is false", async () => {
-            const wrapperWithDisabled = ({ children, }) => (react_2.default.createElement(tambo_registry_provider_1.TamboRegistryProvider, { components: mockRegistry },
-                react_2.default.createElement(tambo_context_helpers_provider_1.TamboContextHelpersProvider, { contextHelpers: {
-                        currentTimeContextHelper: () => null,
-                        currentPageContextHelper: () => null,
+            const WrapperWithDisabled = ({ children, }) => {
+                const client = (0, tambo_client_provider_1.useTamboClient)();
+                const queryClient = (0, tambo_client_provider_1.useTamboQueryClient)();
+                return (react_2.default.createElement(tambo_client_provider_1.TamboClientContext.Provider, { value: {
+                        client,
+                        queryClient,
+                        isUpdatingToken: false,
+                        mcpAccessToken: null,
+                        setMcpAccessToken: () => { },
                     } },
-                    react_2.default.createElement(tambo_mcp_token_provider_1.TamboMcpTokenProvider, null,
-                        react_2.default.createElement(tambo_thread_provider_1.TamboThreadProvider, { streaming: false, autoGenerateThreadName: false, autoGenerateNameThreshold: 2 }, children)))));
+                    react_2.default.createElement(tambo_registry_provider_1.TamboRegistryProvider, { components: mockRegistry },
+                        react_2.default.createElement(tambo_context_helpers_provider_1.TamboContextHelpersProvider, { contextHelpers: {
+                                currentTimeContextHelper: () => null,
+                                currentPageContextHelper: () => null,
+                            } },
+                            react_2.default.createElement(tambo_mcp_token_provider_1.TamboMcpTokenProvider, null,
+                                react_2.default.createElement(tambo_thread_provider_1.TamboThreadProvider, { streaming: false, autoGenerateThreadName: false, autoGenerateNameThreshold: 2 }, children))))));
+            };
             const { result } = (0, react_1.renderHook)(() => (0, tambo_thread_provider_1.useTamboThread)(), {
-                wrapper: wrapperWithDisabled,
+                wrapper: WrapperWithDisabled,
             });
             const existingThread = createMockThread({
                 id: "test-thread-1",
@@ -1329,15 +1496,26 @@ describe("TamboThreadProvider", () => {
             expect(mockThreadsApi.generateName).not.toHaveBeenCalled();
         });
         it("should NOT auto-generate when thread already has a name", async () => {
-            const wrapperWithAutoGenerate = ({ children, }) => (react_2.default.createElement(tambo_registry_provider_1.TamboRegistryProvider, { components: mockRegistry },
-                react_2.default.createElement(tambo_context_helpers_provider_1.TamboContextHelpersProvider, { contextHelpers: {
-                        currentTimeContextHelper: () => null,
-                        currentPageContextHelper: () => null,
+            const WrapperWithAutoGenerate = ({ children, }) => {
+                const client = (0, tambo_client_provider_1.useTamboClient)();
+                const queryClient = (0, tambo_client_provider_1.useTamboQueryClient)();
+                return (react_2.default.createElement(tambo_client_provider_1.TamboClientContext.Provider, { value: {
+                        client,
+                        queryClient,
+                        isUpdatingToken: false,
+                        mcpAccessToken: null,
+                        setMcpAccessToken: () => { },
                     } },
-                    react_2.default.createElement(tambo_mcp_token_provider_1.TamboMcpTokenProvider, null,
-                        react_2.default.createElement(tambo_thread_provider_1.TamboThreadProvider, { streaming: false, autoGenerateNameThreshold: 2 }, children)))));
+                    react_2.default.createElement(tambo_registry_provider_1.TamboRegistryProvider, { components: mockRegistry },
+                        react_2.default.createElement(tambo_context_helpers_provider_1.TamboContextHelpersProvider, { contextHelpers: {
+                                currentTimeContextHelper: () => null,
+                                currentPageContextHelper: () => null,
+                            } },
+                            react_2.default.createElement(tambo_mcp_token_provider_1.TamboMcpTokenProvider, null,
+                                react_2.default.createElement(tambo_thread_provider_1.TamboThreadProvider, { streaming: false, autoGenerateNameThreshold: 2 }, children))))));
+            };
             const { result } = (0, react_1.renderHook)(() => (0, tambo_thread_provider_1.useTamboThread)(), {
-                wrapper: wrapperWithAutoGenerate,
+                wrapper: WrapperWithAutoGenerate,
             });
             const threadWithName = createMockThread({
                 id: "test-thread-1",
@@ -1375,15 +1553,26 @@ describe("TamboThreadProvider", () => {
             expect(mockThreadsApi.generateName).not.toHaveBeenCalled();
         });
         it("should NOT auto-generate for placeholder thread", async () => {
-            const wrapperWithAutoGenerate = ({ children, }) => (react_2.default.createElement(tambo_registry_provider_1.TamboRegistryProvider, { components: mockRegistry },
-                react_2.default.createElement(tambo_context_helpers_provider_1.TamboContextHelpersProvider, { contextHelpers: {
-                        currentTimeContextHelper: () => null,
-                        currentPageContextHelper: () => null,
+            const WrapperWithAutoGenerate = ({ children, }) => {
+                const client = (0, tambo_client_provider_1.useTamboClient)();
+                const queryClient = (0, tambo_client_provider_1.useTamboQueryClient)();
+                return (react_2.default.createElement(tambo_client_provider_1.TamboClientContext.Provider, { value: {
+                        client,
+                        queryClient,
+                        isUpdatingToken: false,
+                        mcpAccessToken: null,
+                        setMcpAccessToken: () => { },
                     } },
-                    react_2.default.createElement(tambo_mcp_token_provider_1.TamboMcpTokenProvider, null,
-                        react_2.default.createElement(tambo_thread_provider_1.TamboThreadProvider, { streaming: false, autoGenerateNameThreshold: 2 }, children)))));
+                    react_2.default.createElement(tambo_registry_provider_1.TamboRegistryProvider, { components: mockRegistry },
+                        react_2.default.createElement(tambo_context_helpers_provider_1.TamboContextHelpersProvider, { contextHelpers: {
+                                currentTimeContextHelper: () => null,
+                                currentPageContextHelper: () => null,
+                            } },
+                            react_2.default.createElement(tambo_mcp_token_provider_1.TamboMcpTokenProvider, null,
+                                react_2.default.createElement(tambo_thread_provider_1.TamboThreadProvider, { streaming: false, autoGenerateNameThreshold: 2 }, children))))));
+            };
             const { result } = (0, react_1.renderHook)(() => (0, tambo_thread_provider_1.useTamboThread)(), {
-                wrapper: wrapperWithAutoGenerate,
+                wrapper: WrapperWithAutoGenerate,
             });
             // Stay on placeholder thread
             expect(result.current.thread.id).toBe("placeholder");