update artifact fixed

Author: SamSaffron

lib/ai_bot/tools/update_artifact.rb

@@ -39,7 +39,7 @@ def self.signature
         def self.inject_prompt(prompt:, context:, persona:)
           return if persona.options["do_not_echo_artifact"].to_s == "true"
           # we inject the current artifact content into the last user message
-          if topic_id = context[:topic_id]
+          if topic_id = context.topic_id
             posts = Post.where(topic_id: topic_id)
             artifact = AiArtifact.order("id desc").where(post: posts).first
             if artifact
@@ -113,7 +113,7 @@ def in_progress(instructions:, source: nil)
         end
 
         def invoke
-          post = Post.find_by(id: context[:post_id])
+          post = Post.find_by(id: context.post_id)
           return error_response("No post context found") unless post
 
           artifact = AiArtifact.find_by(id: parameters[:artifact_id])

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

@@ -47,9 +47,7 @@
             persona_options: {
               "update_algorithm" => "full",
             },
-            context: {
-              post_id: post.id,
-            },
+            context: DiscourseAi::AiBot::BotContext.new(messages: [], post: post),
           )
 
         result = tool.invoke {}
@@ -93,9 +91,7 @@
             persona_options: {
               "update_algorithm" => "full",
             },
-            context: {
-              post_id: post.id,
-            },
+            context: DiscourseAi::AiBot::BotContext.new(messages: [], post: post),
           )
 
         result = tool.invoke {}
@@ -119,9 +115,7 @@
             { artifact_id: artifact.id, instructions: "Invalid update" },
             bot_user: bot_user,
             llm: llm_model.to_llm,
-            context: {
-              post_id: post.id,
-            },
+            context: DiscourseAi::AiBot::BotContext.new(messages: [], post: post),
           )
 
         result = tool.invoke {}
@@ -135,9 +129,7 @@
           { artifact_id: -1, instructions: "Update something" },
           bot_user: bot_user,
           llm: llm_model.to_llm,
-          context: {
-            post_id: post.id,
-          },
+          context: DiscourseAi::AiBot::BotContext.new(messages: [], post: post),
         )
 
       result = tool.invoke {}
@@ -163,9 +155,7 @@
             persona_options: {
               "update_algorithm" => "full",
             },
-            context: {
-              post_id: post.id,
-            },
+            context: DiscourseAi::AiBot::BotContext.new(messages: [], post: post),
           )
 
         tool.invoke {}
@@ -196,9 +186,7 @@
             persona_options: {
               "update_algorithm" => "full",
             },
-            context: {
-              post_id: post.id,
-            },
+            context: DiscourseAi::AiBot::BotContext.new(messages: [], post: post),
           )
           .invoke {}
       end
@@ -224,9 +212,7 @@
             persona_options: {
               "update_algorithm" => "full",
             },
-            context: {
-              post_id: post.id,
-            },
+            context: DiscourseAi::AiBot::BotContext.new(messages: [], post: post),
           )
 
         result = tool.invoke {}
@@ -276,9 +262,7 @@
           { artifact_id: artifact.id, instructions: "Change the text to Updated and color to red" },
           bot_user: bot_user,
           llm: llm_model.to_llm,
-          context: {
-            post_id: post.id,
-          },
+          context: DiscourseAi::AiBot::BotContext.new(messages: [], post: post),
           persona_options: {
             "update_algorithm" => "diff",
           },
@@ -346,9 +330,7 @@
           { artifact_id: artifact.id, instructions: "Change the text to Updated and color to red" },
           bot_user: bot_user,
           llm: llm_model.to_llm,
-          context: {
-            post_id: post.id,
-          },
+          context: DiscourseAi::AiBot::BotContext.new(messages: [], post: post),
           persona_options: {
             "update_algorithm" => "diff",
           },