File size: 335 Bytes
ed084d7 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 | """
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
|