rag/retriever.py

from rag.vectorstore import get_vectorstore

def retrieve_documents(query: str, k: int = 3):
    """
    Retrieves the top k relevant documents for the given query.
    """
    vectorstore = get_vectorstore()
    retriever = vectorstore.as_retriever(search_kwargs={"k": k})
    docs = retriever.invoke(query)

    return [doc.page_content for doc in docs]