72 lines
1.8 KiB
Ruby
72 lines
1.8 KiB
Ruby
|
# frozen_string_literal: true
|
||
|
|
|
||
|
|
module DiscourseAi
|
||
|
|
module Summarization
|
||
|
|
class SummaryGenerator
|
||
|
|
def initialize(target)
|
||
|
|
@target = target
|
||
|
|
end
|
||
|
|
|
||
|
|
def summarize!(content_since)
|
||
|
|
content = get_content(content_since)
|
||
|
|
|
||
|
|
send("#{summarization_provider}_summarization", content)
|
||
|
|
end
|
||
|
|
|
||
|
|
private
|
||
|
|
|
||
|
|
attr_reader :target
|
||
|
|
|
||
|
|
def summarization_provider
|
||
|
|
model.starts_with?("gpt") ? "openai" : "discourse"
|
||
|
|
end
|
||
|
|
|
||
|
|
def get_content(content_since)
|
||
|
|
case target
|
||
|
|
in Post
|
||
|
|
target.raw
|
||
|
|
in Topic
|
||
|
|
target.posts.order(:post_number).pluck(:raw).join("\n")
|
||
|
|
in ::Chat::Channel
|
||
|
|
target
|
||
|
|
.chat_messages
|
||
|
|
.where("chat_messages.created_at > ?", content_since.hours.ago)
|
||
|
|
.includes(:user)
|
||
|
|
.order(created_at: :asc)
|
||
|
|
.pluck(:username_lower, :message)
|
||
|
|
.map { "#{_1}: #{_2}" }
|
||
|
|
.join("\n")
|
||
|
|
else
|
||
|
|
raise "Invalid target to classify"
|
||
|
|
end
|
||
|
|
end
|
||
|
|
|
||
|
|
def discourse_summarization(content)
|
||
|
|
::DiscourseAi::Inference::DiscourseClassifier.perform!(
|
||
|
|
"#{SiteSetting.ai_summarization_discourse_service_api_endpoint}/api/v1/classify",
|
||
|
|
model,
|
||
|
|
content,
|
||
|
|
SiteSetting.ai_sentiment_inference_service_api_key,
|
||
|
|
).dig(:summary_text)
|
||
|
|
end
|
||
|
|
|
||
|
|
def openai_summarization(content)
|
||
|
|
messages = [{ role: "system", content: <<~TEXT }]
|
||
|
|
Summarize the following article:\n\n#{content}
|
||
|
|
TEXT
|
||
|
|
|
||
|
|
::DiscourseAi::Inference::OpenAiCompletions.perform!(messages, model).dig(
|
||
|
|
:choices,
|
||
|
|
0,
|
||
|
|
:message,
|
||
|
|
:content,
|
||
|
|
)
|
||
|
|
end
|
||
|
|
|
||
|
|
def model
|
||
|
|
SiteSetting.ai_summarization_model
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|