50 lines
1.5 KiB
Ruby
50 lines
1.5 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
require_relative "support/toxicity_inference_stubs"
|
|
|
|
describe Plugin::Instance do
|
|
before do
|
|
SiteSetting.discourse_ai_enabled = true
|
|
SiteSetting.ai_toxicity_inference_service_api_endpoint = "http://example.com"
|
|
end
|
|
|
|
describe "on reviewable_transitioned_to event" do
|
|
fab!(:post)
|
|
fab!(:admin)
|
|
|
|
it "adjusts model accuracy" do
|
|
ToxicityInferenceStubs.stub_post_classification(post, toxic: true)
|
|
SiteSetting.ai_toxicity_flag_automatically = true
|
|
classification = DiscourseAi::Toxicity::ToxicityClassification.new
|
|
classificator = DiscourseAi::PostClassificator.new(classification)
|
|
classificator.classify!(post)
|
|
reviewable = ReviewableAiPost.find_by(target: post)
|
|
|
|
reviewable.perform admin, :agree_and_keep
|
|
accuracy = ModelAccuracy.find_by(classification_type: classification.type)
|
|
|
|
expect(accuracy.flags_agreed).to eq(1)
|
|
end
|
|
end
|
|
|
|
describe "current_user_serializer#ai_helper_prompts" do
|
|
fab!(:user)
|
|
|
|
before do
|
|
assign_fake_provider_to(:ai_helper_model)
|
|
SiteSetting.ai_helper_enabled = true
|
|
SiteSetting.ai_helper_illustrate_post_model = "disabled"
|
|
Group.find_by(id: Group::AUTO_GROUPS[:admins]).add(user)
|
|
|
|
DiscourseAi::AiHelper::Assistant.clear_prompt_cache!
|
|
end
|
|
|
|
let(:serializer) { CurrentUserSerializer.new(user, scope: Guardian.new(user)) }
|
|
|
|
it "returns the available prompts" do
|
|
expect(serializer.ai_helper_prompts).to be_present
|
|
expect(serializer.ai_helper_prompts.object.count).to eq(6)
|
|
end
|
|
end
|
|
end
|