add sentry as requirement for logging

lib/http.py

@@ -12,6 +12,7 @@
 from lib.queue.worker import QueueWorker
 from lib.logger import logger
 from lib import schemas
+from lib.sentry import sentry_sdk
 
 app = FastAPI()
 

lib/sentry.py

@@ -0,0 +1,9 @@
+import os
+import sentry_sdk
+from lib.helpers import get_environment_setting
+
+sentry_sdk.init(
+    dsn=get_environment_setting('sentry_sdk_dsn'),
+    environment=get_environment_setting("DEPLOY_ENV"),
+    traces_sample_rate=1.0,
+)

requirements.txt

@@ -10,4 +10,5 @@ httpx==0.23.1
 huggingface-hub==0.11.0
 fasttext==0.9.2
 requests==2.31.0
-pytest==7.4.0
+pytest==7.4.0
+sentry-sdk==1.30.0

run_processor.py

@@ -4,6 +4,7 @@
 from lib.queue.processor import QueueProcessor
 from lib.model.model import Model
 from lib.logger import logger
+from lib.sentry import sentry_sdk
 queue = QueueProcessor.create()
 
 logger.info("Beginning callback loop...")

run_worker.py

@@ -4,6 +4,7 @@
 from lib.queue.worker import QueueWorker
 from lib.model.model import Model
 from lib.logger import logger
+from lib.sentry import sentry_sdk
 queue = QueueWorker.create()
 
 model = Model.create()