diff --git a/beagle/settings.py b/beagle/settings.py index 0bd3deab7..d13bf2d3e 100644 --- a/beagle/settings.py +++ b/beagle/settings.py @@ -60,7 +60,7 @@ "drf_multiple_model", "drf_yasg", "advanced_filters", - "ddtrace.contrib.django" + "ddtrace.contrib.django", ] diff --git a/beagle_etl/tasks.py b/beagle_etl/tasks.py index 047bb12be..c5707dc2e 100644 --- a/beagle_etl/tasks.py +++ b/beagle_etl/tasks.py @@ -63,8 +63,8 @@ def process_smile_events(): if message.request_id in update_requests: update_requests.remove(message.request_id) current_span = tracer.current_span() - request_id=message.request_id - current_span.set_tag("request.id", request_id) + request_id = message.request_id + current_span.set_tag("request.id", request_id) logger.info(f"New request: {message.request_id}") new_request.delay(str(message.id)) diff --git a/runner/operator/operator.py b/runner/operator/operator.py index 2cdf609c1..554099807 100644 --- a/runner/operator/operator.py +++ b/runner/operator/operator.py @@ -60,6 +60,7 @@ def get_output_metadata(self): :return: dict """ return {} + def failed_to_create_job(self, error): operator_error = OperatorErrorSerializer( data={"operator_name": self.model.slug, "request_id": self.request_id, "error": error} @@ -81,6 +82,7 @@ def ready_job(self, pipeline, tempo_inputs, job): current_span = tracer.current_span() cmo_request_id = self.request_id span.set_tag("request.id", cmo_request_id) + def on_job_fail(self, run): pass diff --git a/runner/tasks.py b/runner/tasks.py index 72fd12ac4..9799140c7 100644 --- a/runner/tasks.py +++ b/runner/tasks.py @@ -43,7 +43,6 @@ from ddtrace import tracer - logger = logging.getLogger(__name__)