mirror of
https://github.com/discourse/discourse-ai.git
synced 2025-03-06 09:20:14 +00:00
spec updated
This commit is contained in:
parent
e94385927b
commit
3dfa16ad0f
@ -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
|
||||
|
Loading…
x
Reference in New Issue
Block a user