]> git.ipfire.org Git - thirdparty/rspamd.git/commitdiff
[Minor] Insert Reason to result to show gpt answer in history
authoralex081512 <alex@rbbaader.de>
Thu, 13 Mar 2025 18:35:52 +0000 (19:35 +0100)
committerGitHub <noreply@github.com>
Thu, 13 Mar 2025 18:35:52 +0000 (19:35 +0100)
src/plugins/lua/gpt.lua

index d4e77f34c63db9b57791b9ac968adc7e812bb5b3..4d5311305a558fd5d63a3486fcae9740174ed5a5 100644 (file)
@@ -494,8 +494,16 @@ local function insert_results(task, result, sel_part)
     rspamd_logger.errx(task, 'no probability in result')
     return
   end
+    
+  local resultText
+  if result.reason then
+    resultText = tostring(result.probability * 100) .. '% - ' .. result.reason
+  else
+    resultText = tostring(result.probability * 100)
+  end
+    
   if result.probability > 0.5 then
-    task:insert_result('GPT_SPAM', (result.probability - 0.5) * 2, tostring(result.probability))
+    task:insert_result('GPT_SPAM', (result.probability - 0.5) * 2, resultText)
     if settings.autolearn then
       task:set_flag("learn_spam")
     end
@@ -504,7 +512,7 @@ local function insert_results(task, result, sel_part)
       process_categories(task, result.categories)
     end
   else
-    task:insert_result('GPT_HAM', (0.5 - result.probability) * 2, tostring(result.probability))
+    task:insert_result('GPT_HAM', (0.5 - result.probability) * 2, resultText)
     if settings.autolearn then
       task:set_flag("learn_ham")
     end