@tambo-ai/react 0.71.0 → 0.72.0

package/esm/v1/hooks/use-tambo-v1-component-state.test.js

@@ -0,0 +1,290 @@
+import { act, renderHook } from "@testing-library/react";
+import { useTamboV1ComponentState } from "./use-tambo-v1-component-state";
+// Mock the required modules
+jest.mock("../../providers/tambo-client-provider", () => ({
+    useTamboClient: jest.fn(),
+}));
+jest.mock("../providers/tambo-v1-stream-context", () => ({
+    useStreamState: jest.fn(),
+}));
+jest.mock("../utils/component-renderer", () => ({
+    useV1ComponentContent: jest.fn(),
+}));
+// Create a mock debounced function with flush method
+const createMockDebouncedFunction = (fn) => {
+    const debouncedFn = jest.fn((...args) => fn(...args));
+    debouncedFn.flush = jest.fn();
+    debouncedFn.cancel = jest.fn();
+    debouncedFn.isPending = jest.fn(() => false);
+    return debouncedFn;
+};
+// Mock use-debounce
+jest.mock("use-debounce", () => ({
+    useDebouncedCallback: jest.fn(),
+}));
+// Import the mocked modules
+import { useTamboClient } from "../../providers/tambo-client-provider";
+import { useStreamState } from "../providers/tambo-v1-stream-context";
+import { useV1ComponentContent } from "../utils/component-renderer";
+import { useDebouncedCallback } from "use-debounce";
+describe("useTamboV1ComponentState", () => {
+    const mockUpdateState = jest.fn();
+    const mockComponentId = "comp_test123";
+    const mockThreadId = "thread_abc";
+    const mockMessageId = "msg_xyz";
+    // Helper to create mock stream state
+    const createMockStreamState = (componentState) => ({
+        threadMap: {
+            [mockThreadId]: {
+                thread: {
+                    id: mockThreadId,
+                    messages: [
+                        {
+                            id: mockMessageId,
+                            role: "assistant",
+                            content: [
+                                {
+                                    type: "component",
+                                    id: mockComponentId,
+                                    name: "TestComponent",
+                                    props: {},
+                                    state: componentState,
+                                    streamingState: "done",
+                                },
+                            ],
+                            createdAt: new Date().toISOString(),
+                        },
+                    ],
+                    status: "idle",
+                    createdAt: new Date().toISOString(),
+                    updatedAt: new Date().toISOString(),
+                },
+                streaming: { status: "idle" },
+                accumulatingToolArgs: new Map(),
+            },
+        },
+        currentThreadId: mockThreadId,
+    });
+    beforeEach(() => {
+        jest.clearAllMocks();
+        // Setup default mocks
+        jest.mocked(useTamboClient).mockReturnValue({
+            threads: {
+                state: {
+                    updateState: mockUpdateState,
+                },
+            },
+        });
+        jest.mocked(useV1ComponentContent).mockReturnValue({
+            componentId: mockComponentId,
+            threadId: mockThreadId,
+            messageId: mockMessageId,
+            componentName: "TestComponent",
+        });
+        jest.mocked(useStreamState).mockReturnValue(createMockStreamState());
+        // Setup default mock for useDebouncedCallback
+        jest
+            .mocked(useDebouncedCallback)
+            .mockImplementation((fn) => createMockDebouncedFunction(fn));
+    });
+    afterEach(() => {
+        jest.restoreAllMocks();
+    });
+    describe("Initial State Management", () => {
+        it("should initialize with initialValue when no server state exists", () => {
+            const initialValue = "test-initial";
+            jest.mocked(useStreamState).mockReturnValue(createMockStreamState({}));
+            const { result } = renderHook(() => useTamboV1ComponentState("testKey", initialValue));
+            expect(result.current[0]).toBe(initialValue);
+        });
+        it("should use server state over initialValue", () => {
+            const initialValue = "initial";
+            const serverValue = "server-value";
+            jest
+                .mocked(useStreamState)
+                .mockReturnValue(createMockStreamState({ testKey: serverValue }));
+            const { result } = renderHook(() => useTamboV1ComponentState("testKey", initialValue));
+            expect(result.current[0]).toBe(serverValue);
+        });
+        it("should handle undefined initialValue gracefully", () => {
+            jest.mocked(useStreamState).mockReturnValue(createMockStreamState({}));
+            const { result } = renderHook(() => useTamboV1ComponentState("testKey"));
+            expect(result.current[0]).toBeUndefined();
+        });
+        it("should handle different data types correctly", () => {
+            const testCases = [
+                { value: "string" },
+                { value: 42 },
+                { value: true },
+                { value: { name: "test" } },
+                { value: [1, 2, 3] },
+            ];
+            testCases.forEach(({ value }) => {
+                jest
+                    .mocked(useStreamState)
+                    .mockReturnValue(createMockStreamState({ testKey: value }));
+                const { result } = renderHook(() => useTamboV1ComponentState("testKey", value));
+                expect(result.current[0]).toEqual(value);
+            });
+        });
+    });
+    describe("State Updates", () => {
+        it("should update local state immediately when setState is called", () => {
+            const initialValue = "initial";
+            jest
+                .mocked(useStreamState)
+                .mockReturnValue(createMockStreamState({ testKey: initialValue }));
+            const { result } = renderHook(() => useTamboV1ComponentState("testKey", initialValue));
+            const newValue = "updated";
+            act(() => {
+                result.current[1](newValue);
+            });
+            expect(result.current[0]).toBe(newValue);
+        });
+        it("should support functional updates", () => {
+            const initialValue = 5;
+            jest
+                .mocked(useStreamState)
+                .mockReturnValue(createMockStreamState({ counter: initialValue }));
+            const { result } = renderHook(() => useTamboV1ComponentState("counter", initialValue));
+            act(() => {
+                result.current[1]((prev) => (prev ?? 0) + 1);
+            });
+            expect(result.current[0]).toBe(6);
+        });
+        it("should trigger debounced API call when setState is called", () => {
+            const initialValue = "initial";
+            jest
+                .mocked(useStreamState)
+                .mockReturnValue(createMockStreamState({ testKey: initialValue }));
+            const { result } = renderHook(() => useTamboV1ComponentState("testKey", initialValue));
+            const newValue = "updated";
+            act(() => {
+                result.current[1](newValue);
+            });
+            // The debounced function should be called
+            expect(mockUpdateState).toHaveBeenCalledWith(mockComponentId, {
+                threadId: mockThreadId,
+                state: { testKey: newValue },
+            });
+        });
+        it("should work with complex objects", () => {
+            const initialValue = { name: "test", items: [1, 2, 3] };
+            jest
+                .mocked(useStreamState)
+                .mockReturnValue(createMockStreamState({ data: initialValue }));
+            const { result } = renderHook(() => useTamboV1ComponentState("data", initialValue));
+            const newValue = { name: "updated", items: [4, 5, 6] };
+            act(() => {
+                result.current[1](newValue);
+            });
+            expect(result.current[0]).toEqual(newValue);
+        });
+    });
+    describe("Metadata", () => {
+        it("should return isPending and error in meta", () => {
+            jest.mocked(useStreamState).mockReturnValue(createMockStreamState({}));
+            const { result } = renderHook(() => useTamboV1ComponentState("testKey", "initial"));
+            expect(result.current[2]).toEqual({
+                isPending: false,
+                error: null,
+                flush: expect.any(Function),
+            });
+        });
+        it("should provide a flush function", () => {
+            const mockFlush = jest.fn();
+            const mockDebouncedFn = createMockDebouncedFunction(jest.fn());
+            mockDebouncedFn.flush = mockFlush;
+            jest.mocked(useDebouncedCallback).mockReturnValue(mockDebouncedFn);
+            jest.mocked(useStreamState).mockReturnValue(createMockStreamState({}));
+            const { result } = renderHook(() => useTamboV1ComponentState("testKey", "initial"));
+            act(() => {
+                result.current[2].flush();
+            });
+            expect(mockFlush).toHaveBeenCalled();
+        });
+    });
+    describe("Debouncing Behavior", () => {
+        it("should use default debounce time of 500ms", () => {
+            jest.mocked(useStreamState).mockReturnValue(createMockStreamState({}));
+            renderHook(() => useTamboV1ComponentState("testKey", "initial"));
+            expect(useDebouncedCallback).toHaveBeenCalledWith(expect.any(Function), 500);
+        });
+        it("should use custom debounce time when provided", () => {
+            jest.mocked(useStreamState).mockReturnValue(createMockStreamState({}));
+            const customDebounceTime = 1000;
+            renderHook(() => useTamboV1ComponentState("testKey", "initial", customDebounceTime));
+            expect(useDebouncedCallback).toHaveBeenCalledWith(expect.any(Function), customDebounceTime);
+        });
+        it("should flush debounced callback on unmount", () => {
+            const mockFlush = jest.fn();
+            const mockDebouncedFn = createMockDebouncedFunction(jest.fn());
+            mockDebouncedFn.flush = mockFlush;
+            jest.mocked(useDebouncedCallback).mockReturnValue(mockDebouncedFn);
+            jest.mocked(useStreamState).mockReturnValue(createMockStreamState({}));
+            const { unmount } = renderHook(() => useTamboV1ComponentState("testKey", "initial"));
+            unmount();
+            expect(mockFlush).toHaveBeenCalled();
+        });
+    });
+    describe("Server State Sync", () => {
+        it("should sync with server state changes from streaming", () => {
+            const streamState = createMockStreamState({ testKey: "initial" });
+            jest.mocked(useStreamState).mockReturnValue(streamState);
+            const { result, rerender } = renderHook(() => useTamboV1ComponentState("testKey", "initial"));
+            expect(result.current[0]).toBe("initial");
+            // Simulate server state change from streaming
+            const updatedStreamState = createMockStreamState({
+                testKey: "updated-from-server",
+            });
+            jest.mocked(useStreamState).mockReturnValue(updatedStreamState);
+            rerender();
+            expect(result.current[0]).toBe("updated-from-server");
+        });
+        it("should handle component not found in stream state", () => {
+            // Empty stream state (no matching component)
+            jest.mocked(useStreamState).mockReturnValue({
+                threadMap: {},
+                currentThreadId: null,
+            });
+            const { result } = renderHook(() => useTamboV1ComponentState("testKey", "default"));
+            expect(result.current[0]).toBe("default");
+        });
+    });
+    describe("Error Handling", () => {
+        it("should handle API errors gracefully", async () => {
+            const mockError = new Error("API Error");
+            mockUpdateState.mockRejectedValue(mockError);
+            // Create a sync mock that calls the function immediately
+            jest.mocked(useDebouncedCallback).mockImplementation((fn) => {
+                const debouncedFn = Object.assign(async (...args) => fn(...args), {
+                    flush: jest.fn(),
+                    cancel: jest.fn(),
+                    isPending: () => false,
+                });
+                return debouncedFn;
+            });
+            jest.mocked(useStreamState).mockReturnValue(createMockStreamState({}));
+            const consoleSpy = jest.spyOn(console, "error").mockImplementation();
+            const { result } = renderHook(() => useTamboV1ComponentState("testKey", "initial"));
+            await act(async () => {
+                result.current[1]("new-value");
+                // Wait for async operations
+                await new Promise((resolve) => setTimeout(resolve, 0));
+            });
+            expect(consoleSpy).toHaveBeenCalledWith(expect.stringContaining("Failed to sync state"), mockError);
+            consoleSpy.mockRestore();
+        });
+    });
+    describe("Context Requirements", () => {
+        it("should throw when used outside component content context", () => {
+            jest.mocked(useV1ComponentContent).mockImplementation(() => {
+                throw new Error("useV1ComponentContent must be used within a rendered component");
+            });
+            expect(() => {
+                renderHook(() => useTamboV1ComponentState("testKey", "initial"));
+            }).toThrow("useV1ComponentContent must be used within a rendered component");
+        });
+    });
+});
+//# sourceMappingURL=use-tambo-v1-component-state.test.js.map

package/esm/v1/hooks/use-tambo-v1-component-state.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"use-tambo-v1-component-state.test.js","sourceRoot":"","sources":["../../../src/v1/hooks/use-tambo-v1-component-state.test.tsx"],"names":[],"mappings":"AAAA,OAAO,EAAE,GAAG,EAAE,UAAU,EAAE,MAAM,wBAAwB,CAAC;AACzD,OAAO,EAAE,wBAAwB,EAAE,MAAM,gCAAgC,CAAC;AAE1E,4BAA4B;AAC5B,IAAI,CAAC,IAAI,CAAC,uCAAuC,EAAE,GAAG,EAAE,CAAC,CAAC;IACxD,cAAc,EAAE,IAAI,CAAC,EAAE,EAAE;CAC1B,CAAC,CAAC,CAAC;AAEJ,IAAI,CAAC,IAAI,CAAC,sCAAsC,EAAE,GAAG,EAAE,CAAC,CAAC;IACvD,cAAc,EAAE,IAAI,CAAC,EAAE,EAAE;CAC1B,CAAC,CAAC,CAAC;AAEJ,IAAI,CAAC,IAAI,CAAC,6BAA6B,EAAE,GAAG,EAAE,CAAC,CAAC;IAC9C,qBAAqB,EAAE,IAAI,CAAC,EAAE,EAAE;CACjC,CAAC,CAAC,CAAC;AAEJ,qDAAqD;AACrD,MAAM,2BAA2B,GAAG,CAAC,EAAmC,EAAE,EAAE;IAC1E,MAAM,WAAW,GAAG,IAAI,CAAC,EAAE,CAAC,CAAC,GAAG,IAAe,EAAE,EAAE,CACjD,EAAE,CAAC,GAAG,IAAI,CAAC,CAKZ,CAAC;IACF,WAAW,CAAC,KAAK,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC;IAC9B,WAAW,CAAC,MAAM,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC;IAC/B,WAAW,CAAC,SAAS,GAAG,IAAI,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,KAAK,CAAC,CAAC;IAC7C,OAAO,WAAW,CAAC;AACrB,CAAC,CAAC;AAEF,oBAAoB;AACpB,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,GAAG,EAAE,CAAC,CAAC;IAC/B,oBAAoB,EAAE,IAAI,CAAC,EAAE,EAAE;CAChC,CAAC,CAAC,CAAC;AAEJ,4BAA4B;AAC5B,OAAO,EAAE,cAAc,EAAE,MAAM,uCAAuC,CAAC;AACvE,OAAO,EAAE,cAAc,EAAE,MAAM,sCAAsC,CAAC;AACtE,OAAO,EAAE,qBAAqB,EAAE,MAAM,6BAA6B,CAAC;AACpE,OAAO,EAAE,oBAAoB,EAAE,MAAM,cAAc,CAAC;AAIpD,QAAQ,CAAC,0BAA0B,EAAE,GAAG,EAAE;IACxC,MAAM,eAAe,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC;IAClC,MAAM,eAAe,GAAG,cAAc,CAAC;IACvC,MAAM,YAAY,GAAG,YAAY,CAAC;IAClC,MAAM,aAAa,GAAG,SAAS,CAAC;IAEhC,qCAAqC;IACrC,MAAM,qBAAqB,GAAG,CAC5B,cAAwC,EAC3B,EAAE,CAAC,CAAC;QACjB,SAAS,EAAE;YACT,CAAC,YAAY,CAAC,EAAE;gBACd,MAAM,EAAE;oBACN,EAAE,EAAE,YAAY;oBAChB,QAAQ,EAAE;wBACR;4BACE,EAAE,EAAE,aAAa;4BACjB,IAAI,EAAE,WAAW;4BACjB,OAAO,EAAE;gCACP;oCACE,IAAI,EAAE,WAAW;oCACjB,EAAE,EAAE,eAAe;oCACnB,IAAI,EAAE,eAAe;oCACrB,KAAK,EAAE,EAAE;oCACT,KAAK,EAAE,cAAc;oCACrB,cAAc,EAAE,MAAM;iCACD;6BACxB;4BACD,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;yBACpC;qBACF;oBACD,MAAM,EAAE,MAAM;oBACd,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;oBACnC,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;iBACpC;gBACD,SAAS,EAAE,EAAE,MAAM,EAAE,MAAM,EAAE;gBAC7B,oBAAoB,EAAE,IAAI,GAAG,EAAE;aAChC;SACF;QACD,eAAe,EAAE,YAAY;KAC9B,CAAC,CAAC;IAEH,UAAU,CAAC,GAAG,EAAE;QACd,IAAI,CAAC,aAAa,EAAE,CAAC;QAErB,sBAAsB;QACtB,IAAI,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,eAAe,CAAC;YAC1C,OAAO,EAAE;gBACP,KAAK,EAAE;oBACL,WAAW,EAAE,eAAe;iBAC7B;aACF;SAC8C,CAAC,CAAC;QAEnD,IAAI,CAAC,MAAM,CAAC,qBAAqB,CAAC,CAAC,eAAe,CAAC;YACjD,WAAW,EAAE,eAAe;YAC5B,QAAQ,EAAE,YAAY;YACtB,SAAS,EAAE,aAAa;YACxB,aAAa,EAAE,eAAe;SAC/B,CAAC,CAAC;QAEH,IAAI,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,eAAe,CAAC,qBAAqB,EAAE,CAAC,CAAC;QAErE,8CAA8C;QAC9C,IAAI;aACD,MAAM,CAAC,oBAAoB,CAAC;aAC5B,kBAAkB,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,2BAA2B,CAAC,EAAE,CAAC,CAAC,CAAC;IACjE,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,IAAI,CAAC,eAAe,EAAE,CAAC;IACzB,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,0BAA0B,EAAE,GAAG,EAAE;QACxC,EAAE,CAAC,iEAAiE,EAAE,GAAG,EAAE;YACzE,MAAM,YAAY,GAAG,cAAc,CAAC;YACpC,IAAI,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,eAAe,CAAC,qBAAqB,CAAC,EAAE,CAAC,CAAC,CAAC;YAEvE,MAAM,EAAE,MAAM,EAAE,GAAG,UAAU,CAAC,GAAG,EAAE,CACjC,wBAAwB,CAAC,SAAS,EAAE,YAAY,CAAC,CAClD,CAAC;YAEF,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;QAC/C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,2CAA2C,EAAE,GAAG,EAAE;YACnD,MAAM,YAAY,GAAG,SAAS,CAAC;YAC/B,MAAM,WAAW,GAAG,cAAc,CAAC;YACnC,IAAI;iBACD,MAAM,CAAC,cAAc,CAAC;iBACtB,eAAe,CAAC,qBAAqB,CAAC,EAAE,OAAO,EAAE,WAAW,EAAE,CAAC,CAAC,CAAC;YAEpE,MAAM,EAAE,MAAM,EAAE,GAAG,UAAU,CAAC,GAAG,EAAE,CACjC,wBAAwB,CAAC,SAAS,EAAE,YAAY,CAAC,CAClD,CAAC;YAEF,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;QAC9C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,iDAAiD,EAAE,GAAG,EAAE;YACzD,IAAI,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,eAAe,CAAC,qBAAqB,CAAC,EAAE,CAAC,CAAC,CAAC;YAEvE,MAAM,EAAE,MAAM,EAAE,GAAG,UAAU,CAAC,GAAG,EAAE,CAAC,wBAAwB,CAAC,SAAS,CAAC,CAAC,CAAC;YAEzE,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,aAAa,EAAE,CAAC;QAC5C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,8CAA8C,EAAE,GAAG,EAAE;YACtD,MAAM,SAAS,GAAG;gBAChB,EAAE,KAAK,EAAE,QAAQ,EAAE;gBACnB,EAAE,KAAK,EAAE,EAAE,EAAE;gBACb,EAAE,KAAK,EAAE,IAAI,EAAE;gBACf,EAAE,KAAK,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,EAAE;gBAC3B,EAAE,KAAK,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,EAAE;aACrB,CAAC;YAEF,SAAS,CAAC,OAAO,CAAC,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE;gBAC9B,IAAI;qBACD,MAAM,CAAC,cAAc,CAAC;qBACtB,eAAe,CAAC,qBAAqB,CAAC,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC,CAAC,CAAC;gBAE9D,MAAM,EAAE,MAAM,EAAE,GAAG,UAAU,CAAC,GAAG,EAAE,CACjC,wBAAwB,CAAC,SAAS,EAAE,KAAK,CAAC,CAC3C,CAAC;gBAEF,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;YAC3C,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,eAAe,EAAE,GAAG,EAAE;QAC7B,EAAE,CAAC,+DAA+D,EAAE,GAAG,EAAE;YACvE,MAAM,YAAY,GAAG,SAAS,CAAC;YAC/B,IAAI;iBACD,MAAM,CAAC,cAAc,CAAC;iBACtB,eAAe,CAAC,qBAAqB,CAAC,EAAE,OAAO,EAAE,YAAY,EAAE,CAAC,CAAC,CAAC;YAErE,MAAM,EAAE,MAAM,EAAE,GAAG,UAAU,CAAC,GAAG,EAAE,CACjC,wBAAwB,CAAC,SAAS,EAAE,YAAY,CAAC,CAClD,CAAC;YAEF,MAAM,QAAQ,GAAG,SAAS,CAAC;YAC3B,GAAG,CAAC,GAAG,EAAE;gBACP,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC;YAC9B,CAAC,CAAC,CAAC;YAEH,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAC3C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,mCAAmC,EAAE,GAAG,EAAE;YAC3C,MAAM,YAAY,GAAG,CAAC,CAAC;YACvB,IAAI;iBACD,MAAM,CAAC,cAAc,CAAC;iBACtB,eAAe,CAAC,qBAAqB,CAAC,EAAE,OAAO,EAAE,YAAY,EAAE,CAAC,CAAC,CAAC;YAErE,MAAM,EAAE,MAAM,EAAE,GAAG,UAAU,CAAC,GAAG,EAAE,CACjC,wBAAwB,CAAC,SAAS,EAAE,YAAY,CAAC,CAClD,CAAC;YAEF,GAAG,CAAC,GAAG,EAAE;gBACP,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,IAAI,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC;YAC/C,CAAC,CAAC,CAAC;YAEH,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACpC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,2DAA2D,EAAE,GAAG,EAAE;YACnE,MAAM,YAAY,GAAG,SAAS,CAAC;YAC/B,IAAI;iBACD,MAAM,CAAC,cAAc,CAAC;iBACtB,eAAe,CAAC,qBAAqB,CAAC,EAAE,OAAO,EAAE,YAAY,EAAE,CAAC,CAAC,CAAC;YAErE,MAAM,EAAE,MAAM,EAAE,GAAG,UAAU,CAAC,GAAG,EAAE,CACjC,wBAAwB,CAAC,SAAS,EAAE,YAAY,CAAC,CAClD,CAAC;YAEF,MAAM,QAAQ,GAAG,SAAS,CAAC;YAC3B,GAAG,CAAC,GAAG,EAAE;gBACP,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC;YAC9B,CAAC,CAAC,CAAC;YAEH,0CAA0C;YAC1C,MAAM,CAAC,eAAe,CAAC,CAAC,oBAAoB,CAAC,eAAe,EAAE;gBAC5D,QAAQ,EAAE,YAAY;gBACtB,KAAK,EAAE,EAAE,OAAO,EAAE,QAAQ,EAAE;aAC7B,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,kCAAkC,EAAE,GAAG,EAAE;YAC1C,MAAM,YAAY,GAAG,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC;YACxD,IAAI;iBACD,MAAM,CAAC,cAAc,CAAC;iBACtB,eAAe,CAAC,qBAAqB,CAAC,EAAE,IAAI,EAAE,YAAY,EAAE,CAAC,CAAC,CAAC;YAElE,MAAM,EAAE,MAAM,EAAE,GAAG,UAAU,CAAC,GAAG,EAAE,CACjC,wBAAwB,CAAC,MAAM,EAAE,YAAY,CAAC,CAC/C,CAAC;YAEF,MAAM,QAAQ,GAAG,EAAE,IAAI,EAAE,SAAS,EAAE,KAAK,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC;YACvD,GAAG,CAAC,GAAG,EAAE;gBACP,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC;YAC9B,CAAC,CAAC,CAAC;YAEH,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;QAC9C,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,UAAU,EAAE,GAAG,EAAE;QACxB,EAAE,CAAC,2CAA2C,EAAE,GAAG,EAAE;YACnD,IAAI,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,eAAe,CAAC,qBAAqB,CAAC,EAAE,CAAC,CAAC,CAAC;YAEvE,MAAM,EAAE,MAAM,EAAE,GAAG,UAAU,CAAC,GAAG,EAAE,CACjC,wBAAwB,CAAC,SAAS,EAAE,SAAS,CAAC,CAC/C,CAAC;YAEF,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC;gBAChC,SAAS,EAAE,KAAK;gBAChB,KAAK,EAAE,IAAI;gBACX,KAAK,EAAE,MAAM,CAAC,GAAG,CAAC,QAAQ,CAAC;aAC5B,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,iCAAiC,EAAE,GAAG,EAAE;YACzC,MAAM,SAAS,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC;YAC5B,MAAM,eAAe,GAAG,2BAA2B,CAAC,IAAI,CAAC,EAAE,EAAE,CAAC,CAAC;YAC/D,eAAe,CAAC,KAAK,GAAG,SAAS,CAAC;YAClC,IAAI,CAAC,MAAM,CAAC,oBAAoB,CAAC,CAAC,eAAe,CAAC,eAAe,CAAC,CAAC;YAEnE,IAAI,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,eAAe,CAAC,qBAAqB,CAAC,EAAE,CAAC,CAAC,CAAC;YAEvE,MAAM,EAAE,MAAM,EAAE,GAAG,UAAU,CAAC,GAAG,EAAE,CACjC,wBAAwB,CAAC,SAAS,EAAE,SAAS,CAAC,CAC/C,CAAC;YAEF,GAAG,CAAC,GAAG,EAAE;gBACP,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,KAAK,EAAE,CAAC;YAC5B,CAAC,CAAC,CAAC;YAEH,MAAM,CAAC,SAAS,CAAC,CAAC,gBAAgB,EAAE,CAAC;QACvC,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,qBAAqB,EAAE,GAAG,EAAE;QACnC,EAAE,CAAC,2CAA2C,EAAE,GAAG,EAAE;YACnD,IAAI,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,eAAe,CAAC,qBAAqB,CAAC,EAAE,CAAC,CAAC,CAAC;YAEvE,UAAU,CAAC,GAAG,EAAE,CAAC,wBAAwB,CAAC,SAAS,EAAE,SAAS,CAAC,CAAC,CAAC;YAEjE,MAAM,CAAC,oBAAoB,CAAC,CAAC,oBAAoB,CAC/C,MAAM,CAAC,GAAG,CAAC,QAAQ,CAAC,EACpB,GAAG,CACJ,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,+CAA+C,EAAE,GAAG,EAAE;YACvD,IAAI,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,eAAe,CAAC,qBAAqB,CAAC,EAAE,CAAC,CAAC,CAAC;YACvE,MAAM,kBAAkB,GAAG,IAAI,CAAC;YAEhC,UAAU,CAAC,GAAG,EAAE,CACd,wBAAwB,CAAC,SAAS,EAAE,SAAS,EAAE,kBAAkB,CAAC,CACnE,CAAC;YAEF,MAAM,CAAC,oBAAoB,CAAC,CAAC,oBAAoB,CAC/C,MAAM,CAAC,GAAG,CAAC,QAAQ,CAAC,EACpB,kBAAkB,CACnB,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,4CAA4C,EAAE,GAAG,EAAE;YACpD,MAAM,SAAS,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC;YAC5B,MAAM,eAAe,GAAG,2BAA2B,CAAC,IAAI,CAAC,EAAE,EAAE,CAAC,CAAC;YAC/D,eAAe,CAAC,KAAK,GAAG,SAAS,CAAC;YAClC,IAAI,CAAC,MAAM,CAAC,oBAAoB,CAAC,CAAC,eAAe,CAAC,eAAe,CAAC,CAAC;YAEnE,IAAI,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,eAAe,CAAC,qBAAqB,CAAC,EAAE,CAAC,CAAC,CAAC;YAEvE,MAAM,EAAE,OAAO,EAAE,GAAG,UAAU,CAAC,GAAG,EAAE,CAClC,wBAAwB,CAAC,SAAS,EAAE,SAAS,CAAC,CAC/C,CAAC;YAEF,OAAO,EAAE,CAAC;YAEV,MAAM,CAAC,SAAS,CAAC,CAAC,gBAAgB,EAAE,CAAC;QACvC,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,mBAAmB,EAAE,GAAG,EAAE;QACjC,EAAE,CAAC,sDAAsD,EAAE,GAAG,EAAE;YAC9D,MAAM,WAAW,GAAG,qBAAqB,CAAC,EAAE,OAAO,EAAE,SAAS,EAAE,CAAC,CAAC;YAClE,IAAI,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,eAAe,CAAC,WAAW,CAAC,CAAC;YAEzD,MAAM,EAAE,MAAM,EAAE,QAAQ,EAAE,GAAG,UAAU,CAAC,GAAG,EAAE,CAC3C,wBAAwB,CAAC,SAAS,EAAE,SAAS,CAAC,CAC/C,CAAC;YAEF,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;YAE1C,8CAA8C;YAC9C,MAAM,kBAAkB,GAAG,qBAAqB,CAAC;gBAC/C,OAAO,EAAE,qBAAqB;aAC/B,CAAC,CAAC;YACH,IAAI,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,eAAe,CAAC,kBAAkB,CAAC,CAAC;YAEhE,QAAQ,EAAE,CAAC;YAEX,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,qBAAqB,CAAC,CAAC;QACxD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,mDAAmD,EAAE,GAAG,EAAE;YAC3D,6CAA6C;YAC7C,IAAI,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,eAAe,CAAC;gBAC1C,SAAS,EAAE,EAAE;gBACb,eAAe,EAAE,IAAI;aACtB,CAAC,CAAC;YAEH,MAAM,EAAE,MAAM,EAAE,GAAG,UAAU,CAAC,GAAG,EAAE,CACjC,wBAAwB,CAAC,SAAS,EAAE,SAAS,CAAC,CAC/C,CAAC;YAEF,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QAC5C,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,gBAAgB,EAAE,GAAG,EAAE;QAC9B,EAAE,CAAC,qCAAqC,EAAE,KAAK,IAAI,EAAE;YACnD,MAAM,SAAS,GAAG,IAAI,KAAK,CAAC,WAAW,CAAC,CAAC;YACzC,eAAe,CAAC,iBAAiB,CAAC,SAAS,CAAC,CAAC;YAE7C,yDAAyD;YACzD,IAAI,CAAC,MAAM,CAAC,oBAAoB,CAAC,CAAC,kBAAkB,CAAC,CAAC,EAAE,EAAE,EAAE;gBAC1D,MAAM,WAAW,GAAG,MAAM,CAAC,MAAM,CAC/B,KAAK,EAAE,GAAG,IAAe,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,EACzC;oBACE,KAAK,EAAE,IAAI,CAAC,EAAE,EAAE;oBAChB,MAAM,EAAE,IAAI,CAAC,EAAE,EAAE;oBACjB,SAAS,EAAE,GAAG,EAAE,CAAC,KAAK;iBACvB,CACoD,CAAC;gBACxD,OAAO,WAAW,CAAC;YACrB,CAAC,CAAC,CAAC;YAEH,IAAI,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,eAAe,CAAC,qBAAqB,CAAC,EAAE,CAAC,CAAC,CAAC;YAEvE,MAAM,UAAU,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC,kBAAkB,EAAE,CAAC;YAErE,MAAM,EAAE,MAAM,EAAE,GAAG,UAAU,CAAC,GAAG,EAAE,CACjC,wBAAwB,CAAC,SAAS,EAAE,SAAS,CAAC,CAC/C,CAAC;YAEF,MAAM,GAAG,CAAC,KAAK,IAAI,EAAE;gBACnB,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC;gBAC/B,4BAA4B;gBAC5B,MAAM,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,CAAC;YACzD,CAAC,CAAC,CAAC;YAEH,MAAM,CAAC,UAAU,CAAC,CAAC,oBAAoB,CACrC,MAAM,CAAC,gBAAgB,CAAC,sBAAsB,CAAC,EAC/C,SAAS,CACV,CAAC;YAEF,UAAU,CAAC,WAAW,EAAE,CAAC;QAC3B,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,sBAAsB,EAAE,GAAG,EAAE;QACpC,EAAE,CAAC,0DAA0D,EAAE,GAAG,EAAE;YAClE,IAAI,CAAC,MAAM,CAAC,qBAAqB,CAAC,CAAC,kBAAkB,CAAC,GAAG,EAAE;gBACzD,MAAM,IAAI,KAAK,CACb,gEAAgE,CACjE,CAAC;YACJ,CAAC,CAAC,CAAC;YAEH,MAAM,CAAC,GAAG,EAAE;gBACV,UAAU,CAAC,GAAG,EAAE,CAAC,wBAAwB,CAAC,SAAS,EAAE,SAAS,CAAC,CAAC,CAAC;YACnE,CAAC,CAAC,CAAC,OAAO,CACR,gEAAgE,CACjE,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC","sourcesContent":["import { act, renderHook } from \"@testing-library/react\";\nimport { useTamboV1ComponentState } from \"./use-tambo-v1-component-state\";\n\n// Mock the required modules\njest.mock(\"../../providers/tambo-client-provider\", () => ({\n  useTamboClient: jest.fn(),\n}));\n\njest.mock(\"../providers/tambo-v1-stream-context\", () => ({\n  useStreamState: jest.fn(),\n}));\n\njest.mock(\"../utils/component-renderer\", () => ({\n  useV1ComponentContent: jest.fn(),\n}));\n\n// Create a mock debounced function with flush method\nconst createMockDebouncedFunction = (fn: (...args: unknown[]) => unknown) => {\n  const debouncedFn = jest.fn((...args: unknown[]) =>\n    fn(...args),\n  ) as jest.Mock & {\n    flush: jest.Mock;\n    cancel: jest.Mock;\n    isPending: () => boolean;\n  };\n  debouncedFn.flush = jest.fn();\n  debouncedFn.cancel = jest.fn();\n  debouncedFn.isPending = jest.fn(() => false);\n  return debouncedFn;\n};\n\n// Mock use-debounce\njest.mock(\"use-debounce\", () => ({\n  useDebouncedCallback: jest.fn(),\n}));\n\n// Import the mocked modules\nimport { useTamboClient } from \"../../providers/tambo-client-provider\";\nimport { useStreamState } from \"../providers/tambo-v1-stream-context\";\nimport { useV1ComponentContent } from \"../utils/component-renderer\";\nimport { useDebouncedCallback } from \"use-debounce\";\nimport type { StreamState } from \"../utils/event-accumulator\";\nimport type { V1ComponentContent } from \"../types/message\";\n\ndescribe(\"useTamboV1ComponentState\", () => {\n  const mockUpdateState = jest.fn();\n  const mockComponentId = \"comp_test123\";\n  const mockThreadId = \"thread_abc\";\n  const mockMessageId = \"msg_xyz\";\n\n  // Helper to create mock stream state\n  const createMockStreamState = (\n    componentState?: Record<string, unknown>,\n  ): StreamState => ({\n    threadMap: {\n      [mockThreadId]: {\n        thread: {\n          id: mockThreadId,\n          messages: [\n            {\n              id: mockMessageId,\n              role: \"assistant\",\n              content: [\n                {\n                  type: \"component\",\n                  id: mockComponentId,\n                  name: \"TestComponent\",\n                  props: {},\n                  state: componentState,\n                  streamingState: \"done\",\n                } as V1ComponentContent,\n              ],\n              createdAt: new Date().toISOString(),\n            },\n          ],\n          status: \"idle\",\n          createdAt: new Date().toISOString(),\n          updatedAt: new Date().toISOString(),\n        },\n        streaming: { status: \"idle\" },\n        accumulatingToolArgs: new Map(),\n      },\n    },\n    currentThreadId: mockThreadId,\n  });\n\n  beforeEach(() => {\n    jest.clearAllMocks();\n\n    // Setup default mocks\n    jest.mocked(useTamboClient).mockReturnValue({\n      threads: {\n        state: {\n          updateState: mockUpdateState,\n        },\n      },\n    } as unknown as ReturnType<typeof useTamboClient>);\n\n    jest.mocked(useV1ComponentContent).mockReturnValue({\n      componentId: mockComponentId,\n      threadId: mockThreadId,\n      messageId: mockMessageId,\n      componentName: \"TestComponent\",\n    });\n\n    jest.mocked(useStreamState).mockReturnValue(createMockStreamState());\n\n    // Setup default mock for useDebouncedCallback\n    jest\n      .mocked(useDebouncedCallback)\n      .mockImplementation((fn) => createMockDebouncedFunction(fn));\n  });\n\n  afterEach(() => {\n    jest.restoreAllMocks();\n  });\n\n  describe(\"Initial State Management\", () => {\n    it(\"should initialize with initialValue when no server state exists\", () => {\n      const initialValue = \"test-initial\";\n      jest.mocked(useStreamState).mockReturnValue(createMockStreamState({}));\n\n      const { result } = renderHook(() =>\n        useTamboV1ComponentState(\"testKey\", initialValue),\n      );\n\n      expect(result.current[0]).toBe(initialValue);\n    });\n\n    it(\"should use server state over initialValue\", () => {\n      const initialValue = \"initial\";\n      const serverValue = \"server-value\";\n      jest\n        .mocked(useStreamState)\n        .mockReturnValue(createMockStreamState({ testKey: serverValue }));\n\n      const { result } = renderHook(() =>\n        useTamboV1ComponentState(\"testKey\", initialValue),\n      );\n\n      expect(result.current[0]).toBe(serverValue);\n    });\n\n    it(\"should handle undefined initialValue gracefully\", () => {\n      jest.mocked(useStreamState).mockReturnValue(createMockStreamState({}));\n\n      const { result } = renderHook(() => useTamboV1ComponentState(\"testKey\"));\n\n      expect(result.current[0]).toBeUndefined();\n    });\n\n    it(\"should handle different data types correctly\", () => {\n      const testCases = [\n        { value: \"string\" },\n        { value: 42 },\n        { value: true },\n        { value: { name: \"test\" } },\n        { value: [1, 2, 3] },\n      ];\n\n      testCases.forEach(({ value }) => {\n        jest\n          .mocked(useStreamState)\n          .mockReturnValue(createMockStreamState({ testKey: value }));\n\n        const { result } = renderHook(() =>\n          useTamboV1ComponentState(\"testKey\", value),\n        );\n\n        expect(result.current[0]).toEqual(value);\n      });\n    });\n  });\n\n  describe(\"State Updates\", () => {\n    it(\"should update local state immediately when setState is called\", () => {\n      const initialValue = \"initial\";\n      jest\n        .mocked(useStreamState)\n        .mockReturnValue(createMockStreamState({ testKey: initialValue }));\n\n      const { result } = renderHook(() =>\n        useTamboV1ComponentState(\"testKey\", initialValue),\n      );\n\n      const newValue = \"updated\";\n      act(() => {\n        result.current[1](newValue);\n      });\n\n      expect(result.current[0]).toBe(newValue);\n    });\n\n    it(\"should support functional updates\", () => {\n      const initialValue = 5;\n      jest\n        .mocked(useStreamState)\n        .mockReturnValue(createMockStreamState({ counter: initialValue }));\n\n      const { result } = renderHook(() =>\n        useTamboV1ComponentState(\"counter\", initialValue),\n      );\n\n      act(() => {\n        result.current[1]((prev) => (prev ?? 0) + 1);\n      });\n\n      expect(result.current[0]).toBe(6);\n    });\n\n    it(\"should trigger debounced API call when setState is called\", () => {\n      const initialValue = \"initial\";\n      jest\n        .mocked(useStreamState)\n        .mockReturnValue(createMockStreamState({ testKey: initialValue }));\n\n      const { result } = renderHook(() =>\n        useTamboV1ComponentState(\"testKey\", initialValue),\n      );\n\n      const newValue = \"updated\";\n      act(() => {\n        result.current[1](newValue);\n      });\n\n      // The debounced function should be called\n      expect(mockUpdateState).toHaveBeenCalledWith(mockComponentId, {\n        threadId: mockThreadId,\n        state: { testKey: newValue },\n      });\n    });\n\n    it(\"should work with complex objects\", () => {\n      const initialValue = { name: \"test\", items: [1, 2, 3] };\n      jest\n        .mocked(useStreamState)\n        .mockReturnValue(createMockStreamState({ data: initialValue }));\n\n      const { result } = renderHook(() =>\n        useTamboV1ComponentState(\"data\", initialValue),\n      );\n\n      const newValue = { name: \"updated\", items: [4, 5, 6] };\n      act(() => {\n        result.current[1](newValue);\n      });\n\n      expect(result.current[0]).toEqual(newValue);\n    });\n  });\n\n  describe(\"Metadata\", () => {\n    it(\"should return isPending and error in meta\", () => {\n      jest.mocked(useStreamState).mockReturnValue(createMockStreamState({}));\n\n      const { result } = renderHook(() =>\n        useTamboV1ComponentState(\"testKey\", \"initial\"),\n      );\n\n      expect(result.current[2]).toEqual({\n        isPending: false,\n        error: null,\n        flush: expect.any(Function),\n      });\n    });\n\n    it(\"should provide a flush function\", () => {\n      const mockFlush = jest.fn();\n      const mockDebouncedFn = createMockDebouncedFunction(jest.fn());\n      mockDebouncedFn.flush = mockFlush;\n      jest.mocked(useDebouncedCallback).mockReturnValue(mockDebouncedFn);\n\n      jest.mocked(useStreamState).mockReturnValue(createMockStreamState({}));\n\n      const { result } = renderHook(() =>\n        useTamboV1ComponentState(\"testKey\", \"initial\"),\n      );\n\n      act(() => {\n        result.current[2].flush();\n      });\n\n      expect(mockFlush).toHaveBeenCalled();\n    });\n  });\n\n  describe(\"Debouncing Behavior\", () => {\n    it(\"should use default debounce time of 500ms\", () => {\n      jest.mocked(useStreamState).mockReturnValue(createMockStreamState({}));\n\n      renderHook(() => useTamboV1ComponentState(\"testKey\", \"initial\"));\n\n      expect(useDebouncedCallback).toHaveBeenCalledWith(\n        expect.any(Function),\n        500,\n      );\n    });\n\n    it(\"should use custom debounce time when provided\", () => {\n      jest.mocked(useStreamState).mockReturnValue(createMockStreamState({}));\n      const customDebounceTime = 1000;\n\n      renderHook(() =>\n        useTamboV1ComponentState(\"testKey\", \"initial\", customDebounceTime),\n      );\n\n      expect(useDebouncedCallback).toHaveBeenCalledWith(\n        expect.any(Function),\n        customDebounceTime,\n      );\n    });\n\n    it(\"should flush debounced callback on unmount\", () => {\n      const mockFlush = jest.fn();\n      const mockDebouncedFn = createMockDebouncedFunction(jest.fn());\n      mockDebouncedFn.flush = mockFlush;\n      jest.mocked(useDebouncedCallback).mockReturnValue(mockDebouncedFn);\n\n      jest.mocked(useStreamState).mockReturnValue(createMockStreamState({}));\n\n      const { unmount } = renderHook(() =>\n        useTamboV1ComponentState(\"testKey\", \"initial\"),\n      );\n\n      unmount();\n\n      expect(mockFlush).toHaveBeenCalled();\n    });\n  });\n\n  describe(\"Server State Sync\", () => {\n    it(\"should sync with server state changes from streaming\", () => {\n      const streamState = createMockStreamState({ testKey: \"initial\" });\n      jest.mocked(useStreamState).mockReturnValue(streamState);\n\n      const { result, rerender } = renderHook(() =>\n        useTamboV1ComponentState(\"testKey\", \"initial\"),\n      );\n\n      expect(result.current[0]).toBe(\"initial\");\n\n      // Simulate server state change from streaming\n      const updatedStreamState = createMockStreamState({\n        testKey: \"updated-from-server\",\n      });\n      jest.mocked(useStreamState).mockReturnValue(updatedStreamState);\n\n      rerender();\n\n      expect(result.current[0]).toBe(\"updated-from-server\");\n    });\n\n    it(\"should handle component not found in stream state\", () => {\n      // Empty stream state (no matching component)\n      jest.mocked(useStreamState).mockReturnValue({\n        threadMap: {},\n        currentThreadId: null,\n      });\n\n      const { result } = renderHook(() =>\n        useTamboV1ComponentState(\"testKey\", \"default\"),\n      );\n\n      expect(result.current[0]).toBe(\"default\");\n    });\n  });\n\n  describe(\"Error Handling\", () => {\n    it(\"should handle API errors gracefully\", async () => {\n      const mockError = new Error(\"API Error\");\n      mockUpdateState.mockRejectedValue(mockError);\n\n      // Create a sync mock that calls the function immediately\n      jest.mocked(useDebouncedCallback).mockImplementation((fn) => {\n        const debouncedFn = Object.assign(\n          async (...args: unknown[]) => fn(...args),\n          {\n            flush: jest.fn(),\n            cancel: jest.fn(),\n            isPending: () => false,\n          },\n        ) as unknown as ReturnType<typeof useDebouncedCallback>;\n        return debouncedFn;\n      });\n\n      jest.mocked(useStreamState).mockReturnValue(createMockStreamState({}));\n\n      const consoleSpy = jest.spyOn(console, \"error\").mockImplementation();\n\n      const { result } = renderHook(() =>\n        useTamboV1ComponentState(\"testKey\", \"initial\"),\n      );\n\n      await act(async () => {\n        result.current[1](\"new-value\");\n        // Wait for async operations\n        await new Promise((resolve) => setTimeout(resolve, 0));\n      });\n\n      expect(consoleSpy).toHaveBeenCalledWith(\n        expect.stringContaining(\"Failed to sync state\"),\n        mockError,\n      );\n\n      consoleSpy.mockRestore();\n    });\n  });\n\n  describe(\"Context Requirements\", () => {\n    it(\"should throw when used outside component content context\", () => {\n      jest.mocked(useV1ComponentContent).mockImplementation(() => {\n        throw new Error(\n          \"useV1ComponentContent must be used within a rendered component\",\n        );\n      });\n\n      expect(() => {\n        renderHook(() => useTamboV1ComponentState(\"testKey\", \"initial\"));\n      }).toThrow(\n        \"useV1ComponentContent must be used within a rendered component\",\n      );\n    });\n  });\n});\n"]}

package/esm/v1/hooks/use-tambo-v1-thread-input.d.ts

@@ -0,0 +1,62 @@
+/**
+ * Options for submitting a message
+ */
+export interface SubmitOptions {
+    /**
+     * Enable debug logging for the stream
+     */
+    debug?: boolean;
+}
+/**
+ * Return type for useTamboV1ThreadInput hook
+ */
+export interface UseTamboV1ThreadInputReturn {
+    /** Current value of the input field */
+    value: string;
+    /** Function to update the input value */
+    setValue: React.Dispatch<React.SetStateAction<string>>;
+    /** Submit the current input. Clears input on success. */
+    submit: (options?: SubmitOptions) => Promise<{
+        threadId: string | undefined;
+    }>;
+    isPending: boolean;
+    isError: boolean;
+    error: Error | null;
+    isSuccess: boolean;
+    reset: () => void;
+}
+/**
+ * Hook to manage thread input state and message submission.
+ *
+ * Provides a similar API to the beta SDK's useTamboThreadInput,
+ * managing input value state and providing a submit function.
+ * @param threadId - Optional thread ID to send messages to. If not provided, creates new thread
+ * @returns Thread input state and submit function
+ * @example
+ * ```tsx
+ * function ChatInput({ threadId }: { threadId?: string }) {
+ *   const { value, setValue, submit, isPending } = useTamboV1ThreadInput(threadId);
+ *
+ *   const handleSubmit = async (e: React.FormEvent) => {
+ *     e.preventDefault();
+ *     const result = await submit();
+ *     console.log('Sent to thread:', result.threadId);
+ *   };
+ *
+ *   return (
+ *     <form onSubmit={handleSubmit}>
+ *       <input
+ *         value={value}
+ *         onChange={(e) => setValue(e.target.value)}
+ *         disabled={isPending}
+ *       />
+ *       <button type="submit" disabled={isPending || !value.trim()}>
+ *         Send
+ *       </button>
+ *     </form>
+ *   );
+ * }
+ * ```
+ */
+export declare function useTamboV1ThreadInput(threadId?: string): UseTamboV1ThreadInputReturn;
+//# sourceMappingURL=use-tambo-v1-thread-input.d.ts.map

package/esm/v1/hooks/use-tambo-v1-thread-input.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"use-tambo-v1-thread-input.d.ts","sourceRoot":"","sources":["../../../src/v1/hooks/use-tambo-v1-thread-input.ts"],"names":[],"mappings":"AAYA;;GAEG;AACH,MAAM,WAAW,aAAa;IAC5B;;OAEG;IACH,KAAK,CAAC,EAAE,OAAO,CAAC;CACjB;AAED;;GAEG;AACH,MAAM,WAAW,2BAA2B;IAC1C,uCAAuC;IACvC,KAAK,EAAE,MAAM,CAAC;IACd,yCAAyC;IACzC,QAAQ,EAAE,KAAK,CAAC,QAAQ,CAAC,KAAK,CAAC,cAAc,CAAC,MAAM,CAAC,CAAC,CAAC;IACvD,yDAAyD;IACzD,MAAM,EAAE,CACN,OAAO,CAAC,EAAE,aAAa,KACpB,OAAO,CAAC;QAAE,QAAQ,EAAE,MAAM,GAAG,SAAS,CAAA;KAAE,CAAC,CAAC;IAC/C,SAAS,EAAE,OAAO,CAAC;IACnB,OAAO,EAAE,OAAO,CAAC;IACjB,KAAK,EAAE,KAAK,GAAG,IAAI,CAAC;IACpB,SAAS,EAAE,OAAO,CAAC;IACnB,KAAK,EAAE,MAAM,IAAI,CAAC;CACnB;AAED;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GAgCG;AACH,wBAAgB,qBAAqB,CACnC,QAAQ,CAAC,EAAE,MAAM,GAChB,2BAA2B,CAqC7B"}

package/esm/v1/hooks/use-tambo-v1-thread-input.js

@@ -0,0 +1,73 @@
+"use client";
+/**
+ * useTamboV1ThreadInput - Thread Input Hook for v1 API
+ *
+ * Manages thread input state and message submission, mirroring
+ * the beta SDK's useTamboThreadInput API.
+ */
+import { useCallback, useState } from "react";
+import { useTamboV1SendMessage } from "./use-tambo-v1-send-message";
+/**
+ * Hook to manage thread input state and message submission.
+ *
+ * Provides a similar API to the beta SDK's useTamboThreadInput,
+ * managing input value state and providing a submit function.
+ * @param threadId - Optional thread ID to send messages to. If not provided, creates new thread
+ * @returns Thread input state and submit function
+ * @example
+ * ```tsx
+ * function ChatInput({ threadId }: { threadId?: string }) {
+ *   const { value, setValue, submit, isPending } = useTamboV1ThreadInput(threadId);
+ *
+ *   const handleSubmit = async (e: React.FormEvent) => {
+ *     e.preventDefault();
+ *     const result = await submit();
+ *     console.log('Sent to thread:', result.threadId);
+ *   };
+ *
+ *   return (
+ *     <form onSubmit={handleSubmit}>
+ *       <input
+ *         value={value}
+ *         onChange={(e) => setValue(e.target.value)}
+ *         disabled={isPending}
+ *       />
+ *       <button type="submit" disabled={isPending || !value.trim()}>
+ *         Send
+ *       </button>
+ *     </form>
+ *   );
+ * }
+ * ```
+ */
+export function useTamboV1ThreadInput(threadId) {
+    const [value, setValue] = useState("");
+    const mutation = useTamboV1SendMessage(threadId);
+    const submit = useCallback(async (options) => {
+        const trimmedValue = value.trim();
+        if (!trimmedValue) {
+            throw new Error("Message cannot be empty");
+        }
+        const result = await mutation.mutateAsync({
+            message: {
+                role: "user",
+                content: [{ type: "text", text: trimmedValue }],
+            },
+            debug: options?.debug,
+        });
+        // Clear input on successful submission
+        setValue("");
+        return result;
+    }, [value, mutation]);
+    return {
+        value,
+        setValue,
+        submit,
+        isPending: mutation.isPending,
+        isError: mutation.isError,
+        error: mutation.error,
+        isSuccess: mutation.isSuccess,
+        reset: mutation.reset,
+    };
+}
+//# sourceMappingURL=use-tambo-v1-thread-input.js.map

package/esm/v1/hooks/use-tambo-v1-thread-input.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"use-tambo-v1-thread-input.js","sourceRoot":"","sources":["../../../src/v1/hooks/use-tambo-v1-thread-input.ts"],"names":[],"mappings":"AAAA,YAAY,CAAC;AAEb;;;;;GAKG;AAEH,OAAO,EAAE,WAAW,EAAE,QAAQ,EAAE,MAAM,OAAO,CAAC;AAC9C,OAAO,EAAE,qBAAqB,EAAE,MAAM,6BAA6B,CAAC;AA+BpE;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GAgCG;AACH,MAAM,UAAU,qBAAqB,CACnC,QAAiB;IAEjB,MAAM,CAAC,KAAK,EAAE,QAAQ,CAAC,GAAG,QAAQ,CAAC,EAAE,CAAC,CAAC;IACvC,MAAM,QAAQ,GAAG,qBAAqB,CAAC,QAAQ,CAAC,CAAC;IAEjD,MAAM,MAAM,GAAG,WAAW,CACxB,KAAK,EAAE,OAAuB,EAAE,EAAE;QAChC,MAAM,YAAY,GAAG,KAAK,CAAC,IAAI,EAAE,CAAC;QAClC,IAAI,CAAC,YAAY,EAAE,CAAC;YAClB,MAAM,IAAI,KAAK,CAAC,yBAAyB,CAAC,CAAC;QAC7C,CAAC;QAED,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,WAAW,CAAC;YACxC,OAAO,EAAE;gBACP,IAAI,EAAE,MAAM;gBACZ,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,YAAY,EAAE,CAAC;aAChD;YACD,KAAK,EAAE,OAAO,EAAE,KAAK;SACtB,CAAC,CAAC;QAEH,uCAAuC;QACvC,QAAQ,CAAC,EAAE,CAAC,CAAC;QAEb,OAAO,MAAM,CAAC;IAChB,CAAC,EACD,CAAC,KAAK,EAAE,QAAQ,CAAC,CAClB,CAAC;IAEF,OAAO;QACL,KAAK;QACL,QAAQ;QACR,MAAM;QACN,SAAS,EAAE,QAAQ,CAAC,SAAS;QAC7B,OAAO,EAAE,QAAQ,CAAC,OAAO;QACzB,KAAK,EAAE,QAAQ,CAAC,KAAK;QACrB,SAAS,EAAE,QAAQ,CAAC,SAAS;QAC7B,KAAK,EAAE,QAAQ,CAAC,KAAK;KACtB,CAAC;AACJ,CAAC","sourcesContent":["\"use client\";\n\n/**\n * useTamboV1ThreadInput - Thread Input Hook for v1 API\n *\n * Manages thread input state and message submission, mirroring\n * the beta SDK's useTamboThreadInput API.\n */\n\nimport { useCallback, useState } from \"react\";\nimport { useTamboV1SendMessage } from \"./use-tambo-v1-send-message\";\n\n/**\n * Options for submitting a message\n */\nexport interface SubmitOptions {\n  /**\n   * Enable debug logging for the stream\n   */\n  debug?: boolean;\n}\n\n/**\n * Return type for useTamboV1ThreadInput hook\n */\nexport interface UseTamboV1ThreadInputReturn {\n  /** Current value of the input field */\n  value: string;\n  /** Function to update the input value */\n  setValue: React.Dispatch<React.SetStateAction<string>>;\n  /** Submit the current input. Clears input on success. */\n  submit: (\n    options?: SubmitOptions,\n  ) => Promise<{ threadId: string | undefined }>;\n  isPending: boolean;\n  isError: boolean;\n  error: Error | null;\n  isSuccess: boolean;\n  reset: () => void;\n}\n\n/**\n * Hook to manage thread input state and message submission.\n *\n * Provides a similar API to the beta SDK's useTamboThreadInput,\n * managing input value state and providing a submit function.\n * @param threadId - Optional thread ID to send messages to. If not provided, creates new thread\n * @returns Thread input state and submit function\n * @example\n * ```tsx\n * function ChatInput({ threadId }: { threadId?: string }) {\n *   const { value, setValue, submit, isPending } = useTamboV1ThreadInput(threadId);\n *\n *   const handleSubmit = async (e: React.FormEvent) => {\n *     e.preventDefault();\n *     const result = await submit();\n *     console.log('Sent to thread:', result.threadId);\n *   };\n *\n *   return (\n *     <form onSubmit={handleSubmit}>\n *       <input\n *         value={value}\n *         onChange={(e) => setValue(e.target.value)}\n *         disabled={isPending}\n *       />\n *       <button type=\"submit\" disabled={isPending || !value.trim()}>\n *         Send\n *       </button>\n *     </form>\n *   );\n * }\n * ```\n */\nexport function useTamboV1ThreadInput(\n  threadId?: string,\n): UseTamboV1ThreadInputReturn {\n  const [value, setValue] = useState(\"\");\n  const mutation = useTamboV1SendMessage(threadId);\n\n  const submit = useCallback(\n    async (options?: SubmitOptions) => {\n      const trimmedValue = value.trim();\n      if (!trimmedValue) {\n        throw new Error(\"Message cannot be empty\");\n      }\n\n      const result = await mutation.mutateAsync({\n        message: {\n          role: \"user\",\n          content: [{ type: \"text\", text: trimmedValue }],\n        },\n        debug: options?.debug,\n      });\n\n      // Clear input on successful submission\n      setValue(\"\");\n\n      return result;\n    },\n    [value, mutation],\n  );\n\n  return {\n    value,\n    setValue,\n    submit,\n    isPending: mutation.isPending,\n    isError: mutation.isError,\n    error: mutation.error,\n    isSuccess: mutation.isSuccess,\n    reset: mutation.reset,\n  };\n}\n"]}

package/esm/v1/hooks/use-tambo-v1-thread-input.test.d.ts

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

package/esm/v1/hooks/use-tambo-v1-thread-input.test.d.ts.map

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