Update main.py

main.py

@@ -3,6 +3,8 @@ import os
 import time
 import datetime
 import pandas as pd
+from langsmith import traceable
+import uuid
 import langsmith as ls
 from pathlib import Path
 import chainlit as cl
@@ -11,6 +13,7 @@ from mistralai import Mistral, UserMessage, SystemMessage
 from offres_emploi import Api
 from offres_emploi.utils import dt_to_str_iso
 os.environ["GITHUB_TOKEN"] = os.environ["GITHUB_TOKEN"]
+session_id = str(uuid.uuid4())
 
 @cl.step(type="tool", show_input=True)
 @ls.traceable(run_type="tool", name="Retrieve Context : chargement des données Gustaviz")
@@ -134,6 +137,7 @@ def Generation_completion(client, data, question):
     return msg
    
 @cl.step(type="tool", show_input=True)
+@traceable(run_type="chain", name="Mistral Assistant des datas Gustaviz",)
 async def Affichage_reponse(response):
     msg = cl.Message(author="COPILOT",content="")
     
@@ -450,7 +454,7 @@ async def main(message: cl.Message):
     
         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)
+            msg = await Affichage_reponse(response, langsmith_extra={"metadata": {"session_id": session_id}})
             rt.end(outputs={"output": msg})
     
         answer = msg.content