@tambo-ai/react 1.0.1 → 1.0.3

package/dist/v1/utils/event-accumulator.test.js

@@ -56,11 +56,11 @@ describe("createInitialThreadState", () => {
 describe("createInitialState", () => {
     it("creates initial state with placeholder thread", () => {
         const state = (0, event_accumulator_1.createInitialState)();
-        expect(state.currentThreadId).toBe("placeholder");
-        expect(state.threadMap.placeholder).toBeDefined();
-        expect(state.threadMap.placeholder.thread.id).toBe("placeholder");
-        expect(state.threadMap.placeholder.thread.messages).toEqual([]);
-        expect(state.threadMap.placeholder.streaming.status).toBe("idle");
+        expect(state.currentThreadId).toBe(event_accumulator_1.PLACEHOLDER_THREAD_ID);
+        expect(state.threadMap[event_accumulator_1.PLACEHOLDER_THREAD_ID]).toBeDefined();
+        expect(state.threadMap[event_accumulator_1.PLACEHOLDER_THREAD_ID].thread.id).toBe(event_accumulator_1.PLACEHOLDER_THREAD_ID);
+        expect(state.threadMap[event_accumulator_1.PLACEHOLDER_THREAD_ID].thread.messages).toEqual([]);
+        expect(state.threadMap[event_accumulator_1.PLACEHOLDER_THREAD_ID].streaming.status).toBe("idle");
     });
 });
 describe("streamReducer", () => {
@@ -138,6 +138,92 @@ describe("streamReducer", () => {
             expect(consoleWarnSpy).toHaveBeenCalledWith(expect.stringContaining("Unknown custom event name"));
         });
     });
+    describe("START_NEW_THREAD action", () => {
+        it("creates a new thread and sets it as current", () => {
+            const state = (0, event_accumulator_1.createInitialState)();
+            const result = (0, event_accumulator_1.streamReducer)(state, {
+                type: "START_NEW_THREAD",
+                threadId: "new_thread",
+            });
+            expect(result.currentThreadId).toBe("new_thread");
+            expect(result.threadMap.new_thread).toBeDefined();
+            expect(result.threadMap.new_thread.thread.messages).toHaveLength(0);
+        });
+        it("overwrites existing placeholder thread on second call", () => {
+            // First call: start a new thread on the placeholder
+            let state = (0, event_accumulator_1.createInitialState)();
+            const initialThread = {
+                messages: [
+                    {
+                        id: "seed_1",
+                        role: "user",
+                        content: [{ type: "text", text: "Hello" }],
+                    },
+                ],
+            };
+            state = (0, event_accumulator_1.streamReducer)(state, {
+                type: "START_NEW_THREAD",
+                threadId: event_accumulator_1.PLACEHOLDER_THREAD_ID,
+                initialThread,
+            });
+            expect(state.currentThreadId).toBe(event_accumulator_1.PLACEHOLDER_THREAD_ID);
+            expect(state.threadMap[event_accumulator_1.PLACEHOLDER_THREAD_ID].thread.messages).toHaveLength(1);
+            expect(state.threadMap[event_accumulator_1.PLACEHOLDER_THREAD_ID].thread.messages[0].id).toBe("seed_1");
+            // Simulate a completed conversation: add more messages to the placeholder
+            const userMsgStart = {
+                type: core_1.EventType.TEXT_MESSAGE_START,
+                messageId: "extra_msg",
+                role: "user",
+            };
+            const userMsgContent = {
+                type: core_1.EventType.TEXT_MESSAGE_CONTENT,
+                messageId: "extra_msg",
+                delta: "Follow-up",
+            };
+            state = (0, event_accumulator_1.streamReducer)(state, {
+                type: "EVENT",
+                event: userMsgStart,
+                threadId: event_accumulator_1.PLACEHOLDER_THREAD_ID,
+            });
+            state = (0, event_accumulator_1.streamReducer)(state, {
+                type: "EVENT",
+                event: userMsgContent,
+                threadId: event_accumulator_1.PLACEHOLDER_THREAD_ID,
+            });
+            expect(state.threadMap[event_accumulator_1.PLACEHOLDER_THREAD_ID].thread.messages).toHaveLength(2);
+            // Second call: start a new thread again — should reset to fresh state
+            state = (0, event_accumulator_1.streamReducer)(state, {
+                type: "START_NEW_THREAD",
+                threadId: event_accumulator_1.PLACEHOLDER_THREAD_ID,
+                initialThread,
+            });
+            // Should be reset to just the initial seed message, not the old messages
+            expect(state.currentThreadId).toBe(event_accumulator_1.PLACEHOLDER_THREAD_ID);
+            expect(state.threadMap[event_accumulator_1.PLACEHOLDER_THREAD_ID].thread.messages).toHaveLength(1);
+            expect(state.threadMap[event_accumulator_1.PLACEHOLDER_THREAD_ID].thread.messages[0].id).toBe("seed_1");
+        });
+        it("overwrites existing placeholder even without initialThread", () => {
+            let state = (0, event_accumulator_1.createInitialState)();
+            // Add a message to the placeholder
+            state = (0, event_accumulator_1.streamReducer)(state, {
+                type: "EVENT",
+                event: {
+                    type: core_1.EventType.TEXT_MESSAGE_START,
+                    messageId: "msg_1",
+                    role: "user",
+                },
+                threadId: event_accumulator_1.PLACEHOLDER_THREAD_ID,
+            });
+            expect(state.threadMap[event_accumulator_1.PLACEHOLDER_THREAD_ID].thread.messages).toHaveLength(1);
+            // START_NEW_THREAD without initialThread should still reset
+            state = (0, event_accumulator_1.streamReducer)(state, {
+                type: "START_NEW_THREAD",
+                threadId: event_accumulator_1.PLACEHOLDER_THREAD_ID,
+            });
+            expect(state.currentThreadId).toBe(event_accumulator_1.PLACEHOLDER_THREAD_ID);
+            expect(state.threadMap[event_accumulator_1.PLACEHOLDER_THREAD_ID].thread.messages).toHaveLength(0);
+        });
+    });
     describe("RUN_STARTED event", () => {
         it("updates thread status to streaming", () => {
             const state = createTestStreamState("thread_1");
@@ -203,14 +289,14 @@ describe("streamReducer", () => {
                 event: runStartedEvent,
                 threadId: realThreadId,
             });
-            expect(result.threadMap.placeholder.thread.messages).toHaveLength(0);
-            expect(result.currentThreadId).toBe("placeholder");
+            expect(result.threadMap[event_accumulator_1.PLACEHOLDER_THREAD_ID].thread.messages).toHaveLength(0);
+            expect(result.currentThreadId).toBe(event_accumulator_1.PLACEHOLDER_THREAD_ID);
         });
         it("migrates messages from placeholder thread to real thread", () => {
             // Start with initial state (which has placeholder thread)
             const state = (0, event_accumulator_1.createInitialState)();
             // Verify placeholder thread exists
-            expect(state.currentThreadId).toBe("placeholder");
+            expect(state.currentThreadId).toBe(event_accumulator_1.PLACEHOLDER_THREAD_ID);
             // Add a user message to the placeholder thread
             const userMsgStart = {
                 type: core_1.EventType.TEXT_MESSAGE_START,
@@ -229,22 +315,23 @@ describe("streamReducer", () => {
             let stateWithUserMsg = (0, event_accumulator_1.streamReducer)(state, {
                 type: "EVENT",
                 event: userMsgStart,
-                threadId: "placeholder",
+                threadId: event_accumulator_1.PLACEHOLDER_THREAD_ID,
             });
             stateWithUserMsg = (0, event_accumulator_1.streamReducer)(stateWithUserMsg, {
                 type: "EVENT",
                 event: userMsgContent,
-                threadId: "placeholder",
+                threadId: event_accumulator_1.PLACEHOLDER_THREAD_ID,
             });
             stateWithUserMsg = (0, event_accumulator_1.streamReducer)(stateWithUserMsg, {
                 type: "EVENT",
                 event: userMsgEnd,
-                threadId: "placeholder",
+                threadId: event_accumulator_1.PLACEHOLDER_THREAD_ID,
             });
             // Verify placeholder thread has the message
-            expect(stateWithUserMsg.currentThreadId).toBe("placeholder");
-            expect(stateWithUserMsg.threadMap.placeholder.thread.messages).toHaveLength(1);
-            expect(stateWithUserMsg.threadMap.placeholder.thread.messages[0].content[0]).toEqual({
+            expect(stateWithUserMsg.currentThreadId).toBe(event_accumulator_1.PLACEHOLDER_THREAD_ID);
+            expect(stateWithUserMsg.threadMap[event_accumulator_1.PLACEHOLDER_THREAD_ID].thread.messages).toHaveLength(1);
+            expect(stateWithUserMsg.threadMap[event_accumulator_1.PLACEHOLDER_THREAD_ID].thread.messages[0]
+                .content[0]).toEqual({
                 type: "text",
                 text: "Hello",
             });
@@ -262,8 +349,8 @@ describe("streamReducer", () => {
                 threadId: realThreadId,
             });
             // Placeholder thread should be reset to empty (not removed)
-            expect(finalState.threadMap.placeholder).toBeDefined();
-            expect(finalState.threadMap.placeholder.thread.messages).toHaveLength(0);
+            expect(finalState.threadMap[event_accumulator_1.PLACEHOLDER_THREAD_ID]).toBeDefined();
+            expect(finalState.threadMap[event_accumulator_1.PLACEHOLDER_THREAD_ID].thread.messages).toHaveLength(0);
             // Real thread should have the migrated user message
             expect(finalState.threadMap[realThreadId]).toBeDefined();
             expect(finalState.threadMap[realThreadId].thread.messages).toHaveLength(1);
@@ -304,17 +391,17 @@ describe("streamReducer", () => {
             state = (0, event_accumulator_1.streamReducer)(state, {
                 type: "EVENT",
                 event: userMsgStart,
-                threadId: "placeholder",
+                threadId: event_accumulator_1.PLACEHOLDER_THREAD_ID,
             });
             state = (0, event_accumulator_1.streamReducer)(state, {
                 type: "EVENT",
                 event: userMsgContent,
-                threadId: "placeholder",
+                threadId: event_accumulator_1.PLACEHOLDER_THREAD_ID,
             });
             state = (0, event_accumulator_1.streamReducer)(state, {
                 type: "EVENT",
                 event: userMsgEnd,
-                threadId: "placeholder",
+                threadId: event_accumulator_1.PLACEHOLDER_THREAD_ID,
             });
             const realThreadId = "thread_real_123";
             const runStartedEvent = {
@@ -328,7 +415,7 @@ describe("streamReducer", () => {
                 event: runStartedEvent,
                 threadId: realThreadId,
             });
-            expect(result.threadMap.placeholder.thread.messages).toHaveLength(0);
+            expect(result.threadMap[event_accumulator_1.PLACEHOLDER_THREAD_ID].thread.messages).toHaveLength(0);
             expect(result.threadMap[realThreadId].thread.messages).toHaveLength(1);
             expect(result.currentThreadId).toBe("thread_1");
         });
@@ -353,17 +440,17 @@ describe("streamReducer", () => {
             let stateWithUserMsg = (0, event_accumulator_1.streamReducer)(state, {
                 type: "EVENT",
                 event: userMsgStart,
-                threadId: "placeholder",
+                threadId: event_accumulator_1.PLACEHOLDER_THREAD_ID,
             });
             stateWithUserMsg = (0, event_accumulator_1.streamReducer)(stateWithUserMsg, {
                 type: "EVENT",
                 event: userMsgContent,
-                threadId: "placeholder",
+                threadId: event_accumulator_1.PLACEHOLDER_THREAD_ID,
             });
             stateWithUserMsg = (0, event_accumulator_1.streamReducer)(stateWithUserMsg, {
                 type: "EVENT",
                 event: userMsgEnd,
-                threadId: "placeholder",
+                threadId: event_accumulator_1.PLACEHOLDER_THREAD_ID,
             });
             const realThreadId = "thread_real_123";
             const runStartedEvent = {
@@ -375,9 +462,9 @@ describe("streamReducer", () => {
             const result = (0, event_accumulator_1.streamReducer)(stateWithUserMsg, {
                 type: "EVENT",
                 event: runStartedEvent,
-                threadId: "placeholder",
+                threadId: event_accumulator_1.PLACEHOLDER_THREAD_ID,
             });
-            expect(result.threadMap.placeholder.thread.messages).toHaveLength(0);
+            expect(result.threadMap[event_accumulator_1.PLACEHOLDER_THREAD_ID].thread.messages).toHaveLength(0);
             expect(result.threadMap[realThreadId].thread.messages).toHaveLength(1);
             expect(result.currentThreadId).toBe(realThreadId);
         });