@tambo-ai/react 0.74.1 → 1.0.0-rc.3

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

@@ -18,14 +18,20 @@ jest.mock("../../providers/tambo-client-provider", () => ({
     useTamboQueryClient: jest.fn(),
 }));
 jest.mock("../providers/tambo-v1-provider", () => ({
-    useTamboV1Config: jest.fn(),
+    useTamboConfig: jest.fn(),
+}));
+jest.mock("./use-tambo-v1-auth-state", () => ({
+    useTamboAuthState: () => ({
+        status: "identified",
+        source: "userKey",
+    }),
 }));
 jest.mock("../../providers/tambo-context-helpers-provider", () => ({
     useTamboContextHelpers: () => ({
         getAdditionalContext: jest.fn().mockResolvedValue([]),
     }),
 }));
-describe("useTamboV1SendMessage", () => {
+describe("useTamboSendMessage", () => {
     const mockThreadsRunsApi = {
         run: jest.fn(),
         create: jest.fn(),
@@ -44,7 +50,7 @@ describe("useTamboV1SendMessage", () => {
     function TestWrapper({ children }) {
         return (react_2.default.createElement(react_query_1.QueryClientProvider, { client: queryClient },
             react_2.default.createElement(tambo_registry_provider_1.TamboRegistryContext.Provider, { value: mockRegistry },
-                react_2.default.createElement(tambo_v1_stream_context_1.TamboV1StreamProvider, null, children))));
+                react_2.default.createElement(tambo_v1_stream_context_1.TamboStreamProvider, null, children))));
     }
     beforeEach(() => {
         queryClient = new react_query_1.QueryClient({
@@ -55,12 +61,12 @@ describe("useTamboV1SendMessage", () => {
         });
         jest.mocked(tambo_client_provider_1.useTamboClient).mockReturnValue(mockTamboAI);
         jest.mocked(tambo_client_provider_1.useTamboQueryClient).mockReturnValue(queryClient);
-        jest.mocked(tambo_v1_provider_1.useTamboV1Config).mockReturnValue({ userKey: undefined });
+        jest.mocked(tambo_v1_provider_1.useTamboConfig).mockReturnValue({ userKey: undefined });
         mockThreadsRunsApi.run.mockReset();
         mockThreadsRunsApi.create.mockReset();
     });
     it("returns a mutation object", () => {
-        const { result } = (0, react_1.renderHook)(() => (0, use_tambo_v1_send_message_1.useTamboV1SendMessage)("thread_123"), {
+        const { result } = (0, react_1.renderHook)(() => (0, use_tambo_v1_send_message_1.useTamboSendMessage)("thread_123"), {
             wrapper: TestWrapper,
         });
         expect(result.current.mutate).toBeDefined();
@@ -68,7 +74,7 @@ describe("useTamboV1SendMessage", () => {
         expect(result.current.isPending).toBe(false);
     });
     it("returns a mutation object when threadId is not provided", () => {
-        const { result } = (0, react_1.renderHook)(() => (0, use_tambo_v1_send_message_1.useTamboV1SendMessage)(), {
+        const { result } = (0, react_1.renderHook)(() => (0, use_tambo_v1_send_message_1.useTamboSendMessage)(), {
             wrapper: TestWrapper,
         });
         expect(result.current.mutate).toBeDefined();
@@ -245,7 +251,7 @@ describe("createRunStream", () => {
         expect(callArgs.tools).toEqual([]);
     });
 });
-describe("useTamboV1SendMessage mutation", () => {
+describe("useTamboSendMessage mutation", () => {
     const mockThreadsRunsApi = {
         run: jest.fn(),
         create: jest.fn(),
@@ -275,7 +281,7 @@ describe("useTamboV1SendMessage mutation", () => {
         });
         jest.mocked(tambo_client_provider_1.useTamboClient).mockReturnValue(mockTamboAI);
         jest.mocked(tambo_client_provider_1.useTamboQueryClient).mockReturnValue(queryClient);
-        jest.mocked(tambo_v1_provider_1.useTamboV1Config).mockReturnValue({ userKey: undefined });
+        jest.mocked(tambo_v1_provider_1.useTamboConfig).mockReturnValue({ userKey: undefined });
         mockThreadsRunsApi.run.mockReset();
         mockThreadsRunsApi.create.mockReset();
     });
@@ -296,9 +302,9 @@ describe("useTamboV1SendMessage mutation", () => {
         function TestWrapper({ children }) {
             return (react_2.default.createElement(react_query_1.QueryClientProvider, { client: queryClient },
                 react_2.default.createElement(tambo_registry_provider_1.TamboRegistryContext.Provider, { value: mockRegistry },
-                    react_2.default.createElement(tambo_v1_stream_context_1.TamboV1StreamProvider, null, children))));
+                    react_2.default.createElement(tambo_v1_stream_context_1.TamboStreamProvider, null, children))));
         }
-        const { result } = (0, react_1.renderHook)(() => (0, use_tambo_v1_send_message_1.useTamboV1SendMessage)(), {
+        const { result } = (0, react_1.renderHook)(() => (0, use_tambo_v1_send_message_1.useTamboSendMessage)(), {
             wrapper: TestWrapper,
         });
         let mutationResult;
@@ -324,9 +330,9 @@ describe("useTamboV1SendMessage mutation", () => {
         function TestWrapper({ children }) {
             return (react_2.default.createElement(react_query_1.QueryClientProvider, { client: queryClient },
                 react_2.default.createElement(tambo_registry_provider_1.TamboRegistryContext.Provider, { value: mockRegistry },
-                    react_2.default.createElement(tambo_v1_stream_context_1.TamboV1StreamProvider, null, children))));
+                    react_2.default.createElement(tambo_v1_stream_context_1.TamboStreamProvider, null, children))));
         }
-        const { result } = (0, react_1.renderHook)(() => (0, use_tambo_v1_send_message_1.useTamboV1SendMessage)(), {
+        const { result } = (0, react_1.renderHook)(() => (0, use_tambo_v1_send_message_1.useTamboSendMessage)(), {
             wrapper: TestWrapper,
         });
         await expect((0, react_1.act)(async () => {
@@ -401,9 +407,9 @@ describe("useTamboV1SendMessage mutation", () => {
         function TestWrapper({ children }) {
             return (react_2.default.createElement(react_query_1.QueryClientProvider, { client: queryClient },
                 react_2.default.createElement(tambo_registry_provider_1.TamboRegistryContext.Provider, { value: mockRegistry },
-                    react_2.default.createElement(tambo_v1_stream_context_1.TamboV1StreamProvider, null, children))));
+                    react_2.default.createElement(tambo_v1_stream_context_1.TamboStreamProvider, null, children))));
         }
-        const { result } = (0, react_1.renderHook)(() => (0, use_tambo_v1_send_message_1.useTamboV1SendMessage)("thread_123"), {
+        const { result } = (0, react_1.renderHook)(() => (0, use_tambo_v1_send_message_1.useTamboSendMessage)("thread_123"), {
             wrapper: TestWrapper,
         });
         await (0, react_1.act)(async () => {
@@ -499,9 +505,9 @@ describe("useTamboV1SendMessage mutation", () => {
         function TestWrapper({ children }) {
             return (react_2.default.createElement(react_query_1.QueryClientProvider, { client: queryClient },
                 react_2.default.createElement(tambo_registry_provider_1.TamboRegistryContext.Provider, { value: mockRegistry },
-                    react_2.default.createElement(tambo_v1_stream_context_1.TamboV1StreamProvider, null, children))));
+                    react_2.default.createElement(tambo_v1_stream_context_1.TamboStreamProvider, null, children))));
         }
-        const { result } = (0, react_1.renderHook)(() => (0, use_tambo_v1_send_message_1.useTamboV1SendMessage)("thread_123"), {
+        const { result } = (0, react_1.renderHook)(() => (0, use_tambo_v1_send_message_1.useTamboSendMessage)("thread_123"), {
             wrapper: TestWrapper,
         });
         await (0, react_1.act)(async () => {
@@ -575,9 +581,9 @@ describe("useTamboV1SendMessage mutation", () => {
         function TestWrapper({ children }) {
             return (react_2.default.createElement(react_query_1.QueryClientProvider, { client: queryClient },
                 react_2.default.createElement(tambo_registry_provider_1.TamboRegistryContext.Provider, { value: mockRegistry },
-                    react_2.default.createElement(tambo_v1_stream_context_1.TamboV1StreamProvider, null, children))));
+                    react_2.default.createElement(tambo_v1_stream_context_1.TamboStreamProvider, null, children))));
         }
-        const { result } = (0, react_1.renderHook)(() => (0, use_tambo_v1_send_message_1.useTamboV1SendMessage)("thread_123"), {
+        const { result } = (0, react_1.renderHook)(() => (0, use_tambo_v1_send_message_1.useTamboSendMessage)("thread_123"), {
             wrapper: TestWrapper,
         });
         await (0, react_1.act)(async () => {
@@ -646,9 +652,9 @@ describe("useTamboV1SendMessage mutation", () => {
         function TestWrapper({ children }) {
             return (react_2.default.createElement(react_query_1.QueryClientProvider, { client: queryClient },
                 react_2.default.createElement(tambo_registry_provider_1.TamboRegistryContext.Provider, { value: mockRegistry },
-                    react_2.default.createElement(tambo_v1_stream_context_1.TamboV1StreamProvider, null, children))));
+                    react_2.default.createElement(tambo_v1_stream_context_1.TamboStreamProvider, null, children))));
         }
-        const { result } = (0, react_1.renderHook)(() => (0, use_tambo_v1_send_message_1.useTamboV1SendMessage)("thread_123"), {
+        const { result } = (0, react_1.renderHook)(() => (0, use_tambo_v1_send_message_1.useTamboSendMessage)("thread_123"), {
             wrapper: TestWrapper,
         });
         await (0, react_1.act)(async () => {
@@ -677,10 +683,10 @@ describe("useTamboV1SendMessage mutation", () => {
         // (though with an empty registry)
         function TestWrapperWithoutRegistry({ children, }) {
             return (react_2.default.createElement(react_query_1.QueryClientProvider, { client: queryClient },
-                react_2.default.createElement(tambo_v1_stream_context_1.TamboV1StreamProvider, null, children)));
+                react_2.default.createElement(tambo_v1_stream_context_1.TamboStreamProvider, null, children)));
         }
         // Should not throw - default context is used
-        const { result } = (0, react_1.renderHook)(() => (0, use_tambo_v1_send_message_1.useTamboV1SendMessage)("thread_123"), {
+        const { result } = (0, react_1.renderHook)(() => (0, use_tambo_v1_send_message_1.useTamboSendMessage)("thread_123"), {
             wrapper: TestWrapperWithoutRegistry,
         });
         expect(result.current.mutate).toBeDefined();
@@ -702,10 +708,10 @@ describe("useTamboV1SendMessage mutation", () => {
         function TestWrapper({ children }) {
             return (react_2.default.createElement(react_query_1.QueryClientProvider, { client: queryClient },
                 react_2.default.createElement(tambo_registry_provider_1.TamboRegistryContext.Provider, { value: mockRegistry },
-                    react_2.default.createElement(tambo_v1_stream_context_1.TamboV1StreamProvider, null, children))));
+                    react_2.default.createElement(tambo_v1_stream_context_1.TamboStreamProvider, null, children))));
         }
         const invalidateQueriesSpy = jest.spyOn(queryClient, "invalidateQueries");
-        const { result } = (0, react_1.renderHook)(() => (0, use_tambo_v1_send_message_1.useTamboV1SendMessage)("thread_123"), {
+        const { result } = (0, react_1.renderHook)(() => (0, use_tambo_v1_send_message_1.useTamboSendMessage)("thread_123"), {
             wrapper: TestWrapper,
         });
         await (0, react_1.act)(async () => {
@@ -830,9 +836,9 @@ describe("useTamboV1SendMessage mutation", () => {
         function TestWrapper({ children }) {
             return (react_2.default.createElement(react_query_1.QueryClientProvider, { client: queryClient },
                 react_2.default.createElement(tambo_registry_provider_1.TamboRegistryContext.Provider, { value: mockRegistry },
-                    react_2.default.createElement(tambo_v1_stream_context_1.TamboV1StreamProvider, null, children))));
+                    react_2.default.createElement(tambo_v1_stream_context_1.TamboStreamProvider, null, children))));
         }
-        const { result } = (0, react_1.renderHook)(() => (0, use_tambo_v1_send_message_1.useTamboV1SendMessage)("thread_123"), {
+        const { result } = (0, react_1.renderHook)(() => (0, use_tambo_v1_send_message_1.useTamboSendMessage)("thread_123"), {
             wrapper: TestWrapper,
         });
         await (0, react_1.act)(async () => {
@@ -865,6 +871,67 @@ describe("useTamboV1SendMessage mutation", () => {
             content: [{ type: "text", text: "result from tool 2" }],
         });
     });
+    it("uses lastCompletedRunId as previousRunId when streaming.runId is absent", async () => {
+        const mockStream = createAsyncIterator([
+            {
+                type: core_1.EventType.RUN_STARTED,
+                runId: "run_2",
+                threadId: "thread_123",
+            },
+            { type: core_1.EventType.RUN_FINISHED },
+        ]);
+        mockThreadsRunsApi.run.mockResolvedValue(mockStream);
+        const mockRegistry = {
+            componentList: new Map(),
+            toolRegistry: new Map(),
+        };
+        // Provide stream state with lastCompletedRunId but no active streaming.runId
+        const stateWithLastRun = {
+            currentThreadId: "thread_123",
+            threadMap: {
+                thread_123: {
+                    thread: {
+                        id: "thread_123",
+                        messages: [
+                            {
+                                id: "msg_1",
+                                role: "user",
+                                content: [{ type: "text", text: "Hello" }],
+                                createdAt: "2024-01-01T00:00:00.000Z",
+                            },
+                        ],
+                        status: "idle",
+                        createdAt: "2024-01-01T00:00:00.000Z",
+                        updatedAt: "2024-01-01T00:00:00.000Z",
+                        lastRunCancelled: false,
+                    },
+                    streaming: { status: "idle" },
+                    accumulatingToolArgs: new Map(),
+                    lastCompletedRunId: "run_1",
+                },
+            },
+        };
+        function TestWrapper({ children }) {
+            return (react_2.default.createElement(react_query_1.QueryClientProvider, { client: queryClient },
+                react_2.default.createElement(tambo_registry_provider_1.TamboRegistryContext.Provider, { value: mockRegistry },
+                    react_2.default.createElement(tambo_v1_stream_context_1.TamboStreamProvider, { state: stateWithLastRun, dispatch: jest.fn() }, children))));
+        }
+        const { result } = (0, react_1.renderHook)(() => (0, use_tambo_v1_send_message_1.useTamboSendMessage)("thread_123"), {
+            wrapper: TestWrapper,
+        });
+        await (0, react_1.act)(async () => {
+            await result.current.mutateAsync({
+                message: {
+                    role: "user",
+                    content: [{ type: "text", text: "Follow-up" }],
+                },
+            });
+        });
+        // Should have used lastCompletedRunId as previousRunId
+        expect(mockThreadsRunsApi.run).toHaveBeenCalledWith("thread_123", expect.objectContaining({
+            previousRunId: "run_1",
+        }));
+    });
     it("logs error on mutation failure", async () => {
         const consoleSpy = jest
             .spyOn(console, "error")
@@ -878,9 +945,9 @@ describe("useTamboV1SendMessage mutation", () => {
         function TestWrapper({ children }) {
             return (react_2.default.createElement(react_query_1.QueryClientProvider, { client: queryClient },
                 react_2.default.createElement(tambo_registry_provider_1.TamboRegistryContext.Provider, { value: mockRegistry },
-                    react_2.default.createElement(tambo_v1_stream_context_1.TamboV1StreamProvider, null, children))));
+                    react_2.default.createElement(tambo_v1_stream_context_1.TamboStreamProvider, null, children))));
         }
-        const { result } = (0, react_1.renderHook)(() => (0, use_tambo_v1_send_message_1.useTamboV1SendMessage)("thread_123"), {
+        const { result } = (0, react_1.renderHook)(() => (0, use_tambo_v1_send_message_1.useTamboSendMessage)("thread_123"), {
             wrapper: TestWrapper,
         });
         try {
@@ -897,12 +964,12 @@ describe("useTamboV1SendMessage mutation", () => {
             // Expected to throw
         }
         await (0, react_1.waitFor)(() => {
-            expect(consoleSpy).toHaveBeenCalledWith("[useTamboV1SendMessage] Mutation failed:", testError);
+            expect(consoleSpy).toHaveBeenCalledWith("[useTamboSendMessage] Mutation failed:", testError);
         });
         consoleSpy.mockRestore();
     });
 });
-describe("useTamboV1SendMessage auto thread name generation", () => {
+describe("useTamboSendMessage auto thread name generation", () => {
     const mockGenerateName = jest.fn();
     const mockThreadsRunsApi = {
         run: jest.fn(),
@@ -936,7 +1003,7 @@ describe("useTamboV1SendMessage auto thread name generation", () => {
     function TestWrapper({ children }) {
         return (react_2.default.createElement(react_query_1.QueryClientProvider, { client: queryClient },
             react_2.default.createElement(tambo_registry_provider_1.TamboRegistryContext.Provider, { value: mockRegistry },
-                react_2.default.createElement(tambo_v1_stream_context_1.TamboV1StreamProvider, null, children))));
+                react_2.default.createElement(tambo_v1_stream_context_1.TamboStreamProvider, null, children))));
     }
     function createFinishedStream(threadId) {
         return createAsyncIterator([
@@ -977,14 +1044,14 @@ describe("useTamboV1SendMessage auto thread name generation", () => {
     });
     it("calls generateName when message count reaches threshold", async () => {
         // Threshold=2. Pre-mutation=0, 0 + 2 = 2 >= 2 → fire
-        jest.mocked(tambo_v1_provider_1.useTamboV1Config).mockReturnValue({
+        jest.mocked(tambo_v1_provider_1.useTamboConfig).mockReturnValue({
             userKey: undefined,
             autoGenerateThreadName: true,
             autoGenerateNameThreshold: 2,
         });
         mockGenerateName.mockResolvedValue({ name: "Generated Thread Name" });
         mockThreadsRunsApi.run.mockResolvedValue(createFinishedStream("thread_123"));
-        const { result } = (0, react_1.renderHook)(() => (0, use_tambo_v1_send_message_1.useTamboV1SendMessage)("thread_123"), {
+        const { result } = (0, react_1.renderHook)(() => (0, use_tambo_v1_send_message_1.useTamboSendMessage)("thread_123"), {
             wrapper: TestWrapper,
         });
         await (0, react_1.act)(async () => {
@@ -1000,13 +1067,13 @@ describe("useTamboV1SendMessage auto thread name generation", () => {
         });
     });
     it("does not call generateName when autoGenerateThreadName is false", async () => {
-        jest.mocked(tambo_v1_provider_1.useTamboV1Config).mockReturnValue({
+        jest.mocked(tambo_v1_provider_1.useTamboConfig).mockReturnValue({
             userKey: undefined,
             autoGenerateThreadName: false,
             autoGenerateNameThreshold: 3,
         });
         mockThreadsRunsApi.run.mockResolvedValue(createFinishedStream("thread_123"));
-        const { result } = (0, react_1.renderHook)(() => (0, use_tambo_v1_send_message_1.useTamboV1SendMessage)("thread_123"), {
+        const { result } = (0, react_1.renderHook)(() => (0, use_tambo_v1_send_message_1.useTamboSendMessage)("thread_123"), {
             wrapper: TestWrapper,
         });
         await (0, react_1.act)(async () => {
@@ -1022,13 +1089,13 @@ describe("useTamboV1SendMessage auto thread name generation", () => {
     });
     it("does not call generateName when message count is below threshold", async () => {
         // With threshold=3, preMutation=0 messages, 0 + 2 = 2 < 3 → skip
-        jest.mocked(tambo_v1_provider_1.useTamboV1Config).mockReturnValue({
+        jest.mocked(tambo_v1_provider_1.useTamboConfig).mockReturnValue({
             userKey: undefined,
             autoGenerateThreadName: true,
             autoGenerateNameThreshold: 3,
         });
         mockThreadsRunsApi.create.mockResolvedValue(createFinishedStream("new_thread"));
-        const { result } = (0, react_1.renderHook)(() => (0, use_tambo_v1_send_message_1.useTamboV1SendMessage)(), {
+        const { result } = (0, react_1.renderHook)(() => (0, use_tambo_v1_send_message_1.useTamboSendMessage)(), {
             wrapper: TestWrapper,
         });
         await (0, react_1.act)(async () => {
@@ -1043,7 +1110,7 @@ describe("useTamboV1SendMessage auto thread name generation", () => {
     });
     it("logs error when generateName fails (does not throw)", async () => {
         // Threshold=2. Pre-mutation=0, 0 + 2 = 2 >= 2 → fire
-        jest.mocked(tambo_v1_provider_1.useTamboV1Config).mockReturnValue({
+        jest.mocked(tambo_v1_provider_1.useTamboConfig).mockReturnValue({
             userKey: undefined,
             autoGenerateThreadName: true,
             autoGenerateNameThreshold: 2,
@@ -1053,7 +1120,7 @@ describe("useTamboV1SendMessage auto thread name generation", () => {
             .mockImplementation(() => { });
         mockGenerateName.mockRejectedValue(new Error("API error"));
         mockThreadsRunsApi.run.mockResolvedValue(createFinishedStream("thread_123"));
-        const { result } = (0, react_1.renderHook)(() => (0, use_tambo_v1_send_message_1.useTamboV1SendMessage)("thread_123"), {
+        const { result } = (0, react_1.renderHook)(() => (0, use_tambo_v1_send_message_1.useTamboSendMessage)("thread_123"), {
             wrapper: TestWrapper,
         });
         await (0, react_1.act)(async () => {
@@ -1065,18 +1132,18 @@ describe("useTamboV1SendMessage auto thread name generation", () => {
             });
         });
         await (0, react_1.waitFor)(() => {
-            expect(consoleSpy).toHaveBeenCalledWith("[useTamboV1SendMessage] Failed to auto-generate thread name:", expect.any(Error));
+            expect(consoleSpy).toHaveBeenCalledWith("[useTamboSendMessage] Failed to auto-generate thread name:", expect.any(Error));
         });
         consoleSpy.mockRestore();
     });
     it("uses default config values (enabled, threshold=3)", async () => {
         // Don't set autoGenerateThreadName/autoGenerateNameThreshold — defaults apply
-        jest.mocked(tambo_v1_provider_1.useTamboV1Config).mockReturnValue({
+        jest.mocked(tambo_v1_provider_1.useTamboConfig).mockReturnValue({
             userKey: undefined,
         });
         mockGenerateName.mockResolvedValue({ name: "Auto Name" });
         mockThreadsRunsApi.run.mockResolvedValue(createFinishedStream("thread_123"));
-        const { result } = (0, react_1.renderHook)(() => (0, use_tambo_v1_send_message_1.useTamboV1SendMessage)("thread_123"), {
+        const { result } = (0, react_1.renderHook)(() => (0, use_tambo_v1_send_message_1.useTamboSendMessage)("thread_123"), {
             wrapper: TestWrapper,
         });
         await (0, react_1.act)(async () => {
@@ -1091,4 +1158,210 @@ describe("useTamboV1SendMessage auto thread name generation", () => {
         expect(mockGenerateName).not.toHaveBeenCalled();
     });
 });
+describe("useTamboSendMessage tamboStreamableHint", () => {
+    const mockThreadsRunsApi = {
+        run: jest.fn(),
+        create: jest.fn(),
+    };
+    const mockTamboAI = {
+        apiKey: "",
+        threads: {
+            runs: mockThreadsRunsApi,
+        },
+    };
+    let queryClient;
+    function createAsyncIterator(events) {
+        return {
+            [Symbol.asyncIterator]: async function* () {
+                for (const event of events) {
+                    yield event;
+                }
+            },
+        };
+    }
+    beforeEach(() => {
+        queryClient = new react_query_1.QueryClient({
+            defaultOptions: {
+                queries: { retry: false },
+                mutations: { retry: false },
+            },
+        });
+        jest.mocked(tambo_client_provider_1.useTamboClient).mockReturnValue(mockTamboAI);
+        jest.mocked(tambo_client_provider_1.useTamboQueryClient).mockReturnValue(queryClient);
+        jest.mocked(tambo_v1_provider_1.useTamboConfig).mockReturnValue({ userKey: undefined });
+        mockThreadsRunsApi.run.mockReset();
+        mockThreadsRunsApi.create.mockReset();
+    });
+    it("calls streamable tool on TOOL_CALL_ARGS events during streaming", async () => {
+        const streamableToolFn = jest.fn().mockResolvedValue(undefined);
+        const streamableTool = {
+            name: "write_story",
+            description: "Writes a story",
+            tool: streamableToolFn,
+            inputSchema: zod_1.z.object({ text: zod_1.z.string() }),
+            outputSchema: zod_1.z.void(),
+            annotations: { tamboStreamableHint: true },
+        };
+        // Use Record (not Map) for toolRegistry — matches real runtime type
+        const mockRegistry = {
+            componentList: new Map(),
+            toolRegistry: { write_story: streamableTool },
+        };
+        const initialStream = createAsyncIterator([
+            {
+                type: core_1.EventType.RUN_STARTED,
+                runId: "run_1",
+                threadId: "thread_123",
+            },
+            {
+                type: core_1.EventType.TEXT_MESSAGE_START,
+                messageId: "msg_1",
+                role: "assistant",
+            },
+            {
+                type: core_1.EventType.TOOL_CALL_START,
+                toolCallId: "call_1",
+                toolCallName: "write_story",
+                parentMessageId: "msg_1",
+            },
+            {
+                type: core_1.EventType.TOOL_CALL_ARGS,
+                toolCallId: "call_1",
+                delta: '{"text":"Once upon',
+            },
+            {
+                type: core_1.EventType.TOOL_CALL_ARGS,
+                toolCallId: "call_1",
+                delta: ' a time"}',
+            },
+            {
+                type: core_1.EventType.TOOL_CALL_END,
+                toolCallId: "call_1",
+            },
+            {
+                type: core_1.EventType.CUSTOM,
+                name: "tambo.run.awaiting_input",
+                value: {
+                    pendingToolCalls: [
+                        { toolCallId: "call_1", toolName: "write_story", arguments: "{}" },
+                    ],
+                },
+            },
+        ]);
+        const continueStream = createAsyncIterator([
+            {
+                type: core_1.EventType.RUN_STARTED,
+                runId: "run_2",
+                threadId: "thread_123",
+            },
+            { type: core_1.EventType.RUN_FINISHED },
+        ]);
+        mockThreadsRunsApi.run
+            .mockResolvedValueOnce(initialStream)
+            .mockResolvedValueOnce(continueStream);
+        function TestWrapper({ children }) {
+            return (react_2.default.createElement(react_query_1.QueryClientProvider, { client: queryClient },
+                react_2.default.createElement(tambo_registry_provider_1.TamboRegistryContext.Provider, { value: mockRegistry },
+                    react_2.default.createElement(tambo_v1_stream_context_1.TamboStreamProvider, null, children))));
+        }
+        const { result } = (0, react_1.renderHook)(() => (0, use_tambo_v1_send_message_1.useTamboSendMessage)("thread_123"), {
+            wrapper: TestWrapper,
+        });
+        await (0, react_1.act)(async () => {
+            await result.current.mutateAsync({
+                message: {
+                    role: "user",
+                    content: [{ type: "text", text: "Write me a story" }],
+                },
+            });
+        });
+        // Streamable tool should have been called during TOOL_CALL_ARGS events
+        // (at least once per parseable chunk)
+        expect(streamableToolFn.mock.calls.length).toBeGreaterThanOrEqual(1);
+        // Each streaming call should have had partial args
+        const firstStreamingCall = streamableToolFn.mock.calls[0][0];
+        expect(firstStreamingCall).toHaveProperty("text");
+    });
+    it("does NOT call non-streamable tool during TOOL_CALL_ARGS events", async () => {
+        const normalToolFn = jest.fn().mockResolvedValue("result");
+        const normalTool = {
+            name: "normal_tool",
+            description: "Normal tool (not streamable)",
+            tool: normalToolFn,
+            inputSchema: zod_1.z.object({ query: zod_1.z.string() }),
+            outputSchema: zod_1.z.string(),
+            // No tamboStreamableHint
+        };
+        const mockRegistry = {
+            componentList: new Map(),
+            toolRegistry: { normal_tool: normalTool },
+        };
+        const initialStream = createAsyncIterator([
+            {
+                type: core_1.EventType.RUN_STARTED,
+                runId: "run_1",
+                threadId: "thread_123",
+            },
+            {
+                type: core_1.EventType.TEXT_MESSAGE_START,
+                messageId: "msg_1",
+                role: "assistant",
+            },
+            {
+                type: core_1.EventType.TOOL_CALL_START,
+                toolCallId: "call_1",
+                toolCallName: "normal_tool",
+                parentMessageId: "msg_1",
+            },
+            {
+                type: core_1.EventType.TOOL_CALL_ARGS,
+                toolCallId: "call_1",
+                delta: '{"query":"test"}',
+            },
+            {
+                type: core_1.EventType.TOOL_CALL_END,
+                toolCallId: "call_1",
+            },
+            {
+                type: core_1.EventType.CUSTOM,
+                name: "tambo.run.awaiting_input",
+                value: {
+                    pendingToolCalls: [
+                        { toolCallId: "call_1", toolName: "normal_tool", arguments: "{}" },
+                    ],
+                },
+            },
+        ]);
+        const continueStream = createAsyncIterator([
+            {
+                type: core_1.EventType.RUN_STARTED,
+                runId: "run_2",
+                threadId: "thread_123",
+            },
+            { type: core_1.EventType.RUN_FINISHED },
+        ]);
+        mockThreadsRunsApi.run
+            .mockResolvedValueOnce(initialStream)
+            .mockResolvedValueOnce(continueStream);
+        function TestWrapper({ children }) {
+            return (react_2.default.createElement(react_query_1.QueryClientProvider, { client: queryClient },
+                react_2.default.createElement(tambo_registry_provider_1.TamboRegistryContext.Provider, { value: mockRegistry },
+                    react_2.default.createElement(tambo_v1_stream_context_1.TamboStreamProvider, null, children))));
+        }
+        const { result } = (0, react_1.renderHook)(() => (0, use_tambo_v1_send_message_1.useTamboSendMessage)("thread_123"), {
+            wrapper: TestWrapper,
+        });
+        await (0, react_1.act)(async () => {
+            await result.current.mutateAsync({
+                message: {
+                    role: "user",
+                    content: [{ type: "text", text: "Test" }],
+                },
+            });
+        });
+        // Tool should only be called once at awaiting_input, not during streaming
+        expect(normalToolFn).toHaveBeenCalledTimes(1);
+        expect(normalToolFn).toHaveBeenCalledWith({ query: "test" });
+    });
+});
 //# sourceMappingURL=use-tambo-v1-send-message.test.js.map