diff options
| author | Lucas Rolff <lucas@lucasrolff.com> | 2025-06-03 09:24:29 +0200 |
|---|---|---|
| committer | Lucas Rolff <lucas@lucasrolff.com> | 2025-06-03 09:24:29 +0200 |
| commit | deb509430e84201fce4498df4a03008f780de5a6 (patch) | |
| tree | 6c13b11a77c1b0546f99a2621a2445ba727a9859 | |
| parent | c93385967c2410fe62ef8d34ab7ddeea23a9f843 (diff) | |
| download | rspamd-deb509430e84201fce4498df4a03008f780de5a6.tar.gz rspamd-deb509430e84201fce4498df4a03008f780de5a6.zip | |
[Minor] Log the token usage for OpenAI (compatible) plain conversion
| -rw-r--r-- | src/plugins/lua/gpt.lua | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/src/plugins/lua/gpt.lua b/src/plugins/lua/gpt.lua index bdec927e6..51df0bc20 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 |