llama-index-instrumentation

import os from llama_index.core.llms import MockLLM from llama_index.core.settings import Settings from llama_index_instrumentation.opentelemetry import OpenTelemetryInstrumentation # OpenTelemetry setup (CRITICAL for traces to be visible) # For a real application, you'd configure an OTLP/Jaeger exporter, etc. # This example prints traces to console. from opentelemetry import trace from opentelemetry.sdk.resources import Resource from opentelemetry.sdk.trace import TracerProvider from opentelemetry.sdk.trace.export import ConsoleSpanExporter, SimpleSpanProcessor resource = Resource.create({"service.name": "my-llama-app"}) provider = TracerProvider(resource=resource) provider.add_span_processor(SimpleSpanProcessor(ConsoleSpanExporter())) trace.set_tracer_provider(provider) # 1. Initialize OpenTelemetry Instrumentation instrumentation = OpenTelemetryInstrumentation(tracer_provider=trace.get_tracer_provider()) # 2. Enable instrumentation instrumentation.enable() # 3. Configure LlamaIndex LLM Settings.llm = MockLLM() # 4. Perform a LlamaIndex operation response = Settings.llm.complete("Tell me a short story about a dragon.") print(f"LLM Response: {response.text[:50]}...") # 5. (Optional) Disable instrumentation when no longer needed instrumentation.disable()