def retrieve(saved_vector): print("Retrieving similar documents...") retriever = saved_vector.as_retriever( search_type="similarity", search_kwargs={"k": 3} ) return retriever