Fix requirements.txt

Author: chatrtham

src/fastapi_app/__init__.py

@@ -3,9 +3,11 @@
 import os
 
 import azure.identity.aio
+import fastapi
+from azure.monitor.opentelemetry import configure_azure_monitor
 from dotenv import load_dotenv
 from environs import Env
-from fastapi import FastAPI
+from opentelemetry.instrumentation.sqlalchemy import SQLAlchemyInstrumentor
 
 from .globals import global_storage
 from .openai_clients import create_openai_chat_client
@@ -15,7 +17,7 @@
 
 
 @contextlib.asynccontextmanager
-async def lifespan(app: FastAPI):
+async def lifespan(app: fastapi.FastAPI):
     load_dotenv(override=True)
 
     azure_credential = None
@@ -40,6 +42,8 @@ async def lifespan(app: FastAPI):
     global_storage.openai_chat_client = openai_chat_client
     global_storage.openai_chat_model = openai_chat_model
 
+    if os.getenv("APPLICATIONINSIGHTS_CONNECTION_STRING"):
+        SQLAlchemyInstrumentor().instrument(engine=engine.sync_engine)
     yield
 
     await engine.dispose()
@@ -54,7 +58,14 @@ def create_app():
     else:
         logging.basicConfig(level=logging.WARNING)
 
-    app = FastAPI(docs_url="/docs", lifespan=lifespan)
+    # Turn off particularly noisy INFO level logs from Azure Core SDK:
+    logging.getLogger("azure.core.pipeline.policies.http_logging_policy").setLevel(logging.WARNING)
+
+    if os.getenv("APPLICATIONINSIGHTS_CONNECTION_STRING"):
+        logger.info("Configuring Azure Monitor")
+        configure_azure_monitor(logger_name="ragapp")
+
+    app = fastapi.FastAPI(docs_url="/docs", lifespan=lifespan)
 
     from . import api_routes  # noqa
     from . import frontend_routes  # noqa

src/requirements.txt

@@ -7,7 +7,9 @@ async-timeout==4.0.3
 asyncpg==0.29.0
 attrs==23.2.0
 azure-core==1.30.1
+azure-core-tracing-opentelemetry
 azure-identity==1.16.0
+azure-monitor-opentelemetry
 azure-monitor-opentelemetry-exporter==1.0.0b27
 build==1.2.1
 certifi==2024.2.2
@@ -24,7 +26,7 @@ email_validator==2.1.1
 environs==11.0.0
 fastapi==0.111.0
 fastapi-cli==0.0.4
--e git+https://github.com/chatrtham/rag-postgres-openai-python.git@7f1578f2d48381331570f68cc13167c3290ee097#egg=fastapi_app&subdirectory=src
+-e git+https://github.com/chatrtham/rag-postgres-openai-python@20d8fe30d4373e9290dbd18c012f6610393fd5e7#egg=fastapi_app&subdirectory=src
 filelock==3.14.0
 fixedint==0.1.6
 frozenlist==1.4.1
@@ -63,10 +65,18 @@ opentelemetry-exporter-otlp-proto-grpc==1.25.0
 opentelemetry-exporter-otlp-proto-http==1.25.0
 opentelemetry-instrumentation==0.46b0
 opentelemetry-instrumentation-asgi==0.46b0
+opentelemetry-instrumentation-dbapi==0.46b0
+opentelemetry-instrumentation-django==0.46b0
 opentelemetry-instrumentation-fastapi==0.46b0
+opentelemetry-instrumentation-flask==0.46b0
+opentelemetry-instrumentation-psycopg2==0.46b0
 opentelemetry-instrumentation-requests==0.46b0
 opentelemetry-instrumentation-sqlalchemy==0.46b0
+opentelemetry-instrumentation-urllib==0.46b0
+opentelemetry-instrumentation-urllib3==0.46b0
+opentelemetry-instrumentation-wsgi==0.46b0
 opentelemetry-proto==1.25.0
+opentelemetry-resource-detector-azure==0.1.5
 opentelemetry-sdk==1.25.0
 opentelemetry-semantic-conventions==0.46b0
 opentelemetry-util-http==0.46b0