Update main.py

main.py

@@ -186,25 +186,29 @@ def Generation_completion(client, data, question):
 @traceable(run_type="chain", name="Mistral Assistant des datas Gustaviz",)
 async def Affichage_reponse(response, question, data):
     memory = cl.user_session.get("memory")
-    result = cl.Message(author="COPILOT",content="")
+    msg = cl.Message(author="COPILOT",content="")
+    async for chunk in response.astream({"question": question, "data": data},config=RunnableConfig(callbacks=[cl.AsyncLangchainCallbackHandler(stream_final_answer=True)])):
+        time.sleep(0.125)
+        await msg.stream_token(chunk.replace('Ã','é').replace('©','').replace('Ã','è').replace('¨','').replace('â','\'').replace('€','').replace('™','').replace('Å','oe').replace('“','').replace('®','î').replace('´','ô').replace('<','').replace('>','').replace('/',''))
+    
+    answer = msg.content
     
-    #async for chunk in response.astream({"question": question, "data": data},config=RunnableConfig(callbacks=[cl.AsyncLangchainCallbackHandler(stream_final_answer=True)])):
+    #### Avec Traduction #####
+    #msg = response.invoke({"question": question, "data": data})
+    #msg = msg.replace('Ã','é').replace('©','').replace('Ã','è').replace('¨','').replace('â','\'').replace('€','').replace('™','').replace('Å','oe').replace('“','').replace('®','î').replace('´','ô').replace('<','').replace('>','').replace('/','')
+    #msg = GoogleTranslator(source='auto', target='fr').translate(msg[0:4999])
+    #msgList = msg.split(' ')
+    #for chunk in msgList:
     #    time.sleep(0.125)
-    #    await msg.stream_token(chunk.replace('Ã','é').replace('©','').replace('Ã','è').replace('¨','').replace('â','\'').replace('€','').replace('™','').replace('Å','oe').replace('“','').replace('®','î').replace('´','ô').replace('<','').replace('>','').replace('/',''))
+    #    await result.stream_token(chunk + ' ')
+    #### Avec Traduction #####
 
-    msg = response.invoke({"question": question, "data": data})
-    #msg = msg.replace('Ã','é').replace('©','').replace('Ã','è').replace('¨','').replace('â','\'').replace('€','').replace('™','').replace('Å','oe').replace('“','').replace('®','î').replace('´','ô').replace('<','').replace('>','').replace('/','')
-    msg = GoogleTranslator(source='auto', target='fr').translate(msg[0:4999])
-    msgList = msg.split(' ')
-    for chunk in msgList:
-        time.sleep(0.125)
-        await result.stream_token(chunk + ' ')
-        
-    await result.send()
+    
+    await msg.send()
     memory.chat_memory.add_user_message(question)
-    memory.chat_memory.add_ai_message(msg)
+    memory.chat_memory.add_ai_message(answer)
     
-    return msg
+    return answer
 
 @cl.step(type="tool", show_input=True)
 async def Affichage_question_contexte(answer, question, contexte):