fix: add missing timestamp field to ToolCallEndEvent schema and usages

Author: ThomasK33

src/browser/utils/messages/StreamingMessageAggregator.status.test.ts

@@ -41,6 +41,7 @@ describe("StreamingMessageAggregator - Agent Status", () => {
       toolCallId,
       toolName: "status_set",
       result: { success: true, emoji: "🔍", message: "Analyzing code" },
+      timestamp: Date.now(),
     });
 
     const status = aggregator.getAgentStatus();
@@ -81,6 +82,7 @@ describe("StreamingMessageAggregator - Agent Status", () => {
       toolCallId: "tool1",
       toolName: "status_set",
       result: { success: true, emoji: "🔍", message: "Analyzing" },
+      timestamp: Date.now(),
     });
 
     expect(aggregator.getAgentStatus()?.emoji).toBe("🔍");
@@ -104,6 +106,7 @@ describe("StreamingMessageAggregator - Agent Status", () => {
       toolCallId: "tool2",
       toolName: "status_set",
       result: { success: true, emoji: "📝", message: "Writing" },
+      timestamp: Date.now(),
     });
 
     expect(aggregator.getAgentStatus()?.emoji).toBe("📝");
@@ -142,6 +145,7 @@ describe("StreamingMessageAggregator - Agent Status", () => {
       toolCallId: "tool1",
       toolName: "status_set",
       result: { success: true, emoji: "🔍", message: "Working" },
+      timestamp: Date.now(),
     });
 
     expect(aggregator.getAgentStatus()).toBeDefined();
@@ -193,6 +197,7 @@ describe("StreamingMessageAggregator - Agent Status", () => {
       toolCallId: "tool1",
       toolName: "status_set",
       result: { success: false, error: "Something went wrong" },
+      timestamp: Date.now(),
     });
 
     // Status should remain undefined
@@ -229,6 +234,7 @@ describe("StreamingMessageAggregator - Agent Status", () => {
       toolCallId: "tool1",
       toolName: "status_set",
       result: { success: true, emoji: "🔍", message: "First task" },
+      timestamp: Date.now(),
     });
 
     expect(aggregator.getAgentStatus()?.message).toBe("First task");
@@ -292,6 +298,7 @@ describe("StreamingMessageAggregator - Agent Status", () => {
       toolCallId: "tool1",
       toolName: "status_set",
       result: { success: false, error: "emoji must be a single emoji character" },
+      timestamp: Date.now(),
     });
 
     // End the stream to finalize message
@@ -350,6 +357,7 @@ describe("StreamingMessageAggregator - Agent Status", () => {
       toolCallId: "tool1",
       toolName: "status_set",
       result: { success: true, emoji: "🔍", message: "Analyzing code" },
+      timestamp: Date.now(),
     });
 
     // End the stream to finalize message
@@ -533,6 +541,7 @@ describe("StreamingMessageAggregator - Agent Status", () => {
       toolCallId,
       toolName: "status_set",
       result: { success: true, emoji: "✅", message: truncatedMessage },
+      timestamp: Date.now(),
     });
 
     // Should use truncated message from output, not the original input
@@ -576,6 +585,7 @@ describe("StreamingMessageAggregator - Agent Status", () => {
       toolCallId,
       toolName: "status_set",
       result: { success: true, emoji: "🔗", message: "PR submitted", url: testUrl },
+      timestamp: Date.now(),
     });
 
     const status = aggregator.getAgentStatus();
@@ -618,6 +628,7 @@ describe("StreamingMessageAggregator - Agent Status", () => {
       toolCallId: "tool1",
       toolName: "status_set",
       result: { success: true, emoji: "🔗", message: "PR submitted", url: testUrl },
+      timestamp: Date.now(),
     });
 
     expect(aggregator.getAgentStatus()?.url).toBe(testUrl);
@@ -641,6 +652,7 @@ describe("StreamingMessageAggregator - Agent Status", () => {
       toolCallId: "tool2",
       toolName: "status_set",
       result: { success: true, emoji: "✅", message: "Done" },
+      timestamp: Date.now(),
     });
 
     const statusAfterUpdate = aggregator.getAgentStatus();
@@ -668,6 +680,7 @@ describe("StreamingMessageAggregator - Agent Status", () => {
       toolCallId: "tool3",
       toolName: "status_set",
       result: { success: true, emoji: "🔄", message: "New PR", url: newUrl },
+      timestamp: Date.now(),
     });
 
     const finalStatus = aggregator.getAgentStatus();
@@ -709,6 +722,7 @@ describe("StreamingMessageAggregator - Agent Status", () => {
       toolCallId: "tool1",
       toolName: "status_set",
       result: { success: true, emoji: "🔗", message: "PR submitted", url: testUrl },
+      timestamp: Date.now(),
     });
 
     expect(aggregator.getAgentStatus()?.url).toBe(testUrl);
@@ -754,6 +768,7 @@ describe("StreamingMessageAggregator - Agent Status", () => {
       toolCallId: "tool2",
       toolName: "status_set",
       result: { success: true, emoji: "✅", message: "Tests passed" },
+      timestamp: Date.now(),
     });
 
     const finalStatus = aggregator.getAgentStatus();

src/browser/utils/messages/StreamingMessageAggregator.test.ts

@@ -175,6 +175,7 @@ describe("StreamingMessageAggregator", () => {
         toolCallId: "tool1",
         toolName: "todo_write",
         result: { success: true },
+        timestamp: Date.now(),
       });
 
       // Verify todos are set
@@ -230,6 +231,7 @@ describe("StreamingMessageAggregator", () => {
         toolCallId: "tool1",
         toolName: "todo_write",
         result: { success: true },
+        timestamp: Date.now(),
       });
 
       expect(aggregator.getCurrentTodos()).toHaveLength(1);
@@ -363,6 +365,7 @@ describe("StreamingMessageAggregator", () => {
         toolCallId: "tool1",
         toolName: "todo_write",
         result: { success: true },
+        timestamp: Date.now(),
       });
 
       // TODOs should be set

src/common/orpc/schemas/stream.ts

@@ -151,6 +151,7 @@ export const ToolCallEndEventSchema = z.object({
   toolCallId: z.string(),
   toolName: z.string(),
   result: z.unknown(),
+  timestamp: z.number().meta({ description: "When tool call completed (Date.now())" }),
 });
 
 export const ReasoningStartEventSchema = z.object({

src/node/services/mock/mockScenarioPlayer.ts

@@ -315,6 +315,7 @@ export class MockScenarioPlayer {
           toolCallId: event.toolCallId,
           toolName: event.toolName,
           result: event.result,
+          timestamp: Date.now(),
         };
         this.deps.aiService.emit("tool-call-end", payload);
         break;

src/node/services/streamManager.ts

@@ -408,6 +408,7 @@ export class StreamManager extends EventEmitter {
           toolCallId: part.toolCallId,
           toolName: part.toolName,
           result: part.output,
+          timestamp: Date.now(),
         });
       }
     }
@@ -655,6 +656,7 @@ export class StreamManager extends EventEmitter {
       toolCallId,
       toolName,
       result: output,
+      timestamp: Date.now(),
     } as ToolCallEndEvent);
 
     // Schedule partial write