@tambo-ai/react 0.75.0 → 1.0.0-rc.4

package/dist/v1/hooks/use-tambo-v1.test.js

@@ -49,17 +49,17 @@ jest.mock("../providers/tambo-v1-provider", () => {
     const actual = jest.requireActual("../providers/tambo-v1-provider");
     return {
         ...actual,
-        useTamboV1Config: () => ({ userKey: undefined }),
+        useTamboConfig: () => ({ userKey: undefined }),
     };
 });
 jest.mock("./use-tambo-v1-auth-state", () => ({
-    useTamboV1AuthState: () => ({
+    useTamboAuthState: () => ({
         status: "identified",
         source: "userKey",
     }),
 }));
 const tambo_client_provider_2 = require("../../providers/tambo-client-provider");
-describe("useTamboV1", () => {
+describe("useTambo", () => {
     let queryClient;
     const mockTamboClient = {
         apiKey: "",
@@ -92,23 +92,26 @@ describe("useTamboV1", () => {
     function TestWrapper({ children }) {
         return (react_2.default.createElement(react_query_1.QueryClientProvider, { client: queryClient },
             react_2.default.createElement(tambo_registry_provider_1.TamboRegistryContext.Provider, { value: mockRegistry },
-                react_2.default.createElement(tambo_v1_stream_context_1.TamboV1StreamProvider, null, children))));
+                react_2.default.createElement(tambo_v1_stream_context_1.TamboStreamProvider, null, children))));
     }
     function createWrapperWithState(state, registry = mockRegistry) {
         const noopDispatch = () => { };
         return function WrapperWithState({ children, }) {
             return (react_2.default.createElement(react_query_1.QueryClientProvider, { client: queryClient },
                 react_2.default.createElement(tambo_registry_provider_1.TamboRegistryContext.Provider, { value: registry },
-                    react_2.default.createElement(tambo_v1_stream_context_1.TamboV1StreamProvider, { state: state, dispatch: noopDispatch }, children))));
+                    react_2.default.createElement(tambo_v1_stream_context_1.TamboStreamProvider, { state: state, dispatch: noopDispatch }, children))));
         };
     }
     // Wrapper that uses the real reducer so state updates properly
-    function createWrapperWithRealReducer(initialState) {
+    function createWrapperWithRealReducer(initialState, stateRef) {
         return function WrapperWithRealReducer({ children, }) {
             const [state, dispatch] = (0, react_2.useReducer)(event_accumulator_1.streamReducer, initialState);
+            if (stateRef) {
+                stateRef.current = state;
+            }
             return (react_2.default.createElement(react_query_1.QueryClientProvider, { client: queryClient },
                 react_2.default.createElement(tambo_registry_provider_1.TamboRegistryContext.Provider, { value: mockRegistry },
-                    react_2.default.createElement(tambo_v1_stream_context_1.TamboV1StreamProvider, { state: state, dispatch: dispatch }, children))));
+                    react_2.default.createElement(tambo_v1_stream_context_1.TamboStreamProvider, { state: state, dispatch: dispatch }, children))));
         };
     }
     beforeEach(() => {
@@ -120,15 +123,16 @@ describe("useTamboV1", () => {
         jest.mocked(tambo_client_provider_1.useTamboClient).mockReturnValue(mockTamboClient);
         jest.mocked(tambo_client_provider_2.useTamboQueryClient).mockReturnValue(queryClient);
         jest.clearAllMocks();
+        delete mockTamboClient.threads.update;
     });
     it("returns client from useTamboClient", () => {
-        const { result } = (0, react_1.renderHook)(() => (0, use_tambo_v1_1.useTamboV1)(), {
+        const { result } = (0, react_1.renderHook)(() => (0, use_tambo_v1_1.useTambo)(), {
             wrapper: TestWrapper,
         });
         expect(result.current.client).toBe(mockTamboClient);
     });
     it("returns registry functions", () => {
-        const { result } = (0, react_1.renderHook)(() => (0, use_tambo_v1_1.useTamboV1)(), {
+        const { result } = (0, react_1.renderHook)(() => (0, use_tambo_v1_1.useTambo)(), {
             wrapper: TestWrapper,
         });
         expect(result.current.registerComponent).toBe(mockRegistry.registerComponent);
@@ -138,7 +142,7 @@ describe("useTamboV1", () => {
         expect(result.current.toolRegistry).toBe(mockRegistry.toolRegistry);
     });
     it("returns placeholder thread when no threadId provided", () => {
-        const { result } = (0, react_1.renderHook)(() => (0, use_tambo_v1_1.useTamboV1)(), {
+        const { result } = (0, react_1.renderHook)(() => (0, use_tambo_v1_1.useTambo)(), {
             wrapper: TestWrapper,
         });
         // Default state has placeholder thread for optimistic UI
@@ -147,7 +151,7 @@ describe("useTamboV1", () => {
         expect(result.current.messages).toEqual([]);
     });
     it("returns thread state when switched to a thread", () => {
-        const { result } = (0, react_1.renderHook)(() => (0, use_tambo_v1_1.useTamboV1)(), {
+        const { result } = (0, react_1.renderHook)(() => (0, use_tambo_v1_1.useTambo)(), {
             wrapper: TestWrapper,
         });
         // Initialize and switch to thread
@@ -160,7 +164,7 @@ describe("useTamboV1", () => {
         expect(result.current.messages).toEqual([]);
     });
     it("returns default streaming state when thread not loaded", () => {
-        const { result } = (0, react_1.renderHook)(() => (0, use_tambo_v1_1.useTamboV1)(), {
+        const { result } = (0, react_1.renderHook)(() => (0, use_tambo_v1_1.useTambo)(), {
             wrapper: TestWrapper,
         });
         expect(result.current.streamingState).toEqual({ status: "idle" });
@@ -169,7 +173,7 @@ describe("useTamboV1", () => {
         expect(result.current.isWaiting).toBe(false);
     });
     it("returns thread streaming state when thread loaded", () => {
-        const { result } = (0, react_1.renderHook)(() => (0, use_tambo_v1_1.useTamboV1)(), {
+        const { result } = (0, react_1.renderHook)(() => (0, use_tambo_v1_1.useTambo)(), {
             wrapper: TestWrapper,
         });
         // Initialize and switch to thread
@@ -183,13 +187,13 @@ describe("useTamboV1", () => {
         expect(result.current.isWaiting).toBe(false);
     });
     it("provides dispatch function for advanced usage", () => {
-        const { result } = (0, react_1.renderHook)(() => (0, use_tambo_v1_1.useTamboV1)(), {
+        const { result } = (0, react_1.renderHook)(() => (0, use_tambo_v1_1.useTambo)(), {
             wrapper: TestWrapper,
         });
         expect(typeof result.current.dispatch).toBe("function");
     });
     it("provides thread management functions", () => {
-        const { result } = (0, react_1.renderHook)(() => (0, use_tambo_v1_1.useTamboV1)(), {
+        const { result } = (0, react_1.renderHook)(() => (0, use_tambo_v1_1.useTambo)(), {
             wrapper: TestWrapper,
         });
         expect(typeof result.current.initThread).toBe("function");
@@ -197,7 +201,7 @@ describe("useTamboV1", () => {
         expect(typeof result.current.startNewThread).toBe("function");
     });
     it("initializes and switches threads", () => {
-        const { result } = (0, react_1.renderHook)(() => (0, use_tambo_v1_1.useTamboV1)(), {
+        const { result } = (0, react_1.renderHook)(() => (0, use_tambo_v1_1.useTambo)(), {
             wrapper: TestWrapper,
         });
         // Initially has placeholder thread for optimistic UI
@@ -215,7 +219,7 @@ describe("useTamboV1", () => {
         expect(result.current.currentThreadId).toBe("new_thread_1");
     });
     it("starts new thread with generated ID", () => {
-        const { result } = (0, react_1.renderHook)(() => (0, use_tambo_v1_1.useTamboV1)(), {
+        const { result } = (0, react_1.renderHook)(() => (0, use_tambo_v1_1.useTambo)(), {
             wrapper: TestWrapper,
         });
         let newThreadId;
@@ -227,7 +231,7 @@ describe("useTamboV1", () => {
         expect(result.current.thread).toBeDefined();
     });
     it("uses current thread when no threadId argument provided", () => {
-        const { result } = (0, react_1.renderHook)(() => (0, use_tambo_v1_1.useTamboV1)(), {
+        const { result } = (0, react_1.renderHook)(() => (0, use_tambo_v1_1.useTambo)(), {
             wrapper: TestWrapper,
         });
         // Initialize and switch to a thread
@@ -276,7 +280,7 @@ describe("useTamboV1", () => {
                 },
                 currentThreadId: "thread_123",
             };
-            const { result } = (0, react_1.renderHook)(() => (0, use_tambo_v1_1.useTamboV1)(), {
+            const { result } = (0, react_1.renderHook)(() => (0, use_tambo_v1_1.useTambo)(), {
                 wrapper: createWrapperWithState(state),
             });
             expect(result.current.messages).toHaveLength(1);
@@ -314,7 +318,7 @@ describe("useTamboV1", () => {
                 },
                 currentThreadId: "thread_123",
             };
-            const { result } = (0, react_1.renderHook)(() => (0, use_tambo_v1_1.useTamboV1)(), {
+            const { result } = (0, react_1.renderHook)(() => (0, use_tambo_v1_1.useTambo)(), {
                 wrapper: createWrapperWithState(state),
             });
             expect(result.current.messages).toHaveLength(1);
@@ -354,7 +358,7 @@ describe("useTamboV1", () => {
                 },
                 currentThreadId: "thread_123",
             };
-            const { result, rerender } = (0, react_1.renderHook)(() => (0, use_tambo_v1_1.useTamboV1)(), {
+            const { result, rerender } = (0, react_1.renderHook)(() => (0, use_tambo_v1_1.useTambo)(), {
                 wrapper: createWrapperWithState(state),
             });
             const firstRender = result.current.messages[0].content[0].renderedComponent;
@@ -391,7 +395,7 @@ describe("useTamboV1", () => {
                 },
                 currentThreadId: "thread_123",
             };
-            const { result } = (0, react_1.renderHook)(() => (0, use_tambo_v1_1.useTamboV1)(), {
+            const { result } = (0, react_1.renderHook)(() => (0, use_tambo_v1_1.useTambo)(), {
                 wrapper: createWrapperWithState(state),
             });
             const content = result.current.messages[0].content[0];
@@ -431,7 +435,7 @@ describe("useTamboV1", () => {
                 },
                 currentThreadId: "thread_123",
             };
-            const { result } = (0, react_1.renderHook)(() => (0, use_tambo_v1_1.useTamboV1)(), {
+            const { result } = (0, react_1.renderHook)(() => (0, use_tambo_v1_1.useTambo)(), {
                 wrapper: createWrapperWithState(state),
             });
             const content = result.current.messages[0].content[0];
@@ -473,7 +477,7 @@ describe("useTamboV1", () => {
                 },
                 currentThreadId: "thread_123",
             };
-            const { result } = (0, react_1.renderHook)(() => (0, use_tambo_v1_1.useTamboV1)(), {
+            const { result } = (0, react_1.renderHook)(() => (0, use_tambo_v1_1.useTambo)(), {
                 wrapper: createWrapperWithState(state),
             });
             expect(result.current.messages[0].content).toHaveLength(3);
@@ -512,7 +516,7 @@ describe("useTamboV1", () => {
                 currentThreadId: "placeholder",
             };
             // No explicit threadId - uses currentThreadId from state
-            const { result } = (0, react_1.renderHook)(() => (0, use_tambo_v1_1.useTamboV1)(), {
+            const { result } = (0, react_1.renderHook)(() => (0, use_tambo_v1_1.useTambo)(), {
                 wrapper: createWrapperWithState(state),
             });
             // Component should render using the placeholder thread
@@ -554,10 +558,11 @@ describe("useTamboV1", () => {
                 },
                 currentThreadId: "thread_123",
             };
-            const { result } = (0, react_1.renderHook)(() => (0, use_tambo_v1_1.useTamboV1)(), {
+            const { result } = (0, react_1.renderHook)(() => (0, use_tambo_v1_1.useTambo)(), {
                 wrapper: createWrapperWithState(state),
             });
-            const content = result.current.messages[0].content[0];
+            const content = result.current.messages[0]
+                .content[0];
             expect(content.type).toBe("tool_use");
             expect(content.hasCompleted).toBe(false);
             expect(content.statusMessage).toBe("Calling getWeather");
@@ -607,10 +612,11 @@ describe("useTamboV1", () => {
                 },
                 currentThreadId: "thread_123",
             };
-            const { result } = (0, react_1.renderHook)(() => (0, use_tambo_v1_1.useTamboV1)(), {
+            const { result } = (0, react_1.renderHook)(() => (0, use_tambo_v1_1.useTambo)(), {
                 wrapper: createWrapperWithState(state),
             });
-            const content = result.current.messages[0].content[0];
+            const content = result.current.messages[0]
+                .content[0];
             expect(content.hasCompleted).toBe(true);
             expect(content.statusMessage).toBe("Called getWeather");
         });
@@ -649,10 +655,11 @@ describe("useTamboV1", () => {
                 },
                 currentThreadId: "thread_123",
             };
-            const { result } = (0, react_1.renderHook)(() => (0, use_tambo_v1_1.useTamboV1)(), {
+            const { result } = (0, react_1.renderHook)(() => (0, use_tambo_v1_1.useTambo)(), {
                 wrapper: createWrapperWithState(state),
             });
-            const content = result.current.messages[0].content[0];
+            const content = result.current.messages[0]
+                .content[0];
             expect(content.statusMessage).toBe("Fetching weather for NYC...");
             expect(content.tamboDisplayProps?._tambo_statusMessage).toBe("Fetching weather for NYC...");
         });
@@ -704,10 +711,11 @@ describe("useTamboV1", () => {
                 },
                 currentThreadId: "thread_123",
             };
-            const { result } = (0, react_1.renderHook)(() => (0, use_tambo_v1_1.useTamboV1)(), {
+            const { result } = (0, react_1.renderHook)(() => (0, use_tambo_v1_1.useTambo)(), {
                 wrapper: createWrapperWithState(state),
             });
-            const content = result.current.messages[0].content[0];
+            const content = result.current.messages[0]
+                .content[0];
             expect(content.hasCompleted).toBe(true);
             expect(content.statusMessage).toBe("Got weather for NYC");
         });
@@ -731,7 +739,6 @@ describe("useTamboV1", () => {
                                                 units: "celsius",
                                                 _tambo_statusMessage: "Fetching...",
                                                 _tambo_completionStatusMessage: "Done",
-                                                _tambo_displayMessage: "Weather lookup",
                                             },
                                         },
                                     ],
@@ -749,15 +756,15 @@ describe("useTamboV1", () => {
                 },
                 currentThreadId: "thread_123",
             };
-            const { result } = (0, react_1.renderHook)(() => (0, use_tambo_v1_1.useTamboV1)(), {
+            const { result } = (0, react_1.renderHook)(() => (0, use_tambo_v1_1.useTambo)(), {
                 wrapper: createWrapperWithState(state),
             });
-            const content = result.current.messages[0].content[0];
+            const content = result.current.messages[0]
+                .content[0];
             // Input should only have non-_tambo_ properties
             expect(content.input).toEqual({ location: "NYC", units: "celsius" });
             expect(content.input._tambo_statusMessage).toBeUndefined();
             expect(content.input._tambo_completionStatusMessage).toBeUndefined();
-            expect(content.input._tambo_displayMessage).toBeUndefined();
         });
         it("handles tool_use with empty input", () => {
             const state = {
@@ -791,10 +798,11 @@ describe("useTamboV1", () => {
                 },
                 currentThreadId: "thread_123",
             };
-            const { result } = (0, react_1.renderHook)(() => (0, use_tambo_v1_1.useTamboV1)(), {
+            const { result } = (0, react_1.renderHook)(() => (0, use_tambo_v1_1.useTambo)(), {
                 wrapper: createWrapperWithState(state),
             });
-            const content = result.current.messages[0].content[0];
+            const content = result.current.messages[0]
+                .content[0];
             expect(content.input).toEqual({});
             expect(content.hasCompleted).toBe(false);
             expect(content.statusMessage).toBe("Calling getCurrentTime");
@@ -831,10 +839,11 @@ describe("useTamboV1", () => {
                 },
                 currentThreadId: "thread_123",
             };
-            const { result } = (0, react_1.renderHook)(() => (0, use_tambo_v1_1.useTamboV1)(), {
+            const { result } = (0, react_1.renderHook)(() => (0, use_tambo_v1_1.useTambo)(), {
                 wrapper: createWrapperWithState(state),
             });
-            const content = result.current.messages[0].content[0];
+            const content = result.current.messages[0]
+                .content[0];
             expect(content.input).toEqual({});
             expect(content.hasCompleted).toBe(false);
         });
@@ -889,11 +898,13 @@ describe("useTamboV1", () => {
                 },
                 currentThreadId: "thread_123",
             };
-            const { result } = (0, react_1.renderHook)(() => (0, use_tambo_v1_1.useTamboV1)(), {
+            const { result } = (0, react_1.renderHook)(() => (0, use_tambo_v1_1.useTambo)(), {
                 wrapper: createWrapperWithState(state),
             });
-            const tool1 = result.current.messages[0].content[0];
-            const tool2 = result.current.messages[0].content[1];
+            const tool1 = result.current.messages[0]
+                .content[0];
+            const tool2 = result.current.messages[0]
+                .content[1];
             expect(tool1.hasCompleted).toBe(true);
             expect(tool1.statusMessage).toBe("Called getWeather");
             expect(tool2.hasCompleted).toBe(false);
@@ -919,7 +930,7 @@ describe("useTamboV1", () => {
                 },
                 currentThreadId: "thread_123",
             };
-            const { result } = (0, react_1.renderHook)(() => (0, use_tambo_v1_1.useTamboV1)(), {
+            const { result } = (0, react_1.renderHook)(() => (0, use_tambo_v1_1.useTambo)(), {
                 wrapper: createWrapperWithRealReducer(initialState),
             });
             // Before cancel: streaming and lastRunCancelled is false
@@ -954,7 +965,7 @@ describe("useTamboV1", () => {
                 },
                 currentThreadId: "thread_123",
             };
-            const { result } = (0, react_1.renderHook)(() => (0, use_tambo_v1_1.useTamboV1)(), {
+            const { result } = (0, react_1.renderHook)(() => (0, use_tambo_v1_1.useTambo)(), {
                 wrapper: createWrapperWithRealReducer(initialState),
             });
             // Idle before
@@ -987,7 +998,7 @@ describe("useTamboV1", () => {
                 },
                 currentThreadId: "placeholder",
             };
-            const { result } = (0, react_1.renderHook)(() => (0, use_tambo_v1_1.useTamboV1)(), {
+            const { result } = (0, react_1.renderHook)(() => (0, use_tambo_v1_1.useTambo)(), {
                 wrapper: createWrapperWithRealReducer(initialState),
             });
             // Before: streaming on placeholder
@@ -1027,7 +1038,7 @@ describe("useTamboV1", () => {
                 },
                 currentThreadId: "thread_123",
             };
-            const { result } = (0, react_1.renderHook)(() => (0, use_tambo_v1_1.useTamboV1)(), {
+            const { result } = (0, react_1.renderHook)(() => (0, use_tambo_v1_1.useTambo)(), {
                 wrapper: createWrapperWithRealReducer(initialState),
             });
             await (0, react_1.act)(async () => {
@@ -1041,5 +1052,122 @@ describe("useTamboV1", () => {
             consoleWarnSpy.mockRestore();
         });
     });
+    describe("updateThreadName", () => {
+        it("updates local thread name and invalidates caches on success", async () => {
+            const mockUpdate = jest.fn().mockResolvedValue({});
+            // TypeScript SDK will be updated to include this method
+            mockTamboClient.threads.update = mockUpdate;
+            const invalidateQueriesSpy = jest.spyOn(queryClient, "invalidateQueries");
+            const initialState = {
+                threadMap: {
+                    thread_456: {
+                        thread: {
+                            id: "thread_456",
+                            messages: [],
+                            status: "idle",
+                            createdAt: new Date().toISOString(),
+                            updatedAt: new Date().toISOString(),
+                            lastRunCancelled: false,
+                            name: "Old Name",
+                        },
+                        streaming: { status: "idle" },
+                        accumulatingToolArgs: new Map(),
+                    },
+                },
+                currentThreadId: "thread_456",
+            };
+            const { result } = (0, react_1.renderHook)(() => (0, use_tambo_v1_1.useTambo)(), {
+                wrapper: createWrapperWithRealReducer(initialState),
+            });
+            await (0, react_1.act)(async () => {
+                await result.current.updateThreadName("thread_456", "My New Thread");
+            });
+            expect(mockUpdate).toHaveBeenCalledWith("thread_456", {
+                name: "My New Thread",
+            });
+            expect(result.current.thread?.thread.name).toBe("My New Thread");
+            const invalidatedKeys = invalidateQueriesSpy.mock.calls
+                .map(([arg]) => arg.queryKey)
+                .filter(Boolean);
+            expect(invalidatedKeys).toContainEqual(["v1-threads", "list"]);
+            expect(invalidatedKeys).toContainEqual(["v1-threads", "thread_456"]);
+        });
+        it("does not create local thread state when thread isn't loaded", async () => {
+            const mockUpdate = jest.fn().mockResolvedValue({});
+            // TypeScript SDK will be updated to include this method
+            mockTamboClient.threads.update = mockUpdate;
+            const invalidateQueriesSpy = jest.spyOn(queryClient, "invalidateQueries");
+            // Track the internal StreamState from the real reducer so we can assert we
+            // don't create a local thread entry for threads that haven't been loaded.
+            const stateRef = {
+                current: undefined,
+            };
+            const initialState = {
+                threadMap: {},
+                currentThreadId: "placeholder",
+            };
+            const { result } = (0, react_1.renderHook)(() => (0, use_tambo_v1_1.useTambo)(), {
+                wrapper: createWrapperWithRealReducer(initialState, stateRef),
+            });
+            await (0, react_1.act)(async () => {
+                await result.current.updateThreadName("thread_789", "New Title");
+            });
+            expect(mockUpdate).toHaveBeenCalledWith("thread_789", {
+                name: "New Title",
+            });
+            const invalidatedKeys = invalidateQueriesSpy.mock.calls
+                .map(([arg]) => arg.queryKey)
+                .filter(Boolean);
+            expect(invalidatedKeys).toContainEqual(["v1-threads", "list"]);
+            expect(invalidatedKeys).toContainEqual(["v1-threads", "thread_789"]);
+            expect(stateRef.current?.threadMap.thread_789).toBeUndefined();
+            (0, react_1.act)(() => {
+                result.current.switchThread("thread_789");
+            });
+            expect(result.current.thread).toBeUndefined();
+        });
+        it("propagates errors and does not update local state", async () => {
+            const mockUpdate = jest
+                .fn()
+                .mockRejectedValue(new Error("Network error"));
+            // TypeScript SDK will be updated to include this method
+            mockTamboClient.threads.update = mockUpdate;
+            const invalidateQueriesSpy = jest.spyOn(queryClient, "invalidateQueries");
+            const initialState = {
+                threadMap: {
+                    thread_456: {
+                        thread: {
+                            id: "thread_456",
+                            messages: [],
+                            status: "idle",
+                            createdAt: new Date().toISOString(),
+                            updatedAt: new Date().toISOString(),
+                            lastRunCancelled: false,
+                            name: "Old Name",
+                        },
+                        streaming: { status: "idle" },
+                        accumulatingToolArgs: new Map(),
+                    },
+                },
+                currentThreadId: "thread_456",
+            };
+            const { result } = (0, react_1.renderHook)(() => (0, use_tambo_v1_1.useTambo)(), {
+                wrapper: createWrapperWithRealReducer(initialState),
+            });
+            let caughtError;
+            await (0, react_1.act)(async () => {
+                try {
+                    await result.current.updateThreadName("thread_456", "My New Thread");
+                }
+                catch (error) {
+                    caughtError = error;
+                }
+            });
+            expect(caughtError).toBeInstanceOf(Error);
+            expect(caughtError.message).toBe("Network error");
+            expect(result.current.thread?.thread.name).toBe("Old Name");
+            expect(invalidateQueriesSpy).not.toHaveBeenCalled();
+        });
+    });
 });
 //# sourceMappingURL=use-tambo-v1.test.js.map