113 lines
3.7 KiB
Ruby
113 lines
3.7 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
RSpec.describe "Admin AI agent configuration", type: :system, js: true do
|
|
fab!(:admin)
|
|
let(:page_header) { PageObjects::Components::DPageHeader.new }
|
|
let(:form) { PageObjects::Components::FormKit.new("form") }
|
|
|
|
before do
|
|
SiteSetting.ai_bot_enabled = true
|
|
sign_in(admin)
|
|
end
|
|
|
|
it "allows creation of a agent" do
|
|
visit "/admin/plugins/discourse-ai/ai-agents"
|
|
|
|
expect(page_header).to be_visible
|
|
|
|
find(".ai-agent-list-editor__new-button").click()
|
|
|
|
expect(page_header).to be_hidden
|
|
|
|
form.field("name").fill_in("Test Agent")
|
|
form.field("description").fill_in("I am a test agent")
|
|
form.field("system_prompt").fill_in("You are a helpful bot")
|
|
|
|
tool_selector = PageObjects::Components::SelectKit.new("#control-tools .select-kit")
|
|
tool_selector.expand
|
|
tool_selector.select_row_by_value("Read")
|
|
tool_selector.select_row_by_value("ListCategories")
|
|
tool_selector.collapse
|
|
|
|
tool_selector = PageObjects::Components::SelectKit.new("#control-forcedTools .select-kit")
|
|
tool_selector.expand
|
|
tool_selector.select_row_by_value("ListCategories")
|
|
tool_selector.select_row_by_value("Read")
|
|
tool_selector.collapse
|
|
|
|
form.field("forced_tool_count").select(1)
|
|
|
|
form.submit
|
|
|
|
expect(page).not_to have_current_path("/admin/plugins/discourse-ai/ai-agents/new")
|
|
|
|
agent_id = page.current_path.split("/")[-2].to_i
|
|
|
|
agent = AiAgent.find(agent_id)
|
|
expect(agent.name).to eq("Test Agent")
|
|
expect(agent.description).to eq("I am a test agent")
|
|
expect(agent.system_prompt).to eq("You are a helpful bot")
|
|
expect(agent.forced_tool_count).to eq(1)
|
|
|
|
expected_tools = [["Read", { "read_private" => nil }, true], ["ListCategories", {}, true]]
|
|
expect(agent.tools).to contain_exactly(*expected_tools)
|
|
end
|
|
|
|
it "will not allow deletion or editing of system agents" do
|
|
visit "/admin/plugins/discourse-ai/ai-agents/#{DiscourseAi::Agents::Agent.system_agents.values.first}/edit"
|
|
expect(page).not_to have_selector(".ai-agent-editor__delete")
|
|
expect(form.field("system_prompt")).to be_disabled
|
|
end
|
|
|
|
it "will enable agent right away when you click on enable but does not save side effects" do
|
|
agent = Fabricate(:ai_agent, enabled: false)
|
|
|
|
visit "/admin/plugins/discourse-ai/ai-agents/#{agent.id}/edit"
|
|
|
|
form.field("name").fill_in("Test Agent 1")
|
|
form.field("enabled").toggle
|
|
|
|
try_until_success { expect(agent.reload.enabled).to eq(true) }
|
|
|
|
agent.reload
|
|
expect(agent.enabled).to eq(true)
|
|
expect(agent.name).not_to eq("Test Agent 1")
|
|
end
|
|
|
|
it "enabling a agent doesn't reset other fields" do
|
|
agent = Fabricate(:ai_agent, enabled: false)
|
|
updated_name = "Update agent 1"
|
|
|
|
visit "/admin/plugins/discourse-ai/ai-agents/#{agent.id}/edit"
|
|
|
|
form.field("name").fill_in(updated_name)
|
|
form.field("enabled").toggle
|
|
|
|
try_until_success { expect(agent.reload.enabled).to eq(true) }
|
|
|
|
expect(form.field("name").value).to eq(updated_name)
|
|
end
|
|
|
|
it "toggling a agent's priority doesn't reset other fields" do
|
|
agent = Fabricate(:ai_agent, priority: false)
|
|
updated_name = "Update agent 1"
|
|
|
|
visit "/admin/plugins/discourse-ai/ai-agents/#{agent.id}/edit"
|
|
|
|
form.field("name").fill_in(updated_name)
|
|
form.field("priority").toggle
|
|
|
|
try_until_success { expect(agent.reload.priority).to eq(true) }
|
|
|
|
expect(form.field("name").value).to eq(updated_name)
|
|
end
|
|
|
|
it "can navigate the AI plugin with breadcrumbs" do
|
|
visit "/admin/plugins/discourse-ai/ai-agents"
|
|
expect(page).to have_css(".d-breadcrumbs")
|
|
expect(page).to have_css(".d-breadcrumbs__item", count: 4)
|
|
find(".d-breadcrumbs__item", text: I18n.t("admin_js.admin.plugins.title")).click
|
|
expect(page).to have_current_path("/admin/plugins")
|
|
end
|
|
end
|