demoPOC commited on
Commit
7ccd486
·
verified ·
1 Parent(s): ab64196

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -210,7 +210,7 @@ def getRAGChain(customerName, customerDistrict, custDetailsPresent, vectordb,llm
210
  #memory = ConversationBufferWindowMemory(k=3, memory_key="history", input_key="question")
211
  global memory
212
  #memory = ConversationBufferWindowMemory(k=3, memory_key="history", input_key="question", initial_memory=conversation_history)
213
- memory = ConversationBufferMemory(k=3, memory_key="history", input_key="question", initial_memory=conversation_history)
214
 
215
  # chain = RetrievalQA.from_chain_type(
216
  # llm=getLLMModel(llmID),
 
210
  #memory = ConversationBufferWindowMemory(k=3, memory_key="history", input_key="question")
211
  global memory
212
  #memory = ConversationBufferWindowMemory(k=3, memory_key="history", input_key="question", initial_memory=conversation_history)
213
+ memory = ConversationBufferMemory(k=3, memory_key="history", input_key="query", initial_memory=conversation_history)
214
 
215
  # chain = RetrievalQA.from_chain_type(
216
  # llm=getLLMModel(llmID),