async 1st

app.py

@@ -107,21 +107,26 @@ async def start_chat():
 async def main(message: cl.Message):
     chain = cl.user_session.get("chain")
 
-    resp = chain.invoke({"question" : message.content})
+    resp = await chain.ainvoke({"question" : message.content})
     source_documents = resp["context"] 
 
     text_elements = []  # type: List[cl.Text]
 
     resp_msg = resp["response"].content
 
+    #print(source_documents)
+
     if source_documents:
         for source_idx, source_doc in enumerate(source_documents):
             source_name = f"source_{source_idx}"
 
             # Create the text element referenced in the message
             text_elements.append(
-                cl.Text(content=source_doc[0].page_content, name="{} (scr: {})".format(source_name, round(source_doc[1],2)), display="side")
+                cl.Text(content=source_doc.page_content, name="{}".format(source_name), display="side")
             )
+            # text_elements.append(
+            #    cl.Text(content=source_doc.page_content, name="{} (scr: {})".format(source_name, round(source_doc[1],2)), display="side")
+            #)
         source_names = [text_el.name for text_el in text_elements]
 
     if source_names:
@@ -131,7 +136,7 @@ async def main(message: cl.Message):
 
     msg = cl.Message(content=resp_msg, elements=text_elements)
 
-    print(msg.content)
+    #print(msg.content)
     await msg.send()