FEATURE: allow base query to be specified for google tool

This allows to simply scope search results to specific domains

Author: SamSaffron

config/locales/server.en.yml

@@ -205,6 +205,10 @@ en:
       summarizing: "Summarizing topic"
       searching: "Searching for: '%{query}'"
       tool_options:
+        google:
+          base_query:
+            name: "Base Search Query"
+            description: "Base query to use when searching. Example: 'site:example.com' will only include results from example.com."
         read:
           read_private:
             name: "Read Private"

lib/ai_bot/tools/google.rb

@@ -23,15 +23,26 @@ def self.name
           "google"
         end
 
+        def self.accepted_options
+          [option(:base_query, type: :string)]
+        end
+
         def query
           parameters[:query].to_s.strip
         end
 
         def invoke
+          query = self.query
+
           yield(query)
 
           api_key = SiteSetting.ai_google_custom_search_api_key
           cx = SiteSetting.ai_google_custom_search_cx
+
+          if options[:base_query].present?
+            query = "#{options[:base_query]} #{query}"
+          end
+
           escaped_query = CGI.escape(query)
           uri =
             URI(

spec/lib/modules/ai_bot/tools/google_spec.rb

@@ -7,13 +7,14 @@
   let(:progress_blk) { Proc.new {} }
   let(:search) { described_class.new({ query: "some search term" }, bot_user: bot_user, llm: llm) }
 
-  before { SiteSetting.ai_bot_enabled = true }
+  before do
+    SiteSetting.ai_bot_enabled = true
+    SiteSetting.ai_google_custom_search_api_key = "abc"
+    SiteSetting.ai_google_custom_search_cx = "cx"
+  end
 
   describe "#process" do
     it "will not explode if there are no results" do
-      SiteSetting.ai_google_custom_search_api_key = "abc"
-      SiteSetting.ai_google_custom_search_cx = "cx"
-
       json_text = { searchInformation: { totalResults: "0" } }.to_json
 
       stub_request(
@@ -27,10 +28,30 @@
       expect(info).to_not include("oops")
     end
 
-    it "can generate correct info" do
-      SiteSetting.ai_google_custom_search_api_key = "abc"
-      SiteSetting.ai_google_custom_search_cx = "cx"
+    it "supports base_query" do
+      base_query = "site:discourse.org"
+
+      search =
+        described_class.new(
+          { query: "some search term" },
+          bot_user: bot_user,
+          llm: llm,
+          persona_options: {
+            "base_query" => base_query,
+          },
+        )
+
+      json_text = { searchInformation: { totalResults: "0" } }.to_json
 
+      stub_request(
+        :get,
+        "https://www.googleapis.com/customsearch/v1?cx=cx&key=abc&num=10&q=site:discourse.org%20some%20search%20term",
+      ).to_return(status: 200, body: json_text, headers: {})
+
+      search.invoke(&progress_blk)
+    end
+
+    it "can generate correct info" do
       json_text = {
         searchInformation: {
           totalResults: "2",