Gonalb commited on
Commit
25c8513
·
1 Parent(s): 8327038

persist state

Browse files
Files changed (1) hide show
  1. app.py +13 -3
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": [HumanMessage(content=message.content)],
172
  "question": message.content,
173
- "context": []
174
  }
175
 
176
  response = await graph.ainvoke(state)
177
- await cl.Message(content=response["messages"][-1].content).send()
 
 
 
 
 
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()