persist state
Browse files
app.py
CHANGED
|
@@ -163,15 +163,25 @@ compiled_graph = uncompiled_graph.compile()
|
|
| 163 |
@cl.on_chat_start
|
| 164 |
async def start():
|
| 165 |
cl.user_session.set("graph", compiled_graph)
|
|
|
|
| 166 |
|
| 167 |
@cl.on_message
|
| 168 |
async def handle(message: cl.Message):
|
| 169 |
graph = cl.user_session.get("graph")
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 170 |
state = {
|
| 171 |
-
"messages":
|
| 172 |
"question": message.content,
|
| 173 |
-
"context": []
|
| 174 |
}
|
| 175 |
|
| 176 |
response = await graph.ainvoke(state)
|
| 177 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 163 |
@cl.on_chat_start
|
| 164 |
async def start():
|
| 165 |
cl.user_session.set("graph", compiled_graph)
|
| 166 |
+
cl.user_session.set("messages", []) # Inicializar historial de mensajes
|
| 167 |
|
| 168 |
@cl.on_message
|
| 169 |
async def handle(message: cl.Message):
|
| 170 |
graph = cl.user_session.get("graph")
|
| 171 |
+
|
| 172 |
+
# Recuperar historial de mensajes
|
| 173 |
+
messages = cl.user_session.get("messages")
|
| 174 |
+
messages.append(HumanMessage(content=message.content))
|
| 175 |
+
|
| 176 |
state = {
|
| 177 |
+
"messages": messages,
|
| 178 |
"question": message.content,
|
| 179 |
+
"context": []
|
| 180 |
}
|
| 181 |
|
| 182 |
response = await graph.ainvoke(state)
|
| 183 |
+
|
| 184 |
+
# Guardar historial actualizado en la sesión
|
| 185 |
+
cl.user_session.set("messages", state["messages"])
|
| 186 |
+
|
| 187 |
+
await cl.Message(content=response["messages"][-1].content).send()
|