mirror of
https://gh.wpcy.net/https://github.com/discourse/discourse.git
synced 2026-06-19 02:05:37 +08:00
035ce20c43
Adds an AI feature, agent, to discourse-ai. This feature is meant to be used on the new admin dashboard, but is owned by the discourse-ai plugin. The PR includes an outlet for discourse-ai to hook up to. The agent introduced will take crafted data in for now to formulate the headline and we will adjust this accordingly. Note, we're not doing any streaming here for now. The results are cached for 6h and per locale. The AI feature is "hidden" for now
128 lines
4 KiB
Ruby
Vendored
128 lines
4 KiB
Ruby
Vendored
# frozen_string_literal: true
|
|
|
|
RSpec.describe "Admin AI features configuration" do
|
|
fab!(:admin)
|
|
fab!(:llm_model)
|
|
fab!(:summarization_agent, :ai_agent)
|
|
fab!(:group_1, :group)
|
|
fab!(:group_2, :group)
|
|
let(:page_header) { PageObjects::Components::DPageHeader.new }
|
|
let(:form) { PageObjects::Components::FormKit.new("form") }
|
|
let(:ai_features_page) { PageObjects::Pages::AdminAiFeatures.new }
|
|
|
|
before do
|
|
enable_current_plugin
|
|
summarization_agent.allowed_group_ids = [group_1.id, group_2.id]
|
|
summarization_agent.save!
|
|
assign_fake_provider_to(:ai_default_llm_model)
|
|
SiteSetting.ai_summarization_enabled = true
|
|
SiteSetting.ai_summarization_agent = summarization_agent.id
|
|
sign_in(admin)
|
|
end
|
|
|
|
it "lists all persona backed AI features separated by enabled/not enabled" do
|
|
all_modules = DiscourseAi::Configuration::Module.all.select(&:visible?)
|
|
configured_count = all_modules.count(&:enabled?)
|
|
ai_features_page.visit
|
|
ai_features_page.toggle_enabled
|
|
|
|
expect(ai_features_page).to have_listed_modules(configured_count)
|
|
|
|
ai_features_page.toggle_not_enabled
|
|
|
|
expect(ai_features_page).to have_listed_modules(all_modules.size - configured_count)
|
|
|
|
screenshot_marker(label: "ai-admin-features")
|
|
end
|
|
|
|
it "lists the agent used for the corresponding AI feature" do
|
|
ai_features_page.visit
|
|
|
|
ai_features_page.toggle_enabled
|
|
|
|
expect(ai_features_page).to have_feature_agent("topic_summaries", summarization_agent.name)
|
|
end
|
|
|
|
it "lists the groups allowed to use the AI feature" do
|
|
ai_features_page.visit
|
|
|
|
ai_features_page.toggle_enabled
|
|
|
|
expect(ai_features_page).to have_feature_groups("topic_summaries", [group_1.name, group_2.name])
|
|
end
|
|
|
|
it "shows edit page with grouped settings" do
|
|
ai_features_page.visit
|
|
|
|
ai_features_page.click_edit_module("summarization")
|
|
|
|
expect(page).to have_current_path("/admin/plugins/discourse-ai/ai-features/1/edit")
|
|
|
|
expect(page).to have_css(".ai-feature-editor")
|
|
expect(page).to have_css(".form-kit__section")
|
|
expect(page).to have_css(".form-kit__field")
|
|
end
|
|
|
|
it "renders group_list settings as group selectors" do
|
|
SiteSetting.ai_bot_enabled = true
|
|
SiteSetting.ai_bot_allowed_groups = "#{group_1.id}|#{group_2.id}"
|
|
|
|
page.visit(
|
|
"/admin/plugins/discourse-ai/ai-features/#{DiscourseAi::Configuration::Module::BOT_ID}/edit",
|
|
)
|
|
|
|
expect(page).to have_css(".ai-feature-editor")
|
|
|
|
field = form.field("ai_bot_allowed_groups")
|
|
expect(field.component).to have_css(".list-setting")
|
|
expect(field.component).to have_content(group_1.name)
|
|
expect(field.component).to have_content(group_2.name)
|
|
end
|
|
|
|
it "displays LLM names in compact_list settings" do
|
|
llm1 = Fabricate(:llm_model, display_name: "Test LLM Alpha")
|
|
llm2 = Fabricate(:llm_model, display_name: "Test LLM Beta")
|
|
|
|
SiteSetting.ai_bot_enabled = true
|
|
SiteSetting.ai_bot_enabled_llms = "#{llm1.id}|#{llm2.id}"
|
|
|
|
page.visit(
|
|
"/admin/plugins/discourse-ai/ai-features/#{DiscourseAi::Configuration::Module::BOT_ID}/edit",
|
|
)
|
|
|
|
expect(page).to have_css(".ai-feature-editor")
|
|
|
|
field = form.field("ai_bot_enabled_llms")
|
|
expect(field.component).to have_content("Test LLM Alpha")
|
|
expect(field.component).to have_content("Test LLM Beta")
|
|
end
|
|
|
|
context "with external AI features" do
|
|
let(:fake_plugin) do
|
|
plugin = Plugin::Instance.new
|
|
plugin.path = "#{Rails.root.join("spec/fixtures/plugins/my_plugin/plugin.rb")}"
|
|
plugin
|
|
end
|
|
|
|
before do
|
|
DiscourseAi.register_feature(
|
|
module_name: :test_external,
|
|
feature: :test_feature,
|
|
agent_klass: DiscourseAi::TestHelpers::FakeExternalAgent,
|
|
plugin: fake_plugin,
|
|
)
|
|
end
|
|
|
|
after do
|
|
DiscoursePluginRegistry._raw_external_ai_features.reject! do |entry|
|
|
entry[:value][:module_name] == :test_external
|
|
end
|
|
end
|
|
|
|
it "shows external modules from the registry" do
|
|
ai_features_page.visit
|
|
expect(page).to have_content("test_external")
|
|
expect(page).to have_content("test_feature")
|
|
end
|
|
end
|
|
end
|
