Another try

app.py

@@ -85,54 +85,54 @@ async def on_chat_start():
     cl.user_session.set("runnable_chain", runnable_chain)
 
 
-@cl.on_message
-async def on_message(message: cl.Message):
-    runnable_chain = cl.user_session.get("runnable_chain")
-
-    msg = cl.Message(content="")
-
-    async for chunk in runnable_chain.astream(
-            message.content,
-            config=RunnableConfig(callbacks=[cl.LangchainCallbackHandler()]),
-    ):
-        await msg.stream_token(chunk)
-
-    await msg.send()
+# @cl.on_message
+# async def on_message(message: cl.Message):
+#     runnable_chain = cl.user_session.get("runnable_chain")
+#
+#     msg = cl.Message(content="")
+#
+#     async for chunk in runnable_chain.astream(
+#             message.content,
+#             config=RunnableConfig(callbacks=[cl.LangchainCallbackHandler()]),
+#     ):
+#         await msg.stream_token(chunk)
+#
+#     await msg.send()
 
 @cl.on_message
 async def on_message(message: cl.Message):
     runnable = cl.user_session.get("runnable_chain")
     msg = cl.Message(content="")
 
-    class PostMessageHandler(BaseCallbackHandler):
-        """
-        Callback handler for handling the retriever and LLM processes.
-        Used to post the sources of the retrieved documents as a Chainlit element.
-        """
-
-        def __init__(self, msg: cl.Message):
-            BaseCallbackHandler.__init__(self)
-            self.msg = msg
-            self.sources = set()  # To store unique pairs
-
-        def on_retriever_end(self, documents, *, run_id, parent_run_id, **kwargs):
-            for d in documents:
-                source_page_pair = (d.metadata['source'], d.metadata['page'])
-                self.sources.add(source_page_pair)  # Add unique pairs to the set
-
-        def on_llm_end(self, response, *, run_id, parent_run_id, **kwargs):
-            if len(self.sources):
-                sources_text = "\n".join([f"{source}#page={page}" for source, page in self.sources])
-                self.msg.elements.append(
-                    cl.Text(name="Sources", content=sources_text, display="inline")
-                )
+    # class PostMessageHandler(BaseCallbackHandler):
+    #     """
+    #     Callback handler for handling the retriever and LLM processes.
+    #     Used to post the sources of the retrieved documents as a Chainlit element.
+    #     """
+    #
+    #     def __init__(self, msg: cl.Message):
+    #         BaseCallbackHandler.__init__(self)
+    #         self.msg = msg
+    #         self.sources = set()  # To store unique pairs
+    #
+    #     def on_retriever_end(self, documents, *, run_id, parent_run_id, **kwargs):
+    #         for d in documents:
+    #             source_page_pair = (d.metadata['source'], d.metadata['page'])
+    #             self.sources.add(source_page_pair)  # Add unique pairs to the set
+    #
+    #     def on_llm_end(self, response, *, run_id, parent_run_id, **kwargs):
+    #         if len(self.sources):
+    #             sources_text = "\n".join([f"{source}#page={page}" for source, page in self.sources])
+    #             self.msg.elements.append(
+    #                 cl.Text(name="Sources", content=sources_text, display="inline")
+    #             )
 
     async with cl.Step(type="run", name="Movie Assistant"):
         async for chunk in runnable.astream(
             message.content,
             config=RunnableConfig(callbacks=[
                 cl.LangchainCallbackHandler(),
-                PostMessageHandler(msg)
+                # PostMessageHandler(msg)
             ]),
         ):
             await msg.stream_token(chunk)