getting last 4 messages

utils.py

@@ -39,7 +39,7 @@ message_history = CustomMongoDBChatMessageHistory(
     collection_name='3d_printing_revolution'
 )
 
-memory = ConversationBufferWindowMemory(memory_key="chat_history", k=10)
+memory = ConversationBufferWindowMemory(memory_key="chat_history", k=4)
 
 vectorstore_index = None
 
@@ -210,15 +210,11 @@ def get_chat_history(inputs) -> str:
 def generate_answer(question) -> str:
     global vectorstore_index
     chain = get_qa_chain(vectorstore_index)
-    history = memory.chat_memory.messages
-    print("chat history: " + str(history))
+    history = memory.chat_memory.messages[-4:]
     result = chain(
         {"question": question, "chat_history": history})
-    # chat_tuple = (question, result["answer"])
-    # chat_history.append(chat_tuple)
     save_chat_history(question, result)
-    chat_history = [message.content for message in memory.chat_memory.messages]
-    print("chat history after response: " + str(chat_history))
+
     sources = []
     print(result)
 
@@ -234,4 +230,4 @@ def generate_answer(question) -> str:
 def save_chat_history(question, result):
     memory.chat_memory.add_user_message(question)
     memory.chat_memory.add_ai_message(result["answer"])
-    print("chat history on saving: " + str(memory.chat_memory.messages))
+    print("chat history after saving: " + str(memory.chat_memory.messages))