diff --git a/cnn_classification_service/main.py b/cnn_classification_service/main.py index 7285351..0789212 100644 --- a/cnn_classification_service/main.py +++ b/cnn_classification_service/main.py @@ -6,6 +6,7 @@ import pika import json from sentry_sdk.integrations.logging import LoggingIntegration +from sentry_sdk import start_transaction import sentry_sdk from magic_doer import MagicDoer @@ -18,7 +19,8 @@ def message_callback(channel, method, properties, body): logging.warning(f"Invalid message recieved: {e}") return - results = MagicDoer.run_everything(msg) # <- This is where the magic happens + with start_transaction(op="cnn-classification-service", name="classify-soundfile"): + results = MagicDoer.run_everything(msg) # <- This is where the magic happens channel.basic_publish( exchange=os.environ['PIKA_OUTPUT_EXCHANGE'],