from langchain.chains import RetrievalQA def generate_response(llm, vector_store, question): qa_chain = RetrievalQA.from_chain_type( llm=llm, retriever=vector_store.as_retriever(), return_source_documents=True ) result = qa_chain.invoke(question) response = result['result'] source_docs = result['source_documents'] return response, source_docs