]> git.ipfire.org Git - thirdparty/rspamd.git/commitdiff
[Minor] Log the token usage for OpenAI (compatible) plain conversion 5488/head
authorLucas Rolff <lucas@lucasrolff.com>
Tue, 3 Jun 2025 07:24:29 +0000 (09:24 +0200)
committerLucas Rolff <lucas@lucasrolff.com>
Tue, 3 Jun 2025 07:24:29 +0000 (09:24 +0200)
src/plugins/lua/gpt.lua

index bdec927e6211c6f34386641546339c08763ecd54..51df0bc20cc337eb43e03b29a28933d74a526d71 100644 (file)
@@ -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