Roman RizziandPenar Musaraj
c7acb4a6a0
* DEV: Add summary types * Refactor for different summary types * Use enum for summary types * Update lib/summarization/strategies/topic_summary.rb Co-authored-by: Penar Musaraj <[email protected]> * Update lib/summarization/strategies/topic_gist.rb Co-authored-by: Penar Musaraj <[email protected]> * Update lib/summarization/strategies/chat_messages.rb Co-authored-by: Penar Musaraj <[email protected]> * Fix chat_messages single prompt * Small tweak to the chat summarization prompt --------- Co-authored-by: Penar Musaraj <[email protected]>
49 lines
1.4 KiB
Ruby
49 lines
1.4 KiB
Ruby
# 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)
|
|
|
|
if !guardian.can_see_summary?(topic, AiSummary.summary_types[:complete])
|
|
raise Discourse::NotFound
|
|
end
|
|
|
|
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
|
|
end
|
|
end
|
|
end
|