WIP: allow triage using persona from triage

Author: SamSaffron

config/locales/client.en.yml

@@ -146,6 +146,12 @@ en:
             include_personal_messages:
               label: "Include personal messages"
               description: "Also scan and triage personal messages"
+            whisper:
+              label: "Reply as Whisper"
+              description: "Whether the AI's response should be a whisper"
+            reply_persona:
+              label: "Reply Persona"
+              description: "AI Persona to use for replies (must have default LLM), will be prioritized over canned reply"
             model:
               label: "Model"
               description: "Language model used for triage"

discourse_automation/llm_triage.rb

@@ -9,36 +9,58 @@
 
     triggerables %i[post_created_edited]
 
-    field :system_prompt, component: :message, required: false
-    field :search_for_text, component: :text, required: true
-    field :max_post_tokens, component: :text
-    field :stop_sequences, component: :text_list, required: false
+    # TODO move to triggerables
+    field :include_personal_messages, component: :boolean
+
+    # Inputs
     field :model,
           component: :choices,
           required: true,
           extra: {
             content: DiscourseAi::Automation.available_models,
           }
+    field :system_prompt, component: :message, required: false
+    field :search_for_text, component: :text, required: true
+    field :max_post_tokens, component: :text
+    field :stop_sequences, component: :text_list, required: false
+    field :temperature, component: :text
+
+    # Actions
     field :category, component: :category
     field :tags, component: :tags
     field :hide_topic, component: :boolean
     field :flag_post, component: :boolean
-    field :include_personal_messages, component: :boolean
-    field :temperature, component: :text
     field :flag_type,
           component: :choices,
           required: false,
           extra: {
             content: DiscourseAi::Automation.flag_types,
           },
           default: "review"
-    field :canned_reply, component: :message
     field :canned_reply_user, component: :user
+    field :canned_reply, component: :message
+    field :reply_persona,
+          component: :choices,
+          extra: {
+            content: DiscourseAi::Automation.available_persona_choices,
+          }
+    field :whisper, component: :boolean
 
     script do |context, fields|
       post = context["post"]
+
+      post = context["post"]
+      next if post&.user&.bot?
+
+      if post.topic.private_message?
+        include_personal_messages = fields.dig("include_personal_messages", "value")
+        next if !include_personal_messages
+      end
+
       canned_reply = fields.dig("canned_reply", "value")
       canned_reply_user = fields.dig("canned_reply_user", "value")
+      reply_persona_id = fields["reply_persona"]["value"]
+      whisper = fields["whisper"]["value"]
 
       # nothing to do if we already replied
       next if post.user.username == canned_reply_user
@@ -65,11 +87,6 @@
 
       stop_sequences = fields.dig("stop_sequences", "value")
 
-      if post.topic.private_message?
-        include_personal_messages = fields.dig("include_personal_messages", "value")
-        next if !include_personal_messages
-      end
-
       begin
         RateLimiter.new(
           Discourse.system_user,
@@ -94,6 +111,8 @@
           tags: tags,
           canned_reply: canned_reply,
           canned_reply_user: canned_reply_user,
+          reply_persona_id: reply_persona_id,
+          whisper: whisper,
           hide_topic: hide_topic,
           flag_post: flag_post,
           flag_type: flag_type.to_s.to_sym,

lib/automation/llm_triage.rb

@@ -18,7 +18,9 @@ def self.handle(
         automation: nil,
         max_post_tokens: nil,
         stop_sequences: nil,
-        temperature: nil
+        temperature: nil,
+        whisper: nil,
+        reply_persona_id: nil
       )
         if category_id.blank? && tags.blank? && canned_reply.blank? && hide_topic.blank? &&
              flag_post.blank?
@@ -55,13 +57,29 @@ def self.handle(
         if result.present? && result.downcase.include?(search_for_text.downcase)
           user = User.find_by_username(canned_reply_user) if canned_reply_user.present?
           user = user || Discourse.system_user
-          if canned_reply.present?
+          if reply_persona_id.present?
+            ai_persona = AiPersona.find_by(id: persona_id)
+            if ai_persona.present?
+              persona_class = ai_persona.class_instance
+              persona = persona_class.new
+
+              bot_user = ai_persona.user
+              if bot_user.nil?
+                bot = DiscourseAi::AiBot::Bot.as(bot_user, persona: persona)
+                playground = DiscourseAi::AiBot::Playground.new(bot)
+
+                playground.reply_to(post, whisper: whisper, context_style: :topic)
+              end
+            end
+          elsif canned_reply.present?
+            post_type = whisper ? Post.types[:whisper] : Post.types[:regular]
             PostCreator.create!(
               user,
               topic_id: post.topic_id,
               raw: canned_reply,
               reply_to_post_number: post.post_number,
               skip_validations: true,
+              post_type: post_type,
             )
           end