everything is working now

lib/ai_bot/personas/persona.rb

@@ -201,7 +201,9 @@ def craft_prompt(context, llm: nil)
 
           prompt.max_pixels = self.class.vision_max_pixels if self.class.vision_enabled
           prompt.tools = available_tools.map(&:signature) if available_tools
-          available_tools.each { |tool| tool.inject_prompt(prompt: prompt, context: context) }
+          available_tools.each do |tool|
+            tool.inject_prompt(prompt: prompt, context: context, persona: self)
+          end
           prompt
         end
 

lib/ai_bot/tools/create_artifact.rb

@@ -66,7 +66,8 @@ def self.signature
           }
         end
 
-        def self.inject_prompt(prompt:, context:)
+        def self.inject_prompt(prompt:, context:, persona:)
+          return if persona.options["echo_artifact"] != "true"
           # we inject the current artifact content into the last user message
           if topic_id = context[:topic_id]
             posts = Post.where(topic_id: topic_id)
@@ -138,6 +139,13 @@ def description_args
         def generate_artifact_code(post:, user:)
           prompt = build_artifact_prompt(post: post)
           response = +""
+
+          llm =
+            (
+              options[:creator_llm].present? &&
+                LlmModel.find_by(id: options[:creator_llm].to_i)&.to_llm
+            ) || self.llm
+
           llm.generate(prompt, user: user, feature_name: "create_artifact") do |partial_response|
             response << partial_response
           end

lib/ai_bot/tools/tool.rb

@@ -43,7 +43,7 @@ def allow_partial_tool_calls?
             false
           end
 
-          def inject_prompt(prompt:, context:)
+          def inject_prompt(prompt:, context:, persona:)
           end
         end