more tests

Author: SamSaffron

lib/ai_bot/bot_context.rb

@@ -18,7 +18,7 @@ def initialize(
         participants: nil,
         user: nil,
         skip_tool_details: nil,
-        messages:,
+        messages: [],
         custom_instructions: nil
       )
         @participants = participants

lib/ai_bot/tools/read_artifact.rb

@@ -59,7 +59,7 @@ def discourse_artifact?(uri)
         end
 
         def post
-          @post ||= Post.find_by(id: context[:post_id])
+          @post ||= Post.find_by(id: context.post_id)
         end
 
         def handle_discourse_artifact(uri)

lib/ai_helper/assistant.rb

@@ -188,9 +188,10 @@ def generate_image_caption(upload, user)
             messages: [
               {
                 type: :user,
-                content:
+                content: [
                   "Describe this image in a single sentence#{custom_locale_instructions(user)}",
-                upload_ids: [upload.id],
+                  { upload_id: upload.id },
+                ],
               },
             ],
           )

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

@@ -25,9 +25,7 @@
           { url: "#{Discourse.base_url}/discourse-ai/ai-bot/artifacts/#{artifact.id}" },
           bot_user: bot_user,
           llm: llm_model.to_llm,
-          context: {
-            post_id: post2.id,
-          },
+          context: DiscourseAi::AiBot::BotContext.new(post: post),
         )
 
       result = tool.invoke {}
@@ -46,9 +44,7 @@
           { url: "invalid-url" },
           bot_user: bot_user,
           llm: llm_model.to_llm,
-          context: {
-            post_id: post.id,
-          },
+          context: DiscourseAi::AiBot::BotContext.new(post: post),
         )
 
       result = tool.invoke {}
@@ -62,9 +58,7 @@
           { url: "#{Discourse.base_url}/discourse-ai/ai-bot/artifacts/99999" },
           bot_user: bot_user,
           llm: llm_model.to_llm,
-          context: {
-            post_id: post.id,
-          },
+          context: DiscourseAi::AiBot::BotContext.new(post: post),
         )
 
       result = tool.invoke {}
@@ -97,9 +91,7 @@
           { url: "https://example.com" },
           bot_user: bot_user,
           llm: llm_model.to_llm,
-          context: {
-            post_id: post.id,
-          },
+          context: DiscourseAi::AiBot::BotContext.new(post: post),
         )
 
       result = tool.invoke {}
@@ -128,9 +120,7 @@
           { url: "https://example.com" },
           bot_user: bot_user,
           llm: llm_model.to_llm,
-          context: {
-            post_id: post.id,
-          },
+          context: DiscourseAi::AiBot::BotContext.new(post: post),
         )
 
       result = tool.invoke {}