trace.py

import openai
from opentelemetry import trace

tracer = trace.get_tracer("simple_chatbot.example")

with tracer.start_as_current_span("query_llm") as span:
    response = openai.ChatCompletion.create(
        model="gpt-4",
        messages=[
            {"role": "user", "content": "TODO"}
        ]
    )
    result = response["choices"][0]["message"]["content"]