Update main.py

main.py

@@ -50,8 +50,9 @@ def Chargement_des_datas_web(profile):
     fileOpen.close()
     return txt
     
-@cl.step(type="llm", show_input=True)
-def Connexion_Mistral():
+#@cl.step(type="llm", show_input=True)
+@cl.cache
+asyn def Connexion_Mistral():
     return ChatCompletionsClient(
         endpoint="https://models.inference.ai.azure.com",
         credential=AzureKeyCredential(os.environ["GITHUB_TOKEN"]),
@@ -306,14 +307,14 @@ async def chat_profile():
 
 @cl.on_chat_start
 async def on_chat_start():
-    client = Connexion_Mistral()
+    client = await Connexion_Mistral()
     cl.user_session.set("client", client)
 
 @cl.on_message
 async def main(message: cl.Message):
     data = Chargement_des_datas(cl.user_session.get("chat_profile"))
     cl.user_session.set("data", data)
-    cl.user_session.get("client")
+    client = cl.user_session.get("client")
     
     response = Generation_reponse(client, data, message.content)