diff options
| -rw-r--r-- | Omni/Agent/Telegram.hs | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/Omni/Agent/Telegram.hs b/Omni/Agent/Telegram.hs index 993f2e0..148bb6a 100644 --- a/Omni/Agent/Telegram.hs +++ b/Omni/Agent/Telegram.hs @@ -93,6 +93,7 @@ import qualified Omni.Agent.Tools.WebReader as WebReader import qualified Omni.Agent.Tools.WebSearch as WebSearch import qualified Omni.Test as Test import System.Environment (lookupEnv) +import Text.Printf (printf) defaultTelegramConfig :: Text -> [Int] -> Maybe Text -> Text -> Types.TelegramConfig defaultTelegramConfig = Types.defaultTelegramConfig @@ -914,11 +915,13 @@ processEngagedMessage tgConfig provider engineCfg msg uid userName chatId userMe else do _ <- Messages.enqueueImmediate (Just uid) chatId threadId response (Just "agent_response") Nothing unless isGroup <| checkAndSummarize (Types.tgOpenRouterApiKey tgConfig) uid chatId + let cost = Engine.resultTotalCost agentResult + costStr = Text.pack (printf "%.2f" cost) putText <| "Responded to " <> userName <> " (cost: " - <> tshow (Engine.resultTotalCost agentResult) + <> costStr <> " cents)" maxConversationTokens :: Int |