mirror of
https://github.com/discourse/discourse-ai.git
synced 2025-02-08 04:28:18 +00:00
52a7dd2a4b
This is a rather huge refactor with 1 new feature (tool details can be suppressed) Previously we use the name "Command" to describe "Tools", this unifies all the internal language and simplifies the code. We also amended the persona UI to use less DToggles which aligns with our design guidelines. Co-authored-by: Martin Brennan <martin@discourse.org>
149 lines
3.9 KiB
Ruby
149 lines
3.9 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
RSpec.describe AiPersona do
|
|
it "validates context settings" do
|
|
persona =
|
|
AiPersona.new(
|
|
name: "test",
|
|
description: "test",
|
|
system_prompt: "test",
|
|
tools: [],
|
|
allowed_group_ids: [],
|
|
)
|
|
|
|
expect(persona.valid?).to eq(true)
|
|
|
|
persona.max_context_posts = 0
|
|
expect(persona.valid?).to eq(false)
|
|
expect(persona.errors[:max_context_posts]).to eq(["must be greater than 0"])
|
|
|
|
persona.max_context_posts = 1
|
|
expect(persona.valid?).to eq(true)
|
|
|
|
persona.max_context_posts = nil
|
|
expect(persona.valid?).to eq(true)
|
|
end
|
|
|
|
it "allows creation of user" do
|
|
persona =
|
|
AiPersona.create!(
|
|
name: "test",
|
|
description: "test",
|
|
system_prompt: "test",
|
|
tools: [],
|
|
allowed_group_ids: [],
|
|
)
|
|
|
|
user = persona.create_user!
|
|
expect(user.username).to eq("test_bot")
|
|
expect(user.name).to eq("Test")
|
|
expect(user.bot?).to be(true)
|
|
expect(user.id).to be <= AiPersona::FIRST_PERSONA_USER_ID
|
|
end
|
|
|
|
it "removes all rag embeddings when rag params change" do
|
|
persona =
|
|
AiPersona.create!(
|
|
name: "test",
|
|
description: "test",
|
|
system_prompt: "test",
|
|
tools: [],
|
|
allowed_group_ids: [],
|
|
rag_chunk_tokens: 10,
|
|
rag_chunk_overlap_tokens: 5,
|
|
)
|
|
|
|
id =
|
|
RagDocumentFragment.create!(
|
|
ai_persona: persona,
|
|
fragment: "test",
|
|
fragment_number: 1,
|
|
upload: Fabricate(:upload),
|
|
).id
|
|
|
|
persona.rag_chunk_tokens = 20
|
|
persona.save!
|
|
|
|
expect(RagDocumentFragment.exists?(id)).to eq(false)
|
|
end
|
|
|
|
it "defines singleton methods on system persona classes" do
|
|
forum_helper = AiPersona.find_by(name: "Forum Helper")
|
|
forum_helper.update!(
|
|
user_id: 1,
|
|
mentionable: true,
|
|
default_llm: "anthropic:claude-2",
|
|
max_context_posts: 3,
|
|
)
|
|
|
|
klass = forum_helper.class_instance
|
|
|
|
expect(klass.id).to eq(forum_helper.id)
|
|
expect(klass.system).to eq(true)
|
|
# tl 0 by default
|
|
expect(klass.allowed_group_ids).to eq([10])
|
|
expect(klass.user_id).to eq(1)
|
|
expect(klass.mentionable).to eq(true)
|
|
expect(klass.default_llm).to eq("anthropic:claude-2")
|
|
expect(klass.max_context_posts).to eq(3)
|
|
end
|
|
|
|
it "defines singleton methods non persona classes" do
|
|
persona =
|
|
AiPersona.create!(
|
|
name: "test",
|
|
description: "test",
|
|
system_prompt: "test",
|
|
tools: [],
|
|
allowed_group_ids: [],
|
|
default_llm: "anthropic:claude-2",
|
|
max_context_posts: 3,
|
|
mentionable: true,
|
|
user_id: 1,
|
|
)
|
|
|
|
klass = persona.class_instance
|
|
|
|
expect(klass.id).to eq(persona.id)
|
|
expect(klass.system).to eq(false)
|
|
expect(klass.allowed_group_ids).to eq([])
|
|
expect(klass.user_id).to eq(1)
|
|
expect(klass.mentionable).to eq(true)
|
|
expect(klass.default_llm).to eq("anthropic:claude-2")
|
|
expect(klass.max_context_posts).to eq(3)
|
|
end
|
|
|
|
it "does not allow setting allow_chat without a default_llm" do
|
|
persona =
|
|
AiPersona.create(
|
|
name: "test",
|
|
description: "test",
|
|
system_prompt: "test",
|
|
allowed_group_ids: [],
|
|
default_llm: nil,
|
|
allow_chat: true,
|
|
)
|
|
|
|
expect(persona.valid?).to eq(false)
|
|
expect(persona.errors[:default_llm].first).to eq(
|
|
I18n.t("discourse_ai.ai_bot.personas.default_llm_required"),
|
|
)
|
|
end
|
|
|
|
it "does not leak caches between sites" do
|
|
AiPersona.create!(
|
|
name: "pun_bot",
|
|
description: "you write puns",
|
|
system_prompt: "you are pun bot",
|
|
tools: ["ImageCommand"],
|
|
allowed_group_ids: [Group::AUTO_GROUPS[:trust_level_0]],
|
|
)
|
|
|
|
AiPersona.all_personas
|
|
|
|
expect(AiPersona.persona_cache[:value].length).to be > (0)
|
|
RailsMultisite::ConnectionManagement.stubs(:current_db) { "abc" }
|
|
expect(AiPersona.persona_cache[:value]).to eq(nil)
|
|
end
|
|
end
|