langchain_opentelemetry_tracing_with_openinference_instrumentor.py

1from langchain_openai import OpenAI
2from openinference.instrumentation.langchain import LangChainInstrumentor
3from opentelemetry import trace
4from opentelemetry.exporter.otlp.proto.http.trace_exporter import OTLPSpanExporter
5from opentelemetry.sdk.trace import TracerProvider
6from opentelemetry.sdk.trace.export import BatchSpanProcessor
7
8# Setup OpenTelemetry
9endpoint = "http://127.0.0.1:6006/v1/traces"
10trace_provider = TracerProvider()
11trace_provider.add_span_processor(BatchSpanProcessor(OTLPSpanExporter(endpoint)))
12trace.set_tracer_provider(trace_provider)
13
14# Instrument LangChain
15LangChainInstrumentor().instrument()
16
17# Run your LangChain application
18llm = OpenAI()
19response = llm.invoke("Hello, world!")
20print(response)