mgbam commited on
Commit
64203c2
·
verified ·
1 Parent(s): 123bf53

Create telemetry.py

Browse files
Files changed (1) hide show
  1. telemetry.py +12 -0
telemetry.py ADDED
@@ -0,0 +1,12 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # telemetry.py
2
+ from opentelemetry import trace
3
+ from opentelemetry.sdk.resources import Resource
4
+ from opentelemetry.sdk.trace import TracerProvider
5
+ from opentelemetry.exporter.otlp.proto.grpc.trace_exporter import OTLPSpanExporter
6
+ from opentelemetry.sdk.trace.export import BatchSpanProcessor
7
+ from openllmetry.instrumentation import instrument_agent # <- wraps smolagents
8
+
9
+ def init():
10
+ provider = TracerProvider(resource=Resource.create({"service.name": "ai_sql_agent"}))
11
+ provider.add_span_processor(BatchSpanProcessor(OTLPSpanExporter()))
12
+ trace.set_tracer_provider(provider)