upgrade memory func

app.py

@@ -18,6 +18,7 @@ from langchain.utilities import WikipediaAPIWrapper
 from langchain.python import PythonREPL
 from langchain.chains import LLMMathChain
 from langchain.memory import ConversationBufferMemory
+from langchain.agents import ZeroShotAgent
 import azure.cognitiveservices.speech as speechsdk
 import requests
 
@@ -386,9 +387,19 @@ Final Answer: the final answer to the original input question"""
 
 SUFFIX = """Begin!
 
+{chat_history}
 Question: {input}
 Thought:{agent_scratchpad}"""
 
+prompt = ZeroShotAgent.create_prompt(
+    tools, 
+    prefix=PREFIX,
+    suffix=SUFFIX,
+    format_instructions=FORMAT_INSTRUCTIONS, 
+    input_variables=["input", "chat_history", "agent_scratchpad"]
+)
+
+
 agent = initialize_agent(tools, llm, 
                         #  agent=AgentType.ZERO_SHOT_REACT_DESCRIPTION,
                          agent=AgentType.ZERO_SHOT_REACT_DESCRIPTION,
@@ -396,12 +407,12 @@ agent = initialize_agent(tools, llm,
                          handle_parsing_errors = True,
                          max_iterations = int(os.getenv("max_iterations")),
                          early_stopping_method="generate",
-                         agent_kwargs={
-                            'prefix': PREFIX,
-                            'format_instructions': FORMAT_INSTRUCTIONS,
-                            'suffix': SUFFIX
-                         },
-                         memory = memory,
+                        #  agent_kwargs={
+                        #     'prefix': PREFIX,
+                        #     'format_instructions': FORMAT_INSTRUCTIONS,
+                        #     'suffix': SUFFIX
+                        #  },
+                         prompt = prompt,
                          )
 
 print(agent.agent.llm_chain.prompt.template)