@tambo-ai/react 0.74.1 → 0.75.0

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

@@ -400,6 +400,43 @@ describe("streamReducer", () => {
             expect(result.threadMap.thread_1.thread.status).toBe("complete");
             expect(result.threadMap.thread_1.streaming.status).toBe("complete");
         });
+        it("sets lastCompletedRunId from event.runId", () => {
+            const state = createTestStreamState("thread_1");
+            state.threadMap.thread_1.streaming = {
+                status: "streaming",
+                runId: "run_abc",
+            };
+            const event = {
+                type: core_1.EventType.RUN_FINISHED,
+                runId: "run_abc",
+                threadId: "thread_1",
+            };
+            const result = (0, event_accumulator_1.streamReducer)(state, {
+                type: "EVENT",
+                event,
+                threadId: "thread_1",
+            });
+            expect(result.threadMap.thread_1.lastCompletedRunId).toBe("run_abc");
+        });
+        it("prefers event.runId over streaming.runId for lastCompletedRunId", () => {
+            const state = createTestStreamState("thread_1");
+            state.threadMap.thread_1.lastCompletedRunId = "run_old";
+            state.threadMap.thread_1.streaming = {
+                status: "streaming",
+                // runId intentionally undefined
+            };
+            const event = {
+                type: core_1.EventType.RUN_FINISHED,
+                runId: "run_from_event",
+                threadId: "thread_1",
+            };
+            const result = (0, event_accumulator_1.streamReducer)(state, {
+                type: "EVENT",
+                event,
+                threadId: "thread_1",
+            });
+            expect(result.threadMap.thread_1.lastCompletedRunId).toBe("run_from_event");
+        });
     });
     describe("RUN_ERROR event", () => {
         it("updates thread status to error with details", () => {
@@ -925,6 +962,34 @@ describe("streamReducer", () => {
                 });
             }).toThrow("Failed to parse tool call arguments");
         });
+        it("uses parsedToolArgs from action when provided (skips own parsing)", () => {
+            const state = createTestStreamState("thread_1");
+            state.threadMap.thread_1.thread.messages = [
+                {
+                    id: "msg_1",
+                    role: "assistant",
+                    content: [
+                        { type: "tool_use", id: "tool_1", name: "test", input: {} },
+                    ],
+                    createdAt: "2024-01-01T00:00:00.000Z",
+                },
+            ];
+            const preParsed = { city: "Seattle", units: "celsius" };
+            const result = (0, event_accumulator_1.streamReducer)(state, {
+                type: "EVENT",
+                event: {
+                    type: core_1.EventType.TOOL_CALL_ARGS,
+                    toolCallId: "tool_1",
+                    // Deliberately incomplete JSON — if the reducer re-parses, it would not match preParsed
+                    delta: '{"city": "Seat',
+                },
+                threadId: "thread_1",
+                parsedToolArgs: preParsed,
+            });
+            const toolContent = asToolUseContent(result.threadMap.thread_1.thread.messages[0].content, 0);
+            // Should use the pre-parsed value, not the incomplete delta
+            expect(toolContent.input).toEqual(preParsed);
+        });
     });
     describe("custom component events", () => {
         it("handles tambo.component.start event", () => {
@@ -2150,6 +2215,16 @@ describe("streamReducer", () => {
             expect(warnSpy).toHaveBeenCalledWith(expect.stringContaining('unexpected streamingState "streaming"'));
             warnSpy.mockRestore();
         });
+        it("does not modify lastCompletedRunId", () => {
+            const state = createTestStreamState("thread_1");
+            state.threadMap.thread_1.lastCompletedRunId = "run_existing";
+            const result = (0, event_accumulator_1.streamReducer)(state, {
+                type: "LOAD_THREAD_MESSAGES",
+                threadId: "thread_1",
+                messages: [],
+            });
+            expect(result.threadMap.thread_1.lastCompletedRunId).toBe("run_existing");
+        });
         it("handles system role messages from API", () => {
             const state = createTestStreamState("thread_1");
             const result = (0, event_accumulator_1.streamReducer)(state, {
@@ -2175,6 +2250,37 @@ describe("streamReducer", () => {
             expect(result.threadMap.thread_1.thread.messages[1].role).toBe("user");
         });
     });
+    describe("SET_LAST_COMPLETED_RUN_ID action", () => {
+        it("stores lastCompletedRunId on an existing thread", () => {
+            const state = createTestStreamState("thread_1");
+            const result = (0, event_accumulator_1.streamReducer)(state, {
+                type: "SET_LAST_COMPLETED_RUN_ID",
+                threadId: "thread_1",
+                lastCompletedRunId: "run_xyz",
+            });
+            expect(result.threadMap.thread_1.lastCompletedRunId).toBe("run_xyz");
+        });
+        it("overwrites existing lastCompletedRunId", () => {
+            const state = createTestStreamState("thread_1");
+            state.threadMap.thread_1.lastCompletedRunId = "run_old";
+            const result = (0, event_accumulator_1.streamReducer)(state, {
+                type: "SET_LAST_COMPLETED_RUN_ID",
+                threadId: "thread_1",
+                lastCompletedRunId: "run_new",
+            });
+            expect(result.threadMap.thread_1.lastCompletedRunId).toBe("run_new");
+        });
+        it("creates thread state if thread does not exist", () => {
+            const state = createTestStreamState("thread_1");
+            const result = (0, event_accumulator_1.streamReducer)(state, {
+                type: "SET_LAST_COMPLETED_RUN_ID",
+                threadId: "thread_new",
+                lastCompletedRunId: "run_abc",
+            });
+            expect(result.threadMap.thread_new).toBeDefined();
+            expect(result.threadMap.thread_new.lastCompletedRunId).toBe("run_abc");
+        });
+    });
     describe("UPDATE_THREAD_TITLE action", () => {
         it("updates the title on an existing thread", () => {
             const state = createTestStreamState("thread_1");