mirror of
https://github.com/discourse/discourse-ai.git
synced 2025-03-02 23:39:30 +00:00
4e05763a99
Depends on discourse/discourse#20915 Hooks to the full-page-search component using an experimental API and performs an assymetric similarity search using our embeddings database.
32 lines
805 B
Ruby
32 lines
805 B
Ruby
# frozen_string_literal: true
|
|
|
|
module DiscourseAi
|
|
module Embeddings
|
|
class SemanticSearch
|
|
def initialize(guardian, model)
|
|
@guardian = guardian
|
|
@model = model
|
|
end
|
|
|
|
def search_for_topics(query, page = 1)
|
|
limit = Search.per_filter + 1
|
|
offset = (page - 1) * Search.per_filter
|
|
|
|
candidate_ids =
|
|
DiscourseAi::Embeddings::Topic.new.asymmetric_semantic_search(model, query, limit, offset)
|
|
|
|
::Post
|
|
.where(post_type: ::Topic.visible_post_types(guardian.user))
|
|
.public_posts
|
|
.where("topics.visible")
|
|
.where(topic_id: candidate_ids, post_number: 1)
|
|
.order("array_position(ARRAY#{candidate_ids}, topic_id)")
|
|
end
|
|
|
|
private
|
|
|
|
attr_reader :model, :guardian
|
|
end
|
|
end
|
|
end
|