File size: 588 Bytes
64203c2 |
1 2 3 4 5 6 7 8 9 10 11 12 13 |
# telemetry.py
from opentelemetry import trace
from opentelemetry.sdk.resources import Resource
from opentelemetry.sdk.trace import TracerProvider
from opentelemetry.exporter.otlp.proto.grpc.trace_exporter import OTLPSpanExporter
from opentelemetry.sdk.trace.export import BatchSpanProcessor
from openllmetry.instrumentation import instrument_agent # <- wraps smolagents
def init():
provider = TracerProvider(resource=Resource.create({"service.name": "ai_sql_agent"}))
provider.add_span_processor(BatchSpanProcessor(OTLPSpanExporter()))
trace.set_tracer_provider(provider)
|