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

# frozen_string_literal: true

module Jobs
  class StreamPostHelper < ::Jobs::Base
    sidekiq_options retry: false

    def execute(args)
      return unless post = Post.includes(:topic).find_by(id: args[:post_id])
      return unless user = User.find_by(id: args[:user_id])
      return unless args[:text]
      return unless args[:progress_channel]
      return unless args[:client_id]

      topic = post.topic
      reply_to = post.reply_to_post

      return unless user.guardian.can_see?(post)

      helper_mode = args[:prompt]

      if helper_mode == DiscourseAi::AiHelper::Assistant::EXPLAIN
        input = <<~TEXT.strip
          <term>#{args[:text]}</term>
          <context>#{post.raw}</context>
          <topic>#{topic.title}</topic>
          #{reply_to ? "<replyTo>#{reply_to.raw}</replyTo>" : nil}
        TEXT
      else
        input = args[:text]
      end

      DiscourseAi::AiHelper::Assistant.new.stream_prompt(
        helper_mode,
        input,
        user,
        args[:progress_channel],
        custom_prompt: args[:custom_prompt],
        client_id: args[:client_id],
      )
    end
  end
end