RoAr777 commited on
Commit
3b80d25
·
verified ·
1 Parent(s): 9920d46

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -370,8 +370,9 @@ llm = HuggingFaceEndpoint(
370
  timeout=10
371
  )
372
  from langchain.memory import ConversationBufferWindowMemory
 
373
 
374
- llm_chain = LLMChain(prompt=prompt, llm=llm, memory=ConversationBufferWindowMemory(k=20))
375
 
376
  agent_tools = [ipc_tool, crpc_tool, doj_tool, faq_tool]
377
 
 
370
  timeout=10
371
  )
372
  from langchain.memory import ConversationBufferWindowMemory
373
+ from langchain.memory import ChatMessageHistory
374
 
375
+ llm_chain = LLMChain(prompt=prompt, llm=llm, memory=ChatMessageHistory())
376
 
377
  agent_tools = [ipc_tool, crpc_tool, doj_tool, faq_tool]
378