fix specs

Author: SamSaffron

lib/ai_bot/tools/tool.rb

@@ -56,14 +56,17 @@ def initialize(
           persona_options: {},
           bot_user:,
           llm:,
-          context: {}
+          context: nil
         )
           @parameters = parameters
           @tool_call_id = tool_call_id
           @persona_options = persona_options
           @bot_user = bot_user
           @llm = llm
-          @context = context
+          @context = context.nil? ? DiscourseAi::AiBot::BotContext.new(messages: []) : context
+          if !@context.is_a?(DiscourseAi::AiBot::BotContext)
+            raise ArgumentError, "context must be a DiscourseAi::AiBot::Context"
+          end
         end
 
         def name

lib/automation/llm_triage.rb

@@ -42,7 +42,12 @@ def self.handle(
 
         content = llm.tokenizer.truncate(content, max_post_tokens) if max_post_tokens.present?
 
-        prompt.push(type: :user, content: content, upload_ids: post.upload_ids)
+        if post.upload_ids.present?
+          content = [content]
+          content.concat(post.upload_ids.map { |upload_id| { upload_id: upload_id } })
+        end
+
+        prompt.push(type: :user, content: content)
 
         result = nil
 

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

@@ -15,9 +15,7 @@
   let(:llm) { DiscourseAi::Completions::Llm.proxy("custom:#{gpt_35_turbo.id}") }
   let(:progress_blk) { Proc.new {} }
 
-  let(:dall_e) do
-    described_class.new({ prompts: prompts }, llm: llm, bot_user: bot_user, context: {})
-  end
+  let(:dall_e) { described_class.new({ prompts: prompts }, llm: llm, bot_user: bot_user) }
 
   let(:base64_image) do
     "iVBORw0KGgoAAAANSUhEUgAAAAEAAAABCAYAAAAfFcSJAAAADUlEQVR42mP8z8BQDwAEhQGAhKmMIQAAAABJRU5ErkJggg=="
@@ -30,8 +28,6 @@
           { prompts: ["a cat"], aspect_ratio: "tall" },
           llm: llm,
           bot_user: bot_user,
-          context: {
-          },
         )
 
       data = [{ b64_json: base64_image, revised_prompt: "a tall cat" }]

spec/lib/modules/automation/llm_triage_spec.rb

@@ -199,7 +199,7 @@ def triage(**args)
 
       triage_prompt = DiscourseAi::Completions::Llm.prompts.last
 
-      expect(triage_prompt.messages.last[:upload_ids]).to contain_exactly(post_upload.id)
+      expect(triage_prompt.messages.last[:content].last).to eq({ upload_id: post_upload.id })
     end
   end