Add Sentry error tracking support

Author: boddumanohar

pyproject.toml

@@ -31,6 +31,7 @@ dependencies = [
     'python-json-logger>=4.0',
     'celery[sqlalchemy]',
     'asgiref',
+    'sentry-sdk',
 ]
 
 [tool.setuptools]

src/api/__init__.py

@@ -6,6 +6,10 @@
 from importlib.resources import files
 from typing import Any, Literal
 
+import sentry_sdk
+from sentry_sdk.integrations.fastapi import FastApiIntegration
+from sentry_sdk.integrations.starlette import StarletteIntegration
+
 from alembic import command
 from alembic.config import Config
 from fastapi import FastAPI
@@ -89,6 +93,13 @@ def _logging_config() -> dict[str, Any]:
 logging.config.dictConfig(_logging_config())
 logger = logging.getLogger(__name__)
 
+_sentry_dsn = get_settings().sentry_dsn
+if _sentry_dsn:
+    sentry_sdk.init(
+        dsn=_sentry_dsn,
+        integrations=[StarletteIntegration(), FastApiIntegration()],
+    )
+
 
 class _FastAPI(FastAPI):
     def openapi(self) -> dict[str, Any]:

src/api/settings.py

@@ -28,6 +28,7 @@ class Settings(BaseSettings):
     )
     pitr_wal_retention_days: int = 7
     deployment_password_secret: SecretBytes
+    sentry_dsn: str | None = None
 
 
 @lru_cache

src/worker/__init__.py

@@ -1,15 +1,21 @@
+import sentry_sdk
 from celery import Celery
 from pydantic_settings import BaseSettings, SettingsConfigDict
+from sentry_sdk.integrations.celery import CeleryIntegration
 
 
 class Settings(BaseSettings):
     model_config = SettingsConfigDict(env_prefix="vela_", case_sensitive=False)
     broker_url: str
     result_backend: str
+    sentry_dsn: str | None = None
 
 
 _settings = Settings()  # type: ignore[call-arg]
 
+if _settings.sentry_dsn:
+    sentry_sdk.init(dsn=_settings.sentry_dsn, integrations=[CeleryIntegration()])
+
 app = Celery("vela", broker=_settings.broker_url, backend=_settings.result_backend)
 
 # Persist task name, args, and kwargs in celery_taskmeta so AsyncResult can