Update app.py

app.py

@@ -89,21 +89,21 @@ async def on_chat_start():
     cl.user_session.set("runnable", runnable)
 
 
-# @cl.on_message
-# async def on_message(message: cl.Message):
-#     runnable = cl.user_session.get("runnable")  # type: Runnable
-
-#     msg = cl.Message(content="")
+@cl.on_message
+async def on_message(message: cl.Message):
+    runnable = cl.user_session.get("runnable")  # type: Runnable
 
-#     async for chunk in runnable.astream(
-#         {"question": message.content},
-#         config=RunnableConfig(callbacks=[cl.LangchainCallbackHandler()]),
-#     ):
-#         await msg.stream_token(chunk)
+    msg = cl.Message(content="")
 
-#     await msg.send() 
+    async for chunk in runnable.astream(
+        {"question": message.content},
+        config=RunnableConfig(callbacks=[cl.LangchainCallbackHandler()]),
+    ):
+        await msg.stream_token(chunk)
 
-@cl.on_message
-async def main(question):
-    response = await call_chain_async(question.content)
     await cl.Message(content=response).send()
+
+# @cl.on_message
+# async def main(question):
+#     response = await call_chain_async(question.content)
+#     await cl.Message(content=response).send()