discourse/discourse-ai
1
0
Fork 0
mirror of https://github.com/discourse/discourse-ai.git synced 2025-07-08 23:32:45 +00:00
Code Issues Packages Projects Releases Wiki Activity

spec added

Browse Source
This commit is contained in:
rngus2344 2025-03-05 16:21:51 -06:00
parent 20701cb48a
commit e94385927b
No known key found for this signature in database
GPG Key ID: A548FD2A580A5131
1 changed files with 51 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

51
spec/lib/modules/summarization/strategies/topic_summary_spec.rb
View File

@ -6,6 +6,7 @@ RSpec.describe DiscourseAi::Summarization::Strategies::TopicSummary do
fab!(:topic) { Fabricate(:topic, highest_post_number: 25) }
fab!(:post_1) { Fabricate(:post, topic: topic, post_number: 1) }
fab!(:post_2) { Fabricate(:post, topic: topic, post_number: 2) }
fab!(:admin)
describe "#targets_data" do
shared_examples "includes only public-visible topics" do
@ -62,4 +63,54 @@ RSpec.describe DiscourseAi::Summarization::Strategies::TopicSummary do
end
end
end
describe "#summary_extension_prompt" do
context "when ai_summary_consolidator_persona_id siteSetting is set" do
it "returns a prompt with the correct text" do
AiPersona.create!(
name: "TestPersona",
system_prompt: "test prompt",
description: "test",
allowed_group_ids: [Group::AUTO_GROUPS[:trust_level_0]],
)
personaClass =
DiscourseAi::AiBot::Personas::Persona.find_by(user: admin, name: "TestPersona")
SiteSetting.ai_summary_consolidator_persona_id = personaClass.id
expect(
topic_summary
.summary_extension_prompt(nil, [{ id: 1, poster: "test", text: "hello" }])
.messages
.first[
:content
],
).to include("test prompt")
end
end
end
describe "#first_summary_prompt" do
context "when ai_summary_persona_id siteSetting is set" do
it "returns a prompt with the correct text" do
AiPersona.create!(
name: "TestPersona",
system_prompt: "test prompt",
description: "test",
allowed_group_ids: [Group::AUTO_GROUPS[:trust_level_0]],
)
personaClass =
DiscourseAi::AiBot::Personas::Persona.find_by(user: admin, name: "TestPersona")
SiteSetting.ai_summary_persona_id = personaClass.id
expect(
topic_summary
.first_summary_prompt([{ id: 1, poster: "test", text: "hello" }])
.messages
.first[
:content
],
).to include("test prompt")
end
end
end
end