Add ISO 8601 timestamps to conversation context messages

1 files changed, 5 insertions, 3 deletions

diff --git a/Omni/Agent/Memory.hs b/Omni/Agent/Memory.hs
index 0a050b7..9ca2d99 100644
--- a/Omni/Agent/Memory.hs
+++ b/Omni/Agent/Memory.hs
@@ -90,6 +90,7 @@ import qualified Data.List as List
 import qualified Data.Text as Text
 import qualified Data.Text.Encoding as TE
 import Data.Time (UTCTime, getCurrentTime)
+import Data.Time.Format (defaultTimeLocale, formatTime)
 import qualified Data.UUID as UUID
 import qualified Data.UUID.V4 as UUID
 import qualified Data.Vector.Storable as VS
@@ -1415,9 +1416,10 @@ getConversationContext uid chatId maxTokens = do
           | otherwise = (acc, sum (map cmTokensEstimate acc))
 
     formatMsg m =
-      let prefix = case cmRole m of
-            UserRole -> fromMaybe "User" (cmSenderName m) <> ": "
-            AssistantRole -> "Assistant: "
+      let timestamp = Text.pack (formatTime defaultTimeLocale "%Y-%m-%dT%H:%M:%SZ" (cmCreatedAt m))
+          prefix = case cmRole m of
+            UserRole -> "[" <> timestamp <> "] " <> fromMaybe "User" (cmSenderName m) <> ": "
+            AssistantRole -> "[" <> timestamp <> "] Assistant: "
        in prefix <> cmContent m
 
 -- | Summarize old messages and archive them.