@tambo-ai/react 0.71.0 → 0.73.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.test.js

@@ -5,24 +5,33 @@ import { TamboV1StreamProvider, useStreamDispatch, } from "../providers/tambo-v1
 import { useTamboV1Messages } from "./use-tambo-v1-messages";
 describe("useTamboV1Messages", () => {
     function TestWrapper({ children }) {
-        return (React.createElement(TamboV1StreamProvider, { threadId: "thread_123" }, children));
+        return React.createElement(TamboV1StreamProvider, null, children);
     }
     it("returns empty messages when thread has no messages", () => {
-        const { result } = renderHook(() => useTamboV1Messages("thread_123"), {
-            wrapper: TestWrapper,
+        const { result } = renderHook(() => ({
+            messages: useTamboV1Messages("thread_123"),
+            dispatch: useStreamDispatch(),
+        }), { wrapper: TestWrapper });
+        // Initialize thread first
+        act(() => {
+            result.current.dispatch({ type: "INIT_THREAD", threadId: "thread_123" });
         });
-        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([]);
+        expect(result.current.messages.messages).toEqual([]);
+        expect(result.current.messages.hasMessages).toBe(false);
+        expect(result.current.messages.messageCount).toBe(0);
+        expect(result.current.messages.lastMessage).toBeUndefined();
+        expect(result.current.messages.userMessages).toEqual([]);
+        expect(result.current.messages.assistantMessages).toEqual([]);
     });
     it("returns messages after events are dispatched", () => {
         const { result } = renderHook(() => ({
             messages: useTamboV1Messages("thread_123"),
             dispatch: useStreamDispatch(),
         }), { wrapper: TestWrapper });
+        // Initialize thread first
+        act(() => {
+            result.current.dispatch({ type: "INIT_THREAD", threadId: "thread_123" });
+        });
         // Simulate a text message being received
         act(() => {
             result.current.dispatch({
@@ -69,6 +78,10 @@ describe("useTamboV1Messages", () => {
             messages: useTamboV1Messages("thread_123"),
             dispatch: useStreamDispatch(),
         }), { wrapper: TestWrapper });
+        // Initialize thread first
+        act(() => {
+            result.current.dispatch({ type: "INIT_THREAD", threadId: "thread_123" });
+        });
         // Add user message
         act(() => {
             result.current.dispatch({

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

@@ -1 +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"]}
+{"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,oBAAC,qBAAqB,QAAE,QAAQ,CAAyB,CAAC;IACnE,CAAC;IAED,EAAE,CAAC,oDAAoD,EAAE,GAAG,EAAE;QAC5D,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,0BAA0B;QAC1B,GAAG,CAAC,GAAG,EAAE;YACP,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE,IAAI,EAAE,aAAa,EAAE,QAAQ,EAAE,YAAY,EAAE,CAAC,CAAC;QAC3E,CAAC,CAAC,CAAC;QAEH,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;QACrD,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACxD,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,CAAC,CAAC,aAAa,EAAE,CAAC;QAC5D,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;QACzD,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,iBAAiB,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;IAChE,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,0BAA0B;QAC1B,GAAG,CAAC,GAAG,EAAE;YACP,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE,IAAI,EAAE,aAAa,EAAE,QAAQ,EAAE,YAAY,EAAE,CAAC,CAAC;QAC3E,CAAC,CAAC,CAAC;QAEH,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,0BAA0B;QAC1B,GAAG,CAAC,GAAG,EAAE;YACP,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE,IAAI,EAAE,aAAa,EAAE,QAAQ,EAAE,YAAY,EAAE,CAAC,CAAC;QAC3E,CAAC,CAAC,CAAC;QAEH,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 <TamboV1StreamProvider>{children}</TamboV1StreamProvider>;\n  }\n\n  it(\"returns empty messages when thread has no messages\", () => {\n    const { result } = renderHook(\n      () => ({\n        messages: useTamboV1Messages(\"thread_123\"),\n        dispatch: useStreamDispatch(),\n      }),\n      { wrapper: TestWrapper },\n    );\n\n    // Initialize thread first\n    act(() => {\n      result.current.dispatch({ type: \"INIT_THREAD\", threadId: \"thread_123\" });\n    });\n\n    expect(result.current.messages.messages).toEqual([]);\n    expect(result.current.messages.hasMessages).toBe(false);\n    expect(result.current.messages.messageCount).toBe(0);\n    expect(result.current.messages.lastMessage).toBeUndefined();\n    expect(result.current.messages.userMessages).toEqual([]);\n    expect(result.current.messages.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    // Initialize thread first\n    act(() => {\n      result.current.dispatch({ type: \"INIT_THREAD\", threadId: \"thread_123\" });\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    // Initialize thread first\n    act(() => {\n      result.current.dispatch({ type: \"INIT_THREAD\", threadId: \"thread_123\" });\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"]}

package/esm/v1/hooks/use-tambo-v1-send-message.d.ts

@@ -23,6 +23,7 @@ export interface CreateRunStreamParams {
     threadId: string | undefined;
     message: InputMessage;
     registry: TamboRegistry;
+    userKey: string | undefined;
 }
 /**
  * Stream types from the SDK

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

@@ -1 +1 @@
-{"version":3,"file":"use-tambo-v1-send-message.d.ts","sourceRoot":"","sources":["../../../src/v1/hooks/use-tambo-v1-send-message.ts"],"names":[],"mappings":"AAYA,OAAO,KAAK,OAAO,MAAM,0BAA0B,CAAC;AACpD,OAAO,KAAK,EAAE,MAAM,EAAE,MAAM,yCAAyC,CAAC;AAEtE,OAAO,EAEL,KAAK,oBAAoB,IAAI,aAAa,EAC3C,MAAM,yCAAyC,CAAC;AAEjD,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,kBAAkB,CAAC;AASrD;;GAEG;AACH,MAAM,WAAW,kBAAkB;IACjC;;OAEG;IACH,OAAO,EAAE,YAAY,CAAC;IAEtB;;OAEG;IACH,KAAK,CAAC,EAAE,OAAO,CAAC;CACjB;AAED;;GAEG;AACH,MAAM,WAAW,qBAAqB;IACpC,MAAM,EAAE,OAAO,CAAC;IAChB,QAAQ,EAAE,MAAM,GAAG,SAAS,CAAC;IAC7B,OAAO,EAAE,YAAY,CAAC;IACtB,QAAQ,EAAE,aAAa,CAAC;CACzB;AAED;;GAEG;AACH,KAAK,SAAS,GAAG,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;AAC7D,KAAK,YAAY,GAAG,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;AAEnE;;GAEG;AACH,MAAM,WAAW,qBAAqB;IACpC,MAAM,EAAE,SAAS,GAAG,YAAY,CAAC;IACjC,eAAe,EAAE,MAAM,GAAG,SAAS,CAAC;CACrC;AAoDD;;;;;;;GAOG;AACH,wBAAsB,eAAe,CACnC,MAAM,EAAE,qBAAqB,GAC5B,OAAO,CAAC,qBAAqB,CAAC,CAyBhC;AAED;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GA2CG;AACH,wBAAgB,qBAAqB,CAAC,QAAQ,CAAC,EAAE,MAAM;;uCAqHtD"}
+{"version":3,"file":"use-tambo-v1-send-message.d.ts","sourceRoot":"","sources":["../../../src/v1/hooks/use-tambo-v1-send-message.ts"],"names":[],"mappings":"AAYA,OAAO,KAAK,OAAO,MAAM,0BAA0B,CAAC;AACpD,OAAO,KAAK,EAAE,MAAM,EAAE,MAAM,yCAAyC,CAAC;AAEtE,OAAO,EAEL,KAAK,oBAAoB,IAAI,aAAa,EAC3C,MAAM,yCAAyC,CAAC;AAGjD,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,kBAAkB,CAAC;AASrD;;GAEG;AACH,MAAM,WAAW,kBAAkB;IACjC;;OAEG;IACH,OAAO,EAAE,YAAY,CAAC;IAEtB;;OAEG;IACH,KAAK,CAAC,EAAE,OAAO,CAAC;CACjB;AAED;;GAEG;AACH,MAAM,WAAW,qBAAqB;IACpC,MAAM,EAAE,OAAO,CAAC;IAChB,QAAQ,EAAE,MAAM,GAAG,SAAS,CAAC;IAC7B,OAAO,EAAE,YAAY,CAAC;IACtB,QAAQ,EAAE,aAAa,CAAC;IACxB,OAAO,EAAE,MAAM,GAAG,SAAS,CAAC;CAC7B;AAED;;GAEG;AACH,KAAK,SAAS,GAAG,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;AAC7D,KAAK,YAAY,GAAG,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;AAEnE;;GAEG;AACH,MAAM,WAAW,qBAAqB;IACpC,MAAM,EAAE,SAAS,GAAG,YAAY,CAAC;IACjC,eAAe,EAAE,MAAM,GAAG,SAAS,CAAC;CACrC;AAuDD;;;;;;;GAOG;AACH,wBAAsB,eAAe,CACnC,MAAM,EAAE,qBAAqB,GAC5B,OAAO,CAAC,qBAAqB,CAAC,CA2BhC;AAED;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GA2CG;AACH,wBAAgB,qBAAqB,CAAC,QAAQ,CAAC,EAAE,MAAM;;uCAwHtD"}

package/esm/v1/hooks/use-tambo-v1-send-message.js

@@ -11,6 +11,7 @@ import { asTamboCustomEvent } from "../types/event";
 import { useTamboClient } from "../../providers/tambo-client-provider";
 import { TamboRegistryContext, } from "../../providers/tambo-registry-provider";
 import { useStreamDispatch } from "../providers/tambo-v1-stream-context";
+import { useTamboV1Config } from "../providers/tambo-v1-provider";
 import { toAvailableComponents, toAvailableTools, } from "../utils/registry-conversion";
 import { handleEventStream } from "../utils/stream-handler";
 import { executeAllPendingTools } from "../utils/tool-executor";
@@ -25,7 +26,7 @@ import { ToolCallTracker } from "../utils/tool-call-tracker";
  * @returns The continuation stream to process
  */
 async function executeToolsAndContinue(params) {
-    const { event, toolTracker, registry, client, threadId, runId } = params;
+    const { event, toolTracker, registry, client, threadId, runId, userKey } = params;
     const { pendingToolCallIds } = event.value;
     const toolCallsToExecute = toolTracker.getToolCallsById(pendingToolCallIds);
     // Execute tools
@@ -41,6 +42,7 @@ async function executeToolsAndContinue(params) {
         previousRunId: runId,
         availableComponents: toAvailableComponents(registry.componentList),
         tools: toAvailableTools(registry.toolRegistry),
+        userKey,
     });
 }
 /**
@@ -52,7 +54,7 @@ async function executeToolsAndContinue(params) {
  * @returns The stream and initial thread ID (undefined if creating new thread)
  */
 export async function createRunStream(params) {
-    const { client, threadId, message, registry } = params;
+    const { client, threadId, message, registry, userKey } = params;
     // Convert registry components/tools to v1 API format
     const availableComponents = toAvailableComponents(registry.componentList);
     const availableTools = toAvailableTools(registry.toolRegistry);
@@ -62,6 +64,7 @@ export async function createRunStream(params) {
             message,
             availableComponents,
             tools: availableTools,
+            userKey,
         });
         return { stream, initialThreadId: threadId };
     }
@@ -71,6 +74,7 @@ export async function createRunStream(params) {
             message,
             availableComponents,
             tools: availableTools,
+            thread: userKey ? { userKey } : undefined,
         });
         // threadId will be extracted from first event (RUN_STARTED)
         return { stream, initialThreadId: undefined };
@@ -123,6 +127,7 @@ export async function createRunStream(params) {
 export function useTamboV1SendMessage(threadId) {
     const client = useTamboClient();
     const dispatch = useStreamDispatch();
+    const { userKey } = useTamboV1Config();
     const registry = useContext(TamboRegistryContext);
     const queryClient = useQueryClient();
     if (!registry) {
@@ -138,6 +143,7 @@ export function useTamboV1SendMessage(threadId) {
                 threadId,
                 message,
                 registry,
+                userKey,
             });
             let actualThreadId = initialThreadId;
             let runId;
@@ -187,6 +193,7 @@ export function useTamboV1SendMessage(threadId) {
                         client,
                         threadId: actualThreadId,
                         runId,
+                        userKey,
                     });
                 }
                 return { threadId: actualThreadId };

package/esm/v1/hooks/use-tambo-v1-send-message.js.map

@@ -1 +1 @@
-{"version":3,"file":"use-tambo-v1-send-message.js","sourceRoot":"","sources":["../../../src/v1/hooks/use-tambo-v1-send-message.ts"],"names":[],"mappings":"AAAA,YAAY,CAAC;AAEb;;;;GAIG;AAEH,OAAO,EAAE,WAAW,EAAE,cAAc,EAAE,MAAM,uBAAuB,CAAC;AACpE,OAAO,EAAE,UAAU,EAAE,MAAM,OAAO,CAAC;AACnC,OAAO,EAAE,SAAS,EAAsB,MAAM,aAAa,CAAC;AAC5D,OAAO,EAAE,kBAAkB,EAA8B,MAAM,gBAAgB,CAAC;AAGhF,OAAO,EAAE,cAAc,EAAE,MAAM,uCAAuC,CAAC;AACvE,OAAO,EACL,oBAAoB,GAErB,MAAM,yCAAyC,CAAC;AACjD,OAAO,EAAE,iBAAiB,EAAE,MAAM,sCAAsC,CAAC;AAEzE,OAAO,EACL,qBAAqB,EACrB,gBAAgB,GACjB,MAAM,8BAA8B,CAAC;AACtC,OAAO,EAAE,iBAAiB,EAAE,MAAM,yBAAyB,CAAC;AAC5D,OAAO,EAAE,sBAAsB,EAAE,MAAM,wBAAwB,CAAC;AAChE,OAAO,EAAE,eAAe,EAAE,MAAM,4BAA4B,CAAC;AAqD7D;;;;;;;;GAQG;AACH,KAAK,UAAU,uBAAuB,CACpC,MAA0B;IAE1B,MAAM,EAAE,KAAK,EAAE,WAAW,EAAE,QAAQ,EAAE,MAAM,EAAE,QAAQ,EAAE,KAAK,EAAE,GAAG,MAAM,CAAC;IAEzE,MAAM,EAAE,kBAAkB,EAAE,GAAG,KAAK,CAAC,KAAK,CAAC;IAC3C,MAAM,kBAAkB,GAAG,WAAW,CAAC,gBAAgB,CAAC,kBAAkB,CAAC,CAAC;IAE5E,gBAAgB;IAChB,MAAM,WAAW,GAAG,MAAM,sBAAsB,CAC9C,kBAAkB,EAClB,QAAQ,CAAC,YAAY,CACtB,CAAC;IAEF,8CAA8C;IAC9C,WAAW,CAAC,cAAc,CAAC,kBAAkB,CAAC,CAAC;IAE/C,0DAA0D;IAC1D,OAAO,MAAM,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,GAAG,CAAC,QAAQ,EAAE;QAC7C,OAAO,EAAE;YACP,IAAI,EAAE,MAAM;YACZ,OAAO,EAAE,WAAW;SACrB;QACD,aAAa,EAAE,KAAK;QACpB,mBAAmB,EAAE,qBAAqB,CAAC,QAAQ,CAAC,aAAa,CAAC;QAClE,KAAK,EAAE,gBAAgB,CAAC,QAAQ,CAAC,YAAY,CAAC;KAC/C,CAAC,CAAC;AACL,CAAC;AAED;;;;;;;GAOG;AACH,MAAM,CAAC,KAAK,UAAU,eAAe,CACnC,MAA6B;IAE7B,MAAM,EAAE,MAAM,EAAE,QAAQ,EAAE,OAAO,EAAE,QAAQ,EAAE,GAAG,MAAM,CAAC;IAEvD,qDAAqD;IACrD,MAAM,mBAAmB,GAAG,qBAAqB,CAAC,QAAQ,CAAC,aAAa,CAAC,CAAC;IAC1E,MAAM,cAAc,GAAG,gBAAgB,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC;IAE/D,IAAI,QAAQ,EAAE,CAAC;QACb,yBAAyB;QACzB,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,GAAG,CAAC,QAAQ,EAAE;YACrD,OAAO;YACP,mBAAmB;YACnB,KAAK,EAAE,cAAc;SACtB,CAAC,CAAC;QACH,OAAO,EAAE,MAAM,EAAE,eAAe,EAAE,QAAQ,EAAE,CAAC;IAC/C,CAAC;SAAM,CAAC;QACN,oBAAoB;QACpB,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC;YAC9C,OAAO;YACP,mBAAmB;YACnB,KAAK,EAAE,cAAc;SACtB,CAAC,CAAC;QACH,4DAA4D;QAC5D,OAAO,EAAE,MAAM,EAAE,eAAe,EAAE,SAAS,EAAE,CAAC;IAChD,CAAC;AACH,CAAC;AAED;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GA2CG;AACH,MAAM,UAAU,qBAAqB,CAAC,QAAiB;IACrD,MAAM,MAAM,GAAG,cAAc,EAAE,CAAC;IAChC,MAAM,QAAQ,GAAG,iBAAiB,EAAE,CAAC;IACrC,MAAM,QAAQ,GAAG,UAAU,CAAC,oBAAoB,CAAC,CAAC;IAClD,MAAM,WAAW,GAAG,cAAc,EAAE,CAAC;IAErC,IAAI,CAAC,QAAQ,EAAE,CAAC;QACd,MAAM,IAAI,KAAK,CACb,iEAAiE,CAClE,CAAC;IACJ,CAAC;IAED,OAAO,WAAW,CAAC;QACjB,UAAU,EAAE,KAAK,EAAE,OAA2B,EAAE,EAAE;YAChD,MAAM,EAAE,OAAO,EAAE,KAAK,GAAG,KAAK,EAAE,GAAG,OAAO,CAAC;YAE3C,MAAM,WAAW,GAAG,IAAI,eAAe,EAAE,CAAC;YAE1C,wBAAwB;YACxB,MAAM,EAAE,MAAM,EAAE,eAAe,EAAE,GAAG,MAAM,eAAe,CAAC;gBACxD,MAAM;gBACN,QAAQ;gBACR,OAAO;gBACP,QAAQ;aACT,CAAC,CAAC;YAEH,IAAI,cAAc,GAAG,eAAe,CAAC;YACrC,IAAI,KAAyB,CAAC;YAC9B,IAAI,aAAa,GAAoC,MAAM,CAAC;YAE5D,IAAI,CAAC;gBACH,wEAAwE;gBACxE,gFAAgF;gBAChF,mEAAmE;gBACnE,OAAO,IAAI,EAAE,CAAC;oBACZ,IAAI,oBAAuD,CAAC;oBAE5D,4DAA4D;oBAC5D,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,iBAAiB,CAAC,aAAa,EAAE;wBACzD,KAAK;qBACN,CAAC,EAAE,CAAC;wBACH,oDAAoD;wBACpD,IAAI,KAAK,CAAC,IAAI,KAAK,SAAS,CAAC,WAAW,EAAE,CAAC;4BACzC,KAAK,GAAG,KAAK,CAAC,KAAK,CAAC;4BACpB,cAAc,KAAK,KAAK,CAAC,QAAQ,CAAC;wBACpC,CAAC;6BAAM,IAAI,CAAC,cAAc,EAAE,CAAC;4BAC3B,MAAM,IAAI,KAAK,CACb,8DAA8D,KAAK,CAAC,IAAI,EAAE,CAC3E,CAAC;wBACJ,CAAC;wBAED,WAAW,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC;wBAC/B,QAAQ,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,QAAQ,EAAE,cAAc,EAAE,CAAC,CAAC;wBAE7D,8DAA8D;wBAC9D,IAAI,KAAK,CAAC,IAAI,KAAK,SAAS,CAAC,MAAM,EAAE,CAAC;4BACpC,MAAM,WAAW,GAAG,kBAAkB,CAAC,KAAK,CAAC,CAAC;4BAC9C,IAAI,WAAW,EAAE,IAAI,KAAK,0BAA0B,EAAE,CAAC;gCACrD,oBAAoB,GAAG,WAAW,CAAC;gCACnC,MAAM,CAAC,4CAA4C;4BACrD,CAAC;wBACH,CAAC;oBACH,CAAC;oBAED,wDAAwD;oBACxD,IAAI,CAAC,oBAAoB,EAAE,CAAC;wBAC1B,MAAM;oBACR,CAAC;oBAED,4CAA4C;oBAC5C,8EAA8E;oBAC9E,IAAI,CAAC,KAAK,IAAI,CAAC,cAAc,EAAE,CAAC;wBAC9B,MAAM,IAAI,KAAK,CACb,qEAAqE,CACtE,CAAC;oBACJ,CAAC;oBAED,aAAa,GAAG,MAAM,uBAAuB,CAAC;wBAC5C,KAAK,EAAE,oBAAoB;wBAC3B,WAAW;wBACX,QAAQ;wBACR,MAAM;wBACN,QAAQ,EAAE,cAAc;wBACxB,KAAK;qBACN,CAAC,CAAC;gBACL,CAAC;gBAED,OAAO,EAAE,QAAQ,EAAE,cAAc,EAAE,CAAC;YACtC,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,gEAAgE;gBAChE,mEAAmE;gBACnE,IAAI,cAAc,EAAE,CAAC;oBACnB,MAAM,YAAY,GAChB,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,yBAAyB,CAAC;oBACrE,MAAM,UAAU,GAAkB;wBAChC,IAAI,EAAE,SAAS,CAAC,SAAS;wBACzB,OAAO,EAAE,YAAY;qBACtB,CAAC;oBACF,QAAQ,CAAC;wBACP,IAAI,EAAE,OAAO;wBACb,KAAK,EAAE,UAAU;wBACjB,QAAQ,EAAE,cAAc;qBACzB,CAAC,CAAC;gBACL,CAAC;gBACD,MAAM,KAAK,CAAC;YACd,CAAC;QACH,CAAC;QACD,SAAS,EAAE,KAAK,EAAE,MAAM,EAAE,EAAE;YAC1B,qDAAqD;YACrD,MAAM,WAAW,CAAC,iBAAiB,CAAC;gBAClC,QAAQ,EAAE,CAAC,YAAY,EAAE,MAAM,CAAC,QAAQ,CAAC;aAC1C,CAAC,CAAC;QACL,CAAC;QACD,OAAO,EAAE,CAAC,KAAK,EAAE,EAAE;YACjB,OAAO,CAAC,KAAK,CAAC,0CAA0C,EAAE,KAAK,CAAC,CAAC;QACnE,CAAC;KACF,CAAC,CAAC;AACL,CAAC","sourcesContent":["\"use client\";\n\n/**\n * Send Message Hook for v1 API\n *\n * React Query mutation hook for sending messages and handling streaming responses.\n */\n\nimport { useMutation, useQueryClient } from \"@tanstack/react-query\";\nimport { useContext } from \"react\";\nimport { EventType, type RunErrorEvent } from \"@ag-ui/core\";\nimport { asTamboCustomEvent, type RunAwaitingInputEvent } from \"../types/event\";\nimport type TamboAI from \"@tambo-ai/typescript-sdk\";\nimport type { Stream } from \"@tambo-ai/typescript-sdk/core/streaming\";\nimport { useTamboClient } from \"../../providers/tambo-client-provider\";\nimport {\n  TamboRegistryContext,\n  type TamboRegistryContext as TamboRegistry,\n} from \"../../providers/tambo-registry-provider\";\nimport { useStreamDispatch } from \"../providers/tambo-v1-stream-context\";\nimport type { InputMessage } from \"../types/message\";\nimport {\n  toAvailableComponents,\n  toAvailableTools,\n} from \"../utils/registry-conversion\";\nimport { handleEventStream } from \"../utils/stream-handler\";\nimport { executeAllPendingTools } from \"../utils/tool-executor\";\nimport { ToolCallTracker } from \"../utils/tool-call-tracker\";\n\n/**\n * Options for sending a message\n */\nexport interface SendMessageOptions {\n  /**\n   * The message to send\n   */\n  message: InputMessage;\n\n  /**\n   * Enable debug logging for the stream\n   */\n  debug?: boolean;\n}\n\n/**\n * Parameters for creating a run stream\n */\nexport interface CreateRunStreamParams {\n  client: TamboAI;\n  threadId: string | undefined;\n  message: InputMessage;\n  registry: TamboRegistry;\n}\n\n/**\n * Stream types from the SDK\n */\ntype RunStream = Stream<TamboAI.Threads.Runs.RunRunResponse>;\ntype CreateStream = Stream<TamboAI.Threads.Runs.RunCreateResponse>;\n\n/**\n * Result from creating a run stream\n */\nexport interface CreateRunStreamResult {\n  stream: RunStream | CreateStream;\n  initialThreadId: string | undefined;\n}\n\n/**\n * Parameters for executing tools and continuing the run\n */\ninterface ExecuteToolsParams {\n  event: RunAwaitingInputEvent;\n  toolTracker: ToolCallTracker;\n  registry: TamboRegistry;\n  client: TamboAI;\n  threadId: string;\n  runId: string;\n}\n\n/**\n * Executes pending tools and returns a continuation stream.\n *\n * This function does NOT process the continuation stream - it just executes\n * the tools and returns the new stream for the caller to process. This enables\n * the flat loop pattern that correctly handles multi-round tool execution.\n * @param params - The parameters for tool execution\n * @returns The continuation stream to process\n */\nasync function executeToolsAndContinue(\n  params: ExecuteToolsParams,\n): Promise<RunStream> {\n  const { event, toolTracker, registry, client, threadId, runId } = params;\n\n  const { pendingToolCallIds } = event.value;\n  const toolCallsToExecute = toolTracker.getToolCallsById(pendingToolCallIds);\n\n  // Execute tools\n  const toolResults = await executeAllPendingTools(\n    toolCallsToExecute,\n    registry.toolRegistry,\n  );\n\n  // Clear executed tool calls before continuing\n  toolTracker.clearToolCalls(pendingToolCallIds);\n\n  // Return the continuation stream (caller will process it)\n  return await client.threads.runs.run(threadId, {\n    message: {\n      role: \"user\",\n      content: toolResults,\n    },\n    previousRunId: runId,\n    availableComponents: toAvailableComponents(registry.componentList),\n    tools: toAvailableTools(registry.toolRegistry),\n  });\n}\n\n/**\n * Creates a run stream by calling the appropriate API method.\n *\n * If threadId is provided, runs on existing thread via client.threads.runs.run().\n * If no threadId, creates new thread via client.threads.runs.create().\n * @param params - The parameters for creating the run stream\n * @returns The stream and initial thread ID (undefined if creating new thread)\n */\nexport async function createRunStream(\n  params: CreateRunStreamParams,\n): Promise<CreateRunStreamResult> {\n  const { client, threadId, message, registry } = params;\n\n  // Convert registry components/tools to v1 API format\n  const availableComponents = toAvailableComponents(registry.componentList);\n  const availableTools = toAvailableTools(registry.toolRegistry);\n\n  if (threadId) {\n    // Run on existing thread\n    const stream = await client.threads.runs.run(threadId, {\n      message,\n      availableComponents,\n      tools: availableTools,\n    });\n    return { stream, initialThreadId: threadId };\n  } else {\n    // Create new thread\n    const stream = await client.threads.runs.create({\n      message,\n      availableComponents,\n      tools: availableTools,\n    });\n    // threadId will be extracted from first event (RUN_STARTED)\n    return { stream, initialThreadId: undefined };\n  }\n}\n\n/**\n * Hook to send a message and handle streaming responses.\n *\n * This hook handles two scenarios:\n * - If threadId provided: runs on existing thread via client.threads.runs.run()\n * - If no threadId: creates new thread via client.threads.runs.create()\n *\n * The hook:\n * - Sends a user message to the API\n * - Streams AG-UI events in real-time\n * - Dispatches events to the stream reducer\n * - Extracts threadId from events when creating new thread\n * - Handles tool execution (Phase 6)\n * - Invalidates thread queries on completion\n * @param threadId - Optional thread ID to send message to. If not provided, creates new thread\n * @returns React Query mutation object with threadId in mutation result\n * @example\n * ```tsx\n * function ChatInput({ threadId }: { threadId?: string }) {\n *   const sendMessage = useTamboV1SendMessage(threadId);\n *\n *   const handleSubmit = async (text: string) => {\n *     const result = await sendMessage.mutateAsync({\n *       message: {\n *         role: \"user\",\n *         content: [{ type: \"text\", text }],\n *       },\n *     });\n *\n *     // If threadId wasn't provided, a new thread was created\n *     if (!threadId) {\n *       console.log(\"Created thread:\", result.threadId);\n *     }\n *   };\n *\n *   return (\n *     <div>\n *       <input onSubmit={handleSubmit} />\n *       {sendMessage.isPending && <Spinner />}\n *     </div>\n *   );\n * }\n * ```\n */\nexport function useTamboV1SendMessage(threadId?: string) {\n  const client = useTamboClient();\n  const dispatch = useStreamDispatch();\n  const registry = useContext(TamboRegistryContext);\n  const queryClient = useQueryClient();\n\n  if (!registry) {\n    throw new Error(\n      \"useTamboV1SendMessage must be used within TamboRegistryProvider\",\n    );\n  }\n\n  return useMutation({\n    mutationFn: async (options: SendMessageOptions) => {\n      const { message, debug = false } = options;\n\n      const toolTracker = new ToolCallTracker();\n\n      // Create the run stream\n      const { stream, initialThreadId } = await createRunStream({\n        client,\n        threadId,\n        message,\n        registry,\n      });\n\n      let actualThreadId = initialThreadId;\n      let runId: string | undefined;\n      let currentStream: CreateRunStreamResult[\"stream\"] = stream;\n\n      try {\n        // Outer loop handles stream replacement for multi-round tool execution.\n        // When we hit awaiting_input, we execute tools, get a new stream, and continue.\n        // This flat loop pattern correctly handles tool→AI→tool→AI chains.\n        while (true) {\n          let pendingAwaitingInput: RunAwaitingInputEvent | undefined;\n\n          // Process current stream until completion or awaiting_input\n          for await (const event of handleEventStream(currentStream, {\n            debug,\n          })) {\n            // Extract threadId and runId from RUN_STARTED event\n            if (event.type === EventType.RUN_STARTED) {\n              runId = event.runId;\n              actualThreadId ??= event.threadId;\n            } else if (!actualThreadId) {\n              throw new Error(\n                `Expected first event to be RUN_STARTED with threadId, got: ${event.type}`,\n              );\n            }\n\n            toolTracker.handleEvent(event);\n            dispatch({ type: \"EVENT\", event, threadId: actualThreadId });\n\n            // Check for awaiting_input - if found, break to execute tools\n            if (event.type === EventType.CUSTOM) {\n              const customEvent = asTamboCustomEvent(event);\n              if (customEvent?.name === \"tambo.run.awaiting_input\") {\n                pendingAwaitingInput = customEvent;\n                break; // Exit stream loop to handle tool execution\n              }\n            }\n          }\n\n          // If stream finished without awaiting_input, we're done\n          if (!pendingAwaitingInput) {\n            break;\n          }\n\n          // Execute tools and get continuation stream\n          // These checks should never fail since awaiting_input comes after RUN_STARTED\n          if (!runId || !actualThreadId) {\n            throw new Error(\n              \"Cannot continue run after awaiting_input: missing runId or threadId\",\n            );\n          }\n\n          currentStream = await executeToolsAndContinue({\n            event: pendingAwaitingInput,\n            toolTracker,\n            registry,\n            client,\n            threadId: actualThreadId,\n            runId,\n          });\n        }\n\n        return { threadId: actualThreadId };\n      } catch (error) {\n        // Dispatch a synthetic RUN_ERROR event to clean up thread state\n        // This ensures the thread doesn't stay stuck in \"streaming\" status\n        if (actualThreadId) {\n          const errorMessage =\n            error instanceof Error ? error.message : \"Unknown streaming error\";\n          const errorEvent: RunErrorEvent = {\n            type: EventType.RUN_ERROR,\n            message: errorMessage,\n          };\n          dispatch({\n            type: \"EVENT\",\n            event: errorEvent,\n            threadId: actualThreadId,\n          });\n        }\n        throw error;\n      }\n    },\n    onSuccess: async (result) => {\n      // Invalidate thread queries to refetch updated state\n      await queryClient.invalidateQueries({\n        queryKey: [\"v1-threads\", result.threadId],\n      });\n    },\n    onError: (error) => {\n      console.error(\"[useTamboV1SendMessage] Mutation failed:\", error);\n    },\n  });\n}\n"]}
+{"version":3,"file":"use-tambo-v1-send-message.js","sourceRoot":"","sources":["../../../src/v1/hooks/use-tambo-v1-send-message.ts"],"names":[],"mappings":"AAAA,YAAY,CAAC;AAEb;;;;GAIG;AAEH,OAAO,EAAE,WAAW,EAAE,cAAc,EAAE,MAAM,uBAAuB,CAAC;AACpE,OAAO,EAAE,UAAU,EAAE,MAAM,OAAO,CAAC;AACnC,OAAO,EAAE,SAAS,EAAsB,MAAM,aAAa,CAAC;AAC5D,OAAO,EAAE,kBAAkB,EAA8B,MAAM,gBAAgB,CAAC;AAGhF,OAAO,EAAE,cAAc,EAAE,MAAM,uCAAuC,CAAC;AACvE,OAAO,EACL,oBAAoB,GAErB,MAAM,yCAAyC,CAAC;AACjD,OAAO,EAAE,iBAAiB,EAAE,MAAM,sCAAsC,CAAC;AACzE,OAAO,EAAE,gBAAgB,EAAE,MAAM,gCAAgC,CAAC;AAElE,OAAO,EACL,qBAAqB,EACrB,gBAAgB,GACjB,MAAM,8BAA8B,CAAC;AACtC,OAAO,EAAE,iBAAiB,EAAE,MAAM,yBAAyB,CAAC;AAC5D,OAAO,EAAE,sBAAsB,EAAE,MAAM,wBAAwB,CAAC;AAChE,OAAO,EAAE,eAAe,EAAE,MAAM,4BAA4B,CAAC;AAuD7D;;;;;;;;GAQG;AACH,KAAK,UAAU,uBAAuB,CACpC,MAA0B;IAE1B,MAAM,EAAE,KAAK,EAAE,WAAW,EAAE,QAAQ,EAAE,MAAM,EAAE,QAAQ,EAAE,KAAK,EAAE,OAAO,EAAE,GACtE,MAAM,CAAC;IAET,MAAM,EAAE,kBAAkB,EAAE,GAAG,KAAK,CAAC,KAAK,CAAC;IAC3C,MAAM,kBAAkB,GAAG,WAAW,CAAC,gBAAgB,CAAC,kBAAkB,CAAC,CAAC;IAE5E,gBAAgB;IAChB,MAAM,WAAW,GAAG,MAAM,sBAAsB,CAC9C,kBAAkB,EAClB,QAAQ,CAAC,YAAY,CACtB,CAAC;IAEF,8CAA8C;IAC9C,WAAW,CAAC,cAAc,CAAC,kBAAkB,CAAC,CAAC;IAE/C,0DAA0D;IAC1D,OAAO,MAAM,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,GAAG,CAAC,QAAQ,EAAE;QAC7C,OAAO,EAAE;YACP,IAAI,EAAE,MAAM;YACZ,OAAO,EAAE,WAAW;SACrB;QACD,aAAa,EAAE,KAAK;QACpB,mBAAmB,EAAE,qBAAqB,CAAC,QAAQ,CAAC,aAAa,CAAC;QAClE,KAAK,EAAE,gBAAgB,CAAC,QAAQ,CAAC,YAAY,CAAC;QAC9C,OAAO;KACR,CAAC,CAAC;AACL,CAAC;AAED;;;;;;;GAOG;AACH,MAAM,CAAC,KAAK,UAAU,eAAe,CACnC,MAA6B;IAE7B,MAAM,EAAE,MAAM,EAAE,QAAQ,EAAE,OAAO,EAAE,QAAQ,EAAE,OAAO,EAAE,GAAG,MAAM,CAAC;IAEhE,qDAAqD;IACrD,MAAM,mBAAmB,GAAG,qBAAqB,CAAC,QAAQ,CAAC,aAAa,CAAC,CAAC;IAC1E,MAAM,cAAc,GAAG,gBAAgB,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC;IAE/D,IAAI,QAAQ,EAAE,CAAC;QACb,yBAAyB;QACzB,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,GAAG,CAAC,QAAQ,EAAE;YACrD,OAAO;YACP,mBAAmB;YACnB,KAAK,EAAE,cAAc;YACrB,OAAO;SACR,CAAC,CAAC;QACH,OAAO,EAAE,MAAM,EAAE,eAAe,EAAE,QAAQ,EAAE,CAAC;IAC/C,CAAC;SAAM,CAAC;QACN,oBAAoB;QACpB,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC;YAC9C,OAAO;YACP,mBAAmB;YACnB,KAAK,EAAE,cAAc;YACrB,MAAM,EAAE,OAAO,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,CAAC,CAAC,CAAC,SAAS;SAC1C,CAAC,CAAC;QACH,4DAA4D;QAC5D,OAAO,EAAE,MAAM,EAAE,eAAe,EAAE,SAAS,EAAE,CAAC;IAChD,CAAC;AACH,CAAC;AAED;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GA2CG;AACH,MAAM,UAAU,qBAAqB,CAAC,QAAiB;IACrD,MAAM,MAAM,GAAG,cAAc,EAAE,CAAC;IAChC,MAAM,QAAQ,GAAG,iBAAiB,EAAE,CAAC;IACrC,MAAM,EAAE,OAAO,EAAE,GAAG,gBAAgB,EAAE,CAAC;IACvC,MAAM,QAAQ,GAAG,UAAU,CAAC,oBAAoB,CAAC,CAAC;IAClD,MAAM,WAAW,GAAG,cAAc,EAAE,CAAC;IAErC,IAAI,CAAC,QAAQ,EAAE,CAAC;QACd,MAAM,IAAI,KAAK,CACb,iEAAiE,CAClE,CAAC;IACJ,CAAC;IAED,OAAO,WAAW,CAAC;QACjB,UAAU,EAAE,KAAK,EAAE,OAA2B,EAAE,EAAE;YAChD,MAAM,EAAE,OAAO,EAAE,KAAK,GAAG,KAAK,EAAE,GAAG,OAAO,CAAC;YAE3C,MAAM,WAAW,GAAG,IAAI,eAAe,EAAE,CAAC;YAE1C,wBAAwB;YACxB,MAAM,EAAE,MAAM,EAAE,eAAe,EAAE,GAAG,MAAM,eAAe,CAAC;gBACxD,MAAM;gBACN,QAAQ;gBACR,OAAO;gBACP,QAAQ;gBACR,OAAO;aACR,CAAC,CAAC;YAEH,IAAI,cAAc,GAAG,eAAe,CAAC;YACrC,IAAI,KAAyB,CAAC;YAC9B,IAAI,aAAa,GAAoC,MAAM,CAAC;YAE5D,IAAI,CAAC;gBACH,wEAAwE;gBACxE,gFAAgF;gBAChF,mEAAmE;gBACnE,OAAO,IAAI,EAAE,CAAC;oBACZ,IAAI,oBAAuD,CAAC;oBAE5D,4DAA4D;oBAC5D,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,iBAAiB,CAAC,aAAa,EAAE;wBACzD,KAAK;qBACN,CAAC,EAAE,CAAC;wBACH,oDAAoD;wBACpD,IAAI,KAAK,CAAC,IAAI,KAAK,SAAS,CAAC,WAAW,EAAE,CAAC;4BACzC,KAAK,GAAG,KAAK,CAAC,KAAK,CAAC;4BACpB,cAAc,KAAK,KAAK,CAAC,QAAQ,CAAC;wBACpC,CAAC;6BAAM,IAAI,CAAC,cAAc,EAAE,CAAC;4BAC3B,MAAM,IAAI,KAAK,CACb,8DAA8D,KAAK,CAAC,IAAI,EAAE,CAC3E,CAAC;wBACJ,CAAC;wBAED,WAAW,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC;wBAC/B,QAAQ,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,QAAQ,EAAE,cAAc,EAAE,CAAC,CAAC;wBAE7D,8DAA8D;wBAC9D,IAAI,KAAK,CAAC,IAAI,KAAK,SAAS,CAAC,MAAM,EAAE,CAAC;4BACpC,MAAM,WAAW,GAAG,kBAAkB,CAAC,KAAK,CAAC,CAAC;4BAC9C,IAAI,WAAW,EAAE,IAAI,KAAK,0BAA0B,EAAE,CAAC;gCACrD,oBAAoB,GAAG,WAAW,CAAC;gCACnC,MAAM,CAAC,4CAA4C;4BACrD,CAAC;wBACH,CAAC;oBACH,CAAC;oBAED,wDAAwD;oBACxD,IAAI,CAAC,oBAAoB,EAAE,CAAC;wBAC1B,MAAM;oBACR,CAAC;oBAED,4CAA4C;oBAC5C,8EAA8E;oBAC9E,IAAI,CAAC,KAAK,IAAI,CAAC,cAAc,EAAE,CAAC;wBAC9B,MAAM,IAAI,KAAK,CACb,qEAAqE,CACtE,CAAC;oBACJ,CAAC;oBAED,aAAa,GAAG,MAAM,uBAAuB,CAAC;wBAC5C,KAAK,EAAE,oBAAoB;wBAC3B,WAAW;wBACX,QAAQ;wBACR,MAAM;wBACN,QAAQ,EAAE,cAAc;wBACxB,KAAK;wBACL,OAAO;qBACR,CAAC,CAAC;gBACL,CAAC;gBAED,OAAO,EAAE,QAAQ,EAAE,cAAc,EAAE,CAAC;YACtC,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,gEAAgE;gBAChE,mEAAmE;gBACnE,IAAI,cAAc,EAAE,CAAC;oBACnB,MAAM,YAAY,GAChB,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,yBAAyB,CAAC;oBACrE,MAAM,UAAU,GAAkB;wBAChC,IAAI,EAAE,SAAS,CAAC,SAAS;wBACzB,OAAO,EAAE,YAAY;qBACtB,CAAC;oBACF,QAAQ,CAAC;wBACP,IAAI,EAAE,OAAO;wBACb,KAAK,EAAE,UAAU;wBACjB,QAAQ,EAAE,cAAc;qBACzB,CAAC,CAAC;gBACL,CAAC;gBACD,MAAM,KAAK,CAAC;YACd,CAAC;QACH,CAAC;QACD,SAAS,EAAE,KAAK,EAAE,MAAM,EAAE,EAAE;YAC1B,qDAAqD;YACrD,MAAM,WAAW,CAAC,iBAAiB,CAAC;gBAClC,QAAQ,EAAE,CAAC,YAAY,EAAE,MAAM,CAAC,QAAQ,CAAC;aAC1C,CAAC,CAAC;QACL,CAAC;QACD,OAAO,EAAE,CAAC,KAAK,EAAE,EAAE;YACjB,OAAO,CAAC,KAAK,CAAC,0CAA0C,EAAE,KAAK,CAAC,CAAC;QACnE,CAAC;KACF,CAAC,CAAC;AACL,CAAC","sourcesContent":["\"use client\";\n\n/**\n * Send Message Hook for v1 API\n *\n * React Query mutation hook for sending messages and handling streaming responses.\n */\n\nimport { useMutation, useQueryClient } from \"@tanstack/react-query\";\nimport { useContext } from \"react\";\nimport { EventType, type RunErrorEvent } from \"@ag-ui/core\";\nimport { asTamboCustomEvent, type RunAwaitingInputEvent } from \"../types/event\";\nimport type TamboAI from \"@tambo-ai/typescript-sdk\";\nimport type { Stream } from \"@tambo-ai/typescript-sdk/core/streaming\";\nimport { useTamboClient } from \"../../providers/tambo-client-provider\";\nimport {\n  TamboRegistryContext,\n  type TamboRegistryContext as TamboRegistry,\n} from \"../../providers/tambo-registry-provider\";\nimport { useStreamDispatch } from \"../providers/tambo-v1-stream-context\";\nimport { useTamboV1Config } from \"../providers/tambo-v1-provider\";\nimport type { InputMessage } from \"../types/message\";\nimport {\n  toAvailableComponents,\n  toAvailableTools,\n} from \"../utils/registry-conversion\";\nimport { handleEventStream } from \"../utils/stream-handler\";\nimport { executeAllPendingTools } from \"../utils/tool-executor\";\nimport { ToolCallTracker } from \"../utils/tool-call-tracker\";\n\n/**\n * Options for sending a message\n */\nexport interface SendMessageOptions {\n  /**\n   * The message to send\n   */\n  message: InputMessage;\n\n  /**\n   * Enable debug logging for the stream\n   */\n  debug?: boolean;\n}\n\n/**\n * Parameters for creating a run stream\n */\nexport interface CreateRunStreamParams {\n  client: TamboAI;\n  threadId: string | undefined;\n  message: InputMessage;\n  registry: TamboRegistry;\n  userKey: string | undefined;\n}\n\n/**\n * Stream types from the SDK\n */\ntype RunStream = Stream<TamboAI.Threads.Runs.RunRunResponse>;\ntype CreateStream = Stream<TamboAI.Threads.Runs.RunCreateResponse>;\n\n/**\n * Result from creating a run stream\n */\nexport interface CreateRunStreamResult {\n  stream: RunStream | CreateStream;\n  initialThreadId: string | undefined;\n}\n\n/**\n * Parameters for executing tools and continuing the run\n */\ninterface ExecuteToolsParams {\n  event: RunAwaitingInputEvent;\n  toolTracker: ToolCallTracker;\n  registry: TamboRegistry;\n  client: TamboAI;\n  threadId: string;\n  runId: string;\n  userKey: string | undefined;\n}\n\n/**\n * Executes pending tools and returns a continuation stream.\n *\n * This function does NOT process the continuation stream - it just executes\n * the tools and returns the new stream for the caller to process. This enables\n * the flat loop pattern that correctly handles multi-round tool execution.\n * @param params - The parameters for tool execution\n * @returns The continuation stream to process\n */\nasync function executeToolsAndContinue(\n  params: ExecuteToolsParams,\n): Promise<RunStream> {\n  const { event, toolTracker, registry, client, threadId, runId, userKey } =\n    params;\n\n  const { pendingToolCallIds } = event.value;\n  const toolCallsToExecute = toolTracker.getToolCallsById(pendingToolCallIds);\n\n  // Execute tools\n  const toolResults = await executeAllPendingTools(\n    toolCallsToExecute,\n    registry.toolRegistry,\n  );\n\n  // Clear executed tool calls before continuing\n  toolTracker.clearToolCalls(pendingToolCallIds);\n\n  // Return the continuation stream (caller will process it)\n  return await client.threads.runs.run(threadId, {\n    message: {\n      role: \"user\",\n      content: toolResults,\n    },\n    previousRunId: runId,\n    availableComponents: toAvailableComponents(registry.componentList),\n    tools: toAvailableTools(registry.toolRegistry),\n    userKey,\n  });\n}\n\n/**\n * Creates a run stream by calling the appropriate API method.\n *\n * If threadId is provided, runs on existing thread via client.threads.runs.run().\n * If no threadId, creates new thread via client.threads.runs.create().\n * @param params - The parameters for creating the run stream\n * @returns The stream and initial thread ID (undefined if creating new thread)\n */\nexport async function createRunStream(\n  params: CreateRunStreamParams,\n): Promise<CreateRunStreamResult> {\n  const { client, threadId, message, registry, userKey } = params;\n\n  // Convert registry components/tools to v1 API format\n  const availableComponents = toAvailableComponents(registry.componentList);\n  const availableTools = toAvailableTools(registry.toolRegistry);\n\n  if (threadId) {\n    // Run on existing thread\n    const stream = await client.threads.runs.run(threadId, {\n      message,\n      availableComponents,\n      tools: availableTools,\n      userKey,\n    });\n    return { stream, initialThreadId: threadId };\n  } else {\n    // Create new thread\n    const stream = await client.threads.runs.create({\n      message,\n      availableComponents,\n      tools: availableTools,\n      thread: userKey ? { userKey } : undefined,\n    });\n    // threadId will be extracted from first event (RUN_STARTED)\n    return { stream, initialThreadId: undefined };\n  }\n}\n\n/**\n * Hook to send a message and handle streaming responses.\n *\n * This hook handles two scenarios:\n * - If threadId provided: runs on existing thread via client.threads.runs.run()\n * - If no threadId: creates new thread via client.threads.runs.create()\n *\n * The hook:\n * - Sends a user message to the API\n * - Streams AG-UI events in real-time\n * - Dispatches events to the stream reducer\n * - Extracts threadId from events when creating new thread\n * - Handles tool execution (Phase 6)\n * - Invalidates thread queries on completion\n * @param threadId - Optional thread ID to send message to. If not provided, creates new thread\n * @returns React Query mutation object with threadId in mutation result\n * @example\n * ```tsx\n * function ChatInput({ threadId }: { threadId?: string }) {\n *   const sendMessage = useTamboV1SendMessage(threadId);\n *\n *   const handleSubmit = async (text: string) => {\n *     const result = await sendMessage.mutateAsync({\n *       message: {\n *         role: \"user\",\n *         content: [{ type: \"text\", text }],\n *       },\n *     });\n *\n *     // If threadId wasn't provided, a new thread was created\n *     if (!threadId) {\n *       console.log(\"Created thread:\", result.threadId);\n *     }\n *   };\n *\n *   return (\n *     <div>\n *       <input onSubmit={handleSubmit} />\n *       {sendMessage.isPending && <Spinner />}\n *     </div>\n *   );\n * }\n * ```\n */\nexport function useTamboV1SendMessage(threadId?: string) {\n  const client = useTamboClient();\n  const dispatch = useStreamDispatch();\n  const { userKey } = useTamboV1Config();\n  const registry = useContext(TamboRegistryContext);\n  const queryClient = useQueryClient();\n\n  if (!registry) {\n    throw new Error(\n      \"useTamboV1SendMessage must be used within TamboRegistryProvider\",\n    );\n  }\n\n  return useMutation({\n    mutationFn: async (options: SendMessageOptions) => {\n      const { message, debug = false } = options;\n\n      const toolTracker = new ToolCallTracker();\n\n      // Create the run stream\n      const { stream, initialThreadId } = await createRunStream({\n        client,\n        threadId,\n        message,\n        registry,\n        userKey,\n      });\n\n      let actualThreadId = initialThreadId;\n      let runId: string | undefined;\n      let currentStream: CreateRunStreamResult[\"stream\"] = stream;\n\n      try {\n        // Outer loop handles stream replacement for multi-round tool execution.\n        // When we hit awaiting_input, we execute tools, get a new stream, and continue.\n        // This flat loop pattern correctly handles tool→AI→tool→AI chains.\n        while (true) {\n          let pendingAwaitingInput: RunAwaitingInputEvent | undefined;\n\n          // Process current stream until completion or awaiting_input\n          for await (const event of handleEventStream(currentStream, {\n            debug,\n          })) {\n            // Extract threadId and runId from RUN_STARTED event\n            if (event.type === EventType.RUN_STARTED) {\n              runId = event.runId;\n              actualThreadId ??= event.threadId;\n            } else if (!actualThreadId) {\n              throw new Error(\n                `Expected first event to be RUN_STARTED with threadId, got: ${event.type}`,\n              );\n            }\n\n            toolTracker.handleEvent(event);\n            dispatch({ type: \"EVENT\", event, threadId: actualThreadId });\n\n            // Check for awaiting_input - if found, break to execute tools\n            if (event.type === EventType.CUSTOM) {\n              const customEvent = asTamboCustomEvent(event);\n              if (customEvent?.name === \"tambo.run.awaiting_input\") {\n                pendingAwaitingInput = customEvent;\n                break; // Exit stream loop to handle tool execution\n              }\n            }\n          }\n\n          // If stream finished without awaiting_input, we're done\n          if (!pendingAwaitingInput) {\n            break;\n          }\n\n          // Execute tools and get continuation stream\n          // These checks should never fail since awaiting_input comes after RUN_STARTED\n          if (!runId || !actualThreadId) {\n            throw new Error(\n              \"Cannot continue run after awaiting_input: missing runId or threadId\",\n            );\n          }\n\n          currentStream = await executeToolsAndContinue({\n            event: pendingAwaitingInput,\n            toolTracker,\n            registry,\n            client,\n            threadId: actualThreadId,\n            runId,\n            userKey,\n          });\n        }\n\n        return { threadId: actualThreadId };\n      } catch (error) {\n        // Dispatch a synthetic RUN_ERROR event to clean up thread state\n        // This ensures the thread doesn't stay stuck in \"streaming\" status\n        if (actualThreadId) {\n          const errorMessage =\n            error instanceof Error ? error.message : \"Unknown streaming error\";\n          const errorEvent: RunErrorEvent = {\n            type: EventType.RUN_ERROR,\n            message: errorMessage,\n          };\n          dispatch({\n            type: \"EVENT\",\n            event: errorEvent,\n            threadId: actualThreadId,\n          });\n        }\n        throw error;\n      }\n    },\n    onSuccess: async (result) => {\n      // Invalidate thread queries to refetch updated state\n      await queryClient.invalidateQueries({\n        queryKey: [\"v1-threads\", result.threadId],\n      });\n    },\n    onError: (error) => {\n      console.error(\"[useTamboV1SendMessage] Mutation failed:\", error);\n    },\n  });\n}\n"]}