spec updated

This commit is contained in:
rngus2344 2025-03-05 16:39:54 -06:00
parent e94385927b
commit 3dfa16ad0f
No known key found for this signature in database
GPG Key ID: A548FD2A580A5131

View File

@ -64,18 +64,23 @@ RSpec.describe DiscourseAi::Summarization::Strategies::TopicSummary do
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
describe "Summary prompt" do
let!(:ai_persona) do
AiPersona.create!(
name: "TestPersona",
system_prompt: "test prompt",
description: "test",
allowed_group_ids: [Group::AUTO_GROUPS[:trust_level_0]],
)
personaClass =
end
let!(:persona_class) do
DiscourseAi::AiBot::Personas::Persona.find_by(user: admin, name: "TestPersona")
SiteSetting.ai_summary_consolidator_persona_id = personaClass.id
end
context "when ai_summary_consolidator_persona_id siteSetting is set" do
it "returns a summary extension prompt with the correct text" do
SiteSetting.ai_summary_consolidator_persona_id = persona_class.id
expect(
topic_summary
@ -87,20 +92,10 @@ RSpec.describe DiscourseAi::Summarization::Strategies::TopicSummary do
).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
it "returns a first summary prompt with the correct text" do
SiteSetting.ai_summary_persona_id = persona_class.id
expect(
topic_summary