Update chat_agent.py

chat_agent.py

@@ -1,5 +1,5 @@
 from langchain.agents import Tool
-from langchain.memory import ConversationBufferMemory
+from langchain.memory import ConversationBufferMemory ,ConversationBufferWindowMemory
 from langchain.chat_models import ChatOpenAI
 from langchain.agents import initialize_agent
 from llama_index import GPTSimpleVectorIndex
@@ -37,7 +37,7 @@ def create_chatbot(model_name, seed_memory=None):
         Tool(
             name="GPT Index",
             func=lambda q: str(index.query(q,vector_store_query_mode=query_mode)),
-            description="useful for when you want to answer questions about Martin Seligman and psychonogy. The input to this tool should be a complete english sentence.",
+            description="useful for when you want to answer questions about Martin Seligman and positive psychonogy related. The input to this tool should be a complete english sentence.",
             return_direct=True
         ),
     ]
@@ -46,7 +46,8 @@ def create_chatbot(model_name, seed_memory=None):
     #     SystemMessage(content="You are Martin Seligman. You use a tone that is warm and kind.")
     # ]
     #memory = ConversationBufferMemory(memory_key="chat_history")
-    memory = seed_memory if seed_memory is not None else ConversationBufferMemory(memory_key="chat_history")
+    memory = seed_memory if seed_memory is not None else ConversationBufferWindowMemory( k=4 ,memory_key="chat_history")
+    #memory = seed_memory if seed_memory is not None else ConversationBufferMemory(memory_key="chat_history")
     chat = ChatOpenAI(temperature=0, model_name=model_name)
     agent_chain = initialize_agent(tools, chat, agent="conversational-react-description", verbose=True, memory=memory)