on_message

app.py

@@ -178,23 +178,7 @@ async def start():
 
 @cl.on_message
 async def handle(message: cl.Message):
-    graph = cl.user_session.get("graph")
-    state = {"messages": [HumanMessage(content=message.content)]}
-
-    msg = cl.Message(content="")  # Crear un mensaje vacío para el streaming
-    await msg.send()  # Enviar el mensaje vacío inicial
-
-    async for partial_state in graph.astream(state):
-        # Verificar si "messages" está en el estado antes de acceder a él
-        if "messages" in partial_state and partial_state["messages"]:
-            last_message = partial_state["messages"][-1]
-            await msg.update(content=last_message.content)
-
-
-
-# @cl.on_message
-# async def handle(message: cl.Message):
-#   graph = cl.user_session.get("graph")
-#   state = {"messages" : [HumanMessage(content=message.content)]}
-#   response = await graph.ainvoke(state)
-#   await cl.Message(content=response["messages"][-1].content).send()
+  graph = cl.user_session.get("graph")
+  state = {"messages" : [HumanMessage(content=message.content)]}
+  response = await graph.ainvoke(state)
+  await cl.Message(content=response["messages"][-1].content).send()