FEATURE: allow base 64 encoding results from an http call

Author: SamSaffron

lib/personas/tool_runner.rb

@@ -730,6 +730,7 @@ def attach_http(mini_racer_context)
                 in_attached_function do
                   headers = (options && options["headers"]) || {}
                   body = options && options["body"]
+                  base64_encode = options && options["base64Encode"]
 
                   result = {}
                   DiscourseAi::Personas::Tools::Tool.send_http_request(
@@ -738,7 +739,11 @@ def attach_http(mini_racer_context)
                     headers: headers,
                     body: body,
                   ) do |response|
-                    result[:body] = response.body
+                    if base64_encode
+                      result[:body] = Base64.strict_encode64(response.body)
+                    else
+                      result[:body] = response.body
+                    end
                     result[:status] = response.code.to_i
                   end
 

spec/models/ai_tool_spec.rb

@@ -43,6 +43,38 @@ def create_tool(
     expect(runner.invoke).to eq("query" => "test")
   end
 
+  it "can base64 encode binary HTTP responses" do
+    # Create binary data with all possible byte values (0-255)
+    binary_data = (0..255).map(&:chr).join
+    expected_base64 = Base64.strict_encode64(binary_data)
+
+    script = <<~JS
+      function invoke(params) {
+        const result = http.post("https://example.com/binary", {
+          body: "test",
+          base64Encode: true
+        });
+        return result.body;
+      }
+    JS
+
+    tool = create_tool(script: script)
+    runner = tool.runner({}, llm: nil, bot_user: nil)
+
+    stub_request(:post, "https://example.com/binary").to_return(
+      status: 200,
+      body: binary_data,
+      headers: {
+      },
+    )
+
+    result = runner.invoke
+
+    expect(result).to eq(expected_base64)
+    # Verify we can decode back to original binary data
+    expect(Base64.strict_decode64(result).bytes).to eq((0..255).to_a)
+  end
+
   it "can perform HTTP requests with various verbs" do
     %i[post put delete patch].each do |verb|
       script = <<~JS