Update main.py

main.py

@@ -15,7 +15,7 @@ from langchain.schema import StrOutputParser
 from langchain.chains.qa_with_sources import load_qa_with_sources_chain
 from langchain.vectorstores import Pinecone
 import pinecone
-from langchain.memory import ChatMessageHistory, ConversationBufferMemory, ConversationBufferWindowMemory
+from langchain.memory import ChatMessageHistory, ConversationBufferMemory
 import pandas as pd
 import numpy as np
 from anthropic import Anthropic, HUMAN_PROMPT, AI_PROMPT
@@ -155,8 +155,7 @@ async def start():
         
         ########## Chain with streaming ##########
         message_history = ChatMessageHistory()
-        memory = ConversationBufferWindowMemory(
-            k=1,
+        memory = ConversationBufferMemory(
             memory_key="chat_history",
             output_key="answer",
             chat_memory=message_history,