Modularize instrumentation

python/examples/llama-index/backend/instrument.py

@@ -0,0 +1,16 @@
+from openinference.instrumentation.llama_index import LlamaIndexInstrumentor
+from opentelemetry import trace as trace_api
+from opentelemetry.exporter.otlp.proto.http.trace_exporter import OTLPSpanExporter
+from opentelemetry.sdk import trace as trace_sdk
+from opentelemetry.sdk.resources import Resource
+from opentelemetry.sdk.trace.export import SimpleSpanProcessor
+
+
+def instrument():
+    resource = Resource(attributes={})
+    tracer_provider = trace_sdk.TracerProvider(resource=resource)
+    span_exporter = OTLPSpanExporter(endpoint="http://phoenix:6006/v1/traces")
+    span_processor = SimpleSpanProcessor(span_exporter=span_exporter)
+    tracer_provider.add_span_processor(span_processor=span_processor)
+    trace_api.set_tracer_provider(tracer_provider=tracer_provider)
+    LlamaIndexInstrumentor().instrument()

python/examples/llama-index/backend/main.py

@@ -10,21 +10,13 @@
 from fastapi import FastAPI
 from fastapi.middleware.cors import CORSMiddleware
 
-from openinference.instrumentation.llama_index import LlamaIndexInstrumentor
-from opentelemetry import trace as trace_api
-from opentelemetry.exporter.otlp.proto.http.trace_exporter import OTLPSpanExporter
-from opentelemetry.sdk import trace as trace_sdk
-from opentelemetry.sdk.resources import Resource
-from opentelemetry.sdk.trace.export import SimpleSpanProcessor
-
-resource = Resource(attributes={})
-tracer_provider = trace_sdk.TracerProvider(resource=resource)
-span_exporter = OTLPSpanExporter(endpoint="http://127.0.0.1:6006/v1/traces")
-span_processor = SimpleSpanProcessor(span_exporter=span_exporter)
-tracer_provider.add_span_processor(span_processor=span_processor)
-trace_api.set_tracer_provider(tracer_provider=tracer_provider)
-app = FastAPI()
+from instrument import instrument
+
+do_not_instrument = os.getenv("INSTRUMENT_LLAMA_INDEX", "true") == "false"
+if not do_not_instrument:
+    instrument()
 
+app = FastAPI()
 environment = os.getenv("ENVIRONMENT", "dev")  # Default to 'development' if not set
 
 
@@ -43,5 +35,4 @@
 
 
 if __name__ == "__main__":
-    LlamaIndexInstrumentor().instrument()
-    uvicorn.run(app="main:app", host="0.0.0.0", reload=True)
+    uvicorn.run(app="main:app", host="0.0.0.0", port=8000, reload=True)