Spaces:
Sleeping
Sleeping
Update chat_agent.py
Browse files- 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 |
-
|
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 |
|