feat: added timestamp property to context messages

Author: wladimiiir

CHANGELOG.md

@@ -5,6 +5,7 @@
 - correctly rendering the context around the match line in grep power tool output
 - deduplicated skills prioritizing extension then project then global skills
 - added run_prompt tasks tool for running a prompt
+- added timestamp property to context messages
 
 ## [0.66.0]
 

docs-site/docs/features/socketio-events.md

@@ -100,7 +100,8 @@ Emitted when an AI response is fully completed.
       "color": "optional group color",
       "finished": false
     }
-  }
+  },
+  "timestamp": 1716123456789
 }
 ```
 
@@ -243,7 +244,8 @@ Emitted during tool execution.
       "color": "optional group color",
       "finished": false
     }
-  }
+  },
+  "timestamp": 1716123456789
 }
 ```
 
@@ -256,7 +258,8 @@ Emitted when a command is executed.
   "baseDir": "/path/to/project",
   "taskId": "task-uuid",
   "command": "npm install",
-  "output": "Installing dependencies...\nDone."
+  "output": "Installing dependencies...\nDone.",
+  "timestamp": 1716123456789
 }
 ```
 
@@ -308,7 +311,8 @@ Emitted for logging information.
       "color": "optional group color",
       "finished": false
     }
-  }
+  },
+  "timestamp": 1716123456789
 }
 ```
 
@@ -373,7 +377,8 @@ Emitted when a user sends a message.
       "color": "optional group color",
       "finished": false
     }
-  }
+  },
+  "timestamp": 1716123456789
 }
 ```
 

packages/common/src/extensions.ts

@@ -47,6 +47,7 @@ export interface ResponseMessage {
   finished: boolean;
   usageReport?: UsageReportData;
   promptContext?: PromptContext;
+  timestamp?: number;
 }
 
 /**

packages/common/src/types/common.ts

@@ -161,13 +161,15 @@ export interface ResponseCompletedData {
   usageReport?: UsageReportData;
   sequenceNumber?: number;
   promptContext?: PromptContext;
+  timestamp?: number;
 }
 
 export interface CommandOutputData {
   baseDir: string;
   taskId: string;
   command: string;
   output: string;
+  timestamp?: number;
 }
 
 export type LogLevel = 'info' | 'warning' | 'error' | 'loading';
@@ -180,6 +182,7 @@ export interface LogData {
   finished?: boolean;
   promptContext?: PromptContext;
   actionIds?: string[];
+  timestamp?: number;
 }
 
 // System log types (for application-wide logging)
@@ -215,6 +218,7 @@ export interface ToolData {
   usageReport?: UsageReportData;
   promptContext?: PromptContext;
   finished?: boolean;
+  timestamp?: number;
 }
 
 export interface ContextFilesUpdatedData {
@@ -739,6 +743,7 @@ export interface UserMessageData {
   content: string;
   images?: string[];
   promptContext?: PromptContext;
+  timestamp?: number;
 }
 
 export interface MessageRemovedData {

packages/common/src/types/context.ts

@@ -97,6 +97,7 @@ export enum MessageRole {
 // Base interface for all context messages with usage reporting
 interface BaseContextMessage {
   id: string;
+  timestamp?: number;
   usageReport?: UsageReportData;
   promptContext?: PromptContext;
 }

packages/common/src/types/ui-messages.ts

@@ -6,6 +6,7 @@ export interface Message {
   type: 'user' | 'response' | 'loading' | 'reflected-message' | 'command-output' | 'log' | 'tokens-info' | 'tool' | 'group' | 'task-info' | 'assistant-group';
   content: string;
   promptContext?: PromptContext;
+  timestamp?: number;
   children?: Message[];
 }
 

src/main/agent/agent.ts

@@ -783,6 +783,7 @@ export class Agent {
                 ]
               : prompt,
           promptContext,
+          timestamp: Date.now(),
         }
       : null;
 
@@ -2015,13 +2016,15 @@ export class Agent {
           id: currentResponseId,
           usageReport,
           promptContext,
+          timestamp: Date.now(),
         });
       } else if (message.role === 'tool') {
         messages.push({
           ...message,
           // @ts-expect-error the id is there
           id: message.id || uuidv4(),
           promptContext,
+          timestamp: Date.now(),
         });
       }
     });

src/main/events/event-manager.ts

@@ -11,6 +11,7 @@ import {
   QuestionAnsweredData,
   ResponseChunkData,
   ResponseCompletedData,
+  CommandOutputData,
   TerminalData,
   TerminalExitData,
   ToolData,
@@ -185,11 +186,12 @@ export class EventManager {
 
   // Command events
   sendCommandOutput(baseDir: string, taskId: string, command: string, output: string): void {
-    const data = {
+    const data: CommandOutputData = {
       baseDir,
       taskId,
       command,
       output,
+      timestamp: Date.now(),
     };
     this.sendToWindows('command-output', data);
     this.broadcastToEventConnectors('command-output', data);

src/main/task/context-manager.ts

@@ -87,6 +87,7 @@ export class ContextManager {
         role: roleOrMessage,
         content: content || '',
         usageReport,
+        timestamp: Date.now(),
       } as ContextMessage;
     } else {
       message = roleOrMessage;
@@ -913,6 +914,7 @@ export class ContextManager {
                 diff: response.diff,
                 usageReport: response.usageReport,
                 promptContext: message.promptContext,
+                timestamp: message.timestamp,
               };
               messagesData.push(responseCompletedData);
             });
@@ -965,6 +967,7 @@ export class ContextManager {
                       taskId: this.taskId,
                       usageReport: subMessage.usageReport,
                       promptContext: subMessage.promptContext,
+                      timestamp: subMessage.timestamp,
                     };
                     messagesData.push(responseCompletedData);
                   }
@@ -983,6 +986,7 @@ export class ContextManager {
                         args: subPart.input,
                         usageReport: undefined,
                         promptContext: subMessage.promptContext,
+                        timestamp: subMessage.timestamp,
                       };
                       messagesData.push(toolData);
                     }
@@ -997,6 +1001,7 @@ export class ContextManager {
                     taskId: this.taskId,
                     usageReport: subMessage.usageReport,
                     promptContext: subMessage.promptContext,
+                    timestamp: subMessage.timestamp,
                   };
                   messagesData.push(responseCompletedData);
                 }
@@ -1043,6 +1048,7 @@ export class ContextManager {
               diff: message.diff,
               usageReport: message.usageReport,
               promptContext: message.promptContext,
+              timestamp: message.timestamp,
             };
             messagesData.push(responseCompletedData);
 
@@ -1077,6 +1083,7 @@ export class ContextManager {
                 args: toolCall.input,
                 usageReport: message.usageReport,
                 promptContext: message.promptContext,
+                timestamp: message.timestamp,
               };
               messagesData.push(toolData);
             } else if (part.type === 'tool-result') {
@@ -1116,6 +1123,7 @@ export class ContextManager {
             reflectedMessage: message.reflectedMessage,
             usageReport: message.usageReport,
             promptContext: message.promptContext,
+            timestamp: message.timestamp,
           };
           messagesData.push(responseCompletedData);
         }
@@ -1138,6 +1146,7 @@ export class ContextManager {
           content: content,
           images: images && images.length > 0 ? images : undefined,
           promptContext: message.promptContext,
+          timestamp: message.timestamp,
         };
         messagesData.push(userMessageData);
       } else if (message.role === 'tool' && Array.isArray(message.content)) {

src/main/task/task.ts

@@ -806,6 +806,7 @@ export class Task {
       role: MessageRole.User,
       content: prompt,
       promptContext,
+      timestamp: Date.now(),
     });
 
     // Add user message to context
@@ -851,6 +852,7 @@ export class Task {
       role: MessageRole.User,
       content: prompt,
       promptContext,
+      timestamp: Date.now(),
     });
 
     let messages = this.contextManager.toConnectorMessages();
@@ -1358,6 +1360,7 @@ export class Task {
         usageReport,
         sequenceNumber: message.sequenceNumber,
         promptContext: message.promptContext,
+        timestamp: Date.now(),
       };
 
       const extensionResult = await this.extensionManager.dispatchEvent('onResponseCompleted', { response: data }, this.project, this);
@@ -2121,6 +2124,7 @@ export class Task {
       finished,
       promptContext,
       actionIds,
+      timestamp: Date.now(),
     };
 
     this.eventManager.sendLog(data);
@@ -2562,6 +2566,7 @@ export class Task {
       usageReport,
       promptContext,
       finished,
+      timestamp: Date.now(),
     };
 
     if (response && usageReport && saveToDb) {
@@ -2611,6 +2616,7 @@ export class Task {
       content,
       images,
       promptContext,
+      timestamp: Date.now(),
     };
 
     this.eventManager.sendUserMessage(data);