Update main.py

main.py

@@ -101,7 +101,7 @@ async def on_action(action):
     ]
     await cl.Message(author="Datapcc 🌐🌐🌐",content="Fermer le panneau d'information", actions=others).send()
 
-@cl.step(type="retrieval")
+@cl.cache
 def retriever_to_cache():
     os.environ['PINECONE_API_KEY'] = os.environ['PINECONE_API_KEY']
     os.environ['PINECONE_ENVIRONMENT'] = "us-west4-gcp-free"
@@ -176,7 +176,7 @@ async def main(message: cl.Message):
     if chatProfile[0] == "OF":
         chain = cl.user_session.get("conversation_chain")
         cb = cl.AsyncLangchainCallbackHandler()
-        res = await chain.acall("Contexte : Réponds à la question suivante de la manière la plus pertinente, la plus exhaustive et la plus détaillée possible, avec au minimum 3000 tokens jusqu'à 4000 tokens, seulement et strictement dans le contexte et les informations fournies. Question : " + message.content, callbacks=[cb])
+        res = await chain.ainvoke("Contexte : Réponds à la question suivante de la manière la plus pertinente, la plus exhaustive et la plus détaillée possible, avec au minimum 3000 tokens jusqu'à 4000 tokens, seulement et strictement dans le contexte et les informations fournies. Question : " + message.content, callbacks=[cb])
         answer = res["answer"]
         source_documents = res["source_documents"]