| """ | |
| 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 | |