127 lines
4.0 KiB
Ruby
127 lines
4.0 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
RSpec.describe Jobs::HotTopicsGistBatch do
|
|
fab!(:topic_1) { Fabricate(:topic) }
|
|
fab!(:post_1) { Fabricate(:post, topic: topic_1, post_number: 1) }
|
|
fab!(:post_2) { Fabricate(:post, topic: topic_1, post_number: 2) }
|
|
|
|
before do
|
|
assign_fake_provider_to(:ai_summarization_model)
|
|
SiteSetting.ai_summarization_enabled = true
|
|
SiteSetting.ai_summarize_max_hot_topics_gists_per_batch = 100
|
|
end
|
|
|
|
describe "#execute" do
|
|
context "when there is a topic with a hot score" do
|
|
before { TopicHotScore.create!(topic_id: topic_1.id, score: 0.1) }
|
|
|
|
it "does nothing if the plugin is disabled" do
|
|
SiteSetting.discourse_ai_enabled = false
|
|
|
|
subject.execute({})
|
|
|
|
gist = AiSummary.gist.find_by(target: topic_1)
|
|
expect(gist).to be_nil
|
|
end
|
|
|
|
it "does nothing if the summarization module is disabled" do
|
|
SiteSetting.ai_summarization_enabled = false
|
|
|
|
subject.execute({})
|
|
|
|
gist = AiSummary.gist.find_by(target: topic_1)
|
|
expect(gist).to be_nil
|
|
end
|
|
|
|
it "does nothing if hot topics summarization is disabled" do
|
|
SiteSetting.ai_summarize_max_hot_topics_gists_per_batch = 0
|
|
|
|
subject.execute({})
|
|
|
|
gist = AiSummary.gist.find_by(target: topic_1)
|
|
expect(gist).to be_nil
|
|
end
|
|
|
|
it "creates a gist" do
|
|
gist_result = "I'm a gist"
|
|
|
|
DiscourseAi::Completions::Llm.with_prepared_responses([gist_result]) { subject.execute({}) }
|
|
|
|
gist = AiSummary.gist.find_by(target: topic_1)
|
|
expect(gist.summarized_text).to eq(gist_result)
|
|
end
|
|
|
|
context "when we already generated a gist of it" do
|
|
fab!(:ai_gist) do
|
|
Fabricate(
|
|
:topic_ai_gist,
|
|
target: topic_1,
|
|
original_content_sha: AiSummary.build_sha("12"),
|
|
)
|
|
end
|
|
|
|
it "does nothing if the gist is up to date" do
|
|
updated_gist = "They updated me :("
|
|
|
|
DiscourseAi::Completions::Llm.with_prepared_responses([updated_gist]) do
|
|
subject.execute({})
|
|
end
|
|
|
|
gist = AiSummary.gist.find_by(target: topic_1)
|
|
expect(AiSummary.gist.where(target: topic_1).count).to eq(1)
|
|
expect(gist.summarized_text).not_to eq(updated_gist)
|
|
expect(gist.original_content_sha).to eq(ai_gist.original_content_sha)
|
|
end
|
|
|
|
it "regenerates it if it's outdated" do
|
|
Fabricate(:post, topic: topic_1, post_number: 3)
|
|
gist_result = "They updated me"
|
|
|
|
DiscourseAi::Completions::Llm.with_prepared_responses([gist_result]) do
|
|
subject.execute({})
|
|
end
|
|
|
|
gist = AiSummary.gist.find_by(target: topic_1)
|
|
expect(gist.summarized_text).to eq(gist_result)
|
|
expect(gist.original_content_sha).to eq(AiSummary.build_sha("123"))
|
|
end
|
|
end
|
|
end
|
|
|
|
context "when there is a topic but it doesn't have a hot score" do
|
|
it "does nothing" do
|
|
subject.execute({})
|
|
|
|
gist = AiSummary.gist.find_by(target: topic_1)
|
|
expect(gist).to be_nil
|
|
end
|
|
end
|
|
|
|
context "when there are multiple hot topics" do
|
|
fab!(:topic_2) { Fabricate(:topic) }
|
|
fab!(:post_2_1) { Fabricate(:post, topic: topic_2, post_number: 1) }
|
|
fab!(:post_2_2) { Fabricate(:post, topic: topic_2, post_number: 2) }
|
|
|
|
before do
|
|
TopicHotScore.create!(topic_id: topic_1.id, score: 0.2)
|
|
TopicHotScore.create!(topic_id: topic_2.id, score: 0.4)
|
|
end
|
|
|
|
it "processes them by score order" do
|
|
topic_1_gist = "I'm gist of topic 1"
|
|
topic_2_gist = "I'm gist of topic 2"
|
|
|
|
DiscourseAi::Completions::Llm.with_prepared_responses([topic_2_gist, topic_1_gist]) do
|
|
subject.execute({})
|
|
end
|
|
|
|
gist = AiSummary.gist.find_by(target: topic_1)
|
|
expect(gist.summarized_text).to eq(topic_1_gist)
|
|
|
|
gist_2 = AiSummary.gist.find_by(target: topic_2)
|
|
expect(gist_2.summarized_text).to eq(topic_2_gist)
|
|
end
|
|
end
|
|
end
|
|
end
|