Update main.py

main.py

@@ -3,6 +3,7 @@ import os
 import time
 import datetime
 import pandas as pd
+import langsmith as ls
 from pathlib import Path
 import chainlit as cl
 from mistralai.client import MistralClient
@@ -104,6 +105,7 @@ def Connexion_Mistral():
     endpoint = "https://models.inference.ai.azure.com"
     return Mistral(api_key=os.environ["GITHUB_TOKEN"], server_url=endpoint)
 
+@ls.traceable(run_type="llm", name="Connexion à Mistral : paramétrages de la conversation")
 @cl.step(type="tool", show_input=True)
 def Generation_reponse(client, data, question):
     return client.chat.stream(
@@ -445,7 +447,10 @@ async def main(message: cl.Message):
     
         response = Generation_reponse(client, data, message.content)
     
-        msg = await Affichage_reponse(response)
+        app_inputs = {"input": message.content}
+        with ls.trace("Réponse de l'assistant sur les données MIS", "chain", project_name=cl.user_session.get("chat_profile"), inputs=app_inputs) as rt:
+            msg = await Affichage_reponse(response)
+            rt.end(outputs={"output": msg})
     
         answer = msg.content
         await Affichage_question_contexte(answer, message.content)
@@ -461,7 +466,6 @@ async def main(message: cl.Message):
             await cl.Message(author="COPILOT", content="📊 Connexion à l'API France Travail").send()
             df_FT = API_France_Travail_Metier(codeRomeStr)
         
-        print(cl.Starter)
         await cl.Message(author="COPILOT", content="📈 Tableau des emplois les plus représentatifs : " + cl.user_session.get("chat_profile")).send()
         df_intitule = df_FT.groupby('intitule').size().reset_index(name='obs')
         df_intitule = df_intitule.sort_values(by=['obs'], ascending=True)