filter was being overwritten (#206)

Co-authored-by: Vijay Swamidass <[email protected]>

Author: vswamidass-sfdc

app/controllers/base_documents_controller.rb

@@ -55,7 +55,11 @@ def index
 
     if params[:similar_to].present?
       embedding = get_embedding(params[:similar_to])
-      @documents = related_documents_from_embedding_by_libraries(embedding, library_id)
+      # Get similar documents but preserve existing filters
+      similar_docs = related_documents_from_embedding_by_libraries(embedding, library_id)
+      # Apply the similarity search as an additional filter by getting the IDs
+      similar_doc_ids = similar_docs.pluck(:id)
+      @documents = @documents.where(id: similar_doc_ids)
     end
 
     @documents = @documents.search_by_title_and_document(params[:contains]) if params[:contains].present?

app/controllers/concerns/neighbor_concern.rb

@@ -14,6 +14,6 @@ def related_documents_from_embedding(embedding)
   def related_documents_from_embedding_by_libraries(_embedding, _library_ids)
     scope = related_documents_from_embedding(_embedding)
     scope = scope.where(library_id: _library_ids) if _library_ids.present?
-    scope.order(created_at: :desc).limit(100)
+    scope.order(updated_at: :desc).limit(100)
   end
 end