Improve logfire

Author: nikhilwoodruff

src/policyengine_api/main.py

@@ -14,7 +14,8 @@
 console = Console()
 
 # Configure Logfire
-logfire.configure()
+logfire.configure(service_name="policyengine-api")
+logfire.instrument_httpx()
 
 
 @asynccontextmanager

src/policyengine_api/services/database.py

@@ -1,8 +1,10 @@
 from sqlmodel import Session, create_engine
+import logfire
 
 from policyengine_api.config.settings import settings
 
 engine = create_engine(settings.database_url, echo=settings.debug)
+logfire.instrument_sqlalchemy(engine=engine)
 
 
 def get_session():

src/policyengine_api/tasks/celery_app.py

@@ -1,7 +1,12 @@
 from celery import Celery
+import logfire
 
 from policyengine_api.config.settings import settings
 
+# Configure Logfire for worker
+logfire.configure(service_name="policyengine-worker")
+logfire.instrument_httpx()
+
 celery_app = Celery(
     "policyengine",
     broker=settings.celery_broker_url,