otel

Author: pamelafox

src/fastapi_app/__init__.py

@@ -6,6 +6,7 @@
 from dotenv import load_dotenv
 from environs import Env
 from fastapi import FastAPI
+from opentelemetry.instrumentation.fastapi import FastAPIInstrumentor
 
 from .globals import global_storage
 from .openai_clients import create_openai_chat_client, create_openai_embed_client
@@ -63,6 +64,9 @@ def create_app():
 
     app = FastAPI(docs_url="/docs", lifespan=lifespan)
 
+    if os.getenv("APPLICATIONINSIGHTS_CONNECTION_STRING"):
+        FastAPIInstrumentor.instrument_app(app)
+
     from . import api_routes  # noqa
     from . import frontend_routes  # noqa
 

src/pyproject.toml

@@ -14,6 +14,7 @@ dependencies = [
     "openai>=1.34.0,<2.0.0",
     "tiktoken>=0.7.0,<0.8.0",
     "openai-messages-token-helper>=0.1.5,<0.2.0",
+    "opentelemetry-instrumentation-fastapi>=0.46b0,<1.0.0",
 ]
 
 [build-system]