Update app.py

app.py

@@ -3,7 +3,7 @@ from langchain.vectorstores import Chroma
 from langchain.agents import AgentExecutor, Tool
 from langchain.chains import RetrievalQAWithSourcesChain
 from langchain.chat_models import ChatOpenAI
-from langchain.memory import ConversationBufferMemory
+from langchain.memory import ConversationBufferMemory, ConversationSummaryBufferMemory
 from langchain.utilities import WikipediaAPIWrapper
 from langchain.agents import initialize_agent, AgentType
 
@@ -55,10 +55,10 @@ tools = [
     )
 ]
 
-# Create Buffer Memory
-memory = ConversationBufferMemory(memory_key="chat_history", input_key='input', output_key="output", return_messages=True)
-
-#memory = ConversationSummaryBufferMemory(llm=llm, memory_key="chat_history", input_key='input', output_key="output", return_messages=True)
+# Create Conversational Buffer Memory
+#memory = ConversationBufferMemory(memory_key="chat_history", input_key='input', output_key="output", return_messages=True)
+# Create Conversational Summary Buffer Memory
+memory = ConversationSummaryBufferMemory(llm=llm, memory_key="chat_history", input_key='input', output_key="output", return_messages=True)
 
 # Initialize Re-Act agent and create Agent Executor Chain
 react = initialize_agent(tools, llm, agent=AgentType.CHAT_CONVERSATIONAL_REACT_DESCRIPTION, verbose=True, max_iterations=3, early_stopping_method='generate',  memory=memory)