add memory

app.py

@@ -17,6 +17,7 @@ from langchain.tools import DuckDuckGoSearchRun
 from langchain.utilities import WikipediaAPIWrapper
 from langchain.python import PythonREPL
 from langchain.chains import LLMMathChain
+from langchain.memory import ConversationBufferMemory
 import azure.cognitiveservices.speech as speechsdk
 import requests
 
@@ -47,7 +48,7 @@ from langchain.text_splitter import RecursiveCharacterTextSplitter
 from langchain.docstore.document import Document
 
 
-
+memory = ConversationBufferMemory(memory_key="chat_history")
 # Custom document loaders
 class MyElmLoader(UnstructuredEmailLoader):
     """Wrapper to fallback to text/plain when default does not work"""
@@ -384,6 +385,7 @@ Thought: I now know the final answer
 Final Answer: the final answer to the original input question"""
 
 SUFFIX = """Begin!
+{chat_history}
 Question: {input}
 Thought:{agent_scratchpad}"""
 
@@ -401,8 +403,8 @@ agent = initialize_agent(tools, llm,
                          }
                          )
 
-# print(agent.agent.llm_chain.prompt.template)
-print(agent.agent.llm_chain.prompt)
+print(agent.agent.llm_chain.prompt.template)
+# print(agent.agent.llm_chain.prompt)
 
 global vectordb
 vectordb = Chroma(persist_directory='db', embedding_function=embeddings)