mirror of
https://gh.wpcy.net/https://github.com/discourse/discourse.git
synced 2026-05-21 01:53:41 +08:00
fcb098e85d
## 🔍 Overview This update allows you to regenerate short summaries (gists) with the click of a button. This is helpful for admins to fix short summaries when they go wrong. This update adds a regenerate button in the topic admin controls as well as the bulk select dropdown for topic lists. ## 📸 Screenshots <img width="443" height="663" alt="Screenshot 2025-08-14 at 17 28 46" src="https://github.com/user-attachments/assets/25784d3e-38b4-4eb6-8fde-a921ca9243f6" /> <img width="564" height="579" alt="Screenshot 2025-08-14 at 17 28 36" src="https://github.com/user-attachments/assets/26ef3037-2d79-4a32-bca9-baa94430bf18" /> --------- Co-authored-by: Roman Rizzi <rizziromanalejandro@gmail.com>
81 lines
2.4 KiB
Ruby
Vendored
81 lines
2.4 KiB
Ruby
Vendored
# frozen_string_literal: true
|
|
|
|
module DiscourseAi
|
|
module Summarization
|
|
class SummaryController < ::ApplicationController
|
|
requires_plugin ::DiscourseAi::PLUGIN_NAME
|
|
|
|
def show
|
|
topic = Topic.find(params[:topic_id])
|
|
guardian.ensure_can_see!(topic)
|
|
|
|
raise Discourse::NotFound if !guardian.can_see_summary?(topic)
|
|
|
|
RateLimiter.new(current_user, "summary", 6, 5.minutes).performed! if current_user
|
|
|
|
opts = params.permit(:skip_age_check)
|
|
skip_age_check = opts[:skip_age_check] == "true"
|
|
|
|
summarization_service = DiscourseAi::TopicSummarization.for(topic, current_user)
|
|
|
|
if params[:stream] && current_user
|
|
cached_summary = summarization_service.cached_summary
|
|
|
|
if cached_summary && !skip_age_check
|
|
render_serialized(cached_summary, AiTopicSummarySerializer)
|
|
return
|
|
end
|
|
|
|
Jobs.enqueue(
|
|
:stream_topic_ai_summary,
|
|
topic_id: topic.id,
|
|
user_id: current_user.id,
|
|
skip_age_check: skip_age_check,
|
|
)
|
|
|
|
render json: success_json
|
|
else
|
|
hijack do
|
|
summary = summarization_service.summarize(skip_age_check: skip_age_check)
|
|
render_serialized(summary, AiTopicSummarySerializer)
|
|
end
|
|
end
|
|
end
|
|
|
|
def regen_gist
|
|
raise Discourse::InvalidAccess if !guardian.can_request_gists?
|
|
|
|
topics = []
|
|
|
|
if params[:topic_ids].present?
|
|
topic_ids =
|
|
params[:topic_ids].is_a?(String) ? params[:topic_ids].split(",") : params[:topic_ids]
|
|
topics = Topic.where(id: topic_ids)
|
|
elsif params[:topic_id].present?
|
|
topics = [Topic.find(params[:topic_id])]
|
|
else
|
|
raise Discourse::InvalidParameters.new(:topic_id)
|
|
end
|
|
|
|
if current_user && topics.size >= 1
|
|
RateLimiter.new(current_user, "summary", 6, 5.minutes).performed!
|
|
end
|
|
|
|
if topics.size > TopicQuery::DEFAULT_PER_PAGE_COUNT
|
|
raise Discourse::InvalidParameters.new(:topic_ids)
|
|
end
|
|
|
|
topics.each do |topic|
|
|
guardian.ensure_can_see!(topic)
|
|
|
|
summarizer = DiscourseAi::Summarization.topic_gist(topic)
|
|
summarizer.delete_cached_summaries! if summarizer.present?
|
|
|
|
Jobs.enqueue(:fast_track_topic_gist, topic_id: topic.id, force_regenerate: true)
|
|
end
|
|
|
|
render json: success_json
|
|
end
|
|
end
|
|
end
|
|
end
|
