@tambo-ai/react 1.0.2 → 1.1.0

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

@@ -1,5 +1,5 @@
 import { EventType, } from "@ag-ui/core";
-import { createInitialState, createInitialThreadState, streamReducer, } from "./event-accumulator.js";
+import { PLACEHOLDER_THREAD_ID, createInitialState, createInitialThreadState, streamReducer, } from "./event-accumulator.js";
 /**
  * Helper to extract a ToolUseContent from a message content array.
  * @param content - Content array from a message
@@ -54,11 +54,11 @@ describe("createInitialThreadState", () => {
 describe("createInitialState", () => {
     it("creates initial state with placeholder thread", () => {
         const state = 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(PLACEHOLDER_THREAD_ID);
+        expect(state.threadMap[PLACEHOLDER_THREAD_ID]).toBeDefined();
+        expect(state.threadMap[PLACEHOLDER_THREAD_ID].thread.id).toBe(PLACEHOLDER_THREAD_ID);
+        expect(state.threadMap[PLACEHOLDER_THREAD_ID].thread.messages).toEqual([]);
+        expect(state.threadMap[PLACEHOLDER_THREAD_ID].streaming.status).toBe("idle");
     });
 });
 describe("streamReducer", () => {
@@ -136,6 +136,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 = createInitialState();
+            const result = 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 = createInitialState();
+            const initialThread = {
+                messages: [
+                    {
+                        id: "seed_1",
+                        role: "user",
+                        content: [{ type: "text", text: "Hello" }],
+                    },
+                ],
+            };
+            state = streamReducer(state, {
+                type: "START_NEW_THREAD",
+                threadId: PLACEHOLDER_THREAD_ID,
+                initialThread,
+            });
+            expect(state.currentThreadId).toBe(PLACEHOLDER_THREAD_ID);
+            expect(state.threadMap[PLACEHOLDER_THREAD_ID].thread.messages).toHaveLength(1);
+            expect(state.threadMap[PLACEHOLDER_THREAD_ID].thread.messages[0].id).toBe("seed_1");
+            // Simulate a completed conversation: add more messages to the placeholder
+            const userMsgStart = {
+                type: EventType.TEXT_MESSAGE_START,
+                messageId: "extra_msg",
+                role: "user",
+            };
+            const userMsgContent = {
+                type: EventType.TEXT_MESSAGE_CONTENT,
+                messageId: "extra_msg",
+                delta: "Follow-up",
+            };
+            state = streamReducer(state, {
+                type: "EVENT",
+                event: userMsgStart,
+                threadId: PLACEHOLDER_THREAD_ID,
+            });
+            state = streamReducer(state, {
+                type: "EVENT",
+                event: userMsgContent,
+                threadId: PLACEHOLDER_THREAD_ID,
+            });
+            expect(state.threadMap[PLACEHOLDER_THREAD_ID].thread.messages).toHaveLength(2);
+            // Second call: start a new thread again — should reset to fresh state
+            state = streamReducer(state, {
+                type: "START_NEW_THREAD",
+                threadId: PLACEHOLDER_THREAD_ID,
+                initialThread,
+            });
+            // Should be reset to just the initial seed message, not the old messages
+            expect(state.currentThreadId).toBe(PLACEHOLDER_THREAD_ID);
+            expect(state.threadMap[PLACEHOLDER_THREAD_ID].thread.messages).toHaveLength(1);
+            expect(state.threadMap[PLACEHOLDER_THREAD_ID].thread.messages[0].id).toBe("seed_1");
+        });
+        it("overwrites existing placeholder even without initialThread", () => {
+            let state = createInitialState();
+            // Add a message to the placeholder
+            state = streamReducer(state, {
+                type: "EVENT",
+                event: {
+                    type: EventType.TEXT_MESSAGE_START,
+                    messageId: "msg_1",
+                    role: "user",
+                },
+                threadId: PLACEHOLDER_THREAD_ID,
+            });
+            expect(state.threadMap[PLACEHOLDER_THREAD_ID].thread.messages).toHaveLength(1);
+            // START_NEW_THREAD without initialThread should still reset
+            state = streamReducer(state, {
+                type: "START_NEW_THREAD",
+                threadId: PLACEHOLDER_THREAD_ID,
+            });
+            expect(state.currentThreadId).toBe(PLACEHOLDER_THREAD_ID);
+            expect(state.threadMap[PLACEHOLDER_THREAD_ID].thread.messages).toHaveLength(0);
+        });
+    });
     describe("RUN_STARTED event", () => {
         it("updates thread status to streaming", () => {
             const state = createTestStreamState("thread_1");
@@ -201,14 +287,14 @@ describe("streamReducer", () => {
                 event: runStartedEvent,
                 threadId: realThreadId,
             });
-            expect(result.threadMap.placeholder.thread.messages).toHaveLength(0);
-            expect(result.currentThreadId).toBe("placeholder");
+            expect(result.threadMap[PLACEHOLDER_THREAD_ID].thread.messages).toHaveLength(0);
+            expect(result.currentThreadId).toBe(PLACEHOLDER_THREAD_ID);
         });
         it("migrates messages from placeholder thread to real thread", () => {
             // Start with initial state (which has placeholder thread)
             const state = createInitialState();
             // Verify placeholder thread exists
-            expect(state.currentThreadId).toBe("placeholder");
+            expect(state.currentThreadId).toBe(PLACEHOLDER_THREAD_ID);
             // Add a user message to the placeholder thread
             const userMsgStart = {
                 type: EventType.TEXT_MESSAGE_START,
@@ -227,22 +313,23 @@ describe("streamReducer", () => {
             let stateWithUserMsg = streamReducer(state, {
                 type: "EVENT",
                 event: userMsgStart,
-                threadId: "placeholder",
+                threadId: PLACEHOLDER_THREAD_ID,
             });
             stateWithUserMsg = streamReducer(stateWithUserMsg, {
                 type: "EVENT",
                 event: userMsgContent,
-                threadId: "placeholder",
+                threadId: PLACEHOLDER_THREAD_ID,
             });
             stateWithUserMsg = streamReducer(stateWithUserMsg, {
                 type: "EVENT",
                 event: userMsgEnd,
-                threadId: "placeholder",
+                threadId: 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(PLACEHOLDER_THREAD_ID);
+            expect(stateWithUserMsg.threadMap[PLACEHOLDER_THREAD_ID].thread.messages).toHaveLength(1);
+            expect(stateWithUserMsg.threadMap[PLACEHOLDER_THREAD_ID].thread.messages[0]
+                .content[0]).toEqual({
                 type: "text",
                 text: "Hello",
             });
@@ -260,8 +347,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[PLACEHOLDER_THREAD_ID]).toBeDefined();
+            expect(finalState.threadMap[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);
@@ -302,17 +389,17 @@ describe("streamReducer", () => {
             state = streamReducer(state, {
                 type: "EVENT",
                 event: userMsgStart,
-                threadId: "placeholder",
+                threadId: PLACEHOLDER_THREAD_ID,
             });
             state = streamReducer(state, {
                 type: "EVENT",
                 event: userMsgContent,
-                threadId: "placeholder",
+                threadId: PLACEHOLDER_THREAD_ID,
             });
             state = streamReducer(state, {
                 type: "EVENT",
                 event: userMsgEnd,
-                threadId: "placeholder",
+                threadId: PLACEHOLDER_THREAD_ID,
             });
             const realThreadId = "thread_real_123";
             const runStartedEvent = {
@@ -326,7 +413,7 @@ describe("streamReducer", () => {
                 event: runStartedEvent,
                 threadId: realThreadId,
             });
-            expect(result.threadMap.placeholder.thread.messages).toHaveLength(0);
+            expect(result.threadMap[PLACEHOLDER_THREAD_ID].thread.messages).toHaveLength(0);
             expect(result.threadMap[realThreadId].thread.messages).toHaveLength(1);
             expect(result.currentThreadId).toBe("thread_1");
         });
@@ -351,17 +438,17 @@ describe("streamReducer", () => {
             let stateWithUserMsg = streamReducer(state, {
                 type: "EVENT",
                 event: userMsgStart,
-                threadId: "placeholder",
+                threadId: PLACEHOLDER_THREAD_ID,
             });
             stateWithUserMsg = streamReducer(stateWithUserMsg, {
                 type: "EVENT",
                 event: userMsgContent,
-                threadId: "placeholder",
+                threadId: PLACEHOLDER_THREAD_ID,
             });
             stateWithUserMsg = streamReducer(stateWithUserMsg, {
                 type: "EVENT",
                 event: userMsgEnd,
-                threadId: "placeholder",
+                threadId: PLACEHOLDER_THREAD_ID,
             });
             const realThreadId = "thread_real_123";
             const runStartedEvent = {
@@ -373,9 +460,9 @@ describe("streamReducer", () => {
             const result = streamReducer(stateWithUserMsg, {
                 type: "EVENT",
                 event: runStartedEvent,
-                threadId: "placeholder",
+                threadId: PLACEHOLDER_THREAD_ID,
             });
-            expect(result.threadMap.placeholder.thread.messages).toHaveLength(0);
+            expect(result.threadMap[PLACEHOLDER_THREAD_ID].thread.messages).toHaveLength(0);
             expect(result.threadMap[realThreadId].thread.messages).toHaveLength(1);
             expect(result.currentThreadId).toBe(realThreadId);
         });