From deb509430e84201fce4498df4a03008f780de5a6 Mon Sep 17 00:00:00 2001
From: Lucas Rolff <lucas@lucasrolff.com>
Date: Tue, 3 Jun 2025 09:24:29 +0200
Subject: [PATCH] [Minor] Log the token usage for OpenAI (compatible) plain
 conversion

---
 src/plugins/lua/gpt.lua | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/src/plugins/lua/gpt.lua b/src/plugins/lua/gpt.lua
index bdec927e62..51df0bc20c 100644
--- a/src/plugins/lua/gpt.lua
+++ b/src/plugins/lua/gpt.lua
@@ -355,6 +355,10 @@ local function default_openai_plain_conversion(task, input)
   local reason = clean_reply_line(lines[2])
   local categories = lua_util.str_split(clean_reply_line(lines[3]), ',')
 
+  if type(reply.usage) == 'table' then
+    rspamd_logger.infox(task, 'usage: %s tokens', reply.usage.total_tokens)
+  end
+
   if spam_score then
     return spam_score, reason, categories
   end
-- 
2.47.3