Update src/vectorstore/vectorstore.py

src/vectorstore/vectorstore.py

@@ -31,7 +31,7 @@ class VectorStore:
             documents: List of documents to embed
         """
         self.vectorstore = FAISS.from_documents(documents, self.embedding)
-        self.retriever = self.vectorstore.as_retriever()
+        self.retriever = self.vectorstore.as_retriever(search_type="mmr", search_kwargs={"k": 5, "lambda_mult":0.25})
     
     def get_retriever(self):
         """