"Update MultiVectorRetriever to retrieve 2 similar documents instead …

…of 1."
buhe · Feb 18, 2024 · c8da24f · c8da24f
1 parent 8cc143b
commit c8da24f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/Sources/LangChain/retriever/MultiVectorRetriever.swift b/Sources/LangChain/retriever/MultiVectorRetriever.swift
@@ -17,7 +17,7 @@ public class MultiVectorRetriever: BaseRetriever {
     }
 
     public override func _get_relevant_documents(query: String) async throws  -> [Document] {
-        let sub_docs = await self.vectorstore.similaritySearch(query: query, k: 1)
+        let sub_docs = await self.vectorstore.similaritySearch(query: query, k: 2)
         var ids: [String] = []
         for d in sub_docs {
             ids.append(d.metadata[self.id_key]!)