Method: SimpleRag::Retrieve#similarity_search

Defined in:
lib/simple_rag/retrieve.rb

#similarity_search(question_embeddings, k_neighbors_count) ⇒ Object 



23
24
25
26
27
 
# File 'lib/simple_rag/retrieve.rb', line 23

def similarity_search(question_embeddings, k_neighbors_count)
  distances, indices = @index.search(question_embeddings, k_neighbors_count)
  index_array = indices.to_a[0]
  index_array.map { |i| @chunks[i] }
end