@tambo-ai/react 0.70.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-messages.d.ts

@@ -0,0 +1,58 @@
+import type { TamboV1Message } from "../types/message";
+/**
+ * Return type for useTamboV1Messages hook
+ */
+export interface UseTamboV1MessagesReturn {
+    /**
+     * All messages in the thread
+     */
+    messages: TamboV1Message[];
+    /**
+     * The most recent message (last in the list)
+     */
+    lastMessage: TamboV1Message | undefined;
+    /**
+     * User messages only
+     */
+    userMessages: TamboV1Message[];
+    /**
+     * Assistant messages only
+     */
+    assistantMessages: TamboV1Message[];
+    /**
+     * Whether there are any messages
+     */
+    hasMessages: boolean;
+    /**
+     * Total message count
+     */
+    messageCount: number;
+}
+/**
+ * Hook to access messages in a thread.
+ *
+ * Provides filtered views of messages (user-only, assistant-only)
+ * and metadata about the message list.
+ * @param threadId - Thread ID to get messages for
+ * @returns Message list and utilities
+ * @example
+ * ```tsx
+ * function MessageList({ threadId }: { threadId: string }) {
+ *   const { messages, hasMessages, lastMessage } = useTamboV1Messages(threadId);
+ *
+ *   if (!hasMessages) {
+ *     return <EmptyState />;
+ *   }
+ *
+ *   return (
+ *     <div>
+ *       {messages.map(msg => (
+ *         <Message key={msg.id} message={msg} />
+ *       ))}
+ *     </div>
+ *   );
+ * }
+ * ```
+ */
+export declare function useTamboV1Messages(threadId: string): UseTamboV1MessagesReturn;
+//# sourceMappingURL=use-tambo-v1-messages.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"use-tambo-v1-messages.d.ts","sourceRoot":"","sources":["../../../src/v1/hooks/use-tambo-v1-messages.ts"],"names":[],"mappings":"AAWA,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,kBAAkB,CAAC;AAEvD;;GAEG;AACH,MAAM,WAAW,wBAAwB;IACvC;;OAEG;IACH,QAAQ,EAAE,cAAc,EAAE,CAAC;IAE3B;;OAEG;IACH,WAAW,EAAE,cAAc,GAAG,SAAS,CAAC;IAExC;;OAEG;IACH,YAAY,EAAE,cAAc,EAAE,CAAC;IAE/B;;OAEG;IACH,iBAAiB,EAAE,cAAc,EAAE,CAAC;IAEpC;;OAEG;IACH,WAAW,EAAE,OAAO,CAAC;IAErB;;OAEG;IACH,YAAY,EAAE,MAAM,CAAC;CACtB;AAED;;;;;;;;;;;;;;;;;;;;;;;;;GAyBG;AACH,wBAAgB,kBAAkB,CAAC,QAAQ,EAAE,MAAM,GAAG,wBAAwB,CAgB7E"}

package/esm/v1/hooks/use-tambo-v1-messages.js

@@ -0,0 +1,51 @@
+"use client";
+/**
+ * useTamboV1Messages - Messages Hook for v1 API
+ *
+ * Provides access to messages in a thread with streaming state awareness.
+ * Messages are accumulated from AG-UI events during streaming.
+ */
+import { useMemo } from "react";
+import { useStreamState } from "../providers/tambo-v1-stream-context";
+/**
+ * Hook to access messages in a thread.
+ *
+ * Provides filtered views of messages (user-only, assistant-only)
+ * and metadata about the message list.
+ * @param threadId - Thread ID to get messages for
+ * @returns Message list and utilities
+ * @example
+ * ```tsx
+ * function MessageList({ threadId }: { threadId: string }) {
+ *   const { messages, hasMessages, lastMessage } = useTamboV1Messages(threadId);
+ *
+ *   if (!hasMessages) {
+ *     return <EmptyState />;
+ *   }
+ *
+ *   return (
+ *     <div>
+ *       {messages.map(msg => (
+ *         <Message key={msg.id} message={msg} />
+ *       ))}
+ *     </div>
+ *   );
+ * }
+ * ```
+ */
+export function useTamboV1Messages(threadId) {
+    const streamState = useStreamState();
+    const threadState = streamState.threadMap[threadId];
+    return useMemo(() => {
+        const messages = threadState?.thread.messages ?? [];
+        return {
+            messages,
+            lastMessage: messages[messages.length - 1],
+            userMessages: messages.filter((m) => m.role === "user"),
+            assistantMessages: messages.filter((m) => m.role === "assistant"),
+            hasMessages: messages.length > 0,
+            messageCount: messages.length,
+        };
+    }, [threadState?.thread.messages]);
+}
+//# sourceMappingURL=use-tambo-v1-messages.js.map

package/esm/v1/hooks/use-tambo-v1-messages.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"use-tambo-v1-messages.js","sourceRoot":"","sources":["../../../src/v1/hooks/use-tambo-v1-messages.ts"],"names":[],"mappings":"AAAA,YAAY,CAAC;AAEb;;;;;GAKG;AAEH,OAAO,EAAE,OAAO,EAAE,MAAM,OAAO,CAAC;AAChC,OAAO,EAAE,cAAc,EAAE,MAAM,sCAAsC,CAAC;AAsCtE;;;;;;;;;;;;;;;;;;;;;;;;;GAyBG;AACH,MAAM,UAAU,kBAAkB,CAAC,QAAgB;IACjD,MAAM,WAAW,GAAG,cAAc,EAAE,CAAC;IACrC,MAAM,WAAW,GAAG,WAAW,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC;IAEpD,OAAO,OAAO,CAAC,GAAG,EAAE;QAClB,MAAM,QAAQ,GAAG,WAAW,EAAE,MAAM,CAAC,QAAQ,IAAI,EAAE,CAAC;QAEpD,OAAO;YACL,QAAQ;YACR,WAAW,EAAE,QAAQ,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC,CAAC;YAC1C,YAAY,EAAE,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,MAAM,CAAC;YACvD,iBAAiB,EAAE,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,WAAW,CAAC;YACjE,WAAW,EAAE,QAAQ,CAAC,MAAM,GAAG,CAAC;YAChC,YAAY,EAAE,QAAQ,CAAC,MAAM;SAC9B,CAAC;IACJ,CAAC,EAAE,CAAC,WAAW,EAAE,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC;AACrC,CAAC","sourcesContent":["\"use client\";\n\n/**\n * useTamboV1Messages - Messages Hook for v1 API\n *\n * Provides access to messages in a thread with streaming state awareness.\n * Messages are accumulated from AG-UI events during streaming.\n */\n\nimport { useMemo } from \"react\";\nimport { useStreamState } from \"../providers/tambo-v1-stream-context\";\nimport type { TamboV1Message } from \"../types/message\";\n\n/**\n * Return type for useTamboV1Messages hook\n */\nexport interface UseTamboV1MessagesReturn {\n  /**\n   * All messages in the thread\n   */\n  messages: TamboV1Message[];\n\n  /**\n   * The most recent message (last in the list)\n   */\n  lastMessage: TamboV1Message | undefined;\n\n  /**\n   * User messages only\n   */\n  userMessages: TamboV1Message[];\n\n  /**\n   * Assistant messages only\n   */\n  assistantMessages: TamboV1Message[];\n\n  /**\n   * Whether there are any messages\n   */\n  hasMessages: boolean;\n\n  /**\n   * Total message count\n   */\n  messageCount: number;\n}\n\n/**\n * Hook to access messages in a thread.\n *\n * Provides filtered views of messages (user-only, assistant-only)\n * and metadata about the message list.\n * @param threadId - Thread ID to get messages for\n * @returns Message list and utilities\n * @example\n * ```tsx\n * function MessageList({ threadId }: { threadId: string }) {\n *   const { messages, hasMessages, lastMessage } = useTamboV1Messages(threadId);\n *\n *   if (!hasMessages) {\n *     return <EmptyState />;\n *   }\n *\n *   return (\n *     <div>\n *       {messages.map(msg => (\n *         <Message key={msg.id} message={msg} />\n *       ))}\n *     </div>\n *   );\n * }\n * ```\n */\nexport function useTamboV1Messages(threadId: string): UseTamboV1MessagesReturn {\n  const streamState = useStreamState();\n  const threadState = streamState.threadMap[threadId];\n\n  return useMemo(() => {\n    const messages = threadState?.thread.messages ?? [];\n\n    return {\n      messages,\n      lastMessage: messages[messages.length - 1],\n      userMessages: messages.filter((m) => m.role === \"user\"),\n      assistantMessages: messages.filter((m) => m.role === \"assistant\"),\n      hasMessages: messages.length > 0,\n      messageCount: messages.length,\n    };\n  }, [threadState?.thread.messages]);\n}\n"]}

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

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

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

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

package/esm/v1/hooks/use-tambo-v1-messages.test.js

@@ -0,0 +1,132 @@
+import { EventType, } from "@ag-ui/core";
+import { renderHook, act } from "@testing-library/react";
+import React from "react";
+import { TamboV1StreamProvider, useStreamDispatch, } from "../providers/tambo-v1-stream-context";
+import { useTamboV1Messages } from "./use-tambo-v1-messages";
+describe("useTamboV1Messages", () => {
+    function TestWrapper({ children }) {
+        return (React.createElement(TamboV1StreamProvider, { threadId: "thread_123" }, children));
+    }
+    it("returns empty messages when thread has no messages", () => {
+        const { result } = renderHook(() => useTamboV1Messages("thread_123"), {
+            wrapper: TestWrapper,
+        });
+        expect(result.current.messages).toEqual([]);
+        expect(result.current.hasMessages).toBe(false);
+        expect(result.current.messageCount).toBe(0);
+        expect(result.current.lastMessage).toBeUndefined();
+        expect(result.current.userMessages).toEqual([]);
+        expect(result.current.assistantMessages).toEqual([]);
+    });
+    it("returns messages after events are dispatched", () => {
+        const { result } = renderHook(() => ({
+            messages: useTamboV1Messages("thread_123"),
+            dispatch: useStreamDispatch(),
+        }), { wrapper: TestWrapper });
+        // Simulate a text message being received
+        act(() => {
+            result.current.dispatch({
+                type: "EVENT",
+                event: {
+                    type: EventType.TEXT_MESSAGE_START,
+                    messageId: "msg_1",
+                    role: "assistant",
+                },
+                threadId: "thread_123",
+            });
+        });
+        act(() => {
+            result.current.dispatch({
+                type: "EVENT",
+                event: {
+                    type: EventType.TEXT_MESSAGE_CONTENT,
+                    messageId: "msg_1",
+                    delta: "Hello",
+                },
+                threadId: "thread_123",
+            });
+        });
+        act(() => {
+            result.current.dispatch({
+                type: "EVENT",
+                event: {
+                    type: EventType.TEXT_MESSAGE_END,
+                    messageId: "msg_1",
+                },
+                threadId: "thread_123",
+            });
+        });
+        expect(result.current.messages.messages.length).toBe(1);
+        expect(result.current.messages.hasMessages).toBe(true);
+        expect(result.current.messages.messageCount).toBe(1);
+        expect(result.current.messages.lastMessage?.id).toBe("msg_1");
+        expect(result.current.messages.lastMessage?.role).toBe("assistant");
+        expect(result.current.messages.assistantMessages.length).toBe(1);
+        expect(result.current.messages.userMessages.length).toBe(0);
+    });
+    it("filters user and assistant messages correctly", () => {
+        const { result } = renderHook(() => ({
+            messages: useTamboV1Messages("thread_123"),
+            dispatch: useStreamDispatch(),
+        }), { wrapper: TestWrapper });
+        // Add user message
+        act(() => {
+            result.current.dispatch({
+                type: "EVENT",
+                event: {
+                    type: EventType.TEXT_MESSAGE_START,
+                    messageId: "msg_user",
+                    role: "user",
+                },
+                threadId: "thread_123",
+            });
+        });
+        act(() => {
+            result.current.dispatch({
+                type: "EVENT",
+                event: {
+                    type: EventType.TEXT_MESSAGE_END,
+                    messageId: "msg_user",
+                },
+                threadId: "thread_123",
+            });
+        });
+        // Add assistant message
+        act(() => {
+            result.current.dispatch({
+                type: "EVENT",
+                event: {
+                    type: EventType.TEXT_MESSAGE_START,
+                    messageId: "msg_assistant",
+                    role: "assistant",
+                },
+                threadId: "thread_123",
+            });
+        });
+        act(() => {
+            result.current.dispatch({
+                type: "EVENT",
+                event: {
+                    type: EventType.TEXT_MESSAGE_END,
+                    messageId: "msg_assistant",
+                },
+                threadId: "thread_123",
+            });
+        });
+        expect(result.current.messages.messageCount).toBe(2);
+        expect(result.current.messages.userMessages.length).toBe(1);
+        expect(result.current.messages.userMessages[0].id).toBe("msg_user");
+        expect(result.current.messages.assistantMessages.length).toBe(1);
+        expect(result.current.messages.assistantMessages[0].id).toBe("msg_assistant");
+        expect(result.current.messages.lastMessage?.id).toBe("msg_assistant");
+    });
+    it("returns empty messages when threadId does not exist in threadMap", () => {
+        const { result } = renderHook(() => useTamboV1Messages("nonexistent_thread"), {
+            wrapper: TestWrapper,
+        });
+        expect(result.current.messages).toEqual([]);
+        expect(result.current.hasMessages).toBe(false);
+        expect(result.current.messageCount).toBe(0);
+    });
+});
+//# sourceMappingURL=use-tambo-v1-messages.test.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"use-tambo-v1-messages.test.js","sourceRoot":"","sources":["../../../src/v1/hooks/use-tambo-v1-messages.test.tsx"],"names":[],"mappings":"AAAA,OAAO,EACL,SAAS,GAIV,MAAM,aAAa,CAAC;AACrB,OAAO,EAAE,UAAU,EAAE,GAAG,EAAE,MAAM,wBAAwB,CAAC;AACzD,OAAO,KAAK,MAAM,OAAO,CAAC;AAC1B,OAAO,EACL,qBAAqB,EACrB,iBAAiB,GAClB,MAAM,sCAAsC,CAAC;AAC9C,OAAO,EAAE,kBAAkB,EAAE,MAAM,yBAAyB,CAAC;AAE7D,QAAQ,CAAC,oBAAoB,EAAE,GAAG,EAAE;IAClC,SAAS,WAAW,CAAC,EAAE,QAAQ,EAAiC;QAC9D,OAAO,CACL,oBAAC,qBAAqB,IAAC,QAAQ,EAAC,YAAY,IACzC,QAAQ,CACa,CACzB,CAAC;IACJ,CAAC;IAED,EAAE,CAAC,oDAAoD,EAAE,GAAG,EAAE;QAC5D,MAAM,EAAE,MAAM,EAAE,GAAG,UAAU,CAAC,GAAG,EAAE,CAAC,kBAAkB,CAAC,YAAY,CAAC,EAAE;YACpE,OAAO,EAAE,WAAW;SACrB,CAAC,CAAC;QAEH,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;QAC5C,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAC/C,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAC5C,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC,aAAa,EAAE,CAAC;QACnD,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;QAChD,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,iBAAiB,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;IACvD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8CAA8C,EAAE,GAAG,EAAE;QACtD,MAAM,EAAE,MAAM,EAAE,GAAG,UAAU,CAC3B,GAAG,EAAE,CAAC,CAAC;YACL,QAAQ,EAAE,kBAAkB,CAAC,YAAY,CAAC;YAC1C,QAAQ,EAAE,iBAAiB,EAAE;SAC9B,CAAC,EACF,EAAE,OAAO,EAAE,WAAW,EAAE,CACzB,CAAC;QAEF,yCAAyC;QACzC,GAAG,CAAC,GAAG,EAAE;YACP,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC;gBACtB,IAAI,EAAE,OAAO;gBACb,KAAK,EAAE;oBACL,IAAI,EAAE,SAAS,CAAC,kBAAkB;oBAClC,SAAS,EAAE,OAAO;oBAClB,IAAI,EAAE,WAAW;iBACO;gBAC1B,QAAQ,EAAE,YAAY;aACvB,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,GAAG,CAAC,GAAG,EAAE;YACP,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC;gBACtB,IAAI,EAAE,OAAO;gBACb,KAAK,EAAE;oBACL,IAAI,EAAE,SAAS,CAAC,oBAAoB;oBACpC,SAAS,EAAE,OAAO;oBAClB,KAAK,EAAE,OAAO;iBACY;gBAC5B,QAAQ,EAAE,YAAY;aACvB,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,GAAG,CAAC,GAAG,EAAE;YACP,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC;gBACtB,IAAI,EAAE,OAAO;gBACb,KAAK,EAAE;oBACL,IAAI,EAAE,SAAS,CAAC,gBAAgB;oBAChC,SAAS,EAAE,OAAO;iBACI;gBACxB,QAAQ,EAAE,YAAY;aACvB,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACxD,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACvD,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACrD,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,WAAW,EAAE,EAAE,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAC9D,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,WAAW,EAAE,IAAI,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;QACpE,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,iBAAiB,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACjE,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAC9D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+CAA+C,EAAE,GAAG,EAAE;QACvD,MAAM,EAAE,MAAM,EAAE,GAAG,UAAU,CAC3B,GAAG,EAAE,CAAC,CAAC;YACL,QAAQ,EAAE,kBAAkB,CAAC,YAAY,CAAC;YAC1C,QAAQ,EAAE,iBAAiB,EAAE;SAC9B,CAAC,EACF,EAAE,OAAO,EAAE,WAAW,EAAE,CACzB,CAAC;QAEF,mBAAmB;QACnB,GAAG,CAAC,GAAG,EAAE;YACP,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC;gBACtB,IAAI,EAAE,OAAO;gBACb,KAAK,EAAE;oBACL,IAAI,EAAE,SAAS,CAAC,kBAAkB;oBAClC,SAAS,EAAE,UAAU;oBACrB,IAAI,EAAE,MAAM;iBACY;gBAC1B,QAAQ,EAAE,YAAY;aACvB,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,GAAG,CAAC,GAAG,EAAE;YACP,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC;gBACtB,IAAI,EAAE,OAAO;gBACb,KAAK,EAAE;oBACL,IAAI,EAAE,SAAS,CAAC,gBAAgB;oBAChC,SAAS,EAAE,UAAU;iBACC;gBACxB,QAAQ,EAAE,YAAY;aACvB,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,wBAAwB;QACxB,GAAG,CAAC,GAAG,EAAE;YACP,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC;gBACtB,IAAI,EAAE,OAAO;gBACb,KAAK,EAAE;oBACL,IAAI,EAAE,SAAS,CAAC,kBAAkB;oBAClC,SAAS,EAAE,eAAe;oBAC1B,IAAI,EAAE,WAAW;iBACO;gBAC1B,QAAQ,EAAE,YAAY;aACvB,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,GAAG,CAAC,GAAG,EAAE;YACP,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC;gBACtB,IAAI,EAAE,OAAO;gBACb,KAAK,EAAE;oBACL,IAAI,EAAE,SAAS,CAAC,gBAAgB;oBAChC,SAAS,EAAE,eAAe;iBACJ;gBACxB,QAAQ,EAAE,YAAY;aACvB,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACrD,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAC5D,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QACpE,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,iBAAiB,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACjE,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,iBAAiB,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,IAAI,CAC1D,eAAe,CAChB,CAAC;QACF,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,WAAW,EAAE,EAAE,CAAC,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;IACxE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kEAAkE,EAAE,GAAG,EAAE;QAC1E,MAAM,EAAE,MAAM,EAAE,GAAG,UAAU,CAC3B,GAAG,EAAE,CAAC,kBAAkB,CAAC,oBAAoB,CAAC,EAC9C;YACE,OAAO,EAAE,WAAW;SACrB,CACF,CAAC;QAEF,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;QAC5C,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAC/C,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAC9C,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC","sourcesContent":["import {\n  EventType,\n  type TextMessageStartEvent,\n  type TextMessageContentEvent,\n  type TextMessageEndEvent,\n} from \"@ag-ui/core\";\nimport { renderHook, act } from \"@testing-library/react\";\nimport React from \"react\";\nimport {\n  TamboV1StreamProvider,\n  useStreamDispatch,\n} from \"../providers/tambo-v1-stream-context\";\nimport { useTamboV1Messages } from \"./use-tambo-v1-messages\";\n\ndescribe(\"useTamboV1Messages\", () => {\n  function TestWrapper({ children }: { children: React.ReactNode }) {\n    return (\n      <TamboV1StreamProvider threadId=\"thread_123\">\n        {children}\n      </TamboV1StreamProvider>\n    );\n  }\n\n  it(\"returns empty messages when thread has no messages\", () => {\n    const { result } = renderHook(() => useTamboV1Messages(\"thread_123\"), {\n      wrapper: TestWrapper,\n    });\n\n    expect(result.current.messages).toEqual([]);\n    expect(result.current.hasMessages).toBe(false);\n    expect(result.current.messageCount).toBe(0);\n    expect(result.current.lastMessage).toBeUndefined();\n    expect(result.current.userMessages).toEqual([]);\n    expect(result.current.assistantMessages).toEqual([]);\n  });\n\n  it(\"returns messages after events are dispatched\", () => {\n    const { result } = renderHook(\n      () => ({\n        messages: useTamboV1Messages(\"thread_123\"),\n        dispatch: useStreamDispatch(),\n      }),\n      { wrapper: TestWrapper },\n    );\n\n    // Simulate a text message being received\n    act(() => {\n      result.current.dispatch({\n        type: \"EVENT\",\n        event: {\n          type: EventType.TEXT_MESSAGE_START,\n          messageId: \"msg_1\",\n          role: \"assistant\",\n        } as TextMessageStartEvent,\n        threadId: \"thread_123\",\n      });\n    });\n\n    act(() => {\n      result.current.dispatch({\n        type: \"EVENT\",\n        event: {\n          type: EventType.TEXT_MESSAGE_CONTENT,\n          messageId: \"msg_1\",\n          delta: \"Hello\",\n        } as TextMessageContentEvent,\n        threadId: \"thread_123\",\n      });\n    });\n\n    act(() => {\n      result.current.dispatch({\n        type: \"EVENT\",\n        event: {\n          type: EventType.TEXT_MESSAGE_END,\n          messageId: \"msg_1\",\n        } as TextMessageEndEvent,\n        threadId: \"thread_123\",\n      });\n    });\n\n    expect(result.current.messages.messages.length).toBe(1);\n    expect(result.current.messages.hasMessages).toBe(true);\n    expect(result.current.messages.messageCount).toBe(1);\n    expect(result.current.messages.lastMessage?.id).toBe(\"msg_1\");\n    expect(result.current.messages.lastMessage?.role).toBe(\"assistant\");\n    expect(result.current.messages.assistantMessages.length).toBe(1);\n    expect(result.current.messages.userMessages.length).toBe(0);\n  });\n\n  it(\"filters user and assistant messages correctly\", () => {\n    const { result } = renderHook(\n      () => ({\n        messages: useTamboV1Messages(\"thread_123\"),\n        dispatch: useStreamDispatch(),\n      }),\n      { wrapper: TestWrapper },\n    );\n\n    // Add user message\n    act(() => {\n      result.current.dispatch({\n        type: \"EVENT\",\n        event: {\n          type: EventType.TEXT_MESSAGE_START,\n          messageId: \"msg_user\",\n          role: \"user\",\n        } as TextMessageStartEvent,\n        threadId: \"thread_123\",\n      });\n    });\n\n    act(() => {\n      result.current.dispatch({\n        type: \"EVENT\",\n        event: {\n          type: EventType.TEXT_MESSAGE_END,\n          messageId: \"msg_user\",\n        } as TextMessageEndEvent,\n        threadId: \"thread_123\",\n      });\n    });\n\n    // Add assistant message\n    act(() => {\n      result.current.dispatch({\n        type: \"EVENT\",\n        event: {\n          type: EventType.TEXT_MESSAGE_START,\n          messageId: \"msg_assistant\",\n          role: \"assistant\",\n        } as TextMessageStartEvent,\n        threadId: \"thread_123\",\n      });\n    });\n\n    act(() => {\n      result.current.dispatch({\n        type: \"EVENT\",\n        event: {\n          type: EventType.TEXT_MESSAGE_END,\n          messageId: \"msg_assistant\",\n        } as TextMessageEndEvent,\n        threadId: \"thread_123\",\n      });\n    });\n\n    expect(result.current.messages.messageCount).toBe(2);\n    expect(result.current.messages.userMessages.length).toBe(1);\n    expect(result.current.messages.userMessages[0].id).toBe(\"msg_user\");\n    expect(result.current.messages.assistantMessages.length).toBe(1);\n    expect(result.current.messages.assistantMessages[0].id).toBe(\n      \"msg_assistant\",\n    );\n    expect(result.current.messages.lastMessage?.id).toBe(\"msg_assistant\");\n  });\n\n  it(\"returns empty messages when threadId does not exist in threadMap\", () => {\n    const { result } = renderHook(\n      () => useTamboV1Messages(\"nonexistent_thread\"),\n      {\n        wrapper: TestWrapper,\n      },\n    );\n\n    expect(result.current.messages).toEqual([]);\n    expect(result.current.hasMessages).toBe(false);\n    expect(result.current.messageCount).toBe(0);\n  });\n});\n"]}