Update main.py

main.py

@@ -76,32 +76,8 @@ async def on_action(action):
         author="Datapcc 🌐🌐🌐", content="[Lien] 🔗", elements=elements
     ).send()
     await action.remove()
-    
-@cl.action_callback("close_button")
-async def on_action(action):
-    time.sleep(0.5)
-    track = user_session.get("tracker")
-    await track.remove()
-        
-@cl.action_callback("action_button")
-async def on_action(action):
-    task_list = cl.TaskList()
-    # Create the TaskList
-    # Create a task and put it in the running state
-    task1 = cl.Task(title="Processing data Processing data Processing data Processing data Processing data Processing data Processing data Processing data Processing data Processing data Processing data Processing data Processing data Processing data Processing data Processing data Processing data Processing data Processing data Processing data Processing data \n\n Processing data", status=cl.TaskStatus.READY)
-    await task_list.add_task(task1)
-    task2 = cl.Task(title=action.value, status=cl.TaskStatus.READY)
-    await task_list.add_task(task2)
-    # Perform some action on your end
-    await task_list.send()
-    tracking = user_session.set("tracker", task_list)
-    
-    others = [
-        cl.Action(name="close_button", value="closed", label="Fermer", description="Fermer le volet d'information!")
-    ]
-    await cl.Message(author="Datapcc 🌐🌐🌐",content="Fermer le panneau d'information", actions=others).send()
 
-@cl.cache
+@cl.step(type="retrieval")
 def retriever_to_cache():
     os.environ['PINECONE_API_KEY'] = os.environ['PINECONE_API_KEY']
     os.environ['PINECONE_ENVIRONMENT'] = "us-west4-gcp-free"
@@ -147,6 +123,7 @@ async def start():
         CONDENSE_QUESTION_PROMPT = PromptTemplate.from_template(_template)
         
         ########## Chain with streaming ##########
+        retrieval = wait retriever_to_cache()
         message_history = ChatMessageHistory()
         memory = ConversationBufferMemory(
             memory_key="chat_history",
@@ -165,7 +142,7 @@ async def start():
             chain_type="stuff",
             return_source_documents=True,
             verbose=False,
-            retriever=retriever_to_cache()
+            retriever=retrieval
         )
         cl.user_session.set("conversation_chain", qa)