feat(backend): enable tracing for task workers

This was previously enabled on git agents but got removed during the
transition to task workers.
Prefect 3.1 also adds instrumentation to workflow runs.

Signed-off-by: Fatih Acar <[email protected]>

backend/infrahub/workers/infrahub_async.py

@@ -32,6 +32,7 @@
 from infrahub.services.adapters.workflow import InfrahubWorkflow
 from infrahub.services.adapters.workflow.local import WorkflowLocalExecution
 from infrahub.services.adapters.workflow.worker import WorkflowWorkerExecution
+from infrahub.trace import configure_trace
 from infrahub.workers.utils import inject_service_parameter, load_flow_function
 from infrahub.workflows.models import TASK_RESULT_STORAGE_NAME
 
@@ -90,6 +91,16 @@ async def setup(
 
         self._init_logger()
 
+        # Initialize trace
+        if config.SETTINGS.trace.enable:
+            configure_trace(
+                service="infrahub-task-worker",
+                version=infrahub_version,
+                exporter_type=config.SETTINGS.trace.exporter_type,
+                exporter_endpoint=config.SETTINGS.trace.exporter_endpoint,
+                exporter_protocol=config.SETTINGS.trace.exporter_protocol,
+            )
+
         # Start metric endpoint
         if metric_port is None or metric_port != 0:
             metric_port = metric_port or int(os.environ.get("INFRAHUB_METRICS_PORT", 8000))