Update main.py

main.py

@@ -22,7 +22,6 @@ from IPython.display import display
 
 from literalai import LiteralClient
 literal_client = LiteralClient(api_key=os.getenv("LITERAL_API_KEY"))
-literal_client.instrument_openai()
 
 from surveycaa import surveyCaa
 
@@ -43,7 +42,7 @@ def auth_callback(username: str, password: str):
             identifier=ident + " : 🧑‍🎓 User Datapcc", metadata={"role": "user", "provider": "credentials"}
         )
         
-@literal_client.step(type="run")
+@cl.step(type="run")
 def create_agent(filename: str):
     """
     Create an agent that can access and use a large language model (LLM).
@@ -182,10 +181,8 @@ async def on_message(message: cl.Message):
     agent = create_agent("./public/surveyia.csv")
     cb = cl.AsyncLangchainCallbackHandler()
     try:
-        literal_client.message(content=message.content, type="user_message", name="User")
         #res = await agent.acall("Réponds en langue française à la question suivante : " + message.content, callbacks=[cb])
         res = await agent.ainvoke("Réponds de la manière la plus complète et la plus intelligible, en langue française, à la question suivante : " + message.content + ". Réponds au format markdown ou au format tableau si le résultat nécessite l'affichage d'un tableau.")
-        literal_client.message(content=res['output'], type="assistant_message", name="My Assistant")
         await cl.Message(author="COPILOT",content=GoogleTranslator(source='auto', target='fr').translate(res['output'])).send()
     except ValueError as e:
         res = str(e)