Update app.py

app.py

@@ -42,8 +42,8 @@ from langchain import PromptTemplate
 
 from langchain.chains import RetrievalQA
 #from langchain.memory import ConversationBufferWindowMemory
-#from langchain.memory import ConversationBufferMemory
-from langchain.memory import ConversationMemory
+from langchain.memory import ConversationBufferMemory
+
 
 from transformers import LlamaTokenizer, AutoTokenizer
 
@@ -212,7 +212,7 @@ def getRAGChain(customerName, customerDistrict, custDetailsPresent, vectordb,llm
     global memory 
     #memory = ConversationBufferWindowMemory(k=3, memory_key="history", input_key="question", initial_memory=conversation_history)
     #memory = ConversationBufferMemory(k=3, memory_key="history", input_key="query", initial_memory=conversation_history)
-    memory = ConversationMemory()
+    memory = ConversationBufferMemory()
     
     # chain = RetrievalQA.from_chain_type(
     #     llm=getLLMModel(llmID),