retrieve.py

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