1 files changed, 10 insertions, 0 deletions
diff --git a/app/models/blob_viewer/markup.rb b/app/models/blob_viewer/markup.rb
index f525180048e..37a8e01d0f1 100644
--- a/app/models/blob_viewer/markup.rb
+++ b/app/models/blob_viewer/markup.rb
@@ -9,5 +9,15 @@ module BlobViewer
     self.extensions = Gitlab::MarkupHelper::EXTENSIONS
     self.file_types = %i(readme)
     self.binary = false
+
+    def banzai_render_context
+      {}.tap do |h|
+        h[:rendered] = blob.rendered_markup if blob.respond_to?(:rendered_markup)
+
+        if Feature.enabled?(:cached_markdown_blob, blob.project, default_enabled: true)
+          h[:cache_key] = ['blob', blob.id, 'commit', blob.commit_id]
+        end
+      end
+    end
   end
 end