storage api in update strategy

SamSaffron · SamSaffron · commit 5d95ed0d7ef6 · 2025-06-09T13:13:58.000+10:00
fix incorrect drift in artifact controller
diff --git a/app/controllers/discourse_ai/ai_bot/artifacts_controller.rb b/app/controllers/discourse_ai/ai_bot/artifacts_controller.rb
@@ -20,14 +20,15 @@ def show
         end
 
         name = artifact.name
+        artifact_version = nil
 
         if params[:version].present?
-          artifact = artifact.versions.find_by(version_number: params[:version])
-          raise Discourse::NotFound if !artifact
+          artifact_version = artifact.versions.find_by(version_number: params[:version])
+          raise Discourse::NotFound if !artifact_version
         end
 
-        untrusted_html = build_untrusted_html(artifact, name)
-        trusted_html = build_trusted_html(artifact, name, untrusted_html)
+        untrusted_html = build_untrusted_html(artifact_version || artifact, name)
+        trusted_html = build_trusted_html(artifact, artifact_version, name, untrusted_html)
 
         set_security_headers
         render html: trusted_html.html_safe, layout: false, content_type: "text/html"
@@ -57,7 +58,7 @@ def build_untrusted_html(artifact, name)
         HTML
       end
 
-      def build_trusted_html(artifact, name, untrusted_html)
+      def build_trusted_html(artifact, artifact_version, name, untrusted_html)
         <<~HTML
           <!DOCTYPE html>
           <html>
diff --git a/lib/personas/artifact_update_strategies/base.rb b/lib/personas/artifact_update_strategies/base.rb
@@ -31,6 +31,12 @@ def apply(&progress)
           apply_changes(parsed_changes)
         end
 
+        def storage_api
+          if @artifact.metadata.is_a?(Hash) && @artifact.metadata["requires_storage"]
+            DiscourseAi::Personas::Tools::CreateArtifact.storage_api
+          end
+        end
+
         private
 
         def generate_changes(&progress)
diff --git a/lib/personas/artifact_update_strategies/diff.rb b/lib/personas/artifact_update_strategies/diff.rb
@@ -186,6 +186,8 @@ def system_prompt
             JavaScript libraries must be sourced from the following CDNs, otherwise CSP will reject it:
             #{AiArtifact::ALLOWED_CDN_SOURCES.join("\n")}
 
+            #{storage_api}
+
             Reply Format:
             [HTML]
             (changes or empty if no changes or entire HTML)
diff --git a/lib/personas/artifact_update_strategies/full.rb b/lib/personas/artifact_update_strategies/full.rb
@@ -76,6 +76,8 @@ def system_prompt
             JavaScript libraries must be sourced from the following CDNs, otherwise CSP will reject it:
             #{AiArtifact::ALLOWED_CDN_SOURCES.join("\n")}
 
+            #{storage_api}
+
             Always adhere to the format when replying:
 
             [HTML]
