opentelemetry_openai_v2_instrumentation_chat_completions_tracing.py

1import openai
2from opentelemetry import trace
3from opentelemetry.sdk.trace import TracerProvider
4from opentelemetry.sdk.trace.export import ConsoleSpanExporter, BatchSpanProcessor
5from opentelemetry.instrumentation.openai_v2 import OpenAIInstrumentor
6
7# Setup OpenTelemetry
8provider = TracerProvider()
9processor = BatchSpanProcessor(ConsoleSpanExporter())
10provider.add_span_processor(processor)
11trace.set_tracer_provider(provider)
12
13# Instrument OpenAI
14OpenAIInstrumentor().instrument()
15
16# Use the OpenAI client as usual
17client = openai.OpenAI()
18
19response = client.chat.completions.create(
20    model="gpt-3.5-turbo",
21    messages=[{"role": "user", "content": "Tell me a joke"}]
22)
23
24print(response.choices[0].message.content)