Add basic search mode

Author: xfalcox

app/controllers/discourse_ai/discord/bot_controller.rb

@@ -25,11 +25,10 @@ def interactions
           unless SiteSetting.ai_discord_allowed_guilds_map.include?(interaction.guild_id)
             return head :forbidden
           end
+
           response = { type: 5, data: { content: "Searching..." } }
           hijack { render json: response }
 
-          Rails.logger.info("Discord interaction received: #{body}")
-
           # Respond to Discord command
           Jobs.enqueue(:stream_discord_reply, interaction: body.dup)
         end

app/jobs/regular/stream_discord_reply.rb

@@ -7,7 +7,11 @@ class StreamDiscordReply < ::Jobs::Base
     def execute(args)
       interaction = args[:interaction]
 
-      DiscourseAi::Discord::Bot::PersonaReplier.new(interaction).handle_interaction!
+      if SiteSetting.ai_discord_search_mode == "persona"
+        DiscourseAi::Discord::Bot::PersonaReplier.new(interaction).handle_interaction!
+      else
+        DiscourseAi::Discord::Bot::Search.new(interaction).handle_interaction!
+      end
     end
   end
 end

lib/discord/bot/base.rb

@@ -23,8 +23,6 @@ def create_reply(reply)
             { "Content-Type" => "application/json" },
           )
         @reply_response = JSON.parse(response.body, symbolize_names: true)
-        Rails.logger.info("Discord reply created: #{@reply_response}")
-        @reply_response
       end
 
       def update_reply(reply)
@@ -38,8 +36,6 @@ def update_reply(reply)
             { "Content-Type" => "application/json" },
           )
         @last_update_response = JSON.parse(response.body, symbolize_names: true)
-        Rails.logger.info("Discord reply updated: #{@last_update_response}")
-        @last_update_response
       end
     end
   end

lib/discord/bot/persona_replier.rb

@@ -29,11 +29,11 @@ def handle_interaction!
             next if reply.blank?
 
             if @reply_response.nil?
-              create_reply(reply.dup)
+              create_reply(wrap_links(reply.dup))
             elsif @last_update_response.nil?
-              update_reply(reply.dup)
+              update_reply(wrap_links(reply.dup))
             elsif Time.now - last_update_sent_at > 1
-              update_reply(reply.dup)
+              update_reply(wrap_links(reply.dup))
               last_update_sent_at = Time.now
             end
           end

lib/discord/bot/search.rb

@@ -0,0 +1,37 @@
+# frozen_string_literal: true
+
+module DiscourseAi
+  module Discord::Bot
+    class Search < Base
+      def initialize(body)
+        @search = DiscourseAi::AiBot::Tools::Search
+        super(body)
+      end
+
+      def handle_interaction!
+        results =
+          @search.new(
+            { search_query: @query },
+            persona_options: {
+              "max_results" => 10,
+            },
+            bot_user: nil,
+            llm: nil,
+          ).invoke(&Proc.new {})
+
+        formatted_results = results[:rows].map.with_index { |result, index| <<~RESULT }.join("\n")
+          #{index + 1}. [#{result[0]}](<#{Discourse.base_url}#{result[1]}>)
+          #{result[3].truncate(50)}
+          RESULT
+
+        reply = <<~REPLY
+          Here are the top search results for your query:
+
+          #{formatted_results}
+        REPLY
+
+        create_reply(reply)
+      end
+    end
+  end
+end