Make behavior more robust by echoing artifact.

This means that the architect model can provide better instructions.

Author: SamSaffron

lib/ai_bot/personas/persona.rb

@@ -201,7 +201,7 @@ 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) }
           prompt
         end
 

lib/ai_bot/tools/create_artifact.rb

@@ -66,6 +66,41 @@ def self.signature
           }
         end
 
+        def self.inject_prompt(prompt:, context:)
+          # we inject the current artifact content into the last user message
+          if topic_id = context[:topic_id]
+            posts = Post.where(topic_id: topic_id)
+            artifact = AiArtifact.order("id desc").where(post: posts).first
+            if artifact
+              latest_version = artifact.versions.order(version_number: :desc).first
+              current = latest_version || artifact
+
+              artifact_source = <<~MSG
+                Current Artifact:
+
+                ### HTML
+                ```html
+                #{current.html}
+                ```
+
+                ### CSS
+                ```css
+                #{current.css}
+                ```
+
+                ### JavaScript
+                ```javascript
+                #{current.js}
+                ```
+
+              MSG
+
+              last_message = prompt.messages.last
+              last_message[:content] = "#{artifact_source}\n\n#{last_message[:content]}"
+            end
+          end
+        end
+
         def invoke
           yield parameters[:name] || "New Artifact"
 

lib/ai_bot/tools/tool.rb

@@ -42,6 +42,9 @@ def custom_system_message
           def allow_partial_tool_calls?
             false
           end
+
+          def inject_prompt(prompt:, context:)
+          end
         end
 
         attr_accessor :custom_raw, :parameters