"""
BƯỚC 5: RETRIEVER
"""

from langchain_community.vectorstores import FAISS

RETRIEVER_K = 4


def get_retriever(vectorstore: FAISS, k: int = RETRIEVER_K):
    print(f">>> Creating retriever with k={k} ...")
    retriever = vectorstore.as_retriever(search_kwargs={"k": k})
    print(">>> Retriever ready.\n")
    return retriever