Get informations from langsmith

[aurelienpirot]

Hello, I would like to use a python script (langchain) to retrieve the information created in langsmith when calling an open AI model.

this my script:

import openai
from langsmith import wrappers, traceable
from dotenv import load_dotenv
import os

Charger les variables d'environnement depuis le fichier .env

load_dotenv()

Récupérer la clé API OpenAI

openai_api_key = os.getenv("OPENAI_API_KEY")
if not openai_api_key:
raise ValueError("La clé API OpenAI n'est pas définie. Assurez-vous que le fichier .env contient 'OPENAI_API_KEY'.")

Initialiser le client OpenAI avec la clé

openai.api_key = openai_api_key

Auto-trace LLM calls in-context

client = wrappers.wrap_openai(openai.Client())
@Traceable # Auto-trace this function
def pipeline(user_input: str):
# Effectuer un appel au modèle GPT
result = client.chat.completions.create(
messages=[{"role": "user", "content": user_input}],
model="gpt-4o-mini"
)
# Afficher le contenu du résultat
print("Réponse du modèle :", result.choices[0].message.content)
# Récupérer les métadonnées de LangSmith directement depuis l'objet
trace_info = {
"start_time": getattr(result, "start_time", None), # Utiliser getattr pour éviter les erreurs si l'attribut est absent
"end_time": getattr(result, "end_time", None),
"total_tokens": getattr(result.usage, "total_tokens", None), # Vérifier si result.usage est défini
"prompt_tokens": getattr(result.usage, "prompt_tokens", None),
"completion_tokens": getattr(result.usage, "completion_tokens", None),
}
print("Informations LangSmith :", trace_info)
return result.choices[0].message.content
pipeline("Hello, world!")

i have a new run created in langsmith
but i don't have start_time, end_time, run id, trace id in the response