From: hunter-nl <junobox@gmail.com>
Date: Thu, 14 Aug 2025 10:36:16 +0000 (+0200)
Subject: Update gpt.lua to support newer models with max_completion_tokens
X-Git-Tag: 3.13.0~29^2~9
X-Git-Url: http://git.ipfire.org/gitweb.cgi?a=commitdiff_plain;h=d09b5d24fdaff9dff842fb994c2783ddcc7db4dd;p=thirdparty%2Frspamd.git

Update gpt.lua to support newer models with max_completion_tokens

Newer and reasoning models requires max_completion_tokens instead of max_tokens attribute.
---

diff --git a/src/plugins/lua/gpt.lua b/src/plugins/lua/gpt.lua
index 331dbbce22..968127a287 100644
--- a/src/plugins/lua/gpt.lua
+++ b/src/plugins/lua/gpt.lua
@@ -107,7 +107,7 @@ local categories_map = {}
 local settings = {
   type = 'openai',
   api_key = nil,
-  model = 'gpt-4o-mini',
+  model = 'gpt-5-mini',
   max_tokens = 1000,
   temperature = 0.0,
   timeout = 10,
@@ -683,9 +683,27 @@ local function openai_check(task, content, sel_part)
 
   local from_content, url_content = get_meta_llm_content(task)
 
+  -- Decide which token length field to use for the given model
+  local function get_max_tokens_field(model)
+    if not model then
+      return 'max_tokens' -- default
+    end
+
+    -- Newer models that require max_completion_tokens
+    if model:match('^gpt%-5') or
+       model:match('^o%d') or
+       model:match('^o%d%-mini') or
+       model:match('^gpt%-4%.1') or
+       model:match('reasoning') then
+      return 'max_completion_tokens'
+    end
+
+    -- Default for older/legacy models
+    return 'max_tokens'
+  end
+
   local body = {
     model = settings.model,
-    max_tokens = settings.max_tokens,
     temperature = settings.temperature,
     messages = {
       {
@@ -711,6 +729,10 @@ local function openai_check(task, content, sel_part)
     }
   }
 
+  -- Set the correct token limit field
+  local token_field = get_max_tokens_field(settings.model)
+  body[token_field] = settings.max_tokens
+    
   -- Conditionally add response_format
   if settings.include_response_format then
     body.response_format = { type = "json_object" }