From 2fc07b4bcacf116bda787e06d122af284c1e37a9 Mon Sep 17 00:00:00 2001
From: William Valentin <william.valentin.info@gmail.com>
Date: Thu, 5 Feb 2026 15:51:32 -0800
Subject: [PATCH] feat: add timestamps to session messages

---
 src/session/manager.ts | 11 ++++++++---
 1 file changed, 8 insertions(+), 3 deletions(-)

diff --git a/src/session/manager.ts b/src/session/manager.ts
index e50d161..f54e935 100644
--- a/src/session/manager.ts
+++ b/src/session/manager.ts
@@ -15,9 +15,14 @@ export class ManagedSession implements Session {
     private history: Message[] = []
   ) {}
 
-  addMessage(message: Message): void {
-    this.history.push(message);
-    this.store.addMessage(this.id, message);
+  addMessage(message: Message): Message {
+    const messageWithTimestamp: Message = {
+      ...message,
+      timestamp: Date.now(),
+    };
+    this.history.push(messageWithTimestamp);
+    this.store.addMessage(this.id, messageWithTimestamp);
+    return messageWithTimestamp;
   }
 
   getHistory(): Message[] {