80 lines
2.8 KiB
Ruby
80 lines
2.8 KiB
Ruby
# frozen_string_literal: true
|
|
RSpec.describe "AI personas", type: :system, js: true do
|
|
fab!(:admin)
|
|
fab!(:gpt_4) { Fabricate(:llm_model, name: "gpt-4") }
|
|
|
|
before do
|
|
SiteSetting.ai_bot_enabled = true
|
|
toggle_enabled_bots(bots: [gpt_4])
|
|
sign_in(admin)
|
|
|
|
Group.refresh_automatic_groups!
|
|
end
|
|
|
|
it "remembers the last selected persona" do
|
|
visit "/"
|
|
find(".d-header .ai-bot-button").click()
|
|
persona_selector =
|
|
PageObjects::Components::SelectKit.new(".persona-llm-selector__persona-dropdown")
|
|
|
|
id = DiscourseAi::AiBot::Personas::Persona.all(user: admin).first.id
|
|
|
|
expect(persona_selector).to have_selected_value(id)
|
|
|
|
persona_selector.expand
|
|
persona_selector.select_row_by_value(-2)
|
|
|
|
visit "/"
|
|
find(".d-header .ai-bot-button").click()
|
|
persona_selector =
|
|
PageObjects::Components::SelectKit.new(".persona-llm-selector__persona-dropdown")
|
|
persona_selector.expand
|
|
expect(persona_selector).to have_selected_value(-2)
|
|
end
|
|
|
|
it "allows creation of a persona" do
|
|
visit "/admin/plugins/discourse-ai/ai-personas"
|
|
find(".ai-persona-list-editor__header .btn-primary").click()
|
|
find(".ai-persona-editor__name").set("Test Persona")
|
|
find(".ai-persona-editor__description").fill_in(with: "I am a test persona")
|
|
find(".ai-persona-editor__system_prompt").fill_in(with: "You are a helpful bot")
|
|
|
|
tool_selector = PageObjects::Components::SelectKit.new(".ai-persona-editor__tools")
|
|
tool_selector.expand
|
|
tool_selector.select_row_by_value("Read")
|
|
|
|
find(".ai-persona-editor__save").click()
|
|
|
|
expect(page).not_to have_current_path("/admin/plugins/discourse-ai/ai-personas/new")
|
|
|
|
persona_id = page.current_path.split("/").last.to_i
|
|
|
|
persona = AiPersona.find(persona_id)
|
|
expect(persona.name).to eq("Test Persona")
|
|
expect(persona.description).to eq("I am a test persona")
|
|
expect(persona.system_prompt).to eq("You are a helpful bot")
|
|
expect(persona.tools).to eq([["Read", { "read_private" => nil }]])
|
|
end
|
|
|
|
it "will not allow deletion or editing of system personas" do
|
|
visit "/admin/plugins/discourse-ai/ai-personas/#{DiscourseAi::AiBot::Personas::Persona.system_personas.values.first}"
|
|
expect(page).not_to have_selector(".ai-persona-editor__delete")
|
|
expect(find(".ai-persona-editor__system_prompt")).to be_disabled
|
|
end
|
|
|
|
it "will enable persona right away when you click on enable but does not save side effects" do
|
|
persona = Fabricate(:ai_persona, enabled: false)
|
|
|
|
visit "/admin/plugins/discourse-ai/ai-personas/#{persona.id}"
|
|
|
|
find(".ai-persona-editor__name").set("Test Persona 1")
|
|
PageObjects::Components::DToggleSwitch.new(".ai-persona-editor__enabled").toggle
|
|
|
|
try_until_success { expect(persona.reload.enabled).to eq(true) }
|
|
|
|
persona.reload
|
|
expect(persona.enabled).to eq(true)
|
|
expect(persona.name).not_to eq("Test Persona 1")
|
|
end
|
|
end
|