Adding celery integration

openchatai · Feb 12, 2024 · 57b61c3 · 57b61c3
1 parent 1ab6837
commit 57b61c3
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/llm-server/utils/get_logger.py b/llm-server/utils/get_logger.py
@@ -1,10 +1,12 @@
 import os
 import logging
 import sentry_sdk
+from sentry_sdk.integrations.celery import CeleryIntegration
 
 sentry_sdk.init(
     traces_sample_rate=1.0,
     profiles_sample_rate=1.0,
+    integrations=[CeleryIntegration(propagate_traces=True)],
 )
 
 dsn = os.getenv("SENTRY_DSN")