Update main.py

main.py

@@ -24,6 +24,9 @@ import chainlit as cl
 from chainlit.input_widget import TextInput, Select, Switch, Slider
 from chainlit.playground.config import add_llm_provider
 from chainlit.playground.providers.langchain import LangchainGenericProvider
+from literalai import LiteralClient
+literal_client = LiteralClient(api_key=os.getenv("LITERAL_API_KEY"))
+literal_client.reset_context()
 
 from deep_translator import GoogleTranslator
 
@@ -198,8 +201,9 @@ async def on_message(message: cl.Message):
                     cl.Text(name="Sources", content=sources_text, display="inline")
                 )
         
-    async with cl.Step(type="run", name="Réponse de Mistral"):
-        cb = cl.AsyncLangchainCallbackHandler()
+    async with literal_client.thread(name="Réponse de Mistral") as thread:
+        cb = literal_client.langchain_callback()
+        #cb = cl.AsyncLangchainCallbackHandler()
         with tracing_v2_enabled():
             results = await runnable.acall("Contexte : Vous êtes un chercheur de l'enseignement supérieur et vous êtes doué pour faire des analyses d'articles de recherche sur les thématiques liées à la pédagogie, en fonction des critères définis ci-avant. En fonction des informations suivantes et du contexte suivant seulement et strictement, répondez en langue française strictement à la question ci-dessous à partir du contexte ci-dessous. En plus, tu créeras et tu afficheras 3 questions supplémentaires en relation avec le contexte initial, à chaque étape de la conversation. Tu écriras et tu afficheras les 3 questions supplémentaires en relation avec le contexte initial, avec un titrage de niveau 1 qui a pour titre \"Questions en relation avec le contexte : \". Si vous ne pouvez pas répondre à la question sur la base des informations, dites que vous ne trouvez pas de réponse ou que vous ne parvenez pas à trouver de réponse. Essayez donc de comprendre en profondeur le contexte et répondez uniquement en vous basant sur les informations fournies. Ne générez pas de réponses non pertinentes. Question : " + message.content, callbacks=[cb])
             answer = results["answer"]