add converstation limit

app.py

@@ -10,6 +10,7 @@ import os
 from langchain import OpenAI
 from langchain.chains import ConversationChain
 from langchain.memory import ConversationSummaryBufferMemory
+from langchain.memory import ConversationBufferWindowsMemory
 from langchain.chat_models import ChatOpenAI
 from langchain.callbacks.streaming_stdout import StreamingStdOutCallbackHandler
 from langchain.schema import HumanMessage
@@ -54,7 +55,8 @@ s3 = boto3.client('s3')
 transcribe = boto3.client('transcribe')
 
 
-memory = ConversationSummaryBufferMemory(llm=ChatOpenAI(), max_token_limit=2048, k=3)
+#memory = ConversationSummaryBufferMemory(llm=ChatOpenAI(), max_token_limit=2048)
+memory = ConversationBufferWindowsMemory(k=5)
 conversation = ConversationChain(
     llm=OpenAI(streaming=True, callbacks=[StreamingStdOutCallbackHandler()], max_tokens=2048, temperature=0.5), 
     memory=memory,