mirror of
https://github.com/meta-llama/llama-stack.git
synced 2025-12-07 02:47:21 +00:00
Merge 0ef4f1cd6b into sapling-pr-archive-ehhuang
This commit is contained in:
commit
5da5fe2983
1 changed files with 4 additions and 1 deletions
|
|
@ -79,8 +79,10 @@ class TelemetryAdapter(Telemetry):
|
||||||
metric_reader = PeriodicExportingMetricReader(OTLPMetricExporter())
|
metric_reader = PeriodicExportingMetricReader(OTLPMetricExporter())
|
||||||
metric_provider = MeterProvider(metric_readers=[metric_reader])
|
metric_provider = MeterProvider(metric_readers=[metric_reader])
|
||||||
metrics.set_meter_provider(metric_provider)
|
metrics.set_meter_provider(metric_provider)
|
||||||
|
self.is_otel_endpoint_set = True
|
||||||
else:
|
else:
|
||||||
logger.warning("OTEL_EXPORTER_OTLP_ENDPOINT is not set, skipping telemetry")
|
logger.warning("OTEL_EXPORTER_OTLP_ENDPOINT is not set, skipping telemetry")
|
||||||
|
self.is_otel_endpoint_set = False
|
||||||
|
|
||||||
self.meter = metrics.get_meter(__name__)
|
self.meter = metrics.get_meter(__name__)
|
||||||
self._lock = _global_lock
|
self._lock = _global_lock
|
||||||
|
|
@ -89,7 +91,8 @@ class TelemetryAdapter(Telemetry):
|
||||||
pass
|
pass
|
||||||
|
|
||||||
async def shutdown(self) -> None:
|
async def shutdown(self) -> None:
|
||||||
trace.get_tracer_provider().force_flush()
|
if self.is_otel_endpoint_set:
|
||||||
|
trace.get_tracer_provider().force_flush()
|
||||||
|
|
||||||
async def log_event(self, event: Event, ttl_seconds: int = 604800) -> None:
|
async def log_event(self, event: Event, ttl_seconds: int = 604800) -> None:
|
||||||
if isinstance(event, UnstructuredLogEvent):
|
if isinstance(event, UnstructuredLogEvent):
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue