varios fixes

.chainlit/config.toml

@@ -33,7 +33,7 @@ edit_message = true
 
 # Authorize users to spontaneously upload files with messages
 [features.spontaneous_file_upload]
-    enabled = true
+    enabled = false
     # Define accepted file types using MIME types
     # Examples:
     # 1. For specific file types:

app.py

@@ -178,7 +178,10 @@ async def start():
 
 @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)]}
+    
+    async for partial_state in graph.astream(state):
+        last_message = partial_state["messages"][-1]
+        await cl.Message(content=last_message.content).send()
+