index = VectorStoreIndex.from_documents(documents)

retriever = index.as_retriever(similarity_top_k=2)