The k=RETRIEVAL_K should be passed to the as_retriever method, not the invoke, as in these examples from LangChain https://reference.langchain.com/python/langchain_core/vectorstores/#langchain_core.vectorstores.base.VectorStore.as_retriever
retriever = vectorstore.as_retriever(search_kwargs={"k": RETRIEVAL_K})
|
retriever = vectorstore.as_retriever() |