mirror of
https://gh.wpcy.net/https://github.com/discourse/discourse.git
synced 2026-06-19 03:23:50 +08:00
58 lines
1.9 KiB
Ruby
Vendored
58 lines
1.9 KiB
Ruby
Vendored
# frozen_string_literal: true
|
|
|
|
RSpec.describe "AI agents" do
|
|
fab!(:admin)
|
|
fab!(:gpt_4) { Fabricate(:llm_model, name: "gpt-4") }
|
|
|
|
before do
|
|
enable_current_plugin
|
|
SiteSetting.ai_bot_enabled = true
|
|
SiteSetting.ai_bot_add_to_header = true
|
|
toggle_enabled_bots(bots: [gpt_4])
|
|
sign_in(admin)
|
|
end
|
|
|
|
it "can select and save agent tool options" do
|
|
visit "/admin/plugins/discourse-ai/ai-agents"
|
|
find(".ai-agent-list-editor__new-button").click
|
|
|
|
expect(page).to have_current_path("/admin/plugins/discourse-ai/ai-agents/new")
|
|
|
|
form = PageObjects::Components::FormKit.new("form")
|
|
form.field("name").fill_in("Test Agent")
|
|
form.field("description").fill_in("This is a test agent.")
|
|
form.field("system_prompt").fill_in("You are a helpful assistant.")
|
|
form.field("tools").select("Update Artifact")
|
|
form.field("toolOptions.UpdateArtifact.update_algorithm").select("full")
|
|
form.submit
|
|
|
|
expect(page).to have_current_path(%r{/admin/plugins/discourse-ai/ai-agents/\d+/edit})
|
|
|
|
agent = AiAgent.order("id desc").first
|
|
|
|
expect(agent.name).to eq("Test Agent")
|
|
expect(agent.description).to eq("This is a test agent.")
|
|
expect(agent.tools.count).to eq(1)
|
|
expect(agent.tools.first[0]).to eq("UpdateArtifact")
|
|
expect(agent.tools.first[1]["update_algorithm"]).to eq("full")
|
|
end
|
|
|
|
it "remembers the last selected agent" do
|
|
visit "/"
|
|
find(".d-header .ai-bot-button").click()
|
|
agent_selector = PageObjects::Components::SelectKit.new(".agent-llm-selector__agent-dropdown")
|
|
|
|
id = DiscourseAi::Agents::Agent.all(user: admin).first.id
|
|
|
|
expect(agent_selector).to have_selected_value(id)
|
|
|
|
agent_selector.expand
|
|
agent_selector.select_row_by_value(-2)
|
|
|
|
visit "/"
|
|
find(".d-header .ai-bot-button").click()
|
|
agent_selector = PageObjects::Components::SelectKit.new(".agent-llm-selector__agent-dropdown")
|
|
agent_selector.expand
|
|
expect(agent_selector).to have_selected_value(-2)
|
|
end
|
|
end
|
