Wire up writeAvaLog to actually write audit logs

- Add AuditLog import to Telegram.hs
- Call writeAvaLog for user messages in both handleEngagement paths
- Call writeAvaLog for assistant responses
- Add logs directories to Ava.nix tmpfiles rules

1 files changed, 36 insertions, 0 deletions

diff --git a/Omni/Agent/Telegram.hs b/Omni/Agent/Telegram.hs
index de84e49..0ebf4ca 100644
--- a/Omni/Agent/Telegram.hs
+++ b/Omni/Agent/Telegram.hs
@@ -81,6 +81,7 @@ 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.AuditLog as AuditLog
 import qualified Omni.Agent.Engine as Engine
 import qualified Omni.Agent.Memory as Memory
 import qualified Omni.Agent.Paths as Paths
@@ -870,6 +871,18 @@ handleAuthorizedMessageContinued tgConfig provider engineCfg msg uid userName ch
       now <- getCurrentTime
       _ <- forkIO <| Memory.saveChatHistoryEntry chatId (Just uid) "user" (Just userName) userMessage now
 
+      _ <-
+        forkIO <| do
+          entry <-
+            AuditLog.mkLogEntry
+              (AuditLog.SessionId (tshow chatId))
+              (AuditLog.AgentId "ava")
+              (Just userName)
+              AuditLog.UserMessage
+              (Aeson.String userMessage)
+              AuditLog.emptyMetadata
+          AuditLog.writeAvaLog entry
+
       processEngagedMessage tgConfig provider engineCfg msg uid userName chatId userMessage conversationContext
 
 handleAuthorizedMessageBatch ::
@@ -1005,6 +1018,18 @@ handleAuthorizedMessageBatch tgConfig provider engineCfg msg uid userName chatId
       now <- getCurrentTime
       _ <- forkIO <| Memory.saveChatHistoryEntry chatId (Just uid) "user" (Just userName) userMessage now
 
+      _ <-
+        forkIO <| do
+          entry <-
+            AuditLog.mkLogEntry
+              (AuditLog.SessionId (tshow chatId))
+              (AuditLog.AgentId "ava")
+              (Just userName)
+              AuditLog.UserMessage
+              (Aeson.String userMessage)
+              AuditLog.emptyMetadata
+          AuditLog.writeAvaLog entry
+
       processEngagedMessage tgConfig provider engineCfg msg uid userName chatId userMessage conversationContext
 
 processEngagedMessage ::
@@ -1221,6 +1246,17 @@ processEngagedMessage tgConfig provider engineCfg msg uid userName chatId userMe
       unless (Text.null response) <| do
         nowResp <- getCurrentTime
         _ <- forkIO <| Memory.saveChatHistoryEntry chatId (Just uid) "assistant" (Just "Ava") response nowResp
+        _ <-
+          forkIO <| do
+            entry <-
+              AuditLog.mkLogEntry
+                (AuditLog.SessionId (tshow chatId))
+                (AuditLog.AgentId "ava")
+                Nothing
+                AuditLog.AssistantMessage
+                (Aeson.String response)
+                AuditLog.emptyMetadata
+            AuditLog.writeAvaLog entry
         pure ()
 
       if Text.null response