Update app.py

app.py

@@ -5,7 +5,7 @@ from langchain_community.embeddings import HuggingFaceEmbeddings
 from langchain.chains import RetrievalQA
 from langchain_google_genai import GoogleGenerativeAI
 from langchain.prompts import PromptTemplate
-from langchain.chains import load_qa_chain, RetrievalQA
+#from langchain.chains import load_qa_chain, RetrievalQA
 import requests
 from bs4 import BeautifulSoup
 from urllib.parse import urljoin
@@ -110,12 +110,12 @@ if query and st.session_state.documents_loaded:
         # Create a PromptTemplate for the QA chain
         qa_prompt = PromptTemplate(template="Answer the following question based on the context provided:\n\n{context}\n\nQuestion: {question}\nAnswer:", input_variables=["context", "question"])
 
-        # Load the QA chain
-        qa_chain = load_qa_chain(
-            prompt=qa_prompt,
+        # Create the retrieval QA chain
+        qa_chain = RetrievalQA.from_chain_type(
             retriever=st.session_state.vector_store.as_retriever(),
+            chain_type="map_reduce",
             llm=llm,
-            combine_documents_chain=True
+            prompt=qa_prompt
         )
 
         response = qa_chain({"question": query})
@@ -125,4 +125,3 @@ if query and st.session_state.documents_loaded:
         st.write(response['source'])
 elif query:
     st.warning("Please crawl the CUDA documentation first.")
-