phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-12-04 10:10:36 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix(telemetry): remove dependency on old telemetry config

Browse source 
# What does this PR do?
old telemetry config was removed in #3815

## Test Plan

❯ OTEL_SERVICE_NAME=aloha OTEL_EXPORTER_OTLP_ENDPOINT=http://localhost:4318 uv run llama stack run starter
This commit is contained in:
Eric Huang 2025-10-16 11:24:55 -07:00
parent 3679612b5f
commit 38976b5ac1
1 changed files with 12 additions and 30 deletions
Download patch file Download diff file Expand all files Collapse all files

42
llama_stack/providers/inline/telemetry/meta_reference/telemetry.py
View file

@ -55,17 +55,10 @@ def is_tracing_enabled(tracer):
class TelemetryAdapter(Telemetry): class TelemetryAdapter(Telemetry):
def __init__(self, config: TelemetryConfig, deps: dict[Api, Any]) -> None: def __init__(self, _config: TelemetryConfig, deps: dict[Api, Any]) -> None:
self.config = config
self.datasetio_api = deps.get(Api.datasetio) self.datasetio_api = deps.get(Api.datasetio)
self.meter = None self.meter = None
resource = Resource.create(
{
ResourceAttributes.SERVICE_NAME: self.config.service_name,
}
)
global _TRACER_PROVIDER global _TRACER_PROVIDER
# Initialize the correct span processor based on the provider state. # Initialize the correct span processor based on the provider state.
# This is needed since once the span processor is set, it cannot be unset. # This is needed since once the span processor is set, it cannot be unset.
@ -73,35 +66,24 @@ class TelemetryAdapter(Telemetry):
# Since the library client can be recreated multiple times in a notebook, # Since the library client can be recreated multiple times in a notebook,
# the kernel will hold on to the span processor and cause duplicate spans to be written. # the kernel will hold on to the span processor and cause duplicate spans to be written.
if _TRACER_PROVIDER is None: if _TRACER_PROVIDER is None:
provider = TracerProvider(resource=resource) provider = TracerProvider()
trace.set_tracer_provider(provider) trace.set_tracer_provider(provider)
_TRACER_PROVIDER = provider _TRACER_PROVIDER = provider
# Use single OTLP endpoint for all telemetry signals # Use single OTLP endpoint for all telemetry signals
if TelemetrySink.OTEL_TRACE in self.config.sinks or TelemetrySink.OTEL_METRIC in self.config.sinks:
if self.config.otel_exporter_otlp_endpoint is None:
raise ValueError(
"otel_exporter_otlp_endpoint is required when OTEL_TRACE or OTEL_METRIC is enabled"
)
# Let OpenTelemetry SDK handle endpoint construction automatically # Let OpenTelemetry SDK handle endpoint construction automatically
# The SDK will read OTEL_EXPORTER_OTLP_ENDPOINT and construct appropriate URLs # The SDK will read OTEL_EXPORTER_OTLP_ENDPOINT and construct appropriate URLs
# https://opentelemetry.io/docs/languages/sdk-configuration/otlp-exporter # https://opentelemetry.io/docs/languages/sdk-configuration/otlp-exporter
if TelemetrySink.OTEL_TRACE in self.config.sinks: span_exporter = OTLPSpanExporter()
span_exporter = OTLPSpanExporter() span_processor = BatchSpanProcessor(span_exporter)
span_processor = BatchSpanProcessor(span_exporter) trace.get_tracer_provider().add_span_processor(span_processor)
trace.get_tracer_provider().add_span_processor(span_processor)
if TelemetrySink.OTEL_METRIC in self.config.sinks: metric_reader = PeriodicExportingMetricReader(OTLPMetricExporter())
metric_reader = PeriodicExportingMetricReader(OTLPMetricExporter()) metric_provider = MeterProvider(metric_readers=[metric_reader])
metric_provider = MeterProvider(resource=resource, metric_readers=[metric_reader]) metrics.set_meter_provider(metric_provider)
metrics.set_meter_provider(metric_provider)
if TelemetrySink.CONSOLE in self.config.sinks: self.meter = metrics.get_meter(__name__)
trace.get_tracer_provider().add_span_processor(ConsoleSpanProcessor(print_attributes=True))
if TelemetrySink.OTEL_METRIC in self.config.sinks:
self.meter = metrics.get_meter(__name__)
self._lock = _global_lock self._lock = _global_lock