54 lines
1.5 KiB
Ruby
54 lines
1.5 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
module Jobs
|
|
class StreamTopicAiSummary < ::Jobs::Base
|
|
sidekiq_options retry: false
|
|
|
|
def execute(args)
|
|
return unless topic = Topic.find_by(id: args[:topic_id])
|
|
return unless user = User.find_by(id: args[:user_id])
|
|
|
|
strategy = DiscourseAi::Summarization.topic_summary(topic)
|
|
if strategy.nil? ||
|
|
!Guardian.new(user).can_see_summary?(topic, AiSummary.summary_types[:complete])
|
|
return
|
|
end
|
|
|
|
guardian = Guardian.new(user)
|
|
return unless guardian.can_see?(topic)
|
|
|
|
skip_age_check = !!args[:skip_age_check]
|
|
|
|
streamed_summary = +""
|
|
start = Time.now
|
|
|
|
summary =
|
|
DiscourseAi::TopicSummarization
|
|
.new(strategy, user)
|
|
.summarize(skip_age_check: skip_age_check) do |partial_summary|
|
|
streamed_summary << partial_summary
|
|
|
|
# Throttle updates.
|
|
if (Time.now - start > 0.3) || Rails.env.test?
|
|
payload = { done: false, ai_topic_summary: { summarized_text: streamed_summary } }
|
|
|
|
publish_update(topic, user, payload)
|
|
start = Time.now
|
|
end
|
|
end
|
|
|
|
publish_update(
|
|
topic,
|
|
user,
|
|
AiTopicSummarySerializer.new(summary, { scope: guardian }).as_json.merge(done: true),
|
|
)
|
|
end
|
|
|
|
private
|
|
|
|
def publish_update(topic, user, payload)
|
|
MessageBus.publish("/discourse-ai/summaries/topic/#{topic.id}", payload, user_ids: [user.id])
|
|
end
|
|
end
|
|
end
|