Update main.py

main.py

@@ -247,7 +247,8 @@ async def main(message: cl.Message):
         task1 = cl.Task(title="Connexion à la base documentaire, en attente...", status=cl.TaskStatus.RUNNING)
         await pineconeRetriever.add_task(task1)
         await pineconeRetriever.send()
-        
+        message_id1 = await cl.Message(content="Connexion en cours").send()
+        task1.forId = message_id1
         index_name = os.environ['PINECONE_INDEX_NAME']
         pinecone.init(
             api_key=os.environ['PINECONE_API_KEY'],
@@ -255,15 +256,19 @@ async def main(message: cl.Message):
         )
         embeddings = HuggingFaceEmbeddings()
         task1.status = cl.TaskStatus.DONE
-        task2 = cl.Task(title="Recherche par cosine similarity, en attente...", status=cl.TaskStatus.RUNNING)
+        task2 = cl.Task(title="Recherche sémantique par cosine similarity, en attente...", status=cl.TaskStatus.RUNNING)
         await pineconeRetriever.add_task(task2)
         await pineconeRetriever.send()
+        message_id2 = await cl.Message(content="Recherche sémantique en cours").send()
+        task2.forId = message_id2
         docsearch = Pinecone.from_existing_index(index_name, embeddings)
         retrieve = docsearch.similarity_search(message.content, k=40, filter={'categorie': {'$eq': 'OF'}})
         task2.status = cl.TaskStatus.DONE
         task3 = cl.Task(title="Création du contexte documentaire, en attente...", status=cl.TaskStatus.RUNNING)
         await pineconeRetriever.add_task(task3)
         await pineconeRetriever.send()        
+        message_id3 = await cl.Message(content="Création du contexte en cours").send()
+        task3.forId = message_id3
         context = ''
         metadatas = ''
         text_elements = []
@@ -276,6 +281,7 @@ async def main(message: cl.Message):
             context += '\n' + retrieve[i].page_content
         task3.status = cl.TaskStatus.DONE
         await cl.sleep(20)
+        await pineconeRetriever.send()
         await pineconeRetriever.remove()   
         memory = cl.user_session.get("memory")
         runnable = cl.user_session.get("runnable")  # type: Runnable