43 lines
1.1 KiB
Ruby
43 lines
1.1 KiB
Ruby
# 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]
|
|
|
|
topic = post.topic
|
|
reply_to = post.reply_to_post
|
|
|
|
return unless user.guardian.can_see?(post)
|
|
|
|
prompt = CompletionPrompt.enabled_by_name(args[:prompt])
|
|
|
|
if prompt.id == CompletionPrompt::CUSTOM_PROMPT
|
|
prompt.custom_instruction = args[:custom_prompt]
|
|
end
|
|
|
|
if prompt.name == "explain"
|
|
input = <<~TEXT
|
|
<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(
|
|
prompt,
|
|
input,
|
|
user,
|
|
"/discourse-ai/ai-helper/stream_suggestion/#{post.id}",
|
|
)
|
|
end
|
|
end
|
|
end
|