discourse-ai/app/jobs/regular/stream_topic_ai_summary.rb

# 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