mirror of
https://github.com/discourse/discourse-ai.git
synced 2025-03-09 11:48:47 +00:00
FIX: Only show public visible topics as suggested for anons (#27)
* FIX: Only show public visible topics as suggested for anons * DEV: Add tests for embeddings * Update spec/lib/modules/embeddings/semantic_suggested_spec.rb Co-authored-by: Bianca Nenciu <nbianca@users.noreply.github.com> * Update spec/lib/modules/embeddings/semantic_suggested_spec.rb Co-authored-by: Bianca Nenciu <nbianca@users.noreply.github.com> * move to top --------- Co-authored-by: Bianca Nenciu <nbianca@users.noreply.github.com>
This commit is contained in:
parent
4c960970fa
commit
45950f1bb4
@ -33,7 +33,12 @@ module DiscourseAi
|
|||||||
|
|
||||||
# array_position forces the order of the topics to be preserved
|
# array_position forces the order of the topics to be preserved
|
||||||
candidates =
|
candidates =
|
||||||
::Topic.where(id: candidate_ids).order("array_position(ARRAY#{candidate_ids}, id)")
|
::Topic
|
||||||
|
.visible
|
||||||
|
.listable_topics
|
||||||
|
.secured
|
||||||
|
.where(id: candidate_ids)
|
||||||
|
.order("array_position(ARRAY#{candidate_ids}, id)")
|
||||||
|
|
||||||
{ result: candidates, params: {} }
|
{ result: candidates, params: {} }
|
||||||
end
|
end
|
||||||
|
31
spec/lib/modules/embeddings/entry_point_spec.rb
Normal file
31
spec/lib/modules/embeddings/entry_point_spec.rb
Normal file
@ -0,0 +1,31 @@
|
|||||||
|
# frozen_string_literal: true
|
||||||
|
|
||||||
|
require "rails_helper"
|
||||||
|
|
||||||
|
describe DiscourseAi::Embeddings::EntryPoint do
|
||||||
|
fab!(:user) { Fabricate(:user) }
|
||||||
|
|
||||||
|
describe "registering event callbacks" do
|
||||||
|
context "when creating a topic" do
|
||||||
|
let(:creator) do
|
||||||
|
PostCreator.new(
|
||||||
|
user,
|
||||||
|
raw: "this is the new content for my topic",
|
||||||
|
title: "this is my new topic title",
|
||||||
|
)
|
||||||
|
end
|
||||||
|
|
||||||
|
it "queues a job on create if embeddings is enabled" do
|
||||||
|
SiteSetting.ai_embeddings_enabled = true
|
||||||
|
|
||||||
|
expect { creator.create }.to change(Jobs::GenerateEmbeddings.jobs, :size).by(1)
|
||||||
|
end
|
||||||
|
|
||||||
|
it "does nothing if sentiment analysis is disabled" do
|
||||||
|
SiteSetting.ai_embeddings_enabled = false
|
||||||
|
|
||||||
|
expect { creator.create }.not_to change(Jobs::GenerateEmbeddings.jobs, :size)
|
||||||
|
end
|
||||||
|
end
|
||||||
|
end
|
||||||
|
end
|
38
spec/lib/modules/embeddings/semantic_suggested_spec.rb
Normal file
38
spec/lib/modules/embeddings/semantic_suggested_spec.rb
Normal file
@ -0,0 +1,38 @@
|
|||||||
|
# frozen_string_literal: true
|
||||||
|
|
||||||
|
require "rails_helper"
|
||||||
|
|
||||||
|
describe DiscourseAi::Embeddings::SemanticSuggested do
|
||||||
|
fab!(:target) { Fabricate(:topic) }
|
||||||
|
fab!(:normal_topic_1) { Fabricate(:topic) }
|
||||||
|
fab!(:normal_topic_2) { Fabricate(:topic) }
|
||||||
|
fab!(:normal_topic_3) { Fabricate(:topic) }
|
||||||
|
fab!(:unlisted_topic) { Fabricate(:topic, visible: false) }
|
||||||
|
fab!(:private_topic) { Fabricate(:private_message_topic) }
|
||||||
|
fab!(:secured_category) { Fabricate(:category, read_restricted: true) }
|
||||||
|
fab!(:secured_category_topic) { Fabricate(:topic, category: secured_category) }
|
||||||
|
|
||||||
|
before { SiteSetting.ai_embeddings_semantic_suggested_topics_anons_enabled = true }
|
||||||
|
|
||||||
|
describe "#build_suggested_topics" do
|
||||||
|
before do
|
||||||
|
Discourse.cache.clear
|
||||||
|
described_class.stubs(:search_suggestions).returns(
|
||||||
|
Topic.unscoped.order(id: :desc).limit(10).pluck(:id),
|
||||||
|
)
|
||||||
|
end
|
||||||
|
|
||||||
|
after { Discourse.cache.clear }
|
||||||
|
|
||||||
|
it "returns the suggested topics without non public topics" do
|
||||||
|
suggested = described_class.build_suggested_topics(target, {}, TopicQuery.new(nil))
|
||||||
|
suggested_results = suggested[:result]
|
||||||
|
expect(suggested_results).to include(normal_topic_1)
|
||||||
|
expect(suggested_results).to include(normal_topic_2)
|
||||||
|
expect(suggested_results).to include(normal_topic_3)
|
||||||
|
expect(suggested_results).to_not include(unlisted_topic)
|
||||||
|
expect(suggested_results).to_not include(private_topic)
|
||||||
|
expect(suggested_results).to_not include(secured_category_topic)
|
||||||
|
end
|
||||||
|
end
|
||||||
|
end
|
Loading…
x
Reference in New Issue
Block a user