[Minor] Add from and subject to the prompt

1 files changed, 9 insertions, 0 deletions

diff --git a/src/plugins/lua/gpt.lua b/src/plugins/lua/gpt.lua
index e12bcf807..a2db2123e 100644
--- a/src/plugins/lua/gpt.lua
+++ b/src/plugins/lua/gpt.lua
@@ -15,6 +15,7 @@ limitations under the License.
 ]] --
 
 local N = "gpt"
+local E = {}
 
 if confighelp then
   rspamd_config:add_example(nil, 'gpt',
@@ -247,6 +248,14 @@ local function openai_gpt_check(task)
       },
       {
         role = 'user',
+        content = 'Subject: ' .. task:get_subject() or '',
+      },
+      {
+        role = 'user',
+        content = 'From: ' .. ((task:get_from('mime') or E)[1] or E).name or '',
+      },
+      {
+        role = 'user',
         content = content
       }
     }