deter3 commited on
Commit
e284e34
1 Parent(s): fab3934

Update chat_agent.py

Browse files
Files changed (1) hide show
  1. chat_agent.py +2 -2
chat_agent.py CHANGED
@@ -45,8 +45,8 @@ def create_chatbot(model_name, seed_memory=None):
45
  # messages = [
46
  # SystemMessage(content="You are Martin Seligman. You use a tone that is warm and kind.")
47
  # ]
48
- memory = ConversationBufferMemory(memory_key="chat_history")
49
- #memory = seed_memory if seed_memory is not None else ConversationBufferMemory(memory_key="chat_history")
50
  chat = ChatOpenAI(temperature=0, model_name=model_name)
51
  agent_chain = initialize_agent(tools, chat, agent="conversational-react-description", verbose=True, memory=memory)
52
 
 
45
  # messages = [
46
  # SystemMessage(content="You are Martin Seligman. You use a tone that is warm and kind.")
47
  # ]
48
+ #memory = ConversationBufferMemory(memory_key="chat_history")
49
+ memory = seed_memory if seed_memory is not None else ConversationBufferMemory(memory_key="chat_history")
50
  chat = ChatOpenAI(temperature=0, model_name=model_name)
51
  agent_chain = initialize_agent(tools, chat, agent="conversational-react-description", verbose=True, memory=memory)
52