@tambo-ai/react 0.72.0 → 0.73.0

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

@@ -10,24 +10,33 @@ const tambo_v1_stream_context_1 = require("../providers/tambo-v1-stream-context"
 const use_tambo_v1_messages_1 = require("./use-tambo-v1-messages");
 describe("useTamboV1Messages", () => {
     function TestWrapper({ children }) {
-        return (react_2.default.createElement(tambo_v1_stream_context_1.TamboV1StreamProvider, { threadId: "thread_123" }, children));
+        return react_2.default.createElement(tambo_v1_stream_context_1.TamboV1StreamProvider, null, children);
     }
     it("returns empty messages when thread has no messages", () => {
-        const { result } = (0, react_1.renderHook)(() => (0, use_tambo_v1_messages_1.useTamboV1Messages)("thread_123"), {
-            wrapper: TestWrapper,
+        const { result } = (0, react_1.renderHook)(() => ({
+            messages: (0, use_tambo_v1_messages_1.useTamboV1Messages)("thread_123"),
+            dispatch: (0, tambo_v1_stream_context_1.useStreamDispatch)(),
+        }), { wrapper: TestWrapper });
+        // Initialize thread first
+        (0, react_1.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 } = (0, react_1.renderHook)(() => ({
             messages: (0, use_tambo_v1_messages_1.useTamboV1Messages)("thread_123"),
             dispatch: (0, tambo_v1_stream_context_1.useStreamDispatch)(),
         }), { wrapper: TestWrapper });
+        // Initialize thread first
+        (0, react_1.act)(() => {
+            result.current.dispatch({ type: "INIT_THREAD", threadId: "thread_123" });
+        });
         // Simulate a text message being received
         (0, react_1.act)(() => {
             result.current.dispatch({
@@ -74,6 +83,10 @@ describe("useTamboV1Messages", () => {
             messages: (0, use_tambo_v1_messages_1.useTamboV1Messages)("thread_123"),
             dispatch: (0, tambo_v1_stream_context_1.useStreamDispatch)(),
         }), { wrapper: TestWrapper });
+        // Initialize thread first
+        (0, react_1.act)(() => {
+            result.current.dispatch({ type: "INIT_THREAD", threadId: "thread_123" });
+        });
         // Add user message
         (0, react_1.act)(() => {
             result.current.dispatch({

package/dist/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,sCAKqB;AACrB,kDAAyD;AACzD,kDAA0B;AAC1B,kFAG8C;AAC9C,mEAA6D;AAE7D,QAAQ,CAAC,oBAAoB,EAAE,GAAG,EAAE;IAClC,SAAS,WAAW,CAAC,EAAE,QAAQ,EAAiC;QAC9D,OAAO,CACL,8BAAC,+CAAqB,IAAC,QAAQ,EAAC,YAAY,IACzC,QAAQ,CACa,CACzB,CAAC;IACJ,CAAC;IAED,EAAE,CAAC,oDAAoD,EAAE,GAAG,EAAE;QAC5D,MAAM,EAAE,MAAM,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CAAC,IAAA,0CAAkB,EAAC,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,IAAA,kBAAU,EAC3B,GAAG,EAAE,CAAC,CAAC;YACL,QAAQ,EAAE,IAAA,0CAAkB,EAAC,YAAY,CAAC;YAC1C,QAAQ,EAAE,IAAA,2CAAiB,GAAE;SAC9B,CAAC,EACF,EAAE,OAAO,EAAE,WAAW,EAAE,CACzB,CAAC;QAEF,yCAAyC;QACzC,IAAA,WAAG,EAAC,GAAG,EAAE;YACP,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC;gBACtB,IAAI,EAAE,OAAO;gBACb,KAAK,EAAE;oBACL,IAAI,EAAE,gBAAS,CAAC,kBAAkB;oBAClC,SAAS,EAAE,OAAO;oBAClB,IAAI,EAAE,WAAW;iBACO;gBAC1B,QAAQ,EAAE,YAAY;aACvB,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,IAAA,WAAG,EAAC,GAAG,EAAE;YACP,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC;gBACtB,IAAI,EAAE,OAAO;gBACb,KAAK,EAAE;oBACL,IAAI,EAAE,gBAAS,CAAC,oBAAoB;oBACpC,SAAS,EAAE,OAAO;oBAClB,KAAK,EAAE,OAAO;iBACY;gBAC5B,QAAQ,EAAE,YAAY;aACvB,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,IAAA,WAAG,EAAC,GAAG,EAAE;YACP,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC;gBACtB,IAAI,EAAE,OAAO;gBACb,KAAK,EAAE;oBACL,IAAI,EAAE,gBAAS,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,IAAA,kBAAU,EAC3B,GAAG,EAAE,CAAC,CAAC;YACL,QAAQ,EAAE,IAAA,0CAAkB,EAAC,YAAY,CAAC;YAC1C,QAAQ,EAAE,IAAA,2CAAiB,GAAE;SAC9B,CAAC,EACF,EAAE,OAAO,EAAE,WAAW,EAAE,CACzB,CAAC;QAEF,mBAAmB;QACnB,IAAA,WAAG,EAAC,GAAG,EAAE;YACP,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC;gBACtB,IAAI,EAAE,OAAO;gBACb,KAAK,EAAE;oBACL,IAAI,EAAE,gBAAS,CAAC,kBAAkB;oBAClC,SAAS,EAAE,UAAU;oBACrB,IAAI,EAAE,MAAM;iBACY;gBAC1B,QAAQ,EAAE,YAAY;aACvB,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,IAAA,WAAG,EAAC,GAAG,EAAE;YACP,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC;gBACtB,IAAI,EAAE,OAAO;gBACb,KAAK,EAAE;oBACL,IAAI,EAAE,gBAAS,CAAC,gBAAgB;oBAChC,SAAS,EAAE,UAAU;iBACC;gBACxB,QAAQ,EAAE,YAAY;aACvB,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,wBAAwB;QACxB,IAAA,WAAG,EAAC,GAAG,EAAE;YACP,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC;gBACtB,IAAI,EAAE,OAAO;gBACb,KAAK,EAAE;oBACL,IAAI,EAAE,gBAAS,CAAC,kBAAkB;oBAClC,SAAS,EAAE,eAAe;oBAC1B,IAAI,EAAE,WAAW;iBACO;gBAC1B,QAAQ,EAAE,YAAY;aACvB,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,IAAA,WAAG,EAAC,GAAG,EAAE;YACP,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC;gBACtB,IAAI,EAAE,OAAO;gBACb,KAAK,EAAE;oBACL,IAAI,EAAE,gBAAS,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,IAAA,kBAAU,EAC3B,GAAG,EAAE,CAAC,IAAA,0CAAkB,EAAC,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,sCAKqB;AACrB,kDAAyD;AACzD,kDAA0B;AAC1B,kFAG8C;AAC9C,mEAA6D;AAE7D,QAAQ,CAAC,oBAAoB,EAAE,GAAG,EAAE;IAClC,SAAS,WAAW,CAAC,EAAE,QAAQ,EAAiC;QAC9D,OAAO,8BAAC,+CAAqB,QAAE,QAAQ,CAAyB,CAAC;IACnE,CAAC;IAED,EAAE,CAAC,oDAAoD,EAAE,GAAG,EAAE;QAC5D,MAAM,EAAE,MAAM,EAAE,GAAG,IAAA,kBAAU,EAC3B,GAAG,EAAE,CAAC,CAAC;YACL,QAAQ,EAAE,IAAA,0CAAkB,EAAC,YAAY,CAAC;YAC1C,QAAQ,EAAE,IAAA,2CAAiB,GAAE;SAC9B,CAAC,EACF,EAAE,OAAO,EAAE,WAAW,EAAE,CACzB,CAAC;QAEF,0BAA0B;QAC1B,IAAA,WAAG,EAAC,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,IAAA,kBAAU,EAC3B,GAAG,EAAE,CAAC,CAAC;YACL,QAAQ,EAAE,IAAA,0CAAkB,EAAC,YAAY,CAAC;YAC1C,QAAQ,EAAE,IAAA,2CAAiB,GAAE;SAC9B,CAAC,EACF,EAAE,OAAO,EAAE,WAAW,EAAE,CACzB,CAAC;QAEF,0BAA0B;QAC1B,IAAA,WAAG,EAAC,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,IAAA,WAAG,EAAC,GAAG,EAAE;YACP,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC;gBACtB,IAAI,EAAE,OAAO;gBACb,KAAK,EAAE;oBACL,IAAI,EAAE,gBAAS,CAAC,kBAAkB;oBAClC,SAAS,EAAE,OAAO;oBAClB,IAAI,EAAE,WAAW;iBACO;gBAC1B,QAAQ,EAAE,YAAY;aACvB,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,IAAA,WAAG,EAAC,GAAG,EAAE;YACP,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC;gBACtB,IAAI,EAAE,OAAO;gBACb,KAAK,EAAE;oBACL,IAAI,EAAE,gBAAS,CAAC,oBAAoB;oBACpC,SAAS,EAAE,OAAO;oBAClB,KAAK,EAAE,OAAO;iBACY;gBAC5B,QAAQ,EAAE,YAAY;aACvB,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,IAAA,WAAG,EAAC,GAAG,EAAE;YACP,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC;gBACtB,IAAI,EAAE,OAAO;gBACb,KAAK,EAAE;oBACL,IAAI,EAAE,gBAAS,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,IAAA,kBAAU,EAC3B,GAAG,EAAE,CAAC,CAAC;YACL,QAAQ,EAAE,IAAA,0CAAkB,EAAC,YAAY,CAAC;YAC1C,QAAQ,EAAE,IAAA,2CAAiB,GAAE;SAC9B,CAAC,EACF,EAAE,OAAO,EAAE,WAAW,EAAE,CACzB,CAAC;QAEF,0BAA0B;QAC1B,IAAA,WAAG,EAAC,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,IAAA,WAAG,EAAC,GAAG,EAAE;YACP,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC;gBACtB,IAAI,EAAE,OAAO;gBACb,KAAK,EAAE;oBACL,IAAI,EAAE,gBAAS,CAAC,kBAAkB;oBAClC,SAAS,EAAE,UAAU;oBACrB,IAAI,EAAE,MAAM;iBACY;gBAC1B,QAAQ,EAAE,YAAY;aACvB,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,IAAA,WAAG,EAAC,GAAG,EAAE;YACP,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC;gBACtB,IAAI,EAAE,OAAO;gBACb,KAAK,EAAE;oBACL,IAAI,EAAE,gBAAS,CAAC,gBAAgB;oBAChC,SAAS,EAAE,UAAU;iBACC;gBACxB,QAAQ,EAAE,YAAY;aACvB,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,wBAAwB;QACxB,IAAA,WAAG,EAAC,GAAG,EAAE;YACP,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC;gBACtB,IAAI,EAAE,OAAO;gBACb,KAAK,EAAE;oBACL,IAAI,EAAE,gBAAS,CAAC,kBAAkB;oBAClC,SAAS,EAAE,eAAe;oBAC1B,IAAI,EAAE,WAAW;iBACO;gBAC1B,QAAQ,EAAE,YAAY;aACvB,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,IAAA,WAAG,EAAC,GAAG,EAAE;YACP,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC;gBACtB,IAAI,EAAE,OAAO;gBACb,KAAK,EAAE;oBACL,IAAI,EAAE,gBAAS,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,IAAA,kBAAU,EAC3B,GAAG,EAAE,CAAC,IAAA,0CAAkB,EAAC,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/dist/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/dist/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/dist/v1/hooks/use-tambo-v1-send-message.js

@@ -15,6 +15,7 @@ const event_1 = require("../types/event");
 const tambo_client_provider_1 = require("../../providers/tambo-client-provider");
 const tambo_registry_provider_1 = require("../../providers/tambo-registry-provider");
 const tambo_v1_stream_context_1 = require("../providers/tambo-v1-stream-context");
+const tambo_v1_provider_1 = require("../providers/tambo-v1-provider");
 const registry_conversion_1 = require("../utils/registry-conversion");
 const stream_handler_1 = require("../utils/stream-handler");
 const tool_executor_1 = require("../utils/tool-executor");
@@ -29,7 +30,7 @@ const tool_call_tracker_1 = require("../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
@@ -45,6 +46,7 @@ async function executeToolsAndContinue(params) {
         previousRunId: runId,
         availableComponents: (0, registry_conversion_1.toAvailableComponents)(registry.componentList),
         tools: (0, registry_conversion_1.toAvailableTools)(registry.toolRegistry),
+        userKey,
     });
 }
 /**
@@ -56,7 +58,7 @@ async function executeToolsAndContinue(params) {
  * @returns The stream and initial thread ID (undefined if creating new thread)
  */
 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 = (0, registry_conversion_1.toAvailableComponents)(registry.componentList);
     const availableTools = (0, registry_conversion_1.toAvailableTools)(registry.toolRegistry);
@@ -66,6 +68,7 @@ async function createRunStream(params) {
             message,
             availableComponents,
             tools: availableTools,
+            userKey,
         });
         return { stream, initialThreadId: threadId };
     }
@@ -75,6 +78,7 @@ 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 };
@@ -127,6 +131,7 @@ async function createRunStream(params) {
 function useTamboV1SendMessage(threadId) {
     const client = (0, tambo_client_provider_1.useTamboClient)();
     const dispatch = (0, tambo_v1_stream_context_1.useStreamDispatch)();
+    const { userKey } = (0, tambo_v1_provider_1.useTamboV1Config)();
     const registry = (0, react_1.useContext)(tambo_registry_provider_1.TamboRegistryContext);
     const queryClient = (0, react_query_1.useQueryClient)();
     if (!registry) {
@@ -142,6 +147,7 @@ function useTamboV1SendMessage(threadId) {
                 threadId,
                 message,
                 registry,
+                userKey,
             });
             let actualThreadId = initialThreadId;
             let runId;
@@ -191,6 +197,7 @@ function useTamboV1SendMessage(threadId) {
                         client,
                         threadId: actualThreadId,
                         runId,
+                        userKey,
                     });
                 }
                 return { threadId: actualThreadId };

package/dist/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;;AA8Hb,0CA2BC;AA8CD,sDAqHC;AA1TD;;;;GAIG;AAEH,uDAAoE;AACpE,iCAAmC;AACnC,sCAA4D;AAC5D,0CAAgF;AAGhF,iFAAuE;AACvE,qFAGiD;AACjD,kFAAyE;AAEzE,sEAGsC;AACtC,4DAA4D;AAC5D,0DAAgE;AAChE,kEAA6D;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,IAAA,sCAAsB,EAC9C,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,IAAA,2CAAqB,EAAC,QAAQ,CAAC,aAAa,CAAC;QAClE,KAAK,EAAE,IAAA,sCAAgB,EAAC,QAAQ,CAAC,YAAY,CAAC;KAC/C,CAAC,CAAC;AACL,CAAC;AAED;;;;;;;GAOG;AACI,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,IAAA,2CAAqB,EAAC,QAAQ,CAAC,aAAa,CAAC,CAAC;IAC1E,MAAM,cAAc,GAAG,IAAA,sCAAgB,EAAC,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,SAAgB,qBAAqB,CAAC,QAAiB;IACrD,MAAM,MAAM,GAAG,IAAA,sCAAc,GAAE,CAAC;IAChC,MAAM,QAAQ,GAAG,IAAA,2CAAiB,GAAE,CAAC;IACrC,MAAM,QAAQ,GAAG,IAAA,kBAAU,EAAC,8CAAoB,CAAC,CAAC;IAClD,MAAM,WAAW,GAAG,IAAA,4BAAc,GAAE,CAAC;IAErC,IAAI,CAAC,QAAQ,EAAE,CAAC;QACd,MAAM,IAAI,KAAK,CACb,iEAAiE,CAClE,CAAC;IACJ,CAAC;IAED,OAAO,IAAA,yBAAW,EAAC;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,mCAAe,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,IAAA,kCAAiB,EAAC,aAAa,EAAE;wBACzD,KAAK;qBACN,CAAC,EAAE,CAAC;wBACH,oDAAoD;wBACpD,IAAI,KAAK,CAAC,IAAI,KAAK,gBAAS,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,gBAAS,CAAC,MAAM,EAAE,CAAC;4BACpC,MAAM,WAAW,GAAG,IAAA,0BAAkB,EAAC,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,gBAAS,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;;AAmIb,0CA6BC;AA8CD,sDAwHC;AApUD;;;;GAIG;AAEH,uDAAoE;AACpE,iCAAmC;AACnC,sCAA4D;AAC5D,0CAAgF;AAGhF,iFAAuE;AACvE,qFAGiD;AACjD,kFAAyE;AACzE,sEAAkE;AAElE,sEAGsC;AACtC,4DAA4D;AAC5D,0DAAgE;AAChE,kEAA6D;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,IAAA,sCAAsB,EAC9C,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,IAAA,2CAAqB,EAAC,QAAQ,CAAC,aAAa,CAAC;QAClE,KAAK,EAAE,IAAA,sCAAgB,EAAC,QAAQ,CAAC,YAAY,CAAC;QAC9C,OAAO;KACR,CAAC,CAAC;AACL,CAAC;AAED;;;;;;;GAOG;AACI,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,IAAA,2CAAqB,EAAC,QAAQ,CAAC,aAAa,CAAC,CAAC;IAC1E,MAAM,cAAc,GAAG,IAAA,sCAAgB,EAAC,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,SAAgB,qBAAqB,CAAC,QAAiB;IACrD,MAAM,MAAM,GAAG,IAAA,sCAAc,GAAE,CAAC;IAChC,MAAM,QAAQ,GAAG,IAAA,2CAAiB,GAAE,CAAC;IACrC,MAAM,EAAE,OAAO,EAAE,GAAG,IAAA,oCAAgB,GAAE,CAAC;IACvC,MAAM,QAAQ,GAAG,IAAA,kBAAU,EAAC,8CAAoB,CAAC,CAAC;IAClD,MAAM,WAAW,GAAG,IAAA,4BAAc,GAAE,CAAC;IAErC,IAAI,CAAC,QAAQ,EAAE,CAAC;QACd,MAAM,IAAI,KAAK,CACb,iEAAiE,CAClE,CAAC;IACJ,CAAC;IAED,OAAO,IAAA,yBAAW,EAAC;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,mCAAe,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,IAAA,kCAAiB,EAAC,aAAa,EAAE;wBACzD,KAAK;qBACN,CAAC,EAAE,CAAC;wBACH,oDAAoD;wBACpD,IAAI,KAAK,CAAC,IAAI,KAAK,gBAAS,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,gBAAS,CAAC,MAAM,EAAE,CAAC;4BACpC,MAAM,WAAW,GAAG,IAAA,0BAAkB,EAAC,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,gBAAS,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"]}

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

@@ -10,11 +10,15 @@ const react_2 = __importDefault(require("react"));
 const zod_1 = require("zod");
 const tambo_client_provider_1 = require("../../providers/tambo-client-provider");
 const tambo_registry_provider_1 = require("../../providers/tambo-registry-provider");
+const tambo_v1_provider_1 = require("../providers/tambo-v1-provider");
 const tambo_v1_stream_context_1 = require("../providers/tambo-v1-stream-context");
 const use_tambo_v1_send_message_1 = require("./use-tambo-v1-send-message");
 jest.mock("../../providers/tambo-client-provider", () => ({
     useTamboClient: jest.fn(),
 }));
+jest.mock("../providers/tambo-v1-provider", () => ({
+    useTamboV1Config: jest.fn(),
+}));
 describe("useTamboV1SendMessage", () => {
     const mockThreadsRunsApi = {
         run: jest.fn(),
@@ -34,7 +38,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, { threadId: "thread_123" }, children))));
+                react_2.default.createElement(tambo_v1_stream_context_1.TamboV1StreamProvider, null, children))));
     }
     beforeEach(() => {
         queryClient = new react_query_1.QueryClient({
@@ -44,6 +48,7 @@ describe("useTamboV1SendMessage", () => {
             },
         });
         jest.mocked(tambo_client_provider_1.useTamboClient).mockReturnValue(mockTamboAI);
+        jest.mocked(tambo_v1_provider_1.useTamboV1Config).mockReturnValue({ userKey: undefined });
         mockThreadsRunsApi.run.mockReset();
         mockThreadsRunsApi.create.mockReset();
     });
@@ -117,11 +122,13 @@ describe("createRunStream", () => {
             threadId: "thread_123",
             message: testMessage,
             registry: mockRegistry,
+            userKey: undefined,
         });
         expect(mockThreadsRunsApi.run).toHaveBeenCalledWith("thread_123", {
             message: testMessage,
             availableComponents: expect.any(Array),
             tools: expect.any(Array),
+            userKey: undefined,
         });
         expect(mockThreadsRunsApi.create).not.toHaveBeenCalled();
         expect(result.stream).toBe(mockStream);
@@ -134,11 +141,13 @@ describe("createRunStream", () => {
             threadId: undefined,
             message: testMessage,
             registry: mockRegistry,
+            userKey: undefined,
         });
         expect(mockThreadsRunsApi.create).toHaveBeenCalledWith({
             message: testMessage,
             availableComponents: expect.any(Array),
             tools: expect.any(Array),
+            thread: undefined,
         });
         expect(mockThreadsRunsApi.run).not.toHaveBeenCalled();
         expect(result.stream).toBe(mockStream);
@@ -151,6 +160,7 @@ describe("createRunStream", () => {
             threadId: "thread_123",
             message: testMessage,
             registry: mockRegistry,
+            userKey: undefined,
         });
         const callArgs = mockThreadsRunsApi.run.mock.calls[0][1];
         expect(callArgs.availableComponents).toEqual([
@@ -168,6 +178,7 @@ describe("createRunStream", () => {
             threadId: "thread_123",
             message: testMessage,
             registry: mockRegistry,
+            userKey: undefined,
         });
         const callArgs = mockThreadsRunsApi.run.mock.calls[0][1];
         expect(callArgs.tools).toEqual([
@@ -188,6 +199,7 @@ describe("createRunStream", () => {
             threadId: "thread_123",
             message: testMessage,
             registry: emptyRegistry,
+            userKey: undefined,
         });
         const callArgs = mockThreadsRunsApi.run.mock.calls[0][1];
         expect(callArgs.availableComponents).toEqual([]);
@@ -223,6 +235,7 @@ describe("useTamboV1SendMessage mutation", () => {
             },
         });
         jest.mocked(tambo_client_provider_1.useTamboClient).mockReturnValue(mockTamboAI);
+        jest.mocked(tambo_v1_provider_1.useTamboV1Config).mockReturnValue({ userKey: undefined });
         mockThreadsRunsApi.run.mockReset();
         mockThreadsRunsApi.create.mockReset();
     });
@@ -344,7 +357,7 @@ 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, { threadId: "thread_123" }, children))));
+                    react_2.default.createElement(tambo_v1_stream_context_1.TamboV1StreamProvider, null, children))));
         }
         const { result } = (0, react_1.renderHook)(() => (0, use_tambo_v1_send_message_1.useTamboV1SendMessage)("thread_123"), {
             wrapper: TestWrapper,
@@ -438,7 +451,7 @@ 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, { threadId: "thread_123" }, children))));
+                    react_2.default.createElement(tambo_v1_stream_context_1.TamboV1StreamProvider, null, children))));
         }
         const { result } = (0, react_1.renderHook)(() => (0, use_tambo_v1_send_message_1.useTamboV1SendMessage)("thread_123"), {
             wrapper: TestWrapper,
@@ -510,7 +523,7 @@ 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, { threadId: "thread_123" }, children))));
+                    react_2.default.createElement(tambo_v1_stream_context_1.TamboV1StreamProvider, null, children))));
         }
         const { result } = (0, react_1.renderHook)(() => (0, use_tambo_v1_send_message_1.useTamboV1SendMessage)("thread_123"), {
             wrapper: TestWrapper,
@@ -577,7 +590,7 @@ 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, { threadId: "thread_123" }, children))));
+                    react_2.default.createElement(tambo_v1_stream_context_1.TamboV1StreamProvider, null, children))));
         }
         const { result } = (0, react_1.renderHook)(() => (0, use_tambo_v1_send_message_1.useTamboV1SendMessage)("thread_123"), {
             wrapper: TestWrapper,
@@ -607,7 +620,7 @@ 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, { threadId: "thread_123" }, children)));
+                react_2.default.createElement(tambo_v1_stream_context_1.TamboV1StreamProvider, 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"), {
@@ -632,7 +645,7 @@ 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, { threadId: "thread_123" }, children))));
+                    react_2.default.createElement(tambo_v1_stream_context_1.TamboV1StreamProvider, null, children))));
         }
         const invalidateQueriesSpy = jest.spyOn(queryClient, "invalidateQueries");
         const { result } = (0, react_1.renderHook)(() => (0, use_tambo_v1_send_message_1.useTamboV1SendMessage)("thread_123"), {
@@ -752,7 +765,7 @@ 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, { threadId: "thread_123" }, children))));
+                    react_2.default.createElement(tambo_v1_stream_context_1.TamboV1StreamProvider, null, children))));
         }
         const { result } = (0, react_1.renderHook)(() => (0, use_tambo_v1_send_message_1.useTamboV1SendMessage)("thread_123"), {
             wrapper: TestWrapper,
@@ -800,7 +813,7 @@ 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, { threadId: "thread_123" }, children))));
+                    react_2.default.createElement(tambo_v1_stream_context_1.TamboV1StreamProvider, null, children))));
         }
         const { result } = (0, react_1.renderHook)(() => (0, use_tambo_v1_send_message_1.useTamboV1SendMessage)("thread_123"), {
             wrapper: TestWrapper,