144 lines
4.0 KiB
Ruby
144 lines
4.0 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
require_relative "../../../support/embeddings_generation_stubs"
|
|
require_relative "../../../support/openai_completions_inference_stubs"
|
|
|
|
RSpec.describe DiscourseAi::Embeddings::SemanticSearch do
|
|
fab!(:post) { Fabricate(:post) }
|
|
fab!(:user) { Fabricate(:user) }
|
|
|
|
let(:query) { "test_query" }
|
|
let(:subject) { described_class.new(Guardian.new(user)) }
|
|
|
|
describe "#search_for_topics" do
|
|
let(:hypothetical_post) { "This is an hypothetical post generated from the keyword test_query" }
|
|
|
|
before do
|
|
SiteSetting.ai_embeddings_discourse_service_api_endpoint = "http://test.com"
|
|
|
|
prompt = DiscourseAi::Embeddings::HydeGenerators::OpenAi.new.prompt(query)
|
|
OpenAiCompletionsInferenceStubs.stub_response(
|
|
prompt,
|
|
hypothetical_post,
|
|
req_opts: {
|
|
max_tokens: 400,
|
|
},
|
|
)
|
|
|
|
hyde_embedding = [0.049382, 0.9999]
|
|
EmbeddingsGenerationStubs.discourse_service(
|
|
SiteSetting.ai_embeddings_model,
|
|
hypothetical_post,
|
|
hyde_embedding,
|
|
)
|
|
end
|
|
|
|
after { described_class.clear_cache_for(query) }
|
|
|
|
def stub_candidate_ids(candidate_ids)
|
|
DiscourseAi::Embeddings::VectorRepresentations::AllMpnetBaseV2
|
|
.any_instance
|
|
.expects(:asymmetric_topics_similarity_search)
|
|
.returns(candidate_ids)
|
|
end
|
|
|
|
it "returns the first post of a topic included in the asymmetric search results" do
|
|
stub_candidate_ids([post.topic_id])
|
|
|
|
posts = subject.search_for_topics(query)
|
|
|
|
expect(posts).to contain_exactly(post)
|
|
end
|
|
|
|
describe "applies different scopes to the candidates" do
|
|
context "when the topic is not visible" do
|
|
it "returns an empty list" do
|
|
post.topic.update!(visible: false)
|
|
stub_candidate_ids([post.topic_id])
|
|
|
|
posts = subject.search_for_topics(query)
|
|
|
|
expect(posts).to be_empty
|
|
end
|
|
end
|
|
|
|
context "when the post is not public" do
|
|
it "returns an empty list" do
|
|
pm_post = Fabricate(:private_message_post)
|
|
stub_candidate_ids([pm_post.topic_id])
|
|
|
|
posts = subject.search_for_topics(query)
|
|
|
|
expect(posts).to be_empty
|
|
end
|
|
end
|
|
|
|
context "when the post type is not visible" do
|
|
it "returns an empty list" do
|
|
post.update!(post_type: Post.types[:whisper])
|
|
stub_candidate_ids([post.topic_id])
|
|
|
|
posts = subject.search_for_topics(query)
|
|
|
|
expect(posts).to be_empty
|
|
end
|
|
end
|
|
|
|
context "when the post is not the first post in the topic" do
|
|
it "returns an empty list" do
|
|
reply = Fabricate(:reply)
|
|
reply.topic.first_post.trash!
|
|
stub_candidate_ids([reply.topic_id])
|
|
|
|
posts = subject.search_for_topics(query)
|
|
|
|
expect(posts).to be_empty
|
|
end
|
|
end
|
|
|
|
context "when the post is not a candidate" do
|
|
it "doesn't include it in the results" do
|
|
post_2 = Fabricate(:post)
|
|
stub_candidate_ids([post.topic_id])
|
|
|
|
posts = subject.search_for_topics(query)
|
|
|
|
expect(posts).not_to include(post_2)
|
|
end
|
|
end
|
|
|
|
context "when the post belongs to a secured category" do
|
|
fab!(:group) { Fabricate(:group) }
|
|
fab!(:private_category) { Fabricate(:private_category, group: group) }
|
|
|
|
before do
|
|
post.topic.update!(category: private_category)
|
|
stub_candidate_ids([post.topic_id])
|
|
end
|
|
|
|
it "returns an empty list" do
|
|
posts = subject.search_for_topics(query)
|
|
|
|
expect(posts).to be_empty
|
|
end
|
|
|
|
it "returns the results if the user has access to the category" do
|
|
group.add(user)
|
|
|
|
posts = subject.search_for_topics(query)
|
|
|
|
expect(posts).to contain_exactly(post)
|
|
end
|
|
|
|
context "while searching as anon" do
|
|
it "returns an empty list" do
|
|
posts = described_class.new(Guardian.new(nil)).search_for_topics(query)
|
|
|
|
expect(posts).to be_empty
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|