Update main.py
Browse files
main.py
CHANGED
@@ -438,6 +438,8 @@ async def start():
|
|
438 |
temperature=1.0,
|
439 |
streaming=True
|
440 |
)
|
|
|
|
|
441 |
cl.user_session.set("memory", ConversationBufferMemory(return_messages=True))
|
442 |
memory = cl.user_session.get("memory")
|
443 |
prompt = ChatPromptTemplate.from_messages(
|
|
|
438 |
temperature=1.0,
|
439 |
streaming=True
|
440 |
)
|
441 |
+
if not cl.user_session.get("saveMemory"):
|
442 |
+
cl.user_session.set("saveMemory", "")
|
443 |
cl.user_session.set("memory", ConversationBufferMemory(return_messages=True))
|
444 |
memory = cl.user_session.get("memory")
|
445 |
prompt = ChatPromptTemplate.from_messages(
|