FEATURE: silent triage using tools

Author: SamSaffron

config/locales/client.en.yml

@@ -107,6 +107,9 @@ en:
             whisper:
               label: "Reply as Whisper"
               description: "Whether the persona's response should be a whisper"
+            silent_mode:
+              label: "Silent Mode"
+              description: "In silent mode persona will recieve the content but will not post anything on the forum - useful when performing triage using tools"
         llm_triage:
           fields:
             system_prompt:

discourse_automation/llm_persona_triage.rb

@@ -14,13 +14,15 @@
             content: DiscourseAi::Automation.available_persona_choices,
           }
     field :whisper, component: :boolean
+    field :silent_mode, component: :boolean
 
     script do |context, fields|
       post = context["post"]
       next if post&.user&.bot?
 
-      persona_id = fields["persona"]["value"]
-      whisper = fields["whisper"]["value"]
+      persona_id = fields.dig("persona", "value")
+      whisper = !!fields.dig("whisper", "value")
+      silent_mode = !!fields.dig("silent_mode", "value")
 
       begin
         RateLimiter.new(
@@ -42,6 +44,7 @@
           persona_id: persona_id,
           whisper: whisper,
           automation: self.automation,
+          silent_mode: silent_mode,
         )
       rescue => e
         Discourse.warn_exception(

lib/ai_bot/playground.rb

@@ -188,7 +188,8 @@ def self.reply_to_post(
         whisper: nil,
         add_user_to_pm: false,
         stream_reply: false,
-        auto_set_title: false
+        auto_set_title: false,
+        silent_mode: false
       )
         ai_persona = AiPersona.find_by(id: persona_id)
         raise Discourse::InvalidParameters.new(:persona_id) if !ai_persona
@@ -207,7 +208,15 @@ def self.reply_to_post(
           add_user_to_pm: add_user_to_pm,
           stream_reply: stream_reply,
           auto_set_title: auto_set_title,
+          silent_mode: silent_mode,
         )
+      rescue => e
+        if Rails.env.test?
+          p e
+          puts e.backtrace[0..10]
+        else
+          raise e
+        end
       end
 
       def initialize(bot)
@@ -475,13 +484,19 @@ def reply_to(
         add_user_to_pm: true,
         stream_reply: nil,
         auto_set_title: true,
+        silent_mode: false,
         &blk
       )
         # this is a multithreading issue
         # post custom prompt is needed and it may not
         # be properly loaded, ensure it is loaded
         PostCustomPrompt.none
 
+        if silent_mode
+          auto_set_title = false
+          stream_reply = false
+        end
+
         reply = +""
         post_streamer = nil
 
@@ -590,7 +605,7 @@ def reply_to(
             end
           end
 
-        return if reply.blank?
+        return if reply.blank? || silent_mode
 
         if stream_reply
           post_streamer.finish

lib/ai_bot/tool_runner.rb

@@ -89,6 +89,13 @@ def framework_script
               },
             };
           },
+          createChatMessage: function(params) {
+            const result = _discourse_create_chat_message(params);
+            if (result.error) {
+              throw new Error(result.error);
+            }
+            return result;
+          },
         };
 
         const context = #{JSON.generate(@context)};
@@ -345,6 +352,55 @@ def attach_discourse(mini_racer_context)
           end,
         )
 
+        mini_racer_context.attach(
+          "_discourse_create_chat_message",
+          ->(params) do
+            in_attached_function do
+              params = params.symbolize_keys
+              channel_name = params[:channel_name]
+              username = params[:username]
+              message = params[:message]
+
+              # Validate parameters
+              return { error: "Missing required parameter: channel_name" } if channel_name.blank?
+              return { error: "Missing required parameter: username" } if username.blank?
+              return { error: "Missing required parameter: message" } if message.blank?
+
+              # Find the user
+              user = User.find_by(username: username)
+              return { error: "User not found: #{username}" } if user.nil?
+
+              # Find the channel
+              channel = Chat::Channel.find_by(name: channel_name)
+              if channel.nil?
+                # Try finding by slug if not found by name
+                channel = Chat::Channel.find_by(slug: channel_name.parameterize)
+              end
+              return { error: "Channel not found: #{channel_name}" } if channel.nil?
+
+              begin
+                guardian = Guardian.new(user)
+                message =
+                  ChatSDK::Message.create(
+                    raw: message,
+                    channel_id: channel.id,
+                    guardian: guardian,
+                    enforce_membership: !channel.direct_message_channel?,
+                  )
+
+                {
+                  success: true,
+                  message_id: message.id,
+                  message: message.message,
+                  created_at: message.created_at.iso8601,
+                }
+              rescue => e
+                { error: "Failed to create chat message: #{e.message}" }
+              end
+            end
+          end,
+        )
+
         mini_racer_context.attach(
           "_discourse_search",
           ->(params) do

lib/automation/llm_persona_triage.rb

@@ -2,11 +2,12 @@
 module DiscourseAi
   module Automation
     module LlmPersonaTriage
-      def self.handle(post:, persona_id:, whisper: false, automation: nil)
+      def self.handle(post:, persona_id:, whisper: false, silent_mode: false, automation: nil)
         DiscourseAi::AiBot::Playground.reply_to_post(
           post: post,
           persona_id: persona_id,
           whisper: whisper,
+          silent_mode: silent_mode,
         )
       rescue => e
         Discourse.warn_exception(

spec/lib/discourse_automation/llm_persona_triage_spec.rb

@@ -239,4 +239,92 @@ def add_automation_field(name, value, type: "text")
     # should not inject persona into allowed users
     expect(topic.topic_allowed_users.pluck(:user_id).sort).to eq(original_user_ids.sort)
   end
+
+  describe "LLM Persona Triage with Chat Message Creation" do
+    fab!(:user)
+    fab!(:bot_user) { Fabricate(:user) }
+    fab!(:chat_channel) { Fabricate(:category_channel) }
+
+    fab!(:custom_tool) do
+      AiTool.create!(
+        name: "Chat Notifier",
+        tool_name: "chat_notifier",
+        description: "Creates a chat notification in a channel",
+        parameters: [
+          { name: "channel_id", type: "integer", description: "Chat channel ID" },
+          { name: "message", type: "string", description: "Message to post" },
+        ],
+        script: <<~JS,
+        function invoke(params) {
+          // Create a chat message using the Chat API
+          const result = discourse.createChatMessage({
+            channel_name: '#{chat_channel.name}',
+            username: '#{user.username}',
+            message: params.message
+          });
+
+          chain.setCustomRaw("We are done, stopping chaing");
+
+          return {
+            success: true,
+            message_id: result.message_id,
+            url: result.url,
+            message: params.message
+          };
+        }
+      JS
+        summary: "Notify in chat channel",
+        created_by: Discourse.system_user,
+      )
+    end
+
+    before do
+      SiteSetting.chat_enabled = true
+
+      ai_persona.update!(tools: ["custom-#{custom_tool.id}"])
+
+      # Set up automation fields
+      automation.fields.create!(
+        component: "choices",
+        name: "persona",
+        metadata: {
+          value: ai_persona.id,
+        },
+        target: "script",
+      )
+
+      automation.fields.create!(
+        component: "boolean",
+        name: "silent_mode",
+        metadata: {
+          value: true,
+        },
+        target: "script",
+      )
+    end
+
+    it "can silently analyze a post and create a chat notification" do
+      post = Fabricate(:post, raw: "Please help with my billing issue")
+
+      # Tool response from LLM
+      tool_call =
+        DiscourseAi::Completions::ToolCall.new(
+          name: "chat_notifier",
+          parameters: {
+            "message" => "Hello world!",
+          },
+          id: "tool_call_1",
+        )
+
+      DiscourseAi::Completions::Llm.with_prepared_responses([tool_call]) do
+        automation.running_in_background!
+        automation.trigger!({ "post" => post })
+      end
+
+      expect(post.topic.reload.posts.count).to eq(1)
+
+      expect(chat_channel.chat_messages.count).to eq(1)
+      expect(chat_channel.chat_messages.last.message).to eq("Hello world!")
+    end
+  end
 end