streaming

app.py

@@ -180,8 +180,21 @@ async def start():
 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):
-        last_message = partial_state["messages"][-1]
-        await cl.Message(content=last_message.content).send()
+        # 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()