Add current date/time to Telegram bot system prompt

1 files changed, 10 insertions, 0 deletions

diff --git a/Omni/Agent/Telegram.hs b/Omni/Agent/Telegram.hs
index c5cc465..ff161db 100644
--- a/Omni/Agent/Telegram.hs
+++ b/Omni/Agent/Telegram.hs
@@ -59,6 +59,9 @@ import qualified Data.Aeson as Aeson
 import qualified Data.Aeson.KeyMap as KeyMap
 import qualified Data.ByteString.Lazy as BL
 import qualified Data.Text as Text
+import Data.Time (getCurrentTime, utcToLocalTime)
+import Data.Time.Format (defaultTimeLocale, formatTime)
+import Data.Time.LocalTime (getCurrentTimeZone)
 import qualified Network.HTTP.Client as HTTPClient
 import qualified Network.HTTP.Simple as HTTP
 import qualified Omni.Agent.Engine as Engine
@@ -660,8 +663,15 @@ handleAuthorizedMessage tgConfig provider engineCfg msg uid userName chatId = do
   memories <- Memory.recallMemories uid userMessage 5
   let memoryContext = Memory.formatMemoriesForPrompt memories
 
+  now <- getCurrentTime
+  tz <- getCurrentTimeZone
+  let localTime = utcToLocalTime tz now
+      timeStr = Text.pack (formatTime defaultTimeLocale "%A, %B %d, %Y at %H:%M" localTime)
+
   let systemPrompt =
         telegramSystemPrompt
+          <> "\n\n## Current Date and Time\n"
+          <> timeStr
           <> "\n\n## What you know about this user\n"
           <> memoryContext
           <> "\n\n"