Update main.py

main.py

@@ -83,6 +83,9 @@ async def on_message(message: cl.Message):
     
     msg = cl.Message(content="")
 
+    async for chunk in chain.astream({"input_question": message.content},
+                config=RunnableConfig(callbacks=[cl.AsyncLangchainCallbackHandler(stream_final_answer=True)])):
+            await msg.stream_token(chunk)
     
     
     #cb = cl.AsyncLangchainCallbackHandler()