| """OpenTelemetry init hooks (lazy import — opt-in via `--extra obs`). |
| |
| If `opentelemetry-sdk` is not installed, `init_otel()` is a graceful no-op |
| and `is_enabled()` returns False — calls into the package never raise. |
| """ |
|
|
| from __future__ import annotations |
|
|
| import logging |
| import os |
|
|
| logger = logging.getLogger(__name__) |
|
|
|
|
| def is_enabled() -> bool: |
| try: |
| import opentelemetry |
| except ImportError: |
| return False |
| return True |
|
|
|
|
| def init_otel(service_name: str = "mindxtrain", endpoint: str | None = None) -> bool: |
| """Initialize OpenTelemetry tracing if available; return whether init ran.""" |
| if not is_enabled(): |
| logger.info("opentelemetry-sdk not installed; skipping init_otel") |
| return False |
| try: |
| from opentelemetry import trace |
| from opentelemetry.exporter.otlp.proto.http.trace_exporter import OTLPSpanExporter |
| from opentelemetry.sdk.resources import Resource |
| from opentelemetry.sdk.trace import TracerProvider |
| from opentelemetry.sdk.trace.export import BatchSpanProcessor |
| except ImportError as exc: |
| logger.warning("partial opentelemetry install (%s); skipping init", exc) |
| return False |
|
|
| endpoint = endpoint or os.environ.get("MINDXTRAIN_OTEL_ENDPOINT", "") |
| resource = Resource.create({"service.name": service_name}) |
| provider = TracerProvider(resource=resource) |
| if endpoint: |
| provider.add_span_processor(BatchSpanProcessor(OTLPSpanExporter(endpoint=endpoint))) |
| trace.set_tracer_provider(provider) |
| return True |
|
|
|
|
| __all__ = ["init_otel", "is_enabled"] |
|
|
